EVE/alice-macros/esd_Cascade.C

   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 AliEveCascade* esd_make_cas(TEveTrackPropagator* rnrStyle, AliESDVertex* primVtx,
  11                             AliESDcascade* cas, AliESDtrack* neg, AliESDtrack* pos,
  12                             AliESDtrack* bach,Int_t i) {
  13
  14   AliEveCascade* myCas = new AliEveCascade(rnrStyle);
  15   myCas->SetESDIndex(i);
  16
  17   static Double_t vx,vy,vz, px,py,pz;
  18   cas->GetBPxPyPz(px,py,pz);
  19   myCas->SetBachP(px,py,pz);
  20   cas->GetNPxPyPz(px,py,pz);
  21   myCas->SetNegP(px,py,pz);
  22   cas->GetPPxPyPz(px,py,pz);
  23   myCas->SetPosP(px,py,pz);
  24
  25   cas->GetXYZ(vx,vy,vz); // v0 decay vertex
  26   myCas->SetV0vtx(vx,vy,vz);
  27   cas->GetXYZcascade(vx,vy,vz); // cascade decay vertex
  28   myCas->SetCascadeVtx(vx,vy,vz);
  29
  30   Double_t primx = primVtx->GetXv(),
  31     primy = primVtx->GetYv(),
  32     primz = primVtx->GetZv();
  33
  34   myCas->SetCasCosPointingAngle( cas->GetCascadeCosineOfPointingAngle(primx,primy,primz) );
  35   myCas->SetDecayLength(primx, primy, primz);
  36   myCas->SetDCA_v0_Bach(cas->GetDcaXiDaughters());
  37
  38   Float_t p = neg->GetP(), mc = neg->GetMass();
  39   Float_t betaNeg = p/TMath::Sqrt(p*p + mc*mc);
  40   p = pos->GetP(); mc = pos->GetMass();
  41   Float_t betaPos = p/TMath::Sqrt(p*p + mc*mc);
  42   p = bach->GetP(); mc = bach->GetMass();
  43   Float_t betaBach = p/TMath::Sqrt(p*p + mc*mc);
  44   if (bach->GetSign()<0) betaBach = -betaBach; // sign is stored is this parameter
  45
  46   myCas->SetBeta(betaNeg, betaPos, betaBach);
  47
  48   return myCas;
  49 }
  50
  51
  52 CascadeList* esd_AliEveCascade(Double_t min_pt=0.1, Double_t max_pt=100)
  53 {
  54   printf("THIS SCRIPT DOES NOT WORK.\n"
  55          "AliEveCascade classes have been temporarily removed.\n"
  56          "They need to be cleaned up.\n");
  57   return;
  58
  59   AliESDEvent* esd = AliEveEventManager::AssertESD();
  60   AliESDVertex* primVertex =(AliESDVertex*) esd->GetVertex();
  61
  62   CascadeList* cont = new CascadeList("ESD cascade");
  63   cont->SetMainColor(3); // green
  64   TEveTrackPropagator* rnrStyle = cont->GetPropagator();
  65   rnrStyle->SetMagField( 0.1*esd->GetMagneticField() );
  66
  67   gEve->AddElement(cont);
  68
  69   Int_t count = 0;
  70   //for (Int_t n=0; count<3; n++) {
  71   for (Int_t n=0; n<esd->GetNumberOfCascades(); n++) {
  72
  73     AliESDcascade *cas = esd->GetCascade(n);
  74     Int_t negInd = cas->GetNindex();
  75     Int_t posInd = cas->GetPindex();
  76     Int_t bachInd = cas->GetBindex();
  77     AliESDtrack* negTr = esd->GetTrack(negInd);
  78     AliESDtrack* posTr = esd->GetTrack(posInd);
  79     AliESDtrack* bachTr = esd->GetTrack(bachInd);
  80
  81     if (cas) {
  82       AliEveCascade* myCas = esd_make_cas(rnrStyle, primVertex, cas,
  83                                           negTr, posTr, bachTr, n);
  84       if (myCas) {
  85         gEve->AddElement(myCas, cont);
  86         count++;
  87       }
  88     }
  89   }
  90
  91   cont->SetTitle("CascadeList");
  92
  93   cont->MakeCascades();
  94   gEve->Redraw3D();
  95
  96   return cont;
  97 }