Exclude "neutral" tracks from primary vertex reco
[u/mrichter/AliRoot.git] / EVE / EveBase / AliEveV0.h
CommitLineData
b594a020 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 <TEveVSDStructs.h>
23#include <TEveElement.h>
24#include <TEveTrack.h>
25
26#include <TPolyMarker3D.h>
27#include <TPolyLine3D.h>
28
d4dbbb43 29#include <TPDGCode.h>
30
b594a020 31class TH1F;
32class TH2F;
33
34
35class AliEveV0List;
36
37class AliEveV0 : public TEvePointSet
38{
39 friend class AliEveV0List;
68ca2fe7 40 friend class AliEveV0Editor;
b594a020 41
42public:
43 AliEveV0();
44 AliEveV0(TEveRecTrack* tNeg, TEveRecTrack* tPos, TEveRecV0* v0,
45 TEveTrackPropagator* rs);
46 virtual ~AliEveV0();
47
b594a020 48 void MakeV0();
49
b594a020 50 virtual void SetMainColor(Color_t col)
51 {
e6e8b28b 52 TEvePointSet::SetMainColor(col);
53 fPointingLine->SetLineColor(fMarkerColor);
b594a020 54 }
55
56 void SetRnrStyle(TEveTrackPropagator* rs) { fRnrStyle = rs; }
57
58 Float_t GetDaughterDCA() const { return fDaughterDCA; }
59 void SetDaughterDCA(Float_t dca) { fDaughterDCA = dca; }
60
61 Float_t GetRadius() const { return fRecDecayV.Perp(); }
e1bb28b4 62 Float_t GetPt() const { return fRecDecayP.Perp(); }
b594a020 63
d4dbbb43 64 Float_t GetInvMass(Float_t nPdgCode, Float_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
ca8a6926 69 Bool_t GetOnFlyStatus() const { return fOnFlyStatus; }
70 void SetOnFlyStatus(Bool_t fs) { fOnFlyStatus = fs; }
71
b594a020 72 Int_t GetESDIndex() const { return fESDIndex; }
73 void SetESDIndex(Int_t ind) { fESDIndex = ind;}
74
75 virtual const Text_t* GetName() const { return Form("ESDv0_%i",fESDIndex); }
76 virtual const Text_t* GetTitle() const { return Form("ESDv0_%i",fESDIndex); }
77
b594a020 78 TEveTrackPropagator* GetPropagator() const { return fRnrStyle; }
79
e6e8b28b 80 TEveTrack* GetNegTrack() { return fNegTrack; }
81 TEveTrack* GetPosTrack() { return fPosTrack; }
b594a020 82
e6e8b28b 83 TEveLine* GetPointingLine() { return fPointingLine; }
b594a020 84
85protected:
86 TEveVector fRecBirthV; // Reconstucted birth point of neutral particle
87 TEveVector fRecDecayV; // Point of closest approach
88 TEveVector fRecDecayP;
89
90 TEveTrack *fNegTrack;
91 TEveTrack *fPosTrack;
92
93 TEveTrackPropagator *fRnrStyle;
94
e6e8b28b 95 TEveLine *fPointingLine;
b594a020 96
ca8a6926 97 Int_t fESDIndex; // Index in ESD V0 array.
98 Bool_t fOnFlyStatus; // Reconstructed during tracking.
99 Float_t fDaughterDCA; // Distance at the point of closest approach.
100 Float_t fChi2V0; // Some Chi-square.
b594a020 101
102private:
103 AliEveV0(const AliEveV0&); // Not implemented
104 AliEveV0& operator=(const AliEveV0&); // Not implemented
105
106 ClassDef(AliEveV0, 0); // Visual representation of a AliEveV0.
107};
108
109
110/******************************************************************************/
111// AliEveV0List
112/******************************************************************************/
113
114class AliEveV0List : public TEveElementList
115{
ca8a6926 116 friend class AliEveV0ListEditor;
117
b594a020 118public:
119 AliEveV0List();
120 AliEveV0List(TEveTrackPropagator* rs);
121 AliEveV0List(const Text_t* name, TEveTrackPropagator* rs=0);
e6e8b28b 122 virtual ~AliEveV0List() {}
b594a020 123
124 virtual const Text_t* GetTitle() const { return fTitle; }
125 virtual void SetTitle(const Text_t* t) { fTitle = t; }
126 virtual void SetTracksColor(Color_t cNeg, Color_t cPos) {
127 fNegColor = cNeg; fPosColor = cPos;}
128
129 virtual Bool_t CanEditMainColor() const { return kTRUE; }
130
b594a020 131 void SetRnrStyle(TEveTrackPropagator* rst) { fRnrStyle = rst; }
132 TEveTrackPropagator* GetPropagator() { return fRnrStyle; }
133
134 Bool_t GetRnrV0vtx() const { return fRnrV0vtx; }
135 Bool_t GetRnrV0path() const { return fRnrV0path; }
136 Bool_t GetRnrDaughters() const { return fRnrDaughters; }
137
b594a020 138 void MakeV0s();
b594a020 139
ca8a6926 140 void FilterByRadius(Float_t minR, Float_t maxR);
e1bb28b4 141 void FilterByDaughterDCA(Float_t minDaughterDCA, Float_t maxDaughterDCA);
142 void FilterByPt(Float_t minPt, Float_t maxPt);
ca8a6926 143
b594a020 144protected:
145 TString fTitle;
146
147 TEveTrackPropagator *fRnrStyle;
148
149 Bool_t fRnrDaughters;
150 Bool_t fRnrV0vtx;
151 Bool_t fRnrV0path;
152
153 Color_t fNegColor;
154 Color_t fPosColor;
155
ca8a6926 156 Float_t fMinRCut;
157 Float_t fMaxRCut;
158
e1bb28b4 159 Float_t fMinDaughterDCA;
160 Float_t fMaxDaughterDCA;
161
162 Float_t fMinPt;
163 Float_t fMaxPt;
164
b594a020 165private:
e6e8b28b 166 void Init();
b594a020 167
168 AliEveV0List(const AliEveV0List&); // Not implemented
169 AliEveV0List& operator=(const AliEveV0List&); // Not implemented
170
171 ClassDef(AliEveV0List, 0); // A list of AliEveV0 objecs.
172};
173
174
175#endif