making the container name parameterized
[u/mrichter/AliRoot.git] / PWG / CaloTrackCorrBase / AliCaloTrackReader.cxx
1 /**************************************************************************
2  * Copyright(c) 1998-1999, ALICE Experiment at CERN, All rights reserved. *
3  *                                                                        *
4  * Author: The ALICE Off-line Project.                                    *
5  * Contributors are mentioned in the code where appropriate.              *
6  *                                                                        *
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  **************************************************************************/
15
16 //_________________________________________________________________________
17 // Base class for reading data: MonteCarlo, ESD or AOD, of PHOS EMCAL and 
18 // Central Barrel Tracking detectors (CTS).
19 // Not all MC particles/tracks/clusters are kept, some kinematical/fiducial restrictions are done.
20 // Mother class of : AliCaloTrackESDReader: Fills ESD data in 3 TObjArrays (PHOS, EMCAL, CTS)
21 //                 : AliCaloTrackMCReader : Fills Kinematics data in 3 TObjArrays (PHOS, EMCAL, CTS)
22 //                 : AliCaloTrackAODReader: Fills AOD data in 3 TObjArrays (PHOS, EMCAL, CTS) 
23 //-- Author: Gustavo Conesa (LNF-INFN) 
24 //////////////////////////////////////////////////////////////////////////////
25
26
27 // --- ROOT system ---
28 #include <TFile.h>
29 #include <TGeoManager.h>
30
31 // ---- ANALYSIS system ----
32 #include "AliMCEvent.h"
33 #include "AliAODMCHeader.h"
34 #include "AliGenPythiaEventHeader.h"
35 #include "AliESDEvent.h"
36 #include "AliAODEvent.h"
37 #include "AliVTrack.h"
38 #include "AliVParticle.h"
39 #include "AliMixedEvent.h"
40 #include "AliESDtrack.h"
41 #include "AliESDtrackCuts.h"
42 #include "AliTriggerAnalysis.h"
43 #include "AliESDVZERO.h"
44 #include "AliVCaloCells.h"
45 #include "AliAnalysisManager.h"
46 #include "AliInputEventHandler.h"
47
48 // ---- Detectors ----
49 #include "AliPHOSGeoUtils.h"
50 #include "AliEMCALGeometry.h"
51 #include "AliEMCALRecoUtils.h"
52
53 // ---- CaloTrackCorr ---
54 #include "AliCalorimeterUtils.h"
55 #include "AliCaloTrackReader.h"
56
57 ClassImp(AliCaloTrackReader)
58   
59   
60 //________________________________________
61 AliCaloTrackReader::AliCaloTrackReader() : 
62 TObject(),                   fEventNumber(-1), //fCurrentFileName(""),
63 fDataType(0),                fDebug(0), 
64 fFiducialCut(0x0),           fCheckFidCut(kFALSE), 
65 fComparePtHardAndJetPt(0),   fPtHardAndJetPtFactor(0),
66 fComparePtHardAndClusterPt(0),fPtHardAndClusterPtFactor(0),
67 fCTSPtMin(0),                fEMCALPtMin(0),                  fPHOSPtMin(0), 
68 fCTSPtMax(0),                fEMCALPtMax(0),                  fPHOSPtMax(0), 
69 fEMCALTimeCutMin(-10000),    fEMCALTimeCutMax(10000),
70 fAODBranchList(0x0),
71 fCTSTracks(0x0),             fEMCALClusters(0x0),             fPHOSClusters(0x0),
72 fEMCALCells(0x0),            fPHOSCells(0x0),
73 fInputEvent(0x0),            fOutputEvent(0x0),fMC(0x0),
74 fFillCTS(0),                 fFillEMCAL(0),                   fFillPHOS(0),
75 fFillEMCALCells(0),          fFillPHOSCells(0), 
76 fRecalculateClusters(kFALSE),fSelectEmbeddedClusters(kFALSE),
77 fTrackStatus(0),             fTrackFilterMask(0),             
78 fESDtrackCuts(0),            fConstrainTrack(kFALSE),         fSelectHybridTracks(0),
79 fTrackMult(0),               fTrackMultEtaCut(0.8),
80 fReadStack(kFALSE),          fReadAODMCParticles(kFALSE), 
81 fDeltaAODFileName(""),       fFiredTriggerClassName(""),      
82 fEventTriggerMask(0),        fMixEventTriggerMask(0),         fEventTriggerAtSE(0), 
83 fAnaLED(kFALSE),
84 fTaskName(""),               fCaloUtils(0x0), 
85 fMixedEvent(NULL),           fNMixedEvent(0),                 fVertex(NULL), 
86 fListMixedTracksEvents(),    fListMixedCaloEvents(),
87 fLastMixedTracksEvent(-1),   fLastMixedCaloEvent(-1),
88 fWriteOutputDeltaAOD(kFALSE),fOldAOD(kFALSE),                 fCaloFilterPatch(kFALSE),
89 fEMCALClustersListName(""),  fZvtxCut(0.),                    
90 fAcceptFastCluster(kFALSE),  fRemoveLEDEvents(kTRUE), 
91 fDoEventSelection(kFALSE),   fDoV0ANDEventSelection(kFALSE),  fUseEventsWithPrimaryVertex(kFALSE),
92 fTriggerAnalysis (0x0), 
93 fCentralityClass(""),        fCentralityOpt(0),
94 fEventPlaneMethod(""),       fImportGeometryFromFile(kFALSE), fImportGeometryFilePath("")
95 {
96   //Ctor
97
98   //Initialize parameters
99   InitParameters();
100 }
101
102 //_______________________________________
103 AliCaloTrackReader::~AliCaloTrackReader() 
104 {
105   //Dtor
106   
107   delete fFiducialCut ;
108         
109   if(fAODBranchList)
110   {
111     fAODBranchList->Delete();
112     delete fAODBranchList ;
113   }  
114   
115   if(fCTSTracks)
116   {
117     if(fDataType!=kMC)fCTSTracks->Clear() ; 
118     else              fCTSTracks->Delete() ; 
119     delete fCTSTracks ;
120   }
121   
122   if(fEMCALClusters)
123   {
124     if(fDataType!=kMC)fEMCALClusters->Clear("C") ; 
125     else              fEMCALClusters->Delete() ; 
126     delete fEMCALClusters ;
127   }
128   
129   if(fPHOSClusters)
130   {
131     if(fDataType!=kMC)fPHOSClusters->Clear("C") ; 
132     else              fPHOSClusters->Delete() ; 
133     delete fPHOSClusters ;
134   }
135   
136   if(fVertex)
137   {
138     for (Int_t i = 0; i < fNMixedEvent; i++) 
139     {
140       delete [] fVertex[i] ;
141       
142     }
143     delete [] fVertex ;
144         }
145   
146   delete fESDtrackCuts;
147   delete fTriggerAnalysis;
148   
149   //  Pointers not owned, done by the analysis frame
150   //  if(fInputEvent)  delete fInputEvent ;
151   //  if(fOutputEvent) delete fOutputEvent ;
152   //  if(fMC)          delete fMC ;  
153   //  Pointer not owned, deleted by maker
154   //  if (fCaloUtils) delete fCaloUtils ;
155   
156 }
157
158 //________________________________________________
159 Bool_t AliCaloTrackReader::ComparePtHardAndJetPt()
160 {
161   // Check the event, if the requested ptHard is much smaller than the jet pT, then there is a problem.
162   // Only for PYTHIA.
163   
164   //printf("AliCaloTrackReader::ComparePtHardAndJetPt() - GenHeaderName : %s\n",GetGenEventHeader()->ClassName());
165   
166   if(!strcmp(GetGenEventHeader()->ClassName(), "AliGenPythiaEventHeader"))
167   {
168     TParticle * jet =  0;
169     AliGenPythiaEventHeader* pygeh= (AliGenPythiaEventHeader*) GetGenEventHeader();
170     Int_t nTriggerJets =  pygeh->NTriggerJets();
171     Float_t ptHard = pygeh->GetPtHard();
172     
173     if(fDebug > 1) 
174       printf("AliCaloTrackReader::ComparePtHardAndJetPt() - Njets: %d, pT Hard %f\n",nTriggerJets, ptHard);
175     
176     Float_t tmpjet[]={0,0,0,0};
177     for(Int_t ijet = 0; ijet< nTriggerJets; ijet++)
178     {
179       pygeh->TriggerJet(ijet, tmpjet);
180       jet = new TParticle(94, 21, -1, -1, -1, -1, tmpjet[0],tmpjet[1],tmpjet[2],tmpjet[3], 0,0,0,0);
181       
182       if(fDebug > 1) 
183         printf("AliCaloTrackReader::ComparePtHardAndJetPt() - jet %d; pycell jet pT %f\n",ijet, jet->Pt());
184       
185       //Compare jet pT and pt Hard
186       if(jet->Pt() > fPtHardAndJetPtFactor * ptHard)
187       {
188         printf("AliCaloTrackReader::ComparePtHardAndJetPt() - Reject jet event with : pT Hard %2.2f, pycell jet pT %2.2f, rejection factor %1.1f\n",
189                 ptHard, jet->Pt(), fPtHardAndJetPtFactor);
190         return kFALSE;
191       }
192     }
193     
194     if(jet) delete jet; 
195   }
196   
197   return kTRUE ;
198   
199 }
200
201 //____________________________________________________________________
202 Bool_t AliCaloTrackReader::ComparePtHardAndClusterPt()
203 {
204   // Check the event, if the requested ptHard is smaller than the calorimeter cluster E, then there is a problem.
205   // Only for PYTHIA.
206   
207   if(!strcmp(GetGenEventHeader()->ClassName(), "AliGenPythiaEventHeader"))
208   {
209     AliGenPythiaEventHeader* pygeh= (AliGenPythiaEventHeader*) GetGenEventHeader();
210     Float_t ptHard = pygeh->GetPtHard();
211     
212     Int_t nclusters = fInputEvent->GetNumberOfCaloClusters();
213     for (Int_t iclus =  0; iclus <  nclusters; iclus++) 
214     {
215       AliVCluster * clus = fInputEvent->GetCaloCluster(iclus) ; 
216       Float_t ecluster = clus->E();
217       
218       if(ecluster > fPtHardAndClusterPtFactor*ptHard) 
219       {
220         printf("AliCaloTrackReader::ComparePtHardAndClusterPt() - Reject : ecluster %2.2f, calo %d, factor %2.2f, ptHard %f\n",ecluster,clus->GetType(),fPtHardAndClusterPtFactor,ptHard);
221
222         return kFALSE;
223       }
224     }
225     
226   }
227   
228   return kTRUE ;
229   
230 }
231
232 //____________________________________________
233 AliStack* AliCaloTrackReader::GetStack() const 
234 {
235   //Return pointer to stack
236   if(fMC)
237     return fMC->Stack();
238   else
239   {
240     if(fDebug > 1) printf("AliCaloTrackReader::GetStack() - Stack is not available\n"); 
241     return 0x0 ;
242   }
243 }
244
245 //__________________________________________________
246 TString AliCaloTrackReader::GetFiredTriggerClasses() 
247
248   // List of triggered classes in a TString
249
250   AliESDEvent* esdevent = dynamic_cast<AliESDEvent*> (GetInputEvent());
251   AliAODEvent* aodevent = dynamic_cast<AliAODEvent*> (GetInputEvent());
252     
253   if     (esdevent) return esdevent->GetFiredTriggerClasses();
254   else if(aodevent) return aodevent->GetFiredTriggerClasses();
255   else              return ""; // Mixed Event, MC event, does not have this trigger info
256
257 }
258
259 //______________________________________________
260 AliHeader* AliCaloTrackReader::GetHeader() const 
261 {
262   //Return pointer to header
263   if(fMC)
264   {
265     return fMC->Header();
266   }
267   else
268   {
269     printf("AliCaloTrackReader::Header is not available\n"); 
270     return 0x0 ;
271   }
272 }
273
274 //______________________________________________________________
275 AliGenEventHeader* AliCaloTrackReader::GetGenEventHeader() const 
276 {
277   //Return pointer to Generated event header
278   if     (ReadStack() && fMC)
279   {
280     return fMC->GenEventHeader();
281   }
282   else if(ReadAODMCParticles() && GetAODMCHeader())
283   {
284     //printf("AliCaloTrackReader::GetGenEventHeader() - N headers %d\n",GetAODMCHeader()->GetNCocktailHeaders());
285     if( GetAODMCHeader()->GetNCocktailHeaders() > 0)
286       return GetAODMCHeader()->GetCocktailHeader(0) ;
287     else 
288       return 0x0;
289   }
290   else 
291   {
292     //printf("AliCaloTrackReader::GetGenEventHeader() - MC header not available! \n");
293     return 0;
294   }
295 }
296
297 //____________________________________________________________________
298 TClonesArray* AliCaloTrackReader::GetAODMCParticles(Int_t input) const 
299 {
300   //Return list of particles in AOD. Do it for the corresponding input event.
301   
302   TClonesArray * rv = NULL ; 
303   if(fDataType == kAOD)
304   {
305     if(input == 0)
306     {
307       //Normal input AOD
308       AliAODEvent * evt = dynamic_cast<AliAODEvent*> (fInputEvent) ;
309       if(evt)
310         rv = (TClonesArray*)evt->FindListObject("mcparticles");
311       else  
312         printf("AliCaloTrackReader::GetAODMCParticles() - wrong AOD input index? %d, or non existing tree? \n",input); 
313     }  
314     
315   } 
316   else 
317   {
318     printf("AliCaloTrackReader::GetAODMCParticles() - Input are not AODs\n"); 
319   }
320   
321   return rv ; 
322 }
323
324 //________________________________________________________
325 AliAODMCHeader* AliCaloTrackReader::GetAODMCHeader() const 
326 {
327   //Return MC header in AOD. Do it for the corresponding input event.
328   
329   AliAODMCHeader *mch = NULL;
330   
331   if(fDataType == kAOD)
332   {
333     AliAODEvent * aod = dynamic_cast<AliAODEvent*> (fInputEvent);
334     if(aod) mch = dynamic_cast<AliAODMCHeader*>(aod->FindListObject("mcHeader"));
335   }
336   else 
337   {
338     printf("AliCaloTrackReader::GetAODMCHeader() - Input are not AODs\n");
339   }
340   
341   return mch;
342 }
343
344 //_____________________________
345 void AliCaloTrackReader::Init()
346 {
347   //Init reader. Method to be called in AliAnaPartCorrMaker
348
349   //printf(" AliCaloTrackReader::Init() %p \n",gGeoManager);
350
351   if(fReadStack && fReadAODMCParticles)
352   {
353     printf("AliCaloTrackReader::Init() - Cannot access stack and mcparticles at the same time, change them \n");
354     fReadStack          = kFALSE;
355     fReadAODMCParticles = kFALSE;
356   }
357   
358   // Init geometry, I do not like much to do it like this ...
359   if(fImportGeometryFromFile && !gGeoManager) 
360   {
361     if(fImportGeometryFilePath=="") // If not specified, set a default location
362     fImportGeometryFilePath = "$ALICE_ROOT/OADB/EMCAL/geometry_2011.root"; // "$ALICE_ROOT/EVE/alice-data/default_geo.root"
363
364     printf("AliCaloTrackReader::Init() - Import %s\n",fImportGeometryFilePath.Data());
365     TGeoManager::Import(fImportGeometryFilePath) ; // default need file "geometry.root" in local dir!!!!
366   }
367
368 }
369
370 //_______________________________________
371 void AliCaloTrackReader::InitParameters()
372 {
373   //Initialize the parameters of the analysis.
374   fDataType   = kESD ;
375   fCTSPtMin   = 0.1 ;
376   fEMCALPtMin = 0.1 ;
377   fPHOSPtMin  = 0.1 ;
378   fCTSPtMax   = 1000. ;
379   fEMCALPtMax = 1000. ;
380   fPHOSPtMax  = 1000. ;
381   
382   //Do not filter the detectors input by default.
383   fFillEMCAL      = kFALSE;
384   fFillPHOS       = kFALSE;
385   fFillCTS        = kFALSE;
386   fFillEMCALCells = kFALSE;
387   fFillPHOSCells  = kFALSE;
388   
389   fReadStack             = kFALSE; // Check in the constructor of the other readers if it was set or in the configuration file
390   fReadAODMCParticles    = kFALSE; // Check in the constructor of the other readers if it was set or in the configuration file
391   fDeltaAODFileName      = "deltaAODPartCorr.root";
392   fFiredTriggerClassName = "";
393   fEventTriggerMask      = AliVEvent::kAny;
394   fMixEventTriggerMask   = AliVEvent::kAnyINT;
395   fEventTriggerAtSE      = kTRUE; // Use only events that pass event selection at SE base class
396   
397   fAcceptFastCluster = kTRUE;
398   fAnaLED            = kFALSE;
399   
400   //We want tracks fitted in the detectors:
401   //fTrackStatus=AliESDtrack::kTPCrefit;
402   //fTrackStatus|=AliESDtrack::kITSrefit; 
403   fTrackStatus     = 0;
404   fTrackFilterMask = 128; //For AODs, but what is the difference between fTrackStatus and fTrackFilterMask?
405   
406   fESDtrackCuts = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts(); //initialize with TPC only tracks 
407   
408   fConstrainTrack = kFALSE ; // constrain tracks to vertex
409   
410   fV0ADC[0] = 0;   fV0ADC[1] = 0; 
411   fV0Mul[0] = 0;   fV0Mul[1] = 0; 
412   
413   fZvtxCut   = 10.;
414   
415   fNMixedEvent = 1;
416   
417   fPtHardAndJetPtFactor     = 7.;
418   fPtHardAndClusterPtFactor = 1.;
419
420   //Centrality
421   fCentralityClass  = "V0M";
422   fCentralityOpt    = 10;
423   fCentralityBin[0] = fCentralityBin[1]=-1;
424   
425   fEventPlaneMethod = "V0";
426
427   // Allocate memory (not sure this is the right place)
428   fCTSTracks       = new TObjArray();
429   fEMCALClusters   = new TObjArray();
430   fPHOSClusters    = new TObjArray(); 
431   fTriggerAnalysis = new AliTriggerAnalysis;
432   fAODBranchList   = new TList ;
433
434   fImportGeometryFromFile = kFALSE;
435   
436   fPileUpParam[0] = 3   ; fPileUpParam[1] = 0.8 ;
437   fPileUpParam[2] = 3.0 ; fPileUpParam[3] = 2.0 ; fPileUpParam[4] = 5.0;
438   
439 }
440
441
442 //________________________________________________
443 Bool_t AliCaloTrackReader::IsPileUpFromSPD() const
444 {
445   // Check if event is from pile-up determined by SPD
446   // Default values: (3, 0.8, 3., 2., 5.)
447   return fInputEvent->IsPileupFromSPD((Int_t) fPileUpParam[0] , fPileUpParam[1] , 
448                                               fPileUpParam[2] , fPileUpParam[3] , fPileUpParam[4] ); 
449   //printf("Param : %d, %2.2f, %2.2f, %2.2f, %2.2f\n",(Int_t) fPileUpParam[0], fPileUpParam[1], fPileUpParam[2], fPileUpParam[3], fPileUpParam[4]);
450
451 }
452
453 //________________________________________________________
454 void AliCaloTrackReader::Print(const Option_t * opt) const
455 {
456   
457   //Print some relevant parameters set for the analysis
458   if(! opt)
459     return;
460   
461   printf("***** Print: %s %s ******\n",    GetName(), GetTitle() ) ;
462   printf("Task name      : %s\n",          fTaskName.Data()) ;
463   printf("Data type      : %d\n",          fDataType) ;
464   printf("CTS Min pT     : %2.1f GeV/c\n", fCTSPtMin) ;
465   printf("EMCAL Min pT   : %2.1f GeV/c\n", fEMCALPtMin) ;
466   printf("PHOS Min pT    : %2.1f GeV/c\n", fPHOSPtMin) ;
467   printf("CTS Max pT     : %2.1f GeV/c\n", fCTSPtMax) ;
468   printf("EMCAL Max pT   : %2.1f GeV/c\n", fEMCALPtMax) ;
469   printf("PHOS Max pT    : %2.1f GeV/c\n", fPHOSPtMax) ;
470   printf("EMCAL Time Cut: %3.1f < TOF  < %3.1f\n", fEMCALTimeCutMin, fEMCALTimeCutMax);
471   printf("Use CTS         =     %d\n",     fFillCTS) ;
472   printf("Use EMCAL       =     %d\n",     fFillEMCAL) ;
473   printf("Use PHOS        =     %d\n",     fFillPHOS) ;
474   printf("Use EMCAL Cells =     %d\n",     fFillEMCALCells) ;
475   printf("Use PHOS  Cells =     %d\n",     fFillPHOSCells) ;
476   printf("Track status    =     %d\n", (Int_t) fTrackStatus) ;
477   printf("AODs Track filter mask  =  %d or hybrid %d\n", (Int_t) fTrackFilterMask,fSelectHybridTracks) ;
478   printf("Track Mult Eta Cut =  %d\n", (Int_t) fTrackMultEtaCut) ;
479   printf("Write delta AOD =     %d\n",     fWriteOutputDeltaAOD) ;
480   printf("Recalculate Clusters = %d\n",    fRecalculateClusters) ;
481   
482   printf("Use Triggers selected in SE base class %d; If not what trigger Mask? %d; Trigger max for mixed %d \n", 
483          fEventTriggerAtSE, fEventTriggerMask,fMixEventTriggerMask);
484   
485   if(fComparePtHardAndClusterPt)
486           printf("Compare jet pt and pt hard to accept event, factor = %2.2f",fPtHardAndJetPtFactor);
487   
488   if(fComparePtHardAndClusterPt)
489           printf("Compare cluster pt and pt hard to accept event, factor = %2.2f",fPtHardAndClusterPtFactor);
490   
491   printf("Read Kine from, stack? %d, AOD ? %d \n", fReadStack, fReadAODMCParticles) ;
492   printf("Delta AOD File Name =     %s\n", fDeltaAODFileName.Data()) ;
493   printf("Centrality: Class %s, Option %d, Bin [%d,%d] \n", fCentralityClass.Data(),fCentralityOpt,fCentralityBin[0], fCentralityBin[1]) ;
494   
495   printf("    \n") ;
496   
497
498
499 //_________________________________________________________________________
500 Bool_t AliCaloTrackReader::FillInputEvent(const Int_t iEntry, 
501                                           const char * /*currentFileName*/) 
502 {
503   //Fill the event counter and input lists that are needed, called by the analysis maker.
504     
505   fEventNumber = iEntry;
506   //fCurrentFileName = TString(currentFileName);
507   if(!fInputEvent)
508   {
509           if(fDebug >= 0) printf("AliCaloTrackReader::FillInputEvent() - Input event not available, skip event analysis\n");
510           return kFALSE;
511   }
512   
513   //Select events only fired by a certain trigger configuration if it is provided
514   Int_t eventType = 0;
515   if(fInputEvent->GetHeader())
516           eventType = ((AliVHeader*)fInputEvent->GetHeader())->GetEventType();
517   
518   if (GetFiredTriggerClasses().Contains("FAST")  && !GetFiredTriggerClasses().Contains("ALL") && !fAcceptFastCluster) 
519   {
520     if(fDebug > 0)  printf("AliCaloTrackReader::FillInputEvent - Do not count events from fast cluster, trigger name %s\n",fFiredTriggerClassName.Data());
521     return kFALSE;
522   }
523   
524   //-------------------------------------------------------------------------------------
525   // Reject event if large clusters with large energy
526   // Use only for LHC11a data for the moment, and if input is clusterizer V1 or V1+unfolding
527   // If clusterzer NxN or V2 it does not help
528   //-------------------------------------------------------------------------------------
529   Int_t run = fInputEvent->GetRunNumber();
530   if( fRemoveLEDEvents && run > 146857  && run < 146861 )
531   {
532     //printf("Event %d\n",GetEventNumber());
533     
534     // Count number of cells with energy larger than 0.1 in SM3, cut on this number
535     Int_t ncellsSM3 = 0;
536     for(Int_t icell = 0; icell < fInputEvent->GetEMCALCells()->GetNumberOfCells(); icell++)
537     {
538       Int_t absID = fInputEvent->GetEMCALCells()->GetCellNumber(icell);
539       Int_t sm    = GetCaloUtils()->GetEMCALGeometry()->GetSuperModuleNumber(absID);
540       if(fInputEvent->GetEMCALCells()->GetAmplitude(icell) > 0.1 && sm==3) ncellsSM3++;
541     }
542     
543     Int_t ncellcut = 21;
544     if(fFiredTriggerClassName.Contains("EMC")) ncellcut = 35;
545     
546     if(ncellsSM3 >= ncellcut)
547     {
548       if(fDebug > 0) printf(" AliCaloTrackReader::FillInputEvent() - reject event with ncells in SM3 %d\n",ncellsSM3);
549       return kFALSE;
550     }
551   }// Remove LED events
552   
553   // Reject pure LED events?
554   if( fFiredTriggerClassName  !="" && !fAnaLED)
555   {
556     //printf("Event type %d\n",eventType);
557     if(eventType!=7)
558       return kFALSE; //Only physics event, do not use for simulated events!!!
559     
560     if(fDebug > 0) 
561       printf("AliCaloTrackReader::FillInputEvent() - FiredTriggerClass <%s>, selected class <%s>, compare name %d\n",
562              GetFiredTriggerClasses().Data(),fFiredTriggerClassName.Data(), GetFiredTriggerClasses().Contains(fFiredTriggerClassName));
563     
564     if( !GetFiredTriggerClasses().Contains(fFiredTriggerClassName) ) return kFALSE;
565     else if(fDebug > 0) printf("AliCaloTrackReader::FillInputEvent() - Accepted triggered event\n");
566   }
567   else if(fAnaLED)
568   {
569     //    kStartOfRun =       1,    // START_OF_RUN
570     //    kEndOfRun =         2,    // END_OF_RUN
571     //    kStartOfRunFiles =  3,    // START_OF_RUN_FILES
572     //    kEndOfRunFiles =    4,    // END_OF_RUN_FILES
573     //    kStartOfBurst =     5,    // START_OF_BURST
574     //    kEndOfBurst =       6,    // END_OF_BURST
575     //    kPhysicsEvent =     7,    // PHYSICS_EVENT
576     //    kCalibrationEvent = 8,    // CALIBRATION_EVENT
577     //    kFormatError =      9,    // EVENT_FORMAT_ERROR
578     //    kStartOfData =      10,   // START_OF_DATA
579     //    kEndOfData =        11,   // END_OF_DATA
580     //    kSystemSoftwareTriggerEvent   = 12, // SYSTEM_SOFTWARE_TRIGGER_EVENT
581     //    kDetectorSoftwareTriggerEvent = 13  // DETECTOR_SOFTWARE_TRIGGER_EVENT
582     
583           if(eventType!=7 && fDebug > 1 )printf("AliCaloTrackReader::FillInputEvent() - DO LED, Event Type <%d>, 8 Calibration \n",  eventType);
584           if(eventType!=8)return kFALSE;
585   }
586   
587   //In case of analysis of events with jets, skip those with jet pt > 5 pt hard 
588   if(fComparePtHardAndJetPt) 
589   {
590     if(!ComparePtHardAndJetPt()) return kFALSE ;
591   }
592   
593   if(fComparePtHardAndClusterPt) 
594   {
595     if(!ComparePtHardAndClusterPt()) return kFALSE ;
596   }
597   
598   //Fill Vertex array
599   FillVertexArray();
600   //Reject events with Z vertex too large, only for SE analysis, if not, cut on the analysis code
601   if(!GetMixedEvent() && TMath::Abs(fVertex[0][2]) > fZvtxCut) return kFALSE;  
602   
603   //------------------------------------------------------
604   //Event rejection depending on vertex, pileup, v0and
605   //------------------------------------------------------
606
607   if(fUseEventsWithPrimaryVertex)
608   {
609     if( !CheckForPrimaryVertex() )              return kFALSE;
610     if( TMath::Abs(fVertex[0][0] ) < 1.e-6 && 
611         TMath::Abs(fVertex[0][1] ) < 1.e-6 && 
612         TMath::Abs(fVertex[0][2] ) < 1.e-6    ) return kFALSE;
613   }
614   
615   //printf("Reader : IsPileUp %d, Multi %d\n",IsPileUpFromSPD(),fInputEvent->IsPileupFromSPDInMultBins());
616   
617   if(fDoEventSelection)
618   {
619     if(!fCaloFilterPatch)
620     {
621       // Do not analyze events with pileup
622       Bool_t bPileup = IsPileUpFromSPD();
623       //IsPileupFromSPDInMultBins() // method to try
624       //printf("pile-up %d, %d, %2.2f, %2.2f, %2.2f, %2.2f\n",bPileup, (Int_t) fPileUpParam[0], fPileUpParam[1], fPileUpParam[2], fPileUpParam[3], fPileUpParam[4]);
625       if(bPileup) return kFALSE;
626       
627       if(fDoV0ANDEventSelection)
628       {
629         Bool_t bV0AND = kTRUE; 
630         AliESDEvent* esd = dynamic_cast<AliESDEvent*> (fInputEvent);
631         if(esd) 
632           bV0AND = fTriggerAnalysis->IsOfflineTriggerFired(esd, AliTriggerAnalysis::kV0AND);
633         //else bV0AND = //FIXME FOR AODs
634         if(!bV0AND) return kFALSE;
635       }
636     }//CaloFilter patch
637     else
638     { 
639       if(fInputEvent->GetNumberOfCaloClusters() > 0) 
640       {
641         AliVCluster * calo = fInputEvent->GetCaloCluster(0);
642         if(calo->GetNLabels() == 4)
643         {
644           Int_t * selection = calo->GetLabels();
645           Bool_t bPileup = selection[0];
646           if(bPileup) return kFALSE;
647           
648           Bool_t bGoodV = selection[1]; 
649           if(fUseEventsWithPrimaryVertex && !bGoodV) return kFALSE;
650           
651           if(fDoV0ANDEventSelection)
652           {
653             Bool_t bV0AND = selection[2]; 
654             if(!bV0AND) return kFALSE;
655           }
656           
657           fTrackMult = selection[3];
658           if(fTrackMult == 0) return kFALSE;
659         } 
660         else 
661         {
662           //First filtered AODs, track multiplicity stored there.  
663           fTrackMult = (Int_t) ((AliAODHeader*)fInputEvent->GetHeader())->GetCentrality();
664           if(fTrackMult == 0) return kFALSE;          
665         }
666       }//at least one cluster
667       else 
668       {
669         //printf("AliCaloTrackReader::FillInputEvent() - No clusters in event\n");
670         //Remove events with  vertex (0,0,0), bad vertex reconstruction
671         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;
672         
673         //First filtered AODs, track multiplicity stored there.  
674         fTrackMult = (Int_t) ((AliAODHeader*)fInputEvent->GetHeader())->GetCentrality();
675         if(fTrackMult == 0) return kFALSE;
676       }// no cluster
677     }// CaloFileter patch
678   }// Event selection
679   //------------------------------------------------------
680   
681   //Check if there is a centrality value, PbPb analysis, and if a centrality bin selection is requested
682   //If we need a centrality bin, we select only those events in the corresponding bin.
683   if(GetCentrality() && fCentralityBin[0]>=0 && fCentralityBin[1]>=0 && fCentralityOpt==100)
684   {
685     Int_t cen = GetEventCentrality();
686     if(cen > fCentralityBin[1] || cen < fCentralityBin[0]) return kFALSE; //reject events out of bin.
687   }
688   
689   //Fill the arrays with cluster/tracks/cells data
690   
691   if(fFillEMCALCells) 
692     FillInputEMCALCells();
693   
694   if(fFillPHOSCells)  
695     FillInputPHOSCells();
696   
697   FillInputVZERO();
698   
699   if(fEventTriggerAtSE)
700   {
701     if(fFillCTS)
702     {   
703       FillInputCTS();
704       //Accept events with at least one track
705       if(fTrackMult == 0 && fDoEventSelection) return kFALSE;
706     }
707     if(fFillEMCAL) 
708       FillInputEMCAL();
709     if(fFillPHOS)  
710       FillInputPHOS();
711   }
712   else 
713   {
714     // In case of mixing analysis, all triggers accepted, but trigger particles selected 
715     // only for the specific trigered events selected here. Mixing done only for MB events,
716     // tracks array filled also for those events and not the others.
717
718     AliAnalysisManager *manager = AliAnalysisManager::GetAnalysisManager();
719     AliInputEventHandler *inputHandler = dynamic_cast<AliInputEventHandler*>(manager->GetInputEventHandler());
720     
721     if(!inputHandler) return kFALSE ;  // to content coverity
722     
723     UInt_t isTrigger = inputHandler->IsEventSelected() & fEventTriggerMask;
724     UInt_t isMB      = inputHandler->IsEventSelected() & fMixEventTriggerMask;
725     
726     if(fFillCTS && (isMB || isTrigger))
727     {   
728       FillInputCTS();
729       //Accept events with at least one track
730       if(fTrackMult == 0 && fDoEventSelection) return kFALSE;
731     }
732     
733     if(isTrigger)
734     {
735       if(fFillEMCAL) 
736         FillInputEMCAL();
737       if(fFillPHOS)  
738         FillInputPHOS();
739     } 
740   }
741   
742   return kTRUE ;
743 }
744
745 //___________________________________
746 void AliCaloTrackReader::ResetLists() 
747 {
748   //  Reset lists, called by the analysis maker 
749   
750   if(fCTSTracks)       fCTSTracks     -> Clear();
751   if(fEMCALClusters)   fEMCALClusters -> Clear("C");
752   if(fPHOSClusters)    fPHOSClusters  -> Clear("C");
753   
754   fV0ADC[0] = 0;   fV0ADC[1] = 0; 
755   fV0Mul[0] = 0;   fV0Mul[1] = 0; 
756   
757 }
758
759 //____________________________________________________________
760 void AliCaloTrackReader::SetInputEvent(AliVEvent* const input)  
761 {
762   fInputEvent  = input;
763   fMixedEvent = dynamic_cast<AliMixedEvent*>(GetInputEvent()) ; 
764   if (fMixedEvent) 
765     fNMixedEvent = fMixedEvent->GetNumberOfEvents() ; 
766   
767   //Delete previous vertex
768   if(fVertex)
769   {
770     for (Int_t i = 0; i < fNMixedEvent; i++) 
771     {
772       delete [] fVertex[i] ; 
773     }
774     delete [] fVertex ;
775   }
776   
777   fVertex = new Double_t*[fNMixedEvent] ; 
778   for (Int_t i = 0; i < fNMixedEvent; i++) 
779   {
780     fVertex[i] = new Double_t[3] ; 
781     fVertex[i][0] = 0.0 ; 
782     fVertex[i][1] = 0.0 ; 
783     fVertex[i][2] = 0.0 ; 
784   }
785 }
786
787 //__________________________________________________
788 Int_t AliCaloTrackReader::GetEventCentrality() const 
789 {
790   //Return current event centrality
791   
792   if(GetCentrality())
793   {
794     if     (fCentralityOpt==100) return (Int_t) GetCentrality()->GetCentralityPercentile(fCentralityClass); // 100 bins max
795     else if(fCentralityOpt==10)  return GetCentrality()->GetCentralityClass10(fCentralityClass);// 10 bins max
796     else if(fCentralityOpt==20)  return GetCentrality()->GetCentralityClass5(fCentralityClass); // 20 bins max
797     else 
798     {
799       printf("AliCaloTrackReader::GetEventCentrality() - Unknown centrality option %d, use 10, 20 or 100\n",fCentralityOpt);
800       return -1;
801     } 
802   }
803   else return -1;
804   
805 }
806
807 //_____________________________________________________
808 Double_t AliCaloTrackReader::GetEventPlaneAngle() const 
809 {
810   //Return current event centrality
811   
812   if(GetEventPlane())
813   {
814     Float_t ep =  GetEventPlane()->GetEventplane(GetEventPlaneMethod(), GetInputEvent());
815     
816     if(GetEventPlaneMethod()=="Q" && (ep < 0 || ep > TMath::Pi())) 
817     {
818       if(fDebug > 0 ) printf("AliCaloTrackReader::GetEventPlaneAngle() -  Bad EP for <Q> method : %f\n",ep);
819       return -1000;
820     }
821     else if(GetEventPlaneMethod().Contains("V0")  ) 
822     {
823       if((ep > TMath::Pi()/2 || ep < -TMath::Pi()/2))
824       {
825         if(fDebug > 0 ) printf("AliCaloTrackReader::GetEventPlaneAngle() -  Bad EP for <%s> method : %f\n",GetEventPlaneMethod().Data(), ep);
826         return -1000;
827       }
828       
829       ep+=TMath::Pi()/2; // put same range as for <Q> method
830       
831     }
832   
833     //printf("AliCaloTrackReader::GetEventPlaneAngle() = %f\n",ep);
834     if(fDebug > 0 )
835     {
836       if     (ep > TMath::Pi()) printf("AliCaloTrackReader::GetEventPlaneAngle() - Too large angle = %f\n",ep);
837       else if(ep < 0          ) printf("AliCaloTrackReader::GetEventPlaneAngle() - Negative angle = %f\n" ,ep);
838     }
839     
840     return ep;
841   }
842   else
843   {
844     if(fDataType!=kMC && fDebug > 0) printf("AliCaloTrackReader::GetEventPlaneAngle() -  No EP pointer\n");
845     return -1000;
846   } 
847   
848 }
849
850 //__________________________________________________________
851 void AliCaloTrackReader::GetVertex(Double_t vertex[3]) const 
852 {
853   //Return vertex position to be used for single event analysis
854   vertex[0]=fVertex[0][0];  
855   vertex[1]=fVertex[0][1];  
856   vertex[2]=fVertex[0][2];
857 }
858
859 //____________________________________________________________
860 void AliCaloTrackReader::GetVertex(Double_t vertex[3], 
861                                    const Int_t evtIndex) const 
862 {
863   //Return vertex position for mixed event, recover the vertex in a particular event.
864   
865   vertex[0]=fVertex[evtIndex][0];  vertex[1]=fVertex[evtIndex][1];  vertex[2]=fVertex[evtIndex][2];
866   
867 }
868
869 //________________________________________
870 void AliCaloTrackReader::FillVertexArray() 
871 {
872   
873   //Fill data member with vertex
874   //In case of Mixed event, multiple vertices
875   
876   //Delete previous vertex
877   if(fVertex)
878   {
879     for (Int_t i = 0; i < fNMixedEvent; i++) 
880     {
881       delete [] fVertex[i] ; 
882     }
883     delete [] fVertex ;  
884   }
885   
886   fVertex = new Double_t*[fNMixedEvent] ; 
887   for (Int_t i = 0; i < fNMixedEvent; i++) 
888   {
889     fVertex[i] = new Double_t[3] ; 
890     fVertex[i][0] = 0.0 ; 
891     fVertex[i][1] = 0.0 ; 
892     fVertex[i][2] = 0.0 ; 
893   }          
894   
895   if (!fMixedEvent) 
896   { //Single event analysis
897     if(fDataType!=kMC)
898     {
899       
900       if(fInputEvent->GetPrimaryVertex())
901       {
902         fInputEvent->GetPrimaryVertex()->GetXYZ(fVertex[0]); 
903       }
904       else 
905       {
906         printf("AliCaloTrackReader::FillVertexArray() - NULL primary vertex\n");
907         fVertex[0][0]=0.;   fVertex[0][1]=0.;   fVertex[0][2]=0.;
908       }//Primary vertex pointer do not exist
909       
910     } else
911     {//MC read event 
912       fVertex[0][0]=0.;   fVertex[0][1]=0.;   fVertex[0][2]=0.;
913     }
914     
915     if(fDebug > 1)
916       printf("AliCaloTrackReader::FillVertexArray() - Single Event Vertex : %f,%f,%f\n",fVertex[0][0],fVertex[0][1],fVertex[0][2]);
917     
918   } else 
919   { // MultiEvent analysis
920     for (Int_t iev = 0; iev < fNMixedEvent; iev++) 
921     {
922       if (fMixedEvent->GetVertexOfEvent(iev))
923         fMixedEvent->GetVertexOfEvent(iev)->GetXYZ(fVertex[iev]);
924       else
925       { // no vertex found !!!!
926         AliWarning("No vertex found");
927       }
928       
929       if(fDebug > 1)
930         printf("AliCaloTrackReader::FillVertexArray() - Multi Event %d Vertex : %f,%f,%f\n",iev,fVertex[iev][0],fVertex[iev][1],fVertex[iev][2]);
931       
932     }
933   }
934   
935 }
936
937 //_____________________________________
938 void AliCaloTrackReader::FillInputCTS() 
939 {
940   //Return array with Central Tracking System (CTS) tracks
941   
942   if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputCTS()\n");
943   
944   Double_t pTrack[3] = {0,0,0};
945   
946   Int_t nTracks = fInputEvent->GetNumberOfTracks() ;
947   fTrackMult    = 0;
948   Int_t nstatus = 0;
949   
950   for (Int_t itrack =  0; itrack <  nTracks; itrack++) 
951   {////////////// track loop
952     AliVTrack * track = (AliVTrack*)fInputEvent->GetTrack(itrack) ; // retrieve track from esd
953     
954     //Select tracks under certain conditions, TPCrefit, ITSrefit ... check the set bits
955     if (fTrackStatus && !((track->GetStatus() & fTrackStatus) == fTrackStatus)) 
956       continue ;
957     
958     nstatus++;
959     
960     if     (fDataType==kESD)
961     {
962       AliESDtrack* esdTrack = dynamic_cast<AliESDtrack*> (track);
963       
964       if(esdTrack && fESDtrackCuts->AcceptTrack(esdTrack))
965       {
966         track->GetPxPyPz(pTrack) ;
967
968         if(fConstrainTrack)
969         {
970           if(esdTrack->GetConstrainedParam())
971           {
972             const AliExternalTrackParam* constrainParam = esdTrack->GetConstrainedParam();
973             esdTrack->Set(constrainParam->GetX(),constrainParam->GetAlpha(),constrainParam->GetParameter(),constrainParam->GetCovariance());
974             esdTrack->GetConstrainedPxPyPz(pTrack);
975           }
976           else continue;
977           
978         } // use constrained tracks
979         
980       }
981       else continue;
982       
983     } // ESD
984     else if(fDataType==kAOD)
985     {
986       AliAODTrack *aodtrack = dynamic_cast <AliAODTrack*>(track);
987       
988       if(aodtrack)
989       {
990        if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputCTS():AOD track type: %d (primary %d), hybrid? %d \n",
991                               aodtrack->GetType(),AliAODTrack::kPrimary,
992                               aodtrack->IsHybridGlobalConstrainedGlobal());
993         
994         
995         if (fSelectHybridTracks)
996         {
997           if (!aodtrack->IsHybridGlobalConstrainedGlobal())       continue ;
998         }
999         else 
1000         {
1001           if ( aodtrack->TestFilterBit(fTrackFilterMask)==kFALSE) continue ;
1002         }
1003         
1004         if (aodtrack->GetType()!= AliAODTrack::kPrimary)          continue ;
1005         
1006         if (fDebug > 2 ) printf("AliCaloTrackReader::FillInputCTS(): \t accepted track! \n");
1007         
1008         track->GetPxPyPz(pTrack) ;
1009         
1010       } // aod track exists
1011       else continue ;
1012       
1013     } // AOD
1014     
1015     //Count the tracks in eta < 0.9
1016     //printf("Eta %f cut  %f\n",TMath::Abs(track->Eta()),fTrackMultEtaCut);
1017     if(TMath::Abs(track->Eta())< fTrackMultEtaCut) fTrackMult++;
1018     
1019     TLorentzVector momentum(pTrack[0],pTrack[1],pTrack[2],0);
1020     
1021     if(fCTSPtMin < momentum.Pt() && fCTSPtMax > momentum.Pt())
1022     {
1023       if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"CTS")) 
1024         continue;
1025       
1026       if(fDebug > 2 && momentum.Pt() > 0.1) 
1027         printf("AliCaloTrackReader::FillInputCTS() - Selected tracks E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
1028                momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
1029       
1030       if (fMixedEvent) 
1031       {
1032         track->SetID(itrack);
1033       }
1034       
1035       fCTSTracks->Add(track);        
1036       
1037     }//Pt and Fiducial cut passed. 
1038   }// track loop
1039         
1040   if(fDebug > 1) 
1041     printf("AliCaloTrackReader::FillInputCTS()   - aod entries %d, input tracks %d, pass status %d, multipliticy %d\n", fCTSTracks->GetEntriesFast(), nTracks, nstatus, fTrackMult);//fCTSTracksNormalInputEntries);
1042   
1043 }
1044
1045 //__________________________________________________________________
1046 void AliCaloTrackReader::FillInputEMCALAlgorithm(AliVCluster * clus, 
1047                                                  const Int_t iclus) 
1048 {
1049   //Fill the EMCAL data in the array, do it
1050     
1051   Int_t vindex = 0 ;  
1052   if (fMixedEvent) 
1053     vindex = fMixedEvent->EventIndexForCaloCluster(iclus);
1054   
1055   //Reject clusters with bad channels, close to borders and exotic;
1056   if(!GetCaloUtils()->GetEMCALRecoUtils()->IsGoodCluster(clus,GetCaloUtils()->GetEMCALGeometry(),GetEMCALCells(),fInputEvent->GetBunchCrossNumber())) return;
1057   
1058   //Mask all cells in collumns facing ALICE thick material if requested
1059   if(GetCaloUtils()->GetNMaskCellColumns())
1060   {
1061     Int_t absId   = -1;
1062     Int_t iSupMod = -1;
1063     Int_t iphi    = -1;
1064     Int_t ieta    = -1;
1065     Bool_t shared = kFALSE;
1066     GetCaloUtils()->GetEMCALRecoUtils()->GetMaxEnergyCell(GetCaloUtils()->GetEMCALGeometry(), GetEMCALCells(),clus,absId,iSupMod,ieta,iphi,shared);
1067     if(GetCaloUtils()->MaskFrameCluster(iSupMod, ieta)) return;
1068   }
1069   
1070   if(fSelectEmbeddedClusters)
1071   {
1072     if(clus->GetNLabels()==0 || clus->GetLabel() < 0) return;
1073     //else printf("Embedded cluster,  %d, n label %d label %d  \n",iclus,clus->GetNLabels(),clus->GetLabel());
1074   }
1075   
1076   //Float_t pos[3];
1077   //clus->GetPosition(pos);
1078   //printf("Before Corrections: e %f, x %f, y %f, z %f\n",clus->E(),pos[0],pos[1],pos[2]);
1079   
1080   if(fRecalculateClusters)
1081   {
1082     //Recalibrate the cluster energy 
1083     if(GetCaloUtils()->IsRecalibrationOn())
1084     {
1085       Float_t energy = GetCaloUtils()->RecalibrateClusterEnergy(clus, GetEMCALCells());
1086       
1087       clus->SetE(energy);
1088       //printf("Recalibrated Energy %f\n",clus->E());  
1089       
1090       GetCaloUtils()->RecalculateClusterShowerShapeParameters(GetEMCALCells(),clus);
1091       GetCaloUtils()->RecalculateClusterPID(clus);
1092       
1093     } // recalculate E
1094     
1095     //Recalculate distance to bad channels, if new list of bad channels provided
1096     GetCaloUtils()->RecalculateClusterDistanceToBadChannel(GetEMCALCells(),clus);
1097     
1098     //Recalculate cluster position
1099     if(GetCaloUtils()->IsRecalculationOfClusterPositionOn())
1100     {
1101       GetCaloUtils()->RecalculateClusterPosition(GetEMCALCells(),clus); 
1102       //clus->GetPosition(pos);
1103       //printf("After  Corrections: e %f, x %f, y %f, z %f\n",clus->E(),pos[0],pos[1],pos[2]);
1104     }
1105     
1106     // Recalculate TOF
1107     if(GetCaloUtils()->GetEMCALRecoUtils()->IsTimeRecalibrationOn()) 
1108     {
1109       Double_t tof      = clus->GetTOF();
1110       Float_t  frac     =-1;
1111       Int_t    absIdMax = GetCaloUtils()->GetMaxEnergyCell(fEMCALCells, clus,frac);
1112       
1113       if(fDataType==AliCaloTrackReader::kESD)
1114       { 
1115         tof = fEMCALCells->GetCellTime(absIdMax);
1116       }
1117       
1118       GetCaloUtils()->GetEMCALRecoUtils()->RecalibrateCellTime(absIdMax,fInputEvent->GetBunchCrossNumber(),tof);
1119       
1120       clus->SetTOF(tof);
1121       
1122     }// Time recalibration    
1123   }
1124   
1125   //Correct non linearity
1126   if(GetCaloUtils()->IsCorrectionOfClusterEnergyOn())
1127   {
1128     GetCaloUtils()->CorrectClusterEnergy(clus) ;
1129     //printf("Linearity Corrected Energy %f\n",clus->E());  
1130     
1131     //In case of MC analysis, to match resolution/calibration in real data
1132     Float_t rdmEnergy = GetCaloUtils()->GetEMCALRecoUtils()->SmearClusterEnergy(clus);
1133     // printf("\t Energy %f, smeared %f\n", clus->E(),rdmEnergy);
1134     clus->SetE(rdmEnergy);
1135   }
1136   
1137   TLorentzVector momentum ;
1138   
1139   clus->GetMomentum(momentum, fVertex[vindex]);      
1140   
1141   if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"EMCAL")) return;
1142   
1143   if(fEMCALPtMin > momentum.E() || fEMCALPtMax < momentum.E())         return;
1144   
1145   Double_t tof = clus->GetTOF()*1e9;
1146   if(tof < fEMCALTimeCutMin     || tof > fEMCALTimeCutMax)             return;
1147   
1148   if(fDebug > 2 && momentum.E() > 0.1) 
1149     printf("AliCaloTrackReader::FillInputEMCAL() - Selected clusters E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
1150            momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());
1151   
1152   if (fMixedEvent) 
1153     clus->SetID(iclus) ; 
1154   
1155   fEMCALClusters->Add(clus);    
1156     
1157 }
1158
1159 //_______________________________________
1160 void AliCaloTrackReader::FillInputEMCAL() 
1161 {
1162   //Return array with EMCAL clusters in aod format
1163   
1164   if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputEMCAL()\n");
1165   
1166   // First recalibrate cells, time or energy
1167   //  if(GetCaloUtils()->IsRecalibrationOn())
1168   //    GetCaloUtils()->GetEMCALRecoUtils()->RecalibrateCells(GetCaloUtils()->GetEMCALGeometry(), 
1169   //                                                          GetEMCALCells(), 
1170   //                                                          fInputEvent->GetBunchCrossNumber());
1171   
1172   //Loop to select clusters in fiducial cut and fill container with aodClusters
1173   if(fEMCALClustersListName=="")
1174   {
1175     Int_t nclusters = fInputEvent->GetNumberOfCaloClusters();
1176     for (Int_t iclus =  0; iclus <  nclusters; iclus++) 
1177     {
1178       AliVCluster * clus = 0;
1179       if ( (clus = fInputEvent->GetCaloCluster(iclus)) ) 
1180       {
1181         if (IsEMCALCluster(clus))
1182         {          
1183           FillInputEMCALAlgorithm(clus, iclus);
1184         }//EMCAL cluster
1185       }// cluster exists
1186     }// cluster loop
1187     
1188     //Recalculate track matching
1189     GetCaloUtils()->RecalculateClusterTrackMatching(fInputEvent);
1190     
1191   }//Get the clusters from the input event
1192   else
1193   {
1194     TClonesArray * clusterList = 0x0; 
1195     
1196     if      (fInputEvent->FindListObject(fEMCALClustersListName))
1197     {
1198       clusterList = dynamic_cast<TClonesArray*> (fInputEvent ->FindListObject(fEMCALClustersListName));
1199     }
1200     else if(fOutputEvent)
1201     { 
1202       clusterList = dynamic_cast<TClonesArray*> (fOutputEvent->FindListObject(fEMCALClustersListName));
1203     }
1204     
1205     if(!clusterList)
1206     {
1207       //printf("AliCaloTrackReader::FillInputEMCAL() - Wrong name of list with clusters? Try input event <%s>\n",fEMCALClustersListName.Data());
1208       //List not in output event, try input event
1209       clusterList = dynamic_cast<TClonesArray*> (fInputEvent->FindListObject(fEMCALClustersListName));
1210       if(!clusterList)
1211       {
1212         printf("AliCaloTrackReader::FillInputEMCAL() - Wrong name of list with clusters?  <%s>\n",fEMCALClustersListName.Data());
1213         return;
1214       }
1215     }
1216     
1217     Int_t nclusters = clusterList->GetEntriesFast();
1218     for (Int_t iclus =  0; iclus <  nclusters; iclus++) 
1219     {
1220       AliVCluster * clus = dynamic_cast<AliVCluster*> (clusterList->At(iclus));
1221       //printf("E %f\n",clus->E());
1222       if (clus) FillInputEMCALAlgorithm(clus, iclus);
1223       else printf("AliCaloTrackReader::FillInputEMCAL() - Null cluster in list!\n");
1224     }// cluster loop
1225     
1226     // Recalculate track matching, not necessary if already done in the reclusterization task.
1227     // in case it was not done ...
1228     GetCaloUtils()->RecalculateClusterTrackMatching(fInputEvent,clusterList);
1229     
1230   }
1231   
1232   if(fDebug > 1) printf("AliCaloTrackReader::FillInputEMCAL() - aod entries %d\n",  fEMCALClusters->GetEntriesFast());
1233   
1234 }
1235
1236 //______________________________________
1237 void AliCaloTrackReader::FillInputPHOS() 
1238 {
1239   //Return array with PHOS clusters in aod format
1240   
1241   if(fDebug > 2 ) printf("AliCaloTrackReader::FillInputPHOS()\n");
1242   
1243   //Loop to select clusters in fiducial cut and fill container with aodClusters
1244   Int_t nclusters = fInputEvent->GetNumberOfCaloClusters();
1245   for (Int_t iclus = 0; iclus < nclusters; iclus++) 
1246   {
1247     AliVCluster * clus = 0;
1248     if ( (clus = fInputEvent->GetCaloCluster(iclus)) ) 
1249     {
1250       if (IsPHOSCluster(clus))
1251       {
1252         //Check if the cluster contains any bad channel and if close to calorimeter borders
1253         Int_t vindex = 0 ;  
1254         if (fMixedEvent) 
1255           vindex = fMixedEvent->EventIndexForCaloCluster(iclus);
1256         if( GetCaloUtils()->ClusterContainsBadChannel("PHOS",clus->GetCellsAbsId(), clus->GetNCells())) 
1257           continue;
1258         if(!GetCaloUtils()->CheckCellFiducialRegion(clus, fInputEvent->GetPHOSCells(), fInputEvent, vindex)) 
1259           continue;
1260         
1261         if(fRecalculateClusters)
1262         {
1263           //Recalibrate the cluster energy 
1264           if(GetCaloUtils()->IsRecalibrationOn()) 
1265           {
1266             Float_t energy = GetCaloUtils()->RecalibrateClusterEnergy(clus, (AliAODCaloCells*)GetPHOSCells());
1267             clus->SetE(energy);
1268           }
1269         }
1270         
1271         TLorentzVector momentum ;
1272         
1273         clus->GetMomentum(momentum, fVertex[vindex]);      
1274         
1275         if(fCheckFidCut && !fFiducialCut->IsInFiducialCut(momentum,"PHOS")) continue;
1276         
1277         if(fPHOSPtMin > momentum.E() || fPHOSPtMax < momentum.E())          continue;
1278         
1279         if(fDebug > 2 && momentum.E() > 0.1) 
1280           printf("AliCaloTrackReader::FillInputPHOS() - Selected clusters E %3.2f, pt %3.2f, phi %3.2f, eta %3.2f\n",
1281                  momentum.E(),momentum.Pt(),momentum.Phi()*TMath::RadToDeg(),momentum.Eta());        
1282         
1283         
1284         if (fMixedEvent) 
1285         {
1286           clus->SetID(iclus) ; 
1287         }              
1288         
1289         fPHOSClusters->Add(clus);       
1290         
1291       }//PHOS cluster
1292     }//cluster exists
1293   }//esd cluster loop
1294   
1295   if(fDebug > 1) printf("AliCaloTrackReader::FillInputPHOS()  - aod entries %d\n",  fPHOSClusters->GetEntriesFast());
1296   
1297 }
1298
1299 //____________________________________________
1300 void AliCaloTrackReader::FillInputEMCALCells() 
1301 {
1302   //Return array with EMCAL cells in aod format
1303   
1304   fEMCALCells = fInputEvent->GetEMCALCells(); 
1305   
1306 }
1307
1308 //___________________________________________
1309 void AliCaloTrackReader::FillInputPHOSCells() 
1310 {
1311   //Return array with PHOS cells in aod format
1312   
1313   fPHOSCells = fInputEvent->GetPHOSCells(); 
1314   
1315 }
1316
1317 //_______________________________________
1318 void AliCaloTrackReader::FillInputVZERO()
1319 {
1320   //Fill VZERO information in data member, add all the channels information.
1321   AliVVZERO* v0 = fInputEvent->GetVZEROData();
1322   //printf("Init V0: ADC (%d,%d), Multiplicity (%d,%d) \n",fV0ADC[0],fV0ADC[1],fV0Mul[0],fV0Mul[1]);
1323   
1324   if (v0) 
1325   {
1326     AliESDVZERO* esdV0 = dynamic_cast<AliESDVZERO*> (v0);
1327     for (Int_t i = 0; i < 32; i++)
1328     {
1329       if(esdV0)
1330       {//Only available in ESDs
1331         fV0ADC[0] += (Int_t)esdV0->GetAdcV0C(i);
1332         fV0ADC[1] += (Int_t)esdV0->GetAdcV0A(i);
1333       }
1334       
1335       fV0Mul[0] += (Int_t)v0->GetMultiplicityV0C(i);
1336       fV0Mul[1] += (Int_t)v0->GetMultiplicityV0A(i);
1337     }
1338     if(fDebug > 0)
1339       printf("V0: ADC (%d,%d), Multiplicity (%d,%d) \n",fV0ADC[0],fV0ADC[1],fV0Mul[0],fV0Mul[1]);
1340   }
1341   else
1342   {
1343     if(fDebug > 0)
1344       printf("Cannot retrieve V0 ESD! Run w/ null V0 charges\n ");
1345   }
1346 }
1347
1348
1349 //___________________________________________________________________
1350 Bool_t AliCaloTrackReader::IsEMCALCluster(AliVCluster* cluster) const 
1351 {
1352   // Check if it is a cluster from EMCAL. For old AODs cluster type has
1353   // different number and need to patch here
1354   
1355   if(fDataType==kAOD && fOldAOD)
1356   {
1357     if (cluster->GetType() == 2) return kTRUE;
1358     else                         return kFALSE;
1359   }
1360   else 
1361   {
1362     return cluster->IsEMCAL();
1363   }
1364   
1365 }
1366
1367 //___________________________________________________________________
1368 Bool_t AliCaloTrackReader::IsPHOSCluster(AliVCluster * cluster) const 
1369 {
1370   //Check if it is a cluster from PHOS.For old AODs cluster type has
1371   // different number and need to patch here
1372   
1373   if(fDataType==kAOD && fOldAOD)
1374   {
1375     Int_t type = cluster->GetType();
1376     if (type == 0 || type == 1) return kTRUE;
1377     else                        return kFALSE;
1378   }
1379   else 
1380   {
1381     return cluster->IsPHOS();
1382   }
1383   
1384 }
1385
1386 //________________________________________________
1387 Bool_t AliCaloTrackReader::CheckForPrimaryVertex()
1388 {
1389   //Check if the vertex was well reconstructed, copy from V0Reader of conversion group
1390   //Only for ESDs ...
1391   
1392   AliESDEvent * event = dynamic_cast<AliESDEvent*> (fInputEvent);
1393   if(!event) return kTRUE;
1394   
1395   if(event->GetPrimaryVertexTracks()->GetNContributors() > 0) 
1396   {
1397     return kTRUE;
1398   }
1399   
1400   if(event->GetPrimaryVertexTracks()->GetNContributors() < 1) 
1401   {
1402     // SPD vertex
1403     if(event->GetPrimaryVertexSPD()->GetNContributors() > 0) 
1404     {
1405       //cout<<"spd vertex type::"<< fESDEvent->GetPrimaryVertex()->GetName() << endl;
1406       return kTRUE;
1407       
1408     }
1409     if(event->GetPrimaryVertexSPD()->GetNContributors() < 1) 
1410     {
1411       //      cout<<"bad vertex type::"<< event->GetPrimaryVertex()->GetName() << endl;
1412       return kFALSE;
1413     }
1414   }
1415   
1416   return kFALSE;  
1417   
1418 }
1419
1420 //____________________________________________________________
1421 void  AliCaloTrackReader::SetTrackCuts(AliESDtrackCuts * cuts) 
1422
1423   // Set Track cuts
1424   
1425   if(fESDtrackCuts) delete fESDtrackCuts ;
1426   
1427   fESDtrackCuts = cuts ; 
1428   
1429 }                 
1430