]> git.uio.no Git - u/mrichter/AliRoot.git/blame - EVE/EveDet/AliEveTOFSector.cxx
Changes in order to modify also ITSonly tracks
[u/mrichter/AliRoot.git] / EVE / EveDet / AliEveTOFSector.cxx
CommitLineData
d810d0de 1/**************************************************************************
2 * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. *
3 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for *
51346b82 4 * full copyright notice. *
d810d0de 5 **************************************************************************/
fd54c9e3 6
7// $Id$
8// Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007
9
10//
11// Class to visualize the TOF digit information
12// in TOF sector frame
13//
14// Author: A. De Caro (email: decaro@sa.infn.t)
15//
16
17#include <TGeoManager.h>
18#include <TClonesArray.h>
19#include <TTree.h>
20
21#include <TEveTrans.h>
8616d353 22
8616d353 23#include <AliTOFdigit.h>
28a21457 24#include <AliTOFGeometry.h>
8616d353 25
fd54c9e3 26#include "AliEveTOFSector.h"
d810d0de 27
fd31e9de 28Bool_t AliEveTOFSector::fgStaticInitDone = kFALSE;
d810d0de 29TEveFrameBox* AliEveTOFSector::fgTOFsectorFrameBox = 0;
30TEveRGBAPalette* AliEveTOFSector::fgTOFsectorPalette = 0;
8616d353 31
32//_______________________________________________________
d810d0de 33ClassImp(AliEveTOFSector)
8616d353 34
35/* ************************************************************************ */
36
d810d0de 37AliEveTOFSector::AliEveTOFSector(const Text_t* n, const Text_t* t) :
84aff7a4 38 TEveQuadSet(n, t),
28a21457 39 fTOFgeometry(new AliTOFGeometry()),
8616d353 40 fTOFarray(0x0),
41 fTOFtree(0x0),
42 fSector(-1),
43 fDx(0), fDy(0), fDz(0),
44 fAutoTrans (kTRUE),
ee9d8eca 45 //fMinTime (0), fMaxTime (0),
46 fThreshold (5), fMaxVal (80),
8616d353 47 fSectorID (0),
ee9d8eca 48 //fPlateFlag0(kTRUE), fPlateFlag1(kTRUE), fPlateFlag2(kTRUE), fPlateFlag3(kTRUE), fPlateFlag4(kTRUE),
49 fPlateFlag (0x0),
50 //fFrameColor(4),
51 //fRnrFrame (kFALSE),
8616d353 52 fGeoManager(0)
53{
54
fd54c9e3 55 // default ctr
56
8616d353 57 fPlateFlag = new Bool_t[5];
58 for (Int_t ii=0; ii<5; ii++) fPlateFlag[ii]=kTRUE;
59
ee9d8eca 60
81515106 61 //fGeoManager = AliEveEventManager::AssertGeometry();
8616d353 62 if (!fGeoManager) {
63 printf("ERROR: no TGeo\n");
64 }
fd31e9de 65
ee9d8eca 66}
8616d353 67/* ************************************************************************ */
68
ee9d8eca 69AliEveTOFSector::AliEveTOFSector(TGeoManager *localGeoManager,
70 Int_t nSector)
71 :
72 TEveQuadSet(Form("Sector%i",nSector)),
28a21457 73 fTOFgeometry(new AliTOFGeometry()),
8616d353 74 fTOFarray(0x0),
75 fTOFtree(0x0),
76 fSector(nSector),
77 fDx(0), fDy(0), fDz(0),
78 fAutoTrans (kTRUE),
ee9d8eca 79 //fMinTime (0), fMaxTime (0),
fd31e9de 80 fThreshold (5), fMaxVal (80),
8616d353 81 fSectorID (nSector),
ee9d8eca 82 //fPlateFlag0(kTRUE), fPlateFlag1(kTRUE), fPlateFlag2(kTRUE), fPlateFlag3(kTRUE), fPlateFlag4(kTRUE),
83 fPlateFlag (0x0),
84 //fFrameColor(4),
85 //fRnrFrame (kFALSE),
8616d353 86 fGeoManager(localGeoManager)
87{
88
fd54c9e3 89 // ctr
90
8616d353 91 fPlateFlag = new Bool_t[5];
92 for (Int_t ii=0; ii<5; ii++) fPlateFlag[ii]=kTRUE;
93
94 /*
95 if (!fGeoManager) {
96 printf("ERROR: no TGeo\n");
97 }
98 */
99
100 InitModule();
fd31e9de 101
ee9d8eca 102}
8616d353 103/* ************************************************************************ */
104
d810d0de 105AliEveTOFSector::AliEveTOFSector(TGeoManager *localGeoManager,
8616d353 106 Int_t nSector,
107 TClonesArray *tofArray)
108 :
84aff7a4 109 TEveQuadSet(Form("Sector%i",nSector)),
28a21457 110 fTOFgeometry(new AliTOFGeometry()),
8616d353 111 fTOFarray(tofArray),
112 fTOFtree(0x0),
113 fSector(nSector),
114 fDx(0), fDy(0), fDz(0),
115 fAutoTrans (kTRUE),
ee9d8eca 116 //fMinTime (0), fMaxTime (0),
117 fThreshold (5), fMaxVal (80),
8616d353 118 fSectorID (nSector),
ee9d8eca 119 //fPlateFlag0(kTRUE), fPlateFlag1(kTRUE), fPlateFlag2(kTRUE), fPlateFlag3(kTRUE), fPlateFlag4(kTRUE),
120 fPlateFlag (0x0),
121 //fFrameColor(4),
122 //fRnrFrame (kFALSE),
8616d353 123 fGeoManager(localGeoManager)
124{
125
fd54c9e3 126 // ctr
127
8616d353 128 fPlateFlag = new Bool_t[5];
129 for (Int_t ii=0; ii<5; ii++) fPlateFlag[ii]=kTRUE;
130
131 InitModule();
132
133}
134/* ************************************************************************ */
135
ee9d8eca 136AliEveTOFSector::AliEveTOFSector(TGeoManager *localGeoManager,
137 Int_t nSector,
138 TTree *tofTree)
139 :
140 TEveQuadSet(Form("Sector%i",nSector)),
28a21457 141 fTOFgeometry(new AliTOFGeometry()),
8616d353 142 fTOFarray(0x0),
143 fTOFtree(tofTree),
144 fSector(nSector),
145 fDx(0), fDy(0), fDz(0),
146 fAutoTrans (kTRUE),
ee9d8eca 147 //fMinTime (0), fMaxTime (0),
148 fThreshold (5), fMaxVal (80),
8616d353 149 fSectorID (nSector),
ee9d8eca 150 //fPlateFlag0(kTRUE), fPlateFlag1(kTRUE), fPlateFlag2(kTRUE), fPlateFlag3(kTRUE), fPlateFlag4(kTRUE),
151 fPlateFlag (0x0),
152 //fFrameColor(4),
153 //fRnrFrame (kFALSE),
8616d353 154 fGeoManager(localGeoManager)
155{
156
fd54c9e3 157 // ctr
158
8616d353 159 fPlateFlag = new Bool_t[5];
160 for (Int_t ii=0; ii<5; ii++) fPlateFlag[ii]=kTRUE;
161
162 InitModule();
163
164}
165/* ************************************************************************ */
166
d810d0de 167AliEveTOFSector::~AliEveTOFSector()
8616d353 168{
fd54c9e3 169
170 // dtr
171
8616d353 172 /*
173 fGeoManager = 0x0;
174 delete fGeoManager;
175
176 delete fTOFarray;
177 fTOFarray = 0x0;
178 */
179 delete fPlateFlag;
180
181}
182
183/* ************************************************************************ */
184/*
d810d0de 185void AliEveTOFSector::SetDigitsInfo(AliEveTOFDigitsInfo* info)
8616d353 186{
187 if(fInfo) fInfo->DecRefCount();
188 fInfo = info;
189 if(fInfo) fInfo->IncRefCount();
190
191}
192*/
193/* ************************************************************************ */
d810d0de 194void AliEveTOFSector::InitStatics()
8616d353 195{
fd54c9e3 196 //
197 // To initialize statistic variables
198 //
199
8616d353 200 if (fgStaticInitDone) return;
201
202 Float_t dx = 124.5;
203 Float_t dz = 29.;
204 Float_t dy = 370.6*2.;
84aff7a4 205 fgTOFsectorFrameBox = new TEveFrameBox();
8616d353 206 fgTOFsectorFrameBox->SetAABox(-dx*0.5, -dy*0.5, -dz*0.5, dx, dy, dz);
ab73f3eb 207 fgTOFsectorFrameBox->SetFrameColor(Color_t(32));
55ee7431 208 fgTOFsectorFrameBox->IncRefCount();
8616d353 209
84aff7a4 210 //fgTOFsectorPalette = new TEveRGBAPalette(0, 2048); // TOT
ee9d8eca 211 //fgTOFsectorPalette->SetLimits(0, 2048);
82dabae7 212 //fgTOFsectorPalette = new TEveRGBAPalette(0, 8192); // TDC
213 fgTOFsectorPalette = new TEveRGBAPalette(0, 100000); // TDC
214 fgTOFsectorPalette->SetOverflowAction(2);
215 fgTOFsectorPalette->SetUnderflowAction(2);
55ee7431 216 fgTOFsectorPalette->IncRefCount();
8616d353 217
218 fgStaticInitDone = kTRUE;
219}
220
221/* ************************************************************************ */
d810d0de 222void AliEveTOFSector::InitModule()
8616d353 223{
fd54c9e3 224 //
225 // To initialize TOF sector frame variables
226 //
8616d353 227
228 fDx = fTOFgeometry->XPad()*fTOFgeometry->NpadX();
229 //fDy = fTOFgeometry->XPad()*fTOFgeometry->NpadX();
230 fDz = fTOFgeometry->ZPad()*fTOFgeometry->NpadZ();
231
232 if (!fgStaticInitDone) InitStatics();
233
234 SetFrame(fgTOFsectorFrameBox);
235 SetPalette(fgTOFsectorPalette);
236 //fFrame = fgTOFsectorFrameBox;
237 //fPalette = fgTOFsectorPalette;
238
51346b82 239 LoadQuads();
8616d353 240 ComputeBBox();
241 SetTrans();
242
243}
244
245/* ************************************************************************ */
d810d0de 246void AliEveTOFSector::LoadQuads()
8616d353 247{
fd54c9e3 248 //
249 // Load TOF digit as illuminated pad
250 //
8616d353 251
8616d353 252 //Int_t n_col = gStyle->GetNumberOfColors();
253
254 Int_t vol[5] = {fSectorID, -1, -1, -1, -1};
255 Int_t informations[4] = {-1, -1, -1, -1};
256 Int_t dummy[3] = {-1, -1, -1};
257 Int_t tdc = -1;
258 Int_t tot = -1;
259
260 Double_t **coord = new Double_t*[4];
261 for (Int_t ii=0; ii<4; ii++) coord[ii] = new Double_t[3];
262
ee9d8eca 263 Reset(kQT_FreeQuad, kFALSE, 32);
264
8616d353 265 //printf(" fTOFarray->GetEntries() = %4i \n",fTOFarray->GetEntries());
266
267 if (fTOFtree && !fTOFarray) {
268 //printf("Hello world\n");
269 TClonesArray* digitsTOFnew = new TClonesArray("AliTOFdigit", 300);
270
271 fTOFarray = new TClonesArray("AliTOFdigit", 300);
272 TClonesArray &ldigits = *fTOFarray;
273 Int_t newCounter = 0;
274
275 AliTOFdigit *digs;
276
277 fTOFtree->SetBranchAddress("TOF",&digitsTOFnew);
278 fTOFtree->GetEntry(0);
279 for (Int_t digitNumber=0; digitNumber<digitsTOFnew->GetEntries(); digitNumber++) {
280
281 //if (digitNumber==digitsTOF->GetEntries()-1) printf(" Hello 4 -> %3i digit of %i \n", digitNumber+1, digitsTOF->GetEntries());
51346b82 282
8616d353 283 digs = (AliTOFdigit*)digitsTOFnew->UncheckedAt(digitNumber);
284
285 if (digs->GetSector()!=fSectorID) continue;
286
287 vol[1] = digs->GetPlate(); // Plate Number (0-4)
288 vol[2] = digs->GetStrip(); // Strip Number (0-14/18)
ee9d8eca 289 vol[3] = digs->GetPadx(); // Pad Number in x direction (0-47)
290 vol[4] = digs->GetPadz(); // Pad Number in z direction (0-1)
8616d353 291
292 informations[0] = digs->GetTdc();
293 informations[1] = digs->GetAdc();
294 informations[2] = digs->GetToT();
295 informations[3] = digs->GetTdcND();
296 new (ldigits[newCounter++]) AliTOFdigit(dummy, vol, informations);
297 }
298 }
299
300 AliTOFdigit *tofDigit;
301 //printf(" 0x%lx\n",fTOFarray);
302 for (Int_t ii=0; ii<fTOFarray->GetEntries(); ii++) {
303
304 tofDigit = (AliTOFdigit*)fTOFarray->UncheckedAt(ii);
51346b82 305
8616d353 306 if (fPlateFlag[tofDigit->GetPlate()]) {
307
308 vol[1] = tofDigit->GetPlate();
309 vol[2] = tofDigit->GetStrip();
310 vol[3] = tofDigit->GetPadz();
311 vol[4] = tofDigit->GetPadx();
312
313 tot = tofDigit->GetToT();
314 tdc = tofDigit->GetTdc();
315
316 //if (fSector==4 && fPlate==2 && fStrip==0) printf(" %2i %1i\n", iPadX, iPadZ);
317 //if (iPadX==23 || iPadX==24) printf(" %2i %1i %2i \n", fSector, fPlate, fStrip);
318 //if (vol[0]==4 && vol[1]==2 && vol[2]==7) {
319
320 for (Int_t kk=0; kk<4; kk++) for (Int_t jj=0; jj<3; jj++) coord[kk][jj]=0.;
321
322 fTOFgeometry->DetToSectorRF(vol, coord);
323 /*
324 printf("\n");
325 printf(" %1i %2i, %f %f %f \n", vol[3], vol[4], coord[0][0], coord[0][1], coord[0][2]);
326 printf(" %1i %2i, %f %f %f \n", vol[3], vol[4], coord[1][0], coord[1][1], coord[1][2]);
327 printf(" %1i %2i, %f %f %f \n", vol[3], vol[4], coord[2][0], coord[2][1], coord[2][2]);
328 printf(" %1i %2i, %f %f %f \n", vol[3], vol[4], coord[3][0], coord[3][1], coord[3][2]);
329 */
330 Float_t vertices[12]={(Float_t)coord[0][0], (Float_t)coord[0][1], (Float_t)coord[0][2],
331 (Float_t)coord[1][0], (Float_t)coord[1][1], (Float_t)coord[1][2],
332 (Float_t)coord[2][0], (Float_t)coord[2][1], (Float_t)coord[2][2],
333 (Float_t)coord[3][0], (Float_t)coord[3][1], (Float_t)coord[3][2]};
51346b82 334
8616d353 335 AddQuad(vertices);
336 //AddQuad((Float_t*)coord);
337 //AddQuad(coord[0], coord[1], coord[2], 2.5, 3.5);
338 //AddQuad(-2.5*0.5, -3.5*0.5, 2.5, 3.5);
339
340 // In principle could have color based on number of neigbours. We
341 // can insert the time-of-flight value for each pad
ee9d8eca 342
8616d353 343 //QuadValue((Int_t)tot);
344 QuadValue((Int_t)tdc);
345 QuadId(tofDigit);
51346b82 346
8616d353 347 //}
348 } // closed if control on plates switched on
349 } // closed loop on TOF sector digits
350
351 RefitPlex();
352
353 fTOFarray = 0x0;
51346b82 354
8616d353 355}
356
357/* ************************************************************ */
d810d0de 358void AliEveTOFSector::SetTrans()
8616d353 359{
fd54c9e3 360 //
361 // Set the translation matrix for TOF sector
362 //
363
a15e6d7d 364 InitMainTrans();
365 TEveTrans& t = RefMainTrans();
8616d353 366
367 //Int_t det[5] = {fSector, -1, -1, -1, -1};
368 Char_t path[100];
369
370 Int_t localSector = fSector;
371 if (!fAutoTrans) localSector = 4;
372
373 //fTOFgeometry->GetVolumePath(det,path);
374 fTOFgeometry->GetVolumePath(localSector,path);
375 fGeoManager->cd(path);
376 TGeoHMatrix global = *fGeoManager->GetCurrentMatrix();
377 Double_t *rotMat = global.GetRotationMatrix();
378 Double_t *tr = global.GetTranslation();
379
a15e6d7d 380 t.SetBaseVec(1, rotMat[0], rotMat[3], rotMat[6]);
381 t.SetBaseVec(2, rotMat[1], rotMat[4], rotMat[7]);
382 t.SetBaseVec(3, rotMat[2], rotMat[5], rotMat[8]);
383 t.SetBaseVec(4, tr);
8616d353 384}
385
386//-----------------------------------------------------
387
d810d0de 388void AliEveTOFSector::SetSectorID(Int_t id)
8616d353 389{
fd54c9e3 390 //
391 // Set TOF sector ID to be visualized
392 //
393
8616d353 394 fSectorID = id;
395 fSector = id;
396 if (fAutoTrans)
397 SetTrans(); // Force repositioning.
398
399 LoadQuads();
400}
401
402//-----------------------------------------------------
403
d810d0de 404void AliEveTOFSector::SetPlate(Int_t nPlate, Bool_t r)
8616d353 405{
fd54c9e3 406 //
407 // Set visualization flag to visualize TOF modules
408 //
8616d353 409
410 fPlateFlag[nPlate] = r;
411
412 //printf(" HELLO World ! %i %i %i \n", nPlate, r, fPlateFlag[nPlate]);
413}
414
57ffa5fb 415/******************************************************************************/
8616d353 416
d810d0de 417void AliEveTOFSector::SetThreshold(Short_t t)
8616d353 418{
fd54c9e3 419 //
420 // Set visualization threshold
421 //
422
8616d353 423 fThreshold = TMath::Min(t, (Short_t)(fMaxVal - 1));
424 // ClearColorArray();
425}
426
57ffa5fb 427/******************************************************************************/
8616d353 428
d810d0de 429void AliEveTOFSector::SetMaxVal(Int_t mv)
8616d353 430{
fd54c9e3 431 //
432 // Set visualization max value
433 //
434
8616d353 435 fMaxVal = TMath::Max(mv, (Int_t)(fThreshold + 1));
436 //ClearColorArray();
437}
438
57ffa5fb 439/******************************************************************************/
8616d353 440
d810d0de 441void AliEveTOFSector::DigitSelected(Int_t idx)
8616d353 442{
fd54c9e3 443 //
ee9d8eca 444 // Override control-click from TEveQuadSet
fd54c9e3 445 //
8616d353 446
d794819f 447 TObject* obj = GetId(idx);
8616d353 448 AliTOFdigit* digs = dynamic_cast<AliTOFdigit*>(obj);
d810d0de 449 // printf("AliEveTOFSector::QuadSelected "); Print();
8616d353 450 if (digs)
ee9d8eca 451 printf("\n Sector = %2i Plate = %1i Strip = %2i PadZ = %1i PadX = %2i ToT = %3i Tof = %5i\n",
452 fSector , digs->GetPlate(), digs->GetStrip(), digs->GetPadz(), digs->GetPadx(), digs->GetToT(), digs->GetTdc());
8616d353 453 else printf("\n");
454
455}
456
57ffa5fb 457/******************************************************************************/