EMCALTask library splitting
[u/mrichter/AliRoot.git] / PWGGA / CaloTrackCorrelations / macros / ana.C
CommitLineData
d92b41ad 1/* $Id: $ */
d92b41ad 2//--------------------------------------------------
3// Example macro to do analysis with the
745913ae 4// analysis classes in CaloTrackCorrelations
d92b41ad 5// Can be executed with Root and AliRoot
6//
7// Pay attention to the options and definitions
8// set in the lines below
9//
10// Author : Gustavo Conesa Balbastre (INFN-LNF)
11//
12//-------------------------------------------------
3c867778 13enum anaModes {mLocal=0, mPROOF=1, mPlugin=2, mGRID=3};
2244659d 14//mLocal = 0: Analyze locally files in your computer
3c867778 15//mPROOF = 1: Analyze files on GRID with Plugin
16//mPlugin = 2: Analyze files on GRID with Plugin
17//mGRID = 3: Analyze files on GRID, jobs launched from aliensh
d92b41ad 18
19//---------------------------------------------------------------------------
3c867778 20// Settings to read locally several files, only for "mLocal" mode
21// The different values are default, they can be set with environmental
22// variables: INDIR, PATTERN, NFILES, respectivelly
54769bc0 23
3c867778 24char * kInDir = "/user/data/files/";
3e0577a2 25char * kPattern = ""; // Data are in files kInDir/kPattern+i
ac0610dc 26Int_t kFile = 2;
3c867778 27
d92b41ad 28//---------------------------------------------------------------------------
3c867778 29// Dataset for proof analysis, mode=mPROOF
30// char * kDataset = "/alice/vernet/PbPb_LHC10h_ESD";
31
32char * kDatasetPROOF = "/alice/vernet/LHC11b_149646";
3c867778 33Int_t kDatasetNMaxFiles = 20;
34TString ccin2p3UserName = "arbor" ;
35TString alienUserName = "narbor" ;
36
37//---------------------------------------------------------------------------
38// Collection file for grid analysis
39
d92b41ad 40char * kXML = "collection.xml";
3c867778 41
d92b41ad 42//---------------------------------------------------------------------------
43//Scale histograms from file. Change to kTRUE when xsection file exists
44//Put name of file containing xsection
45//Put number of events per ESD file
46//This is an specific case for normalization of Pythia files.
d92b41ad 47const char * kXSFileName = "pyxsec.root";
54769bc0 48
d92b41ad 49//---------------------------------------------------------------------------
50
193828fd 51//Set some default values, but used values are set in the code!
52
3c867778 53Bool_t kMC = kFALSE; //With real data kMC = kFALSE
193828fd 54TString kInputData = "ESD"; //ESD, AOD, MC, deltaAOD
3c867778 55Int_t kYear = 2011;
193828fd 56TString kCollision = "pp";
dc028d0b 57Bool_t outAOD = kFALSE; //Some tasks doesnt need it.
2244659d 58TString kTreeName;
3c867778 59TString kPass = "";
193828fd 60char kTrigger[1024];
3c867778 61Int_t kRun = 0;
7175a03a 62
3c867778 63//________________________
193828fd 64void ana(Int_t mode=mGRID)
d92b41ad 65{
66 // Main
cd54ca47 67
d92b41ad 68 //--------------------------------------------------------------------
69 // Load analysis libraries
70 // Look at the method below,
71 // change whatever you need for your analysis case
72 // ------------------------------------------------------------------
193828fd 73
3c867778 74 LoadLibraries(mode) ;
193828fd 75 //gSystem->ListLibraries();
2244659d 76
d92b41ad 77 //-------------------------------------------------------------------------------------------------
78 //Create chain from ESD and from cross sections files, look below for options.
c8fe2783 79 //-------------------------------------------------------------------------------------------------
54769bc0 80
193828fd 81 // Set kInputData and kTreeName looking to the kINDIR
cd54ca47 82
193828fd 83 CheckInputData(mode);
cd54ca47 84
193828fd 85 // Check global analysis settings
cd54ca47 86
193828fd 87 CheckEnvironmentVariables();
88
89 printf("*********************************************\n");
90 printf("*** Input data < %s >, pass %s, tree < %s >, MC? < %d > ***\n",kInputData.Data(),kPass.Data(),kTreeName.Data(),kMC);
91 printf("*********************************************\n");
2244659d 92
193828fd 93 TChain * chain = new TChain(kTreeName) ;
d92b41ad 94 TChain * chainxs = new TChain("Xsection") ;
193828fd 95 CreateChain(mode, chain, chainxs);
54769bc0 96
ac0610dc 97 Double_t scale = -1;
98 printf("===== kMC %d, chainxs %p\n",kMC,chainxs);
99 if(kMC && chainxs && chainxs->GetEntries() > 0)
100 {
101 Int_t nfiles = chainxs->GetEntries();
102
103 //Get the cross section
104 Double_t xsection = 0;
105 Float_t ntrials = 0;
106
107 GetAverageXsection(chainxs, xsection, ntrials);
108
1806f9c0 109 Int_t nEventsPerFile = chain->GetEntries() / nfiles;
110
111 Double_t trials = ntrials / nEventsPerFile ;
112
113 scale = xsection/trials;
114
115 printf("Get Cross section : nfiles %d, nevents %d, nevents per file %d \n",nfiles, chain->GetEntries(),nEventsPerFile);
116 printf(" ntrials %d, trials %2.2f, xs %2.2e, scale factor %2.2e\n", ntrials,trials,xsection,scale);
ac0610dc 117
ac0610dc 118 }
119
193828fd 120 printf("*********************************************\n");
121 printf("number of entries # %lld, skipped %d\n", chain->GetEntries()) ;
122 printf("*********************************************\n");
123
3c867778 124 if(!chain)
125 {
193828fd 126 printf("STOP, no chain available\n");
127 return;
128 }
129
130 AliLog::SetGlobalLogLevel(AliLog::kError);//Minimum prints on screen
cd54ca47 131
15882d9f 132 //------------------------------------------
133 // Alien handler part
134 //------------------------------------------
135 AliAnalysisGrid *alienHandler=0x0;
3c867778 136 if(mode==mPlugin)
137 {
15882d9f 138 // Create and configure the alien handler plugin
139 gROOT->LoadMacro("CreateAlienHandler.C");
140 alienHandler = CreateAlienHandler();
141 if (!alienHandler) return;
142 }
143
193828fd 144 //--------------------------------------
145 // Make the analysis manager
146 //-------------------------------------
147 AliAnalysisManager *mgr = new AliAnalysisManager("Manager", "Manager");
148 //AliAnalysisManager::SetUseProgressBar(kTRUE);
149 //mgr->SetSkipTerminate(kTRUE);
150 //mgr->SetNSysInfo(1);
151
3c867778 152 if(mode==mPlugin)
153 {
15882d9f 154 // Connect plugin to the analysis manager
155 mgr->SetGridHandler(alienHandler);
156 }
157
193828fd 158 // MC handler
3c867778 159 if((kMC || kInputData == "MC") && !kInputData.Contains("AOD"))
160 {
193828fd 161 AliMCEventHandler* mcHandler = new AliMCEventHandler();
162 mcHandler->SetReadTR(kFALSE);//Do not search TrackRef file
163 mgr->SetMCtruthEventHandler(mcHandler);
3c867778 164 if( kInputData == "MC")
165 {
193828fd 166 cout<<"MC INPUT EVENT HANDLER"<<endl;
167 mgr->SetInputEventHandler(NULL);
d92b41ad 168 }
193828fd 169 }
170
745913ae 171
193828fd 172 // AOD output handler
3c867778 173 if(kInputData!="deltaAOD" && outAOD)
174 {
193828fd 175 cout<<"Init output handler"<<endl;
176 AliAODHandler* aodoutHandler = new AliAODHandler();
177 aodoutHandler->SetOutputFileName("aod.root");
178 ////aodoutHandler->SetCreateNonStandardAOD();
179 mgr->SetOutputEventHandler(aodoutHandler);
180 }
181
182 //input
183
3c867778 184 if(kInputData == "ESD")
185 {
193828fd 186 // ESD handler
187 AliESDInputHandler *esdHandler = new AliESDInputHandler();
188 esdHandler->SetReadFriends(kFALSE);
189 mgr->SetInputEventHandler(esdHandler);
190 cout<<"ESD handler "<<mgr->GetInputEventHandler()<<endl;
191 }
3c867778 192 else if(kInputData.Contains("AOD"))
193 {
193828fd 194 // AOD handler
195 AliAODInputHandler *aodHandler = new AliAODInputHandler();
196 mgr->SetInputEventHandler(aodHandler);
745913ae 197 if(kInputData == "deltaAOD") aodHandler->AddFriend("deltaAODCaloTrackCorr.root");
193828fd 198 cout<<"AOD handler "<<mgr->GetInputEventHandler()<<endl;
199 }
023e6936 200
745913ae 201 //mgr->RegisterExternalFile("deltaAODCaloTrackCorr.root");
193828fd 202 //mgr->SetDebugLevel(1); // For debugging, do not uncomment if you want no messages.
203
204 TString outputFile = AliAnalysisManager::GetCommonFileName();
205
206 //-------------------------------------------------------------------------
531f682d 207 // Define task, put here any other task that you want to use.
193828fd 208 //-------------------------------------------------------------------------
209
210 // Physics selection
e9de0d57 211 if(kInputData=="ESD")
212 {
213 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
214 AliPhysicsSelectionTask* physSelTask = AddTaskPhysicsSelection(kMC);
215 }
193828fd 216
217 // Centrality
740c3286 218 if(kCollision=="PbPb" && kInputData=="ESD")
3c867778 219 {
2c18833a 220 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskCentrality.C");
221 AliCentralitySelectionTask *taskCentrality = AddTaskCentrality();
2c18833a 222 }
193828fd 223
740c3286 224 if(kCollision=="PbPb")
225 {
226 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskVZEROEPSelection.C");
227 AliVZEROEPSelectionTask * EPV0 = AddTaskVZEROEPSelection();
228
229 gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskEventplane.C");
230 AliEPSelectionTask * EP = AddTaskEventplane();
231 }
232
193828fd 233 // Simple event counting tasks
740c3286 234
235 gROOT->LoadMacro("$ALICE_ROOT/PWGGA/CaloTrackCorrelations/macros/AddTaskCounter.C");
236
3240657f 237 AliAnalysisTask* count = AddTaskCounter(""); // All
740c3286 238 AliAnalysisTask* countmb = AddTaskCounter("MB"); // Min Bias
3240657f 239 AliAnalysisTask* countany = AddTaskCounter("Any");
240 AliAnalysisTask* countint = AddTaskCounter("AnyINT");// Min Bias
241
3c867778 242 if(!kMC)
243 {
740c3286 244 AliAnalysisTaskCounter* countemg = AddTaskCounter("EMCEGA");
245 AliAnalysisTaskCounter* countemj = AddTaskCounter("EMCEJE");
531f682d 246 if(kCollision=="PbPb")
247 {
740c3286 248 AliAnalysisTaskCounter* countcen = AddTaskCounter("Central");
249 AliAnalysisTaskCounter* countsce = AddTaskCounter("SemiCentral");
250 AliAnalysisTaskCounter* countssce= AddTaskCounter("SemiOrCentral");
251 AliAnalysisTaskCounter* countphP = AddTaskCounter("PHOSPb");
531f682d 252 }
740c3286 253 else
254 {
255 AliAnalysisTaskCounter* countem1 = AddTaskCounter("EMC1"); // Trig Th > 1.5 GeV approx
256 AliAnalysisTaskCounter* countem7 = AddTaskCounter("EMC7"); // Trig Th > 4-5 GeV
257 AliAnalysisTaskCounter* countphp = AddTaskCounter("PHOS");
258 }
259 }
745913ae 260 // -----------------
261 // Photon conversion
262 // -----------------
263/*
264 if(kInputData=="ESD"){
265 printf("* Configure photon conversion analysis in macro \n");
266 TString arguments = "-run-on-train -use-own-xyz -force-aod -mc-off ";
531f682d 267 gROOT->LoadMacro("$ALICE_ROOT/PWGGA/GammaConversion/macros/ConfigGammaConversion.C");
745913ae 268 AliAnalysisTaskGammaConversion * taskGammaConversion =
269 ConfigGammaConversion(arguments,mgr->GetCommonInputContainer());
270 taskGammaConversion->SelectCollisionCandidates();
271
272 // Gamma Conversion AOD to AODPWG4Particle
273 AliAnalysisTaskGCPartToPWG4Part * taskGCToPC = new AliAnalysisTaskGCPartToPWG4Part("GCPartToPWG4Part");
274 taskGCToPC->SetGammaCutId("90035620401003321022000000090");
275 mgr->AddTask(taskGCToPC);
276 mgr->ConnectInput (taskGCToPC, 0, mgr->GetCommonInputContainer() );
277 mgr->ConnectOutput (taskGCToPC, 0, mgr->GetCommonOutputContainer());
278 }
279*/
3c867778 280
193828fd 281 Bool_t kPrint = kFALSE;
282 Bool_t deltaAOD = kFALSE;
5fdef45d 283 gROOT->LoadMacro("AddTaskCaloTrackCorr.C"); // $ALICE_ROOT/PWGGA/CaloTrackCorrelations/macros
531f682d 284 gROOT->LoadMacro("$ALICE_ROOT/PWGGA/EMCALTasks/macros/AddTaskEMCALClusterize.C"); // $ALICE_ROOT/PWGGA/EMCALTasks/macros
15882d9f 285
ac0610dc 286 //gROOT->LoadMacro("$ALICE_ROOT/PWGGA/CaloTrackCorrelations/macros/QA/AddTaskCalorimeterQA.C");
287 //AliAnalysisTaskCaloTrackCorrelation * qatask = AddTaskCalorimeterQA(kInputData,kYear,kPrint,kMC);
193828fd 288
dc028d0b 289 //gROOT->LoadMacro("$ALICE_ROOT/PWGGA/EMCALTasks/macros/AddTaskEMCALTriggerQA.C");
290 //AliAnalysisTaskEMCALTriggerQA * qatrigtask = AddTaskEMCALTriggerQA();
291
023e6936 292 // Calibration, bad map ...
293
1806f9c0 294 Bool_t calibEE = kTRUE; // It is set automatically, but here we force to use it or not in any case
295 Bool_t calibTT = kTRUE; // It is set automatically, but here we force to use it or not in any case
ac0610dc 296 if(kRun < 122195 || (kRun > 126437 && kRun < 136851) || kMC) calibTT=kFALSE ; // Recalibration parameters not available for LHC10a,b,c,e,f,g
1806f9c0 297 Bool_t badMap = kTRUE; // It is set automatically, but here we force to use it or not in any case
023e6936 298
531f682d 299 if(kCollision=="pp")
193828fd 300 {
531f682d 301 printf("==================================== \n");
302 printf("CONFIGURE ANALYSIS FOR PP COLLISIONS \n");
303 printf("==================================== \n");
cd54ca47 304
740c3286 305 Bool_t mixing = kTRUE; // mixing in reader and hadron correlation, and pi0
531f682d 306 Bool_t clTM = kTRUE;
307 Bool_t reTM = kFALSE; // Recalculate matches if not already done in clusterizer
308 Bool_t anTM = kTRUE; // Remove matched
309 Bool_t exo = kTRUE; // Remove exotic cells
e9de0d57 310 Bool_t clnonlin = kTRUE; // Apply non linearity (clusterization)
531f682d 311 Bool_t annonlin = kFALSE; // Apply non linearity (analysis)
312 Int_t minEcell = 50; // 50 MeV (10 MeV used in reconstruction)
313 Int_t minEseed = 100; // 100 MeV
314 Int_t dTime = 0; // default, 250 ns
ac0610dc 315 Int_t wTime = 30; // default 425 < T < 825 ns, careful if time calibration is on
d7d24d99 316 Int_t unfMinE = 15; // Remove cells with less than 15 MeV from cluster after unfolding
317 Int_t unfFrac = 1; // Remove cells with less than 1% of cluster energy after unfolding
318
023e6936 319 //Trigger
ac0610dc 320 TString clTrigger = "";
023e6936 321 TString anTrigger = "EMC7";
ac0610dc 322
531f682d 323 if(kMC)
324 {
ac0610dc 325 clTrigger = "";
326 anTrigger = "";
6c4dd754 327 dTime = 0;
328 wTime = 0;
531f682d 329 }
54769bc0 330
531f682d 331 Bool_t selectEvents = kFALSE; // Select events depending on V0, pile-up and vertex quality
332 Bool_t qa = kTRUE; // Do besides calorimeter QA analysis
333 Bool_t hadron = kTRUE; // Do besides charged track correlations analysis
cd54ca47 334
531f682d 335 //Analysis with clusterizer V1
3c867778 336
531f682d 337 TString arrayNameV1 = "";
ac0610dc 338
dc028d0b 339 AliAnalysisTaskEMCALClusterize * clv1 = AddTaskEMCALClusterize(arrayNameV1,outAOD,kMC,exo,"V1",clTrigger, clTM,
d7d24d99 340 minEcell,minEseed,dTime,wTime,unfMinE,unfFrac,
e9de0d57 341 calibEE,badMap,calibTT,clnonlin);
54769bc0 342
531f682d 343 printf("Name of clusterizer1 array: %s\n",arrayNameV1.Data());
344
ac0610dc 345 if(!kMC)
346 {
347
348 AliAnalysisTaskCaloTrackCorrelation *anav1trig = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 349 kYear,kCollision,anTrigger,arrayNameV1,mixing,reTM,anTM,
ac0610dc 350 -1,-1, qa, hadron,calibEE,badMap,calibTT,deltaAOD,kPrint,scale);
351 }
352
353 AliAnalysisTaskCaloTrackCorrelation *anav1mb = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 354 kYear,kCollision,"AnyINT",arrayNameV1,mixing,reTM,anTM,
ac0610dc 355 -1,-1, qa, hadron,calibEE,badMap,calibTT,deltaAOD,kPrint,scale);
356
357
531f682d 358
359 //Analysis with clusterizer V2
360 TString arrayNameV2 = "";
dc028d0b 361 AliAnalysisTaskEMCALClusterize * clv2 = AddTaskEMCALClusterize(arrayNameV2,outAOD,kMC,exo,"V2",clTrigger, clTM,
ac0610dc 362 minEcell,minEseed,dTime,wTime,
e9de0d57 363 calibEE,badMap,calibTT,clnonlin);
ac0610dc 364
531f682d 365 printf("Name of clusterizer2 array: %s\n",arrayNameV2.Data());
366
367 hadron = kFALSE;
ac0610dc 368 if(!kMC)
369 {
370
371
372 AliAnalysisTaskCaloTrackCorrelation *anav2tr = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 373 kYear,kCollision,anTrigger,arrayNameV2,mixing,reTM,anTM,
ac0610dc 374 -1,-1,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint,scale);
375 }
376
377
378 AliAnalysisTaskCaloTrackCorrelation *anav2mb = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 379 kYear,kCollision,"AnyINT",arrayNameV2,mixing,reTM,anTM,
ac0610dc 380 -1,-1, qa, hadron,calibEE,badMap,calibTT,deltaAOD,kPrint,scale);
531f682d 381 }
ac0610dc 382
531f682d 383 if(kCollision=="PbPb")
3c867778 384 {
531f682d 385 printf("====================================== \n");
386 printf("CONFIGURE ANALYSIS FOR PbPb COLLISIONS \n");
387 printf("====================================== \n");
740c3286 388 Bool_t mixing = kTRUE;
531f682d 389 Bool_t clTM = kTRUE;
390 Bool_t reTM = kFALSE; // Recalculate matches if not already done in clusterizer
391 Bool_t anTM = kTRUE; // Remove matched
392 Bool_t exo = kTRUE; // Remove exotic cells
e9de0d57 393 Bool_t clnonlin = kTRUE; // Apply non linearity (clusterization)
531f682d 394 Bool_t annonlin = kFALSE; // Apply non linearity (analysis)
dc028d0b 395 Int_t minEcell = 150; // 50 MeV (10 MeV used in reconstruction)
396 Int_t minEseed = 300; // 100 MeV
531f682d 397 Int_t dTime = 0; // default, 250 ns
398 Int_t wTime = 0; // default 425 < T < 825 ns
d7d24d99 399 Int_t unfMinE = 15; // Remove cells with less than 15 MeV from cluster after unfolding
400 Int_t unfFrac = 1; // Remove cells with less than 1% of cluster energy after unfolding
ac0610dc 401
023e6936 402 // Trigger
ac0610dc 403 TString clTrigger = "";
531f682d 404 TString anTrigger = "EMCGA";
405 if(kMC)
406 {
407 clTrigger = "";
408 anTrigger = "";
6c4dd754 409 dTime = 0;
410 wTime = 0;
531f682d 411 }
3c867778 412
531f682d 413 Bool_t selectEvents = kFALSE; // Select events depending on V0, pile-up and vertex quality
414 Bool_t qa = kTRUE; // Do besides calorimeter QA analysis
415 Bool_t hadron = kTRUE; // Do besides charged track correlations analysis
3c867778 416
531f682d 417 //Analysis with clusterizer V1
3c867778 418
531f682d 419 TString arrayNameV1 = "";
dc028d0b 420 AliAnalysisTaskEMCALClusterize * clv1 = AddTaskEMCALClusterize(arrayNameV1,outAOD,kMC,exo,"V1",clTrigger, clTM,
d7d24d99 421 minEcell,minEseed,dTime,wTime,unfMinE,unfFrac,
e9de0d57 422 calibEE,badMap,calibTT,clnonlin);
3c867778 423
531f682d 424 printf("Name of clusterizer1 array: %s\n",arrayNameV1.Data());
425
dc028d0b 426 if(!kMC)
427 {
428 AliAnalysisTaskCaloTrackCorrelation *anav1c = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 429 kYear,kCollision,anTrigger,arrayNameV1,mixing,reTM,anTM,
dc028d0b 430 0,20,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
431 AliAnalysisTaskCaloTrackCorrelation *anav1m = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 432 kYear,kCollision,anTrigger,arrayNameV1,mixing,reTM,anTM,
dc028d0b 433 20,40,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
434 AliAnalysisTaskCaloTrackCorrelation *anav1p = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 435 kYear,kCollision,anTrigger,arrayNameV1,mixing,reTM,anTM,
dc028d0b 436 60,80,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
437 }
ac0610dc 438
531f682d 439 //Analysis with clusterizer V2
ac0610dc 440
531f682d 441 TString arrayNameV2 = "";
dc028d0b 442 AliAnalysisTaskEMCALClusterize * clv2 = AddTaskEMCALClusterize(arrayNameV2,outAOD,kMC,exo,"V2",clTrigger, clTM,
d7d24d99 443 minEcell,minEseed,dTime,wTime,unfMinE,unfFrac,
e9de0d57 444 calibEE,badMap,calibTT,clnonlin);
531f682d 445
446 printf("Name of clusterizer2 array: %s\n",arrayNameV2.Data());
447
448 hadron = kFALSE;
449
ac0610dc 450 if(!kMC)
451 {
452
453 AliAnalysisTaskCaloTrackCorrelation *anav2cT = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 454 kYear,kCollision,anTrigger,arrayNameV2,mixing,reTM,anTM,
ac0610dc 455 0,20,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
456 AliAnalysisTaskCaloTrackCorrelation *anav2mT = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 457 kYear,kCollision,anTrigger,arrayNameV2,mixing,reTM,anTM,
ac0610dc 458 20,40,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
459 AliAnalysisTaskCaloTrackCorrelation *anav2pT = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 460 kYear,kCollision,anTrigger,arrayNameV2,mixing,reTM,anTM,
ac0610dc 461 60,80,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
462 }
463
464 AliAnalysisTaskCaloTrackCorrelation *anav2cMB = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 465 kYear,kCollision,"AnyINT",arrayNameV2,mixing,reTM,anTM,
ac0610dc 466 0,20,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
467 AliAnalysisTaskCaloTrackCorrelation *anav2mMB = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 468 kYear,kCollision,"AnyINT",arrayNameV2,mixing,reTM,anTM,
ac0610dc 469 20,40,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
470 AliAnalysisTaskCaloTrackCorrelation *anav2pMB = AddTaskCaloTrackCorr(kInputData, "EMCAL", kMC, selectEvents, exo, annonlin, outputFile.Data(),
740c3286 471 kYear,kCollision,"AnyINT",arrayNameV2,mixing,reTM,anTM,
ac0610dc 472 60,80,qa,hadron,calibEE,badMap,calibTT,deltaAOD,kPrint);
473
474
531f682d 475 }
476
ac0610dc 477
478
193828fd 479 //-----------------------
480 // Run the analysis
481 //-----------------------
482 mgr->InitAnalysis();
483 mgr->PrintStatus();
193828fd 484
3c867778 485 if (mode == mPlugin) mgr->StartAnalysis("grid");
486 else if (mode == mPROOF ) mgr->StartAnalysis("proof",chain);
487 else mgr->StartAnalysis("local",chain);
193828fd 488
3c867778 489 cout <<" Analysis ended sucessfully "<< endl ;
d92b41ad 490
491}
492
3c867778 493//_____________________________
494void LoadLibraries(Int_t mode)
193828fd 495{
d92b41ad 496
3c867778 497 if (mode == mPROOF) {
498 //TProof::Mgr("ccalpmaster")->SetROOTVersion("ALICE_v5-27-06b");
499 gROOT->LoadMacro("/afs/in2p3.fr/group/alice/laf/EnableAliRootForLAF.C");
dce3a2ba 500 TProof* proof = EnableAliRootForLAF("ccaplmaster",nPROOFWorkers.Data(),ccin2p3UserName.Data(),alienUserName.Data(),"",kFALSE,kTRUE,kTRUE,"OADB:ANALYSIS:ANALYSISalice:AOD:ESD:CORRFW:STEERBase:EMCALUtils:PHOSUtils:PWGCaloTrackCorrBase:PWGGACaloTrackCorrelations:PWGEMCAL:PWGGAEMCALTasks");
3c867778 501
502 // TProof* proof = TProof::Open("ccaplmaster",Form("workers=%s",nPROOFWorkers.Data()));
503
504 // //proof->ClearPackages();
505 // proof->UploadPackage("STEERBase");
506 // proof->UploadPackage("ESD");
507 // proof->UploadPackage("AOD");
508 // proof->UploadPackage("ANALYSIS");
509 // proof->UploadPackage("OADB");
510 // proof->UploadPackage("ANALYSISalice");
511 // proof->UploadPackage("CORRFW");
512 // //proof->UploadPackage("JETAN");
513 // proof->UploadPackage("PHOSUtils");
514 // proof->UploadPackage("EMCALUtils");
531f682d 515 // proof->UploadPackage("PWGCaloTrackCorrBase");
5fdef45d 516 // proof->UploadPackage("PWGGACaloTrackCorrelations");
dce3a2ba 517 // proof->UploadPackage("PWGEMCAL");
5fdef45d 518 // proof->UploadPackage("PWGGAEMCALTasks");
3c867778 519
520 // proof->EnablePackage("STEERBase");
521 // proof->EnablePackage("ESD");
522 // proof->EnablePackage("AOD");
523 // proof->EnablePackage("ANALYSIS");
524 // proof->EnablePackage("OADB");
525 // proof->EnablePackage("ANALYSISalice");
526 // proof->EnablePackage("CORRFW");
527 // //proof->EnablePackage("JETAN");
528 // proof->EnablePackage("PHOSUtils");
529 // proof->EnablePackage("EMCALUtils");
531f682d 530 // proof->EnablePackage("PWGCaloTrackCorrBase");
5fdef45d 531 // proof->EnablePackage("PWGGACaloTrackCorrelations");
dce3a2ba 532 // proof->EnablePackage("PWGEMCAL");
5fdef45d 533 // proof->EnablePackage("PWGGAEMCALTasks");
3c867778 534 return;
535 }
536
d92b41ad 537 //--------------------------------------
538 // Load the needed libraries most of them already loaded by aliroot
539 //--------------------------------------
540 gSystem->Load("libTree.so");
541 gSystem->Load("libGeom.so");
542 gSystem->Load("libVMC.so");
543 gSystem->Load("libXMLIO.so");
54769bc0 544 gSystem->Load("libMatrix.so");
545 gSystem->Load("libPhysics.so");
193828fd 546 gSystem->Load("libMinuit.so"); // Root + libraries to if reclusterization is done
547
548 gSystem->Load("libSTEERBase.so");
549 gSystem->Load("libGui.so"); // Root + libraries to if reclusterization is done
550 gSystem->Load("libCDB.so"); // Root + libraries to if reclusterization is done
551 gSystem->Load("libESD.so"); // Root + libraries to if reclusterization is done
552 gSystem->Load("libAOD.so");
553 gSystem->Load("libRAWDatabase.so"); // Root + libraries to if reclusterization is done
554 gSystem->Load("libProof.so");
531f682d 555 gSystem->Load("libOADB");
193828fd 556 gSystem->Load("libANALYSIS.so");
557 gSystem->Load("libSTEER.so"); // Root + libraries to if reclusterization is done
558
559 gSystem->Load("libRAWDatarec.so"); // Root + libraries to if reclusterization is done
560 gSystem->Load("libRAWDatasim.so"); // Root + libraries to if reclusterization is done
561 gSystem->Load("libVZERObase.so"); // Root + libraries to if reclusterization is done
562 gSystem->Load("libVZEROrec.so"); // Root + libraries to if reclusterization is done
563
564 gSystem->Load("libEMCALUtils");
565 //SetupPar("EMCALUtils");
566 gSystem->Load("libEMCALraw"); // Root + libraries to if reclusterization is done
567 gSystem->Load("libEMCALbase"); // Root + libraries to if reclusterization is done
568 gSystem->Load("libEMCALsim"); // Root + libraries to if reclusterization is done
569 gSystem->Load("libEMCALrec"); // Root + libraries to if reclusterization is done
570 //SetupPar("EMCALraw");
571 //SetupPar("EMCALbase");
572 //SetupPar("EMCALsim");
573 //SetupPar("EMCALrec");
574
575 gSystem->Load("libANALYSISalice.so");
576 //gSystem->Load("libTENDER.so");
577 //gSystem->Load("libTENDERSupplies.so");
578
579 gSystem->Load("libPHOSUtils");
580 gSystem->Load("libEMCALUtils");
372e67a3 581 gSystem->Load("libPWGCaloTrackCorrBase");
582 gSystem->Load("libPWGGACaloTrackCorrelations");
dce3a2ba 583 gSystem->Load("libPWGEMCAL");
5fdef45d 584 gSystem->Load("libPWGGAEMCALTasks");
372e67a3 585 //SetupPar("PWGCaloTrackCorrBase");
586 //SetupPar("PWGGACaloTrackCorrelations");
dce3a2ba 587 //SetupPar("PWGEMCAL");
5fdef45d 588 //SetupPar("PWGGAEMCALTasks");
15882d9f 589
531f682d 590
15882d9f 591 //gSystem->Load("libJETAN");
592 //gSystem->Load("FASTJETAN");
5fdef45d 593 //gSystem->Load("PWGJE");
745913ae 594
531f682d 595 //gSystem->Load("libCORRFW.so");
596 //gSystem->Load("libPWGGAGammaConv.so");
5fdef45d 597 //SetupPar("PWGGAGammaConv");
15882d9f 598
599 // needed for plugin?
600 gSystem->AddIncludePath("-I$ALICE_ROOT");
601 gSystem->AddIncludePath("-I./");
602
d92b41ad 603}
604
3c867778 605//_________________________________
d92b41ad 606void SetupPar(char* pararchivename)
607{
608 //Load par files, create analysis libraries
609 //For testing, if par file already decompressed and modified
610 //classes then do not decompress.
193828fd 611
d92b41ad 612 TString cdir(Form("%s", gSystem->WorkingDirectory() )) ;
613 TString parpar(Form("%s.par", pararchivename)) ;
193828fd 614
d92b41ad 615 if ( gSystem->AccessPathName(pararchivename) ) {
616 TString processline = Form(".! tar xvzf %s",parpar.Data()) ;
617 gROOT->ProcessLine(processline.Data());
618 }
619
620 TString ocwd = gSystem->WorkingDirectory();
621 gSystem->ChangeDirectory(pararchivename);
622
623 // check for BUILD.sh and execute
624 if (!gSystem->AccessPathName("PROOF-INF/BUILD.sh")) {
625 printf("*******************************\n");
626 printf("*** Building PAR archive ***\n");
627 cout<<pararchivename<<endl;
628 printf("*******************************\n");
629
630 if (gSystem->Exec("PROOF-INF/BUILD.sh")) {
631 Error("runProcess","Cannot Build the PAR Archive! - Abort!");
632 return -1;
633 }
634 }
635 // check for SETUP.C and execute
636 if (!gSystem->AccessPathName("PROOF-INF/SETUP.C")) {
637 printf("*******************************\n");
638 printf("*** Setup PAR archive ***\n");
639 cout<<pararchivename<<endl;
640 printf("*******************************\n");
641 gROOT->Macro("PROOF-INF/SETUP.C");
642 }
643
644 gSystem->ChangeDirectory(ocwd.Data());
645 printf("Current dir: %s\n", ocwd.Data());
646}
647
3c867778 648//______________________________________
649void CheckInputData(const anaModes mode)
650{
193828fd 651 //Sets input data and tree
652
653 TString ocwd = gSystem->WorkingDirectory();
654
655 //---------------------------------------
656 //Local files analysis
657 //---------------------------------------
658 if(mode == mLocal){
659 //If you want to add several ESD files sitting in a common directory INDIR
660 //Specify as environmental variables the directory (INDIR), the number of files
661 //to analyze (NFILES) and the pattern name of the directories with files (PATTERN)
cd54ca47 662
193828fd 663 if(gSystem->Getenv("INDIR"))
664 kInDir = gSystem->Getenv("INDIR") ;
665 else cout<<"INDIR not set, use default: "<<kInDir<<endl;
666
667 TString sindir(kInDir);
668 if (sindir.Contains("pass1")) kPass = "pass1";
669 else if(sindir.Contains("pass2")) kPass = "pass2";
670 else if(sindir.Contains("pass3")) kPass = "pass3";
671
672 if(gSystem->Getenv("PATTERN"))
673 kPattern = gSystem->Getenv("PATTERN") ;
674 else cout<<"PATTERN not set, use default: "<<kPattern<<endl;
675
676 cout<<"INDIR : "<<kInDir<<endl;
677 cout<<"NFILES : "<<kFile<<endl;
678
679 char fileE[120] ;
680 char fileA[120] ;
681 char fileG[120] ;
682 char fileEm[120] ;
683 for (Int_t event = 0 ; event < kFile ; event++) {
3c867778 684 sprintf(fileE, "%s/%s%d/AliESDs.root", kInDir,kPattern,event) ;
685 sprintf(fileA, "%s/%s%d/AliAOD.root", kInDir,kPattern,event) ;
686 sprintf(fileG, "%s/%s%d/galice.root", kInDir,kPattern,event) ;
193828fd 687 sprintf(fileEm, "%s/%s%d/embededAOD.root", kInDir,kPattern,event) ;
688
5eef1db1 689 TFile * fESD = TFile::Open(fileE) ;
690 TFile * fAOD = TFile::Open(fileA) ;
691
193828fd 692 //Check if file exists and add it, if not skip it
3c867778 693 if (fESD)
694 {
695 kTreeName = "esdTree";
193828fd 696 kInputData = "ESD";
5eef1db1 697 TFile * fG = TFile::Open(fileG);
698 if(fG) { kMC = kTRUE; fG->Close();}
699 else kMC = kFALSE;
700
701 // Get run number
702 TTree* esdTree = (TTree*)fESD->Get("esdTree");
703 AliESDEvent* esd = new AliESDEvent();
704 esd->ReadFromTree(esdTree);
705 esdTree->GetEvent(0);
706 kRun = esd->GetRunNumber();
707
193828fd 708 return;
709 }
3c867778 710 else if(fAOD)
711 {
712 kTreeName = "aodTree";
193828fd 713 kInputData = "AOD";
714 if(((TTree*) fAOD->Get("aodTree"))->GetBranch("mcparticles")) kMC=kTRUE;
715 else kMC = kFALSE;
5eef1db1 716
717 // Get run number
718 TTree* aodTree = (TTree*)fAOD->Get("aodTree");
719 AliAODEvent* aod = new AliAODEvent();
0fb69ade 720 aod->ReadFromTree(aodTree);
5eef1db1 721 aodTree->GetEvent(0);
722 kRun = aod->GetRunNumber();
193828fd 723 return;
724 }
3c867778 725 else if(TFile::Open(fileEm))
726 {
5eef1db1 727 kTreeName = "aodTree";
193828fd 728 kInputData = "AOD";
5eef1db1 729 kMC = kTRUE;
730
193828fd 731 return;
732 }
3c867778 733 else if(TFile::Open(fileG))
734 {
5eef1db1 735 kTreeName = "TE";
193828fd 736 kInputData = "MC";
5eef1db1 737 kMC = kTRUE;
193828fd 738 return;
739 }
740 }
741
5eef1db1 742 if(fESD) fESD->Close();
743 if(fAOD) fAOD->Close();
744
193828fd 745 }// local files analysis
746
747 //------------------------------
748 //GRID xml files
749 //-----------------------------
750 else if(mode == mGRID){
751 //Get colection file. It is specified by the environmental
752 //variable XML
753
754 if(gSystem->Getenv("XML") )
755 kXML = gSystem->Getenv("XML");
756 else
757 sprintf(kXML, "collection.xml") ;
758
759 if (!TFile::Open(kXML)) {
760 printf("No collection file with name -- %s -- was found\n",kXML);
761 return ;
762 }
763 else cout<<"XML file "<<kXML<<endl;
764
765 //Load necessary libraries and connect to the GRID
766 gSystem->Load("libNetx.so") ;
767 gSystem->Load("libRAliEn.so");
768 TGrid::Connect("alien://") ;
769
770 //Feed Grid with collection file
771 TGridCollection * collection = (TGridCollection*) TAlienCollection::Open(kXML);
772 if (! collection) {
773 AliError(Form("%s not found", kXML)) ;
774 return kFALSE ;
775 }
776 TGridResult* result = collection->GetGridResult("",0 ,0);
777
778 for (Int_t index = 0; index < result->GetEntries(); index++) {
779 TString alienURL = result->GetKey(index, "turl") ;
780 cout << "================== " << alienURL << endl ;
781
782 if (alienURL.Contains("pass1")) kPass = "pass1";
783 else if(alienURL.Contains("pass2")) kPass = "pass2";
784 else if(alienURL.Contains("pass3")) kPass = "pass3";
785
3c867778 786 kRun = AliAnalysisManager::GetRunFromAlienPath(alienURL.Data());
787 printf("Run number from alien path = %d\n",kRun);
788
193828fd 789 TFile * fAOD = 0 ;
790 //Check if file exists and add it, if not skip it
3c867778 791 if (alienURL.Contains("AliESDs.root"))
792 {
793 kTreeName = "esdTree";
193828fd 794 kInputData = "ESD";
795 alienURL.ReplaceAll("AliESDs.root","galice.root");
796 if(TFile::Open(alienURL)) kMC=kTRUE;
797 else kMC = kFALSE;
193828fd 798 return;
799 }
3c867778 800 else if(alienURL.Contains("AliAOD.root"))
801 {
802 kTreeName = "aodTree";
193828fd 803 kInputData = "AOD";
804 fAOD = TFile::Open(alienURL);
805 if(((TTree*) fAOD->Get("aodTree"))->GetBranch("mcparticles")) kMC=kTRUE;
806 else kMC = kFALSE;
807 return;
808 }
3c867778 809 else if(alienURL.Contains("embededAOD.root"))
810 {
811 kTreeName = "aodTree";
193828fd 812 kInputData = "AOD";
813 kMC=kTRUE;
814 return;
815 }
3c867778 816 else if(alienURL.Contains("galice.root"))
817 {
818 kTreeName = "TE";
193828fd 819 kInputData = "MC";
820 kMC=kTRUE;
821 return;
822 }
823 }
824 }// xml analysis
3c867778 825 //------------------------------
826 //PROOF files
827 //-----------------------------
828 else if(mode == mPROOF){
829
830 TFileCollection* coll = gProof->GetDataSet(kDatasetPROOF)->GetStagedSubset();
831
832 TIter iter(coll->GetList());
833
834 TFileInfo* fileInfo = 0;
835 while ((fileInfo = dynamic_cast<TFileInfo*> (iter())))
836 {
837 if (fileInfo->GetFirstUrl()) {
838 TString ProofURL = fileInfo->GetFirstUrl()->GetUrl();
839 cout << "================== " << ProofURL << endl ;
840
841 if (ProofURL.Contains("pass1")) kPass = "pass1";
842 else if(ProofURL.Contains("pass2")) kPass = "pass2";
843 else if(ProofURL.Contains("pass3")) kPass = "pass3";
844
845 kRun = AliAnalysisManager::GetRunFromAlienPath(ProofURL.Data());
846 printf("Run number from alien path = %d\n",kRun);
847
848 TFile * fAOD = 0 ;
849 //Check if file exists and add it, if not skip it
850 if (ProofURL.Contains("AliESDs.root"))
851 {
852 kTreeName = "esdTree";
853 kInputData = "ESD";
854 alienURL.ReplaceAll("AliESDs.root","galice.root");
855 if(TFile::Open(ProofURL)) kMC=kTRUE;
856 else kMC = kFALSE;
857
858 return;
859 }
860 else if(ProofURL.Contains("AliAOD.root"))
861 {
862 kTreeName = "aodTree";
863 kInputData = "AOD";
864 fAOD = TFile::Open(ProofURL);
865 if(((TTree*) fAOD->Get("aodTree"))->GetBranch("mcparticles")) kMC=kTRUE;
866 else kMC = kFALSE;
867 return;
868 }
869 else if(ProofURL.Contains("embededAOD.root"))
870 {
871 kTreeName = "aodTree";
872 kInputData = "AOD";
873 kMC=kTRUE;
874 return;
875 }
876 else if(ProofURL.Contains("galice.root"))
877 {
878 kTreeName = "TE";
879 kInputData = "MC";
880 kMC=kTRUE;
881 return;
882 }
883 }
884 }
885 }// proof analysis
193828fd 886
887 gSystem->ChangeDirectory(ocwd.Data());
888
889}
d92b41ad 890
193828fd 891//_____________________________________________________________________
892void CreateChain(const anaModes mode, TChain * chain, TChain * chainxs)
893{
d92b41ad 894 //Fills chain with data
895 TString ocwd = gSystem->WorkingDirectory();
896
d92b41ad 897 //---------------------------------------
3c867778 898 // Local files analysis
d92b41ad 899 //---------------------------------------
2244659d 900 if(mode == mLocal){
d92b41ad 901 //If you want to add several ESD files sitting in a common directory INDIR
902 //Specify as environmental variables the directory (INDIR), the number of files
c8fe2783 903 //to analyze (NFILES) and the pattern name of the directories with files (PATTERN)
193828fd 904
d92b41ad 905 if(gSystem->Getenv("INDIR"))
906 kInDir = gSystem->Getenv("INDIR") ;
907 else cout<<"INDIR not set, use default: "<<kInDir<<endl;
908
909 if(gSystem->Getenv("PATTERN"))
910 kPattern = gSystem->Getenv("PATTERN") ;
911 else cout<<"PATTERN not set, use default: "<<kPattern<<endl;
912
c8fe2783 913 if(gSystem->Getenv("NFILES"))
914 kFile = atoi(gSystem->Getenv("NFILES")) ;
915 else cout<<"NFILES not set, use default: "<<kFile<<endl;
d92b41ad 916
917 //Check if env variables are set and are correct
c8fe2783 918 if ( kInDir && kFile) {
919 printf("Get %d files from directory %s\n",kFile,kInDir);
d92b41ad 920 if ( ! gSystem->cd(kInDir) ) {//check if ESDs directory exist
193828fd 921 printf("%s does not exist\n", kInDir) ;
922 return ;
d92b41ad 923 }
193828fd 924
c8fe2783 925 //if(gSystem->Getenv("XSFILE"))
926 //kXSFileName = gSystem->Getenv("XSFILE") ;
927 //else cout<<" XS file name not set, use default: "<<kXSFileName<<endl;
928 char * kGener = gSystem->Getenv("GENER");
929 if(kGener) {
193828fd 930 cout<<"GENER "<<kGener<<endl;
3c867778 931 if (!strcmp(kGener,"PYTHIA")) kXSFileName = "pyxsec.root";
193828fd 932 else if(!strcmp(kGener,"HERWIG")) kXSFileName = "hexsec.root";
933 else cout<<" UNKNOWN GENER, use default: "<<kXSFileName<<endl;
c8fe2783 934 }
935 else cout<<" GENER not set, use default xs file name: "<<kXSFileName<<endl;
193828fd 936
3c867778 937 cout<<"INDIR : "<<kInDir <<endl;
938 cout<<"NFILES : "<<kFile <<endl;
939 cout<<"PATTERN : "<<kPattern <<endl;
c8fe2783 940 cout<<"XSFILE : "<<kXSFileName<<endl;
193828fd 941
7175a03a 942 TString datafile="";
3c867778 943 if (kInputData == "ESD") datafile = "AliESDs.root" ;
944 else if(kInputData.Contains("AOD")) datafile = "AliAOD.root" ;
945 else if(kInputData == "MC") datafile = "galice.root" ;
7175a03a 946
5eef1db1 947 //Loop on ESD/AOD/MC files, add them to chain
d92b41ad 948 Int_t event =0;
949 Int_t skipped=0 ;
950 char file[120] ;
951 char filexs[120] ;
952
c8fe2783 953 for (event = 0 ; event < kFile ; event++) {
3c867778 954 sprintf(file, "%s/%s%d/%s", kInDir,kPattern,event,datafile.Data()) ;
193828fd 955 sprintf(filexs, "%s/%s%d/%s", kInDir,kPattern,event,kXSFileName) ;
5eef1db1 956 TFile * fData = 0 ;
193828fd 957 //Check if file exists and add it, if not skip it
5eef1db1 958 if ( fData = TFile::Open(file)) {
959 if ( fData->Get(kTreeName) ) {
193828fd 960 printf("++++ Adding %s\n", file) ;
961 chain->AddFile(file);
962 chainxs->Add(filexs) ;
963 }
964 }
965 else {
966 printf("---- Skipping %s\n", file) ;
967 skipped++ ;
968 }
d92b41ad 969 }
d92b41ad 970 }
971 else {
972 TString input = "AliESDs.root" ;
973 cout<<">>>>>> No list added, take a single file <<<<<<<<< "<<input<<endl;
974 chain->AddFile(input);
975 }
976
977 }// local files analysis
978
979 //------------------------------
3c867778 980 // GRID xml files
981 //------------------------------
d92b41ad 982 else if(mode == mGRID){
983 //Get colection file. It is specified by the environmental
984 //variable XML
d92b41ad 985
d92b41ad 986 //Feed Grid with collection file
d92b41ad 987 TGridCollection * collection = (TGridCollection*) TAlienCollection::Open(kXML);
988 if (! collection) {
989 AliError(Form("%s not found", kXML)) ;
990 return kFALSE ;
991 }
3c867778 992
d92b41ad 993 TGridResult* result = collection->GetGridResult("",0 ,0);
193828fd 994
d92b41ad 995 // Makes the ESD chain
996 printf("*** Getting the Chain ***\n");
997 for (Int_t index = 0; index < result->GetEntries(); index++) {
998 TString alienURL = result->GetKey(index, "turl") ;
999 cout << "================== " << alienURL << endl ;
1000 chain->Add(alienURL) ;
1001 alienURL.ReplaceAll("AliESDs.root",kXSFileName);
1002 chainxs->Add(alienURL) ;
1003 }
1004 }// xml analysis
1005
3c867778 1006 //------------------------------
1007 // PROOF
1008 //------------------------------
1009 else if (mode == mPROOF) {
1010
1011 TFileCollection* ds= gProof->GetDataSet(kDatasetPROOF)->GetStagedSubset();
1012
1013 gROOT->LoadMacro("/afs/in2p3.fr/group/alice/laf/dataset_management/CreateChainFromDataSet.C");
1014 chain = CreateChainFromDataSet(ds, kTreeName , kDatasetNMaxFiles);
1015 printf("chain has %d entries\n",chain->GetEntries());
1016 }
1017
d92b41ad 1018 gSystem->ChangeDirectory(ocwd.Data());
193828fd 1019
d92b41ad 1020}
1021
193828fd 1022//______________________________
1023void CheckEnvironmentVariables()
1024{
1025
cd54ca47 1026 sprintf(kTrigger,"");
193828fd 1027
cd54ca47 1028 Bool_t bRecalibrate = kFALSE;
1029 Bool_t bBadChannel = kFALSE;
193828fd 1030
cd54ca47 1031 for (int i=0; i< gApplication->Argc();i++){
1032
1033#ifdef VERBOSEARGS
1034
1035 printf("Arg %d: %s\n",i,gApplication->Argv(i));
1036
193828fd 1037#endif
cd54ca47 1038
2c18833a 1039 TString sRun = "";
1040
cd54ca47 1041 if (!(strcmp(gApplication->Argv(i),"--trigger")))
1042 sprintf(trigger,gApplication->Argv(i+1));
193828fd 1043
1044 if (!(strcmp(gApplication->Argv(i),"--recalibrate")))
1045 bRecalibrate = atoi(gApplication->Argv(i+1));
cd54ca47 1046
1047 if (!(strcmp(gApplication->Argv(i),"--badchannel")))
1048 bBadChannel = atoi(gApplication->Argv(i+1));
1049
1050 if (!(strcmp(gApplication->Argv(i),"--run"))){
2c18833a 1051 sRun = gApplication->Argv(i+1);
cd54ca47 1052 if(sRun.Contains("LHC10")) {
1053 kYear = 2010;
1054 }
1055 else {
1056 if(kRun <=0){
1057 kRun = atoi(gApplication->Argv(i+1));
1058 }
1059 else printf("** Run number already set to %d, do not set to %d\n",kRun,atoi(gApplication->Argv(i+1)));
1060 }//numeric run
1061 }//--run available
1062
1063 }// args loop
1064
193828fd 1065 if(!sRun.Contains("LHC10")){
531f682d 1066 if ( kRun < 140000)
1067 {
193828fd 1068 kYear = 2010;
1069 if( kRun >= 136851 ) kCollision = "PbPb";
1070 }
531f682d 1071 else if( kRun < 170600)
1072 {
193828fd 1073 kYear = 2011;
531f682d 1074 if( kRun >= 166500 ) kCollision = "PbPb";
1075 }
1076 else
1077 {
1078 kYear = 2012;
1079
193828fd 1080 }
1081 }
1082
cd54ca47 1083 if(kMC) sprintf(kTrigger,"");
1084
1085 printf("*********************************************\n");
1086 //printf("*** Settings trigger %s, recalibrate %d, remove bad channels %d, year %d, collision %s, run %d ***\n",
1087 // kTrigger,bRecalibrate,bBadChannel, kYear,kCollision.Data(), kRun);
1088 printf("*** Settings year %d, collision %s, run %d ***\n",kYear,kCollision.Data(), kRun);
1089 printf("*********************************************\n");
1090
193828fd 1091}
1092
ac0610dc 1093//_________________________________________________________________
1094void GetAverageXsection(TTree * tree, Double_t & xs, Float_t & ntr)
1095{
1096 // Read the PYTHIA statistics from the file pyxsec.root created by
1097 // the function WriteXsection():
1098 // integrated cross section (xsection) and
1099 // the number of Pyevent() calls (ntrials)
1100 // and calculate the weight per one event xsection/ntrials
1101 // The spectrum calculated by a user should be
1102 // multiplied by this weight, something like this:
1103 // TH1F *userSpectrum ... // book and fill the spectrum
1104 // userSpectrum->Scale(weight)
1105 //
1106 // Yuri Kharlov 19 June 2007
1107 // Gustavo Conesa 15 April 2008
1108 Double_t xsection = 0;
1109 UInt_t ntrials = 0;
1110 xs = 0;
1111 ntr = 0;
1112
1113 Int_t nfiles = tree->GetEntries() ;
1114 if (tree && nfiles > 0) {
1115 tree->SetBranchAddress("xsection",&xsection);
1116 tree->SetBranchAddress("ntrials" ,&ntrials );
1117 for(Int_t i = 0; i < nfiles; i++){
1118 tree->GetEntry(i);
1119 xs += xsection ;
1120 ntr += ntrials ;
1121 cout << "xsection " <<xsection<<" ntrials "<<ntrials<<endl;
1122 }
1123
1124 xs = xs / nfiles;
1125 ntr = ntr / nfiles;
1126 cout << "-----------------------------------------------------------------"<<endl;
1127 cout << "Average of "<< nfiles<<" files: xsection " <<xs<<" ntrials "<<ntr<<endl;
1128 cout << "-----------------------------------------------------------------"<<endl;
1129 }
1130 else cout << " >>>> Empty tree !!!! <<<<< "<<endl;
1131
1132}
193828fd 1133