Merge branch 'feature-movesplit'

[u/mrichter/AliRoot.git] / PWGLF / RESONANCES / macros / lego_train / AddRsnTaskTrain.C

AliAnalysisTask *AddRsnTaskTrain(const char *commonStr,const char *rsnStr,const char *rsnCutStr,TString rsnQualityCutStr="",TString extraMacro="",TString extraMacroArgs="") {
   // rsnStr -> <Name>
   // rsnCutStr -> <CutName>
   // This will use AddRsnPairs<Name>.C
   // and for cuts AddRsnDaughterCuts<CutName>.C
   // and <opt> string is passed to AddRsnDaughterCuts<CutName>.C
   // so you can control different cut settings
   // string "<Name>:mon" means that it will add monitoring histograms from cuts
   // Note : for now you have to set gRsnUseMiniPackage = 0 to have mon histograms
   //    return AddRsnTask("<Name>:mon","<CutName>:<opt>","");
   // or like we are using it now
   //    return AddRsnTask(rsnStr,rsnCutStr,"");

   Printf ("Running AddRsnTaskTrain(\"%s\",\"%s\",\"%s\",\"%s\",\"%s\",\"%s\")",commonStr,rsnStr,rsnCutStr,rsnQualityCutStr.Data(),extraMacro.Data(),extraMacroArgs.Data());
   Bool_t valid;
   AliAnalysisManager::GetGlobalStr("LegoTrainPath",valid);
   if (!valid) {
      TString legoTrainPath = "$ALICE_ROOT/PWGLF/RESONANCES/macros/lego_train";
      AliAnalysisManager::SetGlobalStr("LegoTrainPath",legoTrainPath.Data());
   }

   // Creating Rsn Train Manager
   AliRsnTrainManager *rsnMgr = new AliRsnTrainManager();

   // Rsn Quality Cuts
   if (!RsnLoadMacroTrain("RsnQualityCut.C")) return kFALSE;
   if (!rsnQualityCutStr.IsNull()) AliRsnTrainManager::SetGlobalStr("RsnQualityCut",rsnQualityCutStr.Data());
   //pp_LHC11_p4_120


   if (!RsnLoadMacroTrain("RsnTrainCommonSettings.C")) return kFALSE;
   if (!extraMacro.IsNull()) {
      if (!RsnLoadMacroTrain(extraMacro.Data())) return kFALSE;
   }
   RsnTrainCommonSettings(commonStr,rsnStr,extraMacro,extraMacroArgs);

   rsnMgr->Print();

   if (!RsnLoadMacroTrain("AddRsnTask.C")) return kFALSE;

   // Generate Rsn Analysisa Task
   AliAnalysisTask *task = AddRsnTask(rsnStr,rsnCutStr,"");

   Printf("Saving Rsn Info -> RsnInfo.root");
   TFile *f = TFile::Open("RsnInfo.root","RECREATE");
   AliRsnTrainManager *rsnMgr = AliRsnTrainManager::GetRsnTrainManager();
   TMap *map = rsnMgr->GetGlobals();
   f->WriteObject(map,"RsnInfoMap");
   f->Close();

   // deleting AliRsnTrainManager to reset all previous settings
   delete rsnMgr;

   return task;

}

Bool_t RsnLoadMacroTrain(TString macro,TString path="") {

   Bool_t valid;
   TString lego_path = AliAnalysisManager::GetGlobalStr("RsnLegoTrainPath",valid);
   if (!valid) lego_path = "$ALICE_ROOT/PWGLF/RESONANCES/macros/lego_train";

   if (!gSystem->AccessPathName(macro.Data())) {
      gROOT->LoadMacro(macro.Data());
      Printf("Macro loaded from %s/%s ...",gSystem->pwd(),macro.Data());
      return kTRUE;
   }

   if (!gSystem->AccessPathName(gSystem->ExpandPathName(Form("%s/%s",lego_path.Data(),macro.Data())))) {
      gROOT->LoadMacro(gSystem->ExpandPathName(Form("%s/%s",lego_path.Data(),macro.Data())));
      Printf("Macro loaded from %s ...",gSystem->ExpandPathName(Form("%s/%s",lego_path.Data(),macro.Data())));
      return kTRUE;
   }

   Printf("Error loading %s",macro.Data());

   return kFALSE;
}