00001
00002
00004 #ifndef __InjTre__
00005 #define __InjTre__
00006
00007
00008 #include "CUtils.h"
00009 #include "TFile.h"
00010 #include "TTree.h"
00011
00012 using namespace std;
00013
00018 class InjTre {
00019
00020 public:
00021
00035 InjTre(const string aInjType, const int aVerbose=0);
00036
00040 virtual ~InjTre();
00056 int AddGauss(const double aRa,
00057 const double aDec,
00058 const double aTime,
00059 const double aEccentricity,
00060 const double aPolarization,
00061 const double aAmplitude,
00062 const double aSigma);
00063
00076 int AddSineGauss(const double aRa,
00077 const double aDec,
00078 const double aTime,
00079 const double aEccentricity,
00080 const double aPolarization,
00081 const double aAmplitude,
00082 const double aSigma,
00083 const double aFrequency);
00084
00088 inline int GetNInjections(void){ return InjTree->GetEntries(); };
00089
00093 inline string GetInjectionType(void){ return (string)InjTree->GetTitle(); };
00094
00098 inline void Reset(void){ return InjTree->Reset(); };
00099
00104 bool Write(const string aRootFileName="./myinjections.root");
00105
00106
00107 protected:
00108
00109
00110 int Verbose;
00111
00112 private:
00113
00114 TTree *InjTree;
00115 double inj_ra,
00116 inj_dec,
00117 inj_psi,
00118 inj_time,
00119 inj_amp,
00120 inj_f0,
00121 inj_sigma,
00122 inj_ecc;
00123
00124 ClassDef(InjTre,0)
00125 };
00126
00127 #endif
00128
00129