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 = "local"; // "local", "grid", or "proof" (not yet)
  14
  15   Long64_t nentries=1000000000000000,firstentry=0;
  16   Bool_t useAlienPlugin=kFALSE;
  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("libTENDER.so");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/alignEvents2/boxTRUNK280909_zero/event.",1,48);
  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   //-------------------------------------------------------------------
  62
  63   // Add ESD handler
  64   Bool_t readRP=kTRUE;
  65   AliESDInputHandler *esdH = 0;
  66   if(readRP) {
  67     esdH = new AliESDInputHandlerRP();
  68   } else {
  69     esdH = new AliESDInputHandler();
  70   }
  71   if(readHLT) esdH->SetReadHLT();
  72   mgr->SetInputEventHandler(esdH);
  73
  74   //-------------------------------------------------------------------
  75   // Analysis tasks (wagons of the train)
  76   //
  77   TString taskName;
  78   /*
  79   if(!uselibPWG1) gROOT->LoadMacro("AliAlignmentDataFilterITS.cxx++g");
  80   taskName="AddTaskAlignmentDataFilterITS.C";
  81   taskName.Prepend(loadMacroPath.Data());
  82   gROOT->LoadMacro(taskName.Data());
  83   AliAlignmentDataFilterITS *itsTask = AddTaskAlignmentDataFilterITS();
  84
  85   if(!uselibPWG1) gROOT->LoadMacro("AliTrackMatchingTPCITSCosmics.cxx++g");
  86   taskName="AddTaskTrackMatchingTPCITS.C";
  87   taskName.Prepend(loadMacroPath.Data());
  88   gROOT->LoadMacro(taskName.Data());
  89   AliTrackMatchingTPCITSCosmics *tpcitsTask = AddTaskTrackMatchingTPCITS();
  90   if(readHLT) tpcitsTask->SetReadHLTESD(kTRUE);
  91   */
  92   Bool_t readMC=kTRUE;
  93   if(!uselibPWG1) gROOT->LoadMacro("AliAnalysisTaskVertexESD.cxx++g");
  94   taskName="AddTaskVertexESD.C";
  95   taskName.Prepend(loadMacroPath.Data());
  96   gROOT->LoadMacro(taskName.Data());
  97   //AliAnalysisTaskVertexESD *vtxTask = AddTaskVertexESD(readMC);
  98
  99   if(!uselibPWG1) gROOT->LoadMacro("AliAnalysisTaskITSTrackingCheck.cxx++g");
 100   taskName="AddTaskPerformanceITS.C";
 101   taskName.Prepend(loadMacroPath.Data());
 102   gROOT->LoadMacro(taskName.Data());
 103   AliAnalysisTaskITSTrackingCheck *itsTask = AddTaskPerformanceITS(readMC,readRP);
 104
 105   if(readMC) {
 106     AliMCEventHandler  *mcH = new AliMCEventHandler();
 107     mgr->SetMCtruthEventHandler(mcH);
 108   }
 109
 110   //
 111   // Run the analysis
 112   //
 113   if(chainESD) printf("CHAIN HAS %d ENTRIES\n",(Int_t)chainESD->GetEntries());
 114
 115   if(!mgr->InitAnalysis()) return;
 116   mgr->PrintStatus();
 117   if(analysisMode=="grid" && !useAlienPlugin) analysisMode="local";
 118   mgr->StartAnalysis(analysisMode.Data(),chainESD,nentries,firstentry);
 119
 120   return;
 121 }
 122 //_____________________________________________________________________________
 123 //
 124 AliAnalysisGrid* CreateAlienHandler(TString pluginmode="test",
 125                                     Bool_t uselibPWG1=kFALSE)
 126 {
 127   // Check if user has a valid token, otherwise make one. This has limitations.
 128   // One can always follow the standard procedure of calling alien-token-init then
 129   //   source /tmp/gclient_env_$UID in the current shell.
 130    if (!AliAnalysisGrid::CreateToken()) return NULL;
 131    AliAnalysisAlien *plugin = new AliAnalysisAlien();
 132    // Set the run mode (can be "full", "test", "offline", "submit" or "terminate")
 133    plugin->SetRunMode(pluginmode.Data());
 134    plugin->SetUser("dainesea");
 135    plugin->SetNtestFiles(1);
 136    // Set versions of used packages
 137    plugin->SetAPIVersion("V2.4");
 138    plugin->SetROOTVersion("v5-24-00");
 139    plugin->SetAliROOTVersion("v4-18-07-AN");
 140    // Declare input data to be processed.
 141    // Method 1: Create automatically XML collections using alien 'find' command.
 142    // Define production directory LFN
 143    plugin->SetGridDataDir("/alice/data/2009/LHC09c");
 144    //plugin->SetGridDataDir("/alice/cern.ch/user/s/sitta/output/000088361/");
 145    // Set data search pattern
 146    //plugin->SetDataPattern("AliESDs.root");
 147    plugin->SetDataPattern("ESD.tag.root");
 148    Int_t n=0;
 149    n++; plugin->AddRunNumber("000080015");
 150    n++; plugin->AddRunNumber("000080261");
 151    plugin->SetNrunsPerMaster(n);
 152    // Method 2: Declare existing data files (raw collections, xml collections, root file)
 153    // If no path mentioned data is supposed to be in the work directory (see SetGridWorkingDir())
 154    // XML collections added via this method can be combined with the first method if
 155    // the content is compatible (using or not tags)
 156    // e.g.: find -z -x 80015 /alice/data/2009/LHC09c/000080015/ESDs/ ESD.tag.root > 80015.xml
 157    //plugin->AddDataFile("79876.xml");
 158    //plugin->AddDataFile("80015.xml");
 159    //plugin->AddDataFile("80261.xml");
 160    //   plugin->AddDataFile("/alice/data/2008/LHC08c/000057657/raw/Run57657.Merged.RAW.tag.root");
 161    // Define alien work directory where all files will be copied. Relative to alien $HOME.
 162    plugin->SetGridWorkingDir("analysisITS");
 163    // Declare alien output directory. Relative to working directory.
 164    plugin->SetGridOutputDir("output151009"); // In this case will be $HOME/work/output
 165    // Declare the analysis source files names separated by blancs. To be compiled runtime
 166    // using ACLiC on the worker nodes.
 167    if(!uselibPWG1) {
 168      plugin->SetAnalysisSource("AliAlignmentDataFilterITS.cxx");
 169      plugin->SetAnalysisSource("AliTrackMatchingTPCITSCosmics.cxx");
 170    }
 171    // Declare all libraries (other than the default ones for the framework. These will be
 172    // loaded by the generated analysis macro. Add all extra files (task .cxx/.h) here.
 173    //plugin->SetAdditionalLibs("AliAlignmentDataFilterITS.h AliAlignmentDataFilterITS.cxx libProof.so libRAWDatabase.so libRAWDatarec.so libCDB.so libSTEER.so libITSbase.so libITSrec.so");
 174    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");
 175    if(!uselibPWG1) {
 176      plugin->SetAdditionalLibs("AliAlignmentDataFilterITS.h AliAlignmentDataFilterITS.cxx AliTrackMatchingTPCITSCosmics.h AliTrackMatchingTPCITSCosmics.cxx libProof.so libRAWDatabase.so libRAWDatarec.so libCDB.so libSTEER.so libITSbase.so libITSrec.so");
 177    } else {
 178      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 libTENDER.so libPWG1.so");
 179    }
 180    // Declare the output file names separated by blancs.
 181    // (can be like: file.root or file.root@ALICE::Niham::File)
 182    plugin->SetDefaultOutputs(kTRUE);
 183    // Optionally define the files to be archived.
 184    //   plugin->SetOutputArchive("log_archive.zip:stdout,stderr@ALICE::NIHAM::File root_archive.zip:*.root@ALICE::NIHAM::File");
 185    plugin->SetOutputArchive("log_archive.zip:stdout,stderr");
 186    // Optionally set a name for the generated analysis macro (default MyAnalysis.C)
 187    plugin->SetAnalysisMacro("AnalysisITS.C");
 188    // Optionally set maximum number of input files/subjob (default 100, put 0 to ignore)
 189    plugin->SetSplitMaxInputFileNumber(1);
 190    // Optionally set number of failed jobs that will trigger killing waiting sub-jobs.
 191    //plugin->SetMaxInitFailed(5);
 192    // Optionally resubmit threshold.
 193    //plugin->SetMasterResubmitThreshold(90);
 194    // Optionally set time to live (default 30000 sec)
 195    //plugin->SetTTL(20000);
 196    // Optionally set input format (default xml-single)
 197    plugin->SetInputFormat("xml-single");
 198    // Optionally modify the name of the generated JDL (default analysis.jdl)
 199    plugin->SetJDLName("TaskAnalysisITS.jdl");
 200    // Optionally modify job price (default 1)
 201    //plugin->SetPrice(1);
 202    // Optionally modify split mode (default 'se')
 203    plugin->SetSplitMode("se");
 204    // Optionally set the preferred SE
 205    plugin->SetPreferedSE("ALICE::CNAF::SE");
 206
 207    return plugin;
 208 }
 209 //-----------------------------------------------------------------------------
 210 TChain *CreateESDChain(TString esdpath=".",Int_t ifirst=-1,Int_t ilast=-1) {
 211
 212
 213   TChain *chainESD = new TChain("esdTree");
 214
 215   if(ifirst<0) {
 216     chainESD->Add("AliESDs.root");
 217   } else {
 218     for(Int_t i=ifirst; i<=ilast; i++) {
 219       TString command=".! ln -s ";
 220       command+=esdpath.Data();
 221       command+=i;
 222       command+= "/AliESDs_def.root ";
 223       command+=esdpath.Data();
 224       command+=i;
 225       command+= "/AliESDs.root ";
 226       gROOT->ProcessLine(command.Data());
 227       command.ReplaceAll("AliESDs","AliESDfriends");
 228       gROOT->ProcessLine(command.Data());
 229       TString esdfile=esdpath; esdfile+=i; esdfile.Append("/AliESDs.root");
 230       chainESD->Add(esdfile.Data());
 231     }
 232   }
 233
 234   return chainESD;
 235 }