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