PWG2/FORWARD/analysis2/MakeELossFits.C

   1 /**
   2  * @file   MakeELossFits.C
   3  * @author Christian Holm Christensen <cholm@dalsgaard.hehi.nbi.dk>
   4  * @date   Wed Mar 23 14:08:14 2011
   5  *
   6  * @brief
   7  *
   8  *
   9  * @ingroup pwg2_forward_eloss
  10  */
  11 /**
  12  * Run a pass on ESD data to produce the energ loss fits
  13  *
  14  *
  15  * @ingroup pwg2_forward_eloss
  16  */
  17 void MakeELossFits(const char* esddir,
  18                    Int_t       nEvents=1000,
  19                    Int_t       proof=0,
  20                    Bool_t      mc=false,
  21                    Bool_t      cent=false)
  22 {
  23   // --- Libraries to load -------------------------------------------
  24   gROOT->Macro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadLibs.C");
  25
  26   // --- Check for proof mode, and possibly upload pars --------------
  27   if (proof > 0) {
  28     gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/LoadPars.C");
  29     LoadPars(proof);
  30   }
  31
  32   // --- Our data chain ----------------------------------------------
  33   gROOT->LoadMacro("$ALICE_ROOT/PWG2/FORWARD/analysis2/scripts/MakeChain.C");
  34   TChain* chain = MakeChain("ESD",esddir, true);
  35   // If 0 or less events is select, choose all
  36   if (nEvents <= 0) nEvents = chain->GetEntries();
  37   Info("MakeELossFits", "Will analyse %d events", nEvents);
  38
  39   // --- Set the macro path ------------------------------------------
  40   gROOT->SetMacroPath(Form("%s:$(ALICE_ROOT)/PWG2/FORWARD/analysis2:"
  41                            "$ALICE_ROOT/ANALYSIS/macros",
  42                            gROOT->GetMacroPath()));
  43
  44   // --- Creating the manager and handlers ---------------------------
  45   AliAnalysisManager *mgr  = new AliAnalysisManager("Forward ELoss Train",
  46                                                     "Forward energy loss");
  47   AliAnalysisManager::SetCommonFileName("forward_eloss.root");
  48
  49   AliESDInputHandler *esdHandler = new AliESDInputHandler();
  50   esdHandler->SetInactiveBranches("AliESDACORDE "
  51                                   "AliRawDataErrorLogs "
  52                                   "CaloClusters "
  53                                   "Cascades "
  54                                   "EMCALCells "
  55                                   "EMCALTrigger "
  56                                   "Kinks "
  57                                   "Cascades "
  58                                   "MuonTracks "
  59                                   "TrdTracks "
  60                                   "CaloClusters "
  61                                   "HLTGlobalTrigger");
  62   mgr->SetInputEventHandler(esdHandler);
  63
  64   // AOD output handler
  65   AliAODHandler* aodHandler   = new AliAODHandler();
  66   mgr->SetOutputEventHandler(aodHandler);
  67   aodHandler->SetOutputFileName("AliAODs.root");
  68
  69   // --- Add tasks ---------------------------------------------------
  70   gROOT->LoadMacro("AddTaskPhysicsSelection.C");
  71   AddTaskPhysicsSelection(mc, kTRUE, kFALSE);
  72
  73   gROOT->LoadMacro("AddTaskCentrality.C");
  74   AddTaskCentrality();
  75
  76   AliFMDEnergyFitterTask* task = new AliFMDEnergyFitterTask("fmdEnergyFitter");
  77   mgr->AddTask(task);
  78
  79   // --- Make the output container and connect it --------------------
  80   TString outputfile = AliAnalysisManager::GetCommonFileName();
  81   AliAnalysisDataContainer* histOut =
  82     mgr->CreateContainer("Forward", TList::Class(),
  83                          AliAnalysisManager::kOutputContainer,outputfile);
  84   mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
  85   mgr->ConnectOutput(task, 1, histOut);
  86
  87   // --- Set parameters on the algorithms ----------------------------
  88   // Set the number of SPD tracklets for which we consider the event a
  89   // low flux event
  90   task->GetEventInspector().SetLowFluxCut(1000);
  91   // Set the maximum error on v_z [cm]
  92   task->GetEventInspector().SetMaxVzErr(0.2);
  93   // Set the eta axis to use - note, this overrides whatever is used
  94   // by the rest of the algorithms - but only for the energy fitter
  95   // algorithm.
  96   task->GetEnergyFitter().SetEtaAxis(200, -4, 6);
  97   // Set maximum energy loss to consider
  98   task->GetEnergyFitter().SetMaxE(15);
  99   // Set number of energy loss bins
 100   task->GetEnergyFitter().SetNEbins(100);
 101   // Set whether to use increasing bin sizes
 102   task->GetEnergyFitter().SetUseIncreasingBins(true);
 103   // Set whether to do fit the energy distributions
 104   task->GetEnergyFitter().SetDoFits(kTRUE);
 105   // Set whether to make the correction object
 106   task->GetEnergyFitter().SetDoMakeObject(kTRUE);
 107   // Set the low cut used for energy
 108   task->GetEnergyFitter().SetLowCut(0.4);
 109   // Set the number of bins to subtract from maximum of distributions
 110   // to get the lower bound of the fit range
 111   task->GetEnergyFitter().SetFitRangeBinWidth(4);
 112   // Set the maximum number of landaus to try to fit (max 5)
 113   task->GetEnergyFitter().SetNParticles(5);
 114   // Set the minimum number of entries in the distribution before
 115   // trying to fit to the data
 116   task->GetEnergyFitter().SetMinEntries(1000);
 117   // --- Set limits on fits the energy -------------------------------
 118   // Maximum relative error on parameters
 119   AliFMDCorrELossFit::ELossFit::fgMaxRelError = .12;
 120   // Least weight to use
 121   AliFMDCorrELossFit::ELossFit::fgLeastWeight = 1e-5;
 122   // Maximum value of reduced chi^2
 123   AliFMDCorrELossFit::ELossFit::fgMaxChi2nu   = 10;
 124
 125   // --- Run the analysis --------------------------------------------
 126   TStopwatch t;
 127   if (!mgr->InitAnalysis()) {
 128     Error("RunManager", "Failed to initialize analysis train!");
 129     return;
 130   }
 131   // Some informative output
 132   mgr->PrintStatus();
 133   // mgr->SetDebugLevel(3);
 134   if (mgr->GetDebugLevel() < 1 && proof <= 0) mgr->SetUseProgressBar(kTRUE);
 135
 136   // Run the train
 137   t.Start();
 138   Printf("=== RUNNING ANALYSIS ==================================");
 139   mgr->StartAnalysis(proof > 0 ? "proof" : "local", chain, nEvents);
 140   t.Stop();
 141   t.Print();
 142 }
 143 //
 144 // EOF
 145 //