Fix location dependent paths, enable toggle for CF, manual switch for MC

[u/mrichter/AliRoot.git] / PWGDQ / dielectron / macrosLMEE / AddTask_cbaumann_LMEEpp2010.C

AliAnalysisTask *AddTask_cbaumann_LMEEpp2010(Bool_t runTwo=kFALSE, Bool_t hasMC=kFALSE, Bool_t enableCF=kFALSE){
  //get the current analysis manager
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
    Error("AddTask_cbaumann_LMEEpp2010", "No analysis manager found.");
    return 0;
  }

  //set config file name
  //create config File names: TRAIN_ROOT is for running on GSI train, 
  // ALICE_ROOT for CERN Lego trains
  TString configBasePath("$TRAIN_ROOT/cbaumann_dielectron/");
  TString trainRoot=gSystem->Getenv("TRAIN_ROOT");
  if (trainRoot.IsNull()) configBasePath= "$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE/";
  TString configFile("ConfigLMEEpp2010.C");
  TString configLMEECutLib("LMEECutLib.C");

  TString configLMEECutLibPath(configBasePath+configLMEECutLib);
  TString configFilePath(configBasePath+configFile);



  //AOD Usage not yet testes/avialable-------------------------------------

  if (mgr->GetInputEventHandler()->IsA()==AliAODInputHandler::Class()){
    ::Info("AddTaskLMEEpp2010", "no dedicated AOD configuration");
    //	configFile="$TRAIN_ROOT/util/dielectron/dielectron/macros/ConfigLMEEpp2010AOD.C";	

  }

  //Do we have an MC handler? //Child protected, if auto-recognition fails
  if (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0)
    hasMC=kTRUE;


  //  create task and add it to the manager

  if (!gROOT->GetListOfGlobalFunctions()->FindObject(configLMEECutLib.Data()))
	gROOT->LoadMacro(configLMEECutLibPath.Data());
  if (!gROOT->GetListOfGlobalFunctions()->FindObject(configFile.Data()))
	gROOT->LoadMacro(configFilePath.Data());
  
  LMEECutLib* cutlib = new LMEECutLib();
  AliAnalysisTaskMultiDielectron *task=new AliAnalysisTaskMultiDielectron("MultiDiEData");
  if (!hasMC) task->UsePhysicsSelection();
  task->SetEventFilter(cutlib->GetEventCuts(LMEECutLib::kpp2010TPCandTOF)); //


  //load dielectron configuration file

  //add dielectron analysis with different cuts to the task

  AliDielectron *lowmass1=ConfigLMEEpp2010(1,hasMC,enableCF);
  task->AddDielectron(lowmass1);
  printf("add: %s\n",lowmass1->GetName());

  if (runTwo) {
    AliDielectron *lowmass2=ConfigLMEEpp2010(2,hasMC,enableCF);
    task->AddDielectron(lowmass2);
    printf("add: %s\n",lowmass2->GetName());
  }

  mgr->AddTask(task);

  //create output container
  AliAnalysisDataContainer *coutput1 =
    mgr->CreateContainer("cbaumann_LMEEpp2010_tree",
	TTree::Class(),
	AliAnalysisManager::kExchangeContainer,
	"cbaumann_LMEEpp2010_default.root");

  AliAnalysisDataContainer *cOutputHist1 =
    mgr->CreateContainer("cbaumann_LMEEpp2010_out",
	TList::Class(),
	AliAnalysisManager::kOutputContainer,
	"cbaumann_LMEEpp2010_out.root");
    AliAnalysisDataContainer *cOutputHist2 = 0x0;
  if (enableCF) {
    cOutputHist2 =
      mgr->CreateContainer("cbaumann_LMEEpp2010_CF",
	  TList::Class(),
	  AliAnalysisManager::kOutputContainer,
	  "cbaumann_LMEEpp2010_out.root");
  }
  AliAnalysisDataContainer *cOutputHist3 =
    mgr->CreateContainer("cbaumann_EventStatPbPb2011",
	TH1D::Class(),
	AliAnalysisManager::kOutputContainer,
	"cbaumann_LMEEpp2010_out.root");


  mgr->ConnectInput(task,  0, mgr->GetCommonInputContainer());
  mgr->ConnectOutput(task, 0, coutput1 );
  mgr->ConnectOutput(task, 1, cOutputHist1);
  if (enableCF) {
    mgr->ConnectOutput(task, 2, cOutputHist2);
  }  
  mgr->ConnectOutput(task, 3, cOutputHist3);

  return task;
}