fix coverity
[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){
260       if(!loader->Open("galice.root")){delete loader; loader=NULL;}
261       else{
262         loader->SetDataType(AliEveTRDLoader::kTRDHits | AliEveTRDLoader::kTRDDigits | AliEveTRDLoader::kTRDClusters);
263         break;
264       }
265     }
266   case 0:  
267     loader = new AliEveTRDLoader("Hits");
268     if(loader){
269       if(!loader->Open("TRD.Hits.root")){delete loader; loader=NULL;}
270       else{
271         if(typ!=4) break;
272       }
273     }
274   case 1:
275     if(!loader) loader = new AliEveTRDLoader("Digits");
276     if(loader){
277       if(!loader->Open("TRD.Digits.root")){
278         if(typ==1){delete loader; loader=NULL;}
279       } else {
280         if(typ!=4) break;
281       }
282     }
283   case 2:
284     if(!loader) loader = new AliEveTRDLoader("Clusters");
285     if(loader){
286       if(!loader->Open("TRD.RecPoints.root")){
287         if(typ ==2){delete loader; loader=NULL;}
288       } else {
289         if(typ!=4) break;
290       }
291     }
292   case 3:
293     if(!loader) loader = new AliEveTRDLoader("Tracklets");
294     if(loader){
295       if(!loader->Open("TRD.Tracklets.root")){
296         if(typ ==3) {delete loader; loader=NULL;}
297       } else break;
298     }
299   default: return;
300   }
301   if(!loader) return;
302
303   loader->AddChambers(AliTRDgeometry::GetSector(det),AliTRDgeometry::GetStack(det), AliTRDgeometry::GetLayer(det));
304   // load first event
305   if(!loader->GoToEvent(AliEveEventManager::GetCurrent()->GetEventId())) AliWarning(Form("No data loaded for event %d", AliEveEventManager::GetCurrent()->GetEventId()));
306   // register loader with alieve
307   gEve->AddElement(loader->GetChamber(det), *(BeginParents()));
308   //loader->SpawnEditor();
309   gEve->Redraw3D();
310 }
311
312 ///////////////////////////////////////////////////////////
313 /////////////   AliEveTRDTracklet         /////////////////////
314 ///////////////////////////////////////////////////////////
315
316 //______________________________________________________________________________
317 AliEveTRDTracklet::AliEveTRDTracklet(AliTRDseedV1 *trklt):TEveLine()
318   ,fClusters(NULL)
319 {
320   // Constructor.
321   SetName("tracklet");
322   
323   if(!gGeoManager) AliEveEventManager::AssertGeometry();
324   SetUserData(trklt);
325   // init tracklet line
326   Int_t sec = AliTRDgeometry::GetSector(trklt->GetDetector());
327   Float_t alpha((0.5+sec)*AliTRDgeometry::GetAlpha()),
328           cphi(TMath::Cos(alpha)),
329           sphi(TMath::Sin(alpha));
330   Float_t dx;
331   Float_t x0   = trklt->GetX0();
332   Float_t y0   = trklt->GetYref(0);
333   Float_t z0   = trklt->GetZref(0);
334   Float_t dydx = trklt->GetYref(1);
335   Float_t dzdx = trklt->GetZref(1);
336   Float_t  tilt = trklt->GetTilt();
337   Float_t g[3];
338   AliTRDcluster *c = NULL;
339   for(Int_t ic=0; ic<AliTRDseedV1::kNclusters; ic++){
340     if(!(c = trklt->GetClusters(ic))) continue;
341     if(!fClusters) AddElement(fClusters = new AliEveTRDClusters());
342     dx = x0 - c->GetX();
343     //Float_t yt = y0 - dx*dydx;
344     Float_t zt = z0 - dx*dzdx;
345     // backup yc - for testing purposes
346     Float_t yc = c->GetY(); 
347     c->SetY(yc-tilt*(c->GetZ()-zt));
348     
349     // Ben: Temporary(?) fix -> Issue with wrongly set flag "fIsMisaligned" of AliCluster
350     // Changes can be undone ofter the issue with the flag has been solved
351     //c->GetGlobalXYZ(g);
352     
353     // The following code for the trasformation local-to-global is just adapted from
354     // AliTRDgeometry::RotateBack(...)
355   
356     // tracking to global coordinates transformation
357     Float_t loc[3] = { c->GetX(), c->GetY(), c->GetZ() }; 
358     g[0] = loc[0] * cphi - loc[1] * sphi;
359     g[1] = loc[0] * sphi + loc[1] * cphi;
360     g[2] = loc[2];
361     // Ben: End of fix
362     Int_t id = fClusters->SetNextPoint(g[0], g[1], g[2]);
363     c->SetY(yc);
364     fClusters->SetPointId(id, new AliTRDcluster(*c));
365   } 
366   if(fClusters){
367     fClusters->SetName("TC clusters");
368     fClusters->SetTitle(Form("N[%d]", trklt->GetN()));
369     fClusters->SetMarkerColor(kMagenta);
370   }
371
372   SetTitle(Form("Det[%d] RC[%c] Layer[%d] P[%7.3f]", trklt->GetDetector(), trklt->IsRowCross()?'y':'n', trklt->GetPlane(), trklt->GetMomentum()));
373   SetLineColor(kRed);
374   //SetOwnIds(kTRUE);
375   
376   //trklt->Fit(kTRUE);
377   y0   = trklt->GetYfit(0);
378   dydx = trklt->GetYfit(1);
379   Double_t xg(x0 * cphi - y0 * sphi),
380            yg(x0 * sphi + y0 * cphi);
381   SetPoint(0, xg, yg, z0); 
382   //SetPoint(0, x0, y0, z0);
383
384
385   dx = .5*AliTRDgeometry::CamHght()+AliTRDgeometry::CdrHght();
386   x0 -= dx; 
387   y0 -= dydx*dx,
388   z0 -= dzdx*dx; 
389   xg = x0 * cphi - y0 * sphi;
390   yg = x0 * sphi + y0 * cphi;
391   SetPoint(1, xg, yg, z0);
392   //SetPoint(1, x0, y0, z0);
393 }
394
395
396 AliEveTRDTracklet::~AliEveTRDTracklet() 
397 {
398
399 }
400
401 //______________________________________________________________________________
402 void AliEveTRDTracklet::Print(Option_t *o) const
403 {
404   AliTRDseedV1 *tracklet = (AliTRDseedV1*)GetUserData();
405   if(!tracklet) return;
406   tracklet->Print(o);
407 }
408
409 ///////////////////////////////////////////////////////////
410 /////////////   AliEveTRDTrack         /////////////////////
411 ///////////////////////////////////////////////////////////
412
413 //______________________________________________________________________________
414 AliEveTRDTrack::AliEveTRDTrack(AliTRDtrackV1 *trk) 
415   :TEveLine()
416   ,fTrackState(0)
417   ,fESDStatus(0)
418   ,fAlpha(0.)
419   ,fPoints(NULL)
420   ,fRim(NULL)
421 {
422   // Constructor.
423   SetUserData(trk);
424   SetName("");
425
426 //  AliTRDtrackerV1::SetNTimeBins(24);
427
428   fRim = new AliRieman(trk->GetNumberOfClusters());
429   AliTRDseedV1 *tracklet = NULL;
430   for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++){
431     if(!(tracklet = trk->GetTracklet(il))) continue;
432     if(!tracklet->IsOK()) continue;
433     AddElement(new AliEveTRDTracklet(tracklet));
434
435 //     tracklet->ResetClusterIter(kFALSE);
436 //     while((c = tracklet->PrevCluster())){
437 //    AliTRDcluster *c(NULL);
438 /*    for(Int_t ic=AliTRDseedV1::kNtb; ic--;){
439       if(!(c=tracklet->GetClusters(ic))) continue;
440       Float_t xc = c->GetX();
441       Float_t yc = c->GetY();
442       Float_t zc = c->GetZ();
443       Float_t zt = tracklet->GetZref(0) - (tracklet->GetX0()-xc)*tracklet->GetZref(1); 
444       yc -= tracklet->GetTilt()*(zc-zt);
445       fRim->AddPoint(xc, yc, zc, .05, 2.3);
446     }*/
447   }
448   if(trk->GetNumberOfTracklets()>1) fRim->Update();
449   SetStatus(fTrackState);
450 }
451
452 //______________________________________________________________________________
453 AliEveTRDTrack::~AliEveTRDTrack()
454 {
455   if(fPoints) delete [] fPoints; fPoints = NULL;
456   //delete dynamic_cast<AliTRDtrackV1*>(GetUserData());
457 }
458
459 //______________________________________________________________________________
460 void AliEveTRDTrack::Print(Option_t *o) const
461 {
462   AliTRDtrackV1 *track = (AliTRDtrackV1*)GetUserData();
463   if(!track) return;
464   track->Print(o);
465 }
466
467 //______________________________________________________________________________
468 void AliEveTRDTrack::SetStatus(UChar_t s)
469 {
470   // nothing to be done
471   if(fPoints && fTrackState == s) return;
472   //return;
473   const Int_t nc = AliTRDtrackV1::kMAXCLUSTERSPERTRACK;
474   AliTRDtrackV1 *trk(NULL);
475   if(!(trk=static_cast<AliTRDtrackV1*>(GetUserData()))) {
476     AliError("Failed casting data to TRD track.");
477     return;
478   }
479
480   Bool_t BUILD = kFALSE;
481   if(!fPoints){ 
482     fPoints = new AliTrackPoint[nc];
483
484     // define the radial span of the track in the TRD
485     Double_t xmin = -1., xmax = -1.;
486     Int_t det = 0;
487     AliTRDseedV1 *trklt = NULL;
488     for(Int_t ily=0; ily<AliTRDgeometry::kNlayer; ily++){
489       if(!(trklt = trk->GetTracklet(ily))) continue;
490       if(xmin<0.) xmin = trklt->GetX0() - AliTRDgeometry::CamHght() - AliTRDgeometry::CdrHght();
491       if(trklt->GetX0()>xmax) xmax = trklt->GetX0();
492       det = trklt->GetDetector();
493     }
494     Int_t sec = det/AliTRDgeometry::kNdets;
495     fAlpha = AliTRDgeometry::GetAlpha() * (sec<9 ? sec + .5 : sec - 17.5); //trk->GetAlpha()
496
497     Double_t dx =(xmax - xmin)/nc;
498     for(Int_t ip=0; ip<nc; ip++){
499       fPoints[ip].SetXYZ(xmin, 0., 0.);
500       xmin+=dx;
501     }
502     BUILD = kTRUE;
503   }
504
505   // select track model
506   if(BUILD || ((s&12) != (fTrackState&12))){
507     if(TESTBIT(s, kTrackCosmics)){
508       //printf("Straight track\n");
509       AliTRDtrackerV1::FitLine(trk, NULL, kFALSE, nc, fPoints);
510     } else {
511       if(TESTBIT(s, kTrackModel)){
512         //printf("Kalman track\n");
513         if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitKalman(trk, NULL, kFALSE, nc, fPoints);
514       } else { 
515         //printf("Rieman track rim[%p] nc[%d]\n", (void*)fRim, nc);
516         //if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitRiemanTilt(trk, NULL, kTRUE, nc, fPoints);
517         Float_t x = 0.;
518         for(Int_t ip = nc; ip--;){
519           x = fPoints[ip].GetX();
520           //printf("%2d x[%f] y[%f] z[%f]\n", ip, x, fRim->GetYat(x), fRim->GetZat(x));
521           fPoints[ip].SetXYZ(x, fRim->GetYat(x), fRim->GetZat(x));
522         }
523       }
524     }
525   
526     Float_t global[3];
527     for(Int_t ip=0; ip<nc; ip++){
528       fPoints[ip].Rotate(-fAlpha).GetXYZ(global);
529       SetPoint(ip, global[0], global[1], global[2]);
530       //printf("*** %2d x[%f] y[%f] z[%f]\n", ip, global[0], global[1], global[2]);
531     
532     }
533     SetSmooth(kTRUE);
534   }
535
536   // set color
537   if(BUILD || ((s&3) != (fTrackState&3))){
538     if(TESTBIT(s, kSource)){
539       //printf("Source color\n");
540       if(fESDStatus&AliESDtrack::kTRDin){
541         SetMarkerColor(kGreen);
542         SetLineColor(kGreen);
543       } else {
544         SetMarkerColor(kMagenta);
545         SetLineColor(kMagenta);
546       }
547     } else {
548       if(TESTBIT(s, kPID) == AliTRDpidUtil::kLQ){
549         //printf("PID color kLQPID\n");
550         //trk->GetReconstructor()->SetOption("!nn");
551       } else {
552         //printf("PID color kNNPID\n");
553         //trk->GetReconstructor()->SetOption("nn");
554       }
555       //trk->CookPID();
556   
557       Int_t species = 0; Float_t pid = 0.;
558       for(Int_t is=0; is<AliPID::kSPECIES; ++is) 
559         if(trk->GetPID(is) > pid){
560           pid = trk->GetPID(is);
561                   species = (AliPID::EParticleType) is;
562         }
563       switch(species){
564       case AliPID::kElectron:
565         SetMarkerColor(kRed);
566         SetLineColor(kRed);
567         break;
568       default:
569         SetMarkerColor(kBlue);
570         SetLineColor(kBlue);
571         break;
572       }
573     }
574     SetLineWidth(2);
575   }
576   
577   const Char_t *model = "line";
578   if(!TESTBIT(s, kTrackCosmics)){
579     if(TESTBIT(s, kTrackModel)) model = "kalman";
580     else model = "rieman";
581   }
582   Int_t species = 0; Float_t pid = 0.;
583   for(Int_t is=0; is<AliPID::kSPECIES; is++) 
584     if(trk->GetPID(is) > pid){
585       pid = trk->GetPID(is);
586       species = is;
587     }
588
589   SetTitle(Form(
590     "Tracklets[%d] Clusters[%d]\n"
591     "Reconstruction Source[%s]\n"
592     "PID[%4.1f %4.1f %4.1f %4.1f %4.1f]\n"
593     "MC[%d]", trk->GetNumberOfTracklets(), trk->GetNumberOfClusters(), fESDStatus&AliESDtrack::kTRDin ? "barrel" : "sa",
594     1.E2*trk->GetPID(0), 1.E2*trk->GetPID(1),
595     1.E2*trk->GetPID(2), 1.E2*trk->GetPID(3), 1.E2*trk->GetPID(4), trk->GetLabel()));
596
597   if(GetName()){
598     char id[6]; snprintf(id, 6, "%s", GetName());
599     SetName(Form("%s %s", id, AliPID::ParticleName(species)));
600   }
601
602   // save track status
603   fTrackState = s;
604 }
605
606 //______________________________________________________________________________
607 void AliEveTRDTrack::Load(const Char_t *what) const
608 {
609 // Spread downwards to tracklets the command "what"
610
611   const AliEveTRDTracklet* trklt(NULL);
612   TEveElement::List_ci itrklt=BeginChildren();
613   while(itrklt!=EndChildren()){
614     if((trklt = dynamic_cast<const AliEveTRDTracklet*>(*itrklt))) trklt->Load(what);
615     itrklt++;
616   }
617 }
618
619 //______________________________________________________________________________
620 AliEveTRDTrackletOnline::AliEveTRDTrackletOnline(AliTRDtrackletMCM *tracklet) :
621   TEveLine(),
622   fDetector(-1),
623   fROB(-1),
624   fMCM(-1)
625 {
626   AliTRDtrackletMCM *trkl = new AliTRDtrackletMCM(*tracklet);
627   SetUserData(trkl);
628
629   fDetector = trkl->GetDetector();
630   fROB = trkl->GetROB();
631   fMCM = trkl->GetMCM();
632
633   SetName("sim. tracklet");
634   SetTitle(Form("Det: %i, ROB: %i, MCM: %i, Label: %i\n0x%08x", 
635                 trkl->GetDetector(), trkl->GetROB(), trkl->GetMCM(), trkl->GetLabel(),
636                 trkl->GetTrackletWord()));
637   SetLineColor(kGreen);
638   SetLineWidth(3);
639
640   AliTRDgeometry geo;
641   TGeoHMatrix *matrix = geo.GetClusterMatrix(trkl->GetDetector());
642
643   fDetector = trkl->GetDetector();
644   fROB = trkl->GetROB();
645   fMCM = trkl->GetMCM();
646   
647   Float_t length = 3.;
648   Double_t x[3];
649   Double_t p[3];
650   Double_t p2[3];
651   x[0] = AliTRDgeometry::AnodePos(); 
652   x[1] = trkl->GetY();
653   x[2] = trkl->GetLocalZ();
654
655   matrix->LocalToMaster(x, p);
656   geo.RotateBack(trkl->GetDetector(), p, p2);
657   SetPoint(0, p2[0], p2[1], p2[2]);
658
659   x[0] -= length;
660   x[1] -= length * trkl->GetdYdX();
661   matrix->LocalToMaster(x, p);
662   p[2] *= p[0] / (p[0] + length);
663   geo.RotateBack(trkl->GetDetector(), p, p2);
664   SetPoint(1, p2[0], p2[1], p2[2]);
665 }
666
667 AliEveTRDTrackletOnline::AliEveTRDTrackletOnline(AliTRDtrackletWord *tracklet) :
668   TEveLine(),
669   fDetector(-1),
670   fROB(-1),
671   fMCM(-1)
672 {
673   AliTRDtrackletWord *trkl = new AliTRDtrackletWord(*tracklet);
674   SetUserData(trkl);
675
676   fDetector = trkl->GetDetector();
677   fROB = trkl->GetROB(); 
678   fMCM = trkl->GetMCM(); 
679
680   SetName("raw tracklet");
681   SetTitle(Form("Det: %i, ROB: %i, MCM: %i, Label: %i\n0x%08x", 
682                 trkl->GetDetector(), fROB, fMCM, -1,
683                 trkl->GetTrackletWord()));
684   SetLineColor(kRed);
685   SetLineWidth(3);
686
687   AliTRDgeometry geo;
688   TGeoHMatrix *matrix = geo.GetClusterMatrix(trkl->GetDetector());
689
690   Float_t length = 3.;
691   Double_t x[3];
692   Double_t p[3];
693   Double_t p2[3];
694   x[0] = AliTRDgeometry::AnodePos();
695   x[1] = trkl->GetY();
696   x[2] = trkl->GetLocalZ();
697   
698   matrix->LocalToMaster(x, p);
699   geo.RotateBack(trkl->GetDetector(), p, p2);
700   SetPoint(0, p2[0], p2[1], p2[2]);
701
702   x[0] -= length;
703   x[1] -= length * trkl->GetdYdX();
704   matrix->LocalToMaster(x, p);
705   p[2] *= p[0] / (p[0] + length);
706   geo.RotateBack(trkl->GetDetector(), p, p2);
707   SetPoint(1, p2[0], p2[1], p2[2]);
708 }
709
710 AliEveTRDTrackletOnline::~AliEveTRDTrackletOnline() 
711 {
712   delete ((AliTRDtrackletBase*) GetUserData());
713   SetUserData(0x0);
714 }
715
716 void AliEveTRDTrackletOnline::ShowMCM(Option_t *opt) const
717 {
718   if (fDetector < 0 || fROB < 0 || fMCM < 0)
719     return;
720
721   AliEveTRDmcm *evemcm = new AliEveTRDmcm();
722   evemcm->Init(fDetector, fROB, fMCM);
723   evemcm->LoadDigits();
724   evemcm->Draw(opt);
725
726   TEveElementList *mcmlist = NULL;
727   if (gEve->GetCurrentEvent()) 
728     mcmlist = (TEveElementList*) gEve->GetCurrentEvent()->FindChild("TRD MCMs");
729   if (!mcmlist) {
730     mcmlist = new TEveElementList("TRD MCMs");
731     gEve->AddElement(mcmlist);
732   }
733   gEve->AddElement(evemcm, mcmlist);
734 }
735
736
737 AliEveTRDmcm::AliEveTRDmcm() :
738   TEveElement(),
739   TNamed(),
740   fMCM(new AliTRDmcmSim())
741 {
742   SetName("MCM");
743   SetTitle("Unknown MCM");
744 }
745
746 AliEveTRDmcm::~AliEveTRDmcm()
747 {
748   delete fMCM;
749 }
750
751 Bool_t AliEveTRDmcm::Init(Int_t det, Int_t rob, Int_t mcm)
752 {
753   SetName(Form("MCM: Det. %i, ROB %i, MCM %i", det, rob, mcm));
754   SetTitle(Form("MCM: Det. %i, ROB %i, MCM %i", det, rob, mcm));
755   fMCM->Init(det, rob, mcm);
756   fMCM->Reset();
757   return kTRUE;
758 }
759
760 Bool_t AliEveTRDmcm::LoadDigits()
761 {
762   AliRunLoader *rl = AliEveEventManager::AssertRunLoader();
763   return fMCM->LoadMCM(rl, fMCM->GetDetector(), 
764                        fMCM->GetRobPos(), fMCM->GetMcmPos());
765 }
766
767 Bool_t AliEveTRDmcm::Filter()
768 {
769   fMCM->Filter();
770   return kTRUE;
771 }
772
773 Bool_t AliEveTRDmcm::Tracklet()
774 {
775   fMCM->Tracklet();
776   return kTRUE;
777 }
778
779 void AliEveTRDmcm::Draw(Option_t* option)
780 {
781   const char *mcmname = Form("mcm_%i_%i_%i", fMCM->GetDetector(),
782                              fMCM->GetRobPos(), fMCM->GetMcmPos());
783
784   TCanvas *c = dynamic_cast<TCanvas*> (gROOT->FindObject(mcmname));
785   if (!c)
786     c = gEve->AddCanvasTab("TRD MCM");
787   c->SetTitle(Form("MCM %i on ROB %i of det. %i", 
788                    fMCM->GetMcmPos(), fMCM->GetRobPos(), fMCM->GetDetector()));
789   c->SetName(mcmname);
790   c->cd();
791   fMCM->Draw(option);
792 }
793
794 Bool_t AliEveTRDmcm::AssignPointer(const char* ptrname)
795 {
796   gROOT->ProcessLine(Form("AliTRDmcmSim* %s = (AliTRDmcmSim *)%p", ptrname, (void*)fMCM));
797   return kTRUE;
798 }