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