]>
Commit | Line | Data |
---|---|---|
1 | // $Id$ | |
2 | // Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007 | |
3 | ||
4 | /************************************************************************** | |
5 | * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. * | |
6 | * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for * | |
7 | * full copyright notice. * | |
8 | **************************************************************************/ | |
9 | #ifndef AliEveV0_H | |
10 | #define AliEveV0_H | |
11 | ||
12 | /* Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. * | |
13 | * See cxx source for full Copyright notice */ | |
14 | ||
15 | ||
16 | /*********************************************************************** | |
17 | * This code defines the reconstructed v0 visualized with EVE | |
18 | * | |
19 | * Ludovic Gaudichet (gaudichet@to.infn.it) | |
20 | ************************************************************************/ | |
21 | ||
22 | #include "AliEveTrack.h" | |
23 | #include <TEveVSDStructs.h> | |
24 | ||
25 | #include <TPolyLine3D.h> | |
26 | ||
27 | #include <TPDGCode.h> | |
28 | ||
29 | class TH1F; | |
30 | class TH2F; | |
31 | ||
32 | ||
33 | class AliEveV0List; | |
34 | ||
35 | class AliEveV0 : public TEvePointSet | |
36 | { | |
37 | friend class AliEveV0List; | |
38 | friend class AliEveV0Editor; | |
39 | ||
40 | public: | |
41 | AliEveV0(); | |
42 | AliEveV0(TEveRecTrack* tNeg, TEveRecTrack* tPos, TEveRecV0* v0, | |
43 | TEveTrackPropagator* rs); | |
44 | virtual ~AliEveV0(); | |
45 | ||
46 | void MakeV0(); | |
47 | ||
48 | virtual void SetMainColor(Color_t col) | |
49 | { | |
50 | TEvePointSet::SetMainColor(col); | |
51 | fPointingLine->SetLineColor(fMarkerColor); | |
52 | } | |
53 | ||
54 | void SetRnrStyle(TEveTrackPropagator* rs) { fRnrStyle = rs; } | |
55 | ||
56 | Float_t GetDaughterDCA() const { return fDaughterDCA; } | |
57 | void SetDaughterDCA(Float_t dca) { fDaughterDCA = dca; } | |
58 | ||
59 | Float_t GetPhi() const { return fRecDecayP.Phi(); } | |
60 | Float_t GetEta() const { return fRecDecayP.Eta(); } | |
61 | Float_t GetRadius() const { return fRecDecayV.Perp(); } | |
62 | Float_t GetPt() const { return fRecDecayP.Perp(); } | |
63 | ||
64 | Float_t GetInvMass(Int_t nPdgCode, Int_t pPdgCode) const; | |
65 | Float_t GetK0sInvMass() const { return GetInvMass(kPiMinus,kPiPlus); } | |
66 | Float_t GetLambdaInvMass() const { return GetInvMass(kPiMinus,kProton); } | |
67 | Float_t GetAntiLambdaInvMass() const { return GetInvMass(kProton,kPiPlus); } | |
68 | ||
69 | Bool_t GetOnFlyStatus() const { return fOnFlyStatus; } | |
70 | void SetOnFlyStatus(Bool_t fs) { fOnFlyStatus = fs; } | |
71 | ||
72 | void SetMaxProbPdgPid(Int_t iDaughter, Int_t rPdg, Float_t rPid); | |
73 | Int_t GetNegMaxProbPdg() const { return fNegMaxProbPdg; } | |
74 | Int_t GetPosMaxProbPdg() const { return fPosMaxProbPdg; } | |
75 | Float_t GetNegMaxProbPid() const { return fNegMaxProbPid; } | |
76 | Float_t GetPosMaxProbPid() const { return fPosMaxProbPid; } | |
77 | ||
78 | Int_t GetESDIndex() const { return fESDIndex; } | |
79 | void SetESDIndex(Int_t ind) { fESDIndex = ind;} | |
80 | ||
81 | virtual const Text_t* GetName() const { return Form("ESDv0_%i",fESDIndex); } | |
82 | virtual const Text_t* GetTitle() const { return Form("ESDv0_%i",fESDIndex); } | |
83 | ||
84 | TEveTrackPropagator* GetPropagator() const { return fRnrStyle; } | |
85 | ||
86 | AliEveTrack* GetNegTrack() { return fNegTrack; } | |
87 | AliEveTrack* GetPosTrack() { return fPosTrack; } | |
88 | ||
89 | TEveLine* GetPointingLine() { return fPointingLine; } | |
90 | ||
91 | protected: | |
92 | TEveVector fRecBirthV; // Reconstucted birth point of neutral particle | |
93 | TEveVector fRecDecayV; // Point of closest approach | |
94 | TEveVector fRecDecayP; | |
95 | ||
96 | AliEveTrack *fNegTrack; | |
97 | AliEveTrack *fPosTrack; | |
98 | ||
99 | TEveTrackPropagator *fRnrStyle; | |
100 | ||
101 | TEveLine *fPointingLine; | |
102 | ||
103 | Int_t fESDIndex; // Index in ESD V0 array. | |
104 | Bool_t fOnFlyStatus; // Reconstructed during tracking. | |
105 | Float_t fDaughterDCA; // Distance at the point of closest approach. | |
106 | Float_t fChi2V0; // Some Chi-square. | |
107 | ||
108 | Int_t fNegMaxProbPdg; // Maximum PDG probability for the negative daughter | |
109 | Int_t fPosMaxProbPdg; // Maximum PDG probability for the positive daughter | |
110 | Float_t fNegMaxProbPid; // Maximum PID probability for the negative daughter | |
111 | Float_t fPosMaxProbPid; // Maximum PID probability for the positive daughter | |
112 | ||
113 | private: | |
114 | AliEveV0(const AliEveV0&); // Not implemented | |
115 | AliEveV0& operator=(const AliEveV0&); // Not implemented | |
116 | ||
117 | ClassDef(AliEveV0, 0); // Visual representation of a AliEveV0. | |
118 | }; | |
119 | ||
120 | ||
121 | /******************************************************************************/ | |
122 | // AliEveV0List | |
123 | /******************************************************************************/ | |
124 | ||
125 | class AliEveV0List : public TEveElementList | |
126 | { | |
127 | friend class AliEveV0ListEditor; | |
128 | ||
129 | public: | |
130 | AliEveV0List(); | |
131 | AliEveV0List(TEveTrackPropagator* rs); | |
132 | AliEveV0List(const Text_t* name, TEveTrackPropagator* rs=0); | |
133 | virtual ~AliEveV0List() {} | |
134 | ||
135 | virtual const Text_t* GetTitle() const { return fTitle; } | |
136 | virtual void SetTitle(const Text_t* t) { fTitle = t; } | |
137 | virtual void SetTracksColor(Color_t cNeg, Color_t cPos) { | |
138 | fNegColor = cNeg; fPosColor = cPos;} | |
139 | ||
140 | virtual Bool_t CanEditMainColor() const { return kTRUE; } | |
141 | ||
142 | void SetRnrStyle(TEveTrackPropagator* rst) { fRnrStyle = rst; } | |
143 | TEveTrackPropagator* GetPropagator() { return fRnrStyle; } | |
144 | ||
145 | Bool_t GetRnrV0vtx() const { return fRnrV0vtx; } | |
146 | Bool_t GetRnrV0path() const { return fRnrV0path; } | |
147 | Bool_t GetRnrDaughters() const { return fRnrDaughters; } | |
148 | ||
149 | void MakeV0s(); | |
150 | ||
151 | void FilterByRadius(Float_t minR, Float_t maxR); | |
152 | void FilterByDaughterDCA(Float_t minDaughterDCA, Float_t maxDaughterDCA); | |
153 | void FilterByPt(Float_t minPt, Float_t maxPt); | |
154 | void FilterByCheckedPidMinProb(Int_t rFlag, Int_t rDaughter, Int_t rPid, Float_t rProb); | |
155 | void SetNegCheckedPid(Int_t rNegCheckedPid) {fNegCheckedPid = rNegCheckedPid;} | |
156 | void SetPosCheckedPid(Int_t rPosCheckedPid) {fPosCheckedPid = rPosCheckedPid;} | |
157 | Int_t GetNegCheckedPid() {return fNegCheckedPid;} | |
158 | Int_t GetPosCheckedPid() {return fPosCheckedPid;} | |
159 | ||
160 | void SetNegCheckedProb(Float_t rNegCheckedProb) {fNegCheckedProb = rNegCheckedProb;} | |
161 | void SetPosCheckedProb(Float_t rPosCheckedProb) {fPosCheckedProb = rPosCheckedProb;} | |
162 | Float_t GetNegCheckedProb() {return fNegCheckedProb;} | |
163 | Float_t GetPosCheckedProb() {return fPosCheckedProb;} | |
164 | ||
165 | void FilterByInvariantMass(Float_t minPt, Float_t maxPt, Int_t nPdgCode, Int_t pPdgCode); | |
166 | ||
167 | protected: | |
168 | TString fTitle; | |
169 | ||
170 | TEveTrackPropagator *fRnrStyle; | |
171 | ||
172 | Bool_t fRnrDaughters; | |
173 | Bool_t fRnrV0vtx; | |
174 | Bool_t fRnrV0path; | |
175 | ||
176 | Color_t fNegColor; | |
177 | Color_t fPosColor; | |
178 | ||
179 | Float_t fMinRCut; | |
180 | Float_t fMaxRCut; | |
181 | ||
182 | Float_t fMinDaughterDCA; | |
183 | Float_t fMaxDaughterDCA; | |
184 | ||
185 | Float_t fMinPt; | |
186 | Float_t fMaxPt; | |
187 | ||
188 | Int_t fNegCheckedPid; | |
189 | Int_t fPosCheckedPid; | |
190 | ||
191 | Float_t fNegCheckedProb; | |
192 | Float_t fPosCheckedProb; | |
193 | ||
194 | Float_t fMinInvariantMass; | |
195 | Float_t fMaxInvariantMass; | |
196 | ||
197 | private: | |
198 | void Init(); | |
199 | ||
200 | AliEveV0List(const AliEveV0List&); // Not implemented | |
201 | AliEveV0List& operator=(const AliEveV0List&); // Not implemented | |
202 | ||
203 | ClassDef(AliEveV0List, 0); // A list of AliEveV0 objecs. | |
204 | }; | |
205 | ||
206 | ||
207 | #endif |