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>
41 #include <TSliderBox.h>
44 #include <TVirtualX.h>
45 #include <TVirtualViewer3D.h>
47 // #include <TArrayF.h>
48 // #include <TParticle.h>
50 #include "AliFMDDisplay.h" // ALIFMDDISPLAY_H
51 #include "AliFMDHit.h" // ALIFMDHIT_H
52 #include "AliFMDDigit.h" // ALIFMDDIGIT_H
53 #include "AliFMDSDigit.h" // ALIFMDSDIGIT_H
54 #include "AliFMDRecPoint.h" // ALIFMDRECPOINT_H
55 #include "AliFMDGeometry.h" // ALIFMDGEOMETRY_H
56 #include "AliFMDParameters.h" // ALIFMDPARAMETERS_H
57 #include <AliESDFMD.h> // ALIESDFMD_H
58 // #include <AliLog.h>
59 #include "AliFMDDebug.h" // Better debug macros
61 //____________________________________________________________________
62 ClassImp(AliFMDDisplay)
64 ; // This is here to keep Emacs for indenting the next line
67 //____________________________________________________________________
68 AliFMDDisplay* AliFMDDisplay::fgInstance = 0;
70 //____________________________________________________________________
71 const AliFMDDisplay::Range_t AliFMDDisplay::fgkEdepRange = { 100, 0., 2. };
72 const AliFMDDisplay::Range_t AliFMDDisplay::fgkAdcRange = { 1024, 0., 1023. };
73 const AliFMDDisplay::Range_t AliFMDDisplay::fgkMultRange = { 500, 0., 20. };
76 //____________________________________________________________________
78 AliFMDDisplay::Instance()
80 // Return static instance
81 // If the instance does not exist
86 //____________________________________________________________________
87 AliFMDDisplay::~AliFMDDisplay()
100 if (fPad) delete fPad;
102 if (fSlider) delete fSlider;
103 if (fCanvas) delete fCanvas;
106 //____________________________________________________________________
107 AliFMDDisplay::AliFMDDisplay(Bool_t onlyFMD, const char* gAliceFile)
108 : AliFMDInput(gAliceFile),
133 // Constructor of an FMD display object.
136 SetName("AliFMDDisplay");
137 SetTitle("3D Display of various kinds of FMD data");
139 if (fgInstance) delete fgInstance;
143 //____________________________________________________________________
145 AliFMDDisplay::MakeCanvas(const char** which)
149 // which Which button to put up.
150 gStyle->SetPalette(1);
151 // gStyle->SetCanvasPreferGL(kTRUE);
154 fCanvas = new TCanvas(Form("gl%s", GetName()),
155 Form("%s - Display", GetTitle()),
156 w, Int_t(w / (1-y1)));
157 fCanvas->SetFillColor(1);
158 fCanvas->ToggleEventStatus();
160 fPad = new TPad("glview", "3DView", 0.0, y1, 1.0, 1.0, 1, 0, 0);
163 const char** p = which;
168 AliInfo(Form("Got %d buttons", n));
174 if (TESTBIT(fTreeMask, kDigits) ||
175 TESTBIT(fTreeMask, kRaw)) {
178 fFactor = new TSlider("pedFactor", "Pedestal Factor", xb+.01, 0, 1, yb);
179 fFactor->SetMethod("AliFMDDisplay::Instance()->ChangeFactor()");
180 fFactor->SetRange(3./10, 1);
183 static_cast<TSliderBox*>(fFactor->GetListOfPrimitives()->
184 FindObject("TSliderBox"));
186 sbox->SetToolTipText("Adjust the noise suppression factor by moving "
190 if (TESTBIT(fTreeMask, kHits) ||
191 TESTBIT(fTreeMask, kESD) ||
192 TESTBIT(fTreeMask, kDigits) ||
193 TESTBIT(fTreeMask, kSDigits) ||
194 TESTBIT(fTreeMask, kRaw)) {
196 fSlider = new TSlider("genCut", "Multiplicity cut", 0, 0, xb, yb);
197 fSlider->SetMethod("AliFMDDisplay::Instance()->ChangeCut()");
198 fSlider->SetRange(0,1);
201 static_cast<TSliderBox*>(fSlider->GetListOfPrimitives()->
202 FindObject("TSliderBox"));
204 sbox->SetToolTipText("Adjust lower and upper limit on data signal");
210 while ((m = *(p++))) {
212 AliInfo(Form("Adding button %s", m));
213 TButton* b = new TButton(m, Form("AliFMDDisplay::Instance()->%s()", m),
214 x0, yb, TMath::Min(x0 + dx,.999F), y1);
222 //____________________________________________________________________
224 AliFMDDisplay::ShowOnlyFMD()
229 if (!fGeoManager) return;
230 static bool once = false;
233 AliInfo("Will only show the FMD");
234 TGeoVolume* top = gGeoManager->GetTopVolume();
235 top->InvisibleAll(kTRUE);
236 TGeoIterator next(top);
239 Bool_t hasFMD1 = kFALSE;
240 Bool_t hasFMD2 = kFALSE;
241 Bool_t hasFMD3 = kFALSE;
242 AliInfo("Getting material FMD_Si$");
243 TGeoMaterial* si = gGeoManager->GetMaterial("FMD_Si$"); // kRed
244 AliInfo("Getting material FMD_Carbon$");
245 TGeoMaterial* c = gGeoManager->GetMaterial("FMD_Carbon$"); // kGray
246 AliInfo("Getting material FMD_Aluminum$");
247 TGeoMaterial* al = gGeoManager->GetMaterial("FMD_Aluminum$");// kGray-2
248 AliInfo("Getting material FMD_Copper$");
249 TGeoMaterial* cu = gGeoManager->GetMaterial("FMD_Copper$"); // kGreen-2
250 AliInfo("Getting material FMD_PCB$");
251 TGeoMaterial* pcb = gGeoManager->GetMaterial("FMD_PCB$"); // kGreen+2
252 AliInfo("Getting material FMD_PCB$");
253 TGeoMaterial* chip = gGeoManager->GetMaterial("FMD_Si Chip$");// kGreen+2
255 while ((node = static_cast<TGeoNode*>(next()))) {
256 const char* name = node->GetName();
258 if (!(v = node->GetVolume())) continue;
260 if (name[0] == 'F') {
261 TGeoMaterial* m = (v->IsAssembly() ? 0 : v->GetMaterial());
263 if (m == si) col = kRed;
264 else if (m == c) col = kGray;
265 else if (m == al) col = kYellow+4;
266 else if (m == cu) col = kRed+6;
267 else if (m == pcb) col = kGreen+2;
268 else if (m == chip) col = kGreen+4;
270 v->SetLineColor(col);
271 v->SetFillColor(col);
273 if (name[2] == 'M' && (name[3] == 'T' || name[3] == 'B')) {
274 // Virtual Master half-ring volume - top-level
275 Int_t det = node->GetNumber();
277 case 1: hasFMD1 = true; break;
278 case 2: hasFMD2 = true; break;
279 case 3: hasFMD3 = true; break;
284 else if (name[3] == 'V' && (name[2] == 'T' || name[2] == 'B'))
285 toshow.Add(v); // Virtual Half-ring, bare detectors
286 else if (name[3] == 'H' && (name[2] == 'F' || name[2] == 'B'))
287 toshow.Add(v); // Virtual Hybrid container
288 else if (name[2] == 'S' && name[3] == 'U')
289 toshow.Add(v); // Virtual support structre
290 // else if (name[3] == 'H' && (name[2] == 'F' || name[2] == 'B'))
291 // toshow.Add(v); // Virtual Hybrid container
293 v->SetVisibility(kFALSE);
294 v->SetVisDaughters(kFALSE);
295 v->InvisibleAll(kTRUE);
298 while ((v = static_cast<TGeoVolume*>(i()))) {
299 if (!v->IsAssembly())
300 v->SetVisibility(kTRUE);
301 v->InvisibleAll(kFALSE);
302 v->SetVisDaughters(kTRUE);
308 //____________________________________________________________________
310 AliFMDDisplay::ExecuteEvent(Int_t event, Int_t px, Int_t py)
312 // Execute an event on canvas
314 // event What happened
315 // px, py Pixel coordinates
316 if (px == 0 && py == 0) return;
317 if (!fZoomMode && fPad->GetView()) {
318 fPad->GetView()->ExecuteRotateView(event, px, py);
321 fPad->SetCursor(kCross);
324 fPad->TAttLine::Modify();
325 fX0 = fPad->AbsPixeltoX(px);
326 fY0 = fPad->AbsPixeltoY(py);
327 fXPixel = fOldXPixel = px;
328 fYPixel = fOldYPixel = py;
333 gVirtualX->DrawBox(fXPixel, fYPixel, fOldXPixel, fOldYPixel,
338 gVirtualX->DrawBox(fXPixel, fYPixel, fOldXPixel, fOldYPixel,
342 fPad->GetCanvas()->FeedbackMode(kFALSE);
343 if (px == fXPixel || py == fYPixel) return;
344 fX1 = fPad->AbsPixeltoX(px);
345 fY1 = fPad->AbsPixeltoY(py);
346 if (fX1 < fX0) std::swap(fX0, fX1);
347 if (fY1 < fY0) std::swap(fY0, fY1);
348 fPad->Range(fX0, fY0, fX1, fY1);
354 //____________________________________________________________________
356 AliFMDDisplay::DistancetoPrimitive(Int_t px, Int_t)
358 // Calculate the distance from point to
359 // something in the canvas.
360 // Depends on the zoom mode.
361 fPad->SetCursor(kCross);
362 Float_t xmin = fPad->GetX1();
363 Float_t xmax = fPad->GetX2();
364 Float_t dx = .02 * (xmax - xmin);
365 Float_t x = fPad->AbsPixeltoX(px);
366 if (x < xmin + dx || x > xmax - dx) return 9999;
367 return (fZoomMode ? 0 : 7);
369 //____________________________________________________________________
371 AliFMDDisplay::Init()
373 // Initialize. GEt transforms and such,
374 // so that we can draw thins properly
375 // Returns true on success
376 if (!AliFMDInput::Init()) return kFALSE;
377 AliFMDGeometry* geom = AliFMDGeometry::Instance();
379 geom->InitTransformations();
380 if (TESTBIT(fTreeMask, kDigits) || TESTBIT(fTreeMask, kRaw))
381 AliFMDParameters::Instance()->Init();
383 fMarkers = new TObjArray;
384 fHits = new TObjArray;
385 fMarkers->SetOwner(kTRUE);
386 fHits->SetOwner(kFALSE);
390 //____________________________________________________________________
392 AliFMDDisplay::MakeAux()
394 // MAke the aux canvas
395 // This is used to display spectra
397 const Range_t* range = 0;
398 if (TESTBIT(fTreeMask, kESD)) range = &fgkMultRange;
399 else if (TESTBIT(fTreeMask, kDigits)) range = &fgkAdcRange;
400 else if (TESTBIT(fTreeMask, kSDigits)) range = &fgkAdcRange;
401 else if (TESTBIT(fTreeMask, kRaw)) range = &fgkAdcRange;
402 else if (TESTBIT(fTreeMask, kHits)) range = &fgkEdepRange;
406 fAux = new TCanvas("aux", "Aux");
408 Float_t dBin = (range->fHigh - range->fLow) / range->fNbins;
409 fSpec = new TH1D("spec", "Spectra", range->fNbins,
410 range->fLow-dBin/2, range->fHigh-dBin/2);
411 fSpecCut = static_cast<TH1*>(fSpec->Clone("specCut"));
412 fSpec->SetFillColor(2);
413 fSpec->SetFillStyle(3001);
414 fSpecCut->SetFillColor(4);
415 fSpecCut->SetFillStyle(3001);
423 //____________________________________________________________________
425 AliFMDDisplay::DrawAux()
427 // Draw in the Aux the canvas
428 // For example draw the spectra
434 fSpecCut->Draw("same");
440 //____________________________________________________________________
442 AliFMDDisplay::Begin(Int_t event)
444 // Begin of event. Make canvas is not already done
446 // event The event number
448 const char* m[] = { "Continue",
460 // AliInfo("Clearing canvas");
463 Warning("End", "No geometry manager");
466 AliInfo("Drawing geometry");
468 fGeoManager->GetTopVolume()->Draw();
469 if (fOnlyFMD) ShowOnlyFMD();
470 AliInfo("Adjusting view");
472 if (fPad->GetView()) {
473 fPad->GetView()->SetView(-200, -40, 80, irep);
474 fPad->GetView()->Zoom();
478 return AliFMDInput::Begin(event);
481 //____________________________________________________________________
483 AliFMDDisplay::AtEnd()
485 // Called at of the event.
496 //____________________________________________________________________
498 AliFMDDisplay::Idle()
501 // Sends the ROOT loop into the idle loop,
502 // so that we can go on.
505 gApplication->StartIdleing();
506 gSystem->InnerLoop();
507 gApplication->StopIdleing();
509 AliInfo("After idle loop");
510 if (fMarkers) fMarkers->Delete();
511 if (fHits) fHits->Clear();
512 AliInfo("After clearing caches");
515 //____________________________________________________________________
519 // End of event. Draw everything
522 if (fReturn) return kFALSE;
523 return AliFMDInput::End();
526 //____________________________________________________________________
528 AliFMDDisplay::LookupColor(Float_t x, Float_t min, Float_t max) const
531 // Get a colour from the current palette depending
532 // on the ratio x/max
533 Float_t range = (max-min);
534 Float_t l = fSlider->GetMinimum();
535 Float_t h = fSlider->GetMaximum();
536 if (l == h) { l = 0; h = 1; }
537 Float_t cmin = range * l;
538 Float_t cmax = range * h;
539 Float_t crange = (cmax-cmin);
540 Int_t idx = Int_t((x-cmin) / crange * gStyle->GetNumberOfColors());
541 return gStyle->GetColorPalette(idx);
544 //____________________________________________________________________
546 AliFMDDisplay::SetCut(Float_t l, Float_t h)
548 // Change the cut on the slider.
549 fSlider->SetMinimum(l);
550 fSlider->SetMaximum(h);
554 //____________________________________________________________________
556 AliFMDDisplay::ChangeCut()
558 // Change the cut on the slider.
559 // The factor depends on what is
560 // drawn in the AUX canvas
561 AliInfo(Form("Range is now %7.5f - %7.5f", fSlider->GetMinimum(),
562 fSlider->GetMaximum()));
563 if ((TESTBIT(fTreeMask, kESD) ||
564 TESTBIT(fTreeMask, kDigits) ||
565 TESTBIT(fTreeMask, kSDigits) ||
566 TESTBIT(fTreeMask, kRaw))) {
567 Float_t l = fSlider->GetMinimum();
568 Float_t h = fSlider->GetMaximum();
571 AliInfo(Form("ADC range is now %4d - %4d", int(l), int(h)));
575 //____________________________________________________________________
577 AliFMDDisplay::ChangeFactor()
579 // Change the cut on the slider.
580 // The factor depends on what is
581 // drawn in the AUX canvas
582 AliInfo(Form("Noise factor is now %4.1f, pedestal factor %3.1f",
583 10*fFactor->GetMinimum(),fFactor->GetMaximum()));
587 //____________________________________________________________________
589 AliFMDDisplay::Redisplay()
592 // Redraw markers, hits,
594 if (fMarkers) fMarkers->Delete();
595 if (fHits) fHits->Clear();
596 if (fSpec) fSpec->Reset();
597 if (fSpecCut) fSpecCut->Reset();
601 //____________________________________________________________________
603 AliFMDDisplay::Break()
606 // Redraw markers, hits,
608 if (fMarkers) fMarkers->Delete();
609 if (fHits) fHits->Clear();
610 if (fSpec) fSpec->Reset();
611 if (fSpecCut) fSpecCut->Reset();
615 //____________________________________________________________________
617 AliFMDDisplay::Render()
620 TVirtualViewer3D* viewer = fPad->GetViewer3D("ogl");
624 //____________________________________________________________________
626 AliFMDDisplay::AddMarker(Float_t x, Float_t y, Float_t z,
627 TObject* o, Float_t s, Float_t min, Float_t max)
629 // Add a marker to the display
635 // o Object to refer to
637 // max Maximum of signal
640 Float_t zsize = (s - min) / (max-min) * 10;
641 Float_t r = TMath::Sqrt(x * x + y * y);
642 Float_t theta = TMath::ATan2(r, z);
643 Float_t phi = TMath::ATan2(y, x);
644 Float_t rz = z + (z < 0 ? 1 : -1) * zsize;
645 TMarker3DBox* marker = new TMarker3DBox(x,y,rz,size,size,zsize,theta,phi);
646 if (o) marker->SetRefObject(o);
647 marker->SetLineColor(LookupColor(s, min, max));
648 fMarkers->Add(marker);
650 //____________________________________________________________________
652 AliFMDDisplay::AddMarker(UShort_t det, Char_t rng, UShort_t sec, UShort_t str,
653 TObject* o, Float_t s, Float_t min, Float_t max)
655 // Add a marker to the display
661 // o Object to refer to
663 // max Maximum of signal
665 AliFMDGeometry* geom = AliFMDGeometry::Instance();
667 geom->Detector2XYZ(det, rng, sec, str, x, y, z);
668 AddMarker(x,y,z,o,s,min,max);
671 //____________________________________________________________________
673 AliFMDDisplay::InsideCut(Float_t val, const Float_t& min,
674 const Float_t& max) const
676 Float_t r = max - min;
677 Float_t l = fSlider->GetMinimum();
678 Float_t h = fSlider->GetMaximum();
679 if (l == h) { l = 0; h = 1; }
680 if (val < r * l + min || val > r * h + min) return kFALSE;
685 //____________________________________________________________________
687 AliFMDDisplay::ProcessHit(AliFMDHit* hit, TParticle* /* p */)
692 static const Float_t rMin = fgkEdepRange.fLow;
693 static const Float_t rMax = fgkEdepRange.fHigh;
695 if (!hit) { AliError("No hit"); return kFALSE; }
696 // if (!p) { AliError("No track"); return kFALSE; }
697 Float_t edep = hit->Edep();
699 if (fHits) fHits->Add(hit);
700 if (fSpec) fSpec->Fill(edep);
701 if (!InsideCut(edep, rMin, rMax)) return kTRUE;
702 if (fSpecCut) fSpecCut->Fill(edep);
704 AddMarker(hit->X(), hit->Y(), hit->Z(), hit, edep, rMin, rMax);
708 //____________________________________________________________________
710 AliFMDDisplay::ProcessDigit(AliFMDDigit* digit)
714 // digit Digit information
715 static const Float_t rMin = fgkAdcRange.fLow;
716 static const Float_t rMax = fgkAdcRange.fHigh;
718 if (!digit) { AliError("No digit"); return kFALSE; }
720 AliFMDParameters* parm = AliFMDParameters::Instance();
721 UShort_t det = digit->Detector();
722 Char_t ring = digit->Ring();
723 UShort_t sec = digit->Sector();
724 UShort_t str = digit->Strip();
725 Double_t ped = parm->GetPedestal(det,ring, sec, str);
726 Double_t pedW = parm->GetPedestalWidth(det,ring, sec, str);
727 Double_t threshold = (ped * fFactor->GetMaximum()
728 + pedW * fFactor->GetMinimum());
729 if (threshold > fgkAdcRange.fHigh) threshold = fgkAdcRange.fHigh;
730 Float_t counts = digit->Counts();
732 AliFMDDebug(10, ("FMD%d%c[%02d,%03d] counts %4d threshold %4d",
733 det, ring, sec, str, Int_t(counts), Int_t(threshold)));
734 if (fHits) fHits->Add(digit);
735 if (fSpec) fSpec->Fill(counts);
736 if (!InsideCut(counts-threshold, rMin, rMax)) return kTRUE;
737 if (fSpecCut) fSpecCut->Fill(counts);
740 AddMarker(det, ring, sec, str, digit, counts, rMin, rMax);
744 //____________________________________________________________________
746 AliFMDDisplay::ProcessSDigit(AliFMDSDigit* sdigit)
750 // sdigit Digit information
751 static const Float_t rMin = fgkAdcRange.fLow;
752 static const Float_t rMax = fgkAdcRange.fHigh;
754 if (!sdigit) { AliError("No sdigit"); return kFALSE; }
756 UShort_t det = sdigit->Detector();
757 Char_t ring = sdigit->Ring();
758 UShort_t sec = sdigit->Sector();
759 UShort_t str = sdigit->Strip();
760 Float_t counts = sdigit->Counts();
762 if (fHits) fHits->Add(sdigit);
763 if (fSpec) fSpec->Fill(counts);
764 if (!InsideCut(counts, rMin, rMax)) return kTRUE;
765 if (fSpecCut) fSpecCut->Fill(counts);
767 AddMarker(det, ring, sec, str, sdigit, counts, rMin, rMax);
771 //____________________________________________________________________
773 AliFMDDisplay::ProcessRawDigit(AliFMDDigit* digit)
777 // digit Digit information
778 AliFMDDebug(50, ("Forwarding call of ProcessRaw to ProcessDigit "
779 "for FMD%d&c[%02d,%03d] %d",
780 digit->Detector(), digit->Ring(), digit->Sector(),
781 digit->Strip(), digit->Counts()));
782 return ProcessDigit(digit);
785 //____________________________________________________________________
787 AliFMDDisplay::ProcessRecPoint(AliFMDRecPoint* recpoint)
789 // Process reconstructed point
791 // recpoint Reconstructed multiplicity/energy
792 static const Float_t rMin = fgkMultRange.fLow;
793 static const Float_t rMax = fgkMultRange.fHigh;
796 if (!recpoint) { AliError("No recpoint"); return kFALSE; }
798 if (!InsideCut(recpoint->Particles(), rMin, rMax)) return kTRUE;
800 if (fHits) fHits->Add(recpoint);
801 AddMarker(recpoint->Detector(), recpoint->Ring(), recpoint->Sector(),
802 recpoint->Strip(), recpoint, recpoint->Particles(), rMin, rMax);
806 //____________________________________________________________________
808 AliFMDDisplay::ProcessESD(UShort_t det, Char_t rng, UShort_t sec, UShort_t str,
809 Float_t, Float_t mult)
811 // Process data from ESD
813 // det,rng,sec,str Detector coordinates.
814 // mult Multiplicity.
815 static const Float_t rMin = fgkMultRange.fLow;
816 static const Float_t rMax = fgkMultRange.fHigh;
818 Double_t cmult = mult;
819 if (fSpec) fSpec->Fill(cmult);
820 if (!InsideCut(cmult, rMin, rMax) || cmult == AliESDFMD::kInvalidMult)
823 AddMarker(det,rng,sec,str, 0, cmult, rMin, rMax);
825 if (fSpecCut) fSpecCut->Fill(cmult);
830 //____________________________________________________________________