]>
Commit | Line | Data |
---|---|---|
a65a7e70 | 1 | #ifndef ALIMCINFOCUTS_H |
2 | #define ALIMCINFOCUTS_H | |
3 | ||
4 | //------------------------------------------------------------------------------ | |
5 | // Class to keep selection cuts for MC tracks. | |
6 | // | |
7 | // Author: J.Otwinowski 04/02/2008 | |
8 | //------------------------------------------------------------------------------ | |
9 | ||
10 | #include <TPDGCode.h> | |
11 | #include "AliAnalysisCuts.h" | |
12 | ||
13 | class TArrayI; | |
14 | ||
15 | class AliMCInfoCuts : public AliAnalysisCuts | |
16 | { | |
17 | public: | |
18 | AliMCInfoCuts(const Char_t* name ="AliMCInfoCuts", const Char_t *title =""); | |
19 | virtual ~AliMCInfoCuts(); | |
20 | ||
21 | // setters | |
22 | void SetMinRowsWithDigits(const Int_t min=0) {fMinRowsWithDigits = min;} | |
23 | void SetMaxR(const Float_t max=1e99) {fMaxR = max;} | |
24 | void SetMaxVz(const Float_t max=1e99) {fMaxVz = max;} | |
25 | void SetRangeTPCSignal(const Float_t min=0, const Float_t max=1e99) {fMinTPCSignal = min; fMaxTPCSignal = max;} | |
26 | void SetMinTrackLength(const Int_t min=0) {fMinTrackLength = min;} | |
27 | ||
28 | // getters | |
29 | Int_t GetMinRowsWithDigits() const {return fMinRowsWithDigits;} | |
30 | Float_t GetMaxR() const {return fMaxR;} | |
31 | Float_t GetMaxVz() const {return fMaxVz;} | |
32 | Float_t GetMinTPCSignal() const {return fMinTPCSignal;} | |
33 | Float_t GetMaxTPCSignal() const {return fMaxTPCSignal;} | |
34 | Float_t GetMinTrackLength() const {return fMinTrackLength;} | |
35 | ||
36 | Float_t GetEP() const {return ep;} | |
37 | Float_t GetEM() const {return em;} | |
38 | Float_t GetMuP() const {return mup;} | |
39 | Float_t GetMuM() const {return mum;} | |
40 | Float_t GetPiP() const {return pip;} | |
41 | Float_t GetPiM() const {return pim;} | |
42 | Float_t GetKP() const {return kp;} | |
43 | Float_t GetKM() const {return km;} | |
44 | Float_t GetProt() const {return prot;} | |
45 | Float_t GetProtBar() const {return protbar;} | |
46 | ||
47 | // cuts init function | |
48 | void InitME(); | |
49 | ||
50 | // check MC tracks | |
51 | virtual Bool_t IsSelected(TObject *) {return kTRUE;} | |
52 | virtual Bool_t IsSelected(TList *) {return kTRUE;} | |
53 | ||
54 | // add particle to array | |
55 | void AddPdgParticle(Int_t idx=-1, Int_t pdgcode=0) const; | |
56 | ||
57 | // check particle in array | |
58 | Bool_t IsPdgParticle(Int_t pdgcode=0) const; | |
59 | ||
60 | // check particle in array | |
61 | Bool_t IsPosPdgParticle(Int_t pdgcode=0) const; | |
62 | ||
63 | // Merge output objects (needed by PROOF) | |
64 | virtual Long64_t Merge(TCollection* list); | |
65 | ||
66 | private: | |
67 | Int_t fMinRowsWithDigits; // min. number of TPC raws with digits | |
68 | Float_t fMaxR; // max. R distance from MC vertex | |
69 | Float_t fMaxVz; // max. Z distance from MC vertex | |
70 | Float_t fMinTPCSignal; // min. TPC Signal calculated from Bethe Bloch formula | |
71 | Float_t fMaxTPCSignal; // max. TPC Signal calculated from Bethe Bloch formula | |
72 | Float_t fMinTrackLength; // min. TPC minimum track length | |
73 | ||
74 | TArrayI* aTrackParticles; // array of tracked particles | |
75 | ||
76 | // PDG tracked particles (later added to aTrackParticles) | |
77 | enum enumData { | |
78 | kNParticles = 10, // number of particles below | |
79 | ep = kPositron, | |
80 | em = kElectron, | |
81 | mup = kMuonPlus, | |
82 | mum = kMuonMinus, | |
83 | pip = kPiPlus, | |
84 | pim = kPiMinus, | |
85 | kp = kKPlus, | |
86 | km = kKMinus, | |
87 | prot = kProton, | |
88 | protbar = kProtonBar | |
89 | }; | |
90 | ||
91 | AliMCInfoCuts(const AliMCInfoCuts&); // not implemented | |
92 | AliMCInfoCuts& operator=(const AliMCInfoCuts&); // not implemented | |
93 | ||
94 | ClassDef(AliMCInfoCuts, 1) | |
95 | }; | |
96 | ||
97 | #endif // ALIMCINFOCUTS_H |