MakeMultClassTrain.C

Go to the documentation of this file.

#include "TrainSetup.C"
#include <sstream>

//====================================================================
class MakeMultClassTrain : public TrainSetup
{
public:
  MakeMultClassTrain(const  TString& name) 
    : TrainSetup(name)
  {
    fOptions.Set("type", "ESD");
  }
protected:
  void CreateTasks(AliAnalysisManager* mgr)
  {
    // --- Output file name ------------------------------------------
    AliAnalysisManager::SetCommonFileName("multClass.root");

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

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

    gROOT->ProcessLine("AliMultEventClassifierTask* task = "
           "new AliMultEventClassifierTask(\"MultEvent\"); "
           "task->Connect(0)");
  }
  //__________________________________________________________________
  const char* ClassName() const { return "MakeMultClassTrain"; }
};
//
// EOF
//