InjGen.h

Go to the documentation of this file.

 
#ifndef __InjGen__
#define __InjGen__
 
#include <TMath.h>
#include <TRandom3.h>
#include "Segments.h"
#include "InjTre.h"
 
using namespace std;
 
class InjGen: public InjTre{
  
public:
  
  InjGen(Segments *aInSegments, const injtype aInjType, const string aInjName="gwl_inj", const unsigned int aVerbose=0);
  
  virtual ~InjGen();
  unsigned int GenerateInjections(void);
  
  inline void SetInjectionTimeSpacing(const double aSpacing){ InjSpacing = aSpacing; };
  
  inline void SetInjectionTimeJitter(const double aJitter){ InjJitterMax = aJitter; };
  
  bool SetAmplitudeDistribution(const string aDistribution="LOG");
  
  inline void SetAmplitudeRange(const double aMin=1e-22, const double aMax=5e-19){
    ampMin = aMin;
    ampMax = aMax;
  };
 
  bool SetSigmaDistribution(const string aDistribution="LOG");
 
  inline void SetSigmaRange(const double aMin=0.01, const double aMax=1.0){
    sigmaMin = aMin;
    sigmaMax = aMax;
  };
 
  bool SetFrequencyDistribution(const string aDistribution="LOG");
 
  inline void SetFrequencyRange(const double aMin=30.0, const double aMax=2048.0){
    f0Min = aMin;
    f0Max = aMax;
  };
 
  bool SetEccentricityDistribution(const string aDistribution="UNIFORM");
 
  inline void SetEccentricityRange(const double aMin=0.0, const double aMax=1.0){
    eccMin = aMin;
    eccMax = aMax;
  };
 
  bool SetPolarizationDistribution(const string aDistribution="UNIFORM");
 
  inline void SetPolarizationRange(const double aMin=0.0, const double aMax=2.0*TMath::Pi()){
    psiMin = aMin;
    psiMax = aMax;
  };
  
 private:
  
  TRandom3 *rand;                
  Segments *InSegments;          
  double InjSpacing;             
  double InjJitterMax;           
 
  // PARAMETERS
  double ampMin;                 
  double ampMax;                 
  unsigned int ampDistribution;  
  double sigmaMin;               
  double sigmaMax;               
  unsigned int sigmaDistribution;
  double f0Min;                  
  double f0Max;                  
  unsigned int f0Distribution;   
  double eccMin;                 
  double eccMax;                 
  unsigned int eccDistribution;  
  double psiMin;                 
  double psiMax;                 
  unsigned int psiDistribution;  
  
  double GetAmplitude(void);
  
  double GetSigma(void);
 
  double GetFrequency(void);
 
  double GetEccentricity(void);
 
  double GetPolarization(void);
  
  
  ClassDef(InjGen,0)
};
 
#endif