EVE/EveBase/AliEveCascade.cxx

   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 //------------------------------------------------------------------------
  11 //
  12 // AliEveCascade class
  13 //
  14 //------------------------------------------------------------------------
  15
  16 //#include <TPolyLine3D.h>
  17 //#include <TColor.h>
  18
  19 //#include <vector>
  20
  21 #include <TDatabasePDG.h>
  22 #include <TEveTrackPropagator.h>
  23 #include <TEveManager.h>
  24
  25 #include "AliEveCascade.h"
  26
  27
  28
  29
  30 ClassImp(AliEveCascade)
  31
  32 //______________________________________________________________________________
  33 AliEveCascade::AliEveCascade() :
  34   TEvePointSet(),
  35
  36   fRecBirthV(),
  37   fRecDecayV(),
  38   fRecDecayP(),
  39   fRecDecayV0(),
  40   fBacTrack(0),
  41   fNegTrack(0),
  42   fPosTrack(0),
  43   fRnrStyle(0),
  44   fPointingCurve(0),
  45   fV0Path(0),
  46   fESDIndex(-1),
  47   fDaughterDCA(999),
  48   fChi2Cascade(-1),
  49   fLambdaP(0.,0.,0.),
  50   fBachP(0.,0.,0.)
  51 {
  52   // Default constructor.
  53
  54   // Override from TEveElement.
  55   fPickable = kTRUE;
  56   fMainColorPtr = &fMarkerColor;
  57 }
  58
  59 //______________________________________________________________________________
  60 AliEveCascade::AliEveCascade(TEveRecTrack* tBac, TEveRecTrack* tNeg, TEveRecTrack* tPos, TEveRecV0* v0, TEveRecCascade* cascade, TEveTrackPropagator* rs) :
  61   TEvePointSet(),
  62
  63   fRecBirthV(cascade->fCascadeBirth),
  64   fRecDecayV(cascade->fCascadeVCa),
  65   fRecDecayP(cascade->fPBac + v0->fPNeg + v0->fPPos),
  66   fRecDecayV0(v0->fVCa),
  67   fBacTrack(new AliEveTrack(tBac, rs)),
  68   fNegTrack(new AliEveTrack(tNeg, rs)),
  69   fPosTrack(new AliEveTrack(tPos, rs)),
  70
  71
  72   fRnrStyle(rs),
  73   fPointingCurve(new TEveLine("Pointing Curve")),
  74   fV0Path(new TEveLine("V0 Path")),
  75   fESDIndex(-1),
  76   fDaughterDCA(999), // DCA between Xi daughters is properly set in esd_make_cascade via the setter
  77   fChi2Cascade(-1),
  78   fLambdaP(0.,0.,0.), // Momemtum of Lambda is properly set in esd_make_cascade via the setter
  79   fBachP(0.,0.,0.)    // Momemtum of Lambda is properly set in esd_make_cascade via the setter
  80 {
  81   // Constructor with full Cascade specification.
  82
  83   // Override from TEveElement.
  84   fPickable = kTRUE;
  85   fMainColorPtr = &fMarkerColor;
  86
  87   fMarkerStyle = 2;
  88   fMarkerColor = kMagenta+2;
  89   fMarkerSize  = 1;
  90
  91   fPointingCurve->SetTitle("Xi pointing curve");
  92   fPointingCurve->SetLineColor(fMarkerColor);
  93   fPointingCurve->SetLineStyle(9);
  94   fPointingCurve->SetLineWidth(2);
  95   fPointingCurve->IncDenyDestroy();
  96   AddElement(fPointingCurve);
  97
  98   fV0Path->SetTitle("V0 path");
  99   fV0Path->SetLineColor(fMarkerColor);
 100   fV0Path->SetLineStyle(3);
 101   fV0Path->SetLineWidth(2);
 102   fV0Path->IncDenyDestroy();
 103   AddElement(fV0Path);
 104
 105   fBacTrack->SetLineColor(kMagenta);
 106   fBacTrack->SetStdTitle();
 107   fBacTrack->IncDenyDestroy();
 108   AddElement(fBacTrack);
 109
 110   fNegTrack->SetLineColor(kCyan+2);  // in V0 = kCyan
 111   fNegTrack->SetStdTitle();
 112   fNegTrack->IncDenyDestroy();
 113   AddElement(fNegTrack);
 114
 115   fPosTrack->SetLineColor(kRed+2);  // in Eve V0 = kRed
 116   fPosTrack->SetStdTitle();
 117   fPosTrack->IncDenyDestroy();
 118   AddElement(fPosTrack);
 119 }
 120
 121 //______________________________________________________________________________
 122 AliEveCascade::~AliEveCascade()
 123 {
 124   // Destructor. Dereferences bachelor track and pointing-line objects.
 125
 126   fBacTrack->DecDenyDestroy();
 127   fNegTrack->DecDenyDestroy();
 128   fPosTrack->DecDenyDestroy();
 129   fPointingCurve->DecDenyDestroy();
 130   fV0Path->DecDenyDestroy();
 131 }
 132
 133 //______________________________________________________________________________
 134 void AliEveCascade::MakeCascade()
 135 {
 136   // Set all dependant components for drawing.
 137
 138   SetPoint(0, fRecDecayV.fX, fRecDecayV.fY, fRecDecayV.fZ);
 139
 140   fBacTrack->MakeTrack();
 141   fNegTrack->MakeTrack();
 142   fPosTrack->MakeTrack();
 143
 144   fPointingCurve->SetPoint(0, fRecBirthV.fX, fRecBirthV.fY, fRecBirthV.fZ);
 145   fPointingCurve->SetPoint(1, fRecDecayV.fX, fRecDecayV.fY, fRecDecayV.fZ);
 146
 147   fV0Path->SetPoint(0, fRecDecayV.fX, fRecDecayV.fY, fRecDecayV.fZ);
 148   fV0Path->SetPoint(1, fRecDecayV0.fX, fRecDecayV0.fY, fRecDecayV0.fZ);
 149 }
 150
 151
 152
 153 //______________________________________________________________________________
 154 Double_t AliEveCascade::GetInvMass(Int_t cascadePdgCodeHyp) const
 155 {
 156   // Returns Invariant Mass assuming the masses of the daughter particles
 157         TEveVector lBachMomentum = fBacTrack->GetMomentum();
 158   // Does not work properly because momenta at the primary vertex for V0 daughters (same for AliEVEv0) + Bach
 159
 160         Double_t lBachMass = 0.0;
 161         if(cascadePdgCodeHyp == kXiMinus || cascadePdgCodeHyp == -kXiMinus )
 162                 lBachMass = TDatabasePDG::Instance()->GetParticle("pi-")->Mass();
 163         else if(cascadePdgCodeHyp == kOmegaMinus || cascadePdgCodeHyp == -kOmegaMinus )
 164                 lBachMass = TDatabasePDG::Instance()->GetParticle("K-")->Mass();
 165
 166         Double_t lLambdaMass = TDatabasePDG::Instance()->GetParticle("Lambda0")->Mass();
 167
 168         printf("\n Cascade : check the mass of the bachelor (%.5f) - the Lambda (%.5f)\n",lBachMass, lLambdaMass);
 169
 170         Double_t eBach   = TMath::Sqrt(lBachMass*lBachMass     + lBachMomentum.Mag2());
 171         //Double_t eBach   = TMath::Sqrt(lBachMass  *lBachMass   + fBachP.Mag2());
 172         Double_t eLambda = TMath::Sqrt(lLambdaMass*lLambdaMass + fLambdaP.Mag2());
 173
 174         TVector3 lCascadeMom(fBachP + fLambdaP);
 175         /*
 176         printf("Ptot(Lambda) : %f GeV/c \n", fLambdaP.Mag());
 177         printf("Ptot(Bach)   : %f GeV/c \n", lBachMomentum.Mag());
 178         printf("Ptot(BachESD): %f GeV/c \n", fBachP.Mag());
 179         printf("Ptot(Casc)   : %f GeV/c \n", fRecDecayP.Mag());
 180         printf("Ptot(CascESD): %f GeV/c \n", lCascadeMom.Mag());
 181         */
 182         //return TMath::Sqrt( (eBach+eLambda)*(eBach+eLambda) - fRecDecayP.Mag2() );    // numerically instable ...
 183         //return TMath::Sqrt( ((eBach+eLambda) - lCascadeMom.Mag()) * ((eBach+eLambda) + lCascadeMom.Mag()) );
 184         return TMath::Sqrt( ((eBach+eLambda) - fRecDecayP.Mag()) * ((eBach+eLambda) + fRecDecayP.Mag()) );
 185 }
 186
 187
 188
 189 //------------------------------------------------------------------------
 190 //
 191 // AliEveCascadeList class
 192 //
 193 //------------------------------------------------------------------------
 194
 195 ClassImp(AliEveCascadeList)
 196
 197 //______________________________________________________________________________
 198 AliEveCascadeList::AliEveCascadeList() :
 199   TEveElementList(),
 200   fTitle(),
 201   fRnrStyle(0),
 202   fRnrDaughters(kTRUE),
 203   fRnrCascadevtx(kTRUE),
 204   fRnrCascadepath(kTRUE),
 205   fBacColor(0),
 206   fMinRCut(0),
 207   fMaxRCut(250),
 208   fMinDaughterDCA(0),
 209   fMaxDaughterDCA(1),
 210   fMinPt(0),
 211   fMaxPt(20),
 212   fInvMassHyp(kXiMinus),
 213   fMinInvariantMass(1.0),
 214   fMaxInvariantMass(6.0)
 215 {
 216   // Default constructor.
 217
 218   fChildClass = AliEveCascade::Class(); // override member from base TEveElementList
 219 }
 220
 221 //______________________________________________________________________________
 222 AliEveCascadeList::AliEveCascadeList(TEveTrackPropagator* rs) :
 223   TEveElementList(),
 224   fTitle(),
 225   fRnrStyle(rs),
 226   fRnrDaughters(kTRUE),
 227   fRnrCascadevtx(kTRUE),
 228   fRnrCascadepath(kTRUE),
 229   fBacColor(0),
 230   fMinRCut(0),
 231   fMaxRCut(250),
 232   fMinDaughterDCA(0),
 233   fMaxDaughterDCA(1),
 234   fMinPt(0),
 235   fMaxPt(20),
 236   fInvMassHyp(kXiMinus),
 237   fMinInvariantMass(1.0),
 238   fMaxInvariantMass(6.0)
 239 {
 240   // Constructor with given track-propagator..
 241
 242   fChildClass = AliEveCascade::Class(); // override member from base TEveElementList
 243
 244   Init();
 245 }
 246
 247 //______________________________________________________________________________
 248 AliEveCascadeList::AliEveCascadeList(const Text_t* name, TEveTrackPropagator* rs) :
 249   TEveElementList(),
 250   fTitle(),
 251   fRnrStyle(rs),
 252   fRnrDaughters(kTRUE),
 253   fRnrCascadevtx(kTRUE),
 254   fRnrCascadepath(kTRUE),
 255   fBacColor(0),
 256   fMinRCut(0),
 257   fMaxRCut(100),
 258   fMinDaughterDCA(0),
 259   fMaxDaughterDCA(1),
 260   fMinPt(0),
 261   fMaxPt(20),
 262   fInvMassHyp(kXiMinus),
 263   fMinInvariantMass(1.0),
 264   fMaxInvariantMass(6.0)
 265 {
 266   // Standard constructor.
 267
 268   fChildClass = AliEveCascade::Class(); // override member from base TEveElementList
 269
 270   Init();
 271   SetName(name);
 272 }
 273
 274 //______________________________________________________________________________
 275 void AliEveCascadeList::Init()
 276 {
 277   // Initialize members needed for drawing operations.
 278
 279   if (fRnrStyle== 0) fRnrStyle = new TEveTrackPropagator;
 280 }
 281
 282 /******************************************************************************/
 283
 284 //______________________________________________________________________________
 285 void AliEveCascadeList::MakeCascades()
 286 {
 287   // Call MakeCascade() for all elements.
 288
 289   for(List_i i=fChildren.begin(); i!=fChildren.end(); ++i) {
 290     ((AliEveCascade*)(*i))->MakeCascade();
 291   }
 292   gEve->Redraw3D();
 293 }
 294
 295 /******************************************************************************/
 296
 297 //______________________________________________________________________________
 298 void AliEveCascadeList::FilterByRadius(Float_t minR, Float_t maxR)
 299 {
 300   // Select visibility of elements based on their axial radius.
 301
 302   fMinRCut = minR;
 303   fMaxRCut = maxR;
 304
 305   for(List_i i = fChildren.begin(); i != fChildren.end(); ++i)
 306   {
 307     AliEveCascade* cascade = (AliEveCascade*) *i;
 308     Float_t  rad = cascade->GetRadius();
 309     Bool_t  show = rad >= fMinRCut && rad <= fMaxRCut;
 310     cascade->SetRnrState(show);
 311   }
 312   ElementChanged();
 313   gEve->Redraw3D();
 314 }
 315
 316 /******************************************************************************/
 317
 318 //______________________________________________________________________________
 319 void AliEveCascadeList::FilterByDaughterDCA(Float_t minDaughterDCA, Float_t maxDaughterDCA)
 320 {
 321   // Select visibility of elements based on the DCA between daughters.
 322
 323   fMinDaughterDCA = minDaughterDCA;
 324   fMaxDaughterDCA = maxDaughterDCA;
 325
 326   for(List_i i = fChildren.begin(); i != fChildren.end(); ++i)
 327   {
 328     AliEveCascade* cascade = (AliEveCascade*) *i;
 329     Float_t  dca = cascade->GetDaughterDCA();
 330     Bool_t  show = dca >= fMinDaughterDCA && dca <= fMaxDaughterDCA;
 331     cascade->SetRnrState(show);
 332   }
 333   ElementChanged();
 334   gEve->Redraw3D();
 335 }
 336
 337 /******************************************************************************/
 338
 339 //______________________________________________________________________________
 340 void AliEveCascadeList::FilterByPt(Float_t minPt, Float_t maxPt)
 341 {
 342   // Select visibility of elements based on the Cascade pt.
 343
 344   fMinPt = minPt;
 345   fMaxPt = maxPt;
 346
 347   for(List_i i = fChildren.begin(); i != fChildren.end(); ++i)
 348   {
 349     AliEveCascade* cascade = (AliEveCascade*) *i;
 350     Float_t  pt = cascade->GetPt();
 351     Bool_t  show = pt >= fMinPt && pt <= fMaxPt;
 352     cascade->SetRnrState(show);
 353   }
 354   ElementChanged();
 355   gEve->Redraw3D();
 356 }
 357
 358 /******************************************************************************/
 359
 360 //______________________________________________________________________________
 361 void AliEveCascadeList::FilterByInvariantMass(Float_t minInvariantMass, Float_t maxInvariantMass, Int_t cascadePdgCodeHyp)
 362 {
 363   // Select visibility of elements based on the Cascade invariant mass.
 364
 365         fMinInvariantMass = minInvariantMass;
 366         fMaxInvariantMass = maxInvariantMass;
 367
 368         for(List_i i = fChildren.begin(); i != fChildren.end(); ++i)
 369         {
 370                 AliEveCascade* cascade = (AliEveCascade*) *i;
 371                 Float_t  invMass = (Float_t)cascade->GetInvMass(cascadePdgCodeHyp);
 372                 Bool_t   show    = invMass >= fMinInvariantMass && invMass <= fMaxInvariantMass;
 373                 cascade->SetRnrState(show);
 374         }
 375         ElementChanged();
 376         gEve->Redraw3D();
 377 }
 378