PWG0/multPbPb/run.C

   1 // TODO:
   2 // 1. Check cuts for 2010 (Jochen?)
   3 // 2. Run with many centrality bins at once
   4 #include <string.h>
   5
   6 enum { kMyRunModeLocal = 0, kMyRunModeCAF, kMyRunModeGRID};
   7
   8 TList * listToLoad = new TList();
   9
  10 TChain * GetAnalysisChain(const char * incollection);
  11
  12 void run(Char_t* data, Long64_t nev = -1, Long64_t offset = 0, Bool_t debug = kFALSE, Int_t runMode = 0, Bool_t isMC = 0,
  13          Int_t centrBin = 0, const char * centrEstimator = "VOM", Int_t useOtherCentralityCut = 0, Int_t trackMin=0, Int_t trackMax=10000,
  14          const char* option = "",TString customSuffix = "", Int_t workers = -1, Bool_t useSingleBin=kTRUE)
  15 {
  16   // runMode:
  17   //
  18   // 0 local
  19   // 1 proof
  20
  21   if (nev < 0)
  22     nev = 1234567890;
  23
  24   InitAndLoadLibs(runMode,workers,debug);
  25
  26   // Create the analysis manager
  27   mgr = new AliAnalysisManager;
  28
  29   // Add ESD handler
  30   AliESDInputHandler* esdH = new AliESDInputHandler;
  31   // Do I need any of this?
  32   esdH->SetInactiveBranches("AliESDACORDE FMD ALIESDTZERO ALIESDZDC AliRawDataErrorLogs CaloClusters Cascades EMCALCells EMCALTrigger ESDfriend Kinks AliESDTZERO ALIESDACORDE MuonTracks TrdTracks");
  33   mgr->SetInputEventHandler(esdH);
  34
  35   if(isMC) {
  36     AliMCEventHandler* handler = new AliMCEventHandler;
  37     handler->SetPreReadMode(AliMCEventHandler::kLmPreRead);
  38     mgr->SetMCtruthEventHandler(handler);
  39   }
  40
  41
  42   // If we are running on grid, we need the alien handler
  43   if (runMode == kMyRunModeGRID) {
  44     // Create and configure the alien handler plugin
  45     gROOT->LoadMacro("CreateAlienHandler.C");
  46     AliAnalysisGrid *alienHandler = CreateAlienHandler(data, listToLoad, "full", isMC);
  47     if (!alienHandler) {
  48       cout << "Cannot create alien handler" << endl;
  49       exit(1);
  50     }
  51     mgr->SetGridHandler(alienHandler);
  52   }
  53
  54
  55
  56   // physics selection
  57   gROOT->ProcessLine(".L $ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
  58   physicsSelectionTask = AddTaskPhysicsSelection(isMC,0);//FIXME
  59
  60   // Centrality
  61   AliCentralitySelectionTask *taskCentr = new AliCentralitySelectionTask("CentralitySelection");
  62   const char * file1 = "$ALICE_ROOT/ANALYSIS/macros/test_AliCentralityBy1D.root";
  63   const char * file2 = "$ALICE_ROOT/ANALYSIS/macros/test_AliCentralityByFunction.root";
  64   // const char * file1 = "$ALICE_ROOT/ANALYSIS/macros/AliCentralityBy1D_LHC10g2a_100.root";
  65   // const char * file2 = "$ALICE_ROOT/ANALYSIS/macros/AliCentralityByFunction_LHC10g2a_100.root";
  66   // const char * file1 = "$ALICE_ROOT/ANALYSIS/macros/AliCentralityBy1D_137161_GLAU.root";
  67   // const char * file2 = "$ALICE_ROOT/ANALYSIS/macros/test_AliCentralityByFunction.root";
  68
  69   taskCentr->SetPercentileFile (file1);
  70   taskCentr->SetPercentileFile2(file2);
  71   //FIXME: include back centrality estimator
  72   //  mgr->AddTask(taskCentr);
  73   //  mgr->ConnectInput (taskCentr,0, mgr->GetCommonInputContainer());
  74
  75   // Create my own centrality selector
  76   AliAnalysisMultPbCentralitySelector * centrSelector = new AliAnalysisMultPbCentralitySelector();
  77   centrSelector->SetIsMC(isMC);
  78   centrSelector->SetCentrTaskFiles(file1,file2); // for bookkeping only
  79   centrSelector->SetCentralityBin(centrBin);
  80   if (!useSingleBin) centrSelector->SetCentralityBin(0); // FIXME: ok?
  81   centrSelector->SetCentralityEstimator(centrEstimator);
  82
  83
  84   if(useOtherCentralityCut == 1){
  85     cout << "Setting centrality by MULT" << endl;
  86     centrSelector->SetUseMultRange();
  87     centrSelector->SetMultRange(trackMin,trackMax);
  88   }
  89   if(useOtherCentralityCut == 2){
  90     cout << "Setting centrality by V0" << endl;
  91
  92     centrSelector->SetUseV0Range();
  93     centrSelector->SetMultRange(trackMin,trackMax);
  94   }
  95   if(useOtherCentralityCut == 3){
  96     cout << "Setting centrality by SPD outer" << endl;
  97     centrSelector->SetUseSPDOuterRange();
  98     centrSelector->SetMultRange(trackMin,trackMax);
  99   }
 100
 101   // Parse option strings
 102   TString optionStr(option);
 103
 104   // remove SAVE option if set
 105   // This  is copied from a macro by Jan. The reason I kept it is that I may want to pass textual options to the new task at some point
 106   Bool_t doSave = kFALSE;
 107   TString optionStr(option);
 108   if (optionStr.Contains("SAVE"))
 109     {
 110       optionStr = optionStr(0,optionStr.Index("SAVE")) + optionStr(optionStr.Index("SAVE")+4, optionStr.Length());
 111       doSave = kTRUE;
 112     }
 113
 114   AliESDtrackCuts * cuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kFALSE);
 115   TString pathsuffix = "";
 116
 117   if(!useSingleBin) pathsuffix += "_AllCentr";
 118
 119   if (optionStr.Contains("DCA")) {
 120     delete cuts;
 121     cuts = AliESDtrackCuts::GetStandardITSPureSATrackCuts2009();
 122     cout << ">>>> USING DCA cut" << endl;
 123     cuts = AliESDtrackCuts::GetStandardITSTPCTrackCuts2010(kTRUE);
 124     pathsuffix+="_DCAcut";
 125   }
 126
 127   if (optionStr.Contains("ITSsa")) {
 128     delete cuts;
 129     cuts = AliESDtrackCuts::GetStandardITSPureSATrackCuts2009();
 130     cout << ">>>> USING ITS sa tracks" << endl;
 131     pathsuffix+="_ITSsa";
 132   }
 133
 134   if (optionStr.Contains("TPC")) {
 135     delete cuts;
 136     cuts = AliESDtrackCuts::GetStandardTPCOnlyTrackCuts();
 137     cout << ">>>> USING TPC only tracks" << endl;
 138     pathsuffix+="_TPC";
 139   }
 140
 141   Bool_t useMCKinematics = isMC;
 142   if (optionStr.Contains("NOMCKIN")) {
 143     cout << ">>>> Ignoring MC kinematics" << endl;
 144     useMCKinematics=kFALSE;
 145     pathsuffix+="_NOMCKIN";
 146   }
 147
 148
 149   // load my task
 150   if (useSingleBin) {
 151     gROOT->ProcessLine(".L $ALICE_ROOT/PWG0/multPbPb/AddTaskMultPbPbTracks.C");
 152     AliAnalysisTaskMultPbTracks * task = AddTaskMultPbPbTracks("multPbPbtracks.root", cuts, centrSelector);
 153     task->SetIsMC(useMCKinematics);
 154     task->SetOfflineTrigger(AliVEvent::kMB);
 155     if(optionStr.Contains("TPC")) task->SetTPCOnly();
 156     if(useMCKinematics) task->GetHistoManager()->SetSuffix("MC");
 157     if(customSuffix!=""){
 158       cout << "Setting custom suffix: " << customSuffix << endl;
 159       task->GetHistoManager()->SetSuffix(customSuffix);
 160     }
 161   } else {
 162     gROOT->ProcessLine(".L $ALICE_ROOT/PWG0/multPbPb/AddTaskMultPbPbTracksAllCentrality.C");
 163     centrSelector->SetUseV0Range(kTRUE);
 164     Int_t ncentr = 11;
 165     const Float_t minCentr[] = {0, 79, 247,577, 1185,2155,3565,5527,8203, 12167,15073};
 166     const Float_t maxCentr[] = {79,247,577,1185,2155,3565,5527,8203,12167,15073,21000};
 167     AliAnalysisTaskMultPbTracks ** tasks = AddTaskMultPbPbTracksAllCentrality("multPbPbtracks.root", cuts, centrSelector, ncentr,minCentr,maxCentr);
 168     for(Int_t icentr = 0; icentr < ncentr; icentr++){
 169       tasks[icentr]->Print();
 170       tasks[icentr]->SetIsMC(useMCKinematics);
 171       tasks[icentr]->SetOfflineTrigger(AliVEvent::kMB);
 172       if(optionStr.Contains("TPC")) tasks[icentr]->SetTPCOnly();
 173       if(useMCKinematics) tasks[icentr]->GetHistoManager()->SetSuffix("MC");
 174       if(customSuffix!=""){
 175         cout << "Setting custom suffix: " << customSuffix+long(icentr) << tasks[icentr] << endl;
 176         tasks[icentr]->GetHistoManager()->SetSuffix(customSuffix+long(icentr));
 177       }
 178     }
 179   }
 180   // Init and run the analy
 181   if (!mgr->InitAnalysis()) return;
 182
 183   mgr->PrintStatus();
 184
 185   if (runMode == kMyRunModeLocal ) {
 186     // If running in local mode, create chain of ESD files
 187     cout << "RUNNING LOCAL, CHAIN" << endl;
 188     TChain * chain = GetAnalysisChain(data);
 189     //    chain->Print();
 190     mgr->StartAnalysis("local",chain,nev);
 191   } else if (runMode == kMyRunModeCAF) {
 192     mgr->StartAnalysis("proof",TString(data)+"#esdTree",nev);
 193   } else if (runMode == kMyRunModeGRID) {
 194     mgr->StartAnalysis("grid");
 195   } else {
 196     cout << "ERROR: unknown run mode" << endl;
 197   }
 198
 199   if (!useOtherCentralityCut) {
 200     pathsuffix = pathsuffix + "_" + centrEstimator + "_bin_"+long(centrBin);
 201   } else if(useOtherCentralityCut==1){
 202     pathsuffix = pathsuffix + "_TrackRange_" + long(trackMin) + "_" + long(trackMax);
 203   } else if(useOtherCentralityCut==2){
 204     pathsuffix = pathsuffix + "_V0Range_" + long(trackMin) + "_" + long(trackMax);
 205   } else if(useOtherCentralityCut==3){
 206     pathsuffix = pathsuffix + "_SPDOutRange_" + long(trackMin) + "_" + long(trackMax);
 207   }
 208   pathsuffix += customSuffix;
 209
 210   if (doSave) MoveOutput(data, pathsuffix.Data());
 211
 212
 213 }
 214
 215
 216 void MoveOutput(const char * data, const char * suffix = ""){
 217
 218   TString path("output/");
 219   path = path + TString(data).Tokenize("/")->Last()->GetName() + suffix;
 220
 221   TString fileName = "multPbPbtracks.root";
 222   gSystem->mkdir(path, kTRUE);
 223   gSystem->Rename(fileName, path + "/" + fileName);
 224   for(Int_t ibin = 0; ibin < 20; ibin++){
 225     TString fileBin = fileName;
 226     fileBin.ReplaceAll(".root",Form("_%2.2d.root",ibin));
 227     gSystem->Rename(fileBin, path + "/" + fileBin);
 228   }
 229
 230   gSystem->Rename("event_stat.root", path + "/event_stat.root");
 231   Printf(">>>>> Moved files to %s", path.Data());
 232 }
 233
 234
 235
 236 TChain * GetAnalysisChain(const char * incollection){
 237   // Builds a chain of esd files
 238   // incollection can be
 239   // - a single root file
 240   // - an xml collection of files on alien
 241   // - a ASCII containing a list of local root files
 242   TChain* analysisChain = 0;
 243   // chain
 244   analysisChain = new TChain("esdTree");
 245   if (TString(incollection).Contains(".root")){
 246     analysisChain->Add(incollection);
 247   }
 248   else if (TString(incollection).Contains("xml")){
 249     TGrid::Connect("alien://");
 250     TAlienCollection * coll = TAlienCollection::Open (incollection);
 251     while(coll->Next()){
 252       analysisChain->Add(TString("alien://")+coll->GetLFN());
 253     }
 254   } else {
 255     ifstream file_collect(incollection);
 256     TString line;
 257     while (line.ReadLine(file_collect) ) {
 258       analysisChain->Add(line.Data());
 259     }
 260   }
 261   analysisChain->GetListOfFiles()->Print();
 262
 263   return analysisChain;
 264 }
 265
 266
 267 void InitAndLoadLibs(Int_t runMode=kMyRunModeLocal, Int_t workers=0,Bool_t debug=0) {
 268   // Loads libs and par files + custom task and classes
 269
 270   // Custom stuff to be loaded
 271   listToLoad->Add(new TObjString("$ALICE_ROOT/ANALYSIS/AliCentralitySelectionTask.cxx+"));
 272   listToLoad->Add(new TObjString("$ALICE_ROOT/PWG1/background/AliHistoListWrapper.cxx+"));
 273   listToLoad->Add(new TObjString("$ALICE_ROOT/PWG0/multPbPb/AliAnalysisMultPbTrackHistoManager.cxx+"));
 274   listToLoad->Add(new TObjString("$ALICE_ROOT/PWG0/multPbPb/AliAnalysisMultPbCentralitySelector.cxx+"));
 275   listToLoad->Add(new TObjString("$ALICE_ROOT/PWG0/multPbPb/AliAnalysisTaskMultPbTracks.cxx+"));
 276
 277
 278   if (runMode == kMyRunModeCAF)
 279     {
 280       cout << "Init in CAF mode" << endl;
 281
 282       gEnv->SetValue("XSec.GSI.DelegProxy", "2");
 283       TProof * p = TProof::Open("alice-caf.cern.ch", workers>0 ? Form("workers=%d",workers) : "");
 284       //    TProof::Open("skaf.saske.sk", workers>0 ? Form("workers=%d",workers) : "");
 285       p->Exec("TObject *o = gEnv->GetTable()->FindObject(\"Proof.UseMergers\"); gEnv->GetTable()->Remove(o);", kTRUE);
 286
 287       // Enable the needed package
 288       gProof->UploadPackage("$ALICE_ROOT/STEERBase");
 289       gProof->EnablePackage("$ALICE_ROOT/STEERBase");
 290       gProof->UploadPackage("$ALICE_ROOT/ESD");
 291       gProof->EnablePackage("$ALICE_ROOT/ESD");
 292       gProof->UploadPackage("$ALICE_ROOT/AOD");
 293       gProof->EnablePackage("$ALICE_ROOT/AOD");
 294       gProof->UploadPackage("$ALICE_ROOT/ANALYSIS");
 295       gProof->EnablePackage("$ALICE_ROOT/ANALYSIS");
 296       gProof->UploadPackage("$ALICE_ROOT/ANALYSISalice");
 297       gProof->EnablePackage("$ALICE_ROOT/ANALYSISalice");
 298       gProof->UploadPackage("$ALICE_ROOT/PWG0base");
 299       gProof->EnablePackage("$ALICE_ROOT/PWG0base");
 300       gROOT->ProcessLine(gSystem->ExpandPathName(".include $ALICE_ROOT/PWG0/multPb"));
 301       gROOT->ProcessLine(gSystem->ExpandPathName(".include $ALICE_ROOT/PWG1/background"));
 302     }
 303   else
 304     {
 305       cout << "Init in Local or Grid mode" << endl;
 306       gSystem->Load("libCore.so");
 307       gSystem->Load("libTree.so");
 308       gSystem->Load("libGeom.so");
 309       gSystem->Load("libVMC.so");
 310       gSystem->Load("libPhysics.so");
 311       gSystem->Load("libSTEERBase");
 312       gSystem->Load("libESD");
 313       gSystem->Load("libAOD");
 314       gSystem->Load("libANALYSIS");
 315       gSystem->Load("libANALYSISalice");
 316       // Use AliRoot includes to compile our task
 317       gROOT->ProcessLine(".include $ALICE_ROOT/include");
 318
 319       // gSystem->Load("libVMC");
 320       // gSystem->Load("libTree");
 321       // gSystem->Load("libSTEERBase");
 322       // gSystem->Load("libESD");
 323       // gSystem->Load("libAOD");
 324       // gSystem->Load("libANALYSIS");
 325       // gSystem->Load("libANALYSISalice");
 326       // gSystem->Load("libPWG0base");
 327
 328       gROOT->ProcessLine(gSystem->ExpandPathName(".include $ALICE_ROOT/PWG0/multPb"));
 329       gROOT->ProcessLine(gSystem->ExpandPathName(".include $ALICE_ROOT/PWG1/background"));
 330       //    gROOT->ProcessLine(gSystem->ExpandPathName(".include $ALICE_ROOT/PWG1/background/"));
 331     }
 332   // Load helper classes
 333   TIterator * iter = listToLoad->MakeIterator();
 334   TObjString * name = 0;
 335   while (name = (TObjString *)iter->Next()) {
 336     gSystem->ExpandPathName(name->String());
 337     cout << name->String().Data();
 338     if (runMode == kMyRunModeCAF) {
 339       gProof->Load(name->String()+(debug?"+g":""));
 340     } else {
 341       gROOT->LoadMacro(name->String()+(debug?"+g":""));
 342     }
 343   }
 344
 345 }