Update timestamps for new AMANDA simulation (17/02/2015)

[u/mrichter/AliRoot.git] / EVGEN / AliGenParam.h

diff --git a/EVGEN/AliGenParam.h b/EVGEN/AliGenParam.h
index bc9a35a891d7b953259e5dd4e0108247903a263d..bbd61cebc653437502af3a4f6ab8b0c4dc80c286 100644 (file)
--- a/EVGEN/AliGenParam.h
+++ b/EVGEN/AliGenParam.h
@@ -6,7 +6,7 @@
 /* $Id$ */
 
 //
-// Class to generate particles from using paramtrized pT and y distributions.
+// Class to generate particles from using parametrized pT and y distributions.
 // Distributions are obtained from pointer to object of type AliGenLib.
 // (For example AliGenMUONlib)
 //
@@ -24,33 +24,57 @@ typedef enum { kAnalog, kNonAnalog} Weighting_t;
 //-------------------------------------------------------------
 class AliGenParam : public AliGenMC
 {
- public:
+public:
     AliGenParam();
-    AliGenParam(Int_t npart, AliGenLib * Library, Int_t param, char* tname = 0);
-    AliGenParam(Int_t npart, Int_t param, char* tname = 0);
+    AliGenParam(Int_t npart, const AliGenLib * Library, Int_t param,   const char*  tname = 0);
+    AliGenParam(Int_t npart, Int_t param, const char* tname = 0, const char*  name  = 0);
     AliGenParam(Int_t npart, Int_t param,
-               Double_t (*PtPara)(Double_t*, Double_t*),
-               Double_t (*YPara )(Double_t*, Double_t*),
+               Double_t (*PtPara)(const Double_t*, const Double_t*),
+               Double_t (*YPara )(const Double_t*, const Double_t*),
+             Double_t (*V2Para)(const Double_t*, const Double_t*),
                Int_t    (*IpPara)(TRandom*)           );
-    AliGenParam(const AliGenParam &Param);
      
     virtual ~AliGenParam();
+    virtual void GenerateN(Int_t ntimes);
     virtual void Generate();
     virtual void Init();
     // select particle type
     virtual void SetParam(Int_t param) {fParam = param;}
+    //Setting the flag for Background transportation while using SetForceDecay()
+    void SetSelectAll(Bool_t selectall) {fSelectAll = selectall;}
     // force decay type
     virtual void SetWeighting(Weighting_t flag = kAnalog) {fAnalog = flag;}    
     virtual void SetDeltaPt(Float_t delta=0.01) {fDeltaPt = delta;}
     virtual void SetDecayer(AliDecayer* decayer) {fDecayer = decayer;}
-    virtual void Draw();
-    AliGenParam & operator=(const AliGenParam & rhs);
- protected:
-    Double_t (*fPtParaFunc)(Double_t*, Double_t*); //! Pointer to Pt parametrisation function
-    Double_t (*fYParaFunc )(Double_t*, Double_t*); //! Pointer to Y parametrisation function
+    virtual void SetForceGammaConversion(Bool_t force=kTRUE) {fForceConv = force;}
+    virtual void SetKeepParent(Bool_t keep=kTRUE){fKeepParent= keep;} //Store parent even if it does not have childs within cuts
+    virtual void SetKeepIfOneChildSelected(Bool_t keep=kTRUE){fKeepIfOneChildSelected = keep;} //Accept parent and child even if other children are not within cut.
+    
+    virtual void Draw(const char * opt);
+    TF1 *  GetPt() { return fPtPara;}
+    TF1 *  GetY() {return fYPara;}
+    Float_t GetRelativeArea(Float_t ptMin, Float_t ptMax, Float_t yMin, Float_t yMax, Float_t phiMin, Float_t phiMax);
+
+    static TVector3 OrthogonalVector(TVector3 &inVec);
+    static void RotateVector(Double_t *pin, Double_t *pout, Double_t costheta, Double_t sintheta,
+                          Double_t cosphi, Double_t sinphi);
+    static double ScreenFunction1(double d);
+    static double ScreenFunction2(double d);
+    double RandomEnergyFraction(double Z, double E);
+    double RandomPolarAngle();
+    double RandomMass(Double_t mh);
+    Int_t VirtualGammaPairProduction(TClonesArray *particles, Int_t nPart);
+    Int_t ForceGammaConversion(TClonesArray *particles, Int_t nPart);
+  
+protected:
+    Double_t (*fPtParaFunc)(const Double_t*, const Double_t*); //! Pointer to Pt parametrisation function
+    Double_t (*fYParaFunc )(const Double_t*, const Double_t*); //! Pointer to Y parametrisation function
     Int_t    (*fIpParaFunc )(TRandom*);    //! Pointer to particle type parametrisation function
+  Double_t (*fV2ParaFunc )(const Double_t*, const Double_t*);//! Pointer to V2 parametrisation function
     TF1* fPtPara;              // Transverse momentum parameterisation
     TF1* fYPara;               // Rapidity parameterisation
+  TF1*        fV2Para;       // v2 parametrization
+  TF1*        fdNdPhi;       // Phi distribution depending on v2
     Int_t       fParam;        // Parameterisation type 
     Float_t     fdNdy0;        // central multiplicity per event
     Float_t     fYWgt;         // Y-weight
@@ -58,8 +82,17 @@ class AliGenParam : public AliGenMC
     Float_t     fBias;         // Biasing factor
     Int_t       fTrials;       // Number of trials
     Float_t     fDeltaPt;      // pT sampling in steps of fDeltaPt
+    Bool_t      fSelectAll;    // Flag for transportation of Background while using SetForceDecay()
     AliDecayer  *fDecayer;     // ! Pointer to pythia object for decays
-  ClassDef(AliGenParam,1) // Generator using parameterised pt- and y-distribution
+    Bool_t      fForceConv;    //
+    Bool_t      fKeepParent;   //  Store parent even if it does not have childs within cuts
+    Bool_t      fKeepIfOneChildSelected; //Accept parent and child even if other children are not within cut.
+
+private:
+    AliGenParam(const AliGenParam &Param);
+    AliGenParam & operator=(const AliGenParam & rhs);
+
+    ClassDef(AliGenParam, 2) // Generator using parameterised pt- and y-distribution
 };
 #endif