]> git.uio.no Git - u/mrichter/AliRoot.git/blame - PWGJE/FlavourJetTasks/macros/AnalysisTrainCorrJetsLocal.C
git://git.uio.no / u / mrichter / AliRoot.git / blame
? search:
summary | shortlog | log | commit | commitdiff | tree
blob | blame (incremental) | history | HEAD
Initial commit of the tasks for the flavor jet analysis; targeted at D-meson+jet...
[u/mrichter/AliRoot.git] / PWGJE / FlavourJetTasks / macros / AnalysisTrainCorrJetsLocal.C
CommitLineData
f6e094bd 1// $Id$
2
3const Int_t mode = 0; // 0-->Local, 1-->Grid interactive
4const Bool_t bIsAOD = kTRUE;
5const Bool_t bIsMC = kFALSE;
6
7const Bool_t bIsPhysSel = kFALSE;
8const Bool_t bIsCentSel = kFALSE;
9const Bool_t bIsEvnPSel = kFALSE;
10const Bool_t bIsRespPID = kFALSE;
11
12const TString setData = "dataset.txt";
13//=============================================================================
14
15const Bool_t bIsEMCalAna = kFALSE;
16
17const Bool_t bFastOnly = kTRUE;
18const Bool_t bHistoW = kTRUE;
19const Double_t dMinEclus = 5.;
20const Double_t dMinPtClus = 5.;
21const Double_t dVz = 10.;
22const Bool_t bVzDiff = kTRUE;
23const Double_t dCentMin = -1.;
24const Double_t dCentMax = -1.;
25const Double_t dMinScaleCT = -1.;
26const Double_t dMaxScaleCT = -1.;
27//=============================================================================
28
29void AnalysisTrainCorrJetsLocal()
30{
31 if (mode==1 && !TGrid::Connect("alien://")) {
32 ::Error("AnalysisTrainCorrJetsLocal.C::AnalysisTrainCorrJetsLocal", "Can not connect to the Grid!");
33 return;
34 }
35 if (LoadLibraries()) {
36 ::Error("AnalysisTrainCorrJetsLocal.C::AnalysisTrainCorrJetsLocal", "Load libraries failed!");
37 return;
38 }
39//=============================================================================
40
41 TChain *chain = CreateAODFriendChain(setData);
42 if (!chain) {
43 ::Error("AnalysisTrainCorrJetsLocal.C::AnalysisTrainCorrJetsLocal", "Creating input chain failed!");
44 return;
45 }
46//=============================================================================
47
48 const UInt_t triggerMask = AliVEvent::kMB;
49 AliAnalysisManager *mgr = new AliAnalysisManager("AnalysisTrainCorrJetsLocal", "Analysis Train Jet Correlation Local");
50
51 if (bIsAOD) {
52 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/train/AddAODHandler.C");
53 AliAODInputHandler *aodIH = AddAODHandler();
54 } else {
55 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/train/AddESDHandler.C");
56 AliESDInputHandler *esdIH = AddESDHandler();
57// esdIH->SetReadFriends(kFALSE);
58 }
59 if (bIsMC && !bIsAOD) {
60 AliMCEventHandler *mctEH = new AliMCEventHandler();
61 mcH->SetPreReadMode(AliMCEventHandler::kLmPreRead);
62 mcH->SetReadTR(kTRUE);
63 mgr->SetMCtruthEventHandler(mcH);
64 }
65//=============================================================================
66
67 if (bIsPhysSel && !bIsAOD && !bIsEMCalAna) {
68 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
69 AliPhysicsSelectionTask *taskPhysSel = AddTaskPhysicsSelection(bIsMC);
70 }
71
72 if (bIsCentSel && !bIsAOD) {
73 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskCentrality.C");
74 AliCentralitySelectionTask *taskCentrality = AddTaskCentrality();
75 if (bIsMC) taskCentrality->SetMCInput();
76 }
77
78 if (bIsEvnPSel) {
79 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskEventplane.C");
80 AliEPSelectionTask *taskEventPlane = AddTaskEventplane();
81 if (bIsMC) taskEventPlane->SetUseMCRP();
82 }
83
84 if (bIsRespPID) {
85 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
86 AliAnalysisTaskSE *taskRespPID = AddTaskPIDResponse(bIsMC);
87 }
88//=============================================================================
89
90 if (bIsEMCalAna) {
91 if (bIsEvnPSel) {
92 gROOT->LoadMacro("$ALICE_ROOT/PWG/EMCAL/macros/AddTaskEmcalPhysicsSelection.C");
93 AliPhysicsSelectionTask *taskPhysSel = AddTaskEmcalPhysicsSelection(bFastOnly, bHistoW, triggerMask,
94 dMinEclus, dMinPtClus,
95 dVz, bVzDiff,
96 bIsCentSel ? dCentMin : -1.,
97 bIsCentSel ? dCentMax : -1.,
98 dMinScaleCT, dMaxScaleCT);
99 }
100
101 gROOT->LoadMacro("$ALICE_ROOT/PWG/EMCAL/macros/AddTaskEmcalSetup.C");
102 AliEmcalSetupTask *taskSetupEMCal = AddTaskEmcalSetup();
103 taskSetupEMCal->SetGeoPath("$ALICE_ROOT/OADB/EMCAL");
104
105 gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskJetPreparation.C");
106 AddTaskJetPreparation(bIsAOD ? "AOD" : "ESD"); //TODO
107 }
108//=============================================================================
109
110 gROOT->LoadMacro("AddTasksFlavorJet.C"); AddTasksFlavorJet();
111 if (mgr->InitAnalysis()) { mgr->PrintStatus(); mgr->StartAnalysis("local",chain); }
112 return;
113}
114
115//=============================================================================
116TChain *CreateAODFriendChain(TString setData)
117{
118 if (!setData.EndsWith(".txt")) return 0;
119
120 TChain *chain = new TChain("aodTree");
121 TChain *cFrid = new TChain("aodTree");
122
123 TString dataFile;
124 ifstream dataList(setData.Data(), ios::in);
125 while (!dataList.eof()) {
126 dataFile.ReadLine(dataList,kFALSE);
127 if (!dataFile.EndsWith("AliAOD.root")) continue;
128 if (!gSystem->AccessPathName(dataFile.Data())) chain->Add(dataFile.Data());
129
130 dataFile.ReplaceAll("AliAOD.root","AliAOD.VertexingHF.root");
131 if (!gSystem->AccessPathName(dataFile.Data())) cFrid->Add(dataFile.Data());
132 } dataList.close();
133
134 chain->AddFriend(cFrid);
135 return chain;
136}
137
138//=============================================================================
139Bool_t LoadLibraries()
140{
141 if (gSystem->Load("libTree") <0) return kTRUE;
142 if (gSystem->Load("libGeom") <0) return kTRUE;
143 if (gSystem->Load("libPhysics") <0) return kTRUE;
144 if (gSystem->Load("libVMC") <0) return kTRUE;
145 if (gSystem->Load("libMinuit") <0) return kTRUE;
146 if (gSystem->Load("libMinuit2") <0) return kTRUE;
147
148 if (gSystem->Load("libCore") <0) return kTRUE;
149 if (gSystem->Load("libXMLIO") <0) return kTRUE;
150 if (gSystem->Load("libXMLParser") <0) return kTRUE;
151 if (gSystem->Load("libProof") <0) return kTRUE;
152 if (gSystem->Load("libProofPlayer")<0) return kTRUE;
153 if (gSystem->Load("libGui") <0) return kTRUE;
154//=============================================================================
155
156 gSystem->AddIncludePath("-Wno-deprecated");
157 gSystem->AddIncludePath("-I$ALICE_ROOT/include");
158 gSystem->AddIncludePath("-I$ALICE_ROOT/PWGHF/vertexingHF");
159 gSystem->AddIncludePath("-I$ALICE_ROOT/EMCAL");
160 gSystem->AddIncludePath("-I$ALICE_ROOT/JETAN");
161 gSystem->AddIncludePath("-I$ALICE_ROOT/JETAN/fastjet");
162
163 if (gSystem->Load("libSTEERBase") <0) return kTRUE;
164 if (gSystem->Load("libESD") <0) return kTRUE;
165 if (gSystem->Load("libAOD") <0) return kTRUE;
166 if (gSystem->Load("libANALYSIS") <0) return kTRUE;
167 if (gSystem->Load("libOADB") <0) return kTRUE;
168 if (gSystem->Load("libANALYSISalice") <0) return kTRUE;
169 if (gSystem->Load("libCORRFW") <0) return kTRUE;
170
171 if (gSystem->Load("libPWGTools") <0) return kTRUE;
172 if (gSystem->Load("libPWGflowBase") <0) return kTRUE;
173 if (gSystem->Load("libPWGflowTasks") <0) return kTRUE;
174 if (gSystem->Load("libPWGHFbase") <0) return kTRUE;
175 if (gSystem->Load("libPWGHFvertexingHF") <0) return kTRUE;
176
177 if (gSystem->Load("libSTAT") <0) return kTRUE;
178 if (gSystem->Load("libEMCALUtils") <0) return kTRUE;
179//if (gSystem->Load("libPHOSUtils") <0) return kTRUE;
180
181 if (gSystem->Load("libCDB") <0) return kTRUE;
182 if (gSystem->Load("libRAWDatabase") <0) return kTRUE;
183 if (gSystem->Load("libRAWDatarec") <0) return kTRUE;
184 if (gSystem->Load("libSTEER") <0) return kTRUE;
185 if (gSystem->Load("libITSbase") <0) return kTRUE;
186 if (gSystem->Load("libITSrec") <0) return kTRUE;
187 if (gSystem->Load("libTPCbase") <0) return kTRUE;
188 if (gSystem->Load("libTPCrec") <0) return kTRUE;
189 if (gSystem->Load("libTRDbase") <0) return kTRUE;
190 if (gSystem->Load("libTRDrec") <0) return kTRUE;
191 if (gSystem->Load("libTOFbase") <0) return kTRUE;
192//if (gSystem->Load("libTOFrec") <0) return kTRUE;
193 if (gSystem->Load("libHMPIDbase") <0) return kTRUE;
194 if (gSystem->Load("libEMCALraw") <0) return kTRUE;
195 if (gSystem->Load("libEMCALbase") <0) return kTRUE;
196 if (gSystem->Load("libEMCALrec") <0) return kTRUE;
197 if (gSystem->Load("libVZERObase") <0) return kTRUE;
198 if (gSystem->Load("libVZEROrec") <0) return kTRUE;
199 if (gSystem->Load("libTENDER") <0) return kTRUE;
200 if (gSystem->Load("libTENDERSupplies") <0) return kTRUE;
201
202 if (gSystem->Load("libCGAL") <0) return kTRUE;
203 if (gSystem->Load("libfastjet") <0) return kTRUE;
204 if (gSystem->Load("libsiscone") <0) return kTRUE;
205 if (gSystem->Load("libSISConePlugin") <0) return kTRUE;
206
207 if (gSystem->Load("libJETAN") <0) return kTRUE;
208 if (gSystem->Load("libFASTJETAN") <0) return kTRUE;
209 if (gSystem->Load("libPWGEMCAL") <0) return kTRUE;
210 if (gSystem->Load("libPWGGAEMCALTasks") <0) return kTRUE;
211 if (gSystem->Load("libPWGJEEMCALJetTasks")<0) return kTRUE;
212
213 if (gROOT->LoadMacro("AliAnalysisTaskSEDmesonsFilterCJ.cxx+") <0) return kTRUE;
214 if (gROOT->LoadMacro("AliAnalysisTaskFlavorJetCorrelations.cxx+")<0) return kTRUE;
215 return kFALSE;
216}