]> git.uio.no Git - u/mrichter/AliRoot.git/blame - EVE/EveBase/AliEveLego.cxx
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Coverity
[u/mrichter/AliRoot.git] / EVE / EveBase / AliEveLego.cxx
CommitLineData
3d94b490 1// $Id$
cce980ec 2// Author: Stefano Carrazza 2010, CERN, stefano.carrazza@cern.ch
3d94b490 3
4/**************************************************************************
5 * Copyright(c) 1998-2009, ALICE Experiment at CERN, all rights reserved. *
6 * See http://aliceinfo.cern.ch/Offline/AliRoot/License.html for *
7 * full copyright notice. *
8 **************************************************************************/
9
10#include "AliESDEvent.h"
11#include "AliEveLego.h"
12#include "AliEveEventManager.h"
13#include "AliEveMultiView.h"
14#include "AliPhysicsSelection.h"
3d94b490 15
16#include "TH2F.h"
17#include "TMath.h"
18#include "TGLViewer.h"
19#include "TEveWindow.h"
20#include "TEveManager.h"
21#include "TEveBrowser.h"
22#include "TEveViewer.h"
23#include "TEveScene.h"
24#include "TEveCaloLegoOverlay.h"
25#include "TEveCalo.h"
26#include "TEveCaloData.h"
27#include "TEveLegoEventHandler.h"
28#include "TEveTrans.h"
29#include "TEveProjectionManager.h"
30#include "TEveProjectionAxes.h"
31#include "TGLWidget.h"
a096a7bf 32#include "TStopwatch.h"
3d94b490 33
34//______________________________________________________________________________
cce980ec 35// This class provides the following features:
36//
37// 1) 2D and 3D calorimeter like histograms of tracks distribution,
38// providing particle selection by charge (positive and negative),
39// and by specie: electrons, muons, pions, kaons and protons.
40//
41// 2) Histograms are plotted around detectors, allowing track selection: all tracks,
42// only primary tracks. It allows pT maximum and minimum threashold.
43//
44// 3) 2D and 3D calorimeter like histograms of particles distribution obtained
45// from all events.
46//
47// 4) Possibility to use AliPhysicsSelection during the all events histograms creation.
48// It is also possible to switch between real data and simulation data (MC).
3d94b490 49//
50
51ClassImp(AliEveLego)
4171aa4e 52Double_t kPi = TMath::Pi();
3d94b490 53
54//______________________________________________________________________________
55AliEveLego::AliEveLego(const char* name) :
56 TEveElementList(name),
6e994a7b 57 fIsMC(kFALSE),
58 fCollisionCandidatesOnly(kFALSE),
59 fParticleTypeId(0),
60 fParticleTypeIdAE(0),
3d94b490 61 fTracksId(1),
3d94b490 62 fMaxPt(10000),
3d94b490 63 fTracksIdAE(0),
a096a7bf 64 fMaxPtAE(10000),
3d94b490 65 fEsd(0),
66 fPhysicsSelection(0),
67 fHistopos(0),
4171aa4e 68 fHistoposAllEvents(0),
3d94b490 69 fHistoneg(0),
4171aa4e 70 fHistonegAllEvents(0),
6e994a7b 71 fHistoElectrons(0),
72 fHistoElectronsAllEvents(0),
73 fHistoMuons(0),
74 fHistoMuonsAllEvents(0),
75 fHistoPions(0),
76 fHistoPionsAllEvents(0),
77 fHistoKaons(0),
78 fHistoKaonsAllEvents(0),
79 fHistoProtons(0),
80 fHistoProtonsAllEvents(0),
3d94b490 81 fData(0),
4171aa4e 82 fDataAllEvents(0),
3d94b490 83 fLego(0),
4171aa4e 84 fLegoAllEvents(0),
3d94b490 85 fCalo3d(0),
4171aa4e 86 fCalo3dAllEvents(0),
3d94b490 87 fGlv(0),
4171aa4e 88 fHisto2dv(0),
89 fHisto2ds(0),
90 fHisto2ds2(0),
91 fHisto2dAllEventsv0(0),
92 fHisto2dAllEventsv1(0),
93 fHisto2dAllEventsv2(0),
94 fHisto2dAllEventsv3(0),
95 fHisto2dAllEventss0(0),
96 fHisto2dAllEventss1(0),
97 fHisto2dAllEventss2(0),
98 fHisto2dAllEventss3(0),
3d94b490 99 fAl(0),
4171aa4e 100 fHisto2dLegoOverlay(0),
101 fHisto2dAllEventsLegoOverlay(0),
6e994a7b 102 fHisto2dAllEventsSlot(0)
3d94b490 103{
104 // Constructor.
105 gEve->AddToListTree(this,0);
106
107 // Get Current ESD event
108 fEsd = AliEveEventManager::AssertESD();
109
6e994a7b 110 // Particles types per default
111 fParticleTypeId = new Bool_t[7];
112 fParticleTypeIdAE = new Bool_t[7];
113
114 for (Int_t s = 0; s < 7; s++)
115 {
cce980ec 116 if (s > 1)
117 {
6e994a7b 118 fParticleTypeId[s] = kFALSE;
119 fParticleTypeIdAE[s] = kFALSE;
120 } else {
121 fParticleTypeId[s] = kTRUE;
122 fParticleTypeIdAE[s] = kTRUE;
123 }
124 }
125
cce980ec 126 // Loading Physics Selection to determine the collision candidates
3d94b490 127 fPhysicsSelection = new AliPhysicsSelection();
1107e2ce 128 fPhysicsSelection->Initialize(fEsd);
3d94b490 129
6e994a7b 130 fHistopos = new TH2F("histopos","Histo 2d positive", 100, -1.5, 1.5, 80, -kPi, kPi);
131 fHistoneg = new TH2F("histoneg","Histo 2d negative", 100, -1.5, 1.5, 80, -kPi, kPi);
132 fHistoElectrons = new TH2F("histoele","Histo 2d electron", 100, -1.5, 1.5, 80, -kPi, kPi);
133 fHistoMuons = new TH2F("histomuo","Histo 2d muons ", 100, -1.5, 1.5, 80, -kPi, kPi);
134 fHistoPions = new TH2F("histopio","Histo 2d pions ", 100, -1.5, 1.5, 80, -kPi, kPi);
135 fHistoKaons = new TH2F("histokao","Histo 2d kaons ", 100, -1.5, 1.5, 80, -kPi, kPi);
136 fHistoProtons = new TH2F("histopro","Histo 2d protons ", 100, -1.5, 1.5, 80, -kPi, kPi);
a096a7bf 137
138 fHistopos->SetDirectory(0);
139 fHistoneg->SetDirectory(0);
6e994a7b 140 fHistoElectrons->SetDirectory(0);
141 fHistoMuons->SetDirectory(0);
142 fHistoPions->SetDirectory(0);
143 fHistoKaons->SetDirectory(0);
144 fHistoProtons->SetDirectory(0);
145
cce980ec 146 // Colors from get_pdg_color() in /alice-macros/kine_tracks.C
6e994a7b 147 static Color_t fElectro = 5;
148 static Color_t fMuon = 30;
149 static Color_t fPion = 3;
150 static Color_t fKaon = 38;
151 static Color_t fProton = 10;
3d94b490 152
cce980ec 153 // Adding data to TEveCaloDataHist
3d94b490 154 fData = new TEveCaloDataHist();
155 fData->AddHistogram(fHistoneg);
156 fData->RefSliceInfo(0).Setup("NegCg:", 0, kBlue);
157 fData->AddHistogram(fHistopos);
158 fData->RefSliceInfo(1).Setup("PosCg:", 0, kRed);
6e994a7b 159 fData->AddHistogram(fHistoElectrons);
cce980ec 160 fData->RefSliceInfo(2).Setup("Elect.:", 0, fElectro);
6e994a7b 161 fData->AddHistogram(fHistoMuons);
162 fData->RefSliceInfo(3).Setup("Muons:", 0, fMuon);
163 fData->AddHistogram(fHistoPions);
164 fData->RefSliceInfo(4).Setup("Pions:", 0, fPion);
165 fData->AddHistogram(fHistoKaons);
166 fData->RefSliceInfo(5).Setup("Kaons:", 0, fKaon);
167 fData->AddHistogram(fHistoProtons);
cce980ec 168 fData->RefSliceInfo(6).Setup("Proto.:", 0, fProton);
6e994a7b 169
3d94b490 170 fData->GetEtaBins()->SetTitleFont(120);
171 fData->GetEtaBins()->SetTitle("h");
172 fData->GetPhiBins()->SetTitleFont(120);
173 fData->GetPhiBins()->SetTitle("f");
174 fData->IncDenyDestroy();
175
cce980ec 176 // Setting up the position of the 3D calorimeter histogram view.
3d94b490 177 fCalo3d = new TEveCalo3D(fData);
178 fCalo3d->SetBarrelRadius(550);
179 fCalo3d->SetEndCapPos(550);
180
cce980ec 181 // Adding data to the Lego object
3d94b490 182 fLego = new TEveCaloLego(fData);
183
cce980ec 184 // Creating projections
3d94b490 185 fAl = AliEveMultiView::Instance();
186 fAl->ImportEventRPhi(fCalo3d);
187 fAl->ImportEventRhoZ(fCalo3d);
188
cce980ec 189 // Update viewers and scenes
3d94b490 190 Update();
191}
192
193//______________________________________________________________________________
194AliEveLego::~AliEveLego()
195{
cce980ec 196 // Deleting variables
3d94b490 197 delete fEsd;
198 delete fPhysicsSelection;
199 delete fHistopos;
4171aa4e 200 delete fHistoposAllEvents;
3d94b490 201 delete fHistoneg;
4171aa4e 202 delete fHistonegAllEvents;
6e994a7b 203 delete fHistoElectrons;
204 delete fHistoElectronsAllEvents;
205 delete fHistoMuons;
206 delete fHistoMuonsAllEvents;
207 delete fHistoPions;
208 delete fHistoPionsAllEvents;
209 delete fHistoKaons;
210 delete fHistoKaonsAllEvents;
211 delete fHistoProtons;
212 delete fHistoProtonsAllEvents;
6e994a7b 213 delete[] fParticleTypeId;
214 delete[] fParticleTypeIdAE;
3d94b490 215 delete fData;
4171aa4e 216 delete fDataAllEvents;
3d94b490 217 delete fLego;
4171aa4e 218 delete fLegoAllEvents;
3d94b490 219 delete fCalo3d;
4171aa4e 220 delete fCalo3dAllEvents;
3d94b490 221 delete fGlv;
4171aa4e 222 delete fHisto2dv;
223 delete fHisto2ds;
224 delete fHisto2ds2;
225 delete fHisto2dAllEventsv0;
226 delete fHisto2dAllEventsv1;
227 delete fHisto2dAllEventsv2;
228 delete fHisto2dAllEventsv3;
229 delete fHisto2dAllEventss0;
230 delete fHisto2dAllEventss1;
231 delete fHisto2dAllEventss2;
232 delete fHisto2dAllEventss3;
3d94b490 233 delete fAl;
4171aa4e 234 delete fHisto2dLegoOverlay;
235 delete fHisto2dAllEventsLegoOverlay;
236 delete fHisto2dAllEventsSlot;
3d94b490 237}
238
4171aa4e 239namespace
3d94b490 240{
4171aa4e 241 //____________________________________________________________________________
242 Double_t getphi(Double_t phi)
243 {
cce980ec 244 // Phi correction for alice
4171aa4e 245
246 if (phi > TMath::Pi()) {
247 phi -= TMath::TwoPi();
248 }
249 return phi;
250 }
3d94b490 251}
252
253//______________________________________________________________________________
254TEveCaloDataHist* AliEveLego::LoadData()
255{
4171aa4e 256 // Load data from ESD tree
3d94b490 257 fHistopos->Reset();
258 fHistoneg->Reset();
6e994a7b 259 fHistoElectrons->Reset();
260 fHistoMuons->Reset();
261 fHistoPions->Reset();
262 fHistoKaons->Reset();
263 fHistoProtons->Reset();
3d94b490 264
265 // Getting current tracks, filling histograms
266 for (int n = 0; n < fEsd->GetNumberOfTracks(); ++n) {
267
6e994a7b 268 AliESDtrack *track = fEsd->GetTrack(n);
269 const Double_t sign = fEsd->GetTrack(n)->GetSign();
270 const Int_t prob = GetParticleType(track);
3d94b490 271
cce980ec 272 // Filling histograms
273 if (sign > 0)
274 fHistopos->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 275
cce980ec 276 if (sign < 0)
277 fHistoneg->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 278
cce980ec 279 if (prob == 0)
280 fHistoElectrons->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 281
cce980ec 282 if (prob == 1)
283 fHistoMuons->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 284
cce980ec 285 if (prob == 2)
286 fHistoPions->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 287
cce980ec 288 if (prob == 3)
289 fHistoKaons->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 290
cce980ec 291 if (prob == 4)
292 fHistoProtons->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
3d94b490 293 }
294
3d94b490 295 fData->DataChanged();
296
297 FilterData();
298
299 return fData;
300}
301
302//______________________________________________________________________________
303TEveCaloDataHist* AliEveLego::LoadAllData()
304{
cce980ec 305 // Load data from all events ESD
4171aa4e 306 fHistoposAllEvents->Reset();
307 fHistonegAllEvents->Reset();
6e994a7b 308 fHistoElectronsAllEvents->Reset();
309 fHistoMuonsAllEvents->Reset();
310 fHistoPionsAllEvents->Reset();
311 fHistoKaonsAllEvents->Reset();
312 fHistoProtonsAllEvents->Reset();
3d94b490 313
314 TTree* t = AliEveEventManager::GetMaster()->GetESDTree();
315
316 // Getting current tracks for each event, filling histograms
6e994a7b 317 Int_t fAcceptedEvents = 0;
3d94b490 318 for (int event = 0; event < t->GetEntries(); event++) {
319 t->GetEntry(event);
6e994a7b 320
321 if (fCollisionCandidatesOnly == kTRUE)
322 if (fPhysicsSelection->IsCollisionCandidate(fEsd) == kFALSE) continue;
323
324 fAcceptedEvents++;
325
326 for (int n = 0; n < fEsd->GetNumberOfTracks(); ++n) {
327
328 AliESDtrack *track = fEsd->GetTrack(n);
329 const Double_t sign = track->GetSign();
330 const Int_t prob = GetParticleType(track);
331
cce980ec 332 if (sign > 0)
333 fHistoposAllEvents->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 334
cce980ec 335 if (sign < 0)
336 fHistonegAllEvents->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 337
cce980ec 338 if (prob == 0)
339 fHistoElectronsAllEvents->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 340
cce980ec 341 if (prob == 1)
342 fHistoMuonsAllEvents->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 343
cce980ec 344 if (prob == 2)
345 fHistoPionsAllEvents->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 346
cce980ec 347 if (prob == 3)
348 fHistoKaonsAllEvents->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
6e994a7b 349
cce980ec 350 if (prob == 4)
351 fHistoProtonsAllEvents->Fill(track->Eta(), getphi(track->Phi()), fabs(track->Pt()));
352 }
3d94b490 353 }
cce980ec 354
355 // Setting the current view to the first event
6e994a7b 356 t->GetEntry(0);
cce980ec 357
358 // Usefull information,
359 // with this we can estimate the event efficiency
6e994a7b 360 printf("Number of events loaded: %i, with AliPhysicsSelection: %i\n",fAcceptedEvents,fCollisionCandidatesOnly);
3d94b490 361
4171aa4e 362 fDataAllEvents->DataChanged();
3d94b490 363
4171aa4e 364 return fDataAllEvents;
3d94b490 365}
366
367//______________________________________________________________________________
368TEveCaloDataHist* AliEveLego::FilterData()
369{
370 // Tracks selection
371 if ( fTracksId == 2 )
372 {
373 fHistopos->Reset();
374 fHistoneg->Reset();
6e994a7b 375 fHistoElectrons->Reset();
376 fHistoMuons->Reset();
377 fHistoPions->Reset();
378 fHistoKaons->Reset();
379 fHistoProtons->Reset();
3d94b490 380
381 const AliESDVertex *pv = fEsd->GetPrimaryVertex();
382
383 for (Int_t n = 0; n < pv->GetNIndices(); n++ )
384 {
385 AliESDtrack *at = fEsd->GetTrack(pv->GetIndices()[n]);
6e994a7b 386 const Double_t sign = at->GetSign();
387 const Int_t prob = GetParticleType(at);
388
389 if (sign > 0)
390 fHistopos->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
391
392 if (sign < 0)
393 fHistoneg->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
394
395 if (prob == 0)
396 fHistoElectrons->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
397
398 if (prob == 1)
399 fHistoMuons->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
400
401 if (prob == 2)
402 fHistoPions->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
403
404 if (prob == 3)
405 fHistoKaons->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
406
407 if (prob == 4)
408 fHistoProtons->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
3d94b490 409 }
410 }
411
412 fData->DataChanged();
413
414 // Max Pt threshold
415 if (GetPtMax() >= fMaxPt){
416 for (Int_t binx = 1; binx <= 100; binx++) {
417 for (Int_t biny = 1; biny <= 80; biny++) {
6e994a7b 418
419 if (fHistopos->GetBinContent(binx, biny) >= fMaxPt)
420 fHistopos->SetBinContent(binx, biny, fMaxPt);
421
422 if (fHistoneg->GetBinContent(binx, biny) >= fMaxPt)
423 fHistoneg->SetBinContent(binx, biny, fMaxPt);
424
425 if (fHistoElectrons->GetBinContent(binx, biny) >= fMaxPt)
426 fHistoElectrons->SetBinContent(binx, biny, fMaxPt);
427
428 if (fHistoMuons->GetBinContent(binx, biny) >= fMaxPt)
429 fHistoMuons->SetBinContent(binx, biny, fMaxPt);
430
431 if (fHistoPions->GetBinContent(binx, biny) >= fMaxPt)
432 fHistoPions->SetBinContent(binx, biny, fMaxPt);
433
434 if (fHistoKaons->GetBinContent(binx, biny) >= fMaxPt)
435 fHistoKaons->SetBinContent(binx, biny, fMaxPt);
436
437 if (fHistoProtons->GetBinContent(binx, biny) >= fMaxPt)
438 fHistoProtons->SetBinContent(binx, biny, fMaxPt);
3d94b490 439 }
440 }
441 }
442
6e994a7b 443 // Particle type filter
444 if ( fParticleTypeId[0] == kFALSE) fHistopos->Reset();
445 if ( fParticleTypeId[1] == kFALSE) fHistoneg->Reset();
446 if ( fParticleTypeId[2] == kFALSE) fHistoElectrons->Reset();
447 if ( fParticleTypeId[3] == kFALSE) fHistoMuons->Reset();
448 if ( fParticleTypeId[4] == kFALSE) fHistoPions->Reset();
449 if ( fParticleTypeId[5] == kFALSE) fHistoKaons->Reset();
450 if ( fParticleTypeId[6] == kFALSE) fHistoProtons->Reset();
3d94b490 451
452 fData->DataChanged();
453
454 return fData;
455}
456
3d94b490 457//______________________________________________________________________________
458TEveCaloDataHist* AliEveLego::FilterAllData()
459{
460 // Tracks selection
461 if ( fTracksIdAE == 2 )
462 {
4171aa4e 463 fHistoposAllEvents->Reset();
464 fHistonegAllEvents->Reset();
6e994a7b 465 fHistoElectronsAllEvents->Reset();
466 fHistoMuonsAllEvents->Reset();
467 fHistoPionsAllEvents->Reset();
468 fHistoKaonsAllEvents->Reset();
469 fHistoProtonsAllEvents->Reset();
3d94b490 470
471 TTree* t = AliEveEventManager::GetMaster()->GetESDTree();
472
473 // Getting current tracks for each event, filling histograms
6e994a7b 474 Int_t fAcceptedEvents = 0;
3d94b490 475 for (int event = 0; event < t->GetEntries(); event++) {
a096a7bf 476
6e994a7b 477 t->GetEntry(event);
478 const AliESDVertex *pv = fEsd->GetPrimaryVertex();
3d94b490 479
6e994a7b 480 if (fCollisionCandidatesOnly == kTRUE)
481 if (fPhysicsSelection->IsCollisionCandidate(fEsd) == kFALSE) continue;
3d94b490 482
6e994a7b 483 fAcceptedEvents++;
a096a7bf 484
6e994a7b 485 for (Int_t n = 0; n < pv->GetNIndices(); n++ )
486 {
487 AliESDtrack *at = fEsd->GetTrack(pv->GetIndices()[n]);
488 const Double_t sign = at->GetSign();
489 const Int_t prob = GetParticleType(at);
490
491 if (sign > 0)
4171aa4e 492 fHistoposAllEvents->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
6e994a7b 493
494 if (sign < 0)
4171aa4e 495 fHistonegAllEvents->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
6e994a7b 496
497 if (prob == 0)
498 fHistoElectronsAllEvents->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
499
500 if (prob == 1)
501 fHistoMuonsAllEvents->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
502
503 if (prob == 2)
504 fHistoPionsAllEvents->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
505
506 if (prob == 3)
507 fHistoKaonsAllEvents->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
508
509 if (prob == 4)
510 fHistoProtonsAllEvents->Fill(at->Eta(), getphi(at->Phi()), fabs(at->Pt()));
511 }
a096a7bf 512 }
6e994a7b 513 t->GetEntry(0);
514 printf("Number of events loaded: %i, with AliPhysicsSelection: %i\n",fAcceptedEvents,fCollisionCandidatesOnly);
515
3d94b490 516 } else {
6e994a7b 517
3d94b490 518 LoadAllData();
6e994a7b 519
520 }
3d94b490 521
4171aa4e 522 fDataAllEvents->DataChanged();
a096a7bf 523
3d94b490 524 // Max Pt threshold
525 if (GetPtMaxAE() >= fMaxPtAE){
526 for (Int_t binx = 1; binx <= 100; binx++) {
527 for (Int_t biny = 1; biny <= 80; biny++) {
6e994a7b 528
529 if (fHistoposAllEvents->GetBinContent(binx, biny) >= fMaxPtAE)
530 fHistoposAllEvents->SetBinContent(binx, biny, fMaxPtAE);
531
532 if (fHistonegAllEvents->GetBinContent(binx, biny) >= fMaxPtAE)
533 fHistonegAllEvents->SetBinContent(binx, biny, fMaxPtAE);
534
535 if (fHistoElectronsAllEvents->GetBinContent(binx, biny) >= fMaxPt)
536 fHistoElectronsAllEvents->SetBinContent(binx, biny, fMaxPt);
537
538 if (fHistoMuonsAllEvents->GetBinContent(binx, biny) >= fMaxPt)
539 fHistoMuonsAllEvents->SetBinContent(binx, biny, fMaxPt);
540
541 if (fHistoPionsAllEvents->GetBinContent(binx, biny) >= fMaxPt)
542 fHistoPionsAllEvents->SetBinContent(binx, biny, fMaxPt);
543
544 if (fHistoKaonsAllEvents->GetBinContent(binx, biny) >= fMaxPt)
545 fHistoKaonsAllEvents->SetBinContent(binx, biny, fMaxPt);
546
547 if (fHistoProtonsAllEvents->GetBinContent(binx, biny) >= fMaxPt)
548 fHistoProtonsAllEvents->SetBinContent(binx, biny, fMaxPt);
3d94b490 549 }
550 }
551 }
552
cce980ec 553 // Particles species and charges filter
6e994a7b 554 if ( fParticleTypeIdAE[0] == kFALSE ) fHistoposAllEvents->Reset();
555 if ( fParticleTypeIdAE[1] == kFALSE ) fHistonegAllEvents->Reset();
556 if ( fParticleTypeIdAE[2] == kFALSE ) fHistoElectronsAllEvents->Reset();
557 if ( fParticleTypeIdAE[3] == kFALSE ) fHistoMuonsAllEvents->Reset();
558 if ( fParticleTypeIdAE[4] == kFALSE ) fHistoPionsAllEvents->Reset();
559 if ( fParticleTypeIdAE[5] == kFALSE ) fHistoKaonsAllEvents->Reset();
560 if ( fParticleTypeIdAE[6] == kFALSE ) fHistoProtonsAllEvents->Reset();
3d94b490 561
4171aa4e 562 fDataAllEvents->DataChanged();
3d94b490 563
564 gEve->Redraw3D(kTRUE);
565
4171aa4e 566 return fDataAllEvents;
3d94b490 567}
568
569//______________________________________________________________________________
570void AliEveLego::Update()
571{
572 // Load/Reload data
573 LoadData();
574
575 // Create new histo2d
576 CreateHistoLego();
577
578 // Create 3d view
579 Create3DView();
580
3d94b490 581 // Update the viewers
582 gEve->Redraw3D(kTRUE);
583}
584
585//______________________________________________________________________________
586TEveCaloLego* AliEveLego::CreateHistoLego()
587{
588 // Viewer initialization, tab creation
4171aa4e 589 if (fHisto2dv == 0) {
3d94b490 590 TEveWindowSlot *fslot = 0;
591 TEveBrowser *fbrowser = gEve->GetBrowser();
592
593 fslot = TEveWindow::CreateWindowInTab(fbrowser->GetTabRight());
594 fslot->MakeCurrent();
4171aa4e 595 fHisto2dv = gEve->SpawnNewViewer("2D Lego Histogram", "2D Lego Histogram");
596 fHisto2ds = gEve->SpawnNewScene("2D Lego Histogram", "2D Lego Histogram");
597 fHisto2dv->AddScene(fHisto2ds);
598 fHisto2dv->SetElementName("2D Lego Viewer");
599 fHisto2ds->SetElementName("2D Lego Scene");
600
601 fGlv = fHisto2dv->GetGLViewer();
602 fHisto2dLegoOverlay = new TEveCaloLegoOverlay();
603 fGlv->AddOverlayElement(fHisto2dLegoOverlay);
3d94b490 604 fGlv->SetCurrentCamera(TGLViewer::kCameraPerspXOY);
605
4171aa4e 606 fHisto2ds->AddElement(fLego);
3d94b490 607
608 // move to real world coordinates
609 fLego->InitMainTrans();
610 Float_t sc = TMath::Min(fLego->GetEtaRng(), fLego->GetPhiRng());
611 fLego->RefMainTrans().SetScale(sc, sc, sc);
612
613 // set event handler to move from perspective to orthographic view.
614 fGlv->SetEventHandler(new TEveLegoEventHandler(fGlv->GetGLWidget(), fGlv, fLego));
615
4171aa4e 616 fHisto2dLegoOverlay->SetCaloLego(fLego);
3d94b490 617 }
618
619 return fLego;
620}
621
622//______________________________________________________________________________
623TEveCaloLego* AliEveLego::CreateHistoLego(TEveWindowSlot *slot)
624{
625 // Viewer initialization, tab creation
4171aa4e 626 if (fHisto2dAllEventsv0 == 0) {
3d94b490 627
628 slot->MakeCurrent();
4171aa4e 629 fHisto2dAllEventsv0 = gEve->SpawnNewViewer("2D Lego Histogram", "2D Lego Histogram");
630 fHisto2dAllEventss0 = gEve->SpawnNewScene("2D Lego Histogram", "2D Lego Histogram");
631 fHisto2dAllEventsv0->AddScene(fHisto2dAllEventss0);
632 fHisto2dAllEventsv0->SetElementName("2D Lego Viewer");
633 fHisto2dAllEventss0->SetElementName("2D Lego Scene");
634
635 TGLViewer* glv = fHisto2dAllEventsv0->GetGLViewer();
636 fHisto2dAllEventsLegoOverlay = new TEveCaloLegoOverlay();
637 glv->AddOverlayElement(fHisto2dAllEventsLegoOverlay);
3d94b490 638 glv->SetCurrentCamera(TGLViewer::kCameraPerspXOY);
639
640 // Plotting histogram lego
4171aa4e 641 fLegoAllEvents = new TEveCaloLego(fDataAllEvents);
642 fHisto2dAllEventss0->AddElement(fLegoAllEvents);
3d94b490 643
644 // Move to real world coordinates
4171aa4e 645 fLegoAllEvents->InitMainTrans();
646 Float_t sc = TMath::Min(fLegoAllEvents->GetEtaRng(), fLegoAllEvents->GetPhiRng());
647 fLegoAllEvents->RefMainTrans().SetScale(sc, sc, sc);
3d94b490 648
649 // Set event handler to move from perspective to orthographic view.
4171aa4e 650 glv->SetEventHandler(new TEveLegoEventHandler(glv->GetGLWidget(), glv, fLegoAllEvents));
3d94b490 651
4171aa4e 652 fHisto2dAllEventsLegoOverlay->SetCaloLego(fLegoAllEvents);
3d94b490 653 }
654
4171aa4e 655 return fLegoAllEvents;
3d94b490 656}
657
658//______________________________________________________________________________
659TEveCalo3D* AliEveLego::Create3DView()
660{
cce980ec 661 // Initialization
4171aa4e 662 if (fHisto2ds2 == 0) {
663 fHisto2ds2 = gEve->SpawnNewScene("3D Histogram", "3D Histogram");
664 gEve->GetDefaultViewer()->AddScene(fHisto2ds2);
665 fHisto2ds2->SetElementName("3D Histogram Scene");
666 fHisto2ds2->AddElement(fCalo3d);
3d94b490 667 }
668
669 return fCalo3d;
670}
671
672//______________________________________________________________________________
673TEveCalo3D* AliEveLego::Create3DView(TEveWindowSlot *slot)
674{
cce980ec 675 // Creates a 3d view for the 3d histogram
4171aa4e 676 if ( fHisto2dAllEventsv1 == 0 ) {
3d94b490 677
678 slot->MakeCurrent();
4171aa4e 679 fHisto2dAllEventsv1 = gEve->SpawnNewViewer("3D Histogram", "3D Histogram");
680 fHisto2dAllEventss1 = gEve->SpawnNewScene("3D Histogram", "3D Histogram");
681 fHisto2dAllEventsv1->AddScene(fHisto2dAllEventss1);
682 fHisto2dAllEventsv1->SetElementName("3D Histogram Viewer");
683 fHisto2dAllEventss1->SetElementName("3D Histogram Scene");
3d94b490 684
4171aa4e 685 fCalo3dAllEvents = new TEveCalo3D(fDataAllEvents);
3d94b490 686
4171aa4e 687 fCalo3dAllEvents->SetBarrelRadius(550);
688 fCalo3dAllEvents->SetEndCapPos(550);
689 fHisto2dAllEventss1->AddElement(fCalo3dAllEvents);
3d94b490 690 }
691
4171aa4e 692 return fCalo3dAllEvents;
3d94b490 693}
694
695//______________________________________________________________________________
4171aa4e 696void AliEveLego::CreateProjections(TEveWindowSlot* slot1, TEveWindowSlot* slot2)
697{
cce980ec 698 // Create projections
4171aa4e 699 if (fHisto2dAllEventsv2 == 0) {
3d94b490 700
701 slot1->MakeCurrent();
4171aa4e 702 fHisto2dAllEventsv2 = gEve->SpawnNewViewer("RPhi projection", "RPhi projection");
703 fHisto2dAllEventss2 = gEve->SpawnNewScene("RPhi projection", "RPhi projection");
704 fHisto2dAllEventsv2->AddScene(fHisto2dAllEventss2);
705 fHisto2dAllEventsv2->SetElementName("RPhi Projection Viewer");
706 fHisto2dAllEventss2->SetElementName("RPhi Projection Scene");
3d94b490 707
708 TEveProjectionManager* mng1 = new TEveProjectionManager();
709 mng1->SetProjection(TEveProjection::kPT_RPhi);
710
4171aa4e 711 TEveProjectionAxes* axeghisto2dAllEventss1 = new TEveProjectionAxes(mng1);
712 fHisto2dAllEventss2->AddElement(axeghisto2dAllEventss1);
713 TEveCalo2D* fcalo2d1 = (TEveCalo2D*) mng1->ImportElements(fCalo3dAllEvents);
714 fHisto2dAllEventss2->AddElement(fcalo2d1);
3d94b490 715
4171aa4e 716 fHisto2dAllEventsv2->GetGLViewer()->SetCurrentCamera(TGLViewer::kCameraOrthoXOY);
3d94b490 717 }
718
cce980ec 719 if (fHisto2dAllEventsv3 == 0)
720 {
3d94b490 721 slot2->MakeCurrent();
4171aa4e 722 fHisto2dAllEventsv3 = gEve->SpawnNewViewer("RhoZ projection", "RhoZ projection");
723 fHisto2dAllEventss3 = gEve->SpawnNewScene("RhoZ projection", "RhoZ projection");
724 fHisto2dAllEventsv3->AddScene(fHisto2dAllEventss3);
725 fHisto2dAllEventsv3->SetElementName("RhoZ Projection Viewer");
726 fHisto2dAllEventss3->SetElementName("RhoZ Projection Viewer");
3d94b490 727
728 TEveProjectionManager* mng2 = new TEveProjectionManager();
729 mng2->SetProjection(TEveProjection::kPT_RhoZ);
730
4171aa4e 731 TEveProjectionAxes* axeghisto2dAllEventss2 = new TEveProjectionAxes(mng2);
732 fHisto2dAllEventss3->AddElement(axeghisto2dAllEventss2);
733 TEveCalo2D* fcalo2d2 = (TEveCalo2D*) mng2->ImportElements(fCalo3dAllEvents);
734 fHisto2dAllEventss3->AddElement(fcalo2d2);
3d94b490 735
4171aa4e 736 fHisto2dAllEventsv3->GetGLViewer()->SetCurrentCamera(TGLViewer::kCameraOrthoXOY);
3d94b490 737 }
3d94b490 738 return;
739}
740
741//______________________________________________________________________________
742TEveCaloDataHist* AliEveLego::LoadAllEvents()
743{
4171aa4e 744 // load all events data from ESD
745 if ( fHisto2dAllEventsSlot == 0 ) {
3d94b490 746
747 printf("Filling histogram...\n");
a096a7bf 748 TStopwatch timer;
749 timer.Start();
3d94b490 750
751 // Creating 2D histograms
4171aa4e 752 fHistoposAllEvents = new TH2F("fHistoposAllEvents","Histo 2d positive",
753 100,-1.5,1.5,80,-kPi,kPi);
754 fHistonegAllEvents = new TH2F("fHistonegAllEvents","Histo 2d negative",
755 100,-1.5,1.5,80,-kPi,kPi);
6e994a7b 756 fHistoElectronsAllEvents = new TH2F("fHistoElectronsAllEvents","Histo 2d electrons",
757 100,-1.5,1.5,80,-kPi,kPi);
758 fHistoMuonsAllEvents = new TH2F("fHistoMuonsAllEvents","Histo 2d muons",
759 100,-1.5,1.5,80,-kPi,kPi);
760 fHistoPionsAllEvents = new TH2F("fHistoPionsAllEvents","Histo 2d pions",
761 100,-1.5,1.5,80,-kPi,kPi);
762 fHistoKaonsAllEvents = new TH2F("fHistoKaonsAllEvents","Histo 2d kaons",
763 100,-1.5,1.5,80,-kPi,kPi);
764 fHistoProtonsAllEvents = new TH2F("fHistoProtonsAllEvents","Histo 2d protons",
765 100,-1.5,1.5,80,-kPi,kPi);
4171aa4e 766
767 fHistoposAllEvents->SetDirectory(0);
768 fHistonegAllEvents->SetDirectory(0);
6e994a7b 769 fHistoElectronsAllEvents->SetDirectory(0);
770 fHistoMuonsAllEvents->SetDirectory(0);
771 fHistoPionsAllEvents->SetDirectory(0);
772 fHistoKaonsAllEvents->SetDirectory(0);
773 fHistoProtonsAllEvents->SetDirectory(0);
774
775 // colors from get_pdg_color() in /alice-macros/kine_tracks.C
776 static Color_t fElectro = 5;
777 static Color_t fMuon = 30;
778 static Color_t fPion = 3;
779 static Color_t fKaon = 38;
780 static Color_t fProton = 10;
4171aa4e 781
782 fDataAllEvents = new TEveCaloDataHist();
783 fDataAllEvents->AddHistogram(fHistonegAllEvents);
784 fDataAllEvents->RefSliceInfo(0).Setup("NegCg:", 0, kBlue);
785 fDataAllEvents->AddHistogram(fHistoposAllEvents);
786 fDataAllEvents->RefSliceInfo(1).Setup("PosCg:", 0, kRed);
6e994a7b 787 fDataAllEvents->AddHistogram(fHistoElectronsAllEvents);
788 fDataAllEvents->RefSliceInfo(2).Setup("Electrons:", 0, fElectro);
789 fDataAllEvents->AddHistogram(fHistoMuonsAllEvents);
790 fDataAllEvents->RefSliceInfo(3).Setup("Muons:", 0, fMuon);
791 fDataAllEvents->AddHistogram(fHistoPionsAllEvents);
792 fDataAllEvents->RefSliceInfo(4).Setup("Pions:", 0, fPion);
793 fDataAllEvents->AddHistogram(fHistoKaonsAllEvents);
794 fDataAllEvents->RefSliceInfo(5).Setup("Kaons:", 0, fKaon);
795 fDataAllEvents->AddHistogram(fHistoProtonsAllEvents);
796 fDataAllEvents->RefSliceInfo(6).Setup("Protons:", 0, fProton);
797
4171aa4e 798 fDataAllEvents->GetEtaBins()->SetTitleFont(120);
799 fDataAllEvents->GetEtaBins()->SetTitle("h");
800 fDataAllEvents->GetPhiBins()->SetTitleFont(120);
801 fDataAllEvents->GetPhiBins()->SetTitle("f");
802 fDataAllEvents->IncDenyDestroy();
3d94b490 803
804 // Creating frames
4171aa4e 805 fHisto2dAllEventsSlot = TEveWindow::CreateWindowInTab(gEve->GetBrowser()->GetTabRight());
806 TEveWindowPack* packH = fHisto2dAllEventsSlot->MakePack();
3d94b490 807 packH->SetElementName("Projections");
808 packH->SetHorizontal();
809 packH->SetShowTitleBar(kFALSE);
810
4171aa4e 811 fHisto2dAllEventsSlot = packH->NewSlot();
812 TEveWindowPack* pack0 = fHisto2dAllEventsSlot->MakePack();
3d94b490 813 pack0->SetShowTitleBar(kFALSE);
814 TEveWindowSlot* slotLeftTop = pack0->NewSlot();
815 TEveWindowSlot* slotLeftBottom = pack0->NewSlot();
816
4171aa4e 817 fHisto2dAllEventsSlot = packH->NewSlot();
818 TEveWindowPack* pack1 = fHisto2dAllEventsSlot->MakePack();
3d94b490 819 pack1->SetShowTitleBar(kFALSE);
820 TEveWindowSlot* slotRightTop = pack1->NewSlot();
821 TEveWindowSlot* slotRightBottom = pack1->NewSlot();
822
823 // Creating viewers and scenes
824 Create3DView(slotLeftTop);
825 CreateHistoLego(slotLeftBottom);
826 CreateProjections(slotRightTop, slotRightBottom);
827
6e994a7b 828 FilterAllData();
3d94b490 829
830 gEve->Redraw3D(kTRUE);
831
832 printf("Filling histogram... Finished\n");
a096a7bf 833 timer.Stop();
834 timer.Print();
3d94b490 835 }
4171aa4e 836 return fDataAllEvents;
3d94b490 837}
838
6e994a7b 839//______________________________________________________________________________
840void AliEveLego::ApplyParticleTypeSelectionAE()
841{
cce980ec 842 // Reload all events applying particle type selection
6e994a7b 843 FilterAllData();
844}
845
3d94b490 846//______________________________________________________________________________
847Float_t AliEveLego::GetPtMax()
848{
cce980ec 849 // Return pT maximum
3d94b490 850 return fData->GetMaxVal(fLego->GetPlotEt());
851}
852
853//______________________________________________________________________________
854Float_t AliEveLego::GetPtMaxAE()
855{
cce980ec 856 // Return pT max from all events
4171aa4e 857 return fDataAllEvents->GetMaxVal(fLegoAllEvents->GetPlotEt());
3d94b490 858}
859
6e994a7b 860//______________________________________________________________________________
861Int_t AliEveLego::GetParticleType(AliESDtrack *track)
862{
cce980ec 863 // Determine the particle type
34ce391d 864 Double_t prob[5]={0.};
6e994a7b 865 track->GetESDpid(prob);
866
867 Double_t max = prob[0];
868 Int_t index = 0;
869
870 for (Int_t i = 1 ; i < 5; i++)
871 {
872 if (prob[i] > max){
873 max = prob[i];
874 index = i;
875 }
876 }
877 return index;
878}
879
880//______________________________________________________________________________
cce980ec 881void AliEveLego::SetParticleType(Int_t id, Bool_t status)
6e994a7b 882{
cce980ec 883 // Activate/deactivate particles types
884 fParticleTypeId[id] = status;
6e994a7b 885
886 Update();
887}
888
889//______________________________________________________________________________
cce980ec 890void AliEveLego::SetParticleTypeAE(Int_t id, Bool_t status)
6e994a7b 891{
cce980ec 892 // Activate/deactivate particles types
893 fParticleTypeIdAE[id] = status;
6e994a7b 894}
895
3d94b490 896//______________________________________________________________________________
897void AliEveLego::SetMaxPt(Double_t val)
898{
899 // Add new maximum
900 fMaxPt = val;
901 Update();
902}
903
904//______________________________________________________________________________
905void AliEveLego::SetMaxPtAE(Double_t val)
906{
907 // Add new maximum
908 fMaxPtAE = val;
909 FilterAllData();
910}
911
912//______________________________________________________________________________
913void AliEveLego::SetThreshold(Double_t val)
914{
915 // Setting up the new threshold for all histograms
916 fData->SetSliceThreshold(0,val);
917 fData->SetSliceThreshold(1,val);
6e994a7b 918 fData->SetSliceThreshold(2,val);
919 fData->SetSliceThreshold(3,val);
920 fData->SetSliceThreshold(4,val);
921 fData->SetSliceThreshold(5,val);
922 fData->SetSliceThreshold(6,val);
3d94b490 923 fData->DataChanged();
924
925 gEve->Redraw3D(kTRUE);
926}
927
928//______________________________________________________________________________
929void AliEveLego::SetThresholdAE(Double_t val)
930{
931 // Setting up the new threshold for all histograms
4171aa4e 932 fDataAllEvents->SetSliceThreshold(0,val);
933 fDataAllEvents->SetSliceThreshold(1,val);
6e994a7b 934 fDataAllEvents->SetSliceThreshold(2,val);
935 fDataAllEvents->SetSliceThreshold(3,val);
936 fDataAllEvents->SetSliceThreshold(4,val);
937 fDataAllEvents->SetSliceThreshold(5,val);
938 fDataAllEvents->SetSliceThreshold(6,val);
4171aa4e 939 fDataAllEvents->DataChanged();
3d94b490 940
941 gEve->Redraw3D(kTRUE);
942}
943
944//______________________________________________________________________________
cce980ec 945void AliEveLego::SwitchDataType(Bool_t status)
3d94b490 946{
cce980ec 947 // Activate/deactivate MC / real data type
948 fIsMC = status;
3d94b490 949
cce980ec 950 // Removing defaul physics selection
6e994a7b 951 delete fPhysicsSelection;
952 fPhysicsSelection = NULL;
3d94b490 953
cce980ec 954 // Re-initialization of physics selection
6e994a7b 955 fPhysicsSelection = new AliPhysicsSelection();
956 fPhysicsSelection->SetAnalyzeMC(fIsMC);
1107e2ce 957 fPhysicsSelection->Initialize(fEsd);
6e994a7b 958 FilterAllData();
3d94b490 959}
960
961//______________________________________________________________________________
6e994a7b 962void AliEveLego::SetCollisionCandidatesOnly()
3d94b490 963{
cce980ec 964 // Activate/deactivate MC / real data type
6e994a7b 965 if (fCollisionCandidatesOnly == 0)
966 {
967 fCollisionCandidatesOnly = 1;
968 } else {
969 fCollisionCandidatesOnly = 0;
970 }
971 FilterAllData();
3d94b490 972}
973
3d94b490 974/******************************************************************************/
975
976