* See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for *
* full copyright notice. *
**************************************************************************/
+#if !defined(__CINT__) || defined(__MAKECINT__)
+#include <TMath.h>
+#include <TEveManager.h>
+#include <TEveVSDStructs.h>
+#include <TEveTrackPropagator.h>
+
+#include <AliExternalTrackParam.h>
+#include <AliESDEvent.h>
+#include <AliESDcascade.h>
+#include <AliESDVertex.h>
+#include <AliEveCascade.h>
+#include <AliEveEventManager.h>
+#endif
+
+void esd_cascade_init_rectrack(TEveRecTrack& rt, const AliExternalTrackParam* tp)
+{
+ Double_t pbuf[3], vbuf[3];
-class AliEveCascade;
-class AliEveCascadeList;
+ rt.fSign = tp->GetSign();
+ tp->GetXYZ(vbuf); rt.fV.Set(vbuf);
+ tp->GetPxPyPz(pbuf); rt.fP.Set(pbuf);
+ // Double_t ep = at->GetP(), mc = at->GetMass();
+ rt.fBeta = 1; // ep/TMath::Sqrt(ep*ep + mc*mc);
+}
-AliEveCascade* esd_make_cas(TEveTrackPropagator* rnrStyle, AliESDVertex* primVtx,
- AliESDcascade* cas, AliESDtrack* neg, AliESDtrack* pos,
- AliESDtrack* bach,Int_t i)
+AliEveCascade* esd_make_cascade(TEveTrackPropagator* rnrStyle, AliESDVertex* primVtx,
+ AliESDtrack* bac, AliESDcascade* cascade, Int_t i)
{
- AliEveCascade* myCas = new AliEveCascade(rnrStyle);
- myCas->SetESDIndex(i);
-
- static Double_t vx,vy,vz, px,py,pz;
- cas->GetBPxPyPz(px,py,pz);
- myCas->SetBachP(px,py,pz);
- cas->GetNPxPyPz(px,py,pz);
- myCas->SetNegP(px,py,pz);
- cas->GetPPxPyPz(px,py,pz);
- myCas->SetPosP(px,py,pz);
-
- cas->GetXYZ(vx,vy,vz); // v0 decay vertex
- myCas->SetV0vtx(vx,vy,vz);
- cas->GetXYZcascade(vx,vy,vz); // cascade decay vertex
- myCas->SetCascadeVtx(vx,vy,vz);
-
- Double_t primx = primVtx->GetXv(),
- primy = primVtx->GetYv(),
- primz = primVtx->GetZv();
-
- myCas->SetCasCosPointingAngle( cas->GetCascadeCosineOfPointingAngle(primx,primy,primz) );
- myCas->SetDecayLength(primx, primy, primz);
- myCas->SetDCA_v0_Bach(cas->GetDcaXiDaughters());
-
- Float_t p = neg->GetP(), mc = neg->GetMass();
- Float_t betaNeg = p/TMath::Sqrt(p*p + mc*mc);
- p = pos->GetP(); mc = pos->GetMass();
- Float_t betaPos = p/TMath::Sqrt(p*p + mc*mc);
- p = bach->GetP(); mc = bach->GetMass();
- Float_t betaBach = p/TMath::Sqrt(p*p + mc*mc);
- if (bach->GetSign()<0) betaBach = -betaBach; // sign is stored is this parameter
-
- myCas->SetBeta(betaNeg, betaPos, betaBach);
-
- return myCas;
+ TEveRecTrack rcPos;
+ TEveRecTrack rcNeg;
+ TEveRecV0 rcV0;
+
+ TEveRecTrack rcBac;
+ TEveRecCascade rcCascade;
+
+ Double_t v[3];
+ cascade->GetXYZ(v[0], v[1], v[2]);
+ rcV0.fVCa.Set(v);
+
+ cascade->GetParamN()->GetXYZ(v); rcV0.fVNeg.Set(v);
+ cascade->GetParamP()->GetXYZ(v); rcV0.fVPos.Set(v);
+
+ rcV0.fV0Birth.Set(primVtx->GetX(), primVtx->GetY(), primVtx->GetZ());
+
+ Double_t pCascade[3]={0.}, pBac[3]={0.}, pNeg[3]={0.}, pPos[3]={0.}, cv[21]={0.};
+ //cascade->GetPxPyPz(pCascade[0], pCascade[1], pCascade[2]);
+ cascade->GetBPxPyPz(pBac[0], pBac[1], pBac[2]);
+ cascade->GetNPxPyPz(pNeg[0], pNeg[1], pNeg[2]);
+ cascade->GetPPxPyPz(pPos[0], pPos[1], pPos[2]);
+
+ rcCascade.fPBac.Set(pBac);
+ rcV0.fPNeg.Set(pNeg);
+ rcV0.fPPos.Set(pPos);
+ /*
+ // Debug
+ printf("\n ESD info \n");
+ printf("Neg : px = %.5f, py = %.5f, pz = %.5f \n",pNeg[0], pNeg[1], pNeg[2]);
+ printf("Pos : px = %.5f, py = %.5f, pz = %.5f \n",pPos[0], pPos[1], pPos[2]);
+ printf("Bach : px = %.5f, py = %.5f, pz = %.5f \n",pBac[0], pBac[1], pBac[2]);
+
+ printf("\n EVE info \n");
+ printf("Neg : px = %.5f, py = %.5f, pz = %.5f \n",rcV0.fPNeg.fX, rcV0.fPNeg.fY, rcV0.fPNeg.fZ);
+ printf("Pos : px = %.5f, py = %.5f, pz = %.5f \n",rcV0.fPPos.fX, rcV0.fPPos.fY, rcV0.fPPos.fZ);
+ printf("Bach : px = %.5f, py = %.5f, pz = %.5f \n",rcCascade.fPBac.fX,rcCascade.fPBac.fY,rcCascade.fPBac.fZ);
+ */
+ Double_t pLambda = TMath::Sqrt((pNeg[0]+pPos[0])* (pNeg[0]+pPos[0]) +
+ (pNeg[1]+pPos[1])* (pNeg[1]+pPos[1]) +
+ (pNeg[2]+pPos[2])* (pNeg[2]+pPos[2]) );
+ Double_t pBach = TMath::Sqrt( pBac[0]*pBac[0] + pBac[1]*pBac[1] + pBac[2]*pBac[2]);
+
+ cascade->GetXYZcascade(v[0], v[1], v[2]);
+ rcCascade.fCascadeVCa.Set(v);
+
+ rcCascade.fCascadeBirth.Set(primVtx->GetX(), primVtx->GetY(), primVtx->GetZ());
+
+ // Simulation data not directly available in AliESDcascade
+ // rcCascade.fDLabel = cascade->GetBindex();
+
+ // Problem: two following lines are not possible: no GetParamB !!
+ // cascade->GetParamB()->GetXYZ(v); rcCascade.fVBac.Set(v);
+ // esd_cascade_init_rectrack(rcBac, cascade->GetParamB());
+ // Solution: create an AliExternalTrackParam with null cv...
+ AliExternalTrackParam *bParam = new AliExternalTrackParam(v,pBac,cv,cascade->Charge());
+ esd_cascade_init_rectrack(rcBac,bParam);
+ rcBac.fIndex = cascade->GetBindex();
+
+ esd_cascade_init_rectrack(rcNeg, cascade->GetParamN());
+ rcNeg.fIndex = cascade->GetNindex();
+ esd_cascade_init_rectrack(rcPos, cascade->GetParamP());
+ rcPos.fIndex = cascade->GetPindex();
+
+
+/*
+ // Debug
+ TEveVector BacMom = rcBac.GetMomentum();
+ printf("Bac mom : px = %f, py = %f, pz =%f, Ptot = %f \n", BacMom.fX,
+ rcBac.GetMomentum().fY,
+ BacMom.fZ,
+ BacMom.Mag());
+*/
+ AliEveCascade* myCascade = new AliEveCascade(&rcBac, &rcNeg, &rcPos, &rcV0, &rcCascade, rnrStyle);
+ myCascade->SetElementName(Form("ESDcascade %d", i));
+
+ /*
+ // Debug
+ TEveVector CascMom(rcCascade.fPBac + rcV0.fPNeg + rcV0.fPPos);
+ printf("Casc mom : px = %f, py = %f, pz =%f, Ptot = %f \n", CascMom.fX, CascMom.fY, CascMom.fZ, CascMom.Mag());
+ */
+ myCascade->SetElementTitle(Form("Info coming directly from AliESDcascade : \n - Charge : %d \n - Cascade decay position : x = %.4f, y = %.4f, z = %.4f, Transv. radius = %.4f cm, Decay Length = %.4f cm\n\n - Pt(Cascade) : %f GeV/c, Ptot(Cascade): %f GeV/c\n - Lambda : px = %.4f, py = %.4f, pz = %.4f, Ptot : %f GeV/c\n - Bach : px = %.4f, py = %.4f, pz = %.4f, Ptot : %f GeV/c\n\n - Eta : %f\n - Phi : %f deg \n - Theta : %f deg\n - DCA : %f cm \n - Cos(Ptg Angle) : %f \n\n - Eff. mass (Xi hyp) : %f GeV/c2",
+ cascade->Charge(),
+ v[0], v[1], v[2], TMath::Sqrt(v[0]*v[0] +v[1]*v[1]), TMath::Sqrt(v[0]*v[0] +v[1]*v[1] + v[2]*v[2] ),
+ cascade->Pt(), cascade->P(),
+ pNeg[0]+pPos[0], pNeg[1]+pPos[1], pNeg[2]+pPos[2], pLambda,
+ pBac[0], pBac[1], pBac[2], pBach,
+ cascade->Eta(),
+ cascade->Phi() * 180/TMath::Pi(),
+ cascade->Theta() * 180/TMath::Pi(),
+ cascade->GetDcaXiDaughters(),
+ cascade->GetCascadeCosineOfPointingAngle(primVtx->GetX(),
+ primVtx->GetY(),
+ primVtx->GetZ()),
+ cascade->GetEffMassXi()
+ )
+ );
+
+
+ myCascade->SetESDIndex(i);
+ myCascade->SetDaughterDCA(cascade->GetDcaXiDaughters());
+ myCascade->SetLambdaP( pNeg[0]+pPos[0], pNeg[1]+pPos[1], pNeg[2]+pPos[2] );
+ myCascade->SetBachP( pBac[0], pBac[1], pBac[2]);
+ return myCascade;
}
-AliEveCascadeList* esd_cascade(Double_t min_pt=0.1, Double_t max_pt=100)
+AliEveCascadeList* esd_cascade()
{
- printf("THIS SCRIPT DOES NOT WORK.\n"
- "AliEveCascade classes have been temporarily removed.\n"
- "They need to be cleaned up.\n");
- return;
-
AliESDEvent* esd = AliEveEventManager::AssertESD();
- AliESDVertex* primVertex =(AliESDVertex*) esd->GetVertex();
- CascadeList* cont = new CascadeList("ESD cascade");
- cont->SetMainColor(3); // green
+ AliESDVertex* primVertex = (AliESDVertex*) esd->GetPrimaryVertex();
+
+ AliEveCascadeList* cont = new AliEveCascadeList("ESD cascade");
+ cont->SetMainColor(kBlue+2);
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->GetNumberOfCascades(); n++) {
-
- AliESDcascade *cas = esd->GetCascade(n);
- Int_t negInd = cas->GetNindex();
- Int_t posInd = cas->GetPindex();
- Int_t bachInd = cas->GetBindex();
- AliESDtrack* negTr = esd->GetTrack(negInd);
- AliESDtrack* posTr = esd->GetTrack(posInd);
- AliESDtrack* bachTr = esd->GetTrack(bachInd);
-
- if (cas) {
- AliEveCascade* myCas = esd_make_cas(rnrStyle, primVertex, cas,
- negTr, posTr, bachTr, n);
- if (myCas) {
- gEve->AddElement(myCas, cont);
- count++;
- }
+ for (Int_t n=0; n<esd->GetNumberOfCascades(); ++n)
+ {
+ AliESDcascade *cascade = esd->GetCascade(n);
+
+ Int_t bacInd = cascade->GetBindex();
+ AliESDtrack* bacTr = esd->GetTrack(bacInd);
+
+ AliEveCascade* myCascade = esd_make_cascade(rnrStyle, primVertex, bacTr, cascade, n);
+ if (myCascade)
+ {
+ gEve->AddElement(myCascade, cont);
+ ++count;
}
}
- cont->SetTitle("CascadeList");
+ cont->SetTitle("Cascade candidates (reco)");
cont->MakeCascades();
gEve->Redraw3D();
git://git.uio.no / u / mrichter / AliRoot.git / blobdiff