PWG4/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="simPbPb/LHC10e18a", // "sim" or "real" etc.
   9                const char *pluginRunMode="full", // "test" or "full" or "terminate"
  10                const char *det = "EMCAL") // "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   if (!submit) {
  33     cout << "local - no submitting" << endl;
  34   }
  35   else {
  36     cout << "submitting to grid" << endl;
  37   }
  38
  39   gROOT->ProcessLine(".L AliAnalysisEtCuts.cxx+g");
  40   gROOT->ProcessLine(".L AliAnalysisEt.cxx+g");
  41   gROOT->ProcessLine(".L AliAnalysisEtMonteCarlo.cxx+g");
  42   gROOT->ProcessLine(".L AliAnalysisEtMonteCarloPhos.cxx+g");
  43   gROOT->ProcessLine(".L AliAnalysisEtMonteCarloEmcal.cxx+g");
  44   gROOT->ProcessLine(".L AliAnalysisEtReconstructed.cxx+g");
  45   gROOT->ProcessLine(".L AliAnalysisEtReconstructedPhos.cxx+g");
  46   gROOT->ProcessLine(".L AliAnalysisEtReconstructedEmcal.cxx+g");
  47   gROOT->ProcessLine(".L AliAnalysisTaskTotEt.cxx+g");
  48
  49   char *kTreeName = "esdTree" ;
  50   TChain * chain   = new TChain(kTreeName,"myESDTree") ;
  51
  52   if(submit){
  53     gSystem->Load("libNetx") ;
  54     gSystem->Load("libgapiUI");
  55     gSystem->Load("libRAliEn");
  56     TGrid::Connect("alien://") ;
  57   }
  58
  59   // Make the analysis manager
  60   AliAnalysisManager *mgr = new AliAnalysisManager("TotEtManager");
  61
  62   TString detStr(det);
  63   TString taskName = "TaskTotEt" + detStr;
  64   TString dataStr(dataType);
  65   TString dataStrName(dataType);
  66   dataStrName.ReplaceAll("/",".");
  67   TString outputName = "Et.ESD." + dataStrName + "." + detStr + ".root";
  68   TString outputDir = "totEt" + dataStr;
  69
  70   cout << " taskName " << taskName
  71        << " outputName " << outputName
  72        << " outputDir (alien) " << outputDir << endl;
  73
  74   if (submit) {
  75     gROOT->LoadMacro("CreateAlienHandlerCaloEtSim.C");
  76     AliAnalysisGrid *alienHandler = CreateAlienHandlerCaloEtSim(outputDir, outputName, pluginRunMode);
  77     if (!alienHandler) return;
  78     mgr->SetGridHandler(alienHandler);
  79   }
  80
  81   AliVEventHandler* esdH = new AliESDInputHandler;
  82   mgr->SetInputEventHandler(esdH);
  83   AliMCEventHandler* handler = new AliMCEventHandler;
  84   if ( dataStr.Contains("sim") ) {
  85     cout << " MC " << endl;
  86     if ( dataStr.Contains("PbPb") ) { // a la: simPbPb/LHC10e18a
  87       cout << " PbPb " << endl;
  88       TString fileLocation = "/home/dsilverm/data/E_T/" + dataStr + "/dir/AliESDs.root";
  89       cout << "fileLocation " << fileLocation.Data() << endl;
  90       chain->Add(fileLocation.Data()); // link to local test file
  91     }
  92     else { // pp
  93       chain->Add("/home/dsilverm/data/E_T/sim/LHC10d1/117222/100/AliESDs.root"); // link to local test file
  94     }
  95     handler->SetReadTR(kFALSE);
  96     mgr->SetMCtruthEventHandler(handler);
  97   }
  98   else { // real data
  99     chain->Add("/home/dsilverm/data/E_T/data/2010/LHC10b/000117222/ESDs/pass2/10000117222021.30/AliESDs.root"); // link to local test file
 100     cout << " not MC " << endl;
 101   }
 102
 103
 104   AliAnalysisTaskTotEt *task1 = new AliAnalysisTaskTotEt(taskName);
 105   mgr->AddTask(task1);
 106
 107   AliAnalysisDataContainer *cinput1 = mgr->GetCommonInputContainer();
 108   AliAnalysisDataContainer *coutput1 = mgr->CreateContainer("out1", TList::Class(), AliAnalysisManager::kOutputContainer, outputName);
 109
 110   //____________________________________________//
 111   mgr->ConnectInput(task1,0,cinput1);
 112   mgr->ConnectOutput(task1,1,coutput1);
 113
 114   mgr->SetDebugLevel(0);
 115
 116   if (!mgr->InitAnalysis()) return;
 117   mgr->PrintStatus();
 118   if(submit){
 119     mgr->StartAnalysis("grid");
 120   }
 121   else{
 122     mgr->StartAnalysis("local",chain);
 123   }
 124
 125   timer.Stop();
 126   timer.Print();
 127 }