Source.h

Go to the documentation of this file.

#ifndef __Source__
#define __Source__
 
#include <TFile.h>
#include <TTree.h>
#include "SpecTime.h"
 
using namespace std;
 
class Source: public SpecTime{
  
 public:
  
 
  Source(const int aNt, const double aTimeMin, const double aTimeMax, const bool aUseTimeLog,
     const int aNe, const double aEnergyMin, const double aEnergyMax, const bool aUseEnergyLog);
 
 
  Source(TH2D* aFlux, TH2D* aRate=NULL);
 
 
  virtual ~Source(void);  
 
  enum sourcetype{
    type_photon,            
    type_cosmic,            
    type_photon_background  
  };
 
 
  int GenerateAstroParticles(const int aN, const ULong64_t aTime, const UInt_t aDuration);
 
 
  void SetName(const string aName);
 
 
  inline string GetName(void){ return src_name; };
 
 
  inline void SetType(const Source::sourcetype aType){ src_type = (int)aType; };
 
 
  void SetPosition(const double aThetaPrime, const double aPhiPrime, const double aConeAngle);
  
 
  inline double GetThetaPrime(const ULong64_t aTime){
    return src_thetap->Eval((double)aTime);
  };
 
 
  inline double GetPhiPrime(const ULong64_t aTime){
    return src_phip->Eval((double)aTime);
  };
 
 
  inline int GetType(void){ return src_type; };
  
 
  inline double GetAstroPhip(void){ return astro_phip; };
  
 
  inline double GetAstroThetap(void){ return astro_thetap; };
  
 
  inline ULong64_t GetAstroTime(void){ return astro_time; };
  
 
  inline double GetAstroEnergy(void){ return astro_energy; };
  
 
  void TreeInit(void);
 
 
  inline void TreeReset(void){
    if(src_tree!=NULL)   src_tree->Reset();
    if(astro_tree!=NULL) astro_tree->Reset();
    return;
  };
 
 
  inline int TreeWrite(void){
    int nb=0;
    if(src_tree!=NULL)   nb+=src_tree->Write();
    if(astro_tree!=NULL) nb+=astro_tree->Write();
    return nb;
  };
  
 
  inline void TreeDelete(void){
    if(src_tree!=NULL)   delete src_tree;
    if(astro_tree!=NULL) delete astro_tree;
    src_tree=NULL;
    astro_tree=NULL;
    return;
  };
 
 
  inline int TreeAstroGetN(void){
    if(astro_tree!=NULL) return astro_tree->GetEntries();
    return 0;
  };
 
 
  inline int TreeAstroLoadEntry(const int aEntryIndex){
    if(astro_tree!=NULL) return astro_tree->GetEntry(aEntryIndex);
    return -1;
  };
 
 private:
 
  // Source definition
  string src_name;             
  TTree *src_tree;             
  int src_type;                
  TGraph *src_thetap;          
  TGraph *src_phip;            
  TGraph *src_cosconeangle;    
 
  // Astroparticles
  TTree *astro_tree;            
  ULong64_t astro_time;         
  double astro_thetap;          
  double astro_phip;            
  double astro_energy;          
 
  // Rotation matrix
  TGraph *src_R[3][3];          
 
 
  void ComputeR(void);
 
  void CommonConstructor(void);
 
 
  ClassDef(Source,0)
};
 
#endif