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 //____________________________________________________________________
72 AliFMDDisplay::Instance()
74 // Return static instance
75 // If the instance does not exist
80 //____________________________________________________________________
81 AliFMDDisplay::~AliFMDDisplay()
94 if (fPad) delete fPad;
96 if (fSlider) delete fSlider;
97 if (fCanvas) delete fCanvas;
100 //____________________________________________________________________
101 AliFMDDisplay::AliFMDDisplay(Bool_t onlyFMD, const char* gAliceFile)
102 : AliFMDInput(gAliceFile),
127 // Constructor of an FMD display object.
130 SetName("AliFMDDisplay");
131 SetTitle("3D Display of various kinds of FMD data");
133 if (fgInstance) delete fgInstance;
137 //____________________________________________________________________
139 AliFMDDisplay::MakeCanvas(const char** which)
143 // which Which button to put up.
144 gStyle->SetPalette(1);
145 // gStyle->SetCanvasPreferGL(kTRUE);
148 fCanvas = new TCanvas(Form("gl%s", GetName()),
149 Form("%s - Display", GetTitle()),
150 w, Int_t(w / (1-y1)));
151 fCanvas->SetFillColor(1);
152 fCanvas->ToggleEventStatus();
154 fPad = new TPad("glview", "3DView", 0.0, y1, 1.0, 1.0, 1, 0, 0);
157 const char** p = which;
162 AliInfo(Form("Got %d buttons", n));
168 if (TESTBIT(fTreeMask, kDigits)) {
171 fFactor = new TSlider("pedFactor", "Pedestal Factor", xb+.01, 0, 1, yb);
172 fFactor->SetMethod("AliFMDDisplay::Instance()->ChangeFactor()");
173 fFactor->SetRange(3./10, 1);
176 static_cast<TSliderBox*>(fFactor->GetListOfPrimitives()->
177 FindObject("TSliderBox"));
179 sbox->SetToolTipText("Adjust the noise suppression factor by moving "
183 if (TESTBIT(fTreeMask, kHits) ||
184 TESTBIT(fTreeMask, kESD) ||
185 TESTBIT(fTreeMask, kDigits) ||
186 TESTBIT(fTreeMask, kSDigits) ||
187 TESTBIT(fTreeMask, kRaw)) {
189 fSlider = new TSlider("genCut", "Multiplicity cut", 0, 0, xb, yb);
190 fSlider->SetMethod("AliFMDDisplay::Instance()->ChangeCut()");
191 fSlider->SetRange(0,1);
194 static_cast<TSliderBox*>(fSlider->GetListOfPrimitives()->
195 FindObject("TSliderBox"));
197 sbox->SetToolTipText("Adjust lower and upper limit on data signal");
203 while ((m = *(p++))) {
205 AliInfo(Form("Adding button %s", m));
206 TButton* b = new TButton(m, Form("AliFMDDisplay::Instance()->%s()", m),
207 x0, yb, TMath::Min(x0 + dx,.999F), y1);
215 //____________________________________________________________________
217 AliFMDDisplay::ShowOnlyFMD()
222 if (!fGeoManager) return;
223 static bool once = false;
226 AliInfo("Will only show the FMD");
227 TGeoVolume* top = gGeoManager->GetTopVolume();
228 top->InvisibleAll(kTRUE);
229 TGeoIterator next(top);
232 Bool_t hasFMD1 = kFALSE;
233 Bool_t hasFMD2 = kFALSE;
234 Bool_t hasFMD3 = kFALSE;
235 AliInfo("Getting material FMD_Si$");
236 TGeoMaterial* si = gGeoManager->GetMaterial("FMD_Si$"); // kRed
237 AliInfo("Getting material FMD_Carbon$");
238 TGeoMaterial* c = gGeoManager->GetMaterial("FMD_Carbon$"); // kGray
239 AliInfo("Getting material FMD_Aluminum$");
240 TGeoMaterial* al = gGeoManager->GetMaterial("FMD_Aluminum$");// kGray-2
241 AliInfo("Getting material FMD_Copper$");
242 TGeoMaterial* cu = gGeoManager->GetMaterial("FMD_Copper$"); // kGreen-2
243 AliInfo("Getting material FMD_PCB$");
244 TGeoMaterial* pcb = gGeoManager->GetMaterial("FMD_PCB$"); // kGreen+2
245 AliInfo("Getting material FMD_PCB$");
246 TGeoMaterial* chip = gGeoManager->GetMaterial("FMD_Si Chip$");// kGreen+2
248 while ((node = static_cast<TGeoNode*>(next()))) {
249 const char* name = node->GetName();
251 if (!(v = node->GetVolume())) continue;
253 if (name[0] == 'F') {
254 TGeoMaterial* m = (v->IsAssembly() ? 0 : v->GetMaterial());
256 if (m == si) col = kRed;
257 else if (m == c) col = kGray;
258 else if (m == al) col = kYellow+4;
259 else if (m == cu) col = kRed+6;
260 else if (m == pcb) col = kGreen+2;
261 else if (m == chip) col = kGreen+4;
263 v->SetLineColor(col);
264 v->SetFillColor(col);
266 if (name[2] == 'M' && (name[3] == 'T' || name[3] == 'B')) {
267 // Virtual Master half-ring volume - top-level
268 Int_t det = node->GetNumber();
270 case 1: hasFMD1 = true; break;
271 case 2: hasFMD2 = true; break;
272 case 3: hasFMD3 = true; break;
277 else if (name[3] == 'V' && (name[2] == 'T' || name[2] == 'B'))
278 toshow.Add(v); // Virtual Half-ring, bare detectors
279 else if (name[3] == 'H' && (name[2] == 'F' || name[2] == 'B'))
280 toshow.Add(v); // Virtual Hybrid container
281 else if (name[2] == 'S' && name[3] == 'U')
282 toshow.Add(v); // Virtual support structre
283 // else if (name[3] == 'H' && (name[2] == 'F' || name[2] == 'B'))
284 // toshow.Add(v); // Virtual Hybrid container
286 v->SetVisibility(kFALSE);
287 v->SetVisDaughters(kFALSE);
288 v->InvisibleAll(kTRUE);
291 while ((v = static_cast<TGeoVolume*>(i()))) {
292 if (!v->IsAssembly())
293 v->SetVisibility(kTRUE);
294 v->InvisibleAll(kFALSE);
295 v->SetVisDaughters(kTRUE);
301 //____________________________________________________________________
303 AliFMDDisplay::ExecuteEvent(Int_t event, Int_t px, Int_t py)
305 // Execute an event on canvas
307 // event What happened
308 // px, py Pixel coordinates
309 if (px == 0 && py == 0) return;
310 if (!fZoomMode && fPad->GetView()) {
311 fPad->GetView()->ExecuteRotateView(event, px, py);
314 fPad->SetCursor(kCross);
317 fPad->TAttLine::Modify();
318 fX0 = fPad->AbsPixeltoX(px);
319 fY0 = fPad->AbsPixeltoY(py);
320 fXPixel = fOldXPixel = px;
321 fYPixel = fOldYPixel = py;
326 gVirtualX->DrawBox(fXPixel, fYPixel, fOldXPixel, fOldYPixel,
331 gVirtualX->DrawBox(fXPixel, fYPixel, fOldXPixel, fOldYPixel,
335 fPad->GetCanvas()->FeedbackMode(kFALSE);
336 if (px == fXPixel || py == fYPixel) return;
337 fX1 = fPad->AbsPixeltoX(px);
338 fY1 = fPad->AbsPixeltoY(py);
339 if (fX1 < fX0) std::swap(fX0, fX1);
340 if (fY1 < fY0) std::swap(fY0, fY1);
341 fPad->Range(fX0, fY0, fX1, fY1);
347 //____________________________________________________________________
349 AliFMDDisplay::DistancetoPrimitive(Int_t px, Int_t)
351 // Calculate the distance from point to
352 // something in the canvas.
353 // Depends on the zoom mode.
354 fPad->SetCursor(kCross);
355 Float_t xmin = fPad->GetX1();
356 Float_t xmax = fPad->GetX2();
357 Float_t dx = .02 * (xmax - xmin);
358 Float_t x = fPad->AbsPixeltoX(px);
359 if (x < xmin + dx || x > xmax - dx) return 9999;
360 return (fZoomMode ? 0 : 7);
362 //____________________________________________________________________
364 AliFMDDisplay::Init()
366 // Initialize. GEt transforms and such,
367 // so that we can draw thins properly
368 // Returns true on success
369 if (!AliFMDInput::Init()) return kFALSE;
370 AliFMDGeometry* geom = AliFMDGeometry::Instance();
372 geom->InitTransformations();
373 if (TESTBIT(fTreeMask, kDigits) || TESTBIT(fTreeMask, kRaw))
374 AliFMDParameters::Instance()->Init();
376 fMarkers = new TObjArray;
377 fHits = new TObjArray;
378 fMarkers->SetOwner(kTRUE);
379 fHits->SetOwner(kFALSE);
383 //____________________________________________________________________
385 AliFMDDisplay::MakeAux()
387 // MAke the aux canvas
388 // This is used to display spectra
390 if ((TESTBIT(fTreeMask, kESD) ||
391 TESTBIT(fTreeMask, kDigits) ||
392 TESTBIT(fTreeMask, kRaw))) {
394 fAux = new TCanvas("aux", "Aux");
396 if (TESTBIT(fTreeMask, kESD))
397 fSpec = new TH1D("spec", "Mult spectra", 500, 0, 10);
399 fSpec = new TH1D("spec", "Adc spectra", 1024, -.5, 1023.5);
400 fSpecCut = static_cast<TH1*>(fSpec->Clone("specCut"));
401 fSpec->SetFillColor(2);
402 fSpec->SetFillStyle(3001);
403 fSpecCut->SetFillColor(4);
404 fSpecCut->SetFillStyle(3001);
413 //____________________________________________________________________
415 AliFMDDisplay::DrawAux()
417 // Draw in the Aux the canvas
418 // For example draw the spectra
424 fSpecCut->Draw("same");
430 //____________________________________________________________________
432 AliFMDDisplay::Begin(Int_t event)
434 // Begin of event. Make canvas is not already done
436 // event The event number
438 const char* m[] = { "Continue",
450 // AliInfo("Clearing canvas");
453 Warning("End", "No geometry manager");
456 AliInfo("Drawing geometry");
458 fGeoManager->GetTopVolume()->Draw();
459 if (fOnlyFMD) ShowOnlyFMD();
460 AliInfo("Adjusting view");
462 if (fPad->GetView()) {
463 fPad->GetView()->SetView(-200, -40, 80, irep);
464 fPad->GetView()->Zoom();
468 return AliFMDInput::Begin(event);
471 //____________________________________________________________________
473 AliFMDDisplay::AtEnd()
475 // Called at of the event.
486 //____________________________________________________________________
488 AliFMDDisplay::Idle()
491 // Sends the ROOT loop into the idle loop,
492 // so that we can go on.
495 gApplication->StartIdleing();
496 gSystem->InnerLoop();
497 gApplication->StopIdleing();
499 AliInfo("After idle loop");
500 if (fMarkers) fMarkers->Delete();
501 if (fHits) fHits->Clear();
502 AliInfo("After clearing caches");
505 //____________________________________________________________________
509 // End of event. Draw everything
512 if (fReturn) return kFALSE;
513 return AliFMDInput::End();
516 //____________________________________________________________________
518 AliFMDDisplay::LookupColor(Float_t x, Float_t min, Float_t max) const
521 // Get a colour from the current palette depending
522 // on the ratio x/max
523 Float_t range = (max-min);
524 Float_t l = fSlider->GetMinimum();
525 Float_t h = fSlider->GetMaximum();
526 if (l == h) { l = 0; h = 1; }
527 Float_t cmin = range * l;
528 Float_t cmax = range * h;
529 Float_t crange = (cmax-cmin);
530 Int_t idx = Int_t((x-cmin) / crange * gStyle->GetNumberOfColors());
531 return gStyle->GetColorPalette(idx);
534 //____________________________________________________________________
536 AliFMDDisplay::ChangeCut()
538 // Change the cut on the slider.
539 // The factor depends on what is
540 // drawn in the AUX canvas
541 AliInfo(Form("Range is now %3.1f - %3.1f", fSlider->GetMinimum(),
542 fSlider->GetMaximum()));
545 //____________________________________________________________________
547 AliFMDDisplay::ChangeFactor()
549 // Change the cut on the slider.
550 // The factor depends on what is
551 // drawn in the AUX canvas
552 AliInfo(Form("Noise factor is now %4.1f, pedestal factor %3.1f",
553 10*fFactor->GetMinimum(),fFactor->GetMaximum()));
557 //____________________________________________________________________
559 AliFMDDisplay::Redisplay()
562 // Redraw markers, hits,
564 if (fMarkers) fMarkers->Delete();
565 if (fHits) fHits->Clear();
566 if (fSpec) fSpec->Reset();
567 if (fSpecCut) fSpecCut->Reset();
571 //____________________________________________________________________
573 AliFMDDisplay::Break()
576 // Redraw markers, hits,
578 if (fMarkers) fMarkers->Delete();
579 if (fHits) fHits->Clear();
580 if (fSpec) fSpec->Reset();
581 if (fSpecCut) fSpecCut->Reset();
585 //____________________________________________________________________
587 AliFMDDisplay::Render()
590 TVirtualViewer3D* viewer = fPad->GetViewer3D("ogl");
594 //____________________________________________________________________
596 AliFMDDisplay::AddMarker(Float_t x, Float_t y, Float_t z,
597 TObject* o, Float_t s, Float_t min, Float_t max)
599 // Add a marker to the display
605 // o Object to refer to
607 // max Maximum of signal
610 Float_t zsize = (s - min) / (max-min) * 10;
611 Float_t r = TMath::Sqrt(x * x + y * y);
612 Float_t theta = TMath::ATan2(r, z);
613 Float_t phi = TMath::ATan2(y, x);
614 Float_t rz = z + (z < 0 ? 1 : -1) * zsize;
615 TMarker3DBox* marker = new TMarker3DBox(x,y,rz,size,size,zsize,theta,phi);
616 if (o) marker->SetRefObject(o);
617 marker->SetLineColor(LookupColor(s, min, max));
618 fMarkers->Add(marker);
620 //____________________________________________________________________
622 AliFMDDisplay::AddMarker(UShort_t det, Char_t rng, UShort_t sec, UShort_t str,
623 TObject* o, Float_t s, Float_t min, Float_t max)
625 // Add a marker to the display
631 // o Object to refer to
633 // max Maximum of signal
635 AliFMDGeometry* geom = AliFMDGeometry::Instance();
637 geom->Detector2XYZ(det, rng, sec, str, x, y, z);
638 AddMarker(x,y,z,o,s,min,max);
641 //____________________________________________________________________
643 AliFMDDisplay::InsideCut(Float_t val, const Float_t& min,
644 const Float_t& max) const
646 Float_t r = max - min;
647 Float_t l = fSlider->GetMinimum();
648 Float_t h = fSlider->GetMaximum();
649 if (l == h) { l = 0; h = 1; }
650 if (val < r * l + min || val > r * h + min) return kFALSE;
655 //____________________________________________________________________
657 AliFMDDisplay::ProcessHit(AliFMDHit* hit, TParticle* /* p */)
663 static const Float_t rMin = 0;
664 static const Float_t rMax = .1;
666 if (!hit) { AliError("No hit"); return kFALSE; }
667 // if (!p) { AliError("No track"); return kFALSE; }
669 if (fHits) fHits->Add(hit);
670 Float_t edep = hit->Edep();
672 if (!InsideCut(edep, rMin, rMax)) return kTRUE;
674 AddMarker(hit->X(), hit->Y(), hit->Z(), hit, edep, rMin, rMax);
678 //____________________________________________________________________
680 AliFMDDisplay::ProcessDigit(AliFMDDigit* digit)
684 // digit Digit information
686 static const Float_t rMin = 0;
687 static const Float_t rMax = 1023;
688 if (!digit) { AliError("No digit"); return kFALSE; }
690 AliFMDParameters* parm = AliFMDParameters::Instance();
691 UShort_t det = digit->Detector();
692 Char_t ring = digit->Ring();
693 UShort_t sec = digit->Sector();
694 UShort_t str = digit->Strip();
695 Double_t ped = parm->GetPedestal(det,ring, sec, str);
696 Double_t pedW = parm->GetPedestalWidth(det,ring, sec, str);
697 Double_t threshold = (ped * fFactor->GetMaximum()
698 + pedW * fFactor->GetMinimum());
699 Float_t counts = digit->Counts();
701 if (fHits) fHits->Add(digit);
702 if (fSpec) fSpec->Fill(counts);
703 if (!InsideCut(counts-threshold, rMin, rMax)) return kTRUE;
704 if (fSpecCut) fSpecCut->Fill(counts);
707 AddMarker(det, ring, sec, str, digit, counts, rMin, rMax);
711 //____________________________________________________________________
713 AliFMDDisplay::ProcessSDigit(AliFMDSDigit* sdigit)
717 // sdigit Digit information
719 static const Float_t rMin = 0;
720 static const Float_t rMax = 1023;
721 if (!sdigit) { AliError("No sdigit"); return kFALSE; }
723 UShort_t det = sdigit->Detector();
724 Char_t ring = sdigit->Ring();
725 UShort_t sec = sdigit->Sector();
726 UShort_t str = sdigit->Strip();
727 Float_t counts = sdigit->Counts();
729 if (fHits) fHits->Add(sdigit);
730 if (fSpec) fSpec->Fill(counts);
731 if (!InsideCut(counts, rMin, rMax)) return kTRUE;
732 if (fSpecCut) fSpecCut->Fill(counts);
735 AddMarker(det, ring, sec, str, sdigit, counts, rMin, rMax);
739 //____________________________________________________________________
741 AliFMDDisplay::ProcessRaw(AliFMDDigit* digit)
745 // digit Digit information
746 return ProcessDigit(digit);
749 //____________________________________________________________________
751 AliFMDDisplay::ProcessRecPoint(AliFMDRecPoint* recpoint)
753 // Process reconstructed point
755 // recpoint Reconstructed multiplicity/energy
756 static const Float_t rMin = 0;
757 static const Float_t rMax = 20;
759 if (!recpoint) { AliError("No recpoint"); return kFALSE; }
761 if (!InsideCut(recpoint->Particles(), rMin, rMax)) return kTRUE;
763 if (fHits) fHits->Add(recpoint);
764 AddMarker(recpoint->Detector(), recpoint->Ring(), recpoint->Sector(),
765 recpoint->Strip(), recpoint, recpoint->Particles(), rMin, rMax);
769 //____________________________________________________________________
771 AliFMDDisplay::ProcessESD(UShort_t det, Char_t rng, UShort_t sec, UShort_t str,
772 Float_t, Float_t mult)
774 // Process data from ESD
776 // det,rng,sec,str Detector coordinates.
777 // mult Multiplicity.
778 static const Float_t rMin = 0;
779 static const Float_t rMax = 20;
781 Double_t cmult = mult;
782 if (fSpec) fSpec->Fill(cmult);
783 if (!InsideCut(cmult, rMin, rMax) || cmult == AliESDFMD::kInvalidMult)
786 AddMarker(det,rng,sec,str, 0, cmult, rMin, rMax);
788 if (fSpecCut) fSpecCut->Fill(cmult);
793 //____________________________________________________________________