[u/mrichter/AliRoot.git] / PWGGA / GammaConv / macros / AddTask_GammaConvCalo_pPb.C

void AddTask_GammaConvCalo_pPb(  Int_t trainConfig = 1,  //change different set of cuts
                              Bool_t isMC   = kFALSE, //run MC
                              Int_t enableQAMesonTask = 0, //enable QA in AliAnalysisTaskGammaConvV1
                              Int_t enableQAPhotonTask = 0, // enable additional QA task
                              TString fileNameInputForWeighting = "MCSpectraInput.root", // path to file for weigting input
                              Int_t doWeightingPart = 0,  //enable Weighting
                              TString generatorName = "DPMJET",
                              TString cutnumberAODBranch = "8000000060084000001500000", // cutnumber for AOD branch
                              Bool_t enableExtendedMatching = kFALSE //enable or disable extended matching histograms for conversion electrons <-> cluster
							) {

	// ================= Load Librariers =================================
	gSystem->Load("libCore.so");  
	gSystem->Load("libTree.so");
	gSystem->Load("libGeom.so");
	gSystem->Load("libVMC.so");
	gSystem->Load("libPhysics.so");
	gSystem->Load("libMinuit");
	gSystem->Load("libSTEERBase");
	gSystem->Load("libESD");
	gSystem->Load("libAOD");
	gSystem->Load("libANALYSIS");
	gSystem->Load("libANALYSISalice");  
	gSystem->Load("libPWGGAGammaConv.so");
	gSystem->Load("libCDB.so");
	gSystem->Load("libSTEER.so");
	gSystem->Load("libSTEERBase.so");
	gSystem->Load("libTENDER.so");
	gSystem->Load("libTENDERSupplies.so");
		
	Int_t isHeavyIon = 2;
	
	// ================== GetAnalysisManager ===============================
	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	if (!mgr) {
		Error(Form("AddTask_GammaConvV1_%i",trainConfig), "No analysis manager found.");
		return ;
	}

	// ================== GetInputEventHandler =============================
	AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
	
	//========= Add PID Reponse to ANALYSIS manager ====
	if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
		gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
		AddTaskPIDResponse(isMC);
	}
	
	Printf("here \n");
	
	//=========  Set Cutnumber for V0Reader ================================
	TString cutnumberPhoton = "060084001001500000000";
	TString cutnumberEvent = "8000000";
	Bool_t doEtaShift = kFALSE;
	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();

	//========= Add V0 Reader to  ANALYSIS manager if not yet existent =====
	if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
		AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
		
		fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
		fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
		fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);

		if (!mgr) {
			Error("AddTask_V0ReaderV1", "No analysis manager found.");
			return;
		}

		AliConvEventCuts *fEventCuts=NULL;
		if(cutnumberEvent!=""){
			fEventCuts= new AliConvEventCuts(cutnumberEvent.Data(),cutnumberEvent.Data());
			fEventCuts->SetPreSelectionCutFlag(kTRUE);
			if(fEventCuts->InitializeCutsFromCutString(cutnumberEvent.Data())){
				fEventCuts->DoEtaShift(doEtaShift);
				fV0ReaderV1->SetEventCuts(fEventCuts);
				fEventCuts->SetFillCutHistograms("",kTRUE);
			}
		}

		// Set AnalysisCut Number
		AliConversionPhotonCuts *fCuts=NULL;
		if(cutnumberPhoton!=""){
			fCuts= new AliConversionPhotonCuts(cutnumberPhoton.Data(),cutnumberPhoton.Data());
			fCuts->SetPreSelectionCutFlag(kTRUE);
			fCuts->SetIsHeavyIon(isHeavyIon);
			if(fCuts->InitializeCutsFromCutString(cutnumberPhoton.Data())){
				fV0ReaderV1->SetConversionCuts(fCuts);
				fCuts->SetFillCutHistograms("",kTRUE);
			}
		}
		if(inputHandler->IsA()==AliAODInputHandler::Class()){
		// AOD mode
			fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
		}
		fV0ReaderV1->Init();

		AliLog::SetGlobalLogLevel(AliLog::kFatal);

		//connect input V0Reader
		mgr->AddTask(fV0ReaderV1);
		mgr->ConnectInput(fV0ReaderV1,0,cinput);

	}

	//================================================
	//========= Add task to the ANALYSIS manager =====
	//================================================
	AliAnalysisTaskGammaConvCalo *task=NULL;
	task= new AliAnalysisTaskGammaConvCalo(Form("GammaConvCalo_%i",trainConfig));
	task->SetIsHeavyIon(isHeavyIon);
	task->SetIsMC(isMC);
	// Cut Numbers to use in Analysis
	Int_t numberOfCuts = 2;
	if (trainConfig==3 || trainConfig==5  ){ numberOfCuts =6;}
	if (trainConfig==6 || trainConfig==7  ){ numberOfCuts =5;}
	
	TString *eventCutArray = new TString[numberOfCuts];
	TString *photonCutArray = new TString[numberOfCuts];
	TString *clusterCutArray = new TString[numberOfCuts];
	TString *mesonCutArray = new TString[numberOfCuts];

	// cluster cuts
	// 0 "ClusterType",  1 "EtaMin", 2 "EtaMax", 3 "PhiMin", 4 "PhiMax", 5 "DistanceToBadChannel", 6 "Timing", 7 "TrackMatching", 8 "ExoticCell",
	// 9 "MinEnergy", 10 "MinNCells", 11 "MinM02", 12 "MaxM02", 13 "MinM20", 14 "MaxM20", 15 "MaximumDispersion", 16 "NLM"
	
	if (trainConfig == 1){ 
		eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000040022030000"; mesonCutArray[0] = "01525065000000"; //standart cut, kINT7 // EMCAL clusters
		eventCutArray[ 1] = "8005201"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000040022030000"; mesonCutArray[1] = "01525065000000"; //standard cut, kEMC7 // EMCAL clusters
	} else if (trainConfig == 2) {	
		eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "20000030022000000"; mesonCutArray[0] = "01525065000000"; //standart cut, kINT7 // PHOS clusters
		eventCutArray[ 1] = "8006201"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "20000030022000000"; mesonCutArray[1] = "01525065000000"; //standard cut, kPHI7	// PHOS clusters	
	} else if (trainConfig == 3){ 
		eventCutArray[ 0] = "8008101"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000040022030000"; mesonCutArray[0] = "01525065000000"; //standart cut, kEMCEGA based on INT7 // EMCAL clusters
		eventCutArray[ 1] = "8008301"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000040022030000"; mesonCutArray[1] = "01525065000000"; //standard cut, kEMCEG1 based on INT7 // EMCAL clusters
		eventCutArray[ 2] = "8008501"; photonCutArray[ 2] = "002092970028250400000"; clusterCutArray[2] = "10000040022030000"; mesonCutArray[2] = "01525065000000"; //standard cut, kEMCEG2 based on INT7 // EMCAL clusters
		eventCutArray[ 3] = "8009101"; photonCutArray[ 3] = "002092970028250400000"; clusterCutArray[3] = "10000040022030000"; mesonCutArray[3] = "01525065000000"; //standard cut, kEMCEJE based on INT7 // EMCAL clusters
		eventCutArray[ 4] = "8009301"; photonCutArray[ 4] = "002092970028250400000"; clusterCutArray[4] = "10000040022030000"; mesonCutArray[4] = "01525065000000"; //standard cut, kEMCEJ1 based on INT7 // EMCAL clusters
		eventCutArray[ 5] = "8009501"; photonCutArray[ 5] = "002092970028250400000"; clusterCutArray[5] = "10000040022030000"; mesonCutArray[5] = "01525065000000"; //standard cut, kEMCEG2 based on INT7 // EMCAL clusters
	} else if (trainConfig == 4){ 
		eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000040052030000"; mesonCutArray[0] = "01525065000000"; //standart cut, kINT7 // EMCAL clusters
		eventCutArray[ 1] = "8005201"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000040052030000"; mesonCutArray[1] = "01525065000000"; //standard cut, kEMC7 // EMCAL clusters	
	} else if (trainConfig == 5){ 
		eventCutArray[ 0] = "8008101"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000040052030000"; mesonCutArray[0] = "01525065000000"; //standart cut, kEMCEGA based on INT7 // EMCAL clusters
		eventCutArray[ 1] = "8008301"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000040052030000"; mesonCutArray[1] = "01525065000000"; //standard cut, kEMCEG1 based on INT7 // EMCAL clusters
		eventCutArray[ 2] = "8008501"; photonCutArray[ 2] = "002092970028250400000"; clusterCutArray[2] = "10000040052030000"; mesonCutArray[2] = "01525065000000"; //standard cut, kEMCEG2 based on INT7 // EMCAL clusters
		eventCutArray[ 3] = "8009101"; photonCutArray[ 3] = "002092970028250400000"; clusterCutArray[3] = "10000040052030000"; mesonCutArray[3] = "01525065000000"; //standard cut, kEMCEJE based on INT7 // EMCAL clusters
		eventCutArray[ 4] = "8009301"; photonCutArray[ 4] = "002092970028250400000"; clusterCutArray[4] = "10000040052030000"; mesonCutArray[4] = "01525065000000"; //standard cut, kEMCEJ1 based on INT7 // EMCAL clusters
		eventCutArray[ 5] = "8009501"; photonCutArray[ 5] = "002092970028250400000"; clusterCutArray[5] = "10000040052030000"; mesonCutArray[5] = "01525065000000"; //standard cut, kEMCEG2 based on INT7 // EMCAL clusters		
	} else if (trainConfig == 6){ //EMCAL
		eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000041052030000"; mesonCutArray[0] = "01525065000000"; 
		eventCutArray[ 1] = "8000001"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000042052030000"; mesonCutArray[1] = "01525065000000"; 
		eventCutArray[ 2] = "8000001"; photonCutArray[ 2] = "002092970028250400000"; clusterCutArray[2] = "10000043052030000"; mesonCutArray[2] = "01525065000000"; 
		eventCutArray[ 3] = "8000001"; photonCutArray[ 3] = "002092970028250400000"; clusterCutArray[3] = "10000044052030000"; mesonCutArray[3] = "01525065000000"; 
		eventCutArray[ 4] = "8000001"; photonCutArray[ 4] = "002092970028250400000"; clusterCutArray[4] = "10000045052030000"; mesonCutArray[4] = "01525065000000"; 
	} else if (trainConfig == 7) { //PHOS
		eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "20000041053200000"; mesonCutArray[0] = "01525065000000"; 
		eventCutArray[ 1] = "8000001"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "20000042053200000"; mesonCutArray[1] = "01525065000000"; 
		eventCutArray[ 2] = "8000001"; photonCutArray[ 2] = "002092970028250400000"; clusterCutArray[2] = "20000043053200000"; mesonCutArray[2] = "01525065000000"; 
		eventCutArray[ 3] = "8000001"; photonCutArray[ 3] = "002092970028250400000"; clusterCutArray[3] = "20000044053200000"; mesonCutArray[3] = "01525065000000"; 
		eventCutArray[ 4] = "8000001"; photonCutArray[ 4] = "002092970028250400000"; clusterCutArray[4] = "20000045053200000"; mesonCutArray[4] = "01525065000000"; 
	} else {
		Error(Form("GammaConvCalo_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration");
		return;
	}

	TList *EventCutList = new TList();
	TList *ConvCutList = new TList();
	TList *ClusterCutList = new TList();
	TList *MesonCutList = new TList();

	TList *HeaderList = new TList();
	if (doWeightingPart==1) {
		TObjString *Header1 = new TObjString("pi0_1");
		HeaderList->Add(Header1);
	}
	if (doWeightingPart==2){
		TObjString *Header3 = new TObjString("eta_2");
		HeaderList->Add(Header3);
	}
	if (doWeightingPart==3) {
		TObjString *Header1 = new TObjString("pi0_1");
		HeaderList->Add(Header1);
		TObjString *Header3 = new TObjString("eta_2");
		HeaderList->Add(Header3);
	}

	EventCutList->SetOwner(kTRUE);
	AliConvEventCuts **analysisEventCuts = new AliConvEventCuts*[numberOfCuts];
	ConvCutList->SetOwner(kTRUE);
	AliConversionPhotonCuts **analysisCuts = new AliConversionPhotonCuts*[numberOfCuts];
	ClusterCutList->SetOwner(kTRUE);
	AliCaloPhotonCuts **analysisClusterCuts = new AliCaloPhotonCuts*[numberOfCuts];
	MesonCutList->SetOwner(kTRUE);
	AliConversionMesonCuts **analysisMesonCuts = new AliConversionMesonCuts*[numberOfCuts];

	for(Int_t i = 0; i<numberOfCuts; i++){
		analysisEventCuts[i] = new AliConvEventCuts();   
		analysisEventCuts[i]->InitializeCutsFromCutString(eventCutArray[i].Data());
		EventCutList->Add(analysisEventCuts[i]);
		analysisEventCuts[i]->SetFillCutHistograms("",kFALSE);
		
		analysisCuts[i] = new AliConversionPhotonCuts();
		analysisCuts[i]->InitializeCutsFromCutString(photonCutArray[i].Data());
		analysisCuts[i]->SetIsHeavyIon(isHeavyIon);
		ConvCutList->Add(analysisCuts[i]);
		analysisCuts[i]->SetFillCutHistograms("",kFALSE);
	
		analysisClusterCuts[i] = new AliCaloPhotonCuts();
		analysisClusterCuts[i]->InitializeCutsFromCutString(clusterCutArray[i].Data());
		ClusterCutList->Add(analysisClusterCuts[i]);
        analysisClusterCuts[i]->SetExtendedMatching(enableExtendedMatching);
		analysisClusterCuts[i]->SetFillCutHistograms("");
		
		analysisMesonCuts[i] = new AliConversionMesonCuts();
		analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data());
		MesonCutList->Add(analysisMesonCuts[i]);
		analysisMesonCuts[i]->SetFillCutHistograms("");
		analysisEventCuts[i]->SetAcceptedHeader(HeaderList);
	}

	task->SetEventCutList(numberOfCuts,EventCutList);
	task->SetConversionCutList(numberOfCuts,ConvCutList);
	task->SetCaloCutList(numberOfCuts,ClusterCutList);
	task->SetMesonCutList(numberOfCuts,MesonCutList);
	task->SetMoveParticleAccordingToVertex(kTRUE);
	task->SetDoMesonAnalysis(kTRUE);
	task->SetDoMesonQA(enableQAMesonTask); //Attention new switch for Pi0 QA
	task->SetDoPhotonQA(enableQAPhotonTask);  //Attention new switch small for Photon QA
	task->SetDoClusterQA(1);  //Attention new switch small for Cluster QA

	//connect containers
	AliAnalysisDataContainer *coutput =
		mgr->CreateContainer(Form("GammaConvCalo_%i",trainConfig), TList::Class(),
							AliAnalysisManager::kOutputContainer,Form("GammaConvCalo_%i.root",trainConfig));

	mgr->AddTask(task);
	mgr->ConnectInput(task,0,cinput);
	mgr->ConnectOutput(task,1,coutput);

	return;

}
Commit	Line	Data
8bdca7f1	1	void AddTask_GammaConvCalo_pPb( Int_t trainConfig = 1, //change different set of cuts
	2	Bool_t isMC = kFALSE, //run MC
	3	Int_t enableQAMesonTask = 0, //enable QA in AliAnalysisTaskGammaConvV1
	4	Int_t enableQAPhotonTask = 0, // enable additional QA task
	5	TString fileNameInputForWeighting = "MCSpectraInput.root", // path to file for weigting input
	6	Int_t doWeightingPart = 0, //enable Weighting
	7	TString generatorName = "DPMJET",
4e8dcc76	8	TString cutnumberAODBranch = "8000000060084000001500000", // cutnumber for AOD branch
4e8dcc76	9	Bool_t enableExtendedMatching = kFALSE //enable or disable extended matching histograms for conversion electrons <-> cluster
8bdca7f1	10	) {
	11
	12	// ================= Load Librariers =================================
	13	gSystem->Load("libCore.so");
	14	gSystem->Load("libTree.so");
	15	gSystem->Load("libGeom.so");
	16	gSystem->Load("libVMC.so");
	17	gSystem->Load("libPhysics.so");
	18	gSystem->Load("libMinuit");
	19	gSystem->Load("libSTEERBase");
	20	gSystem->Load("libESD");
	21	gSystem->Load("libAOD");
	22	gSystem->Load("libANALYSIS");
	23	gSystem->Load("libANALYSISalice");
	24	gSystem->Load("libPWGGAGammaConv.so");
	25	gSystem->Load("libCDB.so");
	26	gSystem->Load("libSTEER.so");
	27	gSystem->Load("libSTEERBase.so");
	28	gSystem->Load("libTENDER.so");
	29	gSystem->Load("libTENDERSupplies.so");
	30
344100c4	31	Int_t isHeavyIon = 2;
344100c4	32
8bdca7f1	33	// ================== GetAnalysisManager ===============================
	34	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	35	if (!mgr) {
	36	Error(Form("AddTask_GammaConvV1_%i",trainConfig), "No analysis manager found.");
	37	return ;
	38	}
	39
	40	// ================== GetInputEventHandler =============================
	41	AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
	42
	43	//========= Add PID Reponse to ANALYSIS manager ====
	44	if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
	45	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
344100c4	46	AddTaskPIDResponse(isMC);
8bdca7f1	47	}
	48
	49	Printf("here \n");
	50
	51	//========= Set Cutnumber for V0Reader ================================
