PWG2/FORWARD/analysis2/MakeAOD.C

   1 /**
   2  * @file
   3  *
   4  * @ingroup pwg2_forward_scripts
   5  */
   6
   7 /**
   8  * Run first pass of the analysis - that is read in ESD and produce AOD
   9  *
  10  * @param esddir    ESD input directory. Any file matching the pattern
  11  *                  *AliESDs*.root are added to the chain
  12  * @param nEvents   Number of events to process.  If 0 or less, then
  13  *                  all events are analysed
  14  * @param proof     Run in proof mode
  15  * @param mc        Run over MC data
  16  *
  17  * If PROOF mode is selected, then Terminate will be run on the master node
  18  * in any case.
  19  *
  20  *
  21  * @ingroup pwg2_forward_scripts
  22  */
  23 void MakeAOD(const char* esddir,
  24              Int_t       nEvents=-1,
  25              Int_t       proof=0,
  26              Bool_t      mc=false)
  27 {
  28   // --- Libraries to load -------------------------------------------
  29   gROOT->Macro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadLibs.C");
  30
  31   // --- Check for proof mode, and possibly upload pars --------------
  32   if (proof> 0) {
  33     gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadPars.C");
  34     LoadPars(proof);
  35   }
  36
  37   // --- Our data chain ----------------------------------------------
  38   gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/MakeESDChain.C");
  39   TChain* chain = MakeESDChain(esddir,true);
  40   // If 0 or less events is select, choose all
  41   if (nEvents <= 0) nEvents = chain->GetEntries();
  42
  43   // --- Creating the manager and handlers ---------------------------
  44   AliAnalysisManager *mgr  = new AliAnalysisManager("Analysis Train",
  45                                                     "FMD analysis train");
  46
  47   // --- ESD input handler -------------------------------------------
  48   AliESDInputHandler *esdHandler = new AliESDInputHandler();
  49   esdHandler->SetInactiveBranches("AliESDACORDE "
  50                                   "AliRawDataErrorLogs "
  51                                   "CaloClusters "
  52                                   "Cascades "
  53                                   "EMCALCells "
  54                                   "EMCALTrigger "
  55                                   "Kinks "
  56                                   "Cascades "
  57                                   "MuonTracks "
  58                                   "TrdTracks "
  59                                   "HLTGlobalTrigger");
  60   mgr->SetInputEventHandler(esdHandler);
  61
  62   // --- Monte Carlo handler -----------------------------------------
  63   if (mc) {
  64     AliMCEventHandler* mcHandler = new AliMCEventHandler();
  65     mgr->SetMCtruthEventHandler(mcHandler);
  66     mcHandler->SetReadTR(true);
  67   }
  68
  69   // --- AOD output handler ------------------------------------------
  70   AliAODHandler* aodHandler   = new AliAODHandler();
  71   mgr->SetOutputEventHandler(aodHandler);
  72   aodHandler->SetOutputFileName("AliAODs.root");
  73
  74   // --- Add tasks ---------------------------------------------------
  75   // Physics selection
  76   gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPhysicsSelection.C");
  77   AddTaskPhysicsSelection(mc, kTRUE, kTRUE);
  78
  79 #if 0
  80   // Centrality
  81   gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/Compile.C");
  82   // gDebug = 10;
  83   Compile("$ALICE_ROOT/PWG2/FORWARD/analysis2/AddTaskCopyHeader.C","+");
  84   // gDebug = 0;
  85   AddTaskCopyHeader();
  86 #endif
  87
  88   // FMD
  89   gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/AddTaskFMD.C");
  90   AddTaskFMD(mc);
  91
  92
  93   // --- Run the analysis --------------------------------------------
  94   TStopwatch t;
  95   if (!mgr->InitAnalysis()) {
  96     Error("MakeAOD", "Failed to initialize analysis train!");
  97     return;
  98   }
  99   // Skip terminate if we're so requested and not in Proof or full mode
 100   mgr->SetSkipTerminate(false);
 101   // Some informative output
 102   mgr->PrintStatus();
 103   // mgr->SetDebugLevel(3);
 104   if (mgr->GetDebugLevel() < 1 && !proof)
 105     mgr->SetUseProgressBar(kTRUE);
 106
 107   // Run the train
 108   t.Start();
 109   Printf("=== RUNNING ANALYSIS ==================================");
 110   mgr->StartAnalysis(proof ? "proof" : "local", chain, nEvents);
 111   t.Stop();
 112   t.Print();
 113 }
 114 //
 115 // EOF
 116 //