Update timestamps for new AMANDA simulation (17/02/2015)
[u/mrichter/AliRoot.git] / EVGEN / AliGenMC.h
1 #ifndef ALIGENMC_H
2 #define ALIGENMC_H
3 /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
4  * See cxx source for full Copyright notice                               */
5
6 /* $Id$ */
7
8
9 // Base class for generators using external MC generators.
10 // For example AliGenPythia using Pythia.
11 // Provides basic functionality: setting of kinematic cuts on 
12 // decay products and particle selection.
13 // andreas.morsch@cern.ch
14
15 #include<TClonesArray.h>
16 #include <TArrayI.h>   
17 #include <TString.h>
18 class TParticle;
19
20 class AliGeometry;
21 class AliGenEventHeader;
22 #include "AliGenerator.h"
23 #include "AliDecayer.h"
24
25 class AliGenMC : public AliGenerator
26 {
27  public:
28     AliGenMC();
29     AliGenMC(Int_t npart);
30     virtual ~AliGenMC();
31     virtual void Init();
32     virtual void SetForceDecay(Decay_t decay = kAll) {fForceDecay = decay;}
33     virtual void SetCutOnChild(Int_t flag = 0) {fCutOnChild = flag;}
34     virtual void SetChildMomentumRange(Float_t pmin = 0, Float_t pmax = 1.e10)
35         {fChildPMin = pmin; fChildPMax = pmax;}
36     virtual void SetChildPtRange(Float_t ptmin = 0, Float_t ptmax = 20.)
37         {fChildPtMin = ptmin; fChildPtMax = ptmax;}
38     virtual void SetChildPhiRange(Float_t phimin = 0., Float_t phimax = 360.)
39         {fChildPhiMin = TMath::Pi()*phimin/180;
40         fChildPhiMax  = TMath::Pi()*phimax/180;}
41     virtual void SetChildThetaRange(Float_t thetamin = 0, Float_t thetamax = 180)
42         {fChildThetaMin = TMath::Pi()*thetamin/180;
43         fChildThetaMax  = TMath::Pi()*thetamax/180;}
44     virtual void SetChildYRange(Float_t ymin = -12, Float_t ymax = 12)
45         {fChildYMin = ymin;
46         fChildYMax  = ymax;}
47     virtual void SetMaximumLifetime(Float_t time = 1.e-15) {fMaxLifeTime = time;}
48    
49     virtual void SetGeometryAcceptance(AliGeometry * GeometryAcceptance=0) {fGeometryAcceptance = GeometryAcceptance;}
50
51     virtual void SetPdgCodeParticleforAcceptanceCut(Int_t PdgCodeParticleforAcceptanceCut=0) {fPdgCodeParticleforAcceptanceCut = PdgCodeParticleforAcceptanceCut;}
52
53     virtual void SetNumberOfAcceptedParticles(Int_t NumberOfAcceptedParticles=2) {fNumberOfAcceptedParticles = NumberOfAcceptedParticles;}
54     
55     virtual Bool_t CheckAcceptanceGeometry(Int_t np, TClonesArray* particles);
56     virtual void   SetProjectile(TString proj="P", Int_t a = 1, Int_t z = 1)
57         {fProjectile = proj; fAProjectile = a; fZProjectile = z;}    
58     virtual void   SetTarget(TString tar="P", Int_t a = 1, Int_t z = 1)
59         {fTarget = tar; fATarget = a; fZTarget = z;}
60     virtual void   SetCrossingAngle(Float_t phiX, Float_t phiY) {fXingAngleX = phiX; fXingAngleY = phiY;}
61     virtual void Boost();
62     virtual void SetDyBoost(Double_t dyBoost = 0.) {fDyBoost = dyBoost;}
63     virtual void BeamCrossAngle();
64     virtual void AddHeader(AliGenEventHeader* header);
65
66  protected:
67     // check if particle is selected as parent particle
68     Bool_t ParentSelected(Int_t ip) const;
69     // check if particle is selected as child particle
70     Bool_t ChildSelected(Int_t ip) const;
71     // all kinematic selection cuts go here 
72     Bool_t KinematicSelection(const TParticle *particle, Int_t flag) const;
73     Int_t  CheckPDGCode(Int_t pdgcode) const;
74
75  protected:
76     TClonesArray fParticles;     //!Particle  List
77     TArrayI      fParentSelect;  //!Parent particles to be selected 
78     TArrayI      fChildSelect;   //!Decay products to be selected
79     Int_t        fCutOnChild;    // Cuts on decay products (children)  are enabled/disabled
80     Float_t      fChildPtMin;    // Children minimum pT
81     Float_t      fChildPtMax;    // Children maximum pT
82     Float_t      fChildPMin;     // Children minimum p
83     Float_t      fChildPMax;     // Children maximum p
84     Float_t      fChildPhiMin;   // Children minimum phi
85     Float_t      fChildPhiMax;   // Children maximum phi
86     Float_t      fChildThetaMin; // Children minimum theta
87     Float_t      fChildThetaMax; // Children maximum theta
88     Float_t      fChildYMin;     // Children minimum y
89     Float_t      fChildYMax;     // Children maximum y
90     Float_t      fXingAngleX;    // Crossing angle X
91     Float_t      fXingAngleY;    // Crossing angle Y    
92     Decay_t      fForceDecay;    // Decay channel forced
93     Float_t      fMaxLifeTime;   // Maximum lifetime for unstable particles
94     Double_t     fDyBoost;       // dy for boost into lab frame
95     AliGeometry* fGeometryAcceptance; // Geometry to which particles must be simulated
96     Int_t        fPdgCodeParticleforAcceptanceCut;  // Abs(PDG Code) of the particle to which the GeometryAcceptance must be applied
97     Int_t        fNumberOfAcceptedParticles;  // Number of accepted particles in GeometryAcceptance with the right Abs(PdgCode) 
98     Int_t        fNprimaries;                 // Number of produced and stored particles
99  
100  private:
101     AliGenMC(const AliGenMC &MC);
102     AliGenMC & operator=(const AliGenMC & rhs);
103     
104     ClassDef(AliGenMC,7)       // AliGenerator implementation for generators using MC methods
105 };
106 #endif
107
108
109
110
111