]> git.uio.no Git - u/mrichter/AliRoot.git/blobdiff - EVE/EveDet/AliEveTRDData.cxx
there have been some changes in the on-line tracklet and GTU track
[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveTRDData.cxx
index ec9148fccfc1e7e9669950e0160a5d14d8e6fc08..96e4e0c811bdc0e11624faa63c943be1ebeafd87 100644 (file)
@@ -7,8 +7,11 @@
  * full copyright notice.                                                 *
  **************************************************************************/
 
+#include "TROOT.h"
 #include "TVector.h"
 #include "TLinearFitter.h"
+#include "TCanvas.h"
+#include "TGeoMatrix.h"
 
 #include "TEveTrans.h"
 #include "TEveManager.h"
@@ -20,6 +23,8 @@
 #include "AliEveTRDLoader.h"
 #include "AliEveTRDLoaderImp.h"
 
+#include "AliGeomManager.h"
+#include "AliESDtrack.h"
 #include "AliLog.h"
 #include "AliPID.h"
 #include "AliTrackPointArray.h"
 #include "AliTRDhit.h"
 #include "AliTRDcluster.h"
 #include "AliTRDseedV1.h"
+#include "AliTRDtrackletMCM.h"
+#include "AliTRDtrackletWord.h"
+#include "AliTRDmcmSim.h"
 #include "AliTRDtrackV1.h"
 #include "AliTRDtrackerV1.h"
 #include "AliTRDpadPlane.h"
 #include "AliTRDdigitsManager.h"
-#include "AliTRDdataArrayDigits.h"
+#include "AliTRDmcmSim.h"
 #include "AliTRDarrayADC.h"
 #include "AliTRDSignalIndex.h"
 #include "AliTRDgeometry.h"
@@ -45,6 +53,8 @@ ClassImp(AliEveTRDDigits)
 ClassImp(AliEveTRDClusters)
 ClassImp(AliEveTRDTracklet)
 ClassImp(AliEveTRDTrack)
+ClassImp(AliEveTRDTrackletOnline)
+ClassImp(AliEveTRDmcm)
 
 ///////////////////////////////////////////////////////////
 /////////////   AliEveTRDDigits       /////////////////////
@@ -83,44 +93,53 @@ void AliEveTRDDigits::ComputeRepresentation()
   Int_t nrows = fData.GetNrow(),
         ncols = fData.GetNcol(),
         ntbs  = fData.GetNtime(),
-        det   = fParent->GetID();
+        det   = fParent->GetID(),
+        ly    = AliTRDgeometry::GetLayer(det),
+        stk   = AliTRDgeometry::GetStack(det),
+        sec   = AliTRDgeometry::GetSector(det),
+        vid   = AliGeomManager::LayerToVolUID(AliGeomManager::kTRD1 + ly, stk + AliTRDgeometry::Nstack() * sec);
   Float_t threshold = fParent->GetDigitsThreshold();
+  Short_t sig[7]={0,0,0,10,0,0,0};
 
   AliTRDtransform transform(det);
   AliTRDgeometry *geo = fParent->fGeo;
   AliTRDpadPlane *pp = geo->GetPadPlane(geo->GetLayer(det), geo->GetStack(det));
 
   // express position in tracking coordinates
+  AliTRDcluster c;
   fData.Expand();
   for (Int_t ir = 0; ir < nrows; ir++) {
     dz = pp->GetRowSize(ir);
     for (Int_t ic = 0; ic < ncols; ic++) {
       dy = pp->GetColSize(ic);
       for (Int_t it = 0; it < ntbs; it++) {
-        q = fData.GetDataUnchecked(ir, ic, it);
+        q = fData.GetData(ir, ic, it);
         if (q < threshold) continue;
-
-        Double_t x[6] = {0., 0., Double_t(q), 0., 0., 0.}; 
+        
+/*        Double_t x[6] = {0., 0., Double_t(q), 0., 0., 0.}; 
         Int_t  roc[3] = {ir, ic, 0}; 
-        Bool_t    out = kTRUE;
-        transform.Transform(&x[0], &roc[0], UInt_t(it), out, 0);
+        Bool_t    out = kTRUE;*/
+        
+        new (&c) AliTRDcluster(det, ic, ir, it, sig, vid);
+        transform.Transform(&c);
 
         scale = q < 512 ? q/512. : 1.;
         color  = 50+int(scale*50.);
-    
-        AddQuad(x[1]-.45*dy, x[2]-.5*dz*scale, x[0], .9*dy, dz*scale);
+        
+        AliDebug(4, Form("y[%f] z[%f] x[%f] w[%f] h[%f]\n", c.GetY(), c.GetZ(), c.GetX(), .9*dy, dz*scale));
+        AddQuad(c.GetY(), c.GetZ(), c.GetX(), .9*dy, dz*scale);
         QuadValue(q);
         QuadColor(color);
         QuadId(new TNamed(Form("Charge%d", q), "dummy title"));
       }  // end time loop
     }  // end col loop
   }  // end row loop
-  fData.Compress(1);
+  fData.Compress();
   
   // rotate to global coordinates
   //RefitPlex();
   TEveTrans& t = RefMainTrans();
-  t.SetRotByAngles((geo->GetSector(det)+.5)*AliTRDgeometry::GetAlpha(), 0.,0.);
+  t.SetRotByAngles((sec+.5)*AliTRDgeometry::GetAlpha(), 0.,0.);
 }
 
 //______________________________________________________________________________
