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 #include <TPDGCode.h>
  30
  31 class TH1F;
  32 class TH2F;
  33
  34
  35 class AliEveV0List;
  36
  37 class AliEveV0 : public TEvePointSet
  38 {
  39   friend class AliEveV0List;
  40   friend class AliEveV0Editor;
  41
  42 public:
  43   AliEveV0();
  44   AliEveV0(TEveRecTrack* tNeg, TEveRecTrack* tPos, TEveRecV0* v0,
  45      TEveTrackPropagator* rs);
  46   virtual ~AliEveV0();
  47
  48   void MakeV0();
  49
  50   virtual void  SetMainColor(Color_t col)
  51   {
  52     TEvePointSet::SetMainColor(col);
  53     fPointingLine->SetLineColor(fMarkerColor);
  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(); }
  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   TEveTrack* GetNegTrack() { return fNegTrack; }
  87   TEveTrack* 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   TEveTrack        *fNegTrack;
  97   TEveTrack        *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