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 }