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
  10 #ifndef AliEveV0_H
  11 #define AliEveV0_H
  12
  13
  14 //------------------------------------------------------------------------------
  15 // AliEveV0
  16 //------------------------------------------------------------------------------
  17 //
  18 // Representation of a reconstructed V0.
  19 //
  20 //
  21 //------------------------------------------------------------------------------
  22 // AliEveV0List
  23 //------------------------------------------------------------------------------
  24 //
  25 // Container for AliEveV0s.
  26 //
  27 // Allows runtime selection by pT and DCA of daughters, radius of V0
  28 // creation and PID priobabilities
  29
  30 //==============================================================================
  31
  32 #include "AliEveTrack.h"
  33 #include <TEveVSDStructs.h>
  34
  35 #include <TPolyLine3D.h>
  36
  37 #include <TPDGCode.h>
  38
  39 class TH1F;
  40 class TH2F;
  41
  42
  43 class AliEveV0List;
  44
  45 class AliEveV0 : public TEvePointSet
  46 {
  47   friend class AliEveV0List;
  48   friend class AliEveV0Editor;
  49
  50 public:
  51   AliEveV0();
  52   AliEveV0(TEveRecTrack* tNeg, TEveRecTrack* tPos, TEveRecV0* v0,
  53      TEveTrackPropagator* rs);
  54   virtual ~AliEveV0();
  55
  56   void MakeV0();
  57
  58   virtual void  SetMainColor(Color_t col)
  59   {
  60     TEvePointSet::SetMainColor(col);
  61     fPointingLine->SetLineColor(fMarkerColor);
  62   }
  63
  64   void SetRnrStyle(TEveTrackPropagator* rs) { fRnrStyle = rs; }
  65
  66   Float_t GetDaughterDCA() const { return fDaughterDCA; }
  67   void SetDaughterDCA(Float_t dca) { fDaughterDCA = dca; }
  68
  69   Float_t GetPhi()    const { return fRecDecayP.Phi(); }
  70   Float_t GetEta()    const { return fRecDecayP.Eta(); }
  71   Float_t GetRadius() const { return fRecDecayV.Perp(); }
  72   Float_t GetPt()     const { return fRecDecayP.Perp(); }
  73   Float_t GetMomentum() const { return fRecDecayP.Mag(); }
  74
  75   Float_t GetInvMass(Int_t nPdgCode, Int_t pPdgCode) const;
  76   Float_t GetK0sInvMass() const { return GetInvMass(kPiMinus,kPiPlus); }
  77   Float_t GetLambdaInvMass() const { return GetInvMass(kPiMinus,kProton); }
  78   Float_t GetAntiLambdaInvMass() const { return GetInvMass(kProton,kPiPlus); }
  79
  80   Bool_t GetOnFlyStatus()    const { return fOnFlyStatus; }
  81   void   SetOnFlyStatus(Bool_t fs) { fOnFlyStatus = fs; }
  82
  83   void    SetMaxProbPdgPid(Int_t iDaughter, Int_t rPdg, Float_t rPid);
  84   Int_t   GetNegMaxProbPdg() const { return fNegMaxProbPdg; }
  85   Int_t   GetPosMaxProbPdg() const { return fPosMaxProbPdg; }
  86   Float_t GetNegMaxProbPid() const { return fNegMaxProbPid; }
  87   Float_t GetPosMaxProbPid() const { return fPosMaxProbPid; }
  88
  89   Int_t GetESDIndex() const { return fESDIndex; }
  90   void  SetESDIndex(Int_t ind) { fESDIndex = ind;}
  91
  92   virtual const Text_t* GetName()  const { return Form("ESDv0_%i",fESDIndex); }
  93   virtual const Text_t* GetTitle() const { return Form("ESDv0_%i",fESDIndex); }
  94
  95   TEveTrackPropagator* GetPropagator() const  { return fRnrStyle; }
  96
  97   AliEveTrack* GetNegTrack() const { return fNegTrack; }
  98   AliEveTrack* GetPosTrack() const { return fPosTrack; }
  99
 100   TEveLine*  GetPointingLine() const { return fPointingLine; }
 101
 102 protected:
 103   TEveVector fRecBirthV;    // Reconstucted birth point of neutral particle
 104   TEveVector fRecDecayV;    // Point of closest approach
 105   TEveVector fRecDecayP;    // Reconstructed momentum of decayed particle.
 106
 107   AliEveTrack        *fNegTrack; // Representation of negative track.
 108   AliEveTrack        *fPosTrack; // Representation of positive track.
 109
 110   TEveTrackPropagator *fRnrStyle; // Track propagator for neg/pos track.
 111
 112   TEveLine         *fPointingLine; // Representation of pointing line.
 113
 114   Int_t             fESDIndex;    // Index in ESD V0 array.
 115   Bool_t            fOnFlyStatus; // Reconstructed during tracking.
 116   Float_t           fDaughterDCA; // Distance at the point of closest approach.
 117   Float_t           fChi2V0;      // Some Chi-square.
 118
 119   Int_t             fNegMaxProbPdg; // Maximum PDG probability for the negative daughter
 120   Int_t             fPosMaxProbPdg; // Maximum PDG probability for the positive daughter
 121   Float_t           fNegMaxProbPid; // Maximum PID probability for the negative daughter
 122   Float_t           fPosMaxProbPid; // Maximum PID probability for the positive daughter
 123
 124 private:
 125   AliEveV0(const AliEveV0&);            // Not implemented
 126   AliEveV0& operator=(const AliEveV0&); // Not implemented
 127
 128   ClassDef(AliEveV0, 0); // Visual representation of a AliEveV0.
 129 };
 130
 131
 132 /******************************************************************************/
 133 // AliEveV0List
 134 /******************************************************************************/
 135
 136 class AliEveV0List : public TEveElementList
 137 {
 138   friend class AliEveV0ListEditor;
 139
 140 public:
 141   AliEveV0List();
 142   AliEveV0List(TEveTrackPropagator* rs);
 143   AliEveV0List(const Text_t* name, TEveTrackPropagator* rs=0);
 144   virtual ~AliEveV0List() {}
 145
 146   virtual const Text_t* GetTitle() const { return fTitle; }
 147   virtual void SetTitle(const Text_t* t) { fTitle = t; }
 148   virtual void SetTracksColor(Color_t cNeg, Color_t cPos) {
 149     fNegColor = cNeg; fPosColor = cPos;}
 150
 151   virtual Bool_t CanEditMainColor() const { return kTRUE; }
 152
 153   void  SetRnrStyle(TEveTrackPropagator* rst) { fRnrStyle = rst; }
 154   TEveTrackPropagator* GetPropagator()        { return fRnrStyle; }
 155
 156   Bool_t GetRnrV0vtx()     const { return fRnrV0vtx; }
 157   Bool_t GetRnrV0path()    const { return fRnrV0path; }
 158   Bool_t GetRnrDaughters() const { return fRnrDaughters; }
 159
 160   void   MakeV0s();
 161
 162   void   FilterByRadius(Float_t minR, Float_t maxR);
 163   void   FilterByDaughterDCA(Float_t minDaughterDCA, Float_t maxDaughterDCA);
 164   void   FilterByPt(Float_t minPt, Float_t maxPt);
 165   void   FilterByCheckedPidMinProb(Int_t rFlag, Int_t rDaughter, Int_t rPid, Float_t rProb);
 166   void   SetNegCheckedPid(Int_t rNegCheckedPid) {fNegCheckedPid = rNegCheckedPid;}
 167   void   SetPosCheckedPid(Int_t rPosCheckedPid) {fPosCheckedPid = rPosCheckedPid;}
 168   Int_t  GetNegCheckedPid() const { return fNegCheckedPid; }
 169   Int_t  GetPosCheckedPid() const { return fPosCheckedPid; }
 170
 171   void   SetNegCheckedProb(Float_t rNegCheckedProb) {fNegCheckedProb = rNegCheckedProb;}
 172   void   SetPosCheckedProb(Float_t rPosCheckedProb) {fPosCheckedProb = rPosCheckedProb;}
 173   Float_t  GetNegCheckedProb() const { return fNegCheckedProb; }
 174   Float_t  GetPosCheckedProb() const { return fPosCheckedProb; }
 175
 176   void   FilterByInvariantMass(Float_t minPt, Float_t maxPt, Int_t nPdgCode, Int_t pPdgCode);
 177
 178 protected:
 179   TString              fTitle;    // Title of the object.
 180
 181   TEveTrackPropagator *fRnrStyle; // Track propagator to be passed do conteined V0s.
 182
 183   Bool_t               fRnrDaughters; // Flag - display daughter tracks.
 184   Bool_t               fRnrV0vtx;     // Flag - display V0 vertex.
 185   Bool_t               fRnrV0path;    // Flag - display V0 path.
 186
 187   Color_t              fNegColor;     // Color for negative tracks.
 188   Color_t              fPosColor;     // Color for positive tracks.
 189
 190   Float_t              fMinRCut;      // Minimum radius cut.
 191   Float_t              fMaxRCut;      // Maximum radius cut.
 192
 193   Float_t              fMinDaughterDCA; // Minimum daughter DCA cut.
 194   Float_t              fMaxDaughterDCA; // Maximum daughter DCA cut.
 195
 196   Float_t              fMinPt;          // Minimum Pt cut.
 197   Float_t              fMaxPt;          // Maximum Pt cut.
 198
 199   Int_t                fNegCheckedPid;  // BORIS ?
 200   Int_t                fPosCheckedPid;  // BORIS ?
 201
 202   Float_t              fNegCheckedProb; // BORIS ?
 203   Float_t              fPosCheckedProb; // BORIS ?
 204
 205   Float_t              fMinInvariantMass; // Minimum invariant mass cut.
 206   Float_t              fMaxInvariantMass; // Maximum invariant mass cut.
 207
 208 private:
 209   void Init();
 210
 211   AliEveV0List(const AliEveV0List&);            // Not implemented
 212   AliEveV0List& operator=(const AliEveV0List&); // Not implemented
 213
 214   ClassDef(AliEveV0List, 0); // A list of AliEveV0 objecs.
 215 };
 216
 217
 218 #endif