]>
Commit | Line | Data |
---|---|---|
1 | #include "AliAnalysis.h" | |
2 | //________________________________ | |
3 | /////////////////////////////////////////////////////////// | |
4 | // | |
5 | // class AliAnalysis | |
6 | // | |
7 | // Base class for analysis. | |
8 | // Each inheriting calss must define 3 methods: | |
9 | // - Init() : that is called before event processing | |
10 | // - ProcessEvent(AliESD*,AliStack*) | |
11 | // - | |
12 | // Piotr.Skowronski@cern.ch | |
13 | // | |
14 | /////////////////////////////////////////////////////////// | |
15 | ||
16 | #include "AliEventCut.h" | |
17 | #include "AliAODPairCut.h" | |
18 | ||
19 | ClassImp(AliAnalysis) | |
20 | ||
21 | AliAnalysis::AliAnalysis(): | |
22 | fEventCut(0x0), | |
23 | fCutOnSim(kTRUE), | |
24 | fCutOnRec(kTRUE), | |
25 | fPairCut(new AliAODPairEmptyCut()),//empty cut - accepts all particles | |
26 | fkPass(&AliAnalysis::PassPartAndTrack), //by default perform cut on both track and particle pair | |
27 | fkPass1(&AliAnalysis::PassPartAndTrack1), //used onluy by ProcessTracksAndParticles | |
28 | fkPass2(&AliAnalysis::PassPartAndTrack2), | |
29 | fkPassPairProp(&AliAnalysis::PassPairPropPartAndTrack) | |
30 | { | |
31 | //ctor | |
32 | } | |
33 | /*********************************************************/ | |
34 | ||
35 | AliAnalysis::AliAnalysis(const char* name,const char* title): | |
36 | TTask(name,title), | |
37 | fEventCut(0x0), | |
38 | fCutOnSim(kTRUE), | |
39 | fCutOnRec(kTRUE), | |
40 | fPairCut(new AliAODPairEmptyCut()),//empty cut - accepts all particles | |
41 | fkPass(&AliAnalysis::PassPartAndTrack), //by default perform cut on both track and particle pair | |
42 | fkPass1(&AliAnalysis::PassPartAndTrack1), //used onluy by ProcessTracksAndParticles | |
43 | fkPass2(&AliAnalysis::PassPartAndTrack2), | |
44 | fkPassPairProp(&AliAnalysis::PassPairPropPartAndTrack) | |
45 | { | |
46 | //ctor | |
47 | } | |
48 | /*********************************************************/ | |
49 | ||
50 | AliAnalysis::~AliAnalysis() | |
51 | { | |
52 | //dtor | |
53 | delete fEventCut; | |
54 | } | |
55 | /*********************************************************/ | |
56 | ||
57 | void AliAnalysis::SetEventCut(AliEventCut* evcut) | |
58 | { | |
59 | //Sets event - makes a private copy | |
60 | delete fEventCut; | |
61 | if (evcut) fEventCut = (AliEventCut*)evcut->Clone(); | |
62 | else fEventCut = 0x0; | |
63 | } | |
64 | /*********************************************************/ | |
65 | ||
66 | Bool_t AliAnalysis::Rejected(AliAOD* recevent, AliAOD* simevent) | |
67 | { | |
68 | //checks the event cut | |
69 | if (fEventCut == 0x0) return kFALSE; | |
70 | ||
71 | if (fCutOnRec) | |
72 | if (fEventCut->Rejected(recevent)) return kTRUE; | |
73 | ||
74 | if (fCutOnSim) | |
75 | if (fEventCut->Rejected(simevent)) return kTRUE; | |
76 | ||
77 | return kFALSE; | |
78 | } | |
79 | /*************************************************************************************/ | |
80 | ||
81 | void AliAnalysis::SetPairCut(AliAODPairCut* cut) | |
82 | { | |
83 | //Sets new Pair Cut. Old one is deleted | |
84 | //Note that it is created new object instead of simple pointer set | |
85 | //I do not want to have pointer | |
86 | //to object created somewhere else | |
87 | //because in that case I could not believe that | |
88 | //it would always exist (sb could delete it) | |
89 | //so we have always own copy | |
90 | ||
91 | if(!cut) | |
92 | { | |
93 | Error("AliHBTFunction::SetPairCut","argument is NULL"); | |
94 | return; | |
95 | } | |
96 | delete fPairCut; | |
97 | fPairCut = (AliAODPairCut*)cut->Clone(); | |
98 | ||
99 | } | |
100 | /******************************************************************/ | |
101 | ||
102 | void AliAnalysis::SetCutsOnSim() | |
103 | { | |
104 | // -- aplies only to Process("TracksAndParticles") | |
105 | // (ProcessTracksAndParticles and ProcessTracksAndParticlesNonIdentAnal) | |
106 | // Only particles properties are checkes against cuts | |
107 | ||
108 | fCutOnRec = kFALSE; | |
109 | fCutOnSim = kTRUE; | |
110 | ||
111 | fkPass = &AliAnalysis::PassPart; | |
112 | fkPass1 = &AliAnalysis::PassPart1; | |
113 | fkPass2 = &AliAnalysis::PassPart2; | |
114 | fkPassPairProp = &AliAnalysis::PassPairPropPart; | |
115 | ||
116 | } | |
117 | /*************************************************************************************/ | |
118 | ||
119 | void AliAnalysis::SetCutsOnRec() | |
120 | { | |
121 | // -- aplies only to Process("TracksAndParticles") | |
122 | // (ProcessTracksAndParticles and ProcessTracksAndParticlesNonIdentAnal) | |
123 | // Only tracks properties are checkes against cuts | |
124 | ||
125 | fCutOnRec = kTRUE; | |
126 | fCutOnSim = kFALSE; | |
127 | ||
128 | fkPass = &AliAnalysis::PassTrack; | |
129 | fkPass1 = &AliAnalysis::PassTrack1; | |
130 | fkPass2 = &AliAnalysis::PassTrack2; | |
131 | fkPassPairProp = &AliAnalysis::PassPairPropTrack; | |
132 | ||
133 | } | |
134 | /*************************************************************************************/ | |
135 | ||
136 | void AliAnalysis::SetCutsOnRecAndSim() | |
137 | { | |
138 | // -- aplies only to Process("TracksAndParticles") | |
139 | // (ProcessTracksAndParticles and ProcessTracksAndParticlesNonIdentAnal) | |
140 | // Both, tracks and particles, properties are checked against cuts | |
141 | ||
142 | fCutOnRec = kTRUE; | |
143 | fCutOnSim = kTRUE; | |
144 | ||
145 | fkPass = &AliAnalysis::PassPartAndTrack; | |
146 | fkPass1 = &AliAnalysis::PassPartAndTrack1; | |
147 | fkPass2 = &AliAnalysis::PassPartAndTrack2; | |
148 | fkPassPairProp = &AliAnalysis::PassPairPropPartAndTrack; | |
149 | } |