e69578774bcbe8c38ca38dcf7197da7ec1a67909
[u/mrichter/AliRoot.git] / PWGLF / totEt / macros / runCaloEt.C
1 //Create by Christine Nattrass, Rebecca Scott, Irakli Martashvili
2 //University of Tennessee at Knoxville
3
4 //by default this runs locally
5 //With the argument true this submits jobs to the grid
6 //As written this requires an xml script tag.xml in the ~/et directory on the grid to submit jobs
7 void runCaloEt(bool submit = false, // true or false 
8                const char *dataType="realPbPb", // "sim" or "real" etc.
9                const char *pluginRunMode="test", // "test" or "full" or "terminate"
10                const char *det = "EMCalDetail") // "PHOS" or "EMCAL"
11 {
12   TStopwatch timer;
13   timer.Start();
14   gSystem->Load("libTree");
15   gSystem->Load("libGeom");
16   gSystem->Load("libVMC");
17   gSystem->Load("libPhysics");
18
19   gSystem->Load("libMinuit");
20
21   gSystem->AddIncludePath("-I$ALICE_ROOT/include");
22   gSystem->AddIncludePath("-I. -I$ALICE_ROOT/EMCAL -I$ALICE_ROOT/ANALYSIS");
23
24   gSystem->Load("libSTEERBase");
25   gSystem->Load("libESD");
26   gSystem->Load("libAOD");
27   
28   gSystem->Load("libANALYSIS");
29   gSystem->Load("libANALYSISalice");
30   gSystem->Load("libCORRFW");
31
32
33
34   if (!submit) { 
35     cout << "local - no submitting" << endl;
36   }
37   else { 
38     cout << "submitting to grid" << endl;
39   }
40    
41   gROOT->ProcessLine(".L AliAnalysisEtCuts.cxx+g");
42   gROOT->ProcessLine(".L AliAnalysisHadEtCorrections.cxx+g");
43   gROOT->ProcessLine(".L AliAnalysisEtCommon.cxx+g");
44   gROOT->ProcessLine(".L AliAnalysisEt.cxx+g");
45   gROOT->ProcessLine(".L AliAnalysisEtMonteCarlo.cxx+g");
46   gROOT->ProcessLine(".L AliAnalysisEtMonteCarloPhos.cxx+g");
47   gROOT->ProcessLine(".L AliAnalysisEtMonteCarloEmcal.cxx+g");
48   gROOT->ProcessLine(".L AliAnalysisEtReconstructed.cxx+g");
49   gROOT->ProcessLine(".L AliAnalysisEtReconstructedPhos.cxx+g");
50   gROOT->ProcessLine(".L AliAnalysisEtReconstructedEmcal.cxx+g");  
51   gROOT->ProcessLine(".L AliAnalysisEtSelectionContainer.cxx+g");
52   gROOT->ProcessLine(".L AliAnalysisEtSelectionHandler.cxx+g");
53   gROOT->ProcessLine(".L AliAnalysisTaskTransverseEnergy.cxx+g");
54   gROOT->ProcessLine(".L AliAnalysisEmEtMonteCarlo.cxx+g");
55   gROOT->ProcessLine(".L AliAnalysisEmEtReconstructed.cxx+g");
56   gROOT->ProcessLine(".L AliAnalysisTaskTotEt.cxx+g");
57
58   gInterpreter->GenerateDictionary("std::map<int, AliPhysicsSelection*>", "AliPhysicsSelection.h;map")  ;
59   gInterpreter->GenerateDictionary("std::pair<int, AliPhysicsSelection*>", "AliPhysicsSelection.h;utility");
60
61   char *kTreeName = "esdTree" ;
62   TChain * chain   = new TChain(kTreeName,"myESDTree") ;
63   
64   if(submit){      
65     gSystem->Load("libNetx") ; 
66     gSystem->Load("libgapiUI");
67     gSystem->Load("libRAliEn"); 
68     TGrid::Connect("alien://") ;
69   }
70   
71   // Make the analysis manager
72   AliAnalysisManager *mgr = new AliAnalysisManager("TotEtManager");
73   
74   TString detStr(det);
75   TString taskName = "TaskTotEt" + detStr;
76   TString dataStr(dataType);
77   TString dataStrName(dataType);
78   dataStrName.ReplaceAll("/",".");
79   TString outputName = "Et.ESD." + dataStrName + "." + detStr + ".root";
80   TString outputDir = "totEt" + dataStr;
81
82   cout << " taskName " << taskName
83        << " outputName " << outputName 
84        << " outputDir (alien) " << outputDir << endl;
85   mgr->SetCommonFileName(outputName.Data());
86   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("out1", TList::Class(), AliAnalysisManager::kOutputContainer, outputName);
87   if (submit) {
88     gROOT->LoadMacro("CreateAlienHandlerCaloEtSim.C");
89     AliAnalysisGrid *alienHandler = CreateAlienHandlerCaloEtSim(outputDir, outputName, pluginRunMode);  
90     if (!alienHandler) return;
91     mgr->SetGridHandler(alienHandler);
92   }
93
94   AliVEventHandler* esdH = new AliESDInputHandler;
95   mgr->SetInputEventHandler(esdH);
96   AliMCEventHandler* handler = new AliMCEventHandler;
97   Bool_t isMc = kTRUE;
98   Bool_t isPb = kFALSE;
99   if ( dataStr.Contains("PbPb") ) { isPb = kTRUE;}
100   if ( dataStr.Contains("sim") ) {
101     cout << " MC " << endl;
102     if ( dataStr.Contains("PbPb") ) { // a la: simPbPb/LHC10e18a
103       cout << " PbPb " << endl;
104       TString fileLocation = "/home/dsilverm/data/E_T/" + dataStr + "/dir/AliESDs.root";
105       cout << "fileLocation " << fileLocation.Data() << endl; 
106       chain->Add(fileLocation.Data()); // link to local test file
107     }
108     else { // pp
109       cout<<"adding pp simulation file"<<endl;
110       chain->Add("/data/LHC10d15/1821/AliESDs.root");
111       //chain->Add("/data/LHC10dpass2/10000126403050.70/AliESDs.root");//data
112       //chain->Add("/home/dsilverm/data/E_T/sim/LHC10d1/117222/100/AliESDs.root"); // link to local test file
113     }
114     handler->SetReadTR(kFALSE);
115     mgr->SetMCtruthEventHandler(handler);
116   }
117   else { // real data
118     isMc = kFALSE;
119     chain->Add("/data/LHC10dpass2/10000126403050.70/AliESDs.root");//data
120     //chain->Add("/home/dsilverm/data/E_T/data/2010/LHC10b/000117222/ESDs/pass2/10000117222021.30/AliESDs.root"); // link to local test file
121     cout << " not MC " << endl;
122   }
123
124
125   if(!isMc && detStr.Contains("EMC")){
126     cout<<"You are running over EMCal data and using the tender supply"<<endl;
127     gSystem->Load("libTENDER.so");
128     gSystem->Load("libTENDERSupplies.so"); 
129     gROOT->ProcessLine(".include $ALICE_ROOT/Tender/"); 
130     gSystem->AddIncludePath("-I$ALICE_ROOT/ANALYSIS "); 
131
132     //this macro is downloaded from the EMCal tender supply twiki 
133     //hopefully it will be replaced by something checked in to aliroot
134     //I have added the function from GetOCDBRecParam.C in Jiri's example to this so that we don't add gobs of macros to the code
135     //I set the defaults to the golden run for PbPb because we are focusing on the golden run, however, this should be thought through!!
136     gROOT->LoadMacro("AddTaskEMCALTenderForEtAnalysis.C");
137     cout<<"WARNING: YOU ARE USING CALIBRATION FACTORS FROM PbPb RUN 137161!!"<<endl;
138 //      // get reco params from grid OCDB
139 //    gROOT->LoadMacro("./GetOCDBRecParam.C");
140 //      // run num, data type pp/PbPb, from grid
141 //Gets calibration factors from grid if jobs are to be submitted to the grid
142         AliEMCALRecParam* pars = GetOCDBRecParam( 137161, "PbPb", submit);
143
144     AliTender *tender = AddTaskEMCALTender( "EMCAL_COMPLETEV1", 0);
145     //this also likely needs modification
146     tender->SelectCollisionCandidates( AliVEvent::kMB | AliVEvent::kEMCEGA | AliVEvent::kEMC1 | AliVEvent::kEMC7 );
147     if(submit){tender->SetDefaultCDBStorage("raw://");} //uncomment if you work on grid
148     else{tender->SetDefaultCDBStorage("local://$ALICE_ROOT/OCDB");} //uncomment if you work local
149
150     if(submit){
151       cout<<"Setting tender to run on grid"<<endl;
152       tender->SetDefaultCDBStorage("raw://"); //uncomment if you work on grid
153     }
154     else{
155       cout<<"Setting tender to run locally"<<endl;
156       tender->SetDefaultCDBStorage("local://$ALICE_ROOT/OCDB"); //uncomment if you work local
157     }
158     // one can sellect what collision candidates to use
159     // triggered sample only: L1 = AliVEvent::kEMCEGA, AliVEvent::kEMCEJE; L0 = AliVEvent::kEMC1, AliVEvent::kEMC7
160     tender->SelectCollisionCandidates( AliVEvent::kAny );
161     tender->SetDebugLevel(2);
162
163     //AliAnalysisDataContainer *coutput3 = mgr->CreateContainer("histosTrgContam", TList::Class(), AliAnalysisManager::kOutputContainer,"AnalysisResults.root");
164     //mgr->ConnectOutput(tender,1,coutput3);
165     cout<<"Output container name "<<AliAnalysisManager::GetCommonFileName()<<endl;
166   }
167
168   if(isMc) cout<<"I am a MC"<<endl;
169   gROOT->ProcessLine(".L $ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
170   
171   AliPhysicsSelectionTask *physicsSelectionTask = AddTaskPhysicsSelection(isMc);//isMC is true when processing monte carlo
172   if(isPb){      
173     cout<<"Adding centrality selection task"<<endl;
174     gROOT->ProcessLine(".L $ALICE_ROOT/ANALYSIS/macros/AddTaskCentrality.C");
175     gROOT->ProcessLine(".L AliCentralitySelectionTask.cxx++g");
176     AliCentralitySelectionTask *centTask = AddTaskCentrality();
177   }
178
179
180
181   AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
182
183
184
185
186
187   AliAnalysisTaskTotEt *task1 = new AliAnalysisTaskTotEt(taskName);
188   task1->SetMcData(isMc);//necessary to tell the task to basically accept all MC events.
189   mgr->AddTask(task1);
190
191   
192   //____________________________________________//
193   mgr->ConnectInput(task1,0,cinput1);
194   mgr->ConnectOutput(task1,1,coutput1);
195
196
197   
198   mgr->SetDebugLevel(0);
199   
200   if (!mgr->InitAnalysis()) return;
201   mgr->PrintStatus();
202   if(submit){
203     mgr->StartAnalysis("grid");
204   }
205   else{
206     mgr->StartAnalysis("local",chain);
207   }
208   
209   timer.Stop();
210   timer.Print();
211 }