2 // Visualization of an EMCAL super module.
4 // Author: Magali Estienne (magali.estienne@cern.ch)
7 #include "AliEveEMCALSModule.h"
11 #include <TEveFrameBox.h>
12 #include <TEveQuadSet.h>
13 #include <TEvePointSet.h>
14 #include <TEveRGBAPalette.h>
26 class TVirtualViewer3D;
27 class AliEveEMCALData;
30 class AliEveEMCALSModuleData;
32 ClassImp(AliEveEMCALSModule)
34 Bool_t AliEveEMCALSModule::fgStaticInit = kFALSE;
35 Float_t AliEveEMCALSModule::fgSMBigBBox[3];
36 Float_t AliEveEMCALSModule::fgSMSmallBBox[3];
37 TEveFrameBox* AliEveEMCALSModule::fgFrameBigBox = 0;
38 TEveFrameBox* AliEveEMCALSModule::fgFrameSmallBox = 0;
39 TEveRGBAPalette* AliEveEMCALSModule::fgFrameDigPalette = 0;
40 TEveRGBAPalette* AliEveEMCALSModule::fgFrameCluPalette = 0;
42 AliEveEMCALSModule::AliEveEMCALSModule(Int_t smid, const Text_t* n, const Text_t* t) :
43 TEveElement(fFrameColor),
48 fFrameColor((Color_t)10),
50 fQuadSet(new TEveQuadSet(n,t)),
51 fQuadSet2(new TEveQuadSet(n,t)),
52 fPointSet(new TEvePointSet(n)),
60 snprintf(name,256,"Full Super Module %02d",smid);
62 snprintf(name,256,"Half Super Module %02d",smid);
67 fPointSet->IncDenyDestroy();
68 AddElement(fPointSet);
70 fQuadSet->IncDenyDestroy();
73 fQuadSet2->IncDenyDestroy();
74 AddElement(fQuadSet2);
78 AliEveEMCALSModule::AliEveEMCALSModule(const AliEveEMCALSModule &esm) :
79 TEveElement(fFrameColor),
82 fEMCALData(esm.fEMCALData),
83 fEMCALSModuleData(esm.fEMCALSModuleData),
84 fFrameColor(esm.fFrameColor),
85 fSModuleID(esm.fSModuleID),
86 fQuadSet(esm.fQuadSet),
87 fQuadSet2(esm.fQuadSet2),
88 fPointSet(esm.fPointSet),
89 fClusterSize(esm.fClusterSize),
90 fHitSize(esm.fHitSize),
95 if (fSModuleID < 10) {
96 sprintf(name,"Full Super Module %02d",fSModuleID);
98 sprintf(name,"Half Super Module %02d",fSModuleID);
103 fPointSet->IncDenyDestroy();
104 AddElement(fPointSet);
106 fQuadSet->IncDenyDestroy();
107 AddElement(fQuadSet);
109 fQuadSet2->IncDenyDestroy();
110 AddElement(fQuadSet2);
114 AliEveEMCALSModule::~AliEveEMCALSModule()
120 fPointSet->DecDenyDestroy();
121 fQuadSet->DecDenyDestroy();
122 fQuadSet2->DecDenyDestroy();
124 if(fEMCALData) fEMCALData->DecRefCount();
127 //______________________________________________________________________________
128 void AliEveEMCALSModule::DropData() const
131 // release the sm data
142 //______________________________________________________________________________
143 void AliEveEMCALSModule::InitStatics(AliEveEMCALSModuleData* md)
146 // Bounding box, Framebox and Palette
149 if (fgStaticInit) return;
150 fgStaticInit = kTRUE;
152 md->GetSModuleBigBox(fgSMBigBBox[0],fgSMBigBBox[1], fgSMBigBBox[2]);
153 md->GetSModuleSmallBox(fgSMSmallBBox[0],fgSMSmallBBox[1], fgSMSmallBBox[2]);
155 fgFrameBigBox = new TEveFrameBox();
156 fgFrameBigBox->SetAABoxCenterHalfSize(0, 0, 0, fgSMBigBBox[0], fgSMBigBBox[1], fgSMBigBBox[2]);
157 fgFrameBigBox->SetFrameColor((Color_t)10);
158 fgFrameBigBox->IncRefCount();
159 fgFrameDigPalette = new TEveRGBAPalette(0,512);
160 fgFrameDigPalette->SetLimits(0, 1024);
161 fgFrameDigPalette->IncRefCount();
163 fgFrameSmallBox = new TEveFrameBox();
164 fgFrameSmallBox->SetAABoxCenterHalfSize(0, 0, 0, fgSMSmallBBox[0], fgSMSmallBBox[1], fgSMSmallBBox[2]);
165 fgFrameSmallBox->SetFrameColor((Color_t)10);
166 fgFrameSmallBox->IncRefCount();
167 fgFrameCluPalette = new TEveRGBAPalette(0,512);
168 fgFrameCluPalette->SetLimits(0, 1024);
169 fgFrameCluPalette->IncRefCount();
172 //______________________________________________________________________________
173 void AliEveEMCALSModule::SetClusterSize(Int_t size)
176 // Cluster point size
179 fClusterSize = TMath::Max(1, size);
182 //______________________________________________________________________________
183 void AliEveEMCALSModule::SetHitSize(Int_t size)
189 fHitSize = TMath::Max(1, size);
192 //______________________________________________________________________________
193 void AliEveEMCALSModule::SetDataSource(AliEveEMCALData* const data)
196 // Set source of data.
199 if (data == fEMCALData) return;
200 if(fEMCALData) fEMCALData->DecRefCount();
202 if(fEMCALData) fEMCALData->IncRefCount();
204 // Get pointer on SM data
205 fEMCALSModuleData = GetSModuleData();
208 //______________________________________________________________________________
209 AliEveEMCALSModuleData* AliEveEMCALSModule::GetSModuleData() const
212 // Return source of data.
215 return fEMCALData ? fEMCALData->GetSModuleData(fSModuleID) : 0;
218 //______________________________________________________________________________
219 void AliEveEMCALSModule::UpdateQuads()
222 // Update hit/digit/cluster representation.
225 vector< vector<Double_t> > bufferDigit;
226 vector< vector<Double_t> > bufferCluster;
227 vector< vector<Float_t> > bufferHit;
228 Int_t nDigits, nClusters, nHits, oldSize;
229 Float_t hitX, hitY, hitZ;
230 Int_t smId = fEMCALSModuleData->GetSmId();
232 //--------------------------
233 // Hits from runloader
234 //--------------------------
238 TEvePointSet* points = fEMCALData->GetPointSetData();
241 sprintf(form,"N=%d", points->Size());
242 points->SetTitle(form);
243 points->SetMarkerSize(.5);
244 points->SetMarkerColor((Color_t)2);
245 fPointSet->AddElement(points);
247 else {printf("There is no hits in Runloader \n"); }
251 if (fEMCALSModuleData != 0) {
254 InitStatics(fEMCALSModuleData);
256 // digits ------------------------
258 // Define TEveQuadSet for digits
259 fQuadSet->SetOwnIds(kTRUE);
260 fQuadSet->Reset(TEveQuadSet::kQT_RectangleYZFixedDimX, kFALSE, 32);
261 fQuadSet->SetDefWidth (fEMCALSModuleData->GetPhiTileSize());
262 fQuadSet->SetDefHeight(fEMCALSModuleData->GetEtaTileSize());
263 fQuadSet->RefMainTrans().SetFrom(*fEMCALSModuleData->GetSModuleMatrix());
264 fQuadSet->SetPalette(fgFrameDigPalette);
265 if(smId<fEMCALSModuleData->GetNsmf())
266 fQuadSet->SetFrame(fgFrameBigBox);
267 else fQuadSet->SetFrame(fgFrameSmallBox);
269 // Get the digit information from the buffer
270 bufferDigit = fEMCALSModuleData->GetDigitBuffer();
271 if(!bufferDigit.empty())
273 nDigits = fEMCALSModuleData->GetNDigits();
274 if(fDebug>1) cout << "nDigits: " << nDigits << endl;
276 for (Int_t id = 0; id < nDigits; id++) {
277 // Int_t iid = (Int_t)bufferDigit[id][0];
278 // Int_t isupMod = (Int_t)bufferDigit[id][1];
279 Double_t iamp = bufferDigit[id][2];
280 Int_t amp = (Int_t)(iamp+0.5);
281 // Double_t ix = bufferDigit[id][3];
282 Double_t iy = bufferDigit[id][4];
283 Double_t iz = bufferDigit[id][5];
285 // Add digit information to the TEveQuadSet
286 fQuadSet->AddQuad(iy, iz);
287 fQuadSet->QuadValue(amp);
290 else { if (fDebug) printf("There is no digits in SM %d \n", smId); }
292 // hits --------------------------
293 bufferHit = fEMCALSModuleData->GetHitBuffer();
294 if(!bufferHit.empty())
297 nHits = fEMCALSModuleData->GetNHits();
298 if(fDebug>1) cout << "nHits: " << nHits << endl;
299 oldSize = fPointSet->GrowFor(nHits);
301 for (Int_t ih = 0; ih < nHits; ih++) {
302 hitX = bufferHit[ih][3];
303 hitY = bufferHit[ih][4];
304 hitZ = bufferHit[ih][5];
305 fPointSet->SetPoint(ih,hitX,hitY,hitZ);
306 sprintf(form,"N=%d", fPointSet->Size());
307 fPointSet->SetTitle(form);
308 fPointSet->SetMarkerSize(.5);
309 fPointSet->SetMarkerColor((Color_t)2);
312 else { if (fDebug) printf("There is no hits in SM %d \n", smId); }
314 // clusters ------------------------
316 // Define TEveQuadSet for clusters
317 fQuadSet2->SetOwnIds(kTRUE);
318 fQuadSet2->Reset(TEveQuadSet::kQT_RectangleYZFixedDimX, kFALSE, 32);
319 fQuadSet2->SetDefWidth (fEMCALSModuleData->GetPhiTileSize());
320 fQuadSet2->SetDefHeight(fEMCALSModuleData->GetEtaTileSize());
321 fQuadSet2->RefMainTrans().SetFrom(*fEMCALSModuleData->GetSModuleMatrix());
322 fQuadSet2->SetPalette(fgFrameCluPalette);
323 if(smId<fEMCALSModuleData->GetNsmf())
324 fQuadSet2->SetFrame(fgFrameBigBox);
325 else fQuadSet2->SetFrame(fgFrameSmallBox);
327 // Get the cluster information from the buffer
328 bufferCluster = fEMCALSModuleData->GetClusterBuffer();
329 if(!bufferCluster.empty())
331 nClusters = fEMCALSModuleData->GetNClusters();
332 if(fDebug>1) cout << "nClusters: " << nClusters << endl;
333 // loop over clusters
334 for (Int_t id = 0; id < nClusters; id++) {
336 cout << "bufferCluster[" << id << "][0]: " << bufferCluster[id][0] << endl;
337 cout << "bufferCluster[" << id << "][1]: " << bufferCluster[id][1] << endl;
338 cout << "bufferCluster[" << id << "][2]: " << bufferCluster[id][2] << endl;
339 cout << "bufferCluster[" << id << "][3]: " << bufferCluster[id][3] << endl;
340 cout << "bufferCluster[" << id << "][4]: " << bufferCluster[id][4] << endl;
342 // Int_t isupMod = (Int_t)bufferCluster[id][0];
343 Double_t iamp = bufferCluster[id][1];
344 Int_t amp = (Int_t)(iamp+0.5);
345 // Double_t ix = bufferCluster[id][2];
346 Double_t iy = bufferCluster[id][3];
347 Double_t iz = bufferCluster[id][4];
349 // Add cluster information to the TEveQuadSet
350 fQuadSet2->AddQuad(iy, iz);
351 fQuadSet2->QuadValue(amp);
352 // fQuadSet2->QuadId(iid);
354 } // end clusters loop
356 else { if (fDebug) printf("There is no clusters in SM %d \n", smId); }
358 } // end if (fEMCALSModuleData != 0)
362 //______________________________________________________________________________
363 void AliEveEMCALSModule::SetSModuleID(Int_t id)
366 // Set id of the SM to display.
370 if (id > 12) id = 12;