MakeQATrain.C

Go to the documentation of this file.

#include "TrainSetup.C"

//====================================================================
class MakeQATrain : public TrainSetup
{
public:
  MakeQATrain(const char* name  = "Forward QA") 
    : TrainSetup(name)
  {
    fOptions.Add("cent", "Use centrality");
    fOptions.Set("type", "ESD");
    fOptions.Add("corr", "DIR", "Corrections dir", "");
  }
protected:
  //__________________________________________________________________
  void CreateTasks(AliAnalysisManager* mgr)
  {
    // --- Output file name ------------------------------------------
    AliAnalysisManager::SetCommonFileName("forward_qa.root");

    // --- Load libraries/pars ---------------------------------------
    fRailway->LoadLibrary("PWGLFforward2");
    
    // --- Set load path ---------------------------------------------
    gROOT->SetMacroPath(Form("%s:$(ALICE_PHYSICS)/PWGLF/FORWARD/analysis2",
           gROOT->GetMacroPath()));

    // --- Check if this is MC ---------------------------------------
    Bool_t mc = mgr->GetMCtruthEventHandler() != 0;

    // --- Add the task ----------------------------------------------
    if (!CoupleSECar("AddTaskForwardQA.C", 
         Form("%d,%d", mc, fOptions.Has("cent")),
         AliVEvent::kAny))
      Fatal("CreateTasks", "Failed to add ForwardQA task");

    TString  cor = "";
    if (fOptions.Has("corr")) cor = fOptions.Get("corr"); 
    if (!cor.IsNull()) {
      fRailway->LoadAux(Form("%s/fmd_corrections.root",cor.Data()), true);
    }
  }
  virtual void CreateCentralitySelection(Bool_t mc)
  {
    if (!fOptions.Has("cent")) return;
    TrainSetup::CreateCentralitySelection(mc);    
  }
  AliVEventHandler* CreateOutputHandler(UShort_t) { return 0; }
  //__________________________________________________________________
  const char* ClassName() const { return "MakeQATrain"; }
};

//
// EOF
//