From Bogdan: Due to problems with color scheme in latest geometries use standard...

[u/mrichter/AliRoot.git] / EVE / alice-macros / esd_V0.C

// $Id$
// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007

/**************************************************************************
 * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for          *
 * full copyright notice.                                                 *
 **************************************************************************/

AliEveV0* esd_make_v0(TEveTrackPropagator* rnrStyle, AliESDVertex* primVtx,
		      AliESDtrack* neg, AliESDtrack* pos, AliESDv0* v0, Int_t i)
{
  if (! v0->GetOnFlyStatus())
  { // v0 on fly do not have the momentum vector filled...
    TEveRecTrack  rcPos;
    TEveRecTrack  rcNeg;
    TEveRecV0     rcV0;

    Double_t p[3];
    v0->GetNPxPyPz(p[0], p[1], p[2]);
    rcV0.P_pos.Set(p);
    v0->GetPPxPyPz(p[0], p[1], p[2]);
    rcV0.P_neg.Set(p);

    v0->GetPxPyPz(p[0], p[1], p[2]);
    Double_t v[3];
    v0->GetXYZ(v[0], v[1], v[2]);


    //   printf(" %f %f %f / %f %f %f    %i\n",p[0], p[1], p[2],
    // 	 v[0], v[1], v[2], v0->GetOnFlyStatus());

    rcV0.V_neg.Set(v); //original track vertices at dca not stored
    rcV0.V_pos.Set(v);
    rcV0.V_ca.Set(v);

    rcV0.d_label[0] = v0->GetNindex();
    rcV0.d_label[1] = v0->GetPindex();

    Double_t ep = neg->GetP(), mc = neg->GetMass();
    rcNeg.beta = ep/TMath::Sqrt(ep*ep + mc*mc);
    ep = pos->GetP(); mc = pos->GetMass();
    rcPos.beta = ep/TMath::Sqrt(ep*ep + mc*mc);


    AliEveV0* myV0 = new AliEveV0(&rcNeg, &rcPos, &rcV0, rnrStyle);
    char ch[50];
    //   sprintf(ch,"ESDv0%i",i);
    //   myV0->SetName(ch);
    //   myV0->SetTitle(ch);
    myV0->SetESDIndex(i);
    myV0->SetDaughterDCA(v0->GetDcaV0Daughters());

    Double_t primx = primVtx->GetXv(),
      primy = primVtx->GetYv(),
      primz = primVtx->GetZv();
    myV0->SetCosPointingAngle(v0->GetV0CosineOfPointingAngle(primx,primy,primz));

    myV0->SetDecayLength(primVtx->GetXv(), primVtx->GetYv(), primVtx->GetZv());

    return myV0;
  } else {
    return 0;
  }

}


V0List* esd_AliEveV0(Double_t min_pt=0.1, Double_t max_pt=100)
{
  printf("THIS SCRIPT DOES NOT WORK.\n"
	 "AliEveV0 classes have been temporarily removed.\n"
	 "They need to be cleaned up.\n");
  return;

  AliESDEvent* esd = AliEveEventManager::AssertESD();
  AliESDVertex* primVertex =(AliESDVertex*) esd->GetVertex();

  V0List* cont = new V0List("ESD v0");
  cont->SetMainColor(Color_t(3)); // green
  TEveTrackPropagator* rnrStyle = cont->GetPropagator();
  rnrStyle->SetMagField( 0.1*esd->GetMagneticField() );

  gEve->AddElement(cont);

  Int_t count = 0;
  //for (Int_t n=0; count<3; n++) {
  for (Int_t n=0; n<esd->GetNumberOfV0s(); n++) {

    AliESDv0 *v0 = esd->GetV0(n);
    if (v0->GetOnFlyStatus()) continue;

    Int_t negInd = v0->GetNindex();
    Int_t posInd = v0->GetPindex();
    AliESDtrack* negTr = esd->GetTrack(negInd);
    AliESDtrack* posTr = esd->GetTrack(posInd);

    AliEveV0* myV0 = esd_make_v0(rnrStyle, primVertex, negTr,posTr, v0, n);
    if (myV0) {
      gEve->AddElement(myV0, cont);
      count++;
    }
  }

  cont->SetTitle("testV0List ");
  cont->UpdateItems();

  cont->MakeV0s();
  gEve->Redraw3D();

  return cont;
}