[u/mrichter/AliRoot.git] / PWG / muon / AddPWG3MuonTrain.C

Int_t AddPWG3MuonTrain(Int_t iESDAnalysis=1,
                       Int_t iAODAnalysis=0,
                       Int_t addMuonDistributions=0,
		       Int_t addSingleMuonAnalysis=0,
		       Int_t addMuonHFAnalysis=0,
		       Int_t addDimuonCFContainer=0,
		       Bool_t usePhysicsSelection=kFALSE) {

  // Analysis wagons for PWG3Muon (Roberta)
  
  TString taskName="",loadMacroPath="$ALICE_ROOT/PWG3/muon/";
  Int_t ntasks=0;
  
  if(addMuonDistributions) {
    taskName="AddTaskMuonDistributions.C"; taskName.Prepend(loadMacroPath.Data());
    gROOT->LoadMacro(taskName.Data());
    Bool_t doInvMassFit = kTRUE;
    if(iESDAnalysis){
      AliAnalysisTaskMuonDistributions *esdmuondistributionstask = AddTaskMuonDistributions("ESD",doInvMassFit);	
      if(usePhysicsSelection)esdmuondistributionstask->SelectCollisionCandidates();
    } else if(iAODAnalysis){
      AliAnalysisTaskMuonDistributions *aodmuondistributionstask = AddTaskMuonDistributions("AOD",doInvMassFit);	
    }
    ntasks++;
  }
  

  if(addSingleMuonAnalysis) {
    taskName="AddTaskSingleMuonAnalysis.C"; taskName.Prepend(loadMacroPath.Data());
    gROOT->LoadMacro(taskName.Data());
    AliAnalysisTaskSingleMu *singlemutask = AddTaskSingleMuonAnalysis();	
    ntasks++;
  }

  if(addMuonHFAnalysis) {
    taskName="AddTaskMuonsHF.C"; taskName.Prepend(loadMacroPath.Data());
    gROOT->LoadMacro(taskName.Data());
    Int_t runMode = 0;
    Bool_t isAOD;
    if(iAODAnalysis==1) isAOD=kTRUE;
    else isAOD=kFALSE;
    Bool_t isTree = kFALSE;
    Bool_t isMC   = kFALSE;
    AliAnalysisTaskSEMuonsHF *muonhftask = AddTaskMuonsHF(runMode, isMC, isTree);	
    ntasks++;
  }

  if(addDimuonCFContainer) {
    taskName="AddTaskDimuonCFContainerBuilder.C"; taskName.Prepend(loadMacroPath.Data());
    gROOT->LoadMacro(taskName.Data());
    Int_t runMode = 0;
    Bool_t isAOD;
    if(iAODAnalysis==1) isAOD=kTRUE;
    else isAOD=kFALSE;
    Bool_t isAcceptance = kFALSE;
    Bool_t readMC   = kFALSE;
    Double_t ebeam = 3500.;
    AliAnalysisTaskDimuonCFContainerBuilder *dimuonCFtask =
    AddTaskDimuonCFContainerBuilder(isAOD,readMC,isAcceptance,ebeam);
    ntasks++;
  }

  return ntasks;
}
Commit	Line	Data
7dd391ec	1	Int_t AddPWG3MuonTrain(Int_t iESDAnalysis=1,
0c99d785	2	Int_t iAODAnalysis=0,
7dd391ec	3	Int_t addMuonDistributions=0,
	4	Int_t addSingleMuonAnalysis=0,
	5	Int_t addMuonHFAnalysis=0,
daf100e1	6	Int_t addDimuonCFContainer=0,
daf100e1	7	Bool_t usePhysicsSelection=kFALSE) {
4292b3b6	8
524828c1	9	// Analysis wagons for PWG3Muon (Roberta)
daf100e1	10
524828c1	11	TString taskName="",loadMacroPath="$ALICE_ROOT/PWG3/muon/";
	12	Int_t ntasks=0;
	13
	14	if(addMuonDistributions) {
	15	taskName="AddTaskMuonDistributions.C"; taskName.Prepend(loadMacroPath.Data());
	16	gROOT->LoadMacro(taskName.Data());
	17	Bool_t doInvMassFit = kTRUE;
	18	if(iESDAnalysis){
	19	AliAnalysisTaskMuonDistributions *esdmuondistributionstask = AddTaskMuonDistributions("ESD",doInvMassFit);
daf100e1	20	if(usePhysicsSelection)esdmuondistributionstask->SelectCollisionCandidates();
524828c1	21	} else if(iAODAnalysis){
	22	AliAnalysisTaskMuonDistributions *aodmuondistributionstask = AddTaskMuonDistributions("AOD",doInvMassFit);
	23	}
	24	ntasks++;
	25	}
daf100e1	26
524828c1	27
	28	if(addSingleMuonAnalysis) {
	29	taskName="AddTaskSingleMuonAnalysis.C"; taskName.Prepend(loadMacroPath.Data());
	30	gROOT->LoadMacro(taskName.Data());
	31	AliAnalysisTaskSingleMu *singlemutask = AddTaskSingleMuonAnalysis();
	32	ntasks++;
	33	}
	34
4292b3b6	35	if(addMuonHFAnalysis) {
	36	taskName="AddTaskMuonsHF.C"; taskName.Prepend(loadMacroPath.Data());
	37	gROOT->LoadMacro(taskName.Data());
	38	Int_t runMode = 0;
	39	Bool_t isAOD;
	40	if(iAODAnalysis==1) isAOD=kTRUE;
	41	else isAOD=kFALSE;
	42	Bool_t isTree = kFALSE;
	43	Bool_t isMC = kFALSE;
	44	AliAnalysisTaskSEMuonsHF *muonhftask = AddTaskMuonsHF(runMode, isMC, isTree);
	45	ntasks++;
	46	}
	47
7dd391ec	48	if(addDimuonCFContainer) {
	49	taskName="AddTaskDimuonCFContainerBuilder.C"; taskName.Prepend(loadMacroPath.Data());
	50	gROOT->LoadMacro(taskName.Data());
	51	Int_t runMode = 0;
	52	Bool_t isAOD;
	53	if(iAODAnalysis==1) isAOD=kTRUE;
	54	else isAOD=kFALSE;
	55	Bool_t isAcceptance = kFALSE;
	56	Bool_t readMC = kFALSE;
	57	Double_t ebeam = 3500.;
	58	AliAnalysisTaskDimuonCFContainerBuilder *dimuonCFtask =
	59	AddTaskDimuonCFContainerBuilder(isAOD,readMC,isAcceptance,ebeam);
	60	ntasks++;
	61	}
	62
524828c1	63	return ntasks;
524828c1	64	}