EVE/EveDet/AliEveTRDLoader.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 "AliEveTRDLoader.h"
  11 #include "AliEveTRDModuleImp.h"
  12 #include "EveBase/AliEveEventManager.h"
  13
  14 #include <TEveManager.h>
  15 #include <TEveGValuators.h>
  16
  17 #include "TGeoManager.h"
  18 #include "TSystem.h"
  19 #include "TFile.h"
  20 #include "TTree.h"
  21 #include "TObjString.h"
  22 #include "TObjArray.h"
  23 #include "TClonesArray.h"
  24
  25 #include <TGLabel.h>
  26 #include <TGButton.h>
  27 #include <TGTextEntry.h>
  28 #include <TGFileDialog.h>
  29
  30 #include "AliLog.h"
  31 #include "AliCDBManager.h"
  32
  33 #include "AliTRDhit.h"
  34 #include "AliTRDcluster.h"
  35 #include "AliTRDdigitsManager.h"
  36 #include "AliTRDgeometry.h"
  37
  38 ClassImp(AliEveTRDLoader)
  39 ClassImp(AliEveTRDLoaderEditor)
  40
  41 ///////////////////////////////////////////////////////////
  42 /////////////     AliEveTRDLoader     /////////////////////
  43 ///////////////////////////////////////////////////////////
  44
  45
  46 //______________________________________________________________________________
  47 AliEveTRDLoader::AliEveTRDLoader(const Text_t* n, const Text_t* t) : TEveElementList(n, t)
  48   ,fDataType(0)
  49   ,fSM(-1)
  50   ,fStack(-1)
  51   ,fLy(-1)
  52   ,fEvent(-1)
  53   ,fGeo(0x0)
  54   ,fFilename("")
  55   ,fDir(".")
  56 {
  57   // Constructor.
  58
  59   AliEveEventManager::AssertGeometry();
  60
  61   fGeo = new AliTRDgeometry();
  62   //fGeo->CreateClusterMatrixArray();
  63 }
  64
  65 //______________________________________________________________________________
  66 void AliEveTRDLoader::AddChambers(int sm, int stk, int ly)
  67 {
  68   // Add specified chambers.
  69
  70   fSM=sm; fStack=stk; fLy=ly;
  71   Int_t ismStart = (sm == -1) ?  0 : sm;
  72   Int_t ismStop  = (sm == -1) ? 18 : sm+1;
  73   Int_t istkStart= (stk == -1)?  0 : stk;
  74   Int_t istkStop = (stk == -1)?  5 : stk+1;
  75   Int_t ilyStart = (ly == -1) ?  0 : ly;
  76   Int_t ilyStop  = (ly == -1) ?  6 : ly+1;
  77
  78   List_i ichmb;
  79   ichmb = fChildren.begin();
  80   while(ichmb != fChildren.end()){
  81     (*ichmb)->SetRnrSelf(kFALSE);
  82     ichmb++;
  83   }
  84
  85   AliEveTRDNode *lSM=0x0, *lSTK=0x0;
  86   AliEveTRDChamber *lCHMB = 0x0;
  87   int det;
  88   for (int ism=ismStart; ism<ismStop; ism++){
  89     if(!(lSM = (AliEveTRDNode*)FindChild(Form("SM%03d", ism)))){
  90       AddElement(lSM = new AliEveTRDNode("SM", ism));
  91       lSM->SetElementTitle(Form("Supermodule %2d", ism));
  92     }
  93     lSM->SetRnrSelf(kTRUE);
  94
  95     for (int istk=istkStart; istk<istkStop; istk++) {
  96       if(!(lSTK = (AliEveTRDNode*)lSM->FindChild("Stack%03d"))){
  97         lSM->AddElement(lSTK = new AliEveTRDNode("Stack", istk));
  98         lSTK->SetElementTitle(Form("SM %2d Stack %1d", ism, istk));
  99       }
 100       lSTK->SetRnrSelf(kTRUE);
 101
 102       for (int ily=ilyStart; ily<ilyStop; ily++) {
 103         det = fGeo->GetDetector(ily, istk, ism);
 104         if(!(lCHMB = (AliEveTRDChamber*)lSTK->FindChild(Form("Chmb%03d", det)))){
 105           lSTK->AddElement(lCHMB = new AliEveTRDChamber(det));
 106           lCHMB->SetGeometry(fGeo);
 107           lCHMB->SetElementTitle(Form("SM %2d Stack %1d Layer %1d", ism, istk, ily));
 108         }
 109         lCHMB->SetRnrSelf(kTRUE);
 110       }
 111     }
 112   }
 113   gEve->Redraw3D();
 114 }
 115
 116 //______________________________________________________________________________
 117 AliEveTRDChamber* AliEveTRDLoader::GetChamber(int d)
 118 {
 119   // Get given chamber.
 120
 121   Int_t ism  = fGeo->GetSector(d),
 122         istk = fGeo->GetStack(d);
 123
 124   AliEveTRDNode *node = 0x0;
 125   if(!(node = (AliEveTRDNode*)FindChild(Form("SM%03d", ism)))) return 0x0;
 126   if(!(node = (AliEveTRDNode*)node->FindChild(Form("Stack%03d", istk)))) return 0x0;
 127   return (AliEveTRDChamber*)node->FindChild(Form("Chmb%03d", d));
 128 }
 129
 130 //______________________________________________________________________________
 131 Bool_t AliEveTRDLoader::GoToEvent(int ev)
 132 {
 133   // Go to given event.
 134
 135   if(!fChildren.size()){
 136     AliWarning("Please select first the chamber that you want to monitor from \"Chamber(s) selector\".");
 137     return kFALSE;
 138   }
 139
 140   fEvent = ev;
 141
 142   Unload();
 143
 144   Int_t ndt(0);
 145   const Char_t *tn[] = {"TreeH", "TreeD", "TreeR", "tracklets-raw"};
 146   const Char_t *fn[] = {"Hits", "Digits", "RecPoints", "Tracklets"};
 147   TTree *t(NULL); TFile *f(NULL);
 148   for(Int_t idt(0); idt<4; idt++){
 149     if(idt==0 && !(fDataType&kTRDHits)) continue;
 150     else if(idt==1 && !(fDataType&kTRDDigits)) continue;
 151     else if(idt==2 && !(fDataType&kTRDClusters)) continue;
 152     else if(idt==3 && !(fDataType&kTRDTracklets)) continue;
 153
 154     if(!(f = TFile::Open(Form("%s/TRD.%s.root", fDir.Data(), fn[idt])))){
 155       AliWarning(Form("File not found \"%s/TRD.%s.root\".", fDir.Data(), fn[idt]));
 156       continue;
 157     }
 158     if(!f->cd(Form("Event%d", ev))){
 159       AliError(Form("Event[%d] not found in file \"%s/TRD.%s.root\".", ev, fDir.Data(), fn[idt]));
 160       f->Close(); //delete f;
 161       continue;
 162     }
 163
 164     if(!(t = (TTree*)gDirectory->Get(tn[idt]))) AliError(Form("Tree[%s] not found for Event[%d].", tn[idt], ev));
 165     else{
 166       switch(idt){
 167       case 0:
 168         if(LoadHits(t)) ndt++;
 169         break;
 170       case 1:
 171         if(LoadDigits(t)) ndt++;
 172         break;
 173       case 2:
 174         if(LoadClusters(t)) ndt++;
 175         break;
 176       case 3:
 177         if(LoadTracklets(t)) ndt++;
 178         break;
 179       }
 180     }
 181     f->Close(); //delete f;
 182   }
 183   gEve->Redraw3D();
 184
 185   return Bool_t(ndt);
 186 }
 187
 188
 189 //______________________________________________________________________________
 190 Bool_t AliEveTRDLoader::LoadHits(TTree *tH)
 191 {
 192   // Load hits.
 193
 194   AliInfo("Loading ...");
 195   if(!fChildren.size()) return kFALSE;
 196
 197   AliEveTRDChamber *chmb(NULL);
 198   TClonesArray *hits = new TClonesArray("AliTRDhit", 100);
 199   tH->SetBranchAddress("TRD", &hits);
 200   Int_t idx, nhits;
 201   for(int iTrack=0; iTrack<tH->GetEntries(); iTrack++){
 202     if(!tH->GetEvent(iTrack)) continue;
 203     if(!(nhits = hits->GetEntriesFast())) continue;
 204
 205     idx = 0;
 206     while(idx < nhits){
 207       Int_t det = ((AliTRDhit*)hits->UncheckedAt(idx))->GetDetector();
 208       chmb = GetChamber(det);
 209       if(chmb) chmb->LoadHits(hits, idx);
 210       else{
 211         AliTRDhit *hit(NULL);
 212         while(idx < nhits){
 213           hit = (AliTRDhit*)hits->UncheckedAt(idx);
 214           if(hit->GetDetector() != det) break;
 215           idx++;
 216         }
 217       }
 218     }
 219     hits->Delete();
 220   }
 221   return kTRUE;
 222 }
 223
 224
 225 //______________________________________________________________________________
 226 Bool_t AliEveTRDLoader::LoadClusters(TTree *tC)
 227 {
 228   // Load clusters.
 229
 230   AliInfo("Loading ...");
 231   if(!fChildren.size()) return kFALSE;
 232
 233   TObjArray *clusters(NULL);
 234   tC->SetBranchAddress("TRDcluster", &clusters);
 235
 236   AliEveTRDChamber *chmb(NULL);
 237   AliTRDcluster *c(NULL);
 238   for(int idet=0; idet<AliTRDgeometry::kNdet; idet++){
 239     tC->GetEntry(idet);
 240     if(!clusters->GetEntriesFast()) continue;
 241     if(!(c = (AliTRDcluster*)clusters->UncheckedAt(0))) continue;
 242     if((chmb = GetChamber(c->GetDetector()))) chmb->LoadClusters(clusters);
 243   }
 244   return kTRUE;
 245 }
 246
 247
 248 //______________________________________________________________________________
 249 Bool_t AliEveTRDLoader::LoadDigits(TTree *tD)
 250 {
 251   // Load digits.
 252
 253   AliInfo("Loading ...");
 254
 255   if(!fChildren.size()) return kFALSE;
 256
 257   AliEveTRDChamber *chmb(NULL);
 258   AliTRDdigitsManager dm;
 259   dm.ReadDigits(tD);
 260   for(int idet=0; idet<AliTRDgeometry::kNdet; idet++){
 261     if(!(chmb=GetChamber(idet))) continue;
 262     chmb->LoadDigits(&dm);
 263   }
 264   return kTRUE;
 265 }
 266
 267
 268 //______________________________________________________________________________
 269 Bool_t AliEveTRDLoader::LoadTracklets(TTree *trklTree)
 270 {
 271   // Load tracklets.
 272
 273   AliInfo("Loading ...");
 274   if(!fChildren.size()) return kFALSE;
 275
 276
 277   AliEveTRDChamber *chmb(NULL);
 278
 279   for(int idet=0; idet<AliTRDgeometry::kNdet; idet++){
 280     if((chmb = GetChamber(idet)))
 281       chmb->LoadTracklets(trklTree);
 282   }
 283
 284   return kTRUE;
 285 }
 286
 287
 288 //______________________________________________________________________________
 289 Bool_t AliEveTRDLoader::Open(const char *filename, const char *dir)
 290 {
 291   // Open given file in given directory.
 292
 293   fFilename = filename;
 294   fDir = dir;
 295   TObjArray *so = fFilename.Tokenize(".");
 296
 297   if(((TObjString*)(*so)[0])->GetString().CompareTo("TRD") != 0){
 298     AliError(Form("Filename %s do not fulfill AliRoot naming conventions.", filename));
 299     return kFALSE;
 300   }
 301   if(gSystem->AccessPathName(Form("%s/%s", dir, filename))){
 302     AliError(Form("Missing file %s/%s", dir, filename));
 303     return kFALSE;
 304   }
 305   if(((TObjString*)(*so)[1])->GetString().CompareTo("Hits") == 0){
 306     if(!(fDataType&kTRDHits)){
 307       AliInfo("Data type set to HITS according to file name.");
 308       fDataType|=kTRDHits;
 309     }
 310   } else   if(((TObjString*)(*so)[1])->GetString().CompareTo("Digits") == 0){
 311     if(!(fDataType&kTRDDigits)){
 312       AliInfo("Data type set to DIGITS according to file name.");
 313       fDataType|=kTRDDigits;
 314     }
 315   } else if(((TObjString*)(*so)[1])->GetString().CompareTo("RecPoints") == 0){
 316     if(!(fDataType&kTRDClusters)){
 317       AliInfo("Data type set to CLUSTERS according to file name.");
 318       fDataType|=kTRDClusters;
 319     }
 320   } else if(((TObjString*)(*so)[1])->GetString().CompareTo("Tracklets") == 0){
 321     if(!(fDataType&kTRDTracklets)){
 322       AliInfo("Data type set to TRACKLETS according to file name.");
 323       fDataType|=kTRDTracklets;
 324     }
 325   } else {
 326     AliError("Filename didn't fulfill naming conventions. No data will be loaded.");
 327     return kFALSE;
 328   }
 329
 330   SetDataLinked();
 331   return kTRUE;
 332 }
 333
 334 //______________________________________________________________________________
 335 void AliEveTRDLoader::Paint(Option_t *option)
 336 {
 337   // Paint object.
 338
 339   AliEveTRDModule *module(NULL);
 340   List_i ichmb = fChildren.begin();
 341   while(ichmb != fChildren.end()){
 342     if((module = dynamic_cast<AliEveTRDModule*>(*ichmb))) module->Paint(option);
 343     ichmb++;
 344   }
 345 }
 346
 347
 348 //______________________________________________________________________________
 349 void AliEveTRDLoader::Unload()
 350 {
 351   // Unload module data.
 352
 353   List_i ichmb = fChildren.begin();
 354   while(ichmb != fChildren.end()){
 355     //(dynamic_cast<AliEveTRDModule*>(*ichmb))->Reset();
 356     ichmb++;
 357   }
 358 }
 359
 360 ///////////////////////////////////////////////////////////
 361 /////////////   AliEveTRDLoaderEditor       /////////////////////
 362 ///////////////////////////////////////////////////////////
 363
 364 //______________________________________________________________________________
 365 AliEveTRDLoaderEditor::AliEveTRDLoaderEditor(const TGWindow* p, Int_t width, Int_t height,
 366                                              UInt_t options, Pixel_t back) :
 367   TGedFrame(p, width, height, options | kVerticalFrame, back),
 368   fM(0), fFile(0), fBrowse(0x0), fEvent(0),
 369   fSMNumber(0), fStackNumber(0), fPlaneNumber(0)
 370 {
 371   // Constructor.
 372
 373   MakeTitle("AliEveTRDLoader");
 374
 375   // file browser frame
 376   Int_t labelW = 42;
 377   TGHorizontalFrame* f = new TGHorizontalFrame(this);
 378   TGHorizontalFrame* g = new TGHorizontalFrame(f, labelW, 0, kFixedWidth);
 379   TGLabel* l = new TGLabel(g, "File: ");
 380   g->AddFrame(l, new TGLayoutHints(kLHintsLeft, 0,0,4,0));
 381   f->AddFrame(g);
 382   fFile = new TGTextEntry(f);
 383   fFile->SetToolTipText("Select TRD data file or galice.root");
 384   fFile->SetWidth(140);
 385   fFile->Connect("DoubleClicked()", "AliEveTRDLoaderEditor", this, "FileOpen()");
 386   f->AddFrame(fFile);
 387
 388   fBrowse = new TGTextButton(f, "Browse");
 389   f->AddFrame(fBrowse);
 390   fBrowse->Connect("Clicked()", "AliEveTRDLoaderEditor", this, "FileOpen()");
 391   AddFrame(f, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,5,5,5,5));
 392
 393
 394   // "Chamber(s) selector" group frame
 395   TGGroupFrame *fGroupFrame1974 = new TGGroupFrame(this,"Chamber(s) selector");
 396   TGVerticalFrame *fVerticalFrame1974 = new TGVerticalFrame(fGroupFrame1974, 150, 50,kVerticalFrame);
 397
 398   fSMNumber = new TEveGValuator(fVerticalFrame1974, "SM:", 0, 0);
 399   fSMNumber->SetShowSlider(kFALSE);
 400   fSMNumber->SetLabelWidth(labelW);
 401   fSMNumber->SetNELength(6);
 402   fSMNumber->Build();
 403   fSMNumber->SetLimits(-1, 17);
 404   fSMNumber->SetToolTip("Supermodule id [-1 for all]");
 405   fVerticalFrame1974->AddFrame(fSMNumber, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterX | kLHintsExpandY,2,2,2,2));
 406
 407   fStackNumber = new TEveGValuator(fVerticalFrame1974, "Stack:", 0, 0);
 408   fStackNumber->SetShowSlider(kFALSE);
 409   fStackNumber->SetLabelWidth(labelW);
 410   fStackNumber->SetNELength(6);
 411   fStackNumber->Build();
 412   fStackNumber->SetLimits(-1, 4);
 413   fStackNumber->SetToolTip("Stack id [-1 for all in this SM]");
 414   fVerticalFrame1974->AddFrame(fStackNumber, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterX | kLHintsExpandY,2,2,2,2));
 415
 416   fPlaneNumber = new TEveGValuator(fVerticalFrame1974, "Plane:", 0, 0);
 417   fPlaneNumber->SetShowSlider(kFALSE);
 418   fPlaneNumber->SetLabelWidth(labelW);
 419   fPlaneNumber->SetNELength(6);
 420   fPlaneNumber->Build();
 421   fPlaneNumber->SetLimits(-1, 5);
 422   fPlaneNumber->SetToolTip("Plane id [-1 for all in this stack]");
 423
 424   fVerticalFrame1974->AddFrame(fPlaneNumber, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterX | kLHintsExpandY,2,2,2,2));
 425
 426   fGroupFrame1974->AddFrame(fVerticalFrame1974, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsExpandY | kLHintsCenterX,2,2,2,2));
 427
 428   TGTextButton *fTextButton2037 = new TGTextButton(fGroupFrame1974,"Select");
 429   fTextButton2037->SetTextJustify(36);
 430   fGroupFrame1974->AddFrame(fTextButton2037, new TGLayoutHints(kLHintsExpandY | kLHintsCenterX,2,2,2,2));
 431   fTextButton2037->SetToolTipText("Apply selection", 400);
 432   fTextButton2037->Connect("Clicked()",
 433                            "AliEveTRDLoaderEditor", this, "AddChambers()");
 434
 435   fGroupFrame1974->SetLayoutManager(new TGHorizontalLayout(fGroupFrame1974));
 436   AddFrame(fGroupFrame1974, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,5,5,5,5));
 437
 438
 439   // Event steering frame
 440   f = new TGHorizontalFrame(this);
 441   TGTextButton *fGoTo = new TGTextButton(f, "GoTo");
 442   fGoTo->SetTextJustify(36);
 443   fGoTo->Resize(164,22);
 444   f->AddFrame(fGoTo, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
 445   fGoTo->SetToolTipText("Skip to event", 400);
 446   fGoTo->Connect("Clicked()", "AliEveTRDLoaderEditor", this, "GoTo()");
 447
 448   fEvent = new TEveGValuator(f, "Event:", 110, 0);
 449   fEvent->SetShowSlider(kFALSE);
 450   fEvent->SetLabelWidth(labelW);
 451   fEvent->SetNELength(6);
 452   fEvent->Build();
 453   fEvent->SetLimits(-1, 1000);
 454   fEvent->SetToolTip("Set event number to be monitored");
 455   fEvent->Connect("ValueSet(Double_t)",
 456                   "AliEveTRDLoaderEditor", this, "SetEvent(Double_t)");
 457   f->AddFrame(fEvent, new TGLayoutHints(kLHintsLeft | kLHintsCenterY, 2,2,2,2));
 458
 459   TGTextButton *fNext = new TGTextButton(f, "Next");
 460   fNext->SetTextJustify(36);
 461   fNext->Resize(164,22);
 462   f->AddFrame(fNext, new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,2,2,2,2));
 463   fNext->SetToolTipText("Next event", 400);
 464   fNext->Connect("Clicked()", "AliEveTRDLoaderEditor", this, "Next()");
 465
 466   AddFrame(f,new TGLayoutHints(kLHintsLeft | kLHintsTop | kLHintsCenterY | kLHintsExpandX,5,5,5,5));
 467
 468 }
 469
 470 //______________________________________________________________________________
 471 void AliEveTRDLoaderEditor::SetModel(TObject* obj)
 472 {
 473   // Set model object.
 474
 475   if(!(fM = dynamic_cast<AliEveTRDLoader*>(obj))) return;
 476
 477   fFile->SetEnabled(!fM->IsDataLinked());
 478   fFile->SetText(gSystem->BaseName(fM->fFilename.Data()));
 479   fBrowse->SetEnabled(!fM->IsDataLinked());
 480
 481   fEvent->SetEnabled(fM->IsDataLinked());
 482   fEvent->GetEntry()->SetIntNumber(fM->fEvent);
 483
 484   fSMNumber->SetEnabled(fM->IsDataLinked());
 485   fSMNumber->GetEntry()->SetIntNumber(fM->fSM);
 486
 487
 488   fStackNumber->SetEnabled(fM->IsDataLinked());
 489   fStackNumber->GetEntry()->SetIntNumber(fM->fStack);
 490
 491
 492   fPlaneNumber->SetEnabled(fM->IsDataLinked());
 493   fPlaneNumber->GetEntry()->SetIntNumber(fM->fLy);
 494 }
 495
 496 //______________________________________________________________________________
 497 void AliEveTRDLoaderEditor::AddChambers()
 498 {
 499   // Slot to add chambers.
 500
 501   fM->fSM    = (int)fSMNumber->GetEntry()->GetNumber();
 502   fM->fStack = (int)fStackNumber->GetEntry()->GetNumber();
 503   fM->fLy    = (int)fPlaneNumber->GetEntry()->GetNumber();
 504   fM->AddChambers(fM->fSM, fM->fStack, fM->fLy);
 505 }
 506
 507 //______________________________________________________________________________
 508 void AliEveTRDLoaderEditor::FileOpen()
 509 {
 510   // Slot for opening of file.
 511
 512   TGFileInfo fi;
 513   fi.fIniDir    = StrDup(gSystem->DirName (fM->fFilename.Data()));
 514   fi.fFilename  = StrDup(gSystem->BaseName(fM->fFilename.Data()));
 515   //  fi.fFileTypes = tpcfiletypes;
 516
 517   new TGFileDialog(fClient->GetRoot(), gEve->GetMainWindow(), kFDOpen, &fi);
 518   if (!fi.fFilename) return;
 519
 520   if(fM->Open(gSystem->BaseName(fi.fFilename), gSystem->DirName (fi.fFilename))){
 521     fFile->SetToolTipText(gSystem->DirName (fi.fFilename));
 522     fFile->SetText       (gSystem->BaseName(fi.fFilename));
 523   } else fFile->Clear();
 524
 525   this->SetModel(fM);
 526 }
 527
 528 void AliEveTRDLoaderEditor::GoTo()
 529 {
 530   // Slot for loading of event.
 531
 532   fM->GoToEvent(fM->fEvent);
 533 }
 534
 535 void AliEveTRDLoaderEditor::Next()
 536 {
 537   // Slot for loading of event.
 538
 539   fM->NextEvent();
 540 }