PWG3/muon/AddPWG3MuonTrain.C

   1 Int_t AddPWG3MuonTrain(Int_t iESDAnalysis,
   2                        Int_t iAODAnalyis,
   3                        Int_t addMuonDistributions,
   4                        Int_t addSingleMuonAnalysis,
   5                        Int_t addMuonHFAnalysis) {
   6
   7   // Analysis wagons for PWG3Muon (Roberta)
   8
   9   TString taskName="",loadMacroPath="$ALICE_ROOT/PWG3/muon/";
  10   Int_t ntasks=0;
  11
  12   if(addMuonDistributions) {
  13     taskName="AddTaskMuonDistributions.C"; taskName.Prepend(loadMacroPath.Data());
  14     gROOT->LoadMacro(taskName.Data());
  15     Bool_t doInvMassFit = kTRUE;
  16     if(iESDAnalysis){
  17       AliAnalysisTaskMuonDistributions *esdmuondistributionstask = AddTaskMuonDistributions("ESD",doInvMassFit);
  18     } else if(iAODAnalysis){
  19       AliAnalysisTaskMuonDistributions *aodmuondistributionstask = AddTaskMuonDistributions("AOD",doInvMassFit);
  20     }
  21     ntasks++;
  22   }
  23
  24   if(addSingleMuonAnalysis) {
  25     taskName="AddTaskSingleMuonAnalysis.C"; taskName.Prepend(loadMacroPath.Data());
  26     gROOT->LoadMacro(taskName.Data());
  27     AliAnalysisTaskSingleMu *singlemutask = AddTaskSingleMuonAnalysis();
  28     ntasks++;
  29   }
  30
  31   if(addMuonHFAnalysis) {
  32     taskName="AddTaskMuonsHF.C"; taskName.Prepend(loadMacroPath.Data());
  33     gROOT->LoadMacro(taskName.Data());
  34     Int_t runMode = 0;
  35     Bool_t isAOD;
  36     if(iAODAnalysis==1) isAOD=kTRUE;
  37     else isAOD=kFALSE;
  38     Bool_t isTree = kFALSE;
  39     Bool_t isMC   = kFALSE;
  40     AliAnalysisTaskSEMuonsHF *muonhftask = AddTaskMuonsHF(runMode, isMC, isTree);
  41     ntasks++;
  42   }
  43
  44   return ntasks;
  45 }