2 // Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
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 **************************************************************************/
10 #include "AliEveTRDModuleImp.h"
11 #include "AliEveTRDData.h"
12 #include "EveBase/AliEveEventManager.h"
15 #include "TGListTree.h"
16 #include "TClonesArray.h"
17 #include "TGeoManager.h"
18 #include "TGeoMatrix.h"
20 #include "TEveManager.h"
21 #include "TEveTrack.h"
22 #include "TEveGeoNode.h"
23 #include "TEveTrans.h"
27 #include "AliCDBManager.h"
28 #include "AliTRDgeometry.h"
29 #include "AliTRDCommonParam.h"
30 #include "AliTRDpadPlane.h"
31 #include "AliTRDhit.h"
32 #include "AliTRDcluster.h"
33 #include "AliTRDtrackingChamber.h"
34 #include "AliTRDtrackletMCM.h"
36 ClassImp(AliEveTRDChamber)
37 ClassImp(AliEveTRDNode)
39 //______________________________________________________________________________
40 AliEveTRDNode::AliEveTRDNode(const char *typ, Int_t det) :
41 TEveElement(), AliEveTRDModule(typ, det)
46 //______________________________________________________________________________
47 void AliEveTRDNode::Paint(Option_t* option)
51 AliEveTRDModule *module(NULL);
52 List_i iter = fChildren.begin();
53 while(iter != fChildren.end()){
54 if((module = dynamic_cast<AliEveTRDModule*>(*iter))) module->Paint(option);
59 //______________________________________________________________________________
60 void AliEveTRDNode::Reset()
64 List_i iter = fChildren.begin();
65 while(iter != fChildren.end()){
66 //(dynamic_cast<AliEveTRDModule*>(*iter))->Reset();
71 //______________________________________________________________________________
72 void AliEveTRDNode::Collapse()
76 TGListTree *list = gEve->GetListTree();
77 AliEveTRDNode *node = 0x0;
78 List_i iter = fChildren.begin();
79 while(iter != fChildren.end()){
80 if((node = dynamic_cast<AliEveTRDNode*>(*iter))) node->Collapse();
81 list->CloseItem(FindListTreeItem(list));
86 //______________________________________________________________________________
87 void AliEveTRDNode::Expand()
91 TGListTree *list = gEve->GetListTree();
92 AliEveTRDNode *node = 0x0;
93 List_i iter = fChildren.begin();
94 while(iter != fChildren.end()){
95 if((node = dynamic_cast<AliEveTRDNode*>(*iter))) node->Expand();
96 list->OpenItem(FindListTreeItem(list));
101 //______________________________________________________________________________
102 void AliEveTRDNode::EnableListElements(Bool_t rnr_self, Bool_t rnr_children)
104 // Enable list elements.
106 SetRnrSelf(rnr_self);
107 AliEveTRDNode *node(NULL);
108 AliEveTRDChamber *chmb(NULL);
109 List_i iter = fChildren.begin();
110 while(iter != fChildren.end()){
111 if((node = dynamic_cast<AliEveTRDNode*>(*iter))){
112 node->SetRnrSelf(rnr_children);
113 node->EnableListElements(rnr_children, rnr_children);
115 if((chmb = dynamic_cast<AliEveTRDChamber*>(*iter))) chmb->EnableListElements(rnr_children, rnr_children);
121 //______________________________________________________________________________
122 void AliEveTRDNode::DisableListElements(Bool_t rnr_self, Bool_t rnr_children)
124 // Disable list elements.
126 SetRnrSelf(rnr_self);
127 AliEveTRDNode *node(NULL);
128 AliEveTRDChamber *chmb(NULL);
129 List_i iter = fChildren.begin();
130 while(iter != fChildren.end()){
131 if((node = dynamic_cast<AliEveTRDNode*>(*iter))){
132 node->SetRnrSelf(rnr_children);
133 node->DisableListElements(rnr_children, rnr_children);
135 if((chmb = dynamic_cast<AliEveTRDChamber*>(*iter))) chmb->DisableListElements(rnr_children, rnr_children);
141 //______________________________________________________________________________
142 void AliEveTRDNode::UpdateLeaves()
146 AliEveTRDModule *module;
147 List_i iter = fChildren.begin();
148 while(iter != fChildren.end()){
149 module = dynamic_cast<AliEveTRDModule*>(*iter);
150 if(!module) continue;
152 module->fRnrHits = fRnrHits;
153 module->fRnrDigits = fRnrDigits;
154 module->fDigitsLog = fDigitsLog;
155 module->fDigitsBox = fDigitsBox;
156 module->fDigitsThreshold = fDigitsThreshold;
157 module->fDigitsNeedRecompute = fDigitsNeedRecompute;
158 module->fRnrRecPoints = fRnrRecPoints;
159 module->fRnrTracklets = fRnrTracklets;
163 AliEveTRDNode *node = 0x0;
164 iter = fChildren.begin();
165 while(iter != fChildren.end()){
166 if((node = dynamic_cast<AliEveTRDNode*>(*iter))) node->UpdateLeaves();
172 //______________________________________________________________________________
173 void AliEveTRDNode::UpdateNode()
177 // Info("UpdateNode()", Form("%s", GetName()));
178 AliEveTRDNode *node = 0x0;
179 List_i iter = fChildren.begin();
180 while(iter != fChildren.end()){
181 if((node = dynamic_cast<AliEveTRDNode*>(*iter))) node->UpdateNode();
186 for(int i=0; i<11; i++) score[i] = 0;
187 AliEveTRDModule *module;
188 iter = fChildren.begin();
189 while(iter != fChildren.end()){
190 module = dynamic_cast<AliEveTRDModule*>(*iter);
191 if(!module) continue;
192 score[0] += (module->fLoadHits) ? 1 : 0;
193 score[1] += (module->fRnrHits) ? 1 : 0;
195 score[2] += (module->fLoadDigits) ? 1 : 0;
196 score[3] += (module->fRnrDigits) ? 1 : 0;
197 score[4] += (module->fDigitsLog) ? 1 : 0;
198 score[5] += (module->fDigitsBox) ? 1 : 0;
199 score[6] += (module->fDigitsNeedRecompute) ? 1 : 0;
201 score[7] += (module->fLoadRecPoints) ? 1 : 0;
202 score[8] += (module->fRnrRecPoints) ? 1 : 0;
204 score[9] += (module->fLoadTracklets) ? 1 : 0;
205 score[10] += (module->fRnrTracklets) ? 1 : 0;
209 Int_t size = fChildren.size();
210 fLoadHits = (score[0] > 0) ? kTRUE : kFALSE;
211 fRnrHits = (score[1] == size) ? kTRUE : kFALSE;
213 fLoadDigits = (score[2] > 0) ? kTRUE : kFALSE;
214 fRnrDigits = (score[3] == size) ? kTRUE : kFALSE;
215 fDigitsLog = (score[4] == size) ? kTRUE : kFALSE;
216 fDigitsBox = (score[5] == size) ? kTRUE : kFALSE;
217 fDigitsNeedRecompute = (score[6] == size) ? kTRUE : kFALSE;
219 fLoadRecPoints = (score[7] > 0) ? kTRUE : kFALSE;
220 fRnrRecPoints = (score[8] == size) ? kTRUE : kFALSE;
222 fLoadTracklets = (score[9] > 0) ? kTRUE : kFALSE;
223 fRnrTracklets = (score[10] == size) ? kTRUE : kFALSE;
227 ///////////////////////////////////////////////////////////
228 //////////// AliEveTRDChamber ////////////////////
229 ///////////////////////////////////////////////////////////
231 //______________________________________________________________________________
232 AliEveTRDChamber::AliEveTRDChamber(Int_t det) :
234 ,AliEveTRDModule("Chmb", det)
251 //______________________________________________________________________________
252 void AliEveTRDChamber::LoadClusters(TObjArray *clusters)
259 AliError(Form("Geometry not set for chamber %d. Please call first AliEveTRDChamber::SetGeometry().", fDet));
264 AddElement(fRecPoints = new AliEveTRDClusters());
265 fRecPoints->SetTitle(Form("Clusters for Det %d", GetID()));
270 Float_t g[3]; //global coordinates
271 AliTRDcluster *c=0x0;
272 Int_t nc = clusters->GetEntriesFast();
273 for(int iclus=0; iclus<nc; iclus++){
274 c = (AliTRDcluster*)clusters->UncheckedAt(iclus);
277 Int_t id = fRecPoints->SetNextPoint(g[0], g[1], g[2]);
278 fRecPoints->SetPointId(id, new AliTRDcluster(*c));
280 fRecPoints->StampObjProps();
281 fLoadRecPoints = kTRUE;
285 //______________________________________________________________________________
286 void AliEveTRDChamber::LoadClusters(AliTRDtrackingChamber *tc)
289 AliError(Form("Geometry not set for chamber %d. Please call first AliEveTRDChamber::SetGeometry().", fDet));
294 AddElement(fRecPoints = new AliEveTRDClusters());
295 fRecPoints->SetTitle(Form("Clusters for Det %d", GetID()));
299 Float_t g[3]; //global coordinates
300 const AliTRDchamberTimeBin *tb = 0x0;
301 for(int itb=0; itb<AliTRDseedV1::kNtb; itb++){
303 if(!(Int_t(*tb))) continue;
304 const AliTRDcluster *c= 0x0; Int_t ic = 0;
305 while((c=tb->GetCluster(ic))){
307 Int_t id = fRecPoints->SetNextPoint(g[0], g[1], g[2]);
308 fRecPoints->SetPointId(id, new AliTRDcluster(*c));
312 fRecPoints->StampObjProps();
313 fLoadRecPoints = kTRUE;
317 //______________________________________________________________________________
318 void AliEveTRDChamber::LoadDigits(AliTRDdigitsManager *digits)
324 AliError(Form("Geometry not set for chamber %d. Please call first AliEveTRDChamber::SetGeometry().", fDet));
328 if(!fDigits) AddElement(fDigits = new AliEveTRDDigits(this));
331 fDigits->SetData(digits);
332 fDigits->StampObjProps();
333 fDigitsNeedRecompute = kTRUE;
337 //______________________________________________________________________________
338 void AliEveTRDChamber::LoadHits(TClonesArray *hits, Int_t &idx)
345 AddElement(fHits = new AliEveTRDHits());
346 fHits->SetTitle(Form("Hits for Det %d", GetID()));
349 Int_t nhits = hits->GetEntriesFast();
351 AliTRDhit *hit = 0x0;
353 hit = (AliTRDhit*)hits->UncheckedAt(idx);
354 if(hit->GetDetector() != fDet) return;
356 Int_t id = fHits->SetNextPoint(hit->X(), hit->Y(), hit->Z());
357 fHits->SetPointId(id, new AliTRDhit(*hit));
358 fHits->StampObjProps();
364 //______________________________________________________________________________
365 void AliEveTRDChamber::LoadTracklets(TTree *trklTree)
372 Error("LoadTracklets()", Form("Geometry not set for chamber %d. Please call first AliEveTRDChamber::SetGeometry().", fDet));
377 fTracklets = new TClonesArray("AliEveTRDTrackletOnline",100);
379 fTracklets->Delete();
380 TEveElementList *trklChild = (TEveElementList*) FindChild("Tracklets");
382 trklChild->Destroy();
386 TBranch *mcmBranch = trklTree->GetBranch("mcmtrklbranch");
390 AliTRDtrackletMCM *trkl = 0x0;
391 mcmBranch->SetAddress(&trkl);
393 TEveElementList* listOfTracklets = new TEveElementList("Tracklets");
394 gEve->AddElement(listOfTracklets, this);
396 for(Int_t iTrkl = 0; iTrkl < mcmBranch->GetEntries(); iTrkl++){
397 mcmBranch->GetEntry(iTrkl);
398 if (trkl->GetDetector() == GetID()) {
399 new ((*fTracklets)[fTracklets->GetEntriesFast()]) AliEveTRDTrackletOnline(trkl);
400 gEve->AddElement(new AliEveTRDTrackletOnline(trkl), listOfTracklets);
403 fLoadTracklets = kTRUE;
407 //______________________________________________________________________________
408 void AliEveTRDChamber::SetGeometry(AliTRDgeometry *geo)
414 Int_t ism = geo->GetSector(fDet);
415 Int_t istk = geo->GetStack(fDet);
416 Int_t ilyr = geo->GetLayer(fDet);
417 Int_t icha = istk*6+ilyr;
419 if(ism>12&&ism<16) idx=3;
420 else if(ism==11||ism==12) idx=2;
422 // define pad plane size in pads
423 AliTRDpadPlane *pp = fGeo->GetPadPlane(ilyr, istk);
424 fNrows = pp->GetNrows();
425 fNcols = pp->GetNcols();
427 // this version for setting the rendarable object is not working very nice
428 // Int_t shape_offset = TEveGeoShape::Class()->GetDataMemberOffset("fShape");
429 // TEveGeoShape* eg_shape = new TEveGeoShape("geometry");
430 // eg_shape->RefMainTrans().SetFrom(* gGeoManager->GetCurrentMatrix());
431 // * (TGeoShape**) (((char*)eg_shape) + shape_offset) = gGeoManager->GetCurrentVolume()->GetShape();
433 // eg_shape->StampColorSelection();
435 AliEveEventManager::AssertGeometry();
437 AliError("Geo manager not available.");
442 // define rendarable volumes
443 if(!gGeoManager->cd(Form("/B077_1/BSEGMO%d_1/BTRD%d_1/UTR%d_1/UTS%d_1/UTI%d_1/UT%02d_1", ism, ism, idx, idx, idx, icha))) return;
445 fShape = new TEveGeoTopNode(gGeoManager, gGeoManager->GetCurrentNode());
446 fShape->RefMainTrans().SetFrom(*gGeoManager->GetCurrentMatrix());
447 fShape->DisableListElements();
448 fShape->SetRnrSelf(kFALSE);
449 // try to set the properties but it is crashing !!
450 // TEveGeoNode *node = 0x0;
451 // if((node = (TEveGeoNode*)fShape->FindChild(Form("UA%02d_1", icha)))) node->SetRnrState(kTRUE);
452 // else AliWarning(Form("Can not retrieve geo node UA%02d_1", icha));