344100c4	52	TString cutnumberPhoton = "060084001001500000000";
	53	TString cutnumberEvent = "8000000";
	54	Bool_t doEtaShift = kFALSE;
8bdca7f1	55	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	56
	57	//========= Add V0 Reader to ANALYSIS manager if not yet existent =====
	58	if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
	59	AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
	60
	61	fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
	62	fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
	63	fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);
	64
	65	if (!mgr) {
	66	Error("AddTask_V0ReaderV1", "No analysis manager found.");
	67	return;
	68	}
	69
344100c4	70	AliConvEventCuts *fEventCuts=NULL;
	71	if(cutnumberEvent!=""){
	72	fEventCuts= new AliConvEventCuts(cutnumberEvent.Data(),cutnumberEvent.Data());
	73	fEventCuts->SetPreSelectionCutFlag(kTRUE);
	74	if(fEventCuts->InitializeCutsFromCutString(cutnumberEvent.Data())){
	75	fEventCuts->DoEtaShift(doEtaShift);
	76	fV0ReaderV1->SetEventCuts(fEventCuts);
	77	fEventCuts->SetFillCutHistograms("",kTRUE);
	78	}
	79	}
	80
8bdca7f1	81	// Set AnalysisCut Number
344100c4	82	AliConversionPhotonCuts *fCuts=NULL;
	83	if(cutnumberPhoton!=""){
	84	fCuts= new AliConversionPhotonCuts(cutnumberPhoton.Data(),cutnumberPhoton.Data());
8bdca7f1	85	fCuts->SetPreSelectionCutFlag(kTRUE);
344100c4	86	fCuts->SetIsHeavyIon(isHeavyIon);
344100c4	87	if(fCuts->InitializeCutsFromCutString(cutnumberPhoton.Data())){
8bdca7f1	88	fV0ReaderV1->SetConversionCuts(fCuts);
	89	fCuts->SetFillCutHistograms("",kTRUE);
	90	}
	91	}
	92	if(inputHandler->IsA()==AliAODInputHandler::Class()){
	93	// AOD mode
	94	fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
	95	}
	96	fV0ReaderV1->Init();
	97
	98	AliLog::SetGlobalLogLevel(AliLog::kFatal);
	99
	100	//connect input V0Reader
	101	mgr->AddTask(fV0ReaderV1);
	102	mgr->ConnectInput(fV0ReaderV1,0,cinput);
	103
	104	}
	105
	106	//================================================
	107	//========= Add task to the ANALYSIS manager =====
	108	//================================================
	109	AliAnalysisTaskGammaConvCalo *task=NULL;
	110	task= new AliAnalysisTaskGammaConvCalo(Form("GammaConvCalo_%i",trainConfig));
