1 /**************************************************************************
2 * Copyright(c) 2004, ALICE Experiment at CERN, All rights reserved. *
4 * Author: The ALICE Off-line Project. *
5 * Contributors are mentioned in the code where appropriate. *
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 **************************************************************************/
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
21 //___________________________________________________________________
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.
29 // Latest changes by Christian Holm Christensen
32 // #include <iostream>
34 // #include <TApplication.h>
35 // #include <TButton.h>
41 // #include <TSlider.h>
43 // #include <TSystem.h>
47 #include "AliFMDPattern.h" // ALIFMDDISPLAY_H
48 #include "AliFMDGeometry.h" // ALIFMDGEOMETRY_H
49 //#include "AliFMDParameters.h" // ALIFMDPARAMETERS_H
50 #include "AliFMDRing.h"
51 // #include "AliFMDDetector.h"
52 #include "AliFMDHit.h"
53 // #include <AliLog.h>
54 #include "AliFMDDebug.h" // Better debug macros
57 //____________________________________________________________________
58 ClassImp(AliFMDPattern)
60 ; // This is here to keep Emacs for indenting the next line
63 //____________________________________________________________________
64 AliFMDPattern::AliFMDPatternDetector::AliFMDPatternDetector(UShort_t id)
69 fInners(id == 1 ? 10 : 0)
78 //____________________________________________________________________
79 AliFMDPattern::AliFMDPatternDetector::~AliFMDPatternDetector()
83 // deletes mother frame
84 if (fFrame) delete fFrame;
87 //____________________________________________________________________
89 AliFMDPattern::AliFMDPatternDetector::DrawShape(TObjArray& a)
99 while ((g = static_cast<TGraph*>(next()))) {
100 g->DrawClone("f same");
101 g->DrawClone("l same");
105 //____________________________________________________________________
107 AliFMDPattern::AliFMDPatternDetector::CopyShapes(TObjArray& src,
115 while ((g = static_cast<TGraph*>(next()))) {
116 TGraph* gg = new TGraph(*g);
117 Double_t* x = gg->GetX();
118 Double_t* y = gg->GetY();
119 for (Int_t i = 0; i < gg->GetN(); i++) {
120 Float_t xx = x[i] * TMath::Cos(ang) - y[i] * TMath::Sin(ang);
121 Float_t yy = x[i] * TMath::Sin(ang) + y[i] * TMath::Cos(ang);
122 gg->SetPoint(i, fx * xx, fy * yy);
124 gg->SetFillStyle(g->GetFillStyle());
125 gg->SetFillColor(g->GetFillColor());
126 gg->SetLineStyle(g->GetLineStyle());
127 gg->SetLineColor(g->GetLineColor());
128 gg->SetLineWidth(g->GetLineWidth());
129 gg->SetMarkerStyle(g->GetMarkerStyle());
130 gg->SetMarkerColor(g->GetMarkerColor());
131 gg->SetMarkerSize(g->GetMarkerSize());
132 TString name(g->GetName());
133 name.ReplaceAll("X", Form("%d",fId));
134 gg->SetName(name.Data());
135 TString title(g->GetTitle());
136 title.ReplaceAll("X", Form("%d",fId));
137 gg->SetTitle(title.Data());
143 //____________________________________________________________________
145 AliFMDPattern::AliFMDPatternDetector::Begin(Int_t nlevel,
154 // nlevel Number of levels
156 // inners Array of inner shapes
157 // outers Array of outer shapes
160 // To make code-checker shut up
161 TStyle* style = gStyle;
162 if (nlevel < 1) nlevel = style->GetNumberOfColors();
164 Double_t rr = 1.05 * r;
166 // The code-checker thinks this is not using the declaration of
167 // TH2F - what a morron!
168 fFrame = new TH2F(Form("fmd%dFrame", fId), Form("FMD%d", fId),
169 100, -rr, rr, 100, -rr, rr);
170 fFrame->SetStats(kFALSE);
173 Double_t ang = (fId == 1 ? -TMath::Pi() / 2 : 0);
174 Double_t fx = (fId == 3 ? -1 : 1); // Flip around Y
175 Double_t fy = (fId == 1 ? 1 : 1); // Flip around X
177 CopyShapes(inners, fInners, ang, fx, fy);
180 CopyShapes(outers, fOuters, ang, fx, fy);
184 for (Int_t i = 0; i < nlevel; i++) {
185 TGraph* g = new TGraph;
186 Int_t idx = Int_t(Float_t(i) / nlevel * style->GetNumberOfColors());
187 Int_t col = style->GetColorPalette(idx);
188 g->SetName(Form("FMD%d_L%02d", fId, i));
189 g->SetMarkerColor(col);
190 g->SetLineColor(col);
191 g->SetFillColor(col);
192 g->SetMarkerSize(i * .2 + .2);
193 g->SetMarkerStyle(2);
194 g->SetEditable(kFALSE);
196 fGraphs.AddAtAndExpand(g, i);
198 // TIter next(&fGraphs);
201 //____________________________________________________________________
203 AliFMDPattern::AliFMDPatternDetector::Clear()
205 // Clear this display.
206 // Simply reset counters to zero.
207 // Avoid deleting memory.
211 //____________________________________________________________________
213 AliFMDPattern::AliFMDPatternDetector::End()
215 // Called when displaying the data.
216 // Simply resets number of points at each level to
217 // the seen number of hits at that level.
218 // Avoid deleting memory.
219 TIter next(&fGraphs);
222 while ((g = static_cast<TGraph*>(next()))) {
223 Int_t cnt = fCounts[i++];
226 g->SetMarkerSize(i * .2 + .2);
235 //____________________________________________________________________
237 AliFMDPattern::AliFMDPatternDetector::AddMarker(Double_t x,
242 // Add a marker at (X,Y,Z). The marker color and size is chosen
243 // relative to the MAX argument.
248 // MAX Maximum value.
250 /** Sigh, for some odd reason, the code-checker does not recognise
251 this a usage of the TMath namespace declaration! Idiot */
252 Int_t i = TMath::Min(Int_t(fCounts.fN * s / max),
253 Int_t(fGraphs.GetEntries()-1));
254 TGraph* g = static_cast<TGraph*>(fGraphs.At(i));
256 g->SetPoint(fCounts[i]++, x, y);
260 //____________________________________________________________________
261 AliFMDPattern::AliFMDPattern(const char* gAliceFile)
262 : AliFMDDisplay(kTRUE, gAliceFile),
274 fEvent(.1, .8, "Event #"),
275 fFMD1Sum(.2, .7, "# in FMD1: "),
276 fFMD2Sum(.2, .6, "# in FMD2: "),
277 fFMD3Sum(.2, .5, "# in FMD3: "),
278 fLine(.15, .47, .85, .47),
279 fTotal(.2, .35, "Total: ")
285 // gAliceFile The galice.root file to use - if any.
288 SetName("AliFMDPattern");
289 SetName("2D display of FMD data");
291 // RemoveLoad(kGeometry);
292 fEvent.SetBit(TLatex::kTextNDC);
293 fFMD1Sum.SetBit(TLatex::kTextNDC);
294 fFMD2Sum.SetBit(TLatex::kTextNDC);
295 fFMD3Sum.SetBit(TLatex::kTextNDC);
296 fLine.SetBit(TLine::kLineNDC);
297 fTotal.SetBit(TLatex::kTextNDC);
300 //____________________________________________________________________
301 AliFMDPattern::~AliFMDPattern()
304 // Free all allocated shapes.
305 // note, that most members are real objects, so we do not need to
306 // deal with them here.
312 //____________________________________________________________________
314 AliFMDPattern::Init()
316 // Initialize. Get transforms and such,
317 if (!AliFMDInput::Init()) return kFALSE;
318 AliFMDGeometry* geom = AliFMDGeometry::Instance();
319 if (!geom) return kFALSE;
321 geom->InitTransformations();
323 Char_t rs[] = { 'I' , 'O', '\0' };
326 AliFMDRing* ring = geom->GetRing(*r);
328 const TObjArray& vs = ring->GetVerticies();
329 TObjArray& gs = (*r == 'I' ? fInners : fOuters);
330 Float_t& mr = (*r == 'I' ? fInnerMax : fOuterMax);
331 Int_t nm = ring->GetNModules();
332 AliFMDDebug(1, ("Making %d modules for %c", nm, *r));
333 for (Int_t m = 0; m < nm; m++) {
334 Int_t nv = vs.GetEntries();
335 Double_t a = TMath::Pi() / 180 * (m * 2 + 1) * ring->GetTheta();
336 TGraph* g = new TGraph(nv+1);
337 Double_t x0 = 0, y0 = 0;
338 gs.AddAtAndExpand(g, m);
339 for (Int_t c = 0; c < nv; c++) {
340 TVector2* v = static_cast<TVector2*>(vs.At(c));
341 mr = TMath::Max(mr, Float_t(v->Mod()));
342 TVector2 w(v->Rotate(a));
343 if (c == 0) { x0 = w.X(); y0 = w.Y(); }
344 g->SetPoint(c, w.X(), w.Y());
346 g->SetName(Form("FMDX%c_%02d%02d", *r, 2*m,2*m+1));
347 g->SetTitle(Form("FMDX%c, sectors %d and %d", *r, 2*m,2*m+1));
348 g->SetPoint(nv, x0, y0);
349 g->SetFillColor((*rs == 'I' ?
350 (m % 2 == 0 ? 18 : 17) :
351 (m % 2 == 0 ? 20 : 23)));
352 g->SetFillStyle(3001);
362 //____________________________________________________________________
364 AliFMDPattern::Begin(Int_t event)
366 // Called at the begining of an event.
370 // EVENT The event number
374 const char* which[] = { "Continue", "Start", "Pause", "Redisplay", 0 };
377 AliFMDGeometry* geom = AliFMDGeometry::Instance();
378 // AliFMDDetector* det;
379 if ((/* det = */ geom->GetDetector(1))) {
381 fFMD1Pad = new TPad("FMD1", "FMD1", 0.0, 0.50, 0.5, 1.0, 0, 0);
384 fFMD1.Begin(-1, fInnerMax, fInners, fOuters);
386 if ((/* det = */ geom->GetDetector(2))) {
388 fFMD2Pad = new TPad("FMD2", "FMD2", 0.5, 0.50, 1.0, 1.0, 0, 0);
391 fFMD2.Begin(-1, fOuterMax, fInners, fOuters);
393 if ((/* det = */ geom->GetDetector(3))) {
395 fFMD3Pad = new TPad("FMD3", "FMD3", 0.0, 0.0, .5, .5, 0, 0);
398 fFMD3.Begin(-1, fOuterMax, fInners, fOuters);
401 fSummary = new TPad("display", "Display", 0.5, 0.0, 1.0, 0.5, 0, 0);
411 fEvent.SetTitle(Form("Event # %6d", event));
419 return AliFMDInput::Begin(event);
422 //____________________________________________________________________
424 AliFMDPattern::Redisplay()
426 // Redraw the displayu
430 AliFMDDisplay::Redisplay();
433 //____________________________________________________________________
435 AliFMDPattern::AtEnd()
437 // Called at the end of an event.
443 fFMD1Pad->Modified();
444 fFMD1Sum.SetTitle(Form("# hits in FMD1: %5d", fFMD1.Total()));
445 total += fFMD1.Total();
448 fFMD2Pad->Modified();
449 fFMD2Sum.SetTitle(Form("# hits in FMD2: %5d", fFMD2.Total()));
450 total += fFMD2.Total();
453 fFMD3Pad->Modified();
454 fFMD3Sum.SetTitle(Form("# hits in FMD3: %5d", fFMD3.Total()));
455 total += fFMD3.Total();
457 fTotal.SetTitle(Form("Total: %5d/51200 (%3d%%)",
458 total, Int_t(100. / 51200 * total)));
459 fSummary->Modified();
465 //____________________________________________________________________
467 AliFMDPattern::ProcessHit(AliFMDHit* hit, TParticle*)
473 // HIT The hit to process.
475 // The TParticle argument is never used.
476 static const Float_t rMin = fgkEdepRange.fLow;
477 static const Float_t rMax = fgkEdepRange.fHigh;
479 if (!hit) { AliError("No hit"); return kFALSE; }
480 // if (!p) { AliError("No track"); return kFALSE; }
481 Float_t edep = hit->Edep();
483 if (fHits) fHits->Add(hit);
484 if (fSpec) fSpec->Fill(edep);
485 if (InsideCut(edep, rMin, rMax) && fSpecCut) fSpecCut->Fill(edep);
487 switch (hit->Detector()) {
488 case 1: fFMD1.AddMarker(hit->X(), hit->Y(), hit->Edep(), rMax); break;
489 case 2: fFMD2.AddMarker(hit->X(), hit->Y(), hit->Edep(), rMax); break;
490 case 3: fFMD3.AddMarker(hit->X(), hit->Y(), hit->Edep(), rMax); break;
496 //____________________________________________________________________
498 AliFMDPattern::AddMarker(UShort_t det, Char_t rng,
499 UShort_t sec, UShort_t str,
500 TObject*, Float_t s, Float_t /*min*/, Float_t max)
502 // Add a marker to the display
508 // o Object to refer to
510 // max Maximum of signal
512 AliFMDPatternDetector* d = 0;
514 case 1: d = &fFMD1; break;
515 case 2: d = &fFMD2; break;
516 case 3: d = &fFMD3; break;
519 AliFMDGeometry* geom = AliFMDGeometry::Instance();
521 geom->Detector2XYZ(det, rng, sec, str, x, y, z);
522 // Make code-checker shut the f**k up
523 TRandom* rand = gRandom;
525 AliFMDRing* r = geom->GetRing(rng);
526 Double_t t = .9 * r->GetTheta() / 2;
527 Double_t a = rand->Uniform(-t,t) * TMath::Pi() / 180;
528 Double_t x1 = x * TMath::Cos(a) - y * TMath::Sin(a);
529 Double_t y1 = x * TMath::Sin(a) + y * TMath::Cos(a);
533 d->AddMarker(x, y, s, max);
536 //____________________________________________________________________