Change Mult binning scheme
[u/mrichter/AliRoot.git] / PWGDQ / dielectron / AliDielectronSignalMC.h
CommitLineData
ba15fdfb 1#ifndef ALIDIELECTRONSIGNALMC_H
2#define ALIDIELECTRONSIGNALMC_H
3
4#include <TNamed.h>
5
6
7/*
5720c765 8 Ionut Cristian Arsene, iarsene@cern.ch
9 */
10
11/*
ba15fdfb 12 Monte Carlo signal definition:
13 Leg #1 <-- Mother #1 <-- Grandmother #1
14 |
15 Leg #2 <-- Mother #2 <-- Grandmother #2
5720c765 16
17 For every leg, mother or grand-mother, a PDG code and a source can be specified.
18
19 1.) For the PDG codes, the PYTHIA standard is used.
20 A few non-existent PYTHIA codes are used to select more than one PYTHIA code. All these are described below
21 and implemented in AliDielectronMC::ComparePDG() function:
22 0 - default, accepts all PYTHIA codes
23 100 - light unflavoured mesons in the code range 100-199
24 200 - --"-- 200-299
25 300 - strange mesons in the code range 300-399
26 400 - charmed mesons in the code range 400-499
27 401 - open charm mesons (all D and D* mesons) 400-439
28 402 - open charm mesons and baryons together 400-439, 4000-4399
29 403 - all charm hadrons (mesons and baryons) 400-499, 4000-4999
30 500 - beauty mesons in the code range 500-599
31 501 - open beauty mesons 500-549
32 502 - open beauty mesons and baryons 500-549, 5000-5499
33 503 - all beauty hadrons 500-599, 5000-5999
34 1000 - light unflavoured baryons in the code range 1000-1999
35 2000 - --"-- 2000-2999
36 3000 - strange baryons in the code range 3000-3999
37 4000 - charmed baryons in the code range 4000-4999
38 4001 - open charm baryons 4000-4399
39 5000 - beauty baryons in the code range 5000-5999
40 5001 - open beauty baryons 5000-5499
41
42 2.) If the exclusion flags are turned ON then the PDG codes required and the conventional codes described above
43 are used to exclude the selected particles.
44
45 3.) If the selection of both charges is switched ON then the PDG codes act on both particles and anti-particles.
46
47 4.) Particles sources implemented:
ba15fdfb 48 1. Primary - particle originating in the physics event
5720c765 49 2. FinalState- stable(final state) particles which reach the detector -> according to AliStack::IsPhysicalPrimary()
50 3. Direct - primary particle which has no mother (e.g. J/psi's added to pythia MC events via generator cocktails,
51 particles generated in a sudden freeze-out in thermal models, initial state particles)
52 4. Secondary - particle created during the GEANT propagation due to interaction of final state primaries with the material
53
54 5.) The 2 legs can originate from the same or different mother particles. This can be specified via the SetMotherRelation()
55 method call.
56
57 6.) The filling of the pure MC step can be switched on using SetFillPureMCStep() method call. This should be used
58 with care since at the pure MC information level there is no cut applied and for abundant particles the combinatorics
59 can be very high.
ba15fdfb 60*/
61
62
63//__________________________________________________________________
64class AliDielectronSignalMC : public TNamed {
65
66 public:
67 enum EBranchRelation {kUndefined=0, kSame, kDifferent};
1ae2dca4 68 enum ESource {kDontCare=0, kPrimary, kFinalState, kDirect, kSecondary, kNoCocktail};
ef37a5a8 69 enum EJpsiRadiativ {kAll=0, kIsRadiative, kIsNotRadiative};
ba15fdfb 70
71 AliDielectronSignalMC();
72 AliDielectronSignalMC(const Char_t* name, const Char_t* title);
73 virtual ~AliDielectronSignalMC();
74
5720c765 75 void SetLegPDGs(Int_t pdg1, Int_t pdg2, Bool_t exclude1=kFALSE, Bool_t exclude2=kFALSE)
76 {fLeg1 = pdg1; fLeg2 = pdg2; fLeg1Exclude=exclude1; fLeg2Exclude=exclude2;}
77 void SetMotherPDGs(Int_t pdg1, Int_t pdg2, Bool_t exclude1=kFALSE, Bool_t exclude2=kFALSE)
78 {fMother1 = pdg1; fMother2 = pdg2; fMother1Exclude=exclude1; fMother2Exclude=exclude2;}
79 void SetGrandMotherPDGs(Int_t pdg1, Int_t pdg2, Bool_t exclude1=kFALSE, Bool_t exclude2=kFALSE)
80 {fGrandMother1 = pdg1; fGrandMother2 = pdg2; fGrandMother1Exclude=exclude1; fGrandMother2Exclude=exclude2;}
81 void SetLegSources(ESource s1, ESource s2) {fLeg1Source = s1; fLeg2Source = s2;}
82 void SetMotherSources(ESource s1, ESource s2) {fMother1Source = s1; fMother2Source = s2;}
83 void SetGrandMotherSources(ESource s1, ESource s2) {fGrandMother1Source = s1; fGrandMother2Source = s2;}
84 void SetCheckBothChargesLegs(Bool_t flag1, Bool_t flag2) {fCheckBothChargesLeg1 = flag1; fCheckBothChargesLeg2 = flag2;}
85 void SetCheckBothChargesMothers(Bool_t flag1, Bool_t flag2) {fCheckBothChargesMother1 = flag1; fCheckBothChargesMother2 = flag2;}
ba15fdfb 86 void SetCheckBothChargesGrandMothers(Bool_t flag1, Bool_t flag2) {fCheckBothChargesGrandMother1 = flag1; fCheckBothChargesGrandMother2 = flag2;}
5720c765 87 void SetMothersRelation(EBranchRelation relation) {fMothersRelation = relation;}
88 void SetFillPureMCStep(Bool_t fill=kTRUE) {fFillPureMCStep = fill;}
89
90 Int_t GetLegPDG(Int_t branch) const {return (branch==1 ? fLeg1 : fLeg2);}
91 Int_t GetMotherPDG(Int_t branch) const {return (branch==1 ? fMother1 : fMother2);}
92 Int_t GetGrandMotherPDG(Int_t branch) const {return (branch==1 ? fGrandMother1 : fGrandMother2);}
93 Bool_t GetLegPDGexclude(Int_t branch) const {return (branch==1 ? fLeg1Exclude : fLeg2Exclude);}
94 Bool_t GetMotherPDGexclude(Int_t branch) const {return (branch==1 ? fMother1Exclude : fMother2Exclude);}
95 Bool_t GetGrandMotherPDGexclude(Int_t branch) const {return (branch==1 ? fGrandMother1Exclude : fGrandMother2Exclude);}
96 ESource GetLegSource(Int_t branch) const {return (branch==1 ? fLeg1Source : fLeg2Source);}
97 ESource GetMotherSource(Int_t branch) const {return (branch==1 ? fMother1Source : fMother2Source);}
98 ESource GetGrandMotherSource(Int_t branch) const {return (branch==1 ? fGrandMother1Source : fGrandMother2Source);}
99 Bool_t GetCheckBothChargesLegs(Int_t branch) const {return (branch==1 ? fCheckBothChargesLeg1 : fCheckBothChargesLeg2);}
100 Bool_t GetCheckBothChargesMothers(Int_t branch) const {return (branch==1 ? fCheckBothChargesMother1 : fCheckBothChargesMother2);}
101 Bool_t GetCheckBothChargesGrandMothers(Int_t branch) const {return (branch==1 ? fCheckBothChargesGrandMother1 : fCheckBothChargesGrandMother2);}
102 EBranchRelation GetMothersRelation() const {return fMothersRelation;}
103 Bool_t GetFillPureMCStep() const {return fFillPureMCStep;}
ba15fdfb 104
ef37a5a8 105 void SetJpsiRadiative(EJpsiRadiativ rad) { fJpsiRadiative=rad; }
106 EJpsiRadiativ GetJpsiRadiative() const { return fJpsiRadiative; }
ba15fdfb 107 private:
5720c765 108 // PDG codes for legs, mothers and grand-mothers
ba15fdfb 109 Int_t fLeg1; // leg 1 PDG
110 Int_t fLeg2; // leg 2 PDG
111 Int_t fMother1; // mother 1 PDG
112 Int_t fMother2; // mother 2 PDG
113 Int_t fGrandMother1; // grandmother 1 PDG
114 Int_t fGrandMother2; // grandmother 2 PDG
5720c765 115
116 // Toggle on/off the use of the PDG codes as inclusion or exclusion
117 // Example: if fLeg1=211 and fLeg1Exclude=kTRUE than all codes will be accepted for leg 1 with
118 // the exception of 211 (pions)
119 Bool_t fLeg1Exclude; // leg 1
120 Bool_t fLeg2Exclude; // leg 2
121 Bool_t fMother1Exclude; // mother 1
122 Bool_t fMother2Exclude; // mother 2
123 Bool_t fGrandMother1Exclude; // grandmother 1
124 Bool_t fGrandMother2Exclude; // grandmother 2
ba15fdfb 125
5720c765 126 // Particle sources
ba15fdfb 127 ESource fLeg1Source; // leg 1 source
128 ESource fLeg2Source; // leg 2 source
129 ESource fMother1Source; // mother 1 source
130 ESource fMother2Source; // mother 2 source
131 ESource fGrandMother1Source; // grandmother 1 source
132 ESource fGrandMother2Source; // grandmother 2 source
133
5720c765 134 // Flaggs whether to check both charges of a given PDG code
ba15fdfb 135 Bool_t fCheckBothChargesLeg1; // check both charges of the legs pdg
136 Bool_t fCheckBothChargesLeg2; // leg2
137 Bool_t fCheckBothChargesMother1; // mother 1
138 Bool_t fCheckBothChargesMother2; // mother 2
139 Bool_t fCheckBothChargesGrandMother1; // grand mother 1
140 Bool_t fCheckBothChargesGrandMother2; // grand mother 2
141
142 EBranchRelation fMothersRelation; // mother 1&2 relation (same, different or whatever)
ef37a5a8 143
144 EJpsiRadiativ fJpsiRadiative; // check for J/psi radiative decay
ba15fdfb 145
146 Bool_t fFillPureMCStep; // check and fill the pure MC step
147
ef37a5a8 148 ClassDef(AliDielectronSignalMC,2);
ba15fdfb 149};
150
151#endif