]>
Commit | Line | Data |
---|---|---|
d810d0de | 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 * | |
51346b82 | 7 | * full copyright notice. * |
d810d0de | 8 | **************************************************************************/ |
ca5fb6a4 | 9 | |
b594a020 | 10 | void esd_v0_init_rectrack(TEveRecTrack& rt, AliExternalTrackParam* tp) |
11 | { | |
12 | Double_t pbuf[3], vbuf[3]; | |
13 | ||
14 | rt.fSign = tp->GetSign(); | |
15 | tp->GetXYZ(vbuf); rt.fV.Set(vbuf); | |
16 | tp->GetPxPyPz(pbuf); rt.fP.Set(pbuf); | |
17 | // Double_t ep = at->GetP(), mc = at->GetMass(); | |
18 | rt.fBeta = 1; // ep/TMath::Sqrt(ep*ep + mc*mc); | |
19 | } | |
20 | ||
d810d0de | 21 | AliEveV0* esd_make_v0(TEveTrackPropagator* rnrStyle, AliESDVertex* primVtx, |
a9f14204 | 22 | AliESDtrack* neg, AliESDtrack* pos, AliESDv0* v0, Int_t i) |
23 | { | |
ca8a6926 | 24 | TEveRecTrack rcPos; |
25 | TEveRecTrack rcNeg; | |
26 | TEveRecV0 rcV0; | |
27 | ||
28 | Double_t p[3]; | |
29 | v0->GetNPxPyPz(p[0], p[1], p[2]); | |
30 | rcV0.fPPos.Set(p); | |
31 | v0->GetPPxPyPz(p[0], p[1], p[2]); | |
32 | rcV0.fPNeg.Set(p); | |
33 | ||
34 | v0->GetPxPyPz(p[0], p[1], p[2]); | |
e6e8b28b | 35 | |
ca8a6926 | 36 | Double_t v[3]; |
e6e8b28b | 37 | |
ca8a6926 | 38 | v0->GetXYZ(v[0], v[1], v[2]); |
e6e8b28b | 39 | rcV0.fVCa.Set(v); |
ca8a6926 | 40 | |
e6e8b28b | 41 | v0->GetParamN()->GetXYZ(v); rcV0.fVNeg.Set(v); |
42 | v0->GetParamP()->GetXYZ(v); rcV0.fVPos.Set(v); | |
ca8a6926 | 43 | |
e6e8b28b | 44 | rcV0.fV0Birth.Set(primVtx->GetXv(), primVtx->GetYv(), primVtx->GetZv()); |
ca8a6926 | 45 | |
e6e8b28b | 46 | // Simulation data not directly available in AliESDv0 |
47 | //rcV0.fDLabel[0] = v0->GetNindex(); | |
48 | //rcV0.fDLabel[1] = v0->GetPindex(); | |
ca8a6926 | 49 | |
50 | esd_v0_init_rectrack(rcNeg, v0->GetParamN()); | |
e6e8b28b | 51 | rcNeg.fIndex = v0->GetNindex(); |
ca8a6926 | 52 | esd_v0_init_rectrack(rcPos, v0->GetParamP()); |
e6e8b28b | 53 | rcPos.fIndex = v0->GetPindex(); |
ca8a6926 | 54 | |
55 | AliEveV0* myV0 = new AliEveV0(&rcNeg, &rcPos, &rcV0, rnrStyle); | |
56 | myV0->SetElementName(Form("ESDv0 %d", i)); | |
57 | myV0->SetElementTitle(Form("OnFly: %d\nDCA %f", | |
58 | v0->GetOnFlyStatus(), | |
59 | v0->GetDcaV0Daughters())); | |
60 | myV0->SetESDIndex(i); | |
61 | myV0->SetOnFlyStatus(v0->GetOnFlyStatus()); | |
62 | myV0->SetDaughterDCA(v0->GetDcaV0Daughters()); | |
63 | ||
64 | return myV0; | |
ca5fb6a4 | 65 | } |
66 | ||
67 | ||
ca8a6926 | 68 | AliEveV0List* esd_V0(Bool_t onFly=kFALSE) |
ca5fb6a4 | 69 | { |
d810d0de | 70 | AliESDEvent* esd = AliEveEventManager::AssertESD(); |
ca8a6926 | 71 | |
72 | AliESDVertex* primVertex = (AliESDVertex*) esd->GetPrimaryVertex(); | |
ca5fb6a4 | 73 | |
b594a020 | 74 | AliEveV0List* cont = new AliEveV0List("ESD v0"); |
ca5fb6a4 | 75 | cont->SetMainColor(Color_t(3)); // green |
84aff7a4 | 76 | TEveTrackPropagator* rnrStyle = cont->GetPropagator(); |
daaa6c4d | 77 | rnrStyle->SetMagField( 0.1*esd->GetMagneticField() ); |
ca5fb6a4 | 78 | |
84aff7a4 | 79 | gEve->AddElement(cont); |
ca5fb6a4 | 80 | |
81 | Int_t count = 0; | |
ca8a6926 | 82 | for (Int_t n=0; n<esd->GetNumberOfV0s(); ++n) |
83 | { | |
ca5fb6a4 | 84 | AliESDv0 *v0 = esd->GetV0(n); |
ca8a6926 | 85 | |
86 | if (v0->GetOnFlyStatus() != onFly) continue; | |
ca5fb6a4 | 87 | |
88 | Int_t negInd = v0->GetNindex(); | |
89 | Int_t posInd = v0->GetPindex(); | |
90 | AliESDtrack* negTr = esd->GetTrack(negInd); | |
91 | AliESDtrack* posTr = esd->GetTrack(posInd); | |
51346b82 | 92 | |
d810d0de | 93 | AliEveV0* myV0 = esd_make_v0(rnrStyle, primVertex, negTr,posTr, v0, n); |
ca8a6926 | 94 | if (myV0) |
95 | { | |
84aff7a4 | 96 | gEve->AddElement(myV0, cont); |
ca8a6926 | 97 | ++count; |
ca5fb6a4 | 98 | } |
99 | } | |
100 | ||
b594a020 | 101 | cont->SetTitle("test"); |
ca5fb6a4 | 102 | cont->UpdateItems(); |
103 | ||
104 | cont->MakeV0s(); | |
84aff7a4 | 105 | gEve->Redraw3D(); |
ca5fb6a4 | 106 | |
107 | return cont; | |
108 | } | |
109 | ||
110 | ||
111 | ||
112 | ||
113 | ||
114 | ||
115 | ||
116 | ||
117 | ||
118 | ||
119 | ||
120 | ||
121 | ||
122 | ||
123 | ||
124 |