FDsimG4Write2ROOT.hh

Go to the documentation of this file.

#ifndef _TelescopeSimulatorLX_FDsimG4Write2ROOT_h_
#define _TelescopeSimulatorLX_FDsimG4Write2ROOT_h_

#include <string>
#include <utl/CoordinateSystemPtr.h>

class TFile;
class TTree;
class TDirectory;

namespace utl {
  class Photon;
}

namespace TelescopeSimulatorLX {
  
  class FDsimG4StoreOpticalHit;

  class FDsimG4Write2ROOT {
    
  public:
    FDsimG4Write2ROOT ();
    FDsimG4Write2ROOT (const std::string& RootFilename);
    ~FDsimG4Write2ROOT ();
    
    void Open(const std::string& RootFilename);
    void SaveAndClose();
    void CreateTree();
    void FillTree();
    void WriteIt();
    void Clean();

    void SetTelCS(utl::CoordinateSystemPtr telCS) {ftelCS=telCS;};
    void SetEyeCS(utl::CoordinateSystemPtr eyeCS) {feyeCS=eyeCS;};

    void SetG4FilterPhoton(FDsimG4StoreOpticalHit* G4_photon);
    void SetG4LensPhoton(FDsimG4StoreOpticalHit* G4_photon);
    void SetG4MirrorPhoton(FDsimG4StoreOpticalHit* G4_photon);
    void SetG4MercedesPhoton(FDsimG4StoreOpticalHit* G4_photon);
    void SetG4FSPhoton(FDsimG4StoreOpticalHit* G4_photon);
    void SetG4Photon(FDsimG4StoreOpticalHit* G4_photon);
    void SetINPhoton(FDsimG4StoreOpticalHit* G4_photon);
    void SetINPhoton(utl::Photon IN_photon , 
                     utl::CoordinateSystemPtr telCS,
                     utl::CoordinateSystemPtr eyeCS);

  private:
    utl::CoordinateSystemPtr ftelCS;
    utl::CoordinateSystemPtr feyeCS;

    TFile *fRootFile;
    TTree *fTreePhotons;
    TDirectory *fSaveDir;

    struct G4_info{
      double x;
      double y;
      double z;
    } fG4Filter_info,fG4Lens_info,fG4Mirror_info,fG4Mercedes_info;

    struct G4_infoFS{
      double x;
      double y;
      double z;
      double alfa;
      double beta;
      double th;
      double phi;
    } fG4FS_info;
    
    struct G4_info2{
      double x;
      double y;
      double z;
      double alfa;
      double beta;
      double E;
      double w;
      double t;
      double LocX;
      double LocY;
      double LocZ;
      double PixX;
      double PixY;
      double PixZ;
      double HitID;
      double PMTid;
    } fG4_info;

    struct IN_info{
      double x;
      double y;
      double z;
      double DX;
      double DY;
      double DZ;  
      double th;
      double phi;
      double wantedPix;
      double wantedTh;
      double wantedPhi;
    } fIN_info;  
    
    struct IN_infoExp{
      double x;
      double y;
      double z;
      double alfa;
      double beta;
      double DX;
      double DY;
      double DZ;  
      double th;
      double phi;
    } fINExp_info;  
    
    struct kn_info{
      int G4Filter;
      int G4Lens;
      int G4Mirror;
      int G4Mercedes;
      int G4FS;
      int G4;
    } fn_info;
    
};
  
} // TelescopeSimulatorLX

#endif // _TelescopeSimulatorLX_FDsimG4Write2ROOT_h_