]>
Commit | Line | Data |
---|---|---|
d810d0de | 1 | // $Id$ |
2 | // Main authors: Matevz Tadel & Alja Mrak-Tadel: 2006, 2007 | |
3b868b75 | 3 | |
d810d0de | 4 | /************************************************************************** |
5 | * Copyright(c) 1998-2008, ALICE Experiment at CERN, all rights reserved. * | |
6 | * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for * | |
51346b82 | 7 | * full copyright notice. * |
d810d0de | 8 | **************************************************************************/ |
9 | ||
10 | #include "AliEveITSScaledModule.h" | |
3b868b75 | 11 | |
0879c50b | 12 | #include <AliITSsegmentationSPD.h> |
13 | #include <AliITSsegmentationSDD.h> | |
14 | #include <AliITSsegmentationSSD.h> | |
15 | ||
3b868b75 | 16 | #include <AliITSdigitSPD.h> |
17 | #include <AliITSdigitSDD.h> | |
18 | #include <AliITSdigitSSD.h> | |
d810d0de | 19 | |
0879c50b | 20 | #include <TMath.h> |
a15e6d7d | 21 | #include <TClonesArray.h> |
3b868b75 | 22 | |
a15e6d7d | 23 | //============================================================================== |
24 | //============================================================================== | |
d810d0de | 25 | // AliEveDigitScaleInfo |
a15e6d7d | 26 | //============================================================================== |
27 | ||
28 | //______________________________________________________________________________ | |
3b868b75 | 29 | // |
a15e6d7d | 30 | // Encapsulates current state of scaling and agglomeration. |
698e2c9b | 31 | |
d810d0de | 32 | ClassImp(AliEveDigitScaleInfo) |
3b868b75 | 33 | |
d810d0de | 34 | AliEveDigitScaleInfo::AliEveDigitScaleInfo(): |
3b868b75 | 35 | fScale(1), |
a15e6d7d | 36 | fStatType (kSTAverage), |
32e219c2 | 37 | fSyncPalette(kFALSE) |
3b868b75 | 38 | { |
39 | } | |
40 | ||
d810d0de | 41 | void AliEveDigitScaleInfo::ScaleChanged(Int_t s) |
3b868b75 | 42 | { |
32e219c2 | 43 | fScale = s; |
51346b82 | 44 | |
d810d0de | 45 | AliEveITSScaledModule* sm; |
84aff7a4 | 46 | std::list<TEveElement*>::iterator i = fBackRefs.begin(); |
32e219c2 | 47 | while (i != fBackRefs.end()) |
48 | { | |
d810d0de | 49 | sm = dynamic_cast<AliEveITSScaledModule*>(*i); |
32e219c2 | 50 | if(sm) sm->LoadQuads(); |
51 | ++i; | |
52 | } | |
3b868b75 | 53 | } |
54 | ||
d810d0de | 55 | void AliEveDigitScaleInfo::StatTypeChanged(Int_t t) |
3b868b75 | 56 | { |
32e219c2 | 57 | fStatType = t; |
58 | fSyncPalette = kTRUE; | |
51346b82 | 59 | |
d810d0de | 60 | AliEveITSScaledModule* sm; |
84aff7a4 | 61 | std::list<TEveElement*>::iterator i = fBackRefs.begin(); |
32e219c2 | 62 | while (i != fBackRefs.end()) |
63 | { | |
d810d0de | 64 | sm = dynamic_cast<AliEveITSScaledModule*>(*i); |
a15e6d7d | 65 | if (sm) sm->SetQuadValues(); |
32e219c2 | 66 | ++i; |
67 | } | |
3b868b75 | 68 | } |
69 | ||
57ffa5fb | 70 | //______________________________________________________________________________ |
a15e6d7d | 71 | // ScaledDigit_t |
3b868b75 | 72 | // |
c76ea574 | 73 | |
a15e6d7d | 74 | AliEveITSScaledModule::ScaledDigit_t::ScaledDigit_t() : |
3b868b75 | 75 | TObject(), |
a15e6d7d | 76 | fN(0), |
77 | fSum(0), fSqrSum(0), | |
78 | fMinI(-1), fMinJ(-1), fMaxI(-1), fMaxJ(-1) | |
3b868b75 | 79 | { |
80 | } | |
81 | ||
a15e6d7d | 82 | AliEveITSScaledModule::ScaledDigit_t::ScaledDigit_t(Int_t di, Int_t dj) : |
3b868b75 | 83 | TObject(), |
a15e6d7d | 84 | fN(0), |
85 | fSum(0), fSqrSum(0), | |
86 | fMinI(di), fMinJ(dj), fMaxI(di), fMaxJ(dj) | |
3b868b75 | 87 | { |
88 | } | |
89 | ||
a15e6d7d | 90 | void AliEveITSScaledModule::ScaledDigit_t::Dump() const |
3b868b75 | 91 | { |
a15e6d7d | 92 | printf("N %d, sum %f, sqr_sum %f", fN, fSum, fSqrSum); |
3b868b75 | 93 | } |
94 | ||
698e2c9b | 95 | |
a15e6d7d | 96 | //============================================================================== |
97 | //============================================================================== | |
d810d0de | 98 | // AliEveITSScaledModule |
a15e6d7d | 99 | //============================================================================== |
698e2c9b | 100 | |
101 | //______________________________________________________________________________ | |
3b868b75 | 102 | // |
698e2c9b | 103 | // Visualization of an ITS module with digits aggregated |
104 | // on a grid of pre-defined size. | |
3b868b75 | 105 | |
d810d0de | 106 | ClassImp(AliEveITSScaledModule) |
3b868b75 | 107 | |
d810d0de | 108 | AliEveITSScaledModule::AliEveITSScaledModule(Int_t gid, AliEveITSDigitsInfo* info, AliEveDigitScaleInfo* si): |
109 | AliEveITSModule("AliEveITSScaledModule", "AliEveITSScaledModule"), | |
3b868b75 | 110 | fNx(-1), |
111 | fNz(-1), | |
112 | fNCx(-1), | |
03dfc4a7 | 113 | fNCz(-1), |
9af86b69 | 114 | fScaleInfo(si), |
115 | fDigitsMap() | |
3b868b75 | 116 | { |
3b868b75 | 117 | SetOwnIds(kTRUE); |
118 | ||
119 | SetDigitsInfo(info); | |
51346b82 | 120 | SetID(gid); |
32e219c2 | 121 | fScaleInfo->IncRefCount(this); |
3b868b75 | 122 | } |
123 | ||
d810d0de | 124 | AliEveITSScaledModule::~AliEveITSScaledModule() |
3b868b75 | 125 | { |
32e219c2 | 126 | fScaleInfo->DecRefCount(this); |
3b868b75 | 127 | } |
128 | ||
57ffa5fb | 129 | /******************************************************************************/ |
3b868b75 | 130 | |
d810d0de | 131 | void AliEveITSScaledModule::LoadQuads() |
3b868b75 | 132 | { |
133 | // Here we still use 'z' for the name of axial coordinates. | |
134 | // The transforamtion matrix aplied rotates y -> z. | |
84aff7a4 | 135 | // We need this as TEveQuadSet offers optimized treatment for |
3b868b75 | 136 | // quads in the x-y plane. |
137 | ||
a4135a45 | 138 | TClonesArray *digits = fInfo->GetDigits(fID, fDetID); |
139 | if (!digits) return; | |
140 | ||
141 | Int_t ndigits = digits->GetEntriesFast(); | |
142 | ||
3b868b75 | 143 | Float_t x, z, zo, dpx, dpz; // orig cells size, pos |
a4135a45 | 144 | Int_t i, j; // orig cells idx |
145 | Int_t c1, c2; // original coordinates | |
3b868b75 | 146 | |
147 | Int_t id; | |
fd31e9de | 148 | std::map<Int_t, Int_t> dmap; |
149 | std::map<Int_t, Int_t>::iterator miter; | |
3b868b75 | 150 | |
03dfc4a7 | 151 | Int_t scale = fScaleInfo->GetScale() -1; |
3b868b75 | 152 | switch(fDetID) |
153 | { | |
51346b82 | 154 | case 0: |
155 | { | |
3b868b75 | 156 | // SPD |
84aff7a4 | 157 | Reset(kQT_RectangleXZFixedY, kFALSE, 32); |
3b868b75 | 158 | |
159 | fNCz = fInfo->fSPDScaleZ[scale]; | |
160 | fNCx = fInfo->fSPDScaleX[scale]; | |
161 | fNz = Int_t(fInfo->fSegSPD->Npz()/fNCz); | |
162 | fNx = Int_t(fInfo->fSegSPD->Npx()/fNCx); | |
51346b82 | 163 | dpz = 2*fDz/fNz; |
3b868b75 | 164 | dpx = 2*fDx/fNx; |
165 | //printf("SPD orig cells (%d, %d) (%d, %d)\n", fInfo->fSegSPD->Npx(), fInfo->fSegSPD->Npz(), Nx, Nz); | |
166 | ||
167 | AliITSdigitSPD *od ; | |
168 | for (Int_t k=0; k<ndigits; ++k) | |
169 | { | |
170 | od = (AliITSdigitSPD*) digits->UncheckedAt(k); | |
171 | ||
172 | fInfo->GetSPDLocalZ(od->GetCoord1(),zo); | |
173 | c1 = od->GetCoord1(); c2 = od->GetCoord2(); | |
9af86b69 | 174 | i = Int_t((zo+fDz)/dpz); |
175 | j = Int_t((od->GetCoord2()*fNx)/fInfo->fSegSPD->Npx()); | |
3b868b75 | 176 | id = j*fNx + i; |
51346b82 | 177 | |
9af86b69 | 178 | ScaledDigit_t* sd = 0; |
3b868b75 | 179 | miter = dmap.find(id); |
180 | if(miter == dmap.end()) | |
51346b82 | 181 | { |
3b868b75 | 182 | dmap[id] = fPlex.Size(); |
183 | z = dpz*(i) - fDz; | |
184 | x = dpx*(j) - fDx; | |
185 | AddQuad(x, z, dpx, dpz); | |
a15e6d7d | 186 | sd = new ScaledDigit_t(c1, c2); |
3b868b75 | 187 | QuadId(sd); |
188 | } | |
51346b82 | 189 | else |
3b868b75 | 190 | { |
a15e6d7d | 191 | sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second)); |
192 | if(c1 < sd->fMinI) | |
193 | sd->fMinI = c1; | |
194 | else if( c1 > sd->fMaxI) | |
195 | sd->fMaxI = c1; | |
196 | ||
197 | if(c2 < sd->fMinJ) | |
198 | sd->fMinJ = c2; | |
199 | else if( c2 > sd->fMaxJ) | |
200 | sd->fMaxJ = c2; | |
3b868b75 | 201 | } |
51346b82 | 202 | |
a15e6d7d | 203 | sd->fN++; |
204 | sd->fSum += od->GetSignal(); | |
205 | sd->fSqrSum += od->GetSignal()*od->GetSignal(); | |
3b868b75 | 206 | } |
207 | break; | |
208 | } | |
51346b82 | 209 | case 1: |
210 | { | |
3b868b75 | 211 | // SDD |
84aff7a4 | 212 | Reset(kQT_RectangleXZFixedY, kFALSE, 32); |
3b868b75 | 213 | |
214 | fNCz = fInfo->fSDDScaleZ[scale]; | |
51346b82 | 215 | fNCx = fInfo->fSDDScaleX[scale]; |
3b868b75 | 216 | fNz = Int_t(fInfo->fSegSDD->Npz()/fNCz); |
217 | fNx = Int_t(fInfo->fSegSDD->Npx()/fNCx); | |
9af86b69 | 218 | dpz = 2*fDz/fNz; |
219 | dpx = 2*fDx/fNx; | |
3b868b75 | 220 | |
9af86b69 | 221 | AliITSdigitSDD *od = 0; |
222 | for (Int_t k = 0; k < ndigits; ++k) | |
223 | { | |
224 | od = (AliITSdigitSDD*)digits->UncheckedAt(k); | |
3b868b75 | 225 | fInfo->fSegSDD->DetToLocal(od->GetCoord2(), od->GetCoord1(),x,z); |
9af86b69 | 226 | z += fDz; |
227 | x += fDx; | |
228 | i = Int_t(z/dpz); | |
229 | j = Int_t(x/dpx); | |
3b868b75 | 230 | //printf("Mod %d coord %d,%d out of %d,%d :: ORIG coord %d,%d out of %d,%d \n",fID, |
231 | // i,j,Nz,Nx,od->GetCoord1(),od->GetCoord2(),fInfo->fSegSDD->Npz(),fInfo->fSegSDD->Npx()); | |
232 | ||
233 | id = j*fNx + i; | |
234 | c1 = od->GetCoord1(); c2 = od->GetCoord2(); | |
51346b82 | 235 | |
9af86b69 | 236 | ScaledDigit_t* sd = 0; |
3b868b75 | 237 | miter = dmap.find(id); |
238 | if(miter == dmap.end()) | |
51346b82 | 239 | { |
3b868b75 | 240 | dmap[id] = fPlex.Size(); |
241 | z = dpz*(i) - fDz; | |
242 | x = dpx*(j) - fDx; | |
243 | AddQuad(x, z, dpx, dpz); | |
a15e6d7d | 244 | sd = new ScaledDigit_t(od->GetCoord1(),od->GetCoord2()); |
3b868b75 | 245 | QuadId(sd); |
246 | } | |
51346b82 | 247 | else |
3b868b75 | 248 | { |
a15e6d7d | 249 | sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second)); |
250 | if(c1 < sd->fMinI) | |
251 | sd->fMinI = c1; | |
252 | else if( c1 > sd->fMaxI) | |
253 | sd->fMaxI = c1; | |
254 | ||
255 | if(c2 < sd->fMinJ) | |
256 | sd->fMinJ = c2; | |
257 | else if( c2 > sd->fMaxJ) | |
258 | sd->fMaxJ = c2; | |
3b868b75 | 259 | } |
a15e6d7d | 260 | sd->fN++; |
261 | sd->fSum += od->GetSignal(); | |
262 | sd->fSqrSum += od->GetSignal()*od->GetSignal(); | |
3b868b75 | 263 | } |
264 | break; | |
51346b82 | 265 | } |
266 | case 2: | |
267 | { | |
268 | // SSD | |
84aff7a4 | 269 | Reset(kQT_LineXZFixedY, kFALSE, 32); |
3b868b75 | 270 | |
51346b82 | 271 | AliITSsegmentationSSD* seg = fInfo->fSegSSD; |
3b868b75 | 272 | Float_t ap, an; // positive/negative angles -> offsets |
273 | seg->Angles(ap, an); | |
274 | ap = TMath::Tan(ap) * fDz; | |
275 | an = - TMath::Tan(an) * fDz; | |
276 | ||
51346b82 | 277 | fNCx = fInfo->fSSDScale[scale]; |
3b868b75 | 278 | fNz = 1; |
279 | fNx = Int_t(fInfo->fSegSSD->Npx()/fNCx); | |
280 | dpz = 2*fDz/fNz; | |
281 | dpx = 2*fDx/fNx; | |
282 | ||
9af86b69 | 283 | AliITSdigitSSD *od = 0; |
3b868b75 | 284 | for (Int_t k=0; k<ndigits; k++) { |
285 | od=(AliITSdigitSSD*)digits->UncheckedAt(k); | |
51346b82 | 286 | if(od->GetCoord1() == 1) |
3b868b75 | 287 | i = 1; // p side |
51346b82 | 288 | else |
3b868b75 | 289 | i= -1; // n side |
290 | j = Int_t(od->GetCoord2()/fNCx); | |
291 | c1 = od->GetCoord1(); c2 = od->GetCoord2(); | |
51346b82 | 292 | id = j*i; |
3b868b75 | 293 | |
9af86b69 | 294 | ScaledDigit_t* sd = 0; |
3b868b75 | 295 | miter = dmap.find(id); |
3b868b75 | 296 | if(miter == dmap.end()) |
51346b82 | 297 | { |
3b868b75 | 298 | // printf("orig digit %d,%d scaled %d,%d \n",od->GetCoord1(),od->GetCoord2(),i,j); |
299 | dmap[id] = fPlex.Size(); | |
300 | z = dpz*(i) - fDz; | |
301 | x = dpx*(j) - fDx; | |
302 | Float_t a = ( od->GetCoord1() == 1) ? ap : an; | |
303 | AddLine(x-a, -fDz, 2*a, 2*fDz); | |
304 | ||
a15e6d7d | 305 | sd = new ScaledDigit_t(c1, c2); |
3b868b75 | 306 | QuadId(sd); |
307 | } | |
51346b82 | 308 | else |
3b868b75 | 309 | { |
a15e6d7d | 310 | sd = dynamic_cast<ScaledDigit_t*>(GetId(miter->second)); |
311 | if(c1 < sd->fMinI) | |
312 | sd->fMinI = c1; | |
313 | else if( c1 > sd->fMaxI) | |
314 | sd->fMaxI = c1; | |
315 | ||
316 | if(c2 < sd->fMinJ) | |
317 | sd->fMinJ = c2; | |
318 | else if( c2 > sd->fMaxJ) | |
319 | sd->fMaxJ = c2; | |
3b868b75 | 320 | } |
a15e6d7d | 321 | sd->fN++; |
322 | sd->fSum += od->GetSignal(); | |
323 | sd->fSqrSum += od->GetSignal()*od->GetSignal(); | |
3b868b75 | 324 | } // for digits |
325 | break; | |
326 | } // end case 2 | |
327 | } // end switch | |
328 | ||
329 | SetQuadValues(); | |
330 | RefitPlex(); | |
331 | } | |
332 | ||
57ffa5fb | 333 | /******************************************************************************/ |
3b868b75 | 334 | |
d810d0de | 335 | void AliEveITSScaledModule::SetQuadValues() |
3b868b75 | 336 | { |
32e219c2 | 337 | if(fScaleInfo->GetSyncPalette()) SyncPalette(); |
338 | ||
a15e6d7d | 339 | Int_t num = fPlex.Size(); |
340 | for (Int_t i = 0 ; i < num; i++) | |
3b868b75 | 341 | { |
a15e6d7d | 342 | ScaledDigit_t* sd = dynamic_cast<ScaledDigit_t*>(GetId(i)); |
3b868b75 | 343 | Int_t v = 0; |
32e219c2 | 344 | switch(fScaleInfo->GetStatType()) |
345 | { | |
346 | using namespace TMath; | |
347 | ||
a15e6d7d | 348 | case AliEveDigitScaleInfo::kSTOccup: |
349 | v = Nint((100.0*sd->fN) / (fNCx*fNCz)); | |
32e219c2 | 350 | break; |
a15e6d7d | 351 | case AliEveDigitScaleInfo::kSTAverage: |
352 | v = Nint((Double_t) sd->fSum / sd->fN); | |
32e219c2 | 353 | break; |
a15e6d7d | 354 | case AliEveDigitScaleInfo::kSTRms: |
355 | v = Nint(Sqrt(sd->fSqrSum) / sd->fN); | |
32e219c2 | 356 | break; |
3b868b75 | 357 | } |
84aff7a4 | 358 | DigitBase_t* qb = GetDigit(i); |
3b868b75 | 359 | qb->fValue = v; |
360 | } | |
361 | } | |
362 | ||
57ffa5fb | 363 | /******************************************************************************/ |
3b868b75 | 364 | |
d810d0de | 365 | void AliEveITSScaledModule::SyncPalette() |
51346b82 | 366 | { |
d810d0de | 367 | // printf("AliEveITSScaledModule::SyncPalette()\n"); |
a15e6d7d | 368 | if(fScaleInfo->GetStatType() == AliEveDigitScaleInfo::kSTOccup) |
32e219c2 | 369 | { |
370 | // SPD | |
d810d0de | 371 | AliEveITSModule::fgSPDPalette->SetLimits(0, 100); |
372 | AliEveITSModule::fgSPDPalette->SetMinMax(0, 100); | |
51346b82 | 373 | |
32e219c2 | 374 | // SDD |
d810d0de | 375 | AliEveITSModule::fgSDDPalette->SetLimits(0, 100); |
376 | AliEveITSModule::fgSDDPalette->SetMinMax(0, 100); | |
32e219c2 | 377 | |
378 | // SSD | |
d810d0de | 379 | AliEveITSModule::fgSSDPalette->SetLimits(0, 100); |
380 | AliEveITSModule::fgSDDPalette->SetMinMax(0, 100); | |
32e219c2 | 381 | } |
382 | else | |
383 | { | |
a15e6d7d | 384 | AliEveITSDigitsInfo& di = *fInfo; |
32e219c2 | 385 | // SPD |
a15e6d7d | 386 | AliEveITSModule::fgSPDPalette->SetLimits(0, di.fSPDHighLim); |
387 | AliEveITSModule::fgSPDPalette->SetMinMax(di.fSPDMinVal, di.fSPDMaxVal); | |
51346b82 | 388 | |
32e219c2 | 389 | // SDD |
a15e6d7d | 390 | AliEveITSModule::fgSDDPalette->SetLimits(0, di.fSDDHighLim); |
391 | AliEveITSModule::fgSDDPalette->SetMinMax(di.fSDDMinVal, di.fSDDMaxVal); | |
32e219c2 | 392 | |
393 | // SSD | |
a15e6d7d | 394 | AliEveITSModule::fgSSDPalette->SetLimits(0, di.fSSDHighLim); |
395 | AliEveITSModule::fgSSDPalette->SetMinMax(di.fSSDMinVal, di.fSSDMaxVal); | |
32e219c2 | 396 | } |
397 | ||
398 | fScaleInfo->SetSyncPalette(kFALSE); | |
399 | } | |
400 | ||
57ffa5fb | 401 | /******************************************************************************/ |
32e219c2 | 402 | |
a15e6d7d | 403 | void AliEveITSScaledModule::GetScaleData(Int_t& cnx, Int_t& cnz, Int_t& total) const |
32e219c2 | 404 | { |
a15e6d7d | 405 | cnx = fNx; |
406 | cnz = fNz; | |
32e219c2 | 407 | total = cnx*cnz; |
408 | } | |
409 | ||
57ffa5fb | 410 | /******************************************************************************/ |
32e219c2 | 411 | |
d810d0de | 412 | void AliEveITSScaledModule::DigitSelected(Int_t idx) |
3b868b75 | 413 | { |
84aff7a4 | 414 | // Override control-click from TEveQuadSet |
d810d0de | 415 | printf("AliEveITSScaledModule::DigitSelected "); Print(); |
3b868b75 | 416 | |
84aff7a4 | 417 | DigitBase_t* qb = GetDigit(idx); |
3b868b75 | 418 | TObject* obj = qb->fId.GetObject(); |
a15e6d7d | 419 | ScaledDigit_t* sd = dynamic_cast<ScaledDigit_t*>(obj); |
3b868b75 | 420 | TClonesArray *digits = fInfo->GetDigits(fID, fDetID); |
421 | Int_t ndigits = digits->GetEntriesFast(); | |
422 | ||
a15e6d7d | 423 | printf("%d digits in cell scaleX = %d, scaleZ = %d \n", sd->fN, fNCx, fNCz); |
3b868b75 | 424 | |
425 | Int_t il = 0; | |
51346b82 | 426 | for(Int_t k=0; k<ndigits; k++) |
3b868b75 | 427 | { |
428 | AliITSdigit *d = (AliITSdigit*) digits->UncheckedAt(k); | |
429 | ||
a15e6d7d | 430 | if(d->GetCoord1()>=sd->fMinI && d->GetCoord1()<=sd->fMaxI && |
431 | d->GetCoord2()>=sd->fMinJ && d->GetCoord2()<=sd->fMaxJ) | |
3b868b75 | 432 | { |
433 | printf("%3d, %3d: %3d", d->GetCoord1(), d->GetCoord2(), d->GetSignal()); | |
434 | printf(" | "); | |
435 | il++; | |
436 | if(il>5) { | |
437 | printf("\n"); | |
438 | il = 0; | |
439 | } | |
440 | } | |
441 | } | |
442 | if(il) printf("\n"); | |
443 | } |