1810f2989697ea9430f3f080c9d2473cb53b104f
[u/mrichter/AliRoot.git] / PWG1 / macros / RunAnalysisITS.C
1 class AliAnalysisGrid;
2
3 void RunAnalysisITS() {
4   //
5   // Macro to analyze ESDs from raw data reconstruction
6   // A.Dainese, andrea.dainese@pd.infn.it
7   //
8   //gSystem->Setenv("alien_CLOSE_SE","ALICE::CNAF::SE");
9
10   gSystem->SetIncludePath("-I. -I$ROOTSYS/include -I$ALICE_ROOT -I$ALICE_ROOT/include -I$ALICE_ROOT/ITS -I$ALICE_ROOT/TPC -I$ALICE_ROOT/CONTAINERS -I$ALICE_ROOT/STEER -I$ALICE_ROOT/TRD -I$ALICE_ROOT/macros -I$ALICE_ROOT/ANALYSIS -g"); 
11
12   //
13   TString analysisMode = "grid"; // "local", "grid", or "proof" (not yet)
14
15   Long64_t nentries=1000000000000000,firstentry=0;
16   Bool_t useAlienPlugin=kTRUE;
17   Bool_t uselibPWG1=kTRUE;
18   TString pluginmode="full";
19   TString loadMacroPath="./";
20   Bool_t readHLT=kFALSE;
21   //
22
23   if(analysisMode=="grid") {
24     // Connect to AliEn
25     TGrid::Connect("alien://");
26   } else if(analysisMode=="proof") {
27     // Connect to the PROOF cluster
28     printf("PROOF mode not yet functional..\n");
29     return;
30     TProof::Open("alicecaf");
31     //TProof::Reset("alicecaf");
32   }
33
34   // Load analysis libraries
35   gSystem->Load("libANALYSIS.so");
36   gSystem->Load("libANALYSISalice.so");
37   if(uselibPWG1) gSystem->Load("libPWG1.so");
38
39   // Create Alien plugin, if requested
40   if(useAlienPlugin) {  
41     AliAnalysisGrid *alienHandler = CreateAlienHandler(pluginmode,uselibPWG1);
42     if(!alienHandler) return;
43   }
44
45   TChain *chainESD = 0;
46   if(!useAlienPlugin) {
47     // Prepare input chain
48     //    chainESD = CreateESDChain("/home/dainesea/alignData/RAWdata_CosmicsSum09/RecoSPD/chunk.",13,13);
49     chainESD=new TChain("esdTree");
50     //chainESD->Add("alien:///alice/cern.ch/user/s/sitta/output/000088361/ESDs/pass1/09000088361017.10/AliESDs.root");
51     chainESD->Add("AliESDs.root");
52   }
53
54   // Create the analysis manager
55   AliAnalysisManager *mgr  = new AliAnalysisManager("My Manager","My Manager");
56   // Enable debug printouts
57   mgr->SetDebugLevel(10);
58   // Connect plug-in to the analysis manager
59   if(useAlienPlugin) mgr->SetGridHandler(alienHandler);
60
61   // Add ESD handler
62   AliESDInputHandler *esdH = new AliESDInputHandler();
63   if(readHLT) esdH->SetReadHLT();
64   mgr->SetInputEventHandler(esdH);
65   //-------------------------------------------------------------------
66
67   
68   //-------------------------------------------------------------------
69   // Analysis tasks (wagons of the train)   
70   //
71   TString taskName;
72     
73   if(!uselibPWG1) gROOT->LoadMacro("AliAlignmentDataFilterITS.cxx++g");
74   taskName="AddTaskAlignmentDataFilterITS.C"; 
75   taskName.Prepend(loadMacroPath.Data());
76   gROOT->LoadMacro(taskName.Data());
77   AliAlignmentDataFilterITS *itsTask = AddTaskAlignmentDataFilterITS();
78     
79   if(!uselibPWG1) gROOT->LoadMacro("AliTrackMatchingTPCITSCosmics.cxx++g");
80   taskName="AddTaskTrackMatchingTPCITS.C"; 
81   taskName.Prepend(loadMacroPath.Data());
82   gROOT->LoadMacro(taskName.Data());
83   AliTrackMatchingTPCITSCosmics *tpcitsTask = AddTaskTrackMatchingTPCITS();
84   if(readHLT) tpcitsTask->SetReadHLTESD(kTRUE);  
85   
86   //
87   // Run the analysis
88   //    
89   if(chainESD) printf("CHAIN HAS %d ENTRIES\n",(Int_t)chainESD->GetEntries());
90
91   if(!mgr->InitAnalysis()) return;
92   mgr->PrintStatus();
93   if(analysisMode=="grid" && !useAlienPlugin) analysisMode="local";
94   mgr->StartAnalysis(analysisMode.Data(),chainESD,nentries,firstentry);
95
96   return;
97 }
98 //_____________________________________________________________________________
99 //
100 AliAnalysisGrid* CreateAlienHandler(TString pluginmode="test",
101                                     Bool_t uselibPWG1=kFALSE)
102 {
103   // Check if user has a valid token, otherwise make one. This has limitations.
104   // One can always follow the standard procedure of calling alien-token-init then
105   //   source /tmp/gclient_env_$UID in the current shell.
106    if (!AliAnalysisGrid::CreateToken()) return NULL;
107    AliAnalysisAlien *plugin = new AliAnalysisAlien();
108    // Set the run mode (can be "full", "test", "offline", "submit" or "terminate")
109    plugin->SetRunMode(pluginmode.Data());
110    plugin->SetUser("dainesea");
111    plugin->SetNtestFiles(1);
112    // Set versions of used packages
113    plugin->SetAPIVersion("V2.4");
114    plugin->SetROOTVersion("v5-24-00");
115    plugin->SetAliROOTVersion("v4-18-07-AN");
116    // Declare input data to be processed.
117    // Method 1: Create automatically XML collections using alien 'find' command.
118    // Define production directory LFN
119    plugin->SetGridDataDir("/alice/data/2009/LHC09c");
120    //plugin->SetGridDataDir("/alice/cern.ch/user/s/sitta/output/000088361/");
121    // Set data search pattern
122    //plugin->SetDataPattern("AliESDs.root");
123    plugin->SetDataPattern("ESD.tag.root");
124    Int_t n=0;
125    n++; plugin->AddRunNumber("000080015");
126    n++; plugin->AddRunNumber("000080261");
127    plugin->SetNrunsPerMaster(n);
128    // Method 2: Declare existing data files (raw collections, xml collections, root file)
129    // If no path mentioned data is supposed to be in the work directory (see SetGridWorkingDir())
130    // XML collections added via this method can be combined with the first method if
131    // the content is compatible (using or not tags)
132    // e.g.: find -z -x 80015 /alice/data/2009/LHC09c/000080015/ESDs/ ESD.tag.root > 80015.xml
133    //plugin->AddDataFile("79876.xml");
134    //plugin->AddDataFile("80015.xml");
135    //plugin->AddDataFile("80261.xml");
136    //   plugin->AddDataFile("/alice/data/2008/LHC08c/000057657/raw/Run57657.Merged.RAW.tag.root");
137    // Define alien work directory where all files will be copied. Relative to alien $HOME.
138    plugin->SetGridWorkingDir("analysisITS");
139    // Declare alien output directory. Relative to working directory.
140    plugin->SetGridOutputDir("output151009"); // In this case will be $HOME/work/output
141    // Declare the analysis source files names separated by blancs. To be compiled runtime
142    // using ACLiC on the worker nodes.
143    if(!uselibPWG1) {
144      plugin->SetAnalysisSource("AliAlignmentDataFilterITS.cxx");
145      plugin->SetAnalysisSource("AliTrackMatchingTPCITSCosmics.cxx");
146    }
147    // Declare all libraries (other than the default ones for the framework. These will be
148    // loaded by the generated analysis macro. Add all extra files (task .cxx/.h) here.
149    //plugin->SetAdditionalLibs("AliAlignmentDataFilterITS.h AliAlignmentDataFilterITS.cxx libProof.so libRAWDatabase.so libRAWDatarec.so libCDB.so libSTEER.so libITSbase.so libITSrec.so");
150    plugin->AddIncludePath("-I. -I$ROOTSYS/include -I$ALICE_ROOT -I$ALICE_ROOT/include -I$ALICE_ROOT/ITS -I$ALICE_ROOT/TPC -I$ALICE_ROOT/CONTAINERS -I$ALICE_ROOT/STEER -I$ALICE_ROOT/TRD -I$ALICE_ROOT/macros -I$ALICE_ROOT/ANALYSIS -g");
151    if(!uselibPWG1) {
152      plugin->SetAdditionalLibs("AliAlignmentDataFilterITS.h AliAlignmentDataFilterITS.cxx AliTrackMatchingTPCITSCosmics.h AliTrackMatchingTPCITSCosmics.cxx libProof.so libRAWDatabase.so libRAWDatarec.so libCDB.so libSTEER.so libITSbase.so libITSrec.so");
153    } else {
154      plugin->SetAdditionalLibs("libGui.so libProof.so libMinuit.so libRAWDatabase.so libRAWDatarec.so libCDB.so libSTEER.so libITSbase.so libITSrec.so libTPCbase.so libTPCrec.so libTRDbase.so libTRDrec.so libPWG1.so");
155    }
156    // Declare the output file names separated by blancs.
157    // (can be like: file.root or file.root@ALICE::Niham::File)
158    plugin->SetDefaultOutputs(kTRUE);
159    // Optionally define the files to be archived.
160    //   plugin->SetOutputArchive("log_archive.zip:stdout,stderr@ALICE::NIHAM::File root_archive.zip:*.root@ALICE::NIHAM::File");
161    plugin->SetOutputArchive("log_archive.zip:stdout,stderr");
162    // Optionally set a name for the generated analysis macro (default MyAnalysis.C)
163    plugin->SetAnalysisMacro("AnalysisITS.C");
164    // Optionally set maximum number of input files/subjob (default 100, put 0 to ignore)
165    plugin->SetSplitMaxInputFileNumber(1);
166    // Optionally set number of failed jobs that will trigger killing waiting sub-jobs.
167    //plugin->SetMaxInitFailed(5);
168    // Optionally resubmit threshold.
169    //plugin->SetMasterResubmitThreshold(90);
170    // Optionally set time to live (default 30000 sec)
171    //plugin->SetTTL(20000);
172    // Optionally set input format (default xml-single)
173    plugin->SetInputFormat("xml-single");
174    // Optionally modify the name of the generated JDL (default analysis.jdl)
175    plugin->SetJDLName("TaskAnalysisITS.jdl");
176    // Optionally modify job price (default 1)
177    //plugin->SetPrice(1);      
178    // Optionally modify split mode (default 'se')    
179    plugin->SetSplitMode("se");
180    // Optionally set the preferred SE    
181    plugin->SetPreferedSE("ALICE::CNAF::SE");
182    
183    return plugin;
184 }
185 //-----------------------------------------------------------------------------
186 TChain *CreateESDChain(TString esdpath=".",Int_t ifirst=-1,Int_t ilast=-1) {
187
188
189   TChain *chainESD = new TChain("esdTree");
190
191   if(ifirst<0) {
192     chainESD->Add("AliESDs.root");
193   } else {
194     for(Int_t i=ifirst; i<=ilast; i++) {
195       TString esdfile=esdpath; esdfile+=i; esdfile.Append("/AliESDs.root");
196       chainESD->Add(esdfile.Data());
197     }
198   }
199   
200   return chainESD;
201 }