Fixes for coverity.
[u/mrichter/AliRoot.git] / JETAN / AliJetReader.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 /* $Id$ */
17  
18 //------------------------------------------------------------------------
19 // Jet reader base class
20 // manages the reading of input for jet algorithms
21 // Authors: jgcn@mda.cinvestav.mx
22 //          magali.estienne@subatech.in2p3.fr
23 //          alexandre.shabetai@cern.ch
24 //
25 // **February 2011
26 // implemented  standard geometry (AliEMCALGeometry) (was AliJetDummyGeo implented separately in ESDReader and AODReader
27 // local2master matrices are now get from $ALICE_ROOT/OADB/PWG4/JetReconstruction/EMCALlocal2master.root
28 // you can choose the geometry (EMCAL_COMPLETE, EMCAL_FIRSTYEARv1, etc) via SetEMCALgeo2bLoad('Name_of_Geometry') in the Readerheader
29 // different options for survey(ed) matrice are provided too
30 // ** August 2011
31 // OADB path changed from  '../OADB/PWG4/JetReconstruction/'  to   '../OADB/EMCAL/'
32 // marco.bregant@subatech.in2p3.fr
33 // ** 2011
34 // - AliJetESD/AODReader classes removed from JETAN. Reader now ESD/AOD independent. It uses VEvent in the AliJetFill* classes.
35 // - EMCal utilities added for bad cells id, calibration, etc.
36 //------------------------------------------------------------------------
37
38 // root
39 #include <TFile.h>
40
41 //AliRoot
42 #include "AliVEvent.h"
43 #include "AliAODEvent.h"
44 #include "AliESDEvent.h"
45 #include "AliJetReader.h"
46 #include "AliJetReaderHeader.h"
47 #include "AliAnalysisManager.h"
48 #include "AliJetFillCalTrkTrack.h" 
49 #include "AliJetFillCalTrkTrackMC.h"
50 #include "AliJetCalTrk.h"
51
52 using std::cout;
53 using std::endl;
54 ClassImp(AliJetReader)
55
56 ////////////////////////////////////////////////////////////////////////
57
58
59 AliJetReader::AliJetReader():
60   fCalTrkEvent(0x0),
61   fFillEvent(0x0),
62   fReaderHeader(0),
63   fFillEventwTrks(0x0), 
64   fDebug(0),
65   fVEvent(0x0),
66   fMCEvent(0x0),
67   fOpt(0)
68 {
69   // Default constructor
70 }
71
72 //-----------------------------------------------------------------------
73 AliJetReader::~AliJetReader()
74 {
75   // Destructor
76   if (fCalTrkEvent) {
77     fCalTrkEvent->Delete();
78     delete fCalTrkEvent;
79   }
80
81   if (fFillEventwTrks) {
82     delete fFillEventwTrks;
83   }
84
85 }
86
87 //-----------------------------------------------------------------------
88 void AliJetReader::InitTasks()
89 {
90   // Initialization
91   fOpt = GetReaderHeader()->GetDetector();
92
93   TString datatype = fReaderHeader->GetDataType();
94   datatype.ToUpper();
95   Bool_t kIsKine = kFALSE;
96   if((!datatype.Contains("AOD") && datatype.Contains("MC")) ||
97      (!datatype.Contains("AOD") && datatype.Contains("MC2")) ){
98     kIsKine = kTRUE;
99   } 
100   Bool_t kIsHighMult = GetReaderHeader()->GetIsHighMult();
101   fCalTrkEvent = new AliJetCalTrkEvent(fOpt,kIsKine,kIsHighMult);
102
103   // Initialize jet analysis
104   CreateTasks();
105
106 }
107
108 //-----------------------------------------------------------------------
109 Bool_t AliJetReader::ProcessEvent()
110 {
111   // Process one event
112   // Charged only or charged+neutral jets
113
114   Bool_t ok = ExecTasks();
115
116   if(!ok) {return kFALSE;}
117
118   return kTRUE;
119
120 }
121
122 //-----------------------------------------------------------------------
123 void AliJetReader::SetInputEvent(const TObject* esd, const TObject* aod, const AliMCEvent* mc)
124 {
125   // set input event pointers
126   if( fReaderHeader->GetDataType().Contains("AOD") && aod) {fVEvent = (AliAODEvent*) aod;}
127   else if( fReaderHeader->GetDataType().Contains("ESD") && esd) {fVEvent = (AliESDEvent*) esd;}
128   else if ( fReaderHeader->GetDataType().Contains("MC") || fReaderHeader->GetDataType().Contains("MC2")) { fMCEvent = (AliMCEvent*) mc;}
129   else {printf("No input event ! ");}
130
131 }
132
133 //-----------------------------------------------------------------------
134 Bool_t AliJetReader::CreateTasks()
135 {
136   // For reader task initialization
137
138   fDebug = fReaderHeader->GetDebug();
139
140   fFillEvent = new AliJetFillCalTrkEvent();
141   if (fOpt>0) {
142     // Tracks
143     if(fOpt%2==!0 && fOpt!=0){
144       fFillEventwTrks = new AliJetFillCalTrkTrack();
145       fFillEventwTrks->SetReaderHeader(fReaderHeader);
146     }
147   }
148   else { // MC/Kine cases
149     fFillEventwTrks = new AliJetFillCalTrkTrackMC();
150     fFillEventwTrks->SetReaderHeader(fReaderHeader);
151   }
152
153   if(fDebug>1) cout << "Tasks instantiated at that stage ! " << endl;
154
155   return kTRUE;
156
157 }
158
159 //-----------------------------------------------------------------------
160 Bool_t AliJetReader::ExecTasks()
161 {
162   // Main function
163   // Fill the reader part
164   
165   fDebug = fReaderHeader->GetDebug();
166
167   if(!fVEvent && !fMCEvent) {
168     return kFALSE;
169   }
170
171   // TPC only or Digits+TPC or Clusters+TPC
172   if(fOpt%2==!0 || fOpt==0){
173     fFillEventwTrks->SetVEvent(fVEvent);
174     fFillEventwTrks->SetMCEvent(fMCEvent);
175     fFillEventwTrks->SetCalTrkEvent(fCalTrkEvent);    
176     fFillEventwTrks->Exec("tpc");
177   }
178
179   return kTRUE;
180
181 }
182
183 //-----------------------------------------------------------------------
184 void AliJetReader::WriteRHeaderToFile() const
185 {
186   // write reader header
187   AliJetReaderHeader *rh = GetReaderHeader();
188   rh->Write();
189
190 }
191
192 //-----------------------------------------------------------------------
193 void AliJetReader::WriteReaderHeader()
194 {
195   // Write the Headers
196   TFile* f = new TFile("jets_local.root", "recreate");
197   WriteRHeaderToFile();
198   f->Close();
199
200 }
201