344100c4	111	task->SetIsHeavyIon(isHeavyIon);
8bdca7f1	112	task->SetIsMC(isMC);
	113	// Cut Numbers to use in Analysis
	114	Int_t numberOfCuts = 2;
eed82f0e	115	if (trainConfig==3 \|\| trainConfig==5 ){ numberOfCuts =6;}
925a074a	116	if (trainConfig==6 \|\| trainConfig==7 ){ numberOfCuts =5;}
d53bf4ac	117
344100c4	118	TString *eventCutArray = new TString[numberOfCuts];
	119	TString *photonCutArray = new TString[numberOfCuts];
	120	TString *clusterCutArray = new TString[numberOfCuts];
8bdca7f1	121	TString *mesonCutArray = new TString[numberOfCuts];
	122
	123	// cluster cuts
	124	// 0 "ClusterType", 1 "EtaMin", 2 "EtaMax", 3 "PhiMin", 4 "PhiMax", 5 "DistanceToBadChannel", 6 "Timing", 7 "TrackMatching", 8 "ExoticCell",
	125	// 9 "MinEnergy", 10 "MinNCells", 11 "MinM02", 12 "MaxM02", 13 "MinM20", 14 "MaxM20", 15 "MaximumDispersion", 16 "NLM"
	126
	127	if (trainConfig == 1){
344100c4	128	eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000040022030000"; mesonCutArray[0] = "01525065000000"; //standart cut, kINT7 // EMCAL clusters
344100c4	129	eventCutArray[ 1] = "8005201"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000040022030000"; mesonCutArray[1] = "01525065000000"; //standard cut, kEMC7 // EMCAL clusters
9df9736d	130	} else if (trainConfig == 2) {
344100c4	131	eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "20000030022000000"; mesonCutArray[0] = "01525065000000"; //standart cut, kINT7 // PHOS clusters
344100c4	132	eventCutArray[ 1] = "8006201"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "20000030022000000"; mesonCutArray[1] = "01525065000000"; //standard cut, kPHI7 // PHOS clusters
ab8c8149	133	} else if (trainConfig == 3){
	134	eventCutArray[ 0] = "8008101"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000040022030000"; mesonCutArray[0] = "01525065000000"; //standart cut, kEMCEGA based on INT7 // EMCAL clusters
	135	eventCutArray[ 1] = "8008301"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000040022030000"; mesonCutArray[1] = "01525065000000"; //standard cut, kEMCEG1 based on INT7 // EMCAL clusters
	136	eventCutArray[ 2] = "8008501"; photonCutArray[ 2] = "002092970028250400000"; clusterCutArray[2] = "10000040022030000"; mesonCutArray[2] = "01525065000000"; //standard cut, kEMCEG2 based on INT7 // EMCAL clusters
	137	eventCutArray[ 3] = "8009101"; photonCutArray[ 3] = "002092970028250400000"; clusterCutArray[3] = "10000040022030000"; mesonCutArray[3] = "01525065000000"; //standard cut, kEMCEJE based on INT7 // EMCAL clusters
	138	eventCutArray[ 4] = "8009301"; photonCutArray[ 4] = "002092970028250400000"; clusterCutArray[4] = "10000040022030000"; mesonCutArray[4] = "01525065000000"; //standard cut, kEMCEJ1 based on INT7 // EMCAL clusters
	139	eventCutArray[ 5] = "8009501"; photonCutArray[ 5] = "002092970028250400000"; clusterCutArray[5] = "10000040022030000"; mesonCutArray[5] = "01525065000000"; //standard cut, kEMCEG2 based on INT7 // EMCAL clusters
eed82f0e	140	} else if (trainConfig == 4){
	141	eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000040052030000"; mesonCutArray[0] = "01525065000000"; //standart cut, kINT7 // EMCAL clusters
	142	eventCutArray[ 1] = "8005201"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000040052030000"; mesonCutArray[1] = "01525065000000"; //standard cut, kEMC7 // EMCAL clusters
	143	} else if (trainConfig == 5){
	144	eventCutArray[ 0] = "8008101"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000040052030000"; mesonCutArray[0] = "01525065000000"; //standart cut, kEMCEGA based on INT7 // EMCAL clusters
	145	eventCutArray[ 1] = "8008301"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000040052030000"; mesonCutArray[1] = "01525065000000"; //standard cut, kEMCEG1 based on INT7 // EMCAL clusters
	146	eventCutArray[ 2] = "8008501"; photonCutArray[ 2] = "002092970028250400000"; clusterCutArray[2] = "10000040052030000"; mesonCutArray[2] = "01525065000000"; //standard cut, kEMCEG2 based on INT7 // EMCAL clusters
	147	eventCutArray[ 3] = "8009101"; photonCutArray[ 3] = "002092970028250400000"; clusterCutArray[3] = "10000040052030000"; mesonCutArray[3] = "01525065000000"; //standard cut, kEMCEJE based on INT7 // EMCAL clusters
	148	eventCutArray[ 4] = "8009301"; photonCutArray[ 4] = "002092970028250400000"; clusterCutArray[4] = "10000040052030000"; mesonCutArray[4] = "01525065000000"; //standard cut, kEMCEJ1 based on INT7 // EMCAL clusters
	149	eventCutArray[ 5] = "8009501"; photonCutArray[ 5] = "002092970028250400000"; clusterCutArray[5] = "10000040052030000"; mesonCutArray[5] = "01525065000000"; //standard cut, kEMCEG2 based on INT7 // EMCAL clusters
925a074a	150	} else if (trainConfig == 6){ //EMCAL
	151	eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "10000041052030000"; mesonCutArray[0] = "01525065000000";
	152	eventCutArray[ 1] = "8000001"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "10000042052030000"; mesonCutArray[1] = "01525065000000";
	153	eventCutArray[ 2] = "8000001"; photonCutArray[ 2] = "002092970028250400000"; clusterCutArray[2] = "10000043052030000"; mesonCutArray[2] = "01525065000000";
	154	eventCutArray[ 3] = "8000001"; photonCutArray[ 3] = "002092970028250400000"; clusterCutArray[3] = "10000044052030000"; mesonCutArray[3] = "01525065000000";
	155	eventCutArray[ 4] = "8000001"; photonCutArray[ 4] = "002092970028250400000"; clusterCutArray[4] = "10000045052030000"; mesonCutArray[4] = "01525065000000";
	156	} else if (trainConfig == 7) { //PHOS
ee4b3d66	157	eventCutArray[ 0] = "8000001"; photonCutArray[ 0] = "002092970028250400000"; clusterCutArray[0] = "20000041053200000"; mesonCutArray[0] = "01525065000000";
	158	eventCutArray[ 1] = "8000001"; photonCutArray[ 1] = "002092970028250400000"; clusterCutArray[1] = "20000042053200000"; mesonCutArray[1] = "01525065000000";
	159	eventCutArray[ 2] = "8000001"; photonCutArray[ 2] = "002092970028250400000"; clusterCutArray[2] = "20000043053200000"; mesonCutArray[2] = "01525065000000";
	160	eventCutArray[ 3] = "8000001"; photonCutArray[ 3] = "002092970028250400000"; clusterCutArray[3] = "20000044053200000"; mesonCutArray[3] = "01525065000000";
	161	eventCutArray[ 4] = "8000001"; photonCutArray[ 4] = "002092970028250400000"; clusterCutArray[4] = "20000045053200000"; mesonCutArray[4] = "01525065000000";
8bdca7f1	162	} else {
	163	Error(Form("GammaConvCalo_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration");
	164	return;
	165	}
	166
344100c4	167	TList *EventCutList = new TList();
8bdca7f1	168	TList *ConvCutList = new TList();
	169	TList *ClusterCutList = new TList();
	170	TList *MesonCutList = new TList();
	171
	172	TList *HeaderList = new TList();
	173	if (doWeightingPart==1) {
	174	TObjString *Header1 = new TObjString("pi0_1");
	175	HeaderList->Add(Header1);
	176	}
	177	if (doWeightingPart==2){
	178	TObjString *Header3 = new TObjString("eta_2");
	179	HeaderList->Add(Header3);
	180	}
	181	if (doWeightingPart==3) {
	182	TObjString *Header1 = new TObjString("pi0_1");
	183	HeaderList->Add(Header1);
	184	TObjString *Header3 = new TObjString("eta_2");
	185	HeaderList->Add(Header3);
	186	}
	187
344100c4	188	EventCutList->SetOwner(kTRUE);
344100c4	189	AliConvEventCuts *analysisEventCuts = new AliConvEventCuts[numberOfCuts];
8bdca7f1	190	ConvCutList->SetOwner(kTRUE);
344100c4	191	AliConversionPhotonCuts *analysisCuts = new AliConversionPhotonCuts[numberOfCuts];
8bdca7f1	192	ClusterCutList->SetOwner(kTRUE);
	193	AliCaloPhotonCuts *analysisClusterCuts = new AliCaloPhotonCuts[numberOfCuts];
	194	MesonCutList->SetOwner(kTRUE);
	195	AliConversionMesonCuts *analysisMesonCuts = new AliConversionMesonCuts[numberOfCuts];
	196
	197	for(Int_t i = 0; i<numberOfCuts; i++){
344100c4	198	analysisEventCuts[i] = new AliConvEventCuts();
	199	analysisEventCuts[i]->InitializeCutsFromCutString(eventCutArray[i].Data());
	200	EventCutList->Add(analysisEventCuts[i]);
	201	analysisEventCuts[i]->SetFillCutHistograms("",kFALSE);
	202
	203	analysisCuts[i] = new AliConversionPhotonCuts();
	204	analysisCuts[i]->InitializeCutsFromCutString(photonCutArray[i].Data());
	205	analysisCuts[i]->SetIsHeavyIon(isHeavyIon);
8bdca7f1	206	ConvCutList->Add(analysisCuts[i]);
8bdca7f1	207	analysisCuts[i]->SetFillCutHistograms("",kFALSE);
344100c4	208
8bdca7f1	209	analysisClusterCuts[i] = new AliCaloPhotonCuts();
344100c4	210	analysisClusterCuts[i]->InitializeCutsFromCutString(clusterCutArray[i].Data());
8bdca7f1	211	ClusterCutList->Add(analysisClusterCuts[i]);
4e8dcc76	212	analysisClusterCuts[i]->SetExtendedMatching(enableExtendedMatching);
8bdca7f1	213	analysisClusterCuts[i]->SetFillCutHistograms("");
	214
	215	analysisMesonCuts[i] = new AliConversionMesonCuts();
	216	analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data());
	217	MesonCutList->Add(analysisMesonCuts[i]);
	218	analysisMesonCuts[i]->SetFillCutHistograms("");
344100c4	219	analysisEventCuts[i]->SetAcceptedHeader(HeaderList);
8bdca7f1	220	}
8bdca7f1	221
344100c4	222	task->SetEventCutList(numberOfCuts,EventCutList);
8bdca7f1	223	task->SetConversionCutList(numberOfCuts,ConvCutList);
	224	task->SetCaloCutList(numberOfCuts,ClusterCutList);
	225	task->SetMesonCutList(numberOfCuts,MesonCutList);
	226	task->SetMoveParticleAccordingToVertex(kTRUE);
	227	task->SetDoMesonAnalysis(kTRUE);
	228	task->SetDoMesonQA(enableQAMesonTask); //Attention new switch for Pi0 QA
	229	task->SetDoPhotonQA(enableQAPhotonTask); //Attention new switch small for Photon QA
fd94f234	230	task->SetDoClusterQA(1); //Attention new switch small for Cluster QA
8bdca7f1	231
	232	//connect containers
	233	AliAnalysisDataContainer *coutput =
	234	mgr->CreateContainer(Form("GammaConvCalo_%i",trainConfig), TList::Class(),
	235	AliAnalysisManager::kOutputContainer,Form("GammaConvCalo_%i.root",trainConfig));
	236
	237	mgr->AddTask(task);
	238	mgr->ConnectInput(task,0,cinput);
	239	mgr->ConnectOutput(task,1,coutput);
	240
	241	return;
	242
	243	}