// Author: Akira Okumura <mailto:oxon@mac.com>
/******************************************************************************
 * Copyright (C) 2006-, Akira Okumura                                         *
 * All rights reserved.                                                       *
 *****************************************************************************/

#ifndef A_OPTICS_MANAGER_H
#define A_OPTICS_MANAGER_H

#include "TGeoManager.h"
#include "TMath.h"

#include "AFocalSurface.h"
#include "ALens.h"
#include "AMirror.h"
#include "AObscuration.h"
#include "AOpticalComponent.h"
#include "ARayArray.h"

///////////////////////////////////////////////////////////////////////////////
//
// AOpticsManager
//
// Manager of optics
//
///////////////////////////////////////////////////////////////////////////////

class AOpticsManager : public TGeoManager {
 private:
  Int_t fLimit;                      // Maximum number of crossing calculations
  Bool_t fDisableFresnelReflection;  // disable Fresnel reflection
  TClass* fClassList[5];

  static void* Thread(void* args);

  void DoFresnel(Double_t n1, Double_t n2, Double_t k2, ARay& ray,
                 TGeoNavigator* nav, TGeoNode* currentNode, TGeoNode* nextNode);
  void DoReflection(Double_t n1, ARay& ray, TGeoNavigator* nav,
                    TGeoNode* currentNode, TGeoNode* nextNode,
                    TVector3* normal = 0);
  TVector3 GetFacetNormal(TGeoNavigator* nav, TGeoNode* currentNode,
                          TGeoNode* nextNode);

 public:
  enum {
    kLens = 0,
    kObs = 1,
    kMirror = 2,
    kFocus = 3,
    kOpt = 4,
    kOther = 5,
    kNull = 6
  };

  AOpticsManager();
  AOpticsManager(const char* name, const char* title);
  virtual ~AOpticsManager();

  static Double_t km() { return 1e3 * m(); };
  static Double_t m() { return 1e2 * cm(); };
  static Double_t cm() { return 1; };
  static Double_t mm() { return 1e-3 * m(); };
  static Double_t um() { return 1e-6 * m(); };
  static Double_t nm() { return 1e-9 * m(); };
  static Double_t inch() { return 2.54 * cm(); };
  static Double_t s() { return 1.; };
  static Double_t ms() { return 1e-3 * s(); };
  static Double_t us() { return 1e-6 * s(); };
  static Double_t ns() { return 1e-9 * s(); };
  static Double_t deg() { return TMath::DegToRad(); };
  static Double_t rad() { return 1.; }

  void DisableFresnelReflection(Bool_t disable) {
    fDisableFresnelReflection = disable;
  }
  Bool_t IsFocalSurface(TGeoNode* node) const {
    return node ? node->GetVolume()->IsA() == fClassList[kFocus] : kFALSE;
  };
  Bool_t IsLens(TGeoNode* node) const {
    return node ? node->GetVolume()->IsA() == fClassList[kLens] : kFALSE;
  };
  Bool_t IsMirror(TGeoNode* node) const {
    return node ? node->GetVolume()->IsA() == fClassList[kMirror] : kFALSE;
  };
  Bool_t IsObscuration(TGeoNode* node) const {
    return node ? node->GetVolume()->IsA() == fClassList[kObs] : kFALSE;
  };
  Bool_t IsOpticalComponent(TGeoNode* node) const {
    return node ? node->GetVolume()->IsA() == fClassList[kOpt] : kFALSE;
  };
  void SetLimit(Int_t n);
  void TraceNonSequential(ARay& ray);
  void TraceNonSequential(ARay* ray) {
    if (ray) TraceNonSequential(*ray);
  }
  void TraceNonSequential(ARayArray& array);
  void TraceNonSequential(ARayArray* array) {
    if (array) TraceNonSequential(*array);
  }
  void TraceNonSequential(TObjArray* array);

  ClassDef(AOpticsManager, 1)
};

#endif  // A_OPTICS_MANAGER_H