PWG2/FORWARD/analysis2/MakeFlow.C

   1 /**
   2  * @file   MakeFlow.C
   3  * @author Alexander Hansen
   4  * @date   Wed Mar 23 12:11:33 2011
   5  *
   6  * @brief
   7  *
   8  * @ingroup pwg2_forward_scripts_makers
   9  *
  10  */
  11 /**
  12  * Script to analyse AOD input for flow
  13  *
  14  * Takes either a single (AOD) .root file as input or a .txt
  15  * The .txt file is expected to contain the path to the files
  16  * from the current directory or the absolute path.
  17  *
  18  * @par Inputs:
  19  *
  20  *
  21  * @par Outputs:
  22  * -
  23  *
  24  * @ingroup pwg2_forward_flow
  25  */
  26 void MakeFlow(TString data      = "",
  27               Int_t   nevents   = 0,
  28               TString type      = "",
  29               Int_t   etabins   = 40,
  30               Int_t   zVertex   = 2,
  31               TString addFlow   = "",
  32               Int_t   addFType  = 0,
  33               Int_t   addFOrder = 0,
  34               Bool_t  proof     = false)
  35 {
  36   Bool_t proof = kFALSE;
  37
  38   // --- Load libs ---------------------------------------------------
  39   gROOT->Macro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadLibs.C");
  40
  41   // --- Check for proof mode, and possibly upload pars --------------
  42   if (proof> 0) {
  43     gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadPars.C");
  44     if (!LoadPars(proof)) {
  45       AliError("MakeFlow", "Failed to load PARs");
  46       return;
  47     }
  48   }
  49
  50   // --- Set the macro path ------------------------------------------
  51   gROOT->SetMacroPath(Form("%s:$(ALICE_ROOT)/PWG2/FORWARD/analysis2:"
  52                            "$ALICE_ROOT/ANALYSIS/macros",
  53                            gROOT->GetMacroPath()));
  54
  55   // --- Add to chain either AOD ------------------------------------
  56   if (data.IsNull()) {
  57     AliError("You didn't add a data file");
  58     return;
  59   }
  60   TChain* chain = new TChain("aodTree");
  61
  62   if (data.Contains(".txt")) MakeChain(data, chain);
  63
  64   if (data.Contains(".root")) {
  65     TFile* test = TFile::Open(data.Data())
  66     if (!test) {
  67       AliError(Form("AOD file %s not found", data.Data()));
  68       return;
  69     }
  70     test->Close(); // Remember to close!
  71     chain->Add(data.Data());
  72   }
  73
  74   // --- Initiate the event handlers --------------------------------
  75   AliAnalysisManager *mgr  = new AliAnalysisManager("Forward Flow",
  76                                                     "Flow in the forward region");
  77
  78   // --- AOD input handler -------------------------------------------
  79   AliAODInputHandler *aodInputHandler = new AliAODInputHandler();
  80   mgr->SetInputEventHandler(aodInputHandler);
  81
  82   // --- Create output handler ---------------------------------------
  83   AliAODHandler* aodOut = new AliAODHandler();
  84   mgr->SetOutputEventHandler(aodOut);
  85   aodOut->SetOutputFileName("AliAODs.Flow.root");
  86
  87   // --- Add the tasks ---------------------------------------------
  88   gROOT->LoadMacro("AddTaskForwardFlow.C");
  89   AddTaskForwardFlow(type, etabins, zVertex, addFlow, addFType, addFOrder);
  90
  91   // --- Run the analysis --------------------------------------------
  92   TStopwatch t;
  93   if (!mgr->InitAnalysis()) {
  94     Error("MakeFlow", "Failed to initialize analysis train!");
  95     return;
  96   }
  97   mgr->PrintStatus();
  98   //
  99   if (proof) mgr->SetDebugLevel(3);
 100   if (mgr->GetDebugLevel() < 1 && !proof)
 101     mgr->SetUseProgressBar(kTRUE,nEvents < 10000 ? 10 : 100);
 102
 103   t.Start();
 104   if (nevents == 0) mgr->StartAnalysis("local", chain);
 105   if (nevents != 0) mgr->StartAnalysis("local", chain, nevents);
 106   t.Stop();
 107   t.Print();
 108 }
 109 //----------------------------------------------------------------
 110 void MakeChain(TString data = "", TChain* chain = 0)
 111 {
 112   // creates chain of files in a given directory or file containing a list.
 113
 114   // Open the input stream
 115   ifstream in;
 116   in.open(data.Data());
 117
 118   // Read the input list of files and add them to the chain
 119   TString line;
 120   while(in.good())
 121   {
 122     in >> line;
 123
 124     if (line.Length() == 0)
 125       continue;
 126
 127     if (TFile::Open(line))
 128       chain->Add(line);
 129   }
 130
 131   in.close();
 132
 133   return;
 134 }
 135 //
 136 // EOF
 137 //