8 #include <TObjString.h>
10 #include <ANALYSIS/AliAnalysisManager.h>
11 #include <ANALYSIS/AliMultiInputEventHandler.h>
12 #include <STEER/AOD/AliAODHandler.h>
13 #include <TStopwatch.h>
16 Bool_t RunALICE(TString anSrc = "grid",
17 TString anMode = "terminate",
18 TString input="aod" /*or "esd"*/,
19 TString inputMC="" /*or "mc"*/,
20 Long64_t nEvents = 1e10,
23 TString alirsnliteManagers ="",
24 Bool_t useMultiHandler=kTRUE,
25 TString alirsnlitesrc ="$ALICE_ROOT",
26 TString alirsnlitetasks =""
30 anSrc.ToLower(); anMode.ToLower(); input.ToLower(); inputMC.ToLower();
32 // loads libs and setup include paths
33 if (LoadLibsBase(alirsnlitesrc)) return kFALSE;
35 // reset manager if already exists
36 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
38 mgr = new AliAnalysisManager("AliRsnLiteAM","AliRsnLite Analysis Manager");
40 Bool_t useAODOut = kFALSE;
41 CreateInputHandlers(input,inputMC,useAODOut,useMultiHandler);
43 // add default grid handler
44 gROOT->LoadMacro("SetupAnalysisPlugin.C");
45 AliAnalysisGrid *analysisPlugin = SetupAnalysisPlugin(anMode.Data());
46 if (!analysisPlugin) { Printf("Error : analysisPlugin is null !!!"); return kFALSE; }
47 mgr->SetGridHandler(analysisPlugin);
48 if (!dsName.IsNull()) {
49 if (!anSrc.CompareTo("proof") && !anMode.CompareTo("full")) {
50 analysisPlugin->SetProofDataSet(dsName.Data());
51 Printf(Form("Using DataSet %s ...",dsName.Data()));
53 analysisPlugin->SetFileForTestMode(dsName.Data());
54 Printf(Form("Using Test file %s ...",dsName.Data()));
58 TList *listManagers = CreateListOfManagersFromDir(alirsnliteManagers,alirsnlitetasks);
59 if (!listManagers) { Printf("Error : CreateListOfManagersFromDir failed !!!"); return kFALSE;}
62 if (!AddAllManagers(listManagers, anSrc, anMode,input,inputMC)) { Printf("Error : AddAllManagers failed !!!"); return kFALSE;}
67 if (!RunAnalysisManager(anSrc, anMode.Data(), nEvents, nSkip)) { Printf("Error : RunAnalysisManager failed !!!"); return kFALSE;}
71 Printf("Working directory is %s ...", gSystem->WorkingDirectory());
77 Int_t LoadLibsBase(TString alirsnlitesrc) {
79 if (gSystem->Load("libTree.so") < 0) {num++; return num;}
80 if (gSystem->Load("libGeom.so") < 0) {num++; return num;}
81 if (gSystem->Load("libVMC.so") < 0) {num++; return num;}
82 if (gSystem->Load("libMinuit.so") < 0) {num++; return num;}
83 if (gSystem->Load("libPhysics.so") < 0) {num++; return num;}
84 if (gSystem->Load("libSTEERBase.so") < 0) {num++; return num;}
85 if (gSystem->Load("libESD.so") < 0) {num++; return num;}
86 if (gSystem->Load("libAOD.so") < 0) {num++; return num;}
87 if (gSystem->Load("libANALYSIS.so") < 0) {num++; return num;}
88 if (gSystem->Load("libOADB.so") < 0) {num++; return num;}
89 if (gSystem->Load("libANALYSISalice.so") < 0) {num++; return num;}
91 gSystem->AddIncludePath(Form("-I\"%s/include\"", gSystem->ExpandPathName(alirsnlitesrc.Data())));
92 gROOT->ProcessLine(Form(".include %s/include", gSystem->ExpandPathName(alirsnlitesrc.Data())));
95 Bool_t CreateInputHandlers(TString input,TString inputMC,Bool_t useAODOut=kFALSE,Bool_t useMultiHandler=kTRUE) {
97 Bool_t useMC = !inputMC.CompareTo("mc");
99 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
100 if (!mgr) { Printf("Error [CreateInputHandlers] : mgr is null !!!"); return kFALSE; }
102 if (useMultiHandler) {
103 AliMultiInputEventHandler *inputHandler = new AliMultiInputEventHandler();
104 if (!input.CompareTo("esd")) {
105 inputHandler->AddInputEventHandler(new AliESDInputHandler());
106 if (useMC) inputHandler->AddInputEventHandler(new AliMCEventHandler());
107 } else if (!input.CompareTo("aod")) {
108 inputHandler->AddInputEventHandler(new AliAODInputHandler());
111 mgr->SetInputEventHandler(inputHandler);
113 if (!input.CompareTo("esd")) {
114 mgr->SetInputEventHandler(new AliESDInputHandler());
115 if (useMC) mgr->SetMCtruthEventHandler(new AliMCEventHandler());
116 } else if (!input.CompareTo("aod")) {
117 mgr->SetInputEventHandler(new AliAODInputHandler());
119 mgr->SetInputEventHandler(inputHandler);
123 AliAODHandler *aodHandler = new AliAODHandler();
124 aodHandler->SetOutputFileName("AliAOD.root");
125 mgr->SetOutputEventHandler(aodHandler);
130 TList *CreateListOfManagersFromDir(TString listManagersNames="",TString dir="") {
132 TList *listManagers = new TList;
136 if (listManagersNames.IsNull()) {
137 if (dir.IsNull() || gSystem->AccessPathName(gSystem->ExpandPathName(dir.Data()))) {
138 Printf("Error [CreateListOfManagersFromDir] : Dir '%s' doesn't exists !!!",dir.Data());
141 dirsStr = gSystem->GetFromPipe(Form("ls %s",dir.Data()));
142 dirs = dirsStr.Tokenize("\n");
144 dirsStr = listManagersNames;
145 dirs = dirsStr.Tokenize(" ");
150 TObjString *str,*strtmp;
151 TObjArray *mydirstrTok;
152 TString mydirstr,main,prefix;
153 while ((str = (TObjString *)next.Next())) {
154 // TODO add direcotry
155 mydirstr = str->GetString();
156 if (mydirstr.IsNull()) continue;
158 Printf("Adding %s .,,",mydirstr.Data());
159 mydirstrTok = mydirstr.Tokenize("_");
161 main = ((TObjString *)mydirstrTok->At(0))->GetString();
163 strtmp = (TObjString *)mydirstrTok->At(1);
164 if (strtmp) prefix = strtmp->GetString(); else prefix="";
166 listManagers->Add(new TNamed(main,prefix));
172 Bool_t AddAllManagers(TList *listManagers,TString anSrc, TString anMode,TString input,TString inputMC) {
173 TIter next(listManagers);
176 while ((name = (TNamed *)next.Next())) {
177 if (!AddAnalysisManager(name->GetName(), anSrc, anMode,input,inputMC,name->GetTitle(),Form("%d",counter++))) {
178 Printf("Error: Problem adding %s",name->GetName());
186 Bool_t AddAnalysisManager(TString managerMacro, TString anSrc, TString anMode,TString input,TString inputMC, TString postfix,TString idStr) {
187 gROOT->LoadMacro(Form("%s.C", managerMacro.Data()));
188 return gROOT->ProcessLine(Form("%s\(\"%s\",\"%s\",\"%s\"\,\"%s\",\"%s\",\"%s\"\);", managerMacro.Data(), anSrc.Data(), anMode.Data(),input.Data(),inputMC.Data(), postfix.Data(),idStr.Data()));
191 Bool_t RunAnalysisManager(TString anSrc = "proof", TString anMode = "test", Long64_t nEvents = 1e10, Long64_t nSkip = 0) {
193 AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
195 if (!mgr) { Printf("Error [RunAnalysisManager] : mgr is null !!!"); return kFALSE; }
201 if ((!anSrc.CompareTo("proof")) || (!anSrc.CompareTo("local"))) {
202 mgr->StartAnalysis(anSrc.Data(), nEvents, nSkip);
204 mgr->StartAnalysis(anSrc.Data());