@@ -128,7 +147,7 @@ void AliEveTRDDigits::SetData(AliTRDdigitsManager *digits)
 {
   // Set data source.
 
-  Int_t det = fParent->GetID();
+  Int_t det(fParent->GetID());
   AliTRDarrayADC *data = digits->GetDigits(det);
   if(!data->GetDim()) return;
   data->Expand();
@@ -150,23 +169,23 @@ void AliEveTRDDigits::SetData(AliTRDdigitsManager *digits)
       }
       adc = data->GetData(row, col, time);
       if(adc <= 1) continue;
-      fData.SetDataUnchecked(row, col, time, adc);
+      fData.SetData(row, col, time, adc);
       //fIndex->AddIndexTBin(row,col,time);
       //printf("\tr[%d] c[%d] t[%d] ADC[%d]\n", row, col, time, adc);
     } 
   }
-  fData.Compress(1);
+  fData.Compress();
 }
 
 
-//______________________________________________________________________________
-void AliEveTRDDigits::Paint(Option_t *option)
-{
-  // Paint the object.
-
-  if(fParent->GetDigitsBox()) fBoxes.Paint(option);
-  else TEveQuadSet::Paint(option);
-}
+// //______________________________________________________________________________
+// void AliEveTRDDigits::Paint(Option_t *option)
+// {
+//   // Paint the object.
+// 
+//   if(fParent->GetDigitsBox()) fBoxes.Paint(option);
+//   else TEveQuadSet::Paint(option);
+// }
 
 //______________________________________________________________________________
 void AliEveTRDDigits::Reset()
