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 AliFMDDisplay.cxx
17 @author Christian Holm Christensen <cholm@nbi.dk>
18 @date Mon Mar 27 12:39:09 2006
19 @brief FMD 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
33 #include <TApplication.h>
36 #include <TGeoManager.h>
38 #include <TMarker3DBox.h>
43 #include <TVirtualX.h>
44 // #include <TArrayF.h>
45 // #include <TParticle.h>
47 #include "AliFMDDisplay.h" // ALIFMDDISPLAY_H
48 #include "AliFMDHit.h" // ALIFMDHIT_H
49 #include "AliFMDDigit.h" // ALIFMDDIGIT_H
50 #include "AliFMDRecPoint.h" // ALIFMDRECPOINT_H
51 #include "AliFMDGeometry.h" // ALIFMDGEOMETRY_H
52 #include "AliFMDParameters.h" // ALIFMDPARAMETERS_H
53 #include <AliESDFMD.h> // ALIESDFMD_H
54 // #include <AliLog.h>
55 #include "AliFMDDebug.h" // Better debug macros
57 //____________________________________________________________________
58 ClassImp(AliFMDDisplay)
60 ; // This is here to keep Emacs for indenting the next line
63 //____________________________________________________________________
64 AliFMDDisplay* AliFMDDisplay::fgInstance = 0;
66 //____________________________________________________________________
68 AliFMDDisplay::Instance()
70 // Return static instance
71 // If the instance does not exist
76 //____________________________________________________________________
77 AliFMDDisplay::~AliFMDDisplay()
90 if (fPad) delete fPad;
92 if (fSlider) delete fSlider;
93 if (fCanvas) delete fCanvas;
96 //____________________________________________________________________
97 AliFMDDisplay::AliFMDDisplay(Bool_t onlyFMD, const char* gAliceFile)
98 : AliFMDInput(gAliceFile),
123 // Constructor of an FMD display object.
127 if (fgInstance) delete fgInstance;
129 SetMultiplicityCut();
133 //____________________________________________________________________
135 AliFMDDisplay::MakeCanvas(const char** which)
139 // which Which button to put up.
140 gStyle->SetPalette(1);
143 fCanvas = new TCanvas("display", "Display", w, Int_t(w / (1-y1)));
144 fCanvas->SetFillColor(1);
145 fCanvas->ToggleEventStatus();
147 fPad = new TPad("view", "3DView", 0.0, y1, 1.0, 1.0, 1, 0, 0);
152 if (TESTBIT(fTreeMask, kESD) ||
153 TESTBIT(fTreeMask, kDigits) ||
154 TESTBIT(fTreeMask, kRaw)) {
156 fSlider = new TSlider("multCut", "Multiplicity cut", 0, 0, 1, yb);
157 fSlider->SetMethod("AliFMDDisplay::Instance()->ChangeCut()");
159 fSlider->SetMinimum(TESTBIT(fTreeMask, kESD) ? fMultCut * 10 :
160 fPedestalFactor * 10);
162 const char** p = which;
167 AliInfo(Form("Got %d buttons", n));
171 while ((m = *(p++))) {
173 AliInfo(Form("Adding button %s", m));
174 TButton* b = new TButton(m, Form("AliFMDDisplay::Instance()->%s()", m),
175 x0, yb, x0 + dx, y1);
183 //____________________________________________________________________
185 AliFMDDisplay::ShowOnlyFMD()
190 if (!fGeoManager) return;
191 static bool once = false;
194 AliInfo("Will only show the FMD");
195 TGeoVolume* top = gGeoManager->GetTopVolume();
196 top->InvisibleAll(kTRUE);
197 TGeoIterator next(top);
200 Bool_t hasFMD1 = kFALSE;
201 Bool_t hasFMD2 = kFALSE;
202 Bool_t hasFMD3 = kFALSE;
204 while ((node = static_cast<TGeoNode*>(next()))) {
205 const char* name = node->GetName();
207 if (!(v = node->GetVolume())) continue;
209 if (name[0] == 'F') {
210 if (name[2] == 'M' && (name[3] == 'T' || name[3] == 'B')) {
211 // Virtual Master half-ring volume - top-level
212 Int_t det = node->GetNumber();
214 case 1: hasFMD1 = true; break;
215 case 2: hasFMD2 = true; break;
216 case 3: hasFMD3 = true; break;
221 else if (name[3] == 'V' && (name[2] == 'T' || name[2] == 'B'))
222 toshow.Add(v); // Virtual Half-ring, bare detectors
223 // else if (name[3] == 'H' && (name[2] == 'F' || name[2] == 'B'))
224 // toshow.Add(v); // Virtual Hybrid container
226 v->SetVisibility(kFALSE);
227 v->SetVisDaughters(kFALSE);
228 v->InvisibleAll(kTRUE);
231 while ((v = static_cast<TGeoVolume*>(i()))) {
232 v->SetVisibility(kTRUE);
233 v->SetVisDaughters(kTRUE);
234 v->InvisibleAll(kFALSE);
239 //____________________________________________________________________
241 AliFMDDisplay::ExecuteEvent(Int_t event, Int_t px, Int_t py)
243 // Execute an event on canvas
245 // event What happened
246 // px, py Pixel coordinates
247 if (px == 0 && py == 0) return;
248 if (!fZoomMode && fPad->GetView()) {
249 fPad->GetView()->ExecuteRotateView(event, px, py);
252 fPad->SetCursor(kCross);
255 fPad->TAttLine::Modify();
256 fX0 = fPad->AbsPixeltoX(px);
257 fY0 = fPad->AbsPixeltoY(py);
258 fXPixel = fOldXPixel = px;
259 fYPixel = fOldYPixel = py;
264 gVirtualX->DrawBox(fXPixel, fYPixel, fOldXPixel, fOldYPixel,
269 gVirtualX->DrawBox(fXPixel, fYPixel, fOldXPixel, fOldYPixel,
273 fPad->GetCanvas()->FeedbackMode(kFALSE);
274 if (px == fXPixel || py == fYPixel) return;
275 fX1 = fPad->AbsPixeltoX(px);
276 fY1 = fPad->AbsPixeltoY(py);
277 if (fX1 < fX0) std::swap(fX0, fX1);
278 if (fY1 < fY0) std::swap(fY0, fY1);
279 fPad->Range(fX0, fY0, fX1, fY1);
285 //____________________________________________________________________
287 AliFMDDisplay::DistancetoPrimitive(Int_t px, Int_t)
289 // Calculate the distance from point to
290 // something in the canvas.
291 // Depends on the zoom mode.
292 fPad->SetCursor(kCross);
293 Float_t xmin = fPad->GetX1();
294 Float_t xmax = fPad->GetX2();
295 Float_t dx = .02 * (xmax - xmin);
296 Float_t x = fPad->AbsPixeltoX(px);
297 if (x < xmin + dx || x > xmax - dx) return 9999;
298 return (fZoomMode ? 0 : 7);
300 //____________________________________________________________________
302 AliFMDDisplay::Init()
304 // Initialize. GEt transforms and such,
305 // so that we can draw thins properly
306 // Returns true on success
307 if (!AliFMDInput::Init()) return kFALSE;
308 AliFMDGeometry* geom = AliFMDGeometry::Instance();
310 geom->InitTransformations();
311 if (TESTBIT(fTreeMask, kDigits) || TESTBIT(fTreeMask, kRaw))
312 AliFMDParameters::Instance()->Init();
314 fMarkers = new TObjArray;
315 fHits = new TObjArray;
316 fMarkers->SetOwner(kTRUE);
317 fHits->SetOwner(kFALSE);
321 //____________________________________________________________________
323 AliFMDDisplay::MakeAux()
325 // MAke the aux canvas
326 // This is used to display spectra
328 if ((TESTBIT(fTreeMask, kESD) ||
329 TESTBIT(fTreeMask, kDigits) ||
330 TESTBIT(fTreeMask, kRaw))) {
332 fAux = new TCanvas("aux", "Aux");
334 if (TESTBIT(fTreeMask, kESD))
335 fSpec = new TH1D("spec", "Mult spectra", 500, 0, 10);
337 fSpec = new TH1D("spec", "Adc spectra", 1024, -.5, 1023.5);
338 fSpecCut = static_cast<TH1*>(fSpec->Clone("specCut"));
339 fSpec->SetFillColor(2);
340 fSpec->SetFillStyle(3001);
341 fSpecCut->SetFillColor(4);
342 fSpecCut->SetFillStyle(3001);
351 //____________________________________________________________________
353 AliFMDDisplay::DrawAux()
355 // Draw in the Aux the canvas
356 // For example draw the spectra
362 fSpecCut->Draw("same");
368 //____________________________________________________________________
370 AliFMDDisplay::Begin(Int_t event)
372 // Begin of event. Make canvas is not already done
374 // event The event number
376 const char* m[] = { "Continue", "Zoom", "Pick", "Redisplay", 0 };
381 // AliInfo("Clearing canvas");
384 Warning("End", "No geometry manager");
387 AliInfo("Drawing geometry");
389 fGeoManager->GetTopVolume()->Draw();
390 if (fOnlyFMD) ShowOnlyFMD();
391 AliInfo("Adjusting view");
393 if (fPad->GetView()) {
394 fPad->GetView()->SetView(-200, -40, 80, irep);
395 fPad->GetView()->Zoom();
399 return AliFMDInput::Begin(event);
402 //____________________________________________________________________
404 AliFMDDisplay::AtEnd()
406 // Called at of the event.
417 //____________________________________________________________________
419 AliFMDDisplay::Idle()
422 // Sends the ROOT loop into the idle loop,
423 // so that we can go on.
426 gApplication->StartIdleing();
427 gSystem->InnerLoop();
428 gApplication->StopIdleing();
430 AliInfo("After idle loop");
431 if (fMarkers) fMarkers->Delete();
432 if (fHits) fHits->Clear();
433 AliInfo("After clearing caches");
436 //____________________________________________________________________
440 // End of event. Draw everything
443 return AliFMDInput::End();
446 //____________________________________________________________________
448 AliFMDDisplay::LookupColor(Float_t x, Float_t max) const
451 // Get a colour from the current palette depending
452 // on the ratio x/max
453 Int_t idx = Int_t(x / max * gStyle->GetNumberOfColors());
454 return gStyle->GetColorPalette(idx);
457 //____________________________________________________________________
459 AliFMDDisplay::ChangeCut()
461 // Change the cut on the slider.
462 // The factor depends on what is
463 // drawn in the AUX canvas
464 fMultCut = fSlider->GetMinimum() * 10;
465 fPedestalFactor = fSlider->GetMinimum() * 10;
466 AliInfo(Form("Multiplicity cut: %7.5f, Pedestal factor: %7.4f (%6.5f)",
467 fMultCut, fPedestalFactor, fSlider->GetMinimum()));
471 //____________________________________________________________________
473 AliFMDDisplay::Redisplay()
476 // Redraw markers, hits,
478 if (fMarkers) fMarkers->Delete();
479 if (fHits) fHits->Clear();
480 if (fSpec) fSpec->Reset();
481 if (fSpecCut) fSpecCut->Reset();
486 //____________________________________________________________________
488 AliFMDDisplay::AddMarker(UShort_t det, Char_t rng, UShort_t sec, UShort_t str,
489 TObject* o, Float_t s, Float_t max)
491 // Add a marker to the display
497 // o Object to refer to
499 // max Maximum of signal
501 AliFMDGeometry* geom = AliFMDGeometry::Instance();
503 geom->Detector2XYZ(det, rng, sec, str, x, y, z);
505 Float_t zsize = s / max * 10;
506 Float_t r = TMath::Sqrt(x * x + y * y);
507 Float_t theta = TMath::ATan2(r, z);
508 Float_t phi = TMath::ATan2(y, x);
509 Float_t rz = z + (z < 0 ? 1 : -1) * zsize;
510 TMarker3DBox* marker = new TMarker3DBox(x,y,rz,size,size,zsize,theta,phi);
511 if (o) marker->SetRefObject(o);
512 marker->SetLineColor(LookupColor(s, max));
513 fMarkers->Add(marker);
517 //____________________________________________________________________
519 AliFMDDisplay::ProcessHit(AliFMDHit* hit, TParticle* /* p */)
524 if (!hit) { AliError("No hit"); return kFALSE; }
525 // if (!p) { AliError("No track"); return kFALSE; }
527 if (fHits) fHits->Add(hit);
529 Float_t zsize = TMath::Sqrt(hit->Edep() * 20);
530 Float_t z = hit->Z() + (hit->Z() < 0 ? 1 : -1) * zsize;
531 Float_t pt = TMath::Sqrt(hit->Py()*hit->Py()+hit->Px()*hit->Px());
532 Float_t theta = TMath::ATan2(pt, hit->Pz());
533 Float_t phi = TMath::ATan2(hit->Py(), hit->Px());
534 TMarker3DBox* marker = new TMarker3DBox(hit->X(), hit->Y(), z,
535 size, size, zsize, theta, phi);
536 marker->SetLineColor(LookupColor(hit->Edep(), 1));
537 marker->SetRefObject(hit);
538 fMarkers->Add(marker);
542 //____________________________________________________________________
544 AliFMDDisplay::ProcessDigit(AliFMDDigit* digit)
548 // digit Digit information
549 if (!digit) { AliError("No digit"); return kFALSE; }
551 AliFMDParameters* parm = AliFMDParameters::Instance();
552 UShort_t det = digit->Detector();
553 Char_t ring = digit->Ring();
554 UShort_t sec = digit->Sector();
555 UShort_t str = digit->Strip();
556 Double_t ped = parm->GetPedestal(det,ring, sec, str);
557 Double_t pedW = parm->GetPedestalWidth(det,ring, sec, str);
558 Double_t threshold = ped + fPedestalFactor * pedW;
559 Float_t counts = digit->Counts();
560 AliFMDDebug(10, ("FMD%d%c[%2d,%3d] ADC: %d > %d (=%4.2f+%4.2f*%4.2f)",
561 digit->Detector(), digit->Ring(), digit->Sector(),
562 digit->Strip(), Int_t(counts), Int_t(threshold),
563 ped, fPedestalFactor, pedW));
564 if (fSpec) fSpec->Fill(counts);
565 if (counts < threshold) return kTRUE;
566 if (fHits) fHits->Add(digit);
567 if (fSpecCut) fSpecCut->Fill(counts);
569 AddMarker(det, ring, sec, str, digit, counts, 1024);
573 //____________________________________________________________________
575 AliFMDDisplay::ProcessRaw(AliFMDDigit* digit)
579 // digit Digit information
580 return ProcessDigit(digit);
583 //____________________________________________________________________
585 AliFMDDisplay::ProcessRecPoint(AliFMDRecPoint* recpoint)
587 // Process reconstructed point
589 // recpoint Reconstructed multiplicity/energy
590 if (!recpoint) { AliError("No recpoint"); return kFALSE; }
591 if (recpoint->Particles() < fMultCut) return kTRUE;
592 if (fHits) fHits->Add(recpoint);
593 AddMarker(recpoint->Detector(), recpoint->Ring(), recpoint->Sector(),
594 recpoint->Strip(), recpoint, recpoint->Particles(), 20);
598 //____________________________________________________________________
600 AliFMDDisplay::ProcessESD(UShort_t det, Char_t rng, UShort_t sec, UShort_t str,
601 Float_t, Float_t mult)
603 // Process data from ESD
605 // det,rng,sec,str Detector coordinates.
606 // mult Multiplicity.
607 Double_t cmult = mult;
608 if (fSpec) fSpec->Fill(cmult);
609 if (cmult < fMultCut || cmult == AliESDFMD::kInvalidMult) return kTRUE;
610 AddMarker(det,rng,sec,str, 0, cmult, 20);
611 if (fSpecCut) fSpecCut->Fill(cmult);
615 //____________________________________________________________________