]>
Commit | Line | Data |
---|---|---|
a9579262 | 1 | /************************************************************************** |
2 | * Copyright(c) 2004, ALICE Experiment at CERN, All rights reserved. * | |
3 | * * | |
4 | * Author: The ALICE Off-line Project. * | |
5 | * Contributors are mentioned in the code where appropriate. * | |
6 | * * | |
7 | * Permission to use, copy, modify and distribute this software and its * | |
8 | * documentation strictly for non-commercial purposes is hereby granted * | |
9 | * without fee, provided that the above copyright notice appears in all * | |
10 | * copies and that both the copyright notice and this permission notice * | |
11 | * appear in the supporting documentation. The authors make no claims * | |
12 | * about the suitability of this software for any purpose. It is * | |
13 | * provided "as is" without express or implied warranty. * | |
14 | **************************************************************************/ | |
15 | /* $Id$ */ | |
16 | /** @file AliFMDPattern.cxx | |
17 | @author Christian Holm Christensen <cholm@nbi.dk> | |
18 | @date Mon Mar 27 12:39:09 2006 | |
19 | @brief FMD 2D Event display | |
20 | */ | |
21 | //___________________________________________________________________ | |
22 | // | |
23 | // The classes defined here, are utility classes for reading in data | |
24 | // for the FMD. They are put in a seperate library to not polute the | |
25 | // normal libraries. The classes are intended to be used as base | |
26 | // classes for customized class that do some sort of analysis on the | |
27 | // various types of data produced by the FMD. | |
28 | // | |
29 | // Latest changes by Christian Holm Christensen | |
30 | // | |
9edefa04 | 31 | |
f95a63c4 | 32 | // #include <iostream> |
9edefa04 | 33 | |
f95a63c4 | 34 | // #include <TApplication.h> |
35 | // #include <TButton.h> | |
9edefa04 | 36 | #include <TCanvas.h> |
37 | #include <TH2F.h> | |
38 | #include <TMath.h> | |
39 | #include <TPad.h> | |
40 | #include <TRandom.h> | |
f95a63c4 | 41 | // #include <TSlider.h> |
9edefa04 | 42 | #include <TStyle.h> |
f95a63c4 | 43 | // #include <TSystem.h> |
9edefa04 | 44 | #include <TVector2.h> |
f95a63c4 | 45 | // #include <TView.h> |
46 | #include <TGraph.h> | |
a9579262 | 47 | #include "AliFMDPattern.h" // ALIFMDDISPLAY_H |
48 | #include "AliFMDGeometry.h" // ALIFMDGEOMETRY_H | |
f95a63c4 | 49 | //#include "AliFMDParameters.h" // ALIFMDPARAMETERS_H |
a9579262 | 50 | #include "AliFMDRing.h" |
f95a63c4 | 51 | // #include "AliFMDDetector.h" |
a9579262 | 52 | #include "AliFMDHit.h" |
f95a63c4 | 53 | // #include <AliLog.h> |
54 | #include "AliFMDDebug.h" // Better debug macros | |
55 | class AliFMDDetector; | |
a9579262 | 56 | |
57 | //____________________________________________________________________ | |
58 | ClassImp(AliFMDPattern) | |
59 | #if 0 | |
60 | ; // This is here to keep Emacs for indenting the next line | |
61 | #endif | |
62 | ||
63 | //____________________________________________________________________ | |
f95a63c4 | 64 | AliFMDPattern::AliFMDPatternDetector::AliFMDPatternDetector(UShort_t id) |
a9579262 | 65 | : fId(id), |
66 | fCounts(0), | |
67 | fGraphs(0), | |
68 | fFrame(0) | |
f95a63c4 | 69 | { |
70 | // CTOR | |
71 | // | |
72 | // Parameters: | |
73 | // | |
74 | // ID Identifier | |
75 | } | |
a9579262 | 76 | |
77 | //____________________________________________________________________ | |
f95a63c4 | 78 | AliFMDPattern::AliFMDPatternDetector::~AliFMDPatternDetector() |
a9579262 | 79 | { |
f95a63c4 | 80 | // DTOR |
81 | // Destructor - | |
82 | // deletes mother frame | |
a9579262 | 83 | if (fFrame) delete fFrame; |
84 | } | |
85 | ||
86 | //____________________________________________________________________ | |
87 | void | |
f95a63c4 | 88 | AliFMDPattern::AliFMDPatternDetector::DrawShape(TObjArray& a) |
a9579262 | 89 | { |
f95a63c4 | 90 | // Draw all shapes. |
91 | // | |
92 | // Paramters | |
93 | // | |
94 | // a Array of shapes | |
95 | // | |
a9579262 | 96 | TIter next(&a); |
97 | TGraph* g = 0; | |
98 | while ((g = static_cast<TGraph*>(next()))) { | |
99 | g->DrawClone("f same"); | |
100 | g->DrawClone("l same"); | |
101 | } | |
102 | } | |
103 | ||
104 | //____________________________________________________________________ | |
105 | void | |
cf291b42 | 106 | AliFMDPattern::AliFMDPatternDetector::Begin(Int_t nlevel, |
107 | Double_t r, | |
108 | TObjArray& inners, | |
109 | TObjArray& outers) | |
a9579262 | 110 | { |
f95a63c4 | 111 | // Start of a run. |
112 | // | |
113 | // Parameters | |
114 | // | |
115 | // nlevel Number of levels | |
116 | // r Radius | |
117 | // inners Array of inner shapes | |
118 | // outers Array of outer shapes | |
119 | // | |
120 | ||
121 | // To make code-checker shut up | |
122 | TStyle* style = gStyle; | |
123 | if (nlevel < 1) nlevel = style->GetNumberOfColors(); | |
a9579262 | 124 | fCounts.Set(nlevel); |
125 | if (!fFrame) { | |
f95a63c4 | 126 | // The code-checker thinks this is not using the declaration of |
127 | // TH2F - what a morron! | |
a9579262 | 128 | fFrame = new TH2F(Form("fmd%dFrame", fId), Form("FMD%d", fId), |
129 | 10, -r, r, 10, -r, r); | |
130 | fFrame->SetStats(kFALSE); | |
131 | fFrame->Draw(); | |
132 | } | |
133 | DrawShape(inners); | |
134 | if (fId != 1) DrawShape(outers); | |
135 | for (Int_t i = 0; i < nlevel; i++) { | |
136 | TGraph* g = new TGraph; | |
f95a63c4 | 137 | Int_t idx = Int_t(Float_t(i) / nlevel * style->GetNumberOfColors()); |
138 | Int_t col = style->GetColorPalette(idx); | |
a9579262 | 139 | g->SetName(Form("FMD%d_L%02d", fId, i)); |
140 | g->SetMarkerColor(col); | |
141 | g->SetLineColor(col); | |
142 | g->SetFillColor(col); | |
143 | g->SetMarkerSize(i * .2 + .2); | |
144 | g->SetMarkerStyle(2); | |
145 | g->Draw("same p"); | |
146 | fGraphs.AddAtAndExpand(g, i); | |
147 | } | |
cf291b42 | 148 | // TIter next(&fGraphs); |
a9579262 | 149 | } |
150 | ||
151 | //____________________________________________________________________ | |
152 | void | |
f95a63c4 | 153 | AliFMDPattern::AliFMDPatternDetector::Clear() |
a9579262 | 154 | { |
f95a63c4 | 155 | // Clear this display. |
156 | // Simply reset counters to zero. | |
157 | // Avoid deleting memory. | |
a9579262 | 158 | fCounts.Reset(0); |
159 | } | |
160 | ||
161 | //____________________________________________________________________ | |
162 | void | |
f95a63c4 | 163 | AliFMDPattern::AliFMDPatternDetector::End() |
a9579262 | 164 | { |
f95a63c4 | 165 | // Called when displaying the data. |
166 | // Simply resets number of points at each level to | |
167 | // the seen number of hits at that level. | |
168 | // Avoid deleting memory. | |
a9579262 | 169 | TIter next(&fGraphs); |
170 | TGraph* g = 0; | |
171 | Int_t i = 0; | |
172 | while ((g = static_cast<TGraph*>(next()))) g->Set(fCounts[i++]); | |
173 | } | |
174 | //____________________________________________________________________ | |
175 | void | |
cf291b42 | 176 | AliFMDPattern::AliFMDPatternDetector::AddMarker(Double_t x, |
177 | Double_t y, | |
178 | Float_t s, | |
179 | Float_t max) | |
a9579262 | 180 | { |
f95a63c4 | 181 | // Add a marker at (X,Y,Z). The marker color and size is chosen |
182 | // relative to the MAX argument. | |
183 | // | |
184 | // Parameters | |
185 | // | |
186 | // X,Y,Z Coordiantes | |
187 | // MAX Maximum value. | |
188 | // | |
189 | /** Sigh, for some odd reason, the code-checker does not recognise | |
190 | this a usage of the TMath namespace declaration! Idiot */ | |
191 | Int_t i = TMath::Min(Int_t(fCounts.fN * s / max), | |
a9579262 | 192 | Int_t(fGraphs.GetEntries()-1)); |
193 | TGraph* g = static_cast<TGraph*>(fGraphs.At(i)); | |
194 | if (!g) return; | |
195 | g->SetPoint(fCounts[i]++, x, y); | |
196 | } | |
197 | ||
198 | ||
199 | //____________________________________________________________________ | |
200 | AliFMDPattern::AliFMDPattern(const char* gAliceFile) | |
201 | : AliFMDDisplay(kTRUE, gAliceFile), | |
17e542eb | 202 | fInners(0), |
203 | fOuters(0), | |
a9579262 | 204 | fInnerMax(0), |
205 | fOuterMax(0), | |
206 | fFMD1Pad(0), | |
207 | fFMD1(1), | |
208 | fFMD2Pad(0), | |
209 | fFMD2(2), | |
210 | fFMD3Pad(0), | |
211 | fFMD3(3), | |
17e542eb | 212 | fSummary(0), |
a9579262 | 213 | fEvent(.1, .8, "Event #"), |
214 | fFMD1Sum(.2, .7, "# in FMD1: "), | |
215 | fFMD2Sum(.2, .6, "# in FMD2: "), | |
216 | fFMD3Sum(.2, .5, "# in FMD3: "), | |
217 | fLine(.15, .47, .85, .47), | |
218 | fTotal(.2, .35, "Total: ") | |
219 | { | |
f95a63c4 | 220 | // Constructor. |
221 | // | |
222 | // Parameters | |
223 | // | |
224 | // gAliceFile The galice.root file to use - if any. | |
225 | // | |
226 | ||
a9579262 | 227 | // RemoveLoad(kGeometry); |
228 | fEvent.SetBit(TLatex::kTextNDC); | |
229 | fFMD1Sum.SetBit(TLatex::kTextNDC); | |
230 | fFMD2Sum.SetBit(TLatex::kTextNDC); | |
231 | fFMD3Sum.SetBit(TLatex::kTextNDC); | |
232 | fLine.SetBit(TLine::kLineNDC); | |
233 | fTotal.SetBit(TLatex::kTextNDC); | |
234 | } | |
235 | ||
236 | //____________________________________________________________________ | |
237 | AliFMDPattern::~AliFMDPattern() | |
238 | { | |
f95a63c4 | 239 | // DTOR |
240 | // Free all allocated shapes. | |
241 | // note, that most members are real objects, so we do not need to | |
242 | // deal with them here. | |
a9579262 | 243 | fInners.Delete(); |
244 | fOuters.Delete(); | |
245 | } | |
246 | ||
247 | ||
248 | //____________________________________________________________________ | |
249 | Bool_t | |
250 | AliFMDPattern::Init() | |
251 | { | |
f95a63c4 | 252 | // Initialize. Get transforms and such, |
a9579262 | 253 | if (!AliFMDInput::Init()) return kFALSE; |
254 | AliFMDGeometry* geom = AliFMDGeometry::Instance(); | |
69893a66 | 255 | if (!geom) return kFALSE; |
a9579262 | 256 | geom->Init(); |
257 | geom->InitTransformations(); | |
258 | ||
259 | Char_t rs[] = { 'I' , 'O', '\0' }; | |
260 | Char_t *r = rs; | |
261 | do { | |
262 | AliFMDRing* ring = geom->GetRing(*r); | |
263 | if (!ring) continue; | |
264 | const TObjArray& vs = ring->GetVerticies(); | |
265 | TObjArray& gs = (*r == 'I' ? fInners : fOuters); | |
266 | Float_t& mr = (*r == 'I' ? fInnerMax : fOuterMax); | |
267 | Int_t nm = ring->GetNModules(); | |
268 | AliInfo(Form("Making %d modules for %c", nm, *r)); | |
269 | for (Int_t m = 0; m < nm; m++) { | |
270 | Int_t nv = vs.GetEntries(); | |
271 | Double_t a = TMath::Pi() / 180 * (m * 2 + 1) * ring->GetTheta(); | |
272 | TGraph* g = new TGraph(nv+1); | |
273 | Double_t x0 = 0, y0 = 0; | |
274 | gs.AddAtAndExpand(g, m); | |
275 | for (Int_t c = 0; c < nv; c++) { | |
276 | TVector2* v = static_cast<TVector2*>(vs.At(c)); | |
277 | mr = TMath::Max(mr, Float_t(v->Mod())); | |
278 | TVector2 w(v->Rotate(a)); | |
279 | if (c == 0) { x0 = w.X(); y0 = w.Y(); } | |
280 | g->SetPoint(c, w.X(), w.Y()); | |
281 | } | |
282 | g->SetName(Form("FMDX%c_%02d", *r, m)); | |
283 | g->SetPoint(nv, x0, y0); | |
284 | g->SetFillColor((*rs == 'I' ? | |
285 | (m % 2 == 0 ? 18 : 17) : | |
286 | (m % 2 == 0 ? 20 : 23))); | |
287 | g->SetFillStyle(3001); | |
288 | g->SetLineColor(1); | |
289 | g->SetLineWidth(1); | |
290 | g->SetLineStyle(2); | |
291 | } | |
292 | } while (*(++r)); | |
293 | ||
294 | return kTRUE; | |
295 | } | |
296 | ||
297 | //____________________________________________________________________ | |
298 | Bool_t | |
299 | AliFMDPattern::Begin(Int_t event) | |
300 | { | |
f95a63c4 | 301 | // Called at the begining of an event. |
302 | // | |
303 | // Parameters | |
304 | // | |
305 | // EVENT The event number | |
306 | // | |
97b4001e | 307 | MakeAux(); |
a9579262 | 308 | if (!fCanvas) { |
309 | const char* which[] = { "Continue", "Redisplay", 0 }; | |
310 | MakeCanvas(which); | |
a9579262 | 311 | |
312 | AliFMDGeometry* geom = AliFMDGeometry::Instance(); | |
f95a63c4 | 313 | // AliFMDDetector* det; |
314 | if ((/* det = */ geom->GetDetector(1))) { | |
a9579262 | 315 | fPad->cd(); |
316 | fFMD1Pad = new TPad("FMD1", "FMD1", 0.0, 0.50, 0.5, 1.0, 0, 0); | |
317 | fFMD1Pad->Draw(); | |
318 | fFMD1Pad->cd(); | |
97b4001e | 319 | fFMD1.Begin(-1, fInnerMax, fInners, fOuters); |
a9579262 | 320 | } |
f95a63c4 | 321 | if ((/* det = */ geom->GetDetector(2))) { |
a9579262 | 322 | fPad->cd(); |
323 | fFMD2Pad = new TPad("FMD2", "FMD2", 0.5, 0.50, 1.0, 1.0, 0, 0); | |
324 | fFMD2Pad->Draw(); | |
325 | fFMD2Pad->cd(); | |
97b4001e | 326 | fFMD2.Begin(-1, fOuterMax, fInners, fOuters); |
a9579262 | 327 | } |
f95a63c4 | 328 | if ((/* det = */ geom->GetDetector(3))) { |
a9579262 | 329 | fPad->cd(); |
330 | fFMD3Pad = new TPad("FMD3", "FMD3", 0.0, 0.0, .5, .5, 0, 0); | |
331 | fFMD3Pad->Draw(); | |
332 | fFMD3Pad->cd(); | |
97b4001e | 333 | fFMD3.Begin(-1, fOuterMax, fInners, fOuters); |
a9579262 | 334 | } |
335 | fPad->cd(); | |
336 | fSummary = new TPad("display", "Display", 0.5, 0.0, 1.0, 0.5, 0, 0); | |
337 | fSummary->Draw(); | |
338 | fSummary->cd(); | |
339 | fEvent.Draw(); | |
340 | fFMD1Sum.Draw(); | |
341 | fFMD2Sum.Draw(); | |
342 | fFMD3Sum.Draw(); | |
343 | fLine.Draw(); | |
344 | fTotal.Draw(); | |
345 | } | |
346 | fEvent.SetTitle(Form("Event # %6d", event)); | |
347 | ||
348 | fCanvas->Modified(); | |
349 | fCanvas->Update(); | |
350 | fCanvas->cd(); | |
351 | fFMD1.Clear(); | |
352 | fFMD2.Clear(); | |
353 | fFMD3.Clear(); | |
354 | return AliFMDInput::Begin(event); | |
355 | } | |
356 | ||
357 | //____________________________________________________________________ | |
358 | void | |
359 | AliFMDPattern::Redisplay() | |
360 | { | |
f95a63c4 | 361 | // Redraw the displayu |
a9579262 | 362 | fFMD1.Clear(); |
363 | fFMD2.Clear(); | |
364 | fFMD3.Clear(); | |
365 | AliFMDDisplay::Redisplay(); | |
366 | } | |
367 | ||
368 | //____________________________________________________________________ | |
369 | void | |
370 | AliFMDPattern::AtEnd() | |
371 | { | |
f95a63c4 | 372 | // Called at the end of an event. |
a9579262 | 373 | DrawAux(); |
374 | ||
375 | Int_t total = 0; | |
376 | ||
377 | fFMD1.End(); | |
378 | fFMD1Pad->Modified(); | |
379 | fFMD1Sum.SetTitle(Form("# hits in FMD1: %5d", fFMD1.Total())); | |
380 | total += fFMD1.Total(); | |
381 | ||
382 | fFMD2.End(); | |
383 | fFMD2Pad->Modified(); | |
384 | fFMD2Sum.SetTitle(Form("# hits in FMD2: %5d", fFMD2.Total())); | |
385 | total += fFMD2.Total(); | |
386 | ||
387 | fFMD3.End(); | |
388 | fFMD3Pad->Modified(); | |
389 | fFMD3Sum.SetTitle(Form("# hits in FMD3: %5d", fFMD3.Total())); | |
390 | total += fFMD3.Total(); | |
391 | ||
392 | fTotal.SetTitle(Form("Total: %5d/51200 (%3d%%)", | |
393 | total, Int_t(100. / 51200 * total))); | |
394 | fSummary->Modified(); | |
395 | fCanvas->Modified(); | |
396 | fCanvas->Update(); | |
397 | fCanvas->cd(); | |
398 | } | |
399 | ||
400 | //____________________________________________________________________ | |
401 | Bool_t | |
402 | AliFMDPattern::ProcessHit(AliFMDHit* hit, TParticle*) | |
403 | { | |
f95a63c4 | 404 | // Process a hit. |
405 | // | |
406 | // Parameters | |
407 | // | |
408 | // HIT The hit to process. | |
409 | // | |
410 | // The TParticle argument is never used. | |
a9579262 | 411 | switch (hit->Detector()) { |
412 | case 1: fFMD1.AddMarker(hit->X(), hit->Y(), hit->Edep(), 1); break; | |
413 | case 2: fFMD2.AddMarker(hit->X(), hit->Y(), hit->Edep(), 1); break; | |
414 | case 3: fFMD3.AddMarker(hit->X(), hit->Y(), hit->Edep(), 1); break; | |
415 | } | |
416 | return kTRUE; | |
417 | } | |
418 | ||
419 | ||
420 | //____________________________________________________________________ | |
421 | void | |
f95a63c4 | 422 | AliFMDPattern::AddMarker(UShort_t det, Char_t rng, |
423 | UShort_t sec, UShort_t str, | |
cf291b42 | 424 | TObject*, Float_t s, Float_t /*min*/, Float_t max) |
a9579262 | 425 | { |
426 | // Add a marker to the display | |
427 | // | |
428 | // det Detector | |
429 | // rng Ring | |
430 | // sec Sector | |
431 | // str Strip | |
432 | // o Object to refer to | |
433 | // s Signal | |
434 | // max Maximum of signal | |
435 | // | |
f95a63c4 | 436 | AliFMDPatternDetector* d = 0; |
a9579262 | 437 | switch (det) { |
438 | case 1: d = &fFMD1; break; | |
439 | case 2: d = &fFMD2; break; | |
440 | case 3: d = &fFMD3; break; | |
441 | } | |
442 | if (!d) return; | |
443 | AliFMDGeometry* geom = AliFMDGeometry::Instance(); | |
444 | Double_t x, y, z; | |
445 | geom->Detector2XYZ(det, rng, sec, str, x, y, z); | |
f95a63c4 | 446 | // Make code-checker shut the f**k up |
447 | TRandom* rand = gRandom; | |
a9579262 | 448 | if (true) { |
449 | AliFMDRing* r = geom->GetRing(rng); | |
450 | Double_t t = .9 * r->GetTheta() / 2; | |
f95a63c4 | 451 | Double_t a = rand->Uniform(-t,t) * TMath::Pi() / 180; |
a9579262 | 452 | Double_t x1 = x * TMath::Cos(a) - y * TMath::Sin(a); |
453 | Double_t y1 = x * TMath::Sin(a) + y * TMath::Cos(a); | |
454 | x = x1; | |
455 | y = y1; | |
456 | } | |
457 | d->AddMarker(x, y, s, max); | |
458 | } | |
459 | ||
460 | //____________________________________________________________________ | |
461 | // | |
462 | // EOF | |
463 | // |