@@ -202,7 +221,7 @@ void AliEveTRDHits::PointSelected(Int_t n)
 {
   // Handle an individual point selection from GL.
 
-  AliTRDhit *h = 0x0;
+  AliTRDhit *h = NULL;
   if(!(h = dynamic_cast<AliTRDhit*>(GetPointId(n)))) return;
   printf("Id[%3d] Det[%3d] Reg[%c] TR[%c] Q[%3d] MC[%d] t[%f]\n", 
     n, h->GetDetector(), 
@@ -234,16 +253,9 @@ void AliEveTRDClusters::PointSelected(Int_t n)
   // Handle an individual point selection from GL.
 
   AliTRDcluster *c = dynamic_cast<AliTRDcluster*>(GetPointId(n));
-  printf("\nDetector             : %d\n", c->GetDetector());
-  printf("Charge               : %f\n", c->GetQ());
-  printf("Sum S                : %4.0f\n", c->GetSumS());
-  printf("Time bin             : %d\n", c->GetLocalTimeBin());
-  printf("Signals              : ");
-  Short_t *cSignals = c->GetSignals();
-  for(Int_t ipad=0; ipad<7; ipad++) printf("%d ", cSignals[ipad]); printf("\n");
-  printf("Central pad          : %d\n", c->GetPadCol());
-  printf("MC track labels      : ");
-  for(Int_t itrk=0; itrk<3; itrk++) printf("%d ", c->GetLabel(itrk)); printf("\n");
+  if(!c) return;
+  c->Print();
+  Emit("PointSelected(Int_t)", n);
   // Bool_t    AliCluster::GetGlobalCov(Float_t* cov) const
   // Bool_t    AliCluster::GetGlobalXYZ(Float_t* xyz) const
   // Float_t   AliCluster::GetSigmaY2() const
@@ -254,7 +266,7 @@ void AliEveTRDClusters::PointSelected(Int_t n)
 //______________________________________________________________________________
 void AliEveTRDClusters::Print(Option_t *o) const
 {
-  AliTRDcluster *c = 0x0;
+  AliTRDcluster *c = NULL;
 
   for(Int_t n = GetN(); n--;){
     if(!(c = dynamic_cast<AliTRDcluster*>(GetPointId(n)))) continue;
@@ -279,13 +291,13 @@ void AliEveTRDClusters::Load(Char_t *w, Bool_t stk) const
     return;
   }
 
-  AliTRDcluster *c = 0x0;
+  AliTRDcluster *c = NULL;
   Int_t n = 0;
   while((n = GetN() && !(c = dynamic_cast<AliTRDcluster*>(GetPointId(n))))) n++;
   if(!c) return;
 
   Int_t det = c->GetDetector();
-  AliEveTRDLoader *loader = 0x0;
+  AliEveTRDLoader *loader = NULL;
   switch(typ){
   case 0:  
     loader = new AliEveTRDLoader("Hits");
@@ -338,7 +350,7 @@ void AliEveTRDClusters::Load(Char_t *w, Bool_t stk) const
 
 //______________________________________________________________________________
 AliEveTRDTracklet::AliEveTRDTracklet(AliTRDseedV1 *trklt):TEveLine()
-  ,fClusters(0x0)
+  ,fClusters(NULL)
 {
   // Constructor.
   SetName("tracklet");
@@ -352,8 +364,8 @@ AliEveTRDTracklet::AliEveTRDTracklet(AliTRDseedV1 *trklt):TEveLine()
   Float_t dzdx = trklt->GetZref(1);
   Float_t tilt = trklt->GetTilt();
   Float_t g[3];
-  AliTRDcluster *c = 0x0;
-  for(Int_t ic=0; ic<AliTRDseed::knTimebins; ic++){
+  AliTRDcluster *c = NULL;
+  for(Int_t ic=0; ic<AliTRDseedV1::kNclusters; ic++){
     if(!(c = trklt->GetClusters(ic))) continue;
     if(!fClusters) AddElement(fClusters = new AliEveTRDClusters());
     dx = x0 - c->GetX();
@@ -400,6 +412,12 @@ AliEveTRDTracklet::AliEveTRDTracklet(AliTRDseedV1 *trklt):TEveLine()
   //SetPoint(1, x0, y0, z0);
 }
 
+
+AliEveTRDTracklet::~AliEveTRDTracklet() 
+{
+
+}
+
 //______________________________________________________________________________
 void AliEveTRDTracklet::Print(Option_t *o) const
 {
@@ -418,8 +436,8 @@ AliEveTRDTrack::AliEveTRDTrack(AliTRDtrackV1 *trk)
   ,fTrackState(0)
   ,fESDStatus(0)
   ,fAlpha(0.)
-  ,fPoints(0x0)
-  ,fRim(0x0)
+  ,fPoints(NULL)
+  ,fRim(NULL)
 {
   // Constructor.
   SetUserData(trk);
@@ -427,33 +445,34 @@ AliEveTRDTrack::AliEveTRDTrack(AliTRDtrackV1 *trk)
 
   AliTRDtrackerV1::SetNTimeBins(24);
 
-
   fRim = new AliRieman(trk->GetNumberOfClusters());
-  AliTRDseedV1 *tracklet = 0x0;
+  AliTRDseedV1 *tracklet = NULL;
   for(Int_t il=0; il<AliTRDgeometry::kNlayer; il++){
     if(!(tracklet = trk->GetTracklet(il))) continue;
     if(!tracklet->IsOK()) continue;
     AddElement(new AliEveTRDTracklet(tracklet));
 
-    AliTRDcluster *c = 0x0;
-    tracklet->ResetClusterIter(kFALSE);
-    while((c = tracklet->PrevCluster())){
+    AliTRDcluster *c = NULL;
+//     tracklet->ResetClusterIter(kFALSE);
+//     while((c = tracklet->PrevCluster())){
+    for(Int_t ic=AliTRDseedV1::kNtb; ic--;){
+      if(!(c=tracklet->GetClusters(ic))) continue;
       Float_t xc = c->GetX();
       Float_t yc = c->GetY();
       Float_t zc = c->GetZ();
       Float_t zt = tracklet->GetZref(0) - (tracklet->GetX0()-xc)*tracklet->GetZref(1); 
       yc -= tracklet->GetTilt()*(zc-zt);
-      fRim->AddPoint(xc, yc, zc, 1, 10);
+      fRim->AddPoint(xc, yc, zc, .05, 2.3);
     }
   }
-  fRim->Update();
+  if(trk->GetNumberOfTracklets()>1) fRim->Update();
   SetStatus(fTrackState);
 }
 
 //______________________________________________________________________________
 AliEveTRDTrack::~AliEveTRDTrack()
 {
-  if(fPoints) delete [] fPoints; fPoints = 0x0;
+  if(fPoints) delete [] fPoints; fPoints = NULL;
   //delete dynamic_cast<AliTRDtrackV1*>(GetUserData());
 }
 
@@ -472,7 +491,11 @@ void AliEveTRDTrack::SetStatus(UChar_t s)
   if(fPoints && fTrackState == s) return;
 
   const Int_t nc = AliTRDtrackV1::kMAXCLUSTERSPERTRACK;
-  AliTRDtrackV1 *trk = (AliTRDtrackV1*)GetUserData();
+  AliTRDtrackV1 *trk(NULL);
+  if(!(trk=static_cast<AliTRDtrackV1*>(GetUserData()))) {
+    AliError("Failed casting data to TRD track.");
+    return;
+  }
 
   Bool_t BUILD = kFALSE;
   if(!fPoints){ 
@@ -481,11 +504,11 @@ void AliEveTRDTrack::SetStatus(UChar_t s)
     // define the radial span of the track in the TRD
     Double_t xmin = -1., xmax = -1.;
     Int_t det = 0;
-    AliTRDseedV1 *trklt = 0x0;
-    for(Int_t ily=AliTRDgeometry::kNlayer; ily--;){
+    AliTRDseedV1 *trklt = NULL;
+    for(Int_t ily=0; ily<AliTRDgeometry::kNlayer; ily++){
       if(!(trklt = trk->GetTracklet(ily))) continue;
       if(xmin<0.) xmin = trklt->GetX0() - AliTRDgeometry::CamHght() - AliTRDgeometry::CdrHght();
-      if(xmax<0.) xmax = trklt->GetX0();
+      if(trklt->GetX0()>xmax) xmax = trklt->GetX0();
       det = trklt->GetDetector();
     }
     Int_t sec = det/AliTRDgeometry::kNdets;
@@ -503,14 +526,14 @@ void AliEveTRDTrack::SetStatus(UChar_t s)
   if(BUILD || ((s&12) != (fTrackState&12))){
     if(TESTBIT(s, kTrackCosmics)){
       //printf("Straight track\n");
-      AliTRDtrackerV1::FitLine(trk, 0x0, kFALSE, nc, fPoints);
+      AliTRDtrackerV1::FitLine(trk, NULL, kFALSE, nc, fPoints);
     } else {
       if(TESTBIT(s, kTrackModel)){
         //printf("Kalman track\n");
-        if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitKalman(trk, 0x0, kFALSE, nc, fPoints);
+        if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitKalman(trk, NULL, kFALSE, nc, fPoints);
       } else { 
         //printf("Rieman track\n");
-        // if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitRiemanTilt(trk, 0x0, kTRUE, nc, fPoints);
+        if(trk->GetNumberOfTracklets() >=4) AliTRDtrackerV1::FitRiemanTilt(trk, NULL, kTRUE, nc, fPoints);
         Float_t x = 0.;
         for(Int_t ip = nc; ip--;){
           x = fPoints[ip].GetX();
@@ -546,7 +569,7 @@ void AliEveTRDTrack::SetStatus(UChar_t s)
         //printf("PID color kNNPID\n");
         //trk->GetReconstructor()->SetOption("nn");
       }
-      trk->CookPID();
+      //trk->CookPID();
   
       Int_t species = 0; Float_t pid = 0.;
       for(Int_t is=0; is<AliPID::kSPECIES; is++) 
@@ -596,3 +619,184 @@ void AliEveTRDTrack::SetStatus(UChar_t s)
   // save track status
   fTrackState = s;
 }
+
+
+AliEveTRDTrackletOnline::AliEveTRDTrackletOnline(AliTRDtrackletMCM *tracklet) :
+  TEveLine(),
+  fDetector(-1),
+  fROB(-1),
+  fMCM(-1)
+{
+  AliTRDtrackletMCM *trkl = new AliTRDtrackletMCM(*tracklet);
+  SetUserData(trkl);
+
+  fDetector = trkl->GetDetector();
+  fROB = trkl->GetROB();
+  fMCM = trkl->GetMCM();
+
+  SetName("sim. tracklet");
+  SetTitle(Form("Det: %i, ROB: %i, MCM: %i, Label: %i\n0x%08x", 
+                trkl->GetDetector(), trkl->GetROB(), trkl->GetMCM(), trkl->GetLabel(),
+                trkl->GetTrackletWord()));
+  SetLineColor(kGreen);
+  SetLineWidth(3);
+
+  AliTRDgeometry geo;
+  TGeoHMatrix *matrix = geo.GetClusterMatrix(trkl->GetDetector());
+
+  fDetector = trkl->GetDetector();
+  fROB = trkl->GetROB();
+  fMCM = trkl->GetMCM();
+  
+  Float_t length = 3.;
+  Double_t x[3];
+  Double_t p[3];
+  Double_t p2[3];
+  x[0] = AliTRDgeometry::AnodePos(); 
+  x[1] = trkl->GetY();
+  x[2] = trkl->GetLocalZ();
+
+  matrix->LocalToMaster(x, p);
+  geo.RotateBack(trkl->GetDetector(), p, p2);
+  SetPoint(0, p2[0], p2[1], p2[2]);
+
+  x[0] -= length;
+  x[1] -= length * trkl->GetdYdX();
+  matrix->LocalToMaster(x, p);
+  p[2] *= p[0] / (p[0] + length);
+  geo.RotateBack(trkl->GetDetector(), p, p2);
+  SetPoint(1, p2[0], p2[1], p2[2]);
+}
+
+AliEveTRDTrackletOnline::AliEveTRDTrackletOnline(AliTRDtrackletWord *tracklet) :
+  TEveLine(),
+  fDetector(-1),
+  fROB(-1),
+  fMCM(-1)
+{
+  AliTRDtrackletWord *trkl = new AliTRDtrackletWord(*tracklet);
+  SetUserData(trkl);
+
+  fDetector = trkl->GetDetector();
+  fROB = trkl->GetROB(); 
+  fMCM = trkl->GetMCM(); 
+
+  SetName("raw tracklet");
+  SetTitle(Form("Det: %i, ROB: %i, MCM: %i, Label: %i\n0x%08x", 
+                trkl->GetDetector(), fROB, fMCM, -1,
+                trkl->GetTrackletWord()));
+  SetLineColor(kRed);
+  SetLineWidth(3);
+
+  AliTRDgeometry geo;
+  TGeoHMatrix *matrix = geo.GetClusterMatrix(trkl->GetDetector());
+
+  Float_t length = 3.;
+  Double_t x[3];
+  Double_t p[3];
+  Double_t p2[3];
+  x[0] = AliTRDgeometry::AnodePos();
+  x[1] = trkl->GetY();
+  x[2] = trkl->GetLocalZ();
+  
+  matrix->LocalToMaster(x, p);
+  geo.RotateBack(trkl->GetDetector(), p, p2);
+  SetPoint(0, p2[0], p2[1], p2[2]);
+
+  x[0] -= length;
+  x[1] -= length * trkl->GetdYdX();
+  matrix->LocalToMaster(x, p);
+  p[2] *= p[0] / (p[0] + length);
+  geo.RotateBack(trkl->GetDetector(), p, p2);
+  SetPoint(1, p2[0], p2[1], p2[2]);
+}
+
+AliEveTRDTrackletOnline::~AliEveTRDTrackletOnline() 
+{
+  delete ((AliTRDtrackletBase*) GetUserData());
+  SetUserData(0x0);
+}
+
+void AliEveTRDTrackletOnline::ShowMCM(Option_t *opt) const
+{
+  if (fDetector < 0 || fROB < 0 || fMCM < 0)
+    return;
+
+  AliEveTRDmcm *evemcm = new AliEveTRDmcm();
+  evemcm->Init(fDetector, fROB, fMCM);
+  evemcm->LoadDigits();
+  evemcm->Draw(opt);
+
+  TEveElementList *mcmlist = NULL;
+  if (gEve->GetCurrentEvent()) 
+    mcmlist = (TEveElementList*) gEve->GetCurrentEvent()->FindChild("TRD MCMs");
+  if (!mcmlist) {
+    mcmlist = new TEveElementList("TRD MCMs");
+    gEve->AddElement(mcmlist);
+  }
+  gEve->AddElement(evemcm, mcmlist);
+}
+
+
+AliEveTRDmcm::AliEveTRDmcm() :
+  TEveElement(),
+  TNamed(),
+  fMCM(new AliTRDmcmSim())
+{
+  SetName("MCM");
+  SetTitle("Unknown MCM");
+}
+
+AliEveTRDmcm::~AliEveTRDmcm()
+{
+  delete fMCM;
+}
+
+Bool_t AliEveTRDmcm::Init(Int_t det, Int_t rob, Int_t mcm)
+{
+  SetName(Form("MCM: Det. %i, ROB %i, MCM %i", det, rob, mcm));
+  SetTitle(Form("MCM: Det. %i, ROB %i, MCM %i", det, rob, mcm));
+  fMCM->Init(det, rob, mcm);
+  fMCM->Reset();
+  return kTRUE;
+}
+
+Bool_t AliEveTRDmcm::LoadDigits()
+{
+  AliRunLoader *rl = AliEveEventManager::AssertRunLoader();
+  return fMCM->LoadMCM(rl, fMCM->GetDetector(), 
+                       fMCM->GetRobPos(), fMCM->GetMcmPos());
+}
+
+Bool_t AliEveTRDmcm::Filter()
+{
+  fMCM->Filter();
+  return kTRUE;
+}
+
+Bool_t AliEveTRDmcm::Tracklet()
+{
+  fMCM->Tracklet();
+  return kTRUE;
+}
+
+void AliEveTRDmcm::Draw(Option_t* option)
+{
+  const char *mcmname = Form("mcm_%i_%i_%i", fMCM->GetDetector(),
+                             fMCM->GetRobPos(), fMCM->GetMcmPos());
+
+  TCanvas *c = dynamic_cast<TCanvas*> (gROOT->FindObject(mcmname));
+  if (!c)
+    c = gEve->AddCanvasTab("TRD MCM");
+  c->SetTitle(Form("MCM %i on ROB %i of det. %i", 
+                   fMCM->GetMcmPos(), fMCM->GetRobPos(), fMCM->GetDetector()));
+  c->SetName(mcmname);
+  c->cd();
+  fMCM->Draw(option);
+}
+
+Bool_t AliEveTRDmcm::AssignPointer(const char* ptrname)
+{
+  gROOT->ProcessLine(Form("AliTRDmcmSim* %s = (AliTRDmcmSim *) 0x%x", ptrname, fMCM));
+  return kTRUE;
+}