1 /**************************************************************************
2 * Copyright(c) 1998-1999, 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 **************************************************************************/
17 //_________________________________________________________________________
18 // Base class for reading data: MonteCarlo, ESD or AOD, of PHOS EMCAL and
19 // Central Barrel Tracking detectors (CTS).
20 // Not all MC particles/tracks/clusters are kept, some kinematical/fiducial restrictions are done.
21 // Mother class of : AliCaloTrackESDReader: Fills ESD data in 3 TObjArrays (PHOS, EMCAL, CTS)
22 // : AliCaloTrackMCReader: Fills Kinematics data in 3 TObjArrays (PHOS, EMCAL, CTS)
23 // : AliCaloTrackReader: Fills AOD data in 3 TObjArrays (PHOS, EMCAL, CTS)
25 // This part is commented: Mixing analysis can be done, input AOD with events
26 // is opened in the AliCaloTrackReader::Init()
28 //-- Author: Gustavo Conesa (LNF-INFN)
29 //////////////////////////////////////////////////////////////////////////////
32 // --- ROOT system ---
36 //---- ANALYSIS system ----
37 #include "AliMCEvent.h"
38 #include "AliAODMCHeader.h"
39 #include "AliGenPythiaEventHeader.h"
40 #include "AliESDEvent.h"
41 #include "AliAODEvent.h"
42 #include "AliVTrack.h"
43 #include "AliVParticle.h"
44 #include "AliMixedEvent.h"
45 #include "AliESDtrack.h"
46 #include "AliESDtrackCuts.h"
47 #include "AliTriggerAnalysis.h"
48 #include "AliESDVZERO.h"
50 //---- PartCorr/EMCAL ---
51 #include "AliEMCALRecoUtils.h"
52 #include "AliCaloTrackReader.h"
54 ClassImp(AliCaloTrackReader)
57 //____________________________________________________________________________
58 AliCaloTrackReader::AliCaloTrackReader() :
59 TObject(), fEventNumber(-1), //fCurrentFileName(""),
60 fDataType(0), fDebug(0),
61 fFiducialCut(0x0), fCheckFidCut(kFALSE), fComparePtHardAndJetPt(kFALSE), fPtHardAndJetPtFactor(7),
62 fCTSPtMin(0), fEMCALPtMin(0),fPHOSPtMin(0), fAODBranchList(new TList ),
63 fCTSTracks(new TObjArray()), fEMCALClusters(new TObjArray()), fPHOSClusters(new TObjArray()),
64 fEMCALCells(0x0), fPHOSCells(0x0),
65 fInputEvent(0x0), fOutputEvent(0x0),fMC(0x0),
66 fFillCTS(0),fFillEMCAL(0),fFillPHOS(0),
67 fFillEMCALCells(0),fFillPHOSCells(0), fSelectEmbeddedClusters(kFALSE),
68 fRemoveSuspiciousClusters(kFALSE), fSmearClusterEnergy(kFALSE), fRandom(),
69 // fSecondInputAODTree(0x0), fSecondInputAODEvent(0x0),
70 // fSecondInputFileName(""),fSecondInputFirstEvent(0),
71 // fCTSTracksNormalInputEntries(0), fEMCALClustersNormalInputEntries(0),
72 // fPHOSClustersNormalInputEntries(0),
73 fTrackStatus(0), fESDtrackCuts(0), fTrackMult(0), fTrackMultEtaCut(0.8),
74 fReadStack(kFALSE), fReadAODMCParticles(kFALSE),
75 fDeltaAODFileName("deltaAODPartCorr.root"),fFiredTriggerClassName(""),
76 fAnaLED(kFALSE),fTaskName(""),fCaloUtils(0x0),
77 fMixedEvent(NULL), fNMixedEvent(1), fVertex(NULL),
78 fWriteOutputDeltaAOD(kFALSE),fOldAOD(kFALSE),fCaloFilterPatch(kFALSE),
79 fEMCALClustersListName(""),fZvtxCut(0.),
80 fDoEventSelection(kFALSE), fDoV0ANDEventSelection(kFALSE), fUseEventsWithPrimaryVertex(kFALSE),
81 fTriggerAnalysis (new AliTriggerAnalysis), fCentralityClass("V0M"),fCentralityOpt(10)
86 //Initialize parameters
90 //_________________________________
91 AliCaloTrackReader::~AliCaloTrackReader() {
94 if(fFiducialCut) delete fFiducialCut ;
97 fAODBranchList->Delete();
98 delete fAODBranchList ;
102 if(fDataType!=kMC)fCTSTracks->Clear() ;
103 else fCTSTracks->Delete() ;
108 if(fDataType!=kMC)fEMCALClusters->Clear("C") ;
109 else fEMCALClusters->Delete() ;
110 delete fEMCALClusters ;
114 if(fDataType!=kMC)fPHOSClusters->Clear("C") ;
115 else fPHOSClusters->Delete() ;
116 delete fPHOSClusters ;
120 // delete fEMCALCells ;
124 // delete fPHOSCells ;
128 for (Int_t i = 0; i < fNMixedEvent; i++) {
129 delete [] fVertex[i] ;
135 if(fESDtrackCuts) delete fESDtrackCuts;
136 if(fTriggerAnalysis) delete fTriggerAnalysis;
138 // Pointers not owned, done by the analysis frame
139 // if(fInputEvent) delete fInputEvent ;
140 // if(fOutputEvent) delete fOutputEvent ;
141 // if(fMC) delete fMC ;
143 // if(fSecondInputAODTree){
144 // fSecondInputAODTree->Clear();
145 // delete fSecondInputAODTree;
148 // if(fSecondInputAODEvent) delete fSecondInputAODEvent ;
150 // Pointer not owned, deleted by maker
151 //if (fCaloUtils) delete fCaloUtils ;
155 //_________________________________________________________________________
156 Bool_t AliCaloTrackReader::ComparePtHardAndJetPt(){
157 // Check the event, if the requested ptHard is much larger than the jet pT, then there is a problem.
159 if(!fReadStack) return kTRUE; //Information not filtered to AOD
161 if(!strcmp(GetGenEventHeader()->ClassName(), "AliGenPythiaEventHeader")){
163 AliGenPythiaEventHeader* pygeh= (AliGenPythiaEventHeader*) GetGenEventHeader();
164 Int_t nTriggerJets = pygeh->NTriggerJets();
165 Float_t ptHard = pygeh->GetPtHard();
167 //if(fDebug > 1) printf("AliMCAnalysisUtils::PythiaEventHeader: Njets: %d, pT Hard %f\n",nTriggerJets, ptHard);
168 Float_t tmpjet[]={0,0,0,0};
169 for(Int_t ijet = 0; ijet< nTriggerJets; ijet++){
170 pygeh->TriggerJet(ijet, tmpjet);
171 jet = new TParticle(94, 21, -1, -1, -1, -1, tmpjet[0],tmpjet[1],tmpjet[2],tmpjet[3], 0,0,0,0);
172 //Compare jet pT and pt Hard
173 //if(fDebug > 1) printf("AliMCAnalysisUtils:: %d pycell jet pT %f\n",ijet, jet->Pt());
174 if(jet->Pt() > fPtHardAndJetPtFactor * ptHard) {
175 printf("AliMCAnalysisUtils::PythiaEventHeader: Njets: %d, pT Hard %2.2f, pycell jet pT %2.2f, rejection factor %1.1f\n",
176 nTriggerJets, ptHard, jet->Pt(), fPtHardAndJetPtFactor);
187 //____________________________________________________________________________
188 AliStack* AliCaloTrackReader::GetStack() const {
189 //Return pointer to stack
193 if(fDebug > 1) printf("AliCaloTrackReader::GetStack() - Stack is not available\n");
198 //____________________________________________________________________________
199 AliHeader* AliCaloTrackReader::GetHeader() const {
200 //Return pointer to header
202 return fMC->Header();
204 printf("AliCaloTrackReader::Header is not available\n");
208 //____________________________________________________________________________
209 AliGenEventHeader* AliCaloTrackReader::GetGenEventHeader() const {
210 //Return pointer to Generated event header
212 return fMC->GenEventHeader();
214 printf("AliCaloTrackReader::GenEventHeader is not available\n");
219 //____________________________________________________________________________
220 TClonesArray* AliCaloTrackReader::GetAODMCParticles(Int_t input) const {
221 //Return list of particles in AOD. Do it for the corresponding input event.
223 TClonesArray * rv = NULL ;
224 if(fDataType == kAOD){
228 AliAODEvent * evt = dynamic_cast<AliAODEvent*> (fInputEvent) ;
230 rv = (TClonesArray*)evt->FindListObject("mcparticles");
232 printf("AliCaloTrackReader::GetAODMCParticles() - wrong AOD input index? %d, or non existing tree? \n",input);
234 } //else if(input == 1 && fSecondInputAODEvent){ //Second input AOD
235 // rv = (TClonesArray*) fSecondInputAODEvent->FindListObject("mcparticles");
239 printf("AliCaloTrackReader::GetAODMCParticles() - Input are not AODs\n");
245 //____________________________________________________________________________
246 AliAODMCHeader* AliCaloTrackReader::GetAODMCHeader(Int_t input) const {
247 //Return MC header in AOD. Do it for the corresponding input event.
248 AliAODMCHeader *mch = NULL;
249 if(fDataType == kAOD){
252 mch = (AliAODMCHeader*)((AliAODEvent*)fInputEvent)->FindListObject("mcheader");
254 // //Second input AOD
255 // else if(input == 1){
256 // mch = (AliAODMCHeader*) fSecondInputAODEvent->FindListObject("mcheader");
259 printf("AliCaloTrackReader::GetAODMCHeader() - wrong AOD input index, %d\n",input);
263 printf("AliCaloTrackReader::GetAODMCHeader() - Input are not AODs\n");
269 //_______________________________________________________________
270 void AliCaloTrackReader::Init()
272 //Init reader. Method to be called in AliAnaPartCorrMaker
274 //Get the file with second input events if the filename is given
275 //Get the tree and connect the AODEvent. Only with AODs
277 if(fReadStack && fReadAODMCParticles){
278 printf("AliCaloTrackReader::Init() - Cannot access stack and mcparticles at the same time, change them \n");
280 fReadAODMCParticles = kFALSE;
283 // if(fSecondInputFileName!=""){
284 // if(fDataType == kAOD){
285 // TFile * input2 = new TFile(fSecondInputFileName,"read");
286 // printf("AliCaloTrackReader::Init() - Second input file opened: %s, size %d \n", input2->GetName(), (Int_t) input2->GetSize());
287 // fSecondInputAODTree = (TTree*) input2->Get("aodTree");
288 // if(fSecondInputAODTree) printf("AliCaloTrackReader::Init() - Second input tree opened: %s, entries %d \n",
289 // fSecondInputAODTree->GetName(), (Int_t) fSecondInputAODTree->GetEntries());
291 // printf("AliCaloTrackReader::Init() - Second input tree not available, STOP \n");
294 // fSecondInputAODEvent = new AliAODEvent;
295 // fSecondInputAODEvent->ReadFromTree(fSecondInputAODTree);
296 // if(fSecondInputFirstEvent >= fSecondInputAODTree->GetEntriesFast()){
297 // printf("AliCaloTrackReader::Init() - Requested first event of second input %d, is larger than number of events %d, STOP\n",
298 // fSecondInputFirstEvent, (Int_t) fSecondInputAODTree->GetEntriesFast());
302 // else printf("AliCaloTrackReader::Init() - Second input not added, reader is not AOD\n");
306 //_______________________________________________________________
307 void AliCaloTrackReader::InitParameters()
309 //Initialize the parameters of the analysis.
315 //Do not filter the detectors input by default.
319 fFillEMCALCells = kFALSE;
320 fFillPHOSCells = kFALSE;
322 //fSecondInputFileName = "" ;
323 //fSecondInputFirstEvent = 0 ;
324 fReadStack = kFALSE; // Check in the constructor of the other readers if it was set or in the configuration file
325 fReadAODMCParticles = kFALSE; // Check in the constructor of the other readers if it was set or in the configuration file
326 fDeltaAODFileName = "deltaAODPartCorr.root";
327 fFiredTriggerClassName = "";
331 //We want tracks fitted in the detectors:
332 //fTrackStatus=AliESDtrack::kTPCrefit;
333 //fTrackStatus|=AliESDtrack::kITSrefit;
335 fESDtrackCuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010();
337 fV0ADC[0] = 0; fV0ADC[1] = 0;
338 fV0Mul[0] = 0; fV0Mul[1] = 0;
343 fCentralityBin[0]=fCentralityBin[1]=-1;
346 fSmearClusterEnergy = kFALSE;
347 fSmearClusterParam[0] = 0.07; // * sqrt E term
348 fSmearClusterParam[1] = 0.02; // * E term
349 fSmearClusterParam[2] = 0.00; // constant
353 //________________________________________________________________
354 void AliCaloTrackReader::Print(const Option_t * opt) const
357 //Print some relevant parameters set for the analysis
361 printf("***** Print: %s %s ******\n", GetName(), GetTitle() ) ;
362 printf("Task name : %s\n", fTaskName.Data()) ;
363 printf("Data type : %d\n", fDataType) ;
364 printf("CTS Min pT : %2.1f GeV/c\n", fCTSPtMin) ;
365 printf("EMCAL Min pT : %2.1f GeV/c\n", fEMCALPtMin) ;
366 printf("PHOS Min pT : %2.1f GeV/c\n", fPHOSPtMin) ;
367 printf("Use CTS = %d\n", fFillCTS) ;
368 printf("Use EMCAL = %d\n", fFillEMCAL) ;
369 printf("Use PHOS = %d\n", fFillPHOS) ;
370 printf("Use EMCAL Cells = %d\n", fFillEMCALCells) ;
371 printf("Use PHOS Cells = %d\n", fFillPHOSCells) ;
372 printf("Track status = %d\n", (Int_t) fTrackStatus) ;
373 printf("Track Mult Eta Cut = %d\n", (Int_t) fTrackMultEtaCut) ;
374 printf("Write delta AOD = %d\n", fWriteOutputDeltaAOD) ;
376 if(fComparePtHardAndJetPt)
377 printf("Compare jet pt and pt hard to accept event, factor = %2.2f",fPtHardAndJetPtFactor);
379 // if(fSecondInputFileName!="") {
380 // printf("Second Input File Name = %s\n", fSecondInputFileName.Data()) ;
381 // printf("Second Input First Event = %d\n", fSecondInputFirstEvent) ;
384 printf("Read Kine from, stack? %d, AOD ? %d \n", fReadStack, fReadAODMCParticles) ;
385 printf("Delta AOD File Name = %s\n", fDeltaAODFileName.Data()) ;
386 printf("Centrality: Class %s, Option %d, Bin [%d,%d] \n", fCentralityClass.Data(),fCentralityOpt,fCentralityBin[0], fCentralityBin[1]) ;
392 //___________________________________________________
393 Bool_t AliCaloTrackReader::FillInputEvent(const Int_t iEntry, const char * /*currentFileName*/) {
394 //Fill the event counter and input lists that are needed, called by the analysis maker.
396 fEventNumber = iEntry;
397 //fCurrentFileName = TString(currentFileName);
399 if(fDebug >= 0) printf("AliCaloTrackReader::FillInputEvent() - Input event not available, skip event analysis\n");
402 //Select events only fired by a certain trigger configuration if it is provided
404 if(fInputEvent->GetHeader())
405 eventType = ((AliVHeader*)fInputEvent->GetHeader())->GetEventType();
407 if( fFiredTriggerClassName !="" && !fAnaLED){
409 return kFALSE; //Only physics event, do not use for simulated events!!!
411 printf("AliCaloTrackReader::FillInputEvent() - FiredTriggerClass <%s>, selected class <%s>, compare name %d\n",
412 GetFiredTriggerClasses().Data(),fFiredTriggerClassName.Data(), GetFiredTriggerClasses().Contains(fFiredTriggerClassName));
413 if( !GetFiredTriggerClasses().Contains(fFiredTriggerClassName) ) return kFALSE;
414 else if(fDebug > 0) printf("AliCaloTrackReader::FillInputEvent() - Accepted triggered event\n");
417 // kStartOfRun = 1, // START_OF_RUN
418 // kEndOfRun = 2, // END_OF_RUN
419 // kStartOfRunFiles = 3, // START_OF_RUN_FILES
420 // kEndOfRunFiles = 4, // END_OF_RUN_FILES
421 // kStartOfBurst = 5, // START_OF_BURST
422 // kEndOfBurst = 6, // END_OF_BURST
423 // kPhysicsEvent = 7, // PHYSICS_EVENT
424 // kCalibrationEvent = 8, // CALIBRATION_EVENT
425 // kFormatError = 9, // EVENT_FORMAT_ERROR
426 // kStartOfData = 10, // START_OF_DATA
427 // kEndOfData = 11, // END_OF_DATA
428 // kSystemSoftwareTriggerEvent = 12, // SYSTEM_SOFTWARE_TRIGGER_EVENT
429 // kDetectorSoftwareTriggerEvent = 13 // DETECTOR_SOFTWARE_TRIGGER_EVENT
431 if(eventType!=7 && fDebug > 1 )printf("AliCaloTrackReader::FillInputEvent() - DO LED, Event Type <%d>, 8 Calibration \n", eventType);
432 if(eventType!=8)return kFALSE;
435 //In case of analysis of events with jets, skip those with jet pt > 5 pt hard
436 if(fComparePtHardAndJetPt && GetStack()) {
437 if(!ComparePtHardAndJetPt()) return kFALSE ;
440 //In case of mixing events with other AOD file
441 // if(fDataType == kAOD && fSecondInputAODTree){
444 // printf("AliCaloTrackReader::FillInputEvent() - Get event %d from second input AOD file \n", iEntry+fSecondInputFirstEvent);
445 // if(fSecondInputAODTree->GetEntriesFast() <= iEntry+fSecondInputFirstEvent) {
446 // if(fSecondInputAODTree->GetEntriesFast() == iEntry+fSecondInputFirstEvent)
447 // printf("AliCaloTrackReader::FillInputEvent() - Skip events from event %d, no more events in second AOD file \n", iEntry);
452 // Int_t nbytes = fSecondInputAODTree->GetEvent(iEntry+fSecondInputFirstEvent);
453 // if ( nbytes == 0 ) {//If nothing in AOD
454 // printf("AliCaloTrackReader::FillInputEvent() - Nothing in Second AOD input, STOP\n");
462 //Reject events with Z vertex too large, only for SE analysis, if not, cut on the analysis code
463 if(!GetMixedEvent() && TMath::Abs(fVertex[0][2]) > fZvtxCut) return kFALSE;
465 //------------------------------------------------------
466 //Event rejection depending on vertex, pileup, v0and
467 //------------------------------------------------------
468 if(fDoEventSelection){
469 if(!fCaloFilterPatch){
470 //Do not analyze events with pileup
471 Bool_t bPileup = fInputEvent->IsPileupFromSPD(3, 0.8, 3., 2., 5.); //Default values, if not it does not compile
472 //Bool_t bPileup = event->IsPileupFromSPD();
473 if(bPileup) return kFALSE;
475 if(fDoV0ANDEventSelection){
476 Bool_t bV0AND = kTRUE;
477 AliESDEvent* esd = dynamic_cast<AliESDEvent*> (fInputEvent);
479 bV0AND = fTriggerAnalysis->IsOfflineTriggerFired(esd, AliTriggerAnalysis::kV0AND);
480 //else bV0AND = //FIXME FOR AODs
481 if(!bV0AND) return kFALSE;
484 if(fUseEventsWithPrimaryVertex && !CheckForPrimaryVertex()) return kFALSE;
488 if(fInputEvent->GetNumberOfCaloClusters() > 0) {
489 AliVCluster * calo = fInputEvent->GetCaloCluster(0);
490 if(calo->GetNLabels() == 4){
491 Int_t * selection = calo->GetLabels();
492 Bool_t bPileup = selection[0];
493 if(bPileup) return kFALSE;
495 Bool_t bGoodV = selection[1];
496 if(fUseEventsWithPrimaryVertex && !bGoodV) return kFALSE;
498 if(fDoV0ANDEventSelection){
499 Bool_t bV0AND = selection[2];
500 if(!bV0AND) return kFALSE;
503 fTrackMult = selection[3];
504 if(fTrackMult == 0) return kFALSE;
506 //First filtered AODs, track multiplicity stored there.
507 fTrackMult = (Int_t) ((AliAODHeader*)fInputEvent->GetHeader())->GetCentrality();
508 if(fTrackMult == 0) return kFALSE;
510 }//at least one cluster
512 //printf("AliCaloTrackReader::FillInputEvent() - No clusters in event\n");
513 //Remove events with vertex (0,0,0), bad vertex reconstruction
514 if(fUseEventsWithPrimaryVertex && TMath::Abs(fVertex[0][0]) < 1.e-6 && TMath::Abs(fVertex[0][1]) < 1.e-6 && TMath::Abs(fVertex[0][2]) < 1.e-6) return kFALSE;
516 //First filtered AODs, track multiplicity stored there.
517 fTrackMult = (Int_t) ((AliAODHeader*)fInputEvent->GetHeader())->GetCentrality();
518 if(fTrackMult == 0) return kFALSE;
520 }// CaloFileter patch
522 //------------------------------------------------------
524 //Check if there is a centrality value, PbPb analysis, and if a centrality bin selection is requested
525 //If we need a centrality bin, we select only those events in the corresponding bin.
526 if(GetCentrality() && fCentralityBin[0]>=0 && fCentralityBin[1]>=0 && fCentralityOpt==100){
527 Int_t cen = GetEventCentrality();
528 if(cen > fCentralityBin[1] || cen < fCentralityBin[0]) return kFALSE; //reject events out of bin.
531 //Fill the arrays with cluster/tracks/cells data
533 FillInputEMCALCells();
535 FillInputPHOSCells();
539 //Accept events with at least one track
540 if(fTrackMult == 0 && fDoEventSelection) return kFALSE;
553 //__________________________________________________
554 void AliCaloTrackReader::ResetLists() {
555 // Reset lists, called by the analysis maker
557 if(fCTSTracks) fCTSTracks -> Clear();
558 if(fEMCALClusters) fEMCALClusters -> Clear("C");
559 if(fPHOSClusters) fPHOSClusters -> Clear("C");
560 // if(fEMCALCells) fEMCALCells -> Clear("");
561 // if(fPHOSCells) fPHOSCells -> Clear("");
563 fV0ADC[0] = 0; fV0ADC[1] = 0;
564 fV0Mul[0] = 0; fV0Mul[1] = 0;
568 //____________________________________________________________________________
569 void AliCaloTrackReader::SetInputEvent(AliVEvent* const input)
572 fMixedEvent = dynamic_cast<AliMixedEvent*>(GetInputEvent()) ;
574 fNMixedEvent = fMixedEvent->GetNumberOfEvents() ;
577 //Delete previous vertex
579 for (Int_t i = 0; i < fNMixedEvent; i++) {
580 delete [] fVertex[i] ;
585 fVertex = new Double_t*[fNMixedEvent] ;
586 for (Int_t i = 0; i < fNMixedEvent; i++) {
587 fVertex[i] = new Double_t[3] ;
588 fVertex[i][0] = 0.0 ;
589 fVertex[i][1] = 0.0 ;
590 fVertex[i][2] = 0.0 ;
594 //__________________________________________________
595 Int_t AliCaloTrackReader::GetEventCentrality() const {
596 //Return current event centrality
599 if(fCentralityOpt==100) return (Int_t) GetCentrality()->GetCentralityPercentile(fCentralityClass); // 100 bins max
600 else if(fCentralityOpt==10) return GetCentrality()->GetCentralityClass10(fCentralityClass);// 10 bins max
601 else if(fCentralityOpt==20) return GetCentrality()->GetCentralityClass5(fCentralityClass); // 20 bins max
603 printf("AliAnaPartCorrBaseClass::Unknown centrality option %d, use 10, 20 or 100\n",fCentralityOpt);
611 //____________________________________________________________________________
612 void AliCaloTrackReader::GetVertex(Double_t vertex[3]) const {
613 //Return vertex position to be used for single event analysis
614 vertex[0]=fVertex[0][0];
615 vertex[1]=fVertex[0][1];
616 vertex[2]=fVertex[0][2];
619 //____________________________________________________________________________
620 void AliCaloTrackReader::GetVertex(Double_t vertex[3], const Int_t evtIndex) const {
621 //Return vertex position for mixed event, recover the vertex in a particular event.
623 //Int_t evtIndex = 0; // for single events only one vertex stored in position 0, default value
624 //if (fMixedEvent && clusterID >=0) {
625 // evtIndex=GetMixedEvent()->EventIndexForCaloCluster(clusterID) ;
628 vertex[0]=fVertex[evtIndex][0]; vertex[1]=fVertex[evtIndex][1]; vertex[2]=fVertex[evtIndex][2];
634 //____________________________________________________________________________
635 void AliCaloTrackReader::FillVertexArray() {
637 //Fill data member with vertex
638 //In case of Mixed event, multiple vertices
640 //Delete previous vertex
642 for (Int_t i = 0; i < fNMixedEvent; i++) {
643 delete [] fVertex[i] ;
648 fVertex = new Double_t*[fNMixedEvent] ;
649 for (Int_t i = 0; i < fNMixedEvent; i++) {
650 fVertex[i] = new Double_t[3] ;
651 fVertex[i][0] = 0.0 ;
652 fVertex[i][1] = 0.0 ;
653 fVertex[i][2] = 0.0 ;
656 if (!fMixedEvent) { //Single event analysis
659 if(fInputEvent->GetPrimaryVertex()){
660 fInputEvent->GetPrimaryVertex()->GetXYZ(fVertex[0]);
663 printf("AliCaloTrackReader::FillVertexArray() - NULL primary vertex\n");
664 fVertex[0][0]=0.; fVertex[0][1]=0.; fVertex[0][2]=0.;
665 }//Primary vertex pointer do not exist
667 } else {//MC read event
668 fVertex[0][0]=0.; fVertex[0][1]=0.; fVertex[0][2]=0.;
672 printf("AliCaloTrackReader::FillVertexArray() - Single Event Vertex : %f,%f,%f\n",fVertex[0][0],fVertex[0][1],fVertex[0][2]);
674 } else { // MultiEvent analysis
675 for (Int_t iev = 0; iev < fNMixedEvent; iev++) {
676 if (fMixedEvent->GetVertexOfEvent(iev))
677 fMixedEvent->GetVertexOfEvent(iev)->GetXYZ(fVertex[iev]);
678 else { // no vertex found !!!!
679 AliWarning("No vertex found");
683 printf("AliCaloTrackReader::FillVertexArray() - Multi Event %d Vertex : %f,%f,%f\n",iev,fVertex[iev][0],fVertex[iev][1],fVertex[iev][2]);
690 //____________________________________________________________________________
691 void AliCaloTrackReader::FillInputCTS() {
692 //Return array with Central Tracking System (CTS) tracks
694 if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputCTS()\n");
696 Int_t nTracks = fInputEvent->GetNumberOfTracks() ;
700 for (Int_t itrack = 0; itrack < nTracks; itrack++) {////////////// track loop
701 AliVTrack * track = (AliVTrack*)fInputEvent->GetTrack(itrack) ; // retrieve track from esd
703 //Select tracks under certain conditions, TPCrefit, ITSrefit ... check the set bits
704 if (fTrackStatus && !((track->GetStatus() & fTrackStatus) == fTrackStatus))
709 if(fDataType==kESD && !fESDtrackCuts->AcceptTrack((AliESDtrack*)track)) continue;
711 // Track filter selection
712 //if (fTrackFilter) {
713 // selectInfo = fTrackFilter->IsSelected(esdTrack);
714 // if (!selectInfo && !(esd->GetPrimaryVertex())->UsesTrack(esdTrack->GetID())) continue;
717 //Count the tracks in eta < 0.9
718 //printf("Eta %f cut %f\n",TMath::Abs(track->Eta()),fTrackMultEtaCut);
719 if(TMath::Abs(track->Eta())< fTrackMultEtaCut) fTrackMult++;
721 track->GetPxPyPz(p) ;
722 TLorentzVector momentum(p[0],p[1],p[2],0);
724 if(fCTSPtMin < momentum.Pt()){
726 if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"CTS"))
729 if(fDebug > 2 && momentum.Pt() > 0.1)
730 printf("AliCaloTrackReader::FillInputCTS() - Selected tracks E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
731 momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
734 track->SetID(itrack);
737 fCTSTracks->Add(track);
739 }//Pt and Fiducial cut passed.
742 //fCTSTracksNormalInputEntries = fCTSTracks->GetEntriesFast();
744 printf("AliCaloTrackReader::FillInputCTS() - aod entries %d, input tracks %d, pass status %d, multipliticy %d\n", fCTSTracks->GetEntriesFast(), nTracks, nstatus, fTrackMult);//fCTSTracksNormalInputEntries);
746 // //If second input event available, add the clusters.
747 // if(fSecondInputAODTree && fSecondInputAODEvent){
748 // nTracks = fSecondInputAODEvent->GetNumberOfTracks() ;
749 // if(fDebug > 1) printf("AliCaloTrackReader::FillInputCTS() - Add second input tracks, entries %d\n", nTracks) ;
750 // for (Int_t itrack = 0; itrack < nTracks; itrack++) {////////////// track loop
751 // AliAODTrack * track = ((AliAODEvent*)fSecondInputAODEvent)->GetTrack(itrack) ; // retrieve track from esd
753 // //Select tracks under certain conditions, TPCrefit, ITSrefit ... check the set bits
754 // if (fTrackStatus && !((track->GetStatus() & fTrackStatus) == fTrackStatus)) continue ;
756 // track->GetPxPyPz(p) ;
757 // TLorentzVector momentum(p[0],p[1],p[2],0);
759 // if(fCTSPtMin < momentum.Pt()){
761 // if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"CTS")) continue;
763 // if(fDebug > 2 && momentum.Pt() > 0.1) printf("AliCaloTrackReader::FillInputCTS() - Selected tracks E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
764 // momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
766 // fCTSTracks->Add(track);
768 // }//Pt and Fiducial cut passed.
771 // if(fDebug > 1) printf("AliCaloTrackReader::FillInputCTS() - aod normal entries %d, after second input %d\n", fCTSTracksNormalInputEntries, fCTSTracks->GetEntriesFast());
772 // } //second input loop
776 //____________________________________________________________________________
777 void AliCaloTrackReader::FillInputEMCALAlgorithm(AliVCluster * clus, const Int_t iclus) {
778 //Fill the EMCAL data in the array, do it
782 vindex = fMixedEvent->EventIndexForCaloCluster(iclus);
784 //Check if the cluster contains any bad channel and if close to calorimeter borders
785 if(GetCaloUtils()->ClusterContainsBadChannel("EMCAL",clus->GetCellsAbsId(), clus->GetNCells()))
787 if(!GetCaloUtils()->CheckCellFiducialRegion(clus, (AliVCaloCells*)fInputEvent->GetEMCALCells(), fInputEvent, vindex))
790 //Remove suspicious clusters
791 if(fRemoveSuspiciousClusters){
792 Int_t ncells = clus->GetNCells();
793 Float_t energy = clus->E();
794 Float_t minNCells = 1+energy/3;//-x*x*0.0033
795 if(ncells < minNCells) {
796 //if(energy > 2)printf("AliCaloTrackReader::FillInputEMCALAlgorithm() - Remove cluster: e %2.2f, Ncells %d, min Ncells %2.1f\n",energy,ncells,minNCells);
800 // if(energy > 2)printf("AliCaloTrackReader::FillInputEMCALAlgorithm() - Keep cluster: e %2.2f, Ncells %d, min Ncells %2.1f\n",energy,ncells,minNCells);
804 if(fSelectEmbeddedClusters){
805 if(clus->GetNLabels()==0) return;
806 //else printf("Embedded cluster, %d, n label %d label %d \n",iclus,clus->GetNLabels(),clus->GetLabel());
809 TLorentzVector momentum ;
811 clus->GetMomentum(momentum, fVertex[vindex]);
813 if(fEMCALPtMin < momentum.E()){
815 if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"EMCAL"))
818 if(fDebug > 2 && momentum.E() > 0.1)
819 printf("AliCaloTrackReader::FillInputEMCAL() - Selected clusters E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
820 momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
823 //clus->GetPosition(pos);
824 //printf("Before Corrections: e %f, x %f, y %f, z %f\n",clus->E(),pos[0],pos[1],pos[2]);
826 //Recalibrate the cluster energy
827 if(GetCaloUtils()->IsRecalibrationOn()) {
828 Float_t energy = GetCaloUtils()->RecalibrateClusterEnergy(clus, GetEMCALCells());
830 //printf("Recalibrated Energy %f\n",clus->E());
831 GetCaloUtils()->RecalculateClusterShowerShapeParameters(GetEMCALCells(),clus);
832 GetCaloUtils()->RecalculateClusterPID(clus);
836 //Recalculate distance to bad channels, if new list of bad channels provided
837 GetCaloUtils()->RecalculateClusterDistanceToBadChannel(GetEMCALCells(),clus);
839 //Recalculate cluster position
840 if(GetCaloUtils()->IsRecalculationOfClusterPositionOn()){
841 GetCaloUtils()->RecalculateClusterPosition(GetEMCALCells(),clus);
842 //clus->GetPosition(pos);
843 //printf("After Corrections: e %f, x %f, y %f, z %f\n",clus->E(),pos[0],pos[1],pos[2]);
846 //Correct non linearity
847 if(GetCaloUtils()->IsCorrectionOfClusterEnergyOn()){
848 GetCaloUtils()->CorrectClusterEnergy(clus) ;
849 //printf("Linearity Corrected Energy %f\n",clus->E());
852 //In case of MC analysis, to match resolution/calibration in real data
853 if(fSmearClusterEnergy){
854 Float_t energy = clus->E();
855 Float_t rdmEnergy = fRandom.Gaus(energy,fSmearClusterParam[0]*TMath::Sqrt(energy)+
856 fSmearClusterParam[1]*energy+fSmearClusterParam[2]);
857 clus->SetE(rdmEnergy);
858 if(fDebug > 2) printf("\t Energy %f, smeared %f\n", energy, clus->E());
864 fEMCALClusters->Add(clus);
868 //____________________________________________________________________________
869 void AliCaloTrackReader::FillInputEMCAL() {
870 //Return array with EMCAL clusters in aod format
872 if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputEMCAL()\n");
874 //Loop to select clusters in fiducial cut and fill container with aodClusters
875 if(fEMCALClustersListName==""){
876 Int_t nclusters = fInputEvent->GetNumberOfCaloClusters();
877 for (Int_t iclus = 0; iclus < nclusters; iclus++) {
878 AliVCluster * clus = 0;
879 if ( (clus = fInputEvent->GetCaloCluster(iclus)) ) {
880 if (IsEMCALCluster(clus)){
881 FillInputEMCALAlgorithm(clus, iclus);
886 //Recalculate track matching
887 if(fDataType==kESD)GetCaloUtils()->RecalculateClusterTrackMatching(fInputEvent);
889 }//Get the clusters from the input event
891 TClonesArray * clusterList = dynamic_cast<TClonesArray*> (fOutputEvent->FindListObject(fEMCALClustersListName));
893 printf("AliCaloTrackReader::FillInputEMCAL() - Wrong name of list with clusters? <%s>\n",fEMCALClustersListName.Data());
896 Int_t nclusters = clusterList->GetEntriesFast();
897 for (Int_t iclus = 0; iclus < nclusters; iclus++) {
898 AliVCluster * clus = dynamic_cast<AliVCluster*> (clusterList->At(iclus));
899 //printf("E %f\n",clus->E());
900 if (clus) FillInputEMCALAlgorithm(clus, iclus);
901 else printf("AliCaloTrackReader::FillInputEMCAL() - Null cluster in list!\n");
905 //Recalculate track matching, not necessary, already done in the reclusterization task
906 //GetCaloUtils()->RecalculateClusterTrackMatching(fInputEvent,clusterList);
910 //fEMCALClustersNormalInputEntries = fEMCALClusters->GetEntriesFast();
911 if(fDebug > 1) printf("AliCaloTrackReader::FillInputEMCAL() - aod entries %d\n", fEMCALClusters->GetEntriesFast());//fEMCALClustersNormalInputEntries);
913 //If second input event available, add the clusters.
914 // if(fSecondInputAODTree && fSecondInputAODEvent){
915 // GetSecondInputAODVertex(v);
916 // nclusters = ((AliAODEvent*)fSecondInputAODEvent)->GetNumberOfCaloClusters();
917 // if(fDebug > 1) printf("AliCaloTrackReader::FillInputEMCAL() - Add second input clusters, entries %d\n", nclusters) ;
918 // for (Int_t iclus = 0; iclus < nclusters; iclus++) {
919 // AliAODCaloCluster * clus = 0;
920 // if ( (clus = ((AliAODEvent*)fSecondInputAODEvent)->GetCaloCluster(iclus)) ) {
921 // if (clus->IsEMCAL()){
922 // TLorentzVector momentum ;
923 // clus->GetMomentum(momentum, v);
925 // if(fEMCALPtMin < momentum.Pt()){
927 // if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"EMCAL")) continue;
929 // if(fDebug > 2 && momentum.E() > 0.1) printf("AliCaloTrackReader::FillInputEMCAL() - Selected clusters E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
930 // momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
931 // fEMCALClusters->Add(clus);
932 // }//Pt and Fiducial cut passed.
934 // }// cluster exists
937 // if(fDebug > 1) printf("AliCaloTrackReader::FillInputEMCAL() - aod normal entries %d, after second input %d\n", fEMCALClustersNormalInputEntries, fEMCALClusters->GetEntriesFast());
939 // } //second input loop
942 //____________________________________________________________________________
943 void AliCaloTrackReader::FillInputPHOS() {
944 //Return array with PHOS clusters in aod format
946 if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputPHOS()\n");
948 //Loop to select clusters in fiducial cut and fill container with aodClusters
949 Int_t nclusters = fInputEvent->GetNumberOfCaloClusters();
950 for (Int_t iclus = 0; iclus < nclusters; iclus++) {
951 AliVCluster * clus = 0;
952 if ( (clus = fInputEvent->GetCaloCluster(iclus)) ) {
953 if (IsPHOSCluster(clus)){
954 //Check if the cluster contains any bad channel and if close to calorimeter borders
957 vindex = fMixedEvent->EventIndexForCaloCluster(iclus);
958 if( GetCaloUtils()->ClusterContainsBadChannel("PHOS",clus->GetCellsAbsId(), clus->GetNCells()))
960 if(!GetCaloUtils()->CheckCellFiducialRegion(clus, fInputEvent->GetPHOSCells(), fInputEvent, vindex))
963 TLorentzVector momentum ;
965 clus->GetMomentum(momentum, fVertex[vindex]);
967 if(fPHOSPtMin < momentum.E()){
969 if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"PHOS"))
972 if(fDebug > 2 && momentum.E() > 0.1)
973 printf("AliCaloTrackReader::FillInputPHOS() - Selected clusters E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
974 momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
976 //Recalibrate the cluster energy
977 if(GetCaloUtils()->IsRecalibrationOn()) {
978 Float_t energy = GetCaloUtils()->RecalibrateClusterEnergy(clus, (AliAODCaloCells*)GetPHOSCells());
986 fPHOSClusters->Add(clus);
988 }//Pt and Fiducial cut passed.
993 //fPHOSClustersNormalInputEntries = fPHOSClusters->GetEntriesFast() ;
994 if(fDebug > 1) printf("AliCaloTrackReader::FillInputPHOS() - aod entries %d\n", fPHOSClusters->GetEntriesFast());//fPHOSClustersNormalInputEntries);
996 //If second input event available, add the clusters.
997 // if(fSecondInputAODTree && fSecondInputAODEvent){
998 // GetSecondInputAODVertex(v);
999 // nclusters = ((AliAODEvent*)fSecondInputAODEvent)->GetNumberOfCaloClusters();
1000 // if(fDebug > 1) printf("AliCaloTrackReader::FillInputPHOS() - Add second input clusters, entries %d\n", nclusters);
1001 // for (Int_t iclus = 0; iclus < nclusters; iclus++) {
1002 // AliAODCaloCluster * clus = 0;
1003 // if ( (clus = ((AliAODEvent*)fSecondInputAODEvent)->GetCaloCluster(iclus)) ) {
1004 // if (clus->IsPHOS()){
1005 // TLorentzVector momentum ;
1006 // clus->GetMomentum(momentum, v);
1008 // if(fPHOSPtMin < momentum.Pt()){
1010 // if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"PHOS")) continue;
1012 // if(fDebug > 2 && momentum.E() > 0.1) printf("AliCaloTrackReader::FillInputPHOS() - Selected clusters E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
1013 // momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
1014 // fPHOSClusters->Add(clus);
1015 // }//Pt and Fiducial cut passed.
1017 // }// cluster exists
1019 // if(fDebug > 1) printf("AliCaloTrackReader::FillInputPHOS() - aod normal entries %d, after second input %d\n", fPHOSClustersNormalInputEntries, fPHOSClusters->GetEntriesFast());
1020 // } //second input loop
1024 //____________________________________________________________________________
1025 void AliCaloTrackReader::FillInputEMCALCells() {
1026 //Return array with EMCAL cells in aod format
1028 fEMCALCells = fInputEvent->GetEMCALCells();
1032 //____________________________________________________________________________
1033 void AliCaloTrackReader::FillInputPHOSCells() {
1034 //Return array with PHOS cells in aod format
1036 fPHOSCells = fInputEvent->GetPHOSCells();
1040 //____________________________________________________________________________
1041 void AliCaloTrackReader::FillInputVZERO(){
1042 //Fill VZERO information in data member, add all the channels information.
1043 AliVVZERO* v0 = fInputEvent->GetVZEROData();
1044 //printf("Init V0: ADC (%d,%d), Multiplicity (%d,%d) \n",fV0ADC[0],fV0ADC[1],fV0Mul[0],fV0Mul[1]);
1048 AliESDVZERO* esdV0 = dynamic_cast<AliESDVZERO*> (v0);
1049 for (Int_t i = 0; i < 32; i++)
1051 if(esdV0){//Only available in ESDs
1052 fV0ADC[0] += (Int_t)esdV0->GetAdcV0C(i);
1053 fV0ADC[1] += (Int_t)esdV0->GetAdcV0A(i);
1055 fV0Mul[0] += (Int_t)v0->GetMultiplicityV0C(i);
1056 fV0Mul[1] += (Int_t)v0->GetMultiplicityV0A(i);
1059 printf("V0: ADC (%d,%d), Multiplicity (%d,%d) \n",fV0ADC[0],fV0ADC[1],fV0Mul[0],fV0Mul[1]);
1064 printf("Cannot retrieve V0 ESD! Run w/ null V0 charges\n ");
1069 //____________________________________________________________________________
1070 Bool_t AliCaloTrackReader::IsEMCALCluster(AliVCluster* cluster) const {
1071 // Check if it is a cluster from EMCAL. For old AODs cluster type has
1072 // different number and need to patch here
1074 if(fDataType==kAOD && fOldAOD)
1076 if (cluster->GetType() == 2) return kTRUE;
1081 return cluster->IsEMCAL();
1086 //____________________________________________________________________________
1087 Bool_t AliCaloTrackReader::IsPHOSCluster(AliVCluster * cluster) const {
1088 //Check if it is a cluster from PHOS.For old AODs cluster type has
1089 // different number and need to patch here
1091 if(fDataType==kAOD && fOldAOD)
1093 Int_t type = cluster->GetType();
1094 if (type == 0 || type == 1) return kTRUE;
1099 return cluster->IsPHOS();
1104 //____________________________________________________________________________
1105 Bool_t AliCaloTrackReader::CheckForPrimaryVertex(){
1106 //Check if the vertex was well reconstructed, copy from V0Reader of conversion group
1109 AliESDEvent * event = dynamic_cast<AliESDEvent*> (fInputEvent);
1110 if(!event) return kTRUE;
1112 if(event->GetPrimaryVertexTracks()->GetNContributors() > 0) {
1116 if(event->GetPrimaryVertexTracks()->GetNContributors() < 1) {
1118 if(event->GetPrimaryVertexSPD()->GetNContributors() > 0) {
1119 //cout<<"spd vertex type::"<< fESDEvent->GetPrimaryVertex()->GetName() << endl;
1123 if(event->GetPrimaryVertexSPD()->GetNContributors() < 1) {
1124 // cout<<"bad vertex type::"<< event->GetPrimaryVertex()->GetName() << endl;