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