/afs/in2p3.fr/throng/virgo/virgoDev/GWOLLUM/v1r3/src/Inject/InjTre.h

Go to the documentation of this file.

//  Author : florent robinet (LAL - Orsay): robinet@lal.in2p3.fr
#ifndef __InjTre__
#define __InjTre__


#include "CUtils.h"
#include "TFile.h"
#include "TTree.h"

#define NINJTYPE 2

using namespace std;

class InjTre {
  
 public:
  
  InjTre(const int aInjType, const int aVerbose=0);// constructor
  virtual ~InjTre();        // destructor

  //
  int AddGauss(const double aRa, 
           const double aDec, 
           const double aTime, 
           const double aEccentricity, 
           const double aPolarization, 
           const double aAmplitude, 
           const double aSigma);// add Gauss injection
  int AddSineGauss(const double aRa, 
           const double aDec, 
           const double aTime, 
           const double aEccentricity, 
           const double aPolarization, 
           const double aAmplitude, 
           const double aSigma, 
           const double aFrequency);// add Sine-Gauss injection
  inline void ResetTree(void){ InjTree->Reset(); };// reset injection tree

  // OUTPUT
  bool Write(const string aRootFileName="./myinjections.root");
  inline TTree* GetTree(void){ return InjTree; };

 protected:
  
  // INPUT
  bool status;               // status of the Inject object
  int Verbose;               // verbosity level
  int InjType;               // injection type
  
  // INJECTIONS
  double inj_ra, inj_dec, inj_psi, inj_time, inj_amp, inj_f0, inj_sigma, inj_ecc; 
  TTree *InjTree;            // injection trees

  ClassDef(InjTre,0)
};

#endif

---