1ce0fedf44e04599534d949463d924a1b941bf4d
[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveTRDData.cxx
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 #include "TROOT.h"
11 #include "TStyle.h"
12 #include "TVector.h"
13 #include "TLinearFitter.h"
14 #include "TCanvas.h"
15 #include "TGeoMatrix.h"
16 #include "TGeoManager.h"
17
18 #include "TEveTrans.h"
19 #include "TEveManager.h"
20
21 #include "EveBase/AliEveEventManager.h"
22
23 #include "AliEveTRDData.h"
24 #include "AliEveTRDModuleImp.h"
25 #include "AliEveTRDLoader.h"
26 #include "AliEveTRDLoaderImp.h"
27
28 #include "AliGeomManager.h"
29 #include "AliESDtrack.h"
30 #include "AliLog.h"
31 #include "AliPID.h"
32 #include "AliTrackPointArray.h"
33 #include "AliRieman.h"
34
35 #include "AliTRDhit.h"
36 #include "AliTRDcluster.h"
37 #include "AliTRDseedV1.h"
38 #include "AliTRDtrackletMCM.h"
39 #include "AliTRDtrackletWord.h"
40 #include "AliTRDmcmSim.h"
41 #include "AliTRDtrackV1.h"
42 #include "AliTRDtrackerV1.h"
43 #include "AliTRDpadPlane.h"
44 #include "AliTRDdigitsManager.h"
45 #include "AliTRDmcmSim.h"
46 #include "AliTRDarrayADC.h"
47 #include "AliTRDSignalIndex.h"
48 #include "AliTRDgeometry.h"
49 #include "AliTRDtransform.h"
50 #include "AliTRDReconstructor.h"
51 #include "AliTRDrecoParam.h"
52
53 ClassImp(AliEveTRDHits)
54 ClassImp(AliEveTRDDigits)
55 ClassImp(AliEveTRDClusters)
56 ClassImp(AliEveTRDTracklet)
57 ClassImp(AliEveTRDTrack)
58 ClassImp(AliEveTRDTrackletOnline)
59 ClassImp(AliEveTRDmcm)
60
61 ///////////////////////////////////////////////////////////
62 /////////////   AliEveTRDDigits       /////////////////////
63 ///////////////////////////////////////////////////////////
64
65 //______________________________________________________________________________
66 AliEveTRDDigits::AliEveTRDDigits(AliEveTRDChamber *p) 
67   :TEveQuadSet("digits", "")
68   ,fParent(p)
69 {
70   // Constructor.
71   SetOwnIds(kTRUE);
72   gStyle->SetPalette(1, 0);
73   SetPalette(new TEveRGBAPalette(0, 512));
74   Reset(TEveQuadSet::kQT_RectangleYZ, kFALSE, 32);
75 }
76
77 //______________________________________________________________________________
78 AliEveTRDDigits::~AliEveTRDDigits()
79 {
80 //  AliInfo(GetTitle());
81 }
82
83 //______________________________________________________________________________
84 void AliEveTRDDigits::SetData(AliTRDdigitsManager *digits)
85 {
86   // Set data source.
87
88   Int_t det(fParent->GetID());
89   AliTRDarrayADC *data = digits->GetDigits(det);
90   if(!data->GetDim()) return;
91   data->Expand();
92
93   AliTRDSignalIndex *indexes = digits->GetIndexes(det);
94   if(!indexes->IsAllocated()) digits->BuildIndexes(det);
95
96   Double_t scale, dy, dz;
97   Int_t ly    = AliTRDgeometry::GetLayer(det),
98         stk   = AliTRDgeometry::GetStack(det),
99         sec   = AliTRDgeometry::GetSector(det),
100         vid   = AliGeomManager::LayerToVolUID(AliGeomManager::kTRD1 + ly, stk + AliTRDgeometry::Nstack() * sec);
101   SetNameTitle(Form("digits%03d", det), Form("D-%03d [%02d_%d_%d]", det, sec, stk, ly));
102   Short_t sig[7]={0,0,0,10,0,0,0};
103
104   AliTRDtransform transform(det);
105   AliTRDpadPlane *pp(fParent->fGeo->GetPadPlane(ly, stk));
106
107   Int_t row, col;
108   AliTRDcluster c;
109   indexes->ResetCounters();
110   while (indexes->NextRCIndex(row, col)){
111     dz = pp->GetRowSize(row);
112     dy = pp->GetColSize(col);
113     Short_t *const adc = data->GetDataAddress(row,col);
114     for (Int_t time(0); time<30; time++){     
115       if(data->IsPadCorrupted(row, col, time)) break;
116       if(adc[time] <= 1) continue;
117       new (&c) AliTRDcluster(det, col, row, time, sig, vid);
118       transform.Transform(&c);
119
120       scale = adc[time] < 512 ? adc[time]/512. : 1.;
121       AddQuad(c.GetY()-0.5*dy, c.GetZ()-0.5*dz*scale, c.GetX(), dy*0.95, dz*scale);
122       QuadValue(Int_t(adc[time]));
123       QuadId(new TNamed(Form("ADC %d", adc[time]), Form("det[%3d(%02d_%d_%d)] col[%3d] row[%2d] tb[%2d]", det, sec, stk, ly, col, row, time)));
124     } 
125   }
126
127   // rotate to global coordinates
128   RefitPlex();
129   TEveTrans& t = RefMainTrans();
130   t.SetRotByAngles((sec+.5)*AliTRDgeometry::GetAlpha(), 0.,0.);
131 }
132
133
134 // //______________________________________________________________________________
135 // void AliEveTRDDigits::Paint(Option_t *option)
136 // {
137 //   // Paint the object.
138 // 
139 //   if(fParent->GetDigitsBox()) fBoxes.Paint(option);
140 //   else TEveQuadSet::Paint(option);
141 // }
142
143 // //______________________________________________________________________________
144 // void AliEveTRDDigits::Reset()
145 // {
146 //   // Reset raw and visual data.
147 // 
148 //   TEveQuadSet::Reset(TEveQuadSet::kQT_RectangleYZ, kTRUE, 64);
149 //   // MT fBoxes.fBoxes.clear();
150 //   fData.Reset();
151 // }
152
153 ///////////////////////////////////////////////////////////
154 /////////////   AliEveTRDHits         /////////////////////
155 ///////////////////////////////////////////////////////////
156
157 //______________________________________________________________________________
158 AliEveTRDHits::AliEveTRDHits() : TEvePointSet("hits", 20)
159 {
160   // Constructor.
161   SetMarkerSize(.1);
162   SetMarkerColor(kGreen);
163   SetOwnIds(kTRUE);
164 }
165
166 //______________________________________________________________________________
167 AliEveTRDHits::~AliEveTRDHits()
168 {
169   //AliInfo(GetTitle());
170 }
171
172 //______________________________________________________________________________
173 void AliEveTRDHits::PointSelected(Int_t n)
174 {
175   // Handle an individual point selection from GL.
176
177   AliTRDhit *h = NULL;
178   if(!(h = dynamic_cast<AliTRDhit*>(GetPointId(n)))) return;
179   printf("Id[%3d] Det[%3d] Reg[%c] TR[%c] Q[%3d] MC[%d] t[%f]\n", 
180     n, h->GetDetector(), 
181     h->FromAmplification() ? 'A' : 'D', 
182     h->FromTRphoton() ? 'y' : 'n', 
183     h->GetCharge(), h->GetTrack(), h->GetTime());
184 }
185
186
187 ///////////////////////////////////////////////////////////
188 /////////////   AliEveTRDClusters         /////////////////////
189 ///////////////////////////////////////////////////////////
190
191 //______________________________________________________________________________
192 AliEveTRDClusters::AliEveTRDClusters():AliEveTRDHits()
193 {
194   // Constructor.
195   SetName("clusters");
196
197   SetMarkerSize(.4);
198   SetMarkerStyle(24);
199   SetMarkerColor(kGray);
200   SetOwnIds(kTRUE);
201 }
202
203 //______________________________________________________________________________
204 void AliEveTRDClusters::PointSelected(Int_t n)
205 {
206   // Handle an individual point selection from GL.
207
208   AliTRDcluster *c = dynamic_cast<AliTRDcluster*>(GetPointId(n));
209   if(!c) return;
210   c->Print();
211   Emit("PointSelected(Int_t)", n);
212   // Bool_t     AliCluster::GetGlobalCov(Float_t* cov) const
213   // Bool_t     AliCluster::GetGlobalXYZ(Float_t* xyz) const
214   // Float_t    AliCluster::GetSigmaY2() const
215   // Float_t    AliCluster::GetSigmaYZ() const
216   // Float_t    AliCluster::GetSigmaZ2() const
217 }
218
219 //______________________________________________________________________________
220 void AliEveTRDClusters::Print(Option_t *o) const
221 {
222   AliTRDcluster *c = NULL;
223
224   for(Int_t n = GetN(); n--;){
225     if(!(c = dynamic_cast<AliTRDcluster*>(GetPointId(n)))) continue;
226     c->Print(o);
227   }
228 }
229
230 //______________________________________________________________________________
231 void AliEveTRDClusters::Load(const Char_t *w) const
232 {
233   Int_t typ = -1;
234   if(strcmp(w, "hit")==0) typ = 0;
235   else if(strcmp(w, "dig")==0) typ = 1;
236   else if(strcmp(w, "cls")==0) typ = 2;
237   else if(strcmp(w, "tlt")==0) typ = 3;
238   else if(strcmp(w, "all")==0) typ = 4;
239   else{
240     AliInfo("The following arguments are accepted:");
241     AliInfo("   \"hit\" : loading of MC hits");
242     AliInfo("   \"dig\" : loading of digits");
243     AliInfo("   \"cls\" : loading of reconstructed clusters");
244     AliInfo("   \"tlt\" : loading of online tracklets");
245     AliInfo("   \"all\" : loading everything available");
246     return;
247   }
248
249   AliTRDcluster *c = NULL;
250   Int_t n = 0;
251   while((n = GetN() && !(c = dynamic_cast<AliTRDcluster*>(GetPointId(n))))) n++;
252   if(!c) return;
253
254   Int_t det = c->GetDetector();
255   AliEveTRDLoader *loader(NULL);
256   switch(typ){
257   case 4:
258     loader = new AliEveTRDLoaderSim("MC");
259     if(!loader->Open("galice.root")){delete loader; loader=NULL;}
260     else{
261       loader->SetDataType(AliEveTRDLoader::kTRDHits | AliEveTRDLoader::kTRDDigits | AliEveTRDLoader::kTRDClusters);
262       break;
263     }
264   case 0:  
265     loader = new AliEveTRDLoader("Hits");
266     if(!loader->Open("TRD.Hits.root")){delete loader; loader=NULL;}
267     else{
268       if(typ!=4) break;
269     }
270   case 1:
271     if(!loader) loader = new AliEveTRDLoader("Digits");
272     if(!loader->Open("TRD.Digits.root")){
273       if(typ==1){delete loader; loader=NULL;}
274     } else {
275       if(typ!=4) break;
276     }
277   case 2:
278     if(!loader) loader = new AliEveTRDLoader("Clusters");
279     if(!loader->Open("TRD.RecPoints.root")){
280       if(typ ==2){delete loader; loader=NULL;}
281     } else {
282       if(typ!=4) break;
283     }
284   case 3:
285     if(!loader) loader = new AliEveTRDLoader("Tracklets");
286     if(!loader->Open("TRD.Tracklets.root")){
287       if(typ ==3) {delete loader; loader=NULL;}
288     } else {
289       break;
290     }
291   default: return;
292   }
293   if(!loader) return;
294
295   loader->AddChambers(AliTRDgeometry::GetSector(det),AliTRDgeometry::GetStack(det), AliTRDgeometry::GetLayer(det));
296   // load first event
297   if(!loader->GoToEvent(AliEveEventManager::GetCurrent()->GetEventId())) AliWarning(Form("No data loaded for event %d", AliEveEventManager::GetCurrent()->GetEventId()));
298   // register loader with alieve
299   gEve->AddElement(loader->GetChamber(det), *(BeginParents()));
300   //loader->SpawnEditor();
301   gEve->Redraw3D();
302 }
303
304 ///////////////////////////////////////////////////////////
305 /////////////   AliEveTRDTracklet         /////////////////////
306 ///////////////////////////////////////////////////////////
307
308 //______________________________________________________________________________
309 AliEveTRDTracklet::AliEveTRDTracklet(AliTRDseedV1 *trklt):TEveLine()
310   ,fClusters(NULL)
311 {
312   // Constructor.
313   SetName("tracklet");
314   
315   if(!gGeoManager) AliEveEventManager::AssertGeometry();
316   SetUserData(trklt);
317   // init tracklet line
318   Int_t sec = AliTRDgeometry::GetSector(trklt->GetDetector());
319   Float_t alpha((0.5+sec)*AliTRDgeometry::GetAlpha()),
320           cphi(TMath::Cos(alpha)),
321           sphi(TMath::Sin(alpha));
322   Float_t dx;
323   Float_t x0   = trklt->GetX0();
324   Float_t y0   = trklt->GetYref(0);
325   Float_t z0   = trklt->GetZref(0);
326   Float_t dydx = trklt->GetYref(1);
327   Float_t dzdx = trklt->GetZref(1);
328   Float_t  tilt = trklt->GetTilt();
329   Float_t g[3];
330   AliTRDcluster *c = NULL;
331   for(Int_t ic=0; ic<AliTRDseedV1::kNclusters; ic++){
332     if(!(c = trklt->GetClusters(ic))) continue;
333     if(!fClusters) AddElement(fClusters = new AliEveTRDClusters());
334     dx = x0 - c->GetX();
335     //Float_t yt = y0 - dx*dydx;
336     Float_t zt = z0 - dx*dzdx;
337     // backup yc - for testing purposes
338     Float_t yc = c->GetY(); 
339     c->SetY(yc-tilt*(c->GetZ()-zt));
340     
341     // Ben: Temporary(?) fix -> Issue with wrongly set flag "fIsMisaligned" of AliCluster
342     // Changes can be undone ofter the issue with the flag has been solved
343     //c->GetGlobalXYZ(g);
344     
345     // The following code for the trasformation local-to-global is just adapted from
346     // AliTRDgeometry::RotateBack(...)
347   
348     // tracking to global coordinates transformation
349     Float_t loc[3] = { c->GetX(), c->GetY(), c->GetZ() }; 
350     g[0] = loc[0] * cphi - loc[1] * sphi;
351     g[1] = loc[0] * sphi + loc[1] * cphi;
352     g[2] = loc[2];
353     // Ben: End of fix
354     Int_t id = fClusters->SetNextPoint(g[0], g[1], g[2]);
355     c->SetY(yc);
356     fClusters->SetPointId(id, new AliTRDcluster(*c));
357   } 
358   if(fClusters){
359     fClusters->SetName("TC clusters");
360     fClusters->SetTitle(Form("N[%d]", trklt->GetN()));
361     fClusters->SetMarkerColor(kMagenta);
362   }
363
364   SetTitle(Form("Det[%d] RC[%c] Layer[%d] P[%7.3f]", trklt->GetDetector(), trklt->IsRowCross()?'y':'n', trklt->GetPlane(), trklt->GetMomentum()));
365   SetLineColor(kRed);
366   //SetOwnIds(kTRUE);
367   
368   //trklt->Fit(kTRUE);
369   y0   = trklt->GetYfit(0);
370   dydx = trklt->GetYfit(1);
371   Double_t xg(x0 * cphi - y0 * sphi),
372            yg(x0 * sphi + y0 * cphi);
373   SetPoint(0, xg, yg, z0); 
374   //SetPoint(0, x0, y0, z0);
375
376
377   dx = .5*AliTRDgeometry::CamHght()+AliTRDgeometry::CdrHght();
378   x0 -= dx; 
379   y0 -= dydx*dx,
380   z0 -= dzdx*dx; 
381   xg = x0 * cphi - y0 * sphi;
382   yg = x0 * sphi + y0 * cphi;
383   SetPoint(1, xg, yg, z0);
384   //SetPoint(1, x0, y0, z0);
385 }
386
387
388 AliEveTRDTracklet::~AliEveTRDTracklet() 
389 {
390
391 }
392
393 //______________________________________________________________________________
394 void AliEveTRDTracklet::Print(Option_t *o) const
395 {
396   AliTRDseedV1 *tracklet = (AliTRDseedV1*)GetUserData();
397   if(!tracklet) return;
398   tracklet->Print(o);
399 }
400
401 ///////////////////////////////////////////////////////////
402 /////////////   AliEveTRDTrack         /////////////////////
403 ///////////////////////////////////////////////////////////
404
405 //______________________________________________________________________________
406 AliEveTRDTrack::AliEveTRDTrack(AliTRDtrackV1 *trk) 
407   :TEveLine()
408   ,fTrackState(0)
409   ,fESDStatus(0)
410   ,fAlpha(0.)
411   ,fPoints(NULL)
412   ,fRim(NULL)
413 {
414   // Constructor.
415   SetUserData(trk);
416   SetName("");
417
418 //  AliTRDtrackerV1::SetNTimeBins(24);
419
420   fRim = new AliRieman(trk->GetNumberOfClusters());
421   AliTRDseedV1 *tracklet = NULL;
422   for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++){
423     if(!(tracklet = trk->GetTracklet(il))) continue;
424     if(!tracklet->IsOK()) continue;
425     AddElement(new AliEveTRDTracklet(tracklet));
426
427 //     tracklet->ResetClusterIter(kFALSE);
428 //     while((c = tracklet->PrevCluster())){
429 //    AliTRDcluster *c(NULL);
430 /*    for(Int_t ic=AliTRDseedV1::kNtb; ic--;){
431       if(!(c=tracklet->GetClusters(ic))) continue;
432       Float_t xc = c->GetX();
433       Float_t yc = c->GetY();
434       Float_t zc = c->GetZ();
435       Float_t zt = tracklet->GetZref(0) - (tracklet->GetX0()-xc)*tracklet->GetZref(1); 
436       yc -= tracklet->GetTilt()*(zc-zt);
437       fRim->AddPoint(xc, yc, zc, .05, 2.3);
438     }*/
439   }
440   if(trk->GetNumberOfTracklets()>1) fRim->Update();
441   SetStatus(fTrackState);
442 }
443
444 //______________________________________________________________________________
445 AliEveTRDTrack::~AliEveTRDTrack()
446 {
447   if(fPoints) delete [] fPoints; fPoints = NULL;
448   //delete dynamic_cast<AliTRDtrackV1*>(GetUserData());
449 }
450
451 //______________________________________________________________________________
452 void AliEveTRDTrack::Print(Option_t *o) const
453 {
454   AliTRDtrackV1 *track = (AliTRDtrackV1*)GetUserData();
455   if(!track) return;
456   track->Print(o);
457 }
458
459 //______________________________________________________________________________
460 void AliEveTRDTrack::SetStatus(UChar_t s)
461 {
462   // nothing to be done
463   if(fPoints && fTrackState == s) return;
464   //return;
465   const Int_t nc = AliTRDtrackV1::kMAXCLUSTERSPERTRACK;
466   AliTRDtrackV1 *trk(NULL);
467   if(!(trk=static_cast<AliTRDtrackV1*>(GetUserData()))) {
468     AliError("Failed casting data to TRD track.");
469     return;
470   }
471
472   Bool_t BUILD = kFALSE;
473   if(!fPoints){ 
474     fPoints = new AliTrackPoint[nc];
475
476     // define the radial span of the track in the TRD
477     Double_t xmin = -1., xmax = -1.;
478     Int_t det = 0;
479     AliTRDseedV1 *trklt = NULL;
480     for(Int_t ily=0; ily<AliTRDgeometry::kNlayer; ily++){
481       if(!(trklt = trk->GetTracklet(ily))) continue;
482       if(xmin<0.) xmin = trklt->GetX0() - AliTRDgeometry::CamHght() - AliTRDgeometry::CdrHght();
483       if(trklt->GetX0()>xmax) xmax = trklt->GetX0();
484       det = trklt->GetDetector();
485     }
486     Int_t sec = det/AliTRDgeometry::kNdets;
487     fAlpha = AliTRDgeometry::GetAlpha() * (sec<9 ? sec + .5 : sec - 17.5); //trk->GetAlpha()
488
489     Double_t dx =(xmax - xmin)/nc;
490     for(Int_t ip=0; ip<nc; ip++){
491       fPoints[ip].SetXYZ(xmin, 0., 0.);
492       xmin+=dx;
493     }
494     BUILD = kTRUE;
495   }
496
497   // select track model
498   if(BUILD || ((s&12) != (fTrackState&12))){
499     if(TESTBIT(s, kTrackCosmics)){
500       //printf("Straight track\n");
501       AliTRDtrackerV1::FitLine(trk, NULL, kFALSE, nc, fPoints);
502     } else {
503       if(TESTBIT(s, kTrackModel)){
504         //printf("Kalman track\n");
505         if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitKalman(trk, NULL, kFALSE, nc, fPoints);
506       } else { 
507         //printf("Rieman track rim[%p] nc[%d]\n", (void*)fRim, nc);
508         //if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitRiemanTilt(trk, NULL, kTRUE, nc, fPoints);
509         Float_t x = 0.;
510         for(Int_t ip = nc; ip--;){
511           x = fPoints[ip].GetX();
512           //printf("%2d x[%f] y[%f] z[%f]\n", ip, x, fRim->GetYat(x), fRim->GetZat(x));
513           fPoints[ip].SetXYZ(x, fRim->GetYat(x), fRim->GetZat(x));
514         }
515       }
516     }
517   
518     Float_t global[3];
519     for(Int_t ip=0; ip<nc; ip++){
520       fPoints[ip].Rotate(-fAlpha).GetXYZ(global);
521       SetPoint(ip, global[0], global[1], global[2]);
522       //printf("*** %2d x[%f] y[%f] z[%f]\n", ip, global[0], global[1], global[2]);
523     
524     }
525     SetSmooth(kTRUE);
526   }
527
528   // set color
529   if(BUILD || ((s&3) != (fTrackState&3))){
530     if(TESTBIT(s, kSource)){
531       //printf("Source color\n");
532       if(fESDStatus&AliESDtrack::kTRDin){
533         SetMarkerColor(kGreen);
534         SetLineColor(kGreen);
535       } else {
536         SetMarkerColor(kMagenta);
537         SetLineColor(kMagenta);
538       }
539     } else {
540       if(TESTBIT(s, kPID) == AliTRDpidUtil::kLQ){
541         //printf("PID color kLQPID\n");
542         //trk->GetReconstructor()->SetOption("!nn");
543       } else {
544         //printf("PID color kNNPID\n");
545         //trk->GetReconstructor()->SetOption("nn");
546       }
547       //trk->CookPID();
548   
549       Int_t species = 0; Float_t pid = 0.;
550       for(Int_t is=0; is<AliPID::kSPECIES; ++is) 
551         if(trk->GetPID(is) > pid){
552           pid = trk->GetPID(is);
553                   species = (AliPID::EParticleType) is;
554         }
555       switch(species){
556       case AliPID::kElectron:
557         SetMarkerColor(kRed);
558         SetLineColor(kRed);
559         break;
560       default:
561         SetMarkerColor(kBlue);
562         SetLineColor(kBlue);
563         break;
564       }
565     }
566     SetLineWidth(2);
567   }
568   
569   const Char_t *model = "line";
570   if(!TESTBIT(s, kTrackCosmics)){
571     if(TESTBIT(s, kTrackModel)) model = "kalman";
572     else model = "rieman";
573   }
574   Int_t species = 0; Float_t pid = 0.;
575   for(Int_t is=0; is<AliPID::kSPECIES; is++) 
576     if(trk->GetPID(is) > pid){
577       pid = trk->GetPID(is);
578       species = is;
579     }
580
581   SetTitle(Form(
582     "Tracklets[%d] Clusters[%d]\n"
583     "Reconstruction Source[%s]\n"
584     "PID[%4.1f %4.1f %4.1f %4.1f %4.1f]\n"
585     "MC[%d]", trk->GetNumberOfTracklets(), trk->GetNumberOfClusters(), fESDStatus&AliESDtrack::kTRDin ? "barrel" : "sa",
586     1.E2*trk->GetPID(0), 1.E2*trk->GetPID(1),
587     1.E2*trk->GetPID(2), 1.E2*trk->GetPID(3), 1.E2*trk->GetPID(4), trk->GetLabel()));
588
589   if(GetName()){
590     char id[6]; snprintf(id, 6, "%s", GetName());
591     SetName(Form("%s %s", id, AliPID::ParticleName(species)));
592   }
593
594   // save track status
595   fTrackState = s;
596 }
597
598 //______________________________________________________________________________
599 void AliEveTRDTrack::Load(const Char_t *what) const
600 {
601 // Spread downwards to tracklets the command "what"
602
603   const AliEveTRDTracklet* trklt(NULL);
604   TEveElement::List_ci itrklt=BeginChildren();
605   while(itrklt!=EndChildren()){
606     if((trklt = dynamic_cast<const AliEveTRDTracklet*>(*itrklt))) trklt->Load(what);
607     itrklt++;
608   }
609 }
610
611 //______________________________________________________________________________
612 AliEveTRDTrackletOnline::AliEveTRDTrackletOnline(AliTRDtrackletMCM *tracklet) :
613   TEveLine(),
614   fDetector(-1),
615   fROB(-1),
616   fMCM(-1)
617 {
618   AliTRDtrackletMCM *trkl = new AliTRDtrackletMCM(*tracklet);
619   SetUserData(trkl);
620
621   fDetector = trkl->GetDetector();
622   fROB = trkl->GetROB();
623   fMCM = trkl->GetMCM();
624
625   SetName("sim. tracklet");
626   SetTitle(Form("Det: %i, ROB: %i, MCM: %i, Label: %i\n0x%08x", 
627                 trkl->GetDetector(), trkl->GetROB(), trkl->GetMCM(), trkl->GetLabel(),
628                 trkl->GetTrackletWord()));
629   SetLineColor(kGreen);
630   SetLineWidth(3);
631
632   AliTRDgeometry geo;
633   TGeoHMatrix *matrix = geo.GetClusterMatrix(trkl->GetDetector());
634
635   fDetector = trkl->GetDetector();
636   fROB = trkl->GetROB();
637   fMCM = trkl->GetMCM();
638   
639   Float_t length = 3.;
640   Double_t x[3];
641   Double_t p[3];
642   Double_t p2[3];
643   x[0] = AliTRDgeometry::AnodePos(); 
644   x[1] = trkl->GetY();
645   x[2] = trkl->GetLocalZ();
646
647   matrix->LocalToMaster(x, p);
648   geo.RotateBack(trkl->GetDetector(), p, p2);
649   SetPoint(0, p2[0], p2[1], p2[2]);
650
651   x[0] -= length;
652   x[1] -= length * trkl->GetdYdX();
653   matrix->LocalToMaster(x, p);
654   p[2] *= p[0] / (p[0] + length);
655   geo.RotateBack(trkl->GetDetector(), p, p2);
656   SetPoint(1, p2[0], p2[1], p2[2]);
657 }
658
659 AliEveTRDTrackletOnline::AliEveTRDTrackletOnline(AliTRDtrackletWord *tracklet) :
660   TEveLine(),
661   fDetector(-1),
662   fROB(-1),
663   fMCM(-1)
664 {
665   AliTRDtrackletWord *trkl = new AliTRDtrackletWord(*tracklet);
666   SetUserData(trkl);
667
668   fDetector = trkl->GetDetector();
669   fROB = trkl->GetROB(); 
670   fMCM = trkl->GetMCM(); 
671
672   SetName("raw tracklet");
673   SetTitle(Form("Det: %i, ROB: %i, MCM: %i, Label: %i\n0x%08x", 
674                 trkl->GetDetector(), fROB, fMCM, -1,
675                 trkl->GetTrackletWord()));
676   SetLineColor(kRed);
677   SetLineWidth(3);
678
679   AliTRDgeometry geo;
680   TGeoHMatrix *matrix = geo.GetClusterMatrix(trkl->GetDetector());
681
682   Float_t length = 3.;
683   Double_t x[3];
684   Double_t p[3];
685   Double_t p2[3];
686   x[0] = AliTRDgeometry::AnodePos();
687   x[1] = trkl->GetY();
688   x[2] = trkl->GetLocalZ();
689   
690   matrix->LocalToMaster(x, p);
691   geo.RotateBack(trkl->GetDetector(), p, p2);
692   SetPoint(0, p2[0], p2[1], p2[2]);
693
694   x[0] -= length;
695   x[1] -= length * trkl->GetdYdX();
696   matrix->LocalToMaster(x, p);
697   p[2] *= p[0] / (p[0] + length);
698   geo.RotateBack(trkl->GetDetector(), p, p2);
699   SetPoint(1, p2[0], p2[1], p2[2]);
700 }
701
702 AliEveTRDTrackletOnline::~AliEveTRDTrackletOnline() 
703 {
704   delete ((AliTRDtrackletBase*) GetUserData());
705   SetUserData(0x0);
706 }
707
708 void AliEveTRDTrackletOnline::ShowMCM(Option_t *opt) const
709 {
710   if (fDetector < 0 || fROB < 0 || fMCM < 0)
711     return;
712
713   AliEveTRDmcm *evemcm = new AliEveTRDmcm();
714   evemcm->Init(fDetector, fROB, fMCM);
715   evemcm->LoadDigits();
716   evemcm->Draw(opt);
717
718   TEveElementList *mcmlist = NULL;
719   if (gEve->GetCurrentEvent()) 
720     mcmlist = (TEveElementList*) gEve->GetCurrentEvent()->FindChild("TRD MCMs");
721   if (!mcmlist) {
722     mcmlist = new TEveElementList("TRD MCMs");
723     gEve->AddElement(mcmlist);
724   }
725   gEve->AddElement(evemcm, mcmlist);
726 }
727
728
729 AliEveTRDmcm::AliEveTRDmcm() :
730   TEveElement(),
731   TNamed(),
732   fMCM(new AliTRDmcmSim())
733 {
734   SetName("MCM");
735   SetTitle("Unknown MCM");
736 }
737
738 AliEveTRDmcm::~AliEveTRDmcm()
739 {
740   delete fMCM;
741 }
742
743 Bool_t AliEveTRDmcm::Init(Int_t det, Int_t rob, Int_t mcm)
744 {
745   SetName(Form("MCM: Det. %i, ROB %i, MCM %i", det, rob, mcm));
746   SetTitle(Form("MCM: Det. %i, ROB %i, MCM %i", det, rob, mcm));
747   fMCM->Init(det, rob, mcm);
748   fMCM->Reset();
749   return kTRUE;
750 }
751
752 Bool_t AliEveTRDmcm::LoadDigits()
753 {
754   AliRunLoader *rl = AliEveEventManager::AssertRunLoader();
755   return fMCM->LoadMCM(rl, fMCM->GetDetector(), 
756                        fMCM->GetRobPos(), fMCM->GetMcmPos());
757 }
758
759 Bool_t AliEveTRDmcm::Filter()
760 {
761   fMCM->Filter();
762   return kTRUE;
763 }
764
765 Bool_t AliEveTRDmcm::Tracklet()
766 {
767   fMCM->Tracklet();
768   return kTRUE;
769 }
770
771 void AliEveTRDmcm::Draw(Option_t* option)
772 {
773   const char *mcmname = Form("mcm_%i_%i_%i", fMCM->GetDetector(),
774                              fMCM->GetRobPos(), fMCM->GetMcmPos());
775
776   TCanvas *c = dynamic_cast<TCanvas*> (gROOT->FindObject(mcmname));
777   if (!c)
778     c = gEve->AddCanvasTab("TRD MCM");
779   c->SetTitle(Form("MCM %i on ROB %i of det. %i", 
780                    fMCM->GetMcmPos(), fMCM->GetRobPos(), fMCM->GetDetector()));
781   c->SetName(mcmname);
782   c->cd();
783   fMCM->Draw(option);
784 }
785
786 Bool_t AliEveTRDmcm::AssignPointer(const char* ptrname)
787 {
788   gROOT->ProcessLine(Form("AliTRDmcmSim* %s = (AliTRDmcmSim *)%p", ptrname, (void*)fMCM));
789   return kTRUE;
790 }