1 // Launches the list analyser and loads kinks into the list.
2 // If you already have a list (or rather a list analyser) with objects and you want to add the kinks to this list, do the following:
3 // Right-click the list analyser in the eve-browser and select "ExportToCint". Choose e.g. "list" for the name in the window that pops up.
4 // In the console type ".x ana_list_load_kinks.C(list)"
5 // For more information please see "ana_list.C" or have a look at the class documentation.
8 #if !defined(__CINT__) || defined(__MAKECINT__)
10 #include <TEveManager.h>
11 #include <TEveTrackPropagator.h>
13 #include <AliTRDarrayADC.h>
14 #include <AliTRDReconstructor.h>
15 #include <AliTRDtrackV1.h>
16 #include <AliESDkink.h>
17 #include <AliESDEvent.h>
18 #include <AliESDfriend.h>
19 #include <AliEveTRDData.h>
20 #include <AliEveEventManager.h>
21 #include <AliEveKink.h>
22 #include <AliEveListAnalyser.h>
25 void esd_kink_init_rectrackMother(TEveRecTrack& rt, const AliExternalTrackParam* tp)
27 Double_t pbuf[3], vbuf[3];
29 rt.fSign = tp->GetSign();
30 tp->GetXYZ(vbuf); rt.fV.Set(vbuf);
31 tp->GetPxPyPz(pbuf); rt.fP.Set(pbuf);
33 rt.fBeta = 1; // ep/TMath::Sqrt(ep*ep + mc*mc);
36 void esd_kink_init_rectrackDaughter(TEveRecTrack& rt, const AliExternalTrackParam* tp, TEveVector* svt,TEveVector* spt)
38 rt.fSign = tp->GetSign();
42 rt.fBeta = 1; // ep/TMath::Sqrt(ep*ep + mc*mc);
45 AliEveKink* esd_make_kink(TEveTrackPropagator* rnrStyleMoth,TEveTrackPropagator* rnrStyleDaugh, AliESDtrack* moth, AliESDtrack* daug, AliESDkink* kink, Int_t i)
51 const TVector3 p1(kink->GetMotherP());
52 rcKink.fPMother.Set(p1);
53 const TVector3 p2(kink->GetDaughterP());
54 rcKink.fPDaughter.Set(p2);
56 const TVector3 r1(kink->GetPosition());
57 rcKink.fVKink.Set(r1);
59 for (Int_t j=0; j<3; ++j) rcKink.fKinkAngle[j]=kink->GetAngle(j);
63 moth->GetTPCInnerParam()->GetXYZ(r2); rcKink.fVMother.Set(r2);
64 daug->GetOuterParam()->GetXYZ(r); rcKink.fVDaughter.Set(r);
66 esd_kink_init_rectrackMother(rcMoth, (moth->GetTPCInnerParam()));
67 rcMoth.fIndex = moth->GetID();
69 esd_kink_init_rectrackDaughter(rcDaug, daug->GetOuterParam(), &rcKink.fVKink, &rcKink.fPDaughter);
70 rcDaug.fIndex = daug->GetID();
72 AliEveKink* myKink = new AliEveKink(&rcMoth, &rcDaug, &rcKink, rnrStyleMoth,rnrStyleDaugh);
74 myKink->SetElementName(Form("ESDkink %d \n", i));
75 myKink->SetESDKinkIndex(i);
77 for (Int_t j=0; j<3; ++j) myKink->SetKinkAngle(j, kink->GetAngle(j));
78 Double_t daugProbability[10];
80 daug->GetESDpid(daugProbability);
83 // ****** Tentative particle type "concentrations"
84 Double_t c[5]={0.01, 0.01, 0.85, 0.10, 0.05};
87 AliPID daugPid(daugProbability);
89 Int_t daugMostProbPdg = 0;
91 switch (daugPid.GetMostProbable()){
93 daugMostProbPdg = 11; break;
95 daugMostProbPdg = 13; break;
97 daugMostProbPdg = 211; break;
99 daugMostProbPdg = 13; break;
102 Float_t daugMaxProbPid = daugPid.GetProbability(daugPid.GetMostProbable());
104 myKink->SetMaxProbPdgPid(daugMostProbPdg,daugMaxProbPid);//????????????
109 void ana_list_load_kinks(AliEveListAnalyser* objects = 0, TEveElement *cont = 0)
111 Bool_t noListProvided = kTRUE;
112 if (objects) noListProvided = kFALSE;
114 // Link data containers
115 AliESDfriend *eventESDfriend = 0x0;
116 if(!(eventESDfriend = AliEveEventManager::AssertESDfriend())){
117 Warning("ana_list_load_tracks", "AliESDfriend not found");
121 AliESDEvent* esd = AliEveEventManager::AssertESD();
123 AliEveEventManager::AssertGeometry();
125 if (!objects) objects = new AliEveListAnalyser("Analysis Objects");
128 AliEveKinkList* list = new AliEveKinkList("ESD kink");
129 list->SetMainColor(3); // green
130 TEveTrackPropagator* rnrStyleMoth = list->GetPropagatorMoth();
131 rnrStyleMoth->SetMagField( 0.1*esd->GetMagneticField() );
132 TEveTrackPropagator* rnrStyleDaugh = list->GetPropagatorDaugh();
133 rnrStyleDaugh->SetMagField( 0.1*esd->GetMagneticField() );
134 rnrStyleDaugh->SetMaxR(520);
135 //gEve->AddElement(list);
137 // Number of elements already in the list
138 Int_t nOld = objects->NumChildren();
142 for (Int_t n = 0; n < esd->GetNumberOfTracks(); ++n)
144 AliESDtrack* mtrack = esd->GetTrack(n);
145 if (!mtrack) continue;
146 // GetKinkIndex < 0 in the following!!! => -GetKinkIndex > 0
147 if(mtrack->GetKinkIndex(0) < 0)
149 AliESDkink *kink = new AliESDkink;
151 kink = esd->GetKink(-mtrack->GetKinkIndex(0) - 1);
154 for (Int_t m = 0; m < esd->GetNumberOfTracks(); ++m)
156 AliESDtrack * dtrack = esd->GetTrack(m);
157 if (!dtrack) continue;
159 if((dtrack->GetKinkIndex(0) > 0 ) && (dtrack->GetKinkIndex(0) == -mtrack->GetKinkIndex(0)))
161 AliESDtrack* mothTr = esd->GetTrack(n);
162 AliESDtrack* daugTr = esd->GetTrack(m);
163 if (!mothTr || !daugTr) continue;
165 AliEveKink* myKink = esd_make_kink(rnrStyleMoth, rnrStyleDaugh, mothTr, daugTr, kink, (-mtrack->GetKinkIndex(0)-1));
168 myKink->SetUserData(kink);
169 myKink->SetMarkerColor(3);
170 gEve->AddElement(myKink, list);
171 objects->AddElement(myKink);
172 myKink->SetName(Form("[%4d] kink", count + nOld));
176 } // inner track loop
177 } //mother kink index <0
178 } // Outer track loop
182 objects->SetTitle(Form("Objects %d", objects->NumChildren()));
183 objects->StampObjProps();
185 // If a new list analyser has been created, add it to eve.
186 if (noListProvided) gEve->AddElement(objects, cont);
190 // TGLViewer *v = gEve->GetDefaultGLViewer();
191 // v->SetCurrentCamera(TGLViewer::kCameraOrthoXOY);
192 // ((TGLOrthoCamera&)v->CurrentCamera()).SetEnableRotate(kTRUE);