HLT/TPCLib/tracking-ca/AliHLTTPCCADisplay.cxx

   1 // $Id$
   2 // **************************************************************************
   3 // This file is property of and copyright by the ALICE HLT Project          *
   4 // ALICE Experiment at CERN, All rights reserved.                           *
   5 //                                                                          *
   6 // Primary Authors: Sergey Gorbunov <sergey.gorbunov@kip.uni-heidelberg.de> *
   7 //                  Ivan Kisel <kisel@kip.uni-heidelberg.de>                *
   8 //                  for The ALICE HLT Project.                              *
   9 //                                                                          *
  10 // Permission to use, copy, modify and distribute this software and its     *
  11 // documentation strictly for non-commercial purposes is hereby granted     *
  12 // without fee, provided that the above copyright notice appears in all     *
  13 // copies and that both the copyright notice and this permission notice     *
  14 // appear in the supporting documentation. The authors make no claims       *
  15 // about the suitability of this software for any purpose. It is            *
  16 // provided "as is" without express or implied warranty.                    *
  17 //                                                                          *
  18 //***************************************************************************
  19
  20
  21 #include "AliHLTTPCCADisplay.h"
  22
  23
  24 #include "AliHLTTPCCATracker.h"
  25 #include "AliHLTTPCCAGBTracker.h"
  26 #include "AliHLTTPCCARow.h"
  27 #include "AliHLTTPCCATrack.h"
  28 #include "AliHLTTPCCAGBTrack.h"
  29 #include "AliHLTTPCCAGBHit.h"
  30 #include "AliHLTTPCCAPerformance.h"
  31 #include "AliHLTTPCCAMCTrack.h"
  32 #include "AliHLTTPCCAOutTrack.h"
  33
  34 #include "TString.h"
  35 #include "Riostream.h"
  36 #include "TMath.h"
  37 #include "TStyle.h"
  38 #include "TCanvas.h"
  39 #include "TApplication.h"
  40
  41
  42 class AliHLTTPCCADisplay::AliHLTTPCCADisplayTmpHit
  43 {
  44
  45 public:
  46
  47   Int_t ID() const { return fHitID; }
  48   Double_t S() const { return fS; }
  49   Double_t Z() const { return fZ; }
  50
  51   void SetID( Int_t v ){ fHitID = v; }
  52   void SetS( Double_t v){ fS = v; }
  53   void SetZ( Double_t v){ fZ = v; }
  54
  55   static Bool_t CompareHitDS( const AliHLTTPCCADisplayTmpHit &a,
  56                               const AliHLTTPCCADisplayTmpHit  &b )
  57   {
  58     return (a.fS < b.fS);
  59   }
  60
  61   static Bool_t CompareHitZ( const AliHLTTPCCADisplayTmpHit &a,
  62                              const AliHLTTPCCADisplayTmpHit  &b )
  63   {
  64     return (a.fZ < b.fZ);
  65   }
  66
  67 private:
  68
  69   Int_t fHitID; // hit ID
  70   Double_t fS;  // hit position on the XY track curve
  71   Double_t fZ;  // hit Z position
  72
  73 };
  74
  75
  76
  77 AliHLTTPCCADisplay &AliHLTTPCCADisplay::Instance()
  78 {
  79   // reference to static object
  80   static AliHLTTPCCADisplay gAliHLTTPCCADisplay;
  81   static Bool_t firstCall = 1;
  82   if( firstCall ){
  83     if( !gApplication ) new TApplication("myapp",0,0);
  84     gAliHLTTPCCADisplay.Init();
  85     firstCall = 0;
  86   }
  87   return gAliHLTTPCCADisplay;
  88 }
  89
  90 AliHLTTPCCADisplay::AliHLTTPCCADisplay() : fYX(0), fZX(0), fAsk(1), fSliceView(1), fSlice(0),fGB(0), fPerf(0),
  91                                            fCos(1), fSin(0), fZMin(-250), fZMax(250),fYMin(-250), fYMax(250),fSliceCos(1), fSliceSin(0),
  92                                            fRInnerMin(83.65), fRInnerMax(133.3), fROuterMin(133.5), fROuterMax(247.7),
  93                                            fTPCZMin(-250.), fTPCZMax(250), fArc(), fLine(), fPLine(), fMarker(), fBox(), fCrown(), fLatex(), fDrawOnlyRef(0)
  94 {
  95   fPerf = &(AliHLTTPCCAPerformance::Instance());
  96   // constructor
  97 }
  98
  99
 100 AliHLTTPCCADisplay::AliHLTTPCCADisplay( const AliHLTTPCCADisplay& )
 101   : fYX(0), fZX(0), fAsk(1), fSliceView(1), fSlice(0), fGB(0), fPerf(0),
 102     fCos(1), fSin(0), fZMin(-250), fZMax(250), fYMin(-250), fYMax(250), fSliceCos(1), fSliceSin(0),
 103     fRInnerMin(83.65), fRInnerMax(133.3), fROuterMin(133.5), fROuterMax(247.7),
 104     fTPCZMin(-250.), fTPCZMax(250), fArc(), fLine(), fPLine(), fMarker(), fBox(), fCrown(), fLatex(), fDrawOnlyRef(0)
 105 {
 106   // dummy
 107 }
 108
 109 const AliHLTTPCCADisplay& AliHLTTPCCADisplay::operator=( const AliHLTTPCCADisplay& ) const
 110 {
 111   // dummy
 112   return *this;
 113 }
 114
 115 AliHLTTPCCADisplay::~AliHLTTPCCADisplay()
 116 {
 117   // destructor
 118   delete fYX;
 119   delete fZX;
 120 }
 121
 122 void AliHLTTPCCADisplay::Init()
 123 {
 124   // initialization
 125   gStyle->SetCanvasBorderMode(0);
 126   gStyle->SetCanvasBorderSize(1);
 127   gStyle->SetCanvasColor(0);
 128   fYX = new TCanvas ("YX", "YX window", -1, 0, 600, 600);
 129   fZX = new TCanvas ("ZX", "ZX window", -610, 0, 590, 600);
 130   fMarker = TMarker(0.0, 0.0, 20);//6);
 131   fDrawOnlyRef = 0;
 132 }
 133
 134 void AliHLTTPCCADisplay::Update()
 135 {
 136   // update windows
 137   if( !fAsk ) return;
 138   fYX->Update();
 139   fZX->Update();
 140   fYX->Print("YX.pdf");
 141   fZX->Print("ZX.pdf");
 142
 143 }
 144
 145 void AliHLTTPCCADisplay::ClearView()
 146 {
 147   // clear windows
 148   fYX->Clear();
 149   fZX->Clear();
 150 }
 151
 152 void AliHLTTPCCADisplay::Ask()
 153 {
 154   // whait for the pressed key, when "r" pressed, don't ask anymore
 155   char symbol;
 156   if (fAsk){
 157     Update();
 158     std::cout<<"ask> "<<std::endl;
 159     do{
 160       std::cin.get(symbol);
 161       if (symbol == 'r')
 162         fAsk = false;
 163     } while (symbol != '\n');
 164   }
 165 }
 166
 167
 168 void AliHLTTPCCADisplay::SetSliceView()
 169 {
 170   // switch to slice view
 171   fSliceView = 1;
 172 }
 173
 174 void AliHLTTPCCADisplay::SetTPCView()
 175 {
 176   // switch to full TPC view
 177   fSliceView = 0;
 178   fCos = 1;
 179   fSin = 0;
 180   fZMin = fTPCZMin;
 181   fZMax = fTPCZMax;
 182   fYMin = -fROuterMax;
 183   fYMax = fROuterMax;
 184 }
 185
 186
 187 void AliHLTTPCCADisplay::SetGB( AliHLTTPCCAGBTracker * const GBTracker )
 188 {
 189   fGB = GBTracker;
 190 }
 191
 192 void AliHLTTPCCADisplay::SetCurrentSlice( AliHLTTPCCATracker *slice )
 193 {
 194   // set reference to the current CA tracker, and read the current slice geometry
 195   fSlice = slice;
 196   SetSliceTransform( slice );
 197   if( fSliceView ){
 198     fCos = slice->Param().SinAlpha();
 199     fSin = slice->Param().CosAlpha();
 200     fZMin = slice->Param().ZMin();
 201     fZMax = slice->Param().ZMax();
 202     ClearView();
 203     Double_t r0 = .5*(slice->Param().RMax()+slice->Param().RMin());
 204     Double_t dr = .5*(slice->Param().RMax()-slice->Param().RMin());
 205     fYMin = -dr;
 206     fYMax = dr;
 207     Double_t cx = 0;
 208     Double_t cy = r0;
 209     Double_t cz = .5*(slice->Param().ZMax()+slice->Param().ZMin());
 210     Double_t dz = .5*(slice->Param().ZMax()-slice->Param().ZMin())*1.2;
 211     fYX->Range(cx-dr, cy-dr*1.05, cx+dr, cy+dr);
 212     fZX->Range(cz-dz, cy-dr*1.05, cz+dz, cy+dr);
 213
 214     //fYX->Range(cx-dr*.3, cy-dr*1.05, cx+dr*.3, cy-dr*.35);
 215     //fZX->Range(cz-dz, cy-dr*1.05, cz+dz, cy-dr*.3);
 216
 217     //fYX->Range(cx-dr*.3, cy-dr*.8, cx-dr*.1, cy-dr*.75);
 218     //fZX->Range(cz-dz*0, cy-dr*.8, cz+dz, cy-dr*.75);
 219
 220     //fYX->Range(cx-dr*.08, cy-dr*1., cx-dr*.02, cy-dr*0.7);
 221     //fZX->Range(cz-dz*.2, cy-dr*1., cz-dz*.05, cy-dr*0.7);
 222
 223     //Double_t x0 = cx-dr*.1, x1 = cx-dr*.05;
 224     //Double_t y0 = cy-dr*1.05, y1 = cy-dr*0.7;
 225     //Double_t z0 = cz-dz*.3, z1 = cz;
 226     //Double_t xc = (x0+x1)/2, yc= (y0+y1)/2, zc=(z0+z1)/2;
 227     //Double_t d = TMath::Max((x1-x0)/2,TMath::Max((y1-y0)/2,(z1-z0)/2));
 228     //fYX->Range(xc-d, yc-d, xc+d, yc+d);
 229     //fZX->Range(zc-d, yc-d, zc+d, yc+d);
 230
 231    }
 232 }
 233
 234 void AliHLTTPCCADisplay::SetSliceTransform( Double_t alpha )
 235 {
 236   fSliceCos = TMath::Cos( alpha );
 237   fSliceSin = TMath::Sin( alpha );
 238 }
 239
 240 void AliHLTTPCCADisplay::SetSliceTransform( AliHLTTPCCATracker *slice )
 241 {
 242   SetSliceTransform(slice->Param().Alpha());
 243 }
 244
 245
 246 void AliHLTTPCCADisplay::DrawTPC()
 247 {
 248   // schematically draw TPC detector
 249   fYX->Range(-fROuterMax, -fROuterMax, fROuterMax, fROuterMax);
 250   //fYX->Range( -fROuterMax*.7, -fROuterMax, fROuterMax*0., -fROuterMax*.5);
 251   fYX->Clear();
 252   {
 253     fArc.SetLineColor(kBlack);
 254     fArc.SetFillStyle(0);
 255     fYX->cd();
 256     for( Int_t iSlice=0; iSlice<18; iSlice++){
 257       fCrown.SetLineColor(kBlack);
 258       fCrown.SetFillStyle(0);
 259       fCrown.DrawCrown(0,0,fRInnerMin, fRInnerMax, 360./18.*iSlice, 360./18.*(iSlice+1) );
 260       fCrown.DrawCrown(0,0,fROuterMin, fROuterMax, 360./18.*iSlice, 360./18.*(iSlice+1) );
 261     }
 262   }
 263   fZX->cd();
 264   fZX->Range( fTPCZMin, -fROuterMax, fTPCZMax*1.1, fROuterMax );
 265   //fZX->Range( fTPCZMax*.1, -fROuterMax, fTPCZMax*.3, -fROuterMax*0.5 );
 266   fZX->Clear();
 267 }
 268
 269 void AliHLTTPCCADisplay::DrawSlice( AliHLTTPCCATracker *slice, Bool_t DrawRows )
 270 {
 271   // draw current the TPC slice
 272   fYX->cd();
 273   Double_t r0 = .5*(slice->Param().RMax()+slice->Param().RMin());
 274   Double_t dr = .5*(slice->Param().RMax()-slice->Param().RMin());
 275   Double_t cx = r0*slice->Param().CosAlpha();
 276   Double_t cy = r0*slice->Param().SinAlpha();
 277   Double_t raddeg = 180./3.1415;
 278   Double_t a0 = raddeg*.5*(slice->Param().AngleMax() + slice->Param().AngleMin());
 279   Double_t da = raddeg*.5*(slice->Param().AngleMax() - slice->Param().AngleMin());
 280   if( fSliceView ){
 281     cx = 0; cy = r0;
 282     a0 = 90.;
 283     fLatex.DrawLatex(cx-dr+dr*.05,cy-dr+dr*.05, Form("YX, Slice %2i",slice->Param().ISlice()));
 284   } else {
 285     a0+= raddeg*TMath::ATan2(fSin, fCos );
 286   }
 287   fArc.SetLineColor(kBlack);
 288   fArc.SetFillStyle(0);
 289   fCrown.SetLineColor(kBlack);
 290   fCrown.SetFillStyle(0);
 291   fCrown.DrawCrown(0,0, fRInnerMin, fRInnerMax, a0-da, a0+da );
 292   fCrown.DrawCrown(0,0, fROuterMin, fROuterMax, a0-da, a0+da );
 293   //fCrown.DrawCrown(0,0, slice->Param().RMin(),slice->Param().RMax(), a0-da, a0+da );
 294
 295   fLine.SetLineColor(kBlack);
 296
 297   fZX->cd();
 298
 299   Double_t cz = .5*(slice->Param().ZMax()+slice->Param().ZMin());
 300   Double_t dz = .5*(slice->Param().ZMax()-slice->Param().ZMin())*1.2;
 301   //fLine.DrawLine(cz+dz, cy-dr, cz+dz, cy+dr );
 302   if( fSliceView ) fLatex.DrawLatex(cz-dz+dz*.05,cy-dr+dr*.05, Form("ZX, Slice %2i",slice->Param().ISlice()));
 303
 304   if( DrawRows ){
 305     fLine.SetLineWidth(1);
 306     fLine.SetLineColor(kBlack);
 307     SetSliceTransform(fSlice);
 308     for( Int_t iRow=0; iRow<fSlice->Param().NRows(); iRow++ ){
 309       Double_t x = fSlice->Row(iRow).X();
 310       Double_t y = fSlice->Row(iRow).MaxY();
 311       Double_t vx0, vy0, vx1, vy1;
 312       Slice2View( x, y, &vx0, &vy0 );
 313       Slice2View( x, -y, &vx1, &vy1 );
 314       fYX->cd();
 315       fLine.DrawLine(vx0, vy0,vx1,vy1);
 316       fZX->cd();
 317       fLine.DrawLine(fTPCZMin, vy0,fTPCZMax,vy1);
 318     }
 319   }
 320
 321 }
 322
 323
 324 void AliHLTTPCCADisplay::Set2Slices( AliHLTTPCCATracker * const slice )
 325 {
 326   //* Set view for two neighbouring slices
 327
 328   fSlice = slice;
 329   fSliceView = 0;
 330   fCos = TMath::Cos(TMath::Pi()/2 - (slice->Param().Alpha()+10./180.*TMath::Pi()));
 331   fSin = TMath::Sin(TMath::Pi()/2 - (slice->Param().Alpha()+10./180.*TMath::Pi()));
 332   fZMin = slice->Param().ZMin();
 333   fZMax = slice->Param().ZMax();
 334   ClearView();
 335   Double_t r0 = .5*(slice->Param().RMax()+slice->Param().RMin());
 336   Double_t dr = .5*(slice->Param().RMax()-slice->Param().RMin());
 337   Double_t cx = 0;
 338   Double_t cy = r0;
 339   fYX->Range(cx-1.3*dr, cy-1.1*dr, cx+1.3*dr, cy+1.1*dr);
 340   fYX->cd();
 341   Int_t islice = slice->Param().ISlice();
 342   Int_t jslice = slice->Param().ISlice()+1;
 343   if( islice==17 ) jslice = 0;
 344   else if( islice==35 ) jslice = 18;
 345   fLatex.DrawLatex(cx-1.3*dr+1.3*dr*.05,cy-dr+dr*.05, Form("YX, Slices %2i/%2i",islice,jslice));
 346   Double_t cz = .5*(slice->Param().ZMax()+slice->Param().ZMin());
 347   Double_t dz = .5*(slice->Param().ZMax()-slice->Param().ZMin())*1.2;
 348   fZX->Range(cz-dz, cy-1.1*dr, cz+dz, cy+1.1*dr);//+dr);
 349   fZX->cd();
 350   fLatex.DrawLatex(cz-dz+dz*.05,cy-dr+dr*.05, Form("ZX, Slices %2i/%2i",islice,jslice));
 351 }
 352
 353 Int_t AliHLTTPCCADisplay::GetColor( Int_t i ) const
 354 {
 355   // Get color with respect to Z coordinate
 356   const Color_t kMyColor[9] = { kGreen, kBlue, kYellow, kCyan, kOrange,
 357                                  kSpring, kTeal, kAzure, kViolet };
 358   if( i<0 ) i= 0;
 359   if( i==0 ) return kBlack;
 360   return kMyColor[(i-1)%9];
 361 }
 362
 363 Int_t AliHLTTPCCADisplay::GetColorZ( Double_t z ) const
 364 {
 365   // Get color with respect to Z coordinate
 366   const Color_t kMyColor[11] = { kGreen, kBlue, kYellow, kMagenta, kCyan,
 367                                  kOrange, kSpring, kTeal, kAzure, kViolet, kPink };
 368
 369   Double_t zz = (z-fZMin)/(fZMax-fZMin);
 370   Int_t iz = (int) (zz*11);
 371   if( iz<0 ) iz = 0;
 372   if( iz>10 ) iz = 10;
 373   return kMyColor[iz];
 374 }
 375
 376 Int_t AliHLTTPCCADisplay::GetColorY( Double_t y ) const
 377 {
 378   // Get color with respect to Z coordinate
 379   const Color_t kMyColor[11] = { kGreen, kBlue, kYellow, kMagenta, kCyan,
 380                                  kOrange, kSpring, kTeal, kAzure, kViolet, kPink };
 381
 382   Double_t yy = (y-fYMin)/(fYMax-fYMin);
 383   Int_t iy = (int) (yy*11);
 384   if( iy<0 ) iy = 0;
 385   if( iy>10 ) iy = 10;
 386   return kMyColor[iy];
 387 }
 388
 389 Int_t AliHLTTPCCADisplay::GetColorK( Double_t k ) const
 390 {
 391   // Get color with respect to Z coordinate
 392   const Color_t kMyColor[11] = { kRed, kBlue, kYellow, kMagenta, kCyan,
 393                                  kOrange, kSpring, kTeal, kAzure, kViolet, kPink };
 394   const Double_t kCLight = 0.000299792458;
 395   const Double_t kBz = 5;
 396   Double_t k2QPt = 100;
 397   if( TMath::Abs(kBz)>1.e-4 ) k2QPt= 1./(kBz*kCLight);
 398   Double_t qPt = k*k2QPt;
 399   Double_t pt = 100;
 400   if( TMath::Abs(qPt) >1.e-4 ) pt = 1./TMath::Abs(qPt);
 401
 402   Double_t yy = (pt-0.1)/(1.-0.1);
 403   Int_t iy = (int) (yy*11);
 404   if( iy<0 ) iy = 0;
 405   if( iy>10 ) iy = 10;
 406   return kMyColor[iy];
 407 }
 408
 409 void AliHLTTPCCADisplay::Global2View( Double_t x, Double_t y, Double_t *xv, Double_t *yv ) const
 410 {
 411   // convert coordinates global->view
 412   *xv = x*fCos + y*fSin;
 413   *yv = y*fCos - x*fSin;
 414 }
 415
 416
 417 void AliHLTTPCCADisplay::Slice2View( Double_t x, Double_t y, Double_t *xv, Double_t *yv ) const
 418 {
 419   // convert coordinates slice->view
 420   Double_t xg = x*fSliceCos - y*fSliceSin;
 421   Double_t yg = y*fSliceCos + x*fSliceSin;
 422   *xv = xg*fCos - yg*fSin;
 423   *yv = yg*fCos + xg*fSin;
 424 }
 425
 426
 427 void AliHLTTPCCADisplay::DrawGBHit( AliHLTTPCCAGBTracker &tracker, Int_t iHit, Int_t color, Size_t width  )
 428 {
 429   // draw hit
 430   AliHLTTPCCAGBHit &h = tracker.Hits()[iHit];
 431   AliHLTTPCCATracker &slice = tracker.Slices()[h.ISlice()];
 432   SetSliceTransform(&slice);
 433
 434   if( color<0 ){
 435     if( fPerf ){
 436       Int_t lab = fPerf->HitLabels()[h.ID()].fLab[0];
 437       color = GetColor(lab+1);
 438       if( lab>=0 ){
 439         AliHLTTPCCAMCTrack &mc = fPerf->MCTracks()[lab];
 440         if( mc.P()>=1. ) color = kRed;
 441         else if(fDrawOnlyRef) return;
 442       }
 443     }else color=GetColorZ( h.Z() );
 444   }
 445   if( width>0 )fMarker.SetMarkerSize(width);
 446   else fMarker.SetMarkerSize(.3);
 447   fMarker.SetMarkerColor(color);
 448   Double_t vx, vy;
 449   Slice2View( h.X(), h.Y(), &vx, &vy );
 450
 451   fYX->cd();
 452   fMarker.DrawMarker(vx, vy);
 453   fZX->cd();
 454   fMarker.DrawMarker(h.Z(), vy);
 455 }
 456
 457 void AliHLTTPCCADisplay::DrawGBHits( AliHLTTPCCAGBTracker &tracker, Int_t color, Size_t width )
 458 {
 459   // draw hits
 460
 461   if( !fPerf ) return;
 462   if( width<0 ) width = .3;
 463
 464   for( Int_t iHit = 0; iHit<tracker.NHits(); iHit++ ){
 465     AliHLTTPCCAGBHit &h = tracker.Hits()[iHit];
 466     Int_t imc = fPerf->HitLabels()[h.ID()].fLab[0];
 467     AliHLTTPCCAMCTrack *mc = (imc>=0) ?&(fPerf->MCTracks()[imc]) :0;
 468     if(fDrawOnlyRef && (!mc || (mc->P()<1))) continue;
 469     Int_t col = color;
 470     if( color<0 ){
 471       col = GetColor(imc+1) ;
 472       if( mc && (mc->P()>=1.) ) col = kRed;
 473     }
 474
 475     AliHLTTPCCATracker &slice = tracker.Slices()[h.ISlice()];
 476     SetSliceTransform(&slice);
 477
 478     fMarker.SetMarkerSize(width);
 479     fMarker.SetMarkerColor(col);
 480     Double_t vx, vy;
 481     Slice2View( h.X(), h.Y(), &vx, &vy );
 482
 483     fYX->cd();
 484     fMarker.DrawMarker(vx, vy);
 485     fZX->cd();
 486     fMarker.DrawMarker(h.Z(), vy);
 487   }
 488 }
 489
 490 void AliHLTTPCCADisplay::DrawSliceHit( Int_t iRow, Int_t iHit, Int_t color, Size_t width )
 491 {
 492   // draw hit
 493   if( !fSlice ) return;
 494   const AliHLTTPCCARow &row = fSlice->Row(iRow);
 495   float y0 = row.Grid().YMin();
 496   float z0 = row.Grid().ZMin();
 497   float stepY = row.HstepY();
 498   float stepZ = row.HstepZ();
 499   const uint4* tmpint4 = fSlice->RowData() + row.FullOffset();
 500   const ushort2 *hits = reinterpret_cast<const ushort2*>(tmpint4);
 501   ushort2 hh = hits[iHit];
 502   Float_t x = row.X();
 503   Float_t y = y0 + hh.x*stepY;
 504   Float_t z = z0 + hh.y*stepZ;
 505
 506   SetSliceTransform(fSlice);
 507
 508   if( color<0 ){
 509     if( fPerf && fGB ){
 510       Int_t id = fGB->FirstSliceHit()[fSlice->Param().ISlice()]+fSlice->HitInputIDs()[row.FirstHit()+iHit];
 511       AliHLTTPCCAGBHit &h = fGB->Hits()[id];
 512       Int_t lab = fPerf->HitLabels()[h.ID()].fLab[0];
 513       color = GetColor(lab+1);
 514       if( lab>=0 ){
 515         AliHLTTPCCAMCTrack &mc = fPerf->MCTracks()[lab];
 516         if( mc.P()>=1. ) color = kRed;
 517         else if(fDrawOnlyRef) return;
 518       }
 519     }else color=GetColorZ( z );
 520   }
 521   if( width>0 )fMarker.SetMarkerSize(width);
 522   else fMarker.SetMarkerSize(.3);
 523   fMarker.SetMarkerColor(color);
 524   Double_t vx, vy;
 525   Slice2View( x, y, &vx, &vy );
 526   fYX->cd();
 527   fMarker.DrawMarker(vx, vy);
 528   fZX->cd();
 529   fMarker.DrawMarker(z, vy);
 530 }
 531
 532 void AliHLTTPCCADisplay::DrawSliceHits( Int_t color, Size_t width )
 533 {
 534
 535   // draw hits
 536
 537   for( Int_t iRow=0; iRow<fSlice->Param().NRows(); iRow++ ){
 538     const AliHLTTPCCARow &row = fSlice->Row(iRow);
 539     for( Int_t ih=0; ih<row.NHits(); ih++ ){
 540       DrawSliceHit( iRow, ih, color, width );
 541     }
 542   }
 543 }
 544
 545
 546 void AliHLTTPCCADisplay::DrawSliceLink( Int_t iRow, Int_t iHit, Int_t colorUp, Int_t colorDn, Int_t width )
 547 {
 548   // draw link between clusters
 549
 550   if( !fPerf || !fGB) return;
 551   AliHLTTPCCAGBTracker &tracker = *fGB;
 552   if( width<0 ) width = 1.;
 553   fLine.SetLineWidth( width );
 554   Int_t colUp = colorUp>=0 ? colorUp :kMagenta;
 555   Int_t colDn = colorDn>=0 ? colorDn :kBlack;
 556   if( iRow<1 || iRow>=fSlice->Param().NRows()-1 ) return;
 557
 558   const AliHLTTPCCARow& row = fSlice->Row(iRow);
 559   const AliHLTTPCCARow& rowUp = fSlice->Row(iRow+1);
 560   const AliHLTTPCCARow& rowDn = fSlice->Row(iRow-1);
 561
 562   Int_t id = fSlice->HitInputIDs()[row.FirstHit()+iHit];
 563   AliHLTTPCCAGBHit &h = tracker.Hits()[tracker.FirstSliceHit()[fSlice->Param().ISlice()]+id];
 564   Short_t iUp = ((Short_t*)(fSlice->RowData() + row.FullOffset()))[row.FullLinkOffset()+iHit];
 565   Short_t iDn = ((Short_t*)(fSlice->RowData() + row.FullOffset()))[row.FullLinkOffset()+row.NHits()+iHit];
 566
 567   if( iUp>=0){
 568     Int_t id1 =fSlice->HitInputIDs()[rowUp.FirstHit()+iUp];
 569     AliHLTTPCCAGBHit &h1 = tracker.Hits()[tracker.FirstSliceHit()[fSlice->Param().ISlice()]+id1];
 570     Double_t vx, vy, vx1, vy1;
 571     Slice2View( h.X(), h.Y(), &vx, &vy );
 572     Slice2View( h1.X(), h1.Y(), &vx1, &vy1 );
 573     fLine.SetLineColor( colUp );
 574     fYX->cd();
 575     fLine.DrawLine(vx-.1,vy,vx1-.1,vy1);
 576     fZX->cd();
 577     fLine.DrawLine(h.Z()-1.,vy,h1.Z()-1.,vy1);
 578   }
 579   if( iDn>=0){
 580     Int_t id1 =fSlice->HitInputIDs()[rowDn.FirstHit()+iDn];
 581     AliHLTTPCCAGBHit &h1 = tracker.Hits()[tracker.FirstSliceHit()[fSlice->Param().ISlice()]+id1];
 582     Double_t vx, vy, vx1, vy1;
 583     Slice2View( h.X(), h.Y(), &vx, &vy );
 584     Slice2View( h1.X(), h1.Y(), &vx1, &vy1 );
 585     fLine.SetLineColor( colDn );
 586     fYX->cd();
 587     fLine.DrawLine(vx+.1,vy,vx1+.1,vy1);
 588     fZX->cd();
 589     fLine.DrawLine(h.Z()+1.,vy,h1.Z()+1.,vy1);
 590   }
 591 }
 592
 593
 594 void AliHLTTPCCADisplay::DrawSliceLinks( Int_t colorUp, Int_t colorDn, Int_t width )
 595 {
 596   // draw links between clusters
 597
 598   for( Int_t iRow=1; iRow<fSlice->Param().NRows()-1; iRow++){
 599     const AliHLTTPCCARow& row = fSlice->Row(iRow);
 600     for( Int_t ih=0; ih<row.NHits(); ih++ ){
 601       DrawSliceLink( iRow, ih, colorUp, colorDn, width );
 602     }
 603   }
 604 }
 605
 606
 607
 608 Int_t AliHLTTPCCADisplay::GetTrackMC( const AliHLTTPCCADisplayTmpHit *vHits, Int_t NHits )
 609 {
 610   // get MC label for the track
 611
 612   AliHLTTPCCAGBTracker &tracker = *fGB;
 613
 614   Int_t label = -1;
 615   Double_t purity = 0;
 616   Int_t *lb = new Int_t[NHits*3];
 617   Int_t nla=0;
 618   //std::cout<<"\n\nTrack hits mc: "<<std::endl;
 619   for( Int_t ihit=0; ihit<NHits; ihit++){
 620     AliHLTTPCCAGBHit &h = tracker.Hits()[vHits[ihit].ID()];
 621     AliHLTTPCCAPerformance::AliHLTTPCCAHitLabel &l = fPerf->HitLabels()[h.ID()];
 622     if(l.fLab[0]>=0 ) lb[nla++]= l.fLab[0];
 623     if(l.fLab[1]>=0 ) lb[nla++]= l.fLab[1];
 624     if(l.fLab[2]>=0 ) lb[nla++]= l.fLab[2];
 625     //std::cout<<ihit<<":  "<<l.fLab[0]<<" "<<l.fLab[1]<<" "<<l.fLab[2]<<std::endl;
 626   }
 627   sort( lb, lb+nla );
 628   Int_t labmax = -1, labcur=-1, lmax = 0, lcurr=0, nh=0;
 629   //std::cout<<"MC track IDs :"<<std::endl;
 630   for( Int_t i=0; i<nla; i++ ){
 631     if( lb[i]!=labcur ){
 632       if( 0 && i>0 && lb[i-1]>=0 ){
 633         AliHLTTPCCAMCTrack &mc = fPerf->MCTracks()[lb[i-1]];
 634         std::cout<<lb[i-1]<<": nhits="<<nh<<", pdg="<< mc.PDG()<<", Pt="<<mc.Pt()<<", P="<<mc.P()
 635                  <<", par="<<mc.Par()[0]<<" "<<mc.Par()[1]<<" "<<mc.Par()[2]
 636                  <<" "<<mc.Par()[3]<<" "<<mc.Par()[4]<<" "<<mc.Par()[5]<<" "<<mc.Par()[6]<<std::endl;
 637
 638       }
 639       nh=0;
 640       if( labcur>=0 && lmax<lcurr ){
 641         lmax = lcurr;
 642         labmax = labcur;
 643       }
 644       labcur = lb[i];
 645       lcurr = 0;
 646     }
 647     lcurr++;
 648     nh++;
 649   }
 650   if( 0 && nla-1>0 && lb[nla-1]>=0 ){
 651     AliHLTTPCCAMCTrack &mc = fPerf->MCTracks()[lb[nla-1]];
 652     std::cout<<lb[nla-1]<<": nhits="<<nh<<", pdg="<< mc.PDG()<<", Pt="<<mc.Pt()<<", P="<<mc.P()
 653              <<", par="<<mc.Par()[0]<<" "<<mc.Par()[1]<<" "<<mc.Par()[2]
 654              <<" "<<mc.Par()[3]<<" "<<mc.Par()[4]<<" "<<mc.Par()[5]<<" "<<mc.Par()[6]<<std::endl;
 655
 656   }
 657   if( labcur>=0 && lmax<lcurr ){
 658     lmax = lcurr;
 659     labmax = labcur;
 660   }
 661   lmax = 0;
 662   for( Int_t ihit=0; ihit<NHits; ihit++){
 663     AliHLTTPCCAGBHit &h = tracker.Hits()[vHits[ihit].ID()];
 664     AliHLTTPCCAPerformance::AliHLTTPCCAHitLabel &l = fPerf->HitLabels()[h.ID()];
 665     if( l.fLab[0] == labmax || l.fLab[1] == labmax || l.fLab[2] == labmax
 666         ) lmax++;
 667   }
 668   label = labmax;
 669   purity = ( (NHits>0) ?double(lmax)/double(NHits) :0 );
 670   if( lb ) delete[] lb;
 671   if( purity<.9 ) label = -1;
 672   return label;
 673 }
 674
 675 Bool_t AliHLTTPCCADisplay::DrawTrack( AliHLTTPCCATrackParam t, Double_t Alpha, const AliHLTTPCCADisplayTmpHit *vHits,
 676                                     Int_t NHits, Int_t color, Int_t width, Bool_t pPoint )
 677 {
 678   // draw track
 679
 680   if(NHits<2 ) return 0;
 681
 682   AliHLTTPCCAGBTracker &tracker = *fGB;
 683   if( width<0 ) width = 2;
 684
 685   if(fDrawOnlyRef ){
 686     Int_t lab = GetTrackMC(vHits, NHits);
 687     if( lab<0 ) return 0;
 688     AliHLTTPCCAMCTrack &mc = fPerf->MCTracks()[lab];
 689     if(mc.P()<1) return 0;
 690   }
 691
 692   if( color < 0 ){
 693     //color = GetColorZ( (vz[0]+vz[mHits-1])/2. );
 694     //color = GetColorK(t.GetKappa());
 695     Int_t lab = GetTrackMC(vHits, NHits);
 696     color = GetColor( lab +1 );
 697     if( lab>=0 ){
 698       AliHLTTPCCAMCTrack &mc = fPerf->MCTracks()[lab];
 699       if( mc.P()>=1. ) color = kRed;
 700     }
 701   }
 702
 703   if( t.SinPhi()>.999 )  t.SetSinPhi( .999 );
 704   else if( t.SinPhi()<-.999 )  t.SetSinPhi( -.999 );
 705
 706   //  Int_t iSlice = fSlice->Param().ISlice();
 707
 708   //sort(vHits, vHits + NHits, AliHLTTPCCADisplayTmpHit::CompareHitZ );
 709
 710   Double_t vx[2000], vy[2000], vz[2000];
 711   Int_t mHits = 0;
 712
 713   //Int_t oldSlice = -1;
 714   Double_t alpha = ( TMath::Abs(Alpha+1)<1.e-4 ) ?fSlice->Param().Alpha() :Alpha;
 715   AliHLTTPCCATrackParam tt = t;
 716
 717   for( Int_t iHit=0; iHit<NHits; iHit++ ){
 718
 719     AliHLTTPCCAGBHit &h = tracker.Hits()[vHits[iHit].ID()];
 720
 721     Double_t hCos = TMath::Cos( alpha - tracker.Slices()[h.ISlice()].Param().Alpha());
 722     Double_t hSin = TMath::Sin( alpha - tracker.Slices()[h.ISlice()].Param().Alpha());
 723     Double_t x0=h.X(), y0=h.Y(), z1=h.Z();
 724     Double_t x1 = x0*hCos + y0*hSin;
 725     Double_t y1 = y0*hCos - x0*hSin;
 726
 727     {
 728       Double_t dx = x1-tt.X();
 729       Double_t dy = y1-tt.Y();
 730       if( dx*dx+dy*dy>1. ){
 731         Double_t dalpha = TMath::ATan2( dy, dx );
 732         if( tt.Rotate(dalpha ) ){
 733           alpha+=dalpha;
 734           hCos = TMath::Cos( alpha - tracker.Slices()[h.ISlice()].Param().Alpha());
 735           hSin = TMath::Sin( alpha - tracker.Slices()[h.ISlice()].Param().Alpha());
 736           x1 = x0*hCos + y0*hSin;
 737           y1 = y0*hCos - x0*hSin;
 738         }
 739       }
 740     }
 741     SetSliceTransform( alpha );
 742
 743     //t.GetDCAPoint( x1, y1, z1, x1, y1, z1 );
 744   std::cout<<"mark 3"<<std::endl;
 745     Bool_t ok = tt.TransportToX(x1,.999);
 746   std::cout<<"mark 4"<<std::endl;
 747   if( 1||ok ){
 748       x1 = tt.X();
 749       y1 = tt.Y();
 750       z1 = tt.Z();
 751     }
 752
 753     Slice2View(x1, y1, &x1, &y1 );
 754     vx[mHits] = x1;
 755     vy[mHits] = y1;
 756     vz[mHits] = z1;
 757     mHits++;
 758     for( int j=0; j<0; j++ ){
 759       x0=h.X()+j; y0=h.Y(); z1=h.Z();
 760       x1 = x0*hCos + y0*hSin;
 761       y1 = y0*hCos - x0*hSin;
 762       ok = tt.TransportToX(x1,.999);
 763       if( ok ){
 764         x1 = tt.X();
 765         y1 = tt.Y();
 766         z1 = tt.Z();
 767       }
 768
 769       Slice2View(x1, y1, &x1, &y1 );
 770       vx[mHits] = x1;
 771       vy[mHits] = y1;
 772       vz[mHits] = z1;
 773       mHits++;
 774     }
 775   }
 776   if( pPoint ){
 777     Double_t x1=t.X(), y1=t.Y(), z1=t.Z();
 778     Double_t a = ( TMath::Abs(Alpha+1)<1.e-4 ) ?fSlice->Param().Alpha() :Alpha;
 779     SetSliceTransform( a );
 780
 781     Slice2View(x1, y1, &x1, &y1 );
 782     Double_t dx = x1 - vx[0];
 783     Double_t dy = y1 - vy[0];
 784     //std::cout<<x1<<" "<<y1<<" "<<vx[0]<<" "<<vy[0]<<" "<<dx<<" "<<dy<<std::endl;
 785     Double_t d0 = dx*dx + dy*dy;
 786     dx = x1 - vx[mHits-1];
 787     dy = y1 - vy[mHits-1];
 788     //std::cout<<x1<<" "<<y1<<" "<<vx[mHits-1]<<" "<<vy[mHits-1]<<" "<<dx<<" "<<dy<<std::endl;
 789     Double_t d1 = dx*dx + dy*dy;
 790     //std::cout<<"d0, d1="<<d0<<" "<<d1<<std::endl;
 791     if( d1<d0 ){
 792       vx[mHits] = x1;
 793       vy[mHits] = y1;
 794       vz[mHits] = z1;
 795       mHits++;
 796     } else {
 797       for( Int_t i = mHits; i>0; i-- ){
 798         vx[i] = vx[i-1];
 799         vy[i] = vy[i-1];
 800         vz[i] = vz[i-1];
 801       }
 802       vx[0] = x1;
 803       vy[0] = y1;
 804       vz[0] = z1;
 805       mHits++;
 806     }
 807   }
 808
 809
 810   fLine.SetLineColor(color);
 811   fLine.SetLineWidth(width);
 812   fArc.SetFillStyle(0);
 813   fArc.SetLineColor(color);
 814   fArc.SetLineWidth(width);
 815   TPolyLine pl;
 816   pl.SetLineColor(color);
 817   pl.SetLineWidth(width);
 818   TPolyLine plZ;
 819   plZ.SetLineColor(color);
 820   plZ.SetLineWidth(width);
 821
 822   fMarker.SetMarkerSize(width/2.);
 823   fMarker.SetMarkerColor(color);
 824
 825   fYX->cd();
 826   pl.DrawPolyLine(mHits,vx,vy);
 827   {
 828     fMarker.DrawMarker(vx[0],vy[0]);
 829     fMarker.DrawMarker(vx[mHits-1],vy[mHits-1]);
 830   }
 831   fZX->cd();
 832   plZ.DrawPolyLine(mHits,vz,vy);
 833   fMarker.DrawMarker(vz[0],vy[0]);
 834   fMarker.DrawMarker(vz[mHits-1],vy[mHits-1]);
 835
 836   fLine.SetLineWidth(1);
 837   return 1;
 838 }
 839
 840
 841 Bool_t AliHLTTPCCADisplay::DrawTracklet( AliHLTTPCCATrackParam &track, const Int_t *hitstore, Int_t color, Int_t width, Bool_t pPoint )
 842 {
 843   // draw tracklet
 844   AliHLTTPCCAGBTracker &tracker = *fGB;
 845   AliHLTTPCCADisplayTmpHit vHits[200];
 846   Int_t nHits = 0;
 847   for( Int_t iRow=0; iRow<fSlice->Param().NRows(); iRow++ ){
 848     Int_t iHit = hitstore[iRow];
 849     if( iHit<0 ) continue;
 850     const AliHLTTPCCARow &row = fSlice->Row(iRow);
 851     Int_t id = fSlice->HitInputIDs()[row.FirstHit()+iHit];
 852     Int_t iGBHit = tracker.FirstSliceHit()[fSlice->Param().ISlice()]+id;
 853     AliHLTTPCCAGBHit &h = tracker.Hits()[iGBHit];
 854     vHits[nHits].SetID( iGBHit );
 855     vHits[nHits].SetS( 0 );
 856     vHits[nHits].SetZ( h.Z() );
 857     nHits++;
 858   }
 859   return DrawTrack( track, -1, vHits, nHits, color,width,pPoint );
 860 }
 861
 862
 863 void AliHLTTPCCADisplay::DrawSliceOutTrack( AliHLTTPCCATrackParam &t, Double_t alpha, Int_t itr, Int_t color, Int_t width )
 864 {
 865   // draw slice track
 866
 867   AliHLTTPCCAOutTrack &track = fSlice->OutTracks()[itr];
 868   if( track.NHits()<2 ) return;
 869
 870   AliHLTTPCCAGBTracker &tracker = *fGB;
 871   AliHLTTPCCADisplayTmpHit vHits[200];
 872
 873   for( Int_t ih=0; ih<track.NHits(); ih++ ){
 874     Int_t id = tracker.FirstSliceHit()[fSlice->Param().ISlice()] + fSlice->OutTrackHits()[track.FirstHitRef()+ih];
 875     AliHLTTPCCAGBHit &h = tracker.Hits()[id];
 876     vHits[ih].SetID( id );
 877     vHits[ih].SetS( 0 );
 878     vHits[ih].SetZ( h.Z() );
 879   }
 880
 881   DrawTrack( t, alpha, vHits, track.NHits(), color, width, 1 );
 882 }
 883
 884 void AliHLTTPCCADisplay::DrawSliceOutTrack( Int_t itr, Int_t color, Int_t width )
 885 {
 886   // draw slice track
 887
 888   AliHLTTPCCAOutTrack &track = fSlice->OutTracks()[itr];
 889   if( track.NHits()<2 ) return;
 890
 891   AliHLTTPCCAGBTracker &tracker = *fGB;
 892   AliHLTTPCCADisplayTmpHit vHits[200];
 893
 894   for( Int_t ih=0; ih<track.NHits(); ih++ ){
 895     Int_t id = tracker.FirstSliceHit()[fSlice->Param().ISlice()] + fSlice->OutTrackHits()[track.FirstHitRef()+ih];
 896     AliHLTTPCCAGBHit &h = tracker.Hits()[id];
 897     vHits[ih].SetID( id );
 898     vHits[ih].SetS( 0 );
 899     vHits[ih].SetZ( h.Z() );
 900   }
 901
 902   DrawTrack( track.StartPoint(), -1, vHits, track.NHits(), color, width );
 903 }
 904
 905
 906 void AliHLTTPCCADisplay::DrawSliceTrack( Int_t itr, Int_t color )
 907 {
 908   // draw slice track
 909
 910   AliHLTTPCCATrack &track = fSlice->Tracks()[itr];
 911   if( track.NHits()<2 ) return;
 912
 913   AliHLTTPCCAGBTracker &tracker = *fGB;
 914   AliHLTTPCCADisplayTmpHit vHits[200];
 915   for( Int_t ith=0; ith<track.NHits(); ith++ ){
 916     Int_t ic = (fSlice->TrackHits()[track.FirstHitID()+ith]);
 917     const AliHLTTPCCARow &row = fSlice->ID2Row(ic);
 918     Int_t ih = fSlice->ID2IHit(ic);
 919     int id = fSlice->HitInputIDs()[row.FirstHit()+ih];
 920     Int_t gbID = tracker.FirstSliceHit()[fSlice->Param().ISlice()] + id;
 921     AliHLTTPCCAGBHit &h = tracker.Hits()[gbID];
 922     vHits[ith].SetID( gbID );
 923     vHits[ith].SetS( 0 );
 924     vHits[ith].SetZ( h.Z() );
 925   }
 926
 927   DrawTrack( track.Param(), -1, vHits, track.NHits(), color,-1 );
 928   //track.Param().Print();
 929 }
 930
 931
 932 void AliHLTTPCCADisplay::DrawGBTrack( Int_t itr, Int_t color, Int_t width )
 933 {
 934   // draw global track
 935
 936   AliHLTTPCCAGBTracker &tracker = *fGB;
 937   AliHLTTPCCADisplayTmpHit vHits[1000];
 938
 939   AliHLTTPCCAGBTrack &track = tracker.Tracks()[itr];
 940   if( track.NHits()<2 ) return;
 941
 942   for( Int_t ih=0; ih<track.NHits(); ih++ ){
 943     Int_t i = tracker.TrackHits()[ track.FirstHitRef() + ih];
 944     AliHLTTPCCAGBHit &h = tracker.Hits()[i];
 945     vHits[ih].SetID( i );
 946     vHits[ih].SetS( 0 );
 947     vHits[ih].SetZ( h.Z() );
 948   }
 949
 950   DrawTrack( track.Param(), track.Alpha(), vHits, track.NHits(), color, width );
 951 }
 952
 953
 954 void AliHLTTPCCADisplay::DrawGBTrackFast( AliHLTTPCCAGBTracker &tracker, Int_t itr, Int_t color )
 955 {
 956   // draw global track
 957
 958   AliHLTTPCCAGBTrack &track = tracker.Tracks()[itr];
 959   if( track.NHits()<2 ) return;
 960   Int_t width = 1;
 961
 962   AliHLTTPCCADisplayTmpHit *vHits = new AliHLTTPCCADisplayTmpHit[track.NHits()];
 963   AliHLTTPCCATrackParam t = track.Param();
 964
 965   for( Int_t ih=0; ih<track.NHits(); ih++ ){
 966     Int_t i = tracker.TrackHits()[ track.FirstHitRef() + ih];
 967     AliHLTTPCCAGBHit *h = &(tracker.Hits()[i]);
 968     vHits[ih].SetID( i );
 969     vHits[ih].SetS( 0 );
 970     vHits[ih].SetZ( h->Z() );
 971   }
 972
 973   sort(vHits, vHits + track.NHits(), AliHLTTPCCADisplayTmpHit::CompareHitZ );
 974   Int_t colorY = color;
 975   {
 976     AliHLTTPCCAGBHit &h1 = tracker.Hits()[ vHits[0].ID()];
 977     AliHLTTPCCAGBHit &h2 = tracker.Hits()[ vHits[track.NHits()-1].ID()];
 978     if( color<0 ) color = GetColorZ( (h1.Z()+h2.Z())/2. );
 979     Double_t gx1, gy1, gx2, gy2;
 980     Slice2View(h1.X(), h1.Y(), &gx1, &gy1 );
 981     Slice2View(h2.X(), h2.Y(), &gx2, &gy2 );
 982     if( colorY<0 ) colorY = GetColorY( (gy1+gy2)/2. );
 983     color = colorY = GetColorK(t.GetQPt());
 984   }
 985
 986   fMarker.SetMarkerColor(color);//kBlue);
 987   fMarker.SetMarkerSize(1.);
 988   fLine.SetLineColor(color);
 989   fLine.SetLineWidth(width);
 990   fArc.SetFillStyle(0);
 991   fArc.SetLineColor(color);
 992   fArc.SetLineWidth(width);
 993   TPolyLine pl;
 994   pl.SetLineColor(colorY);
 995   pl.SetLineWidth(width);
 996
 997   Int_t oldSlice = -1;
 998   Double_t alpha = track.Alpha();
 999   // YX
1000   {
1001
1002     AliHLTTPCCAGBHit &h1 = tracker.Hits()[vHits[0].ID()];
1003     AliHLTTPCCAGBHit &h2 = tracker.Hits()[vHits[track.NHits()-1].ID()];
1004     Float_t x1, y1, z1, x2, y2, z2;
1005     Double_t vx1, vy1, vx2, vy2;
1006
1007     if( h1.ISlice() != oldSlice ){
1008       t.Rotate( tracker.Slices()[h1.ISlice()].Param().Alpha() - alpha);
1009       oldSlice = h1.ISlice();
1010       alpha = tracker.Slices()[h1.ISlice()].Param().Alpha();
1011       SetSliceTransform( &(tracker.Slices()[oldSlice]) );
1012     }
1013     t.GetDCAPoint( h1.X(), h1.Y(), h1.Z(), x1, y1, z1, fSlice->Param().Bz()  );
1014     Slice2View(x1, y1, &vx1, &vy1 );
1015
1016     if( h2.ISlice() != oldSlice ){
1017       t.Rotate( tracker.Slices()[h2.ISlice()].Param().Alpha() - alpha);
1018       oldSlice = h2.ISlice();
1019       alpha = tracker.Slices()[h2.ISlice()].Param().Alpha();
1020       SetSliceTransform( &(tracker.Slices()[oldSlice]) );
1021     }
1022     t.GetDCAPoint( h2.X(), h2.Y(), h2.Z(), x2, y2, z2, fSlice->Param().Bz()  );
1023     Slice2View(x2, y2, &vx2, &vy2 );
1024
1025     Double_t x0 = t.GetX();
1026     Double_t y0 = t.GetY();
1027     Double_t sinPhi = t.GetSinPhi();
1028     Double_t k = t.GetKappa( fSlice->Param().Bz() );
1029     Double_t ex = t.GetCosPhi();
1030     Double_t ey = sinPhi;
1031
1032     if( TMath::Abs(k)>1.e-4 ){
1033
1034       fYX->cd();
1035
1036       Double_t r = 1/TMath::Abs(k);
1037       Double_t xc = x0 -ey*(1/k);
1038       Double_t yc = y0 +ex*(1/k);
1039
1040       Double_t vx, vy;
1041       Slice2View( xc, yc, &vx, &vy );
1042
1043       Double_t a1 = TMath::ATan2(vy1-vy, vx1-vx)/TMath::Pi()*180.;
1044       Double_t a2 = TMath::ATan2(vy2-vy, vx2-vx)/TMath::Pi()*180.;
1045       if( a1<0 ) a1+=360;
1046       if( a2<0 ) a2+=360;
1047       if( a2<a1 ) a2+=360;
1048       Double_t da = TMath::Abs(a2-a1);
1049       if( da>360 ) da-= 360;
1050       if( da>180 ){
1051         da = a1;
1052         a1 = a2;
1053         a2 = da;
1054         if( a2<a1 ) a2+=360;
1055       }
1056       fArc.DrawArc(vx,vy,r, a1,a2,"only");
1057       //fArc.DrawArc(vx,vy,r, 0,360,"only");
1058    } else {
1059       fYX->cd();
1060       fLine.DrawLine(vx1,vy1, vx2, vy2 );
1061     }
1062   }
1063
1064   // ZX
1065   Double_t py[track.NHits()], pz[track.NHits()];
1066
1067   for( Int_t iHit=0; iHit<track.NHits(); iHit++ ){
1068
1069     AliHLTTPCCAGBHit &h1 = tracker.Hits()[vHits[iHit].ID()];
1070     Float_t x1, y1, z1;
1071     Double_t vx1, vy1;
1072     if( h1.ISlice() != oldSlice ){
1073       t.Rotate( tracker.Slices()[h1.ISlice()].Param().Alpha() - alpha);
1074       oldSlice = h1.ISlice();
1075       alpha = tracker.Slices()[h1.ISlice()].Param().Alpha();
1076       SetSliceTransform( &(tracker.Slices()[oldSlice]) );
1077     }
1078     t.GetDCAPoint( h1.X(), h1.Y(), h1.Z(), x1, y1, z1, fSlice->Param().Bz()  );
1079     Slice2View(x1, y1, &vx1, &vy1 );
1080     py[iHit] = vy1;
1081     pz[iHit] = z1;
1082   }
1083
1084
1085   fZX->cd();
1086   pl.DrawPolyLine(track.NHits(),pz,py);
1087
1088   fLine.SetLineWidth(1);
1089   delete[] vHits;
1090 }
1091
1092
1093
1094
1095
1096 #ifdef XXXX
1097
1098
1099
1100
1101 void AliHLTTPCCADisplay::DrawMergedHit( Int_t iRow, Int_t iHit, Int_t color )
1102 {
1103   // connect two cells on display
1104
1105 #ifdef XXX
1106
1107   const AliHLTTPCCARow &row = fSlice->Row(iRow);
1108   AliHLTTPCCAHit &h = row.Hits()[iHit];
1109   AliHLTTPCCAHit &hyz = row.HitsYZ()[iHit];
1110
1111   Double_t x = row.X();
1112   Double_t y = hyz.Y();
1113   Double_t z = hyz.Z();
1114   Double_t x1 = x, x2 = x;
1115   Double_t y1 = y, y2 = y;
1116   Double_t z1 = z, z2 = z;
1117   Int_t iRow1 = iRow, iHit1 = iHit;
1118   Int_t iRow2 = iRow, iHit2 = iHit;
1119
1120   if( fSlice->HitLinksDown()[]>=0 ){
1121     iRow1 = iRow - 1;
1122     iHit1 = h.LinkDown();
1123     AliHLTTPCCARow &row1 = fSlice->Rows()[iRow1];
1124     AliHLTTPCCAHitYZ &h1 = row1.HitsYZ()[iHit1];
1125     x1 = row1.X();
1126     y1 = h1.Y();
1127     z1 = h1.Z();
1128   }
1129   if( h.LinkUp()>=0 ){
1130     iRow2 = iRow+1;
1131     iHit2 = h.LinkUp();
1132     AliHLTTPCCARow &row2 = fSlice->Rows()[iRow2];
1133     AliHLTTPCCAHitYZ &h2 = row2.HitsYZ()[iHit2];
1134     x2 = row2.X();
1135     y2 = h2.Y();
1136     z2 = h2.Z();
1137   }
1138   if( color<0 ) color = GetColorZ( (z+z1+z2)/3. );
1139
1140
1141   Slice2View(x,y, &x, &y );
1142   Slice2View(x1,y1, &x1, &y1 );
1143   Slice2View(x2,y2, &x2, &y2 );
1144
1145   Double_t lx[] = { x1, x, x2 };
1146   Double_t ly[] = { y1, y, y2 };
1147   Double_t lz[] = { z1, z, z2 };
1148
1149   fPLine.SetLineColor(color);
1150   fPLine.SetLineWidth(1);
1151   //fPLine.SetFillColor(color);
1152   fPLine.SetFillStyle(-1);
1153
1154   fYX->cd();
1155   fPLine.DrawPolyLine(3, lx, ly );
1156   fZX->cd();
1157   fPLine.DrawPolyLine(3, lz, ly );
1158   DrawHit( iRow, iHit, color );
1159   DrawHit( iRow1, iHit1, color );
1160   DrawHit( iRow2, iHit2, color );
1161 #endif
1162 }
1163
1164
1165 void AliHLTTPCCADisplay::DrawTrack( AliHLTTPCCATrack &track, Int_t color, Bool_t DrawHits )
1166 {
1167   // draw track
1168
1169   if( track.NHits()<2 ) return;
1170   Int_t width = 2;
1171
1172   AliHLTTPCCADisplayTmpHit *vHits = new AliHLTTPCCADisplayTmpHit[track.NHits()];
1173   AliHLTTPCCATrackParam &t = track.Param();
1174
1175   Int_t iID = track.FirstHitID();
1176   Int_t nhits = 0;
1177   {
1178     Int_t iHit = 0;
1179     for( Int_t ih=0; ih<track.NHits(); ih++ ){
1180       Int_t i = fSlice->TrackHits()[iID];
1181       AliHLTTPCCAHit *h = &(fSlice->ID2Hit( i ));
1182       AliHLTTPCCARow &row = fSlice->ID2Row(i);
1183       vHits[iHit].ID() = i;
1184       vHits[iHit].S() = t.GetS( row.X(), h->Y() );
1185       vHits[iHit].Z() = h->Z();
1186       iHit++;
1187       nhits++;
1188       iID++;
1189     }
1190   }
1191   sort(vHits, vHits + track.NHits(), AliHLTTPCCADisplayTmpHit::CompareHitZ );
1192   //cout<<"Draw track, nhits = "<<nhits<<endl;
1193   {
1194     AliHLTTPCCAHit &c1 = fSlice->ID2Hit(vHits[0].ID());
1195     AliHLTTPCCAHit &c2 = fSlice->ID2Hit(vHits[track.NHits()-1].ID());
1196     if( color<0 ) color = GetColorZ( (c1.Z()+c2.Z())/2. );
1197   }
1198
1199   fMarker.SetMarkerColor(color);//kBlue);
1200   fMarker.SetMarkerSize(1.);
1201   /*
1202   for( Int_t i=0; i<3; i++){
1203     AliHLTTPCCAHit &c1 = fSlice->ID2Hit(track.HitID()[i]);
1204     AliHLTTPCCARow &row1 = fSlice->ID2Row(track.HitID()[i]);
1205     Double_t vx1, vy1;
1206     Slice2View(row1.X(), c1.Y(), &vx1, &vy1 );
1207     fYX->cd();
1208     fMarker.DrawMarker(vx1,vy1);
1209     fZX->cd();
1210     fMarker.DrawMarker(c1.Z(),vy1);
1211   }
1212   */
1213
1214   //DrawTrackletPoint( fSlice->ID2Point(track.PointID()[0]).Param(), kBlack);//color );
1215   //DrawTrackletPoint( fSlice->ID2Point(track.PointID()[1]).Param(), kBlack);//color );
1216   //cout<<"DrawTrack end points x = "<<fSlice->ID2Point(track.PointID()[0]).Param().GetX()<<" "<<fSlice->ID2Point(track.PointID()[1]).Param().GetX()<<endl;
1217   for( Int_t iHit=0; iHit<track.NHits()-1; iHit++ )
1218   {
1219     AliHLTTPCCAHit &c1 = fSlice->ID2Hit(vHits[iHit].ID());
1220     AliHLTTPCCAHit &c2 = fSlice->ID2Hit(vHits[iHit+1].ID());
1221     AliHLTTPCCARow &row1 = fSlice->ID2Row(vHits[iHit].ID());
1222     AliHLTTPCCARow &row2 = fSlice->ID2Row(vHits[iHit+1].ID());
1223     Float_t x1, y1, z1, x2, y2, z2;
1224     t.GetDCAPoint( row1.X(), c1.Y(), c1.Z(), x1, y1, z1,fSlice->Param().Bz()  );
1225     t.GetDCAPoint( row2.X(), c2.Y(), c2.Z(), x2, y2, z2,fSlice->Param().Bz()  );
1226
1227     //if( color<0 ) color = GetColorZ( (z1+z2)/2. );
1228     Double_t vx1, vy1, vx2, vy2;
1229     Slice2View(x1, y1, &vx1, &vy1 );
1230     Slice2View(x2, y2, &vx2, &vy2 );
1231
1232     fLine.SetLineColor( color );
1233     fLine.SetLineWidth( width );
1234
1235     Double_t x0 = t.GetX();
1236     Double_t y0 = t.GetY();
1237     Double_t sinPhi = t.GetSinPhi();
1238     Double_t k = t.GetKappa();
1239     Double_t ex = t.GetCosPhi();
1240     Double_t ey = sinPhi;
1241
1242     if( TMath::Abs(k)>1.e-4 ){
1243
1244       fArc.SetFillStyle(0);
1245       fArc.SetLineColor(color);
1246       fArc.SetLineWidth(width);
1247
1248       fYX->cd();
1249
1250       Double_t r = 1/TMath::Abs(k);
1251       Double_t xc = x0 -ey*(1/k);
1252       Double_t yc = y0 +ex*(1/k);
1253
1254       Double_t vx, vy;
1255       Slice2View( xc, yc, &vx, &vy );
1256
1257       Double_t a1 = TMath::ATan2(vy1-vy, vx1-vx)/TMath::Pi()*180.;
1258       Double_t a2 = TMath::ATan2(vy2-vy, vx2-vx)/TMath::Pi()*180.;
1259       if( a1<0 ) a1+=360;
1260       if( a2<0 ) a2+=360;
1261       if( a2<a1 ) a2+=360;
1262       Double_t da = TMath::Abs(a2-a1);
1263       if( da>360 ) da-= 360;
1264       if( da>180 ){
1265         da = a1;
1266         a1 = a2;
1267         a2 = da;
1268         if( a2<a1 ) a2+=360;
1269       }
1270       fArc.DrawArc(vx,vy,r, a1,a2,"only");
1271       //fArc.DrawArc(vx,vy,r, 0,360,"only");
1272    } else {
1273       fYX->cd();
1274       fLine.DrawLine(vx1,vy1, vx2, vy2 );
1275     }
1276   }
1277
1278   for( Int_t iHit=0; iHit<track.NHits()-1; iHit++ ){
1279     AliHLTTPCCAHit &c1 = fSlice->ID2Hit(vHits[iHit].ID());
1280     AliHLTTPCCAHit &c2 = fSlice->ID2Hit(vHits[iHit+1].ID());
1281     AliHLTTPCCARow &row1 = fSlice->ID2Row(vHits[iHit].ID());
1282     AliHLTTPCCARow &row2 = fSlice->ID2Row(vHits[iHit+1].ID());
1283
1284     //if( DrawHits ) ConnectHits( fSlice->ID2IRow(vHits[iHit].ID()),c1,
1285     //fSlice->ID2IRow(vHits[iHit+1].ID()),c2, color );
1286     Float_t x1, y1, z1, x2, y2, z2;
1287     t.GetDCAPoint( row1.X(), c1.Y(), c1.Z(), x1, y1, z1,fSlice->Param().Bz()  );
1288     t.GetDCAPoint( row2.X(), c2.Y(), c2.Z(), x2, y2, z2,fSlice->Param().Bz()  );
1289
1290     Double_t vx1, vy1, vx2, vy2;
1291     Slice2View(x1, y1, &vx1, &vy1 );
1292     Slice2View(x2, y2, &vx2, &vy2 );
1293
1294     fLine.SetLineColor(color);
1295     fLine.SetLineWidth(width);
1296
1297     fZX->cd();
1298     fLine.DrawLine(z1,vy1, z2, vy2 );
1299   }
1300   fLine.SetLineWidth(1);
1301   delete[] vHits;
1302 }
1303
1304
1305 void AliHLTTPCCADisplay::DrawTrackletPoint( AliHLTTPCCATrackParam &t, Int_t color )
1306 {
1307   // draw tracklet point
1308
1309   Double_t x = t.GetX();
1310   Double_t y = t.GetY();
1311   Double_t sinPhi = t.GetSinPhi();
1312   Double_t z = t.GetZ();
1313   Double_t dzds = t.GetDzDs();
1314   Double_t ex = t.GetCosPhi();
1315   Double_t ey = sinPhi;
1316
1317   Int_t width = 1;
1318
1319   if( color<0 ) color = GetColorZ( t.GetZ() );
1320
1321   fMarker.SetMarkerColor(color);
1322   fMarker.SetMarkerSize(.5);
1323   fLine.SetLineWidth(width);
1324   fLine.SetLineColor(color);
1325
1326   Double_t vx, vy, vex, vey, vdx, vdy;
1327   Double_t dz = TMath::Sqrt(t.GetErr2Z());
1328   Slice2View( x, y, &vx, &vy );
1329   Slice2View( ex, ey, &vex, &vey );
1330   Slice2View( 0, TMath::Sqrt(t.GetErr2Y())*3.5, &vdx, &vdy);
1331   Double_t d = TMath::Sqrt(vex*vex+vey*vey);
1332   vex/=d;
1333   vey/=d;
1334   fYX->cd();
1335   fMarker.DrawMarker(vx,vy);
1336   fLine.DrawLine(vx,vy,vx+vex*4, vy+vey*4);
1337   fLine.DrawLine(vx-vdx,vy-vdy, vx+vdx, vy+vdy );
1338   fZX->cd();
1339   fMarker.DrawMarker(z,vy);
1340   fLine.DrawLine(z,vy,z+dzds*4, vy+vey*4);
1341   fLine.DrawLine(z-3.5*dz,vy-vdy, z+3.5*dz, vy+vdy );
1342   fLine.SetLineWidth(1);
1343 }
1344 #endif //XXXX