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

void AddTask_GammaConvV1_pp2(  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
                              TString cutnumberAODBranch = "0000000060084001001500000" 
                           ) {

	// ================= 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 = 0;	
	   
	// ================== 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);
	}
	
	//=========  Set Cutnumber for V0Reader ================================
		TString cutnumberPhoton = "002084000002200000000";
		TString cutnumberEvent = "0000000"; 
		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())){
				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::kInfo);

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

	}

	//================================================
	//========= Add task to the ANALYSIS manager =====
	//================================================
	AliAnalysisTaskGammaConvV1 *task=NULL;
	task= new AliAnalysisTaskGammaConvV1(Form("GammaConvV1_%i",trainConfig));
	task->SetIsHeavyIon(isHeavyIon);
	task->SetIsMC(isMC);
	// Cut Numbers to use in Analysis
	Int_t numberOfCuts = 1;

	TString *eventCutArray = new TString[numberOfCuts];
	TString *photonCutArray = new TString[numberOfCuts];
	TString *mesonCutArray = new TString[numberOfCuts];

	if (trainConfig == 1){ 
		eventCutArray[ 0] = "0000012"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , only boxes
	} else if (trainConfig == 2) { 
		eventCutArray[ 0] = "0000012"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD, V0AND , only boxes
	} else if (trainConfig == 3) { 
		eventCutArray[ 0] = "0000012"; photonCutArray[ 0] = "002093260003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Gamma pp 2-76TeV , only boxes
	} else if (trainConfig == 4) { 
		eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , only Minbias MC
	} else if (trainConfig == 5) { 
		eventCutArray[ 0] = "0001011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD, V0AND
	} else if (trainConfig == 6) { 
		eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "002093260003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Gamma pp 2-76TeV
	} else if (trainConfig == 7) {    
		eventCutArray[ 0] = "0002011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , only Minbias MC
	} else if (trainConfig == 8) {    
		eventCutArray[ 0] = "0003011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD, V0AND , only Minbias MC
	} else if (trainConfig == 9) {    
		eventCutArray[ 0] = "0002012"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , only Boxes MC
	} else if (trainConfig == 10) { 
		eventCutArray[ 0] = "0003012"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD, V0AND, only Boxes MC
	} else if (trainConfig == 11) { 
		eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , all photon qualities
	} else if (trainConfig == 12) { 
		eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "007093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , all photon qualities, min R = 35 cm
	} else if (trainConfig == 13) { 
		eventCutArray[ 0] = "0002011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , all photon qualities
	} else if (trainConfig == 14) { 
		eventCutArray[ 0] = "0002011"; photonCutArray[ 0] = "007093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , all photon qualities, min R = 35 cm
	} else if (trainConfig == 15) { 
		eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "002092970028250400000"; mesonCutArray[0] = "01525065000000"; //standard cut LHC11h pp 2.76TeV 
	} else {
		Error(Form("GammaConvV1_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration");
		return;
	}

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

	TList *HeaderList = new TList();
	TObjString *Header2 = new TObjString("BOX");
	HeaderList->Add(Header2);

	EventCutList->SetOwner(kTRUE);
	AliConvEventCuts **analysisEventCuts = new AliConvEventCuts*[numberOfCuts];
	ConvCutList->SetOwner(kTRUE);
	AliConversionPhotonCuts **analysisCuts = new AliConversionPhotonCuts*[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());
		ConvCutList->Add(analysisCuts[i]);
		analysisCuts[i]->SetFillCutHistograms("",kFALSE);

		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->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

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

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

	return;

}
Commit	Line	Data
c323a28f	1	void AddTask_GammaConvV1_pp2( Int_t trainConfig = 1, //change different set of cuts
88202ecb	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	TString cutnumberAODBranch = "0000000060084001001500000"
	7	) {
	8
344100c4	9	// ================= Load Librariers =================================
	10	gSystem->Load("libCore.so");
	11	gSystem->Load("libTree.so");
	12	gSystem->Load("libGeom.so");
	13	gSystem->Load("libVMC.so");
	14	gSystem->Load("libPhysics.so");
	15	gSystem->Load("libMinuit");
	16	gSystem->Load("libSTEERBase");
	17	gSystem->Load("libESD");
	18	gSystem->Load("libAOD");
	19	gSystem->Load("libANALYSIS");
	20	gSystem->Load("libANALYSISalice");
	21	gSystem->Load("libPWGGAGammaConv.so");
	22	gSystem->Load("libCDB.so");
	23	gSystem->Load("libSTEER.so");
	24	gSystem->Load("libSTEERBase.so");
	25	gSystem->Load("libTENDER.so");
	26	gSystem->Load("libTENDERSupplies.so");
88202ecb	27
344100c4	28	Int_t isHeavyIon = 0;
	29
	30	// ================== GetAnalysisManager ===============================
	31	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	32	if (!mgr) {
	33	Error(Form("AddTask_GammaConvV1_%i",trainConfig), "No analysis manager found.");
	34	return ;
	35	}
	36
	37	// ================== GetInputEventHandler =============================
	38	AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
	39
	40	//========= Add PID Reponse to ANALYSIS manager ====
	41	if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
	42	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
	43	AddTaskPIDResponse(isMC);
	44	}
	45
	46	//========= Set Cutnumber for V0Reader ================================
	47	TString cutnumberPhoton = "002084000002200000000";
	48	TString cutnumberEvent = "0000000";
	49	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	50
	51	//========= Add V0 Reader to ANALYSIS manager if not yet existent =====
	52	if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
	53	AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
	54
	55	fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
	56	fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
	57	fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);
	58
	59	if (!mgr) {
	60	Error("AddTask_V0ReaderV1", "No analysis manager found.");
	61	return;
	62	}
	63
	64	AliConvEventCuts *fEventCuts=NULL;
	65	if(cutnumberEvent!=""){
	66	fEventCuts= new AliConvEventCuts(cutnumberEvent.Data(),cutnumberEvent.Data());
	67	fEventCuts->SetPreSelectionCutFlag(kTRUE);
	68	if(fEventCuts->InitializeCutsFromCutString(cutnumberEvent.Data())){
	69	fV0ReaderV1->SetEventCuts(fEventCuts);
	70	fEventCuts->SetFillCutHistograms("",kTRUE);
	71	}
	72	}
	73
	74	// Set AnalysisCut Number
	75	AliConversionPhotonCuts *fCuts=NULL;
	76	if(cutnumberPhoton!=""){
	77	fCuts= new AliConversionPhotonCuts(cutnumberPhoton.Data(),cutnumberPhoton.Data());
	78	fCuts->SetPreSelectionCutFlag(kTRUE);
	79	fCuts->SetIsHeavyIon(isHeavyIon);
	80	if(fCuts->InitializeCutsFromCutString(cutnumberPhoton.Data())){
	81	fV0ReaderV1->SetConversionCuts(fCuts);
	82	fCuts->SetFillCutHistograms("",kTRUE);
	83	}
	84	}
	85	if(inputHandler->IsA()==AliAODInputHandler::Class()){
	86	// AOD mode
	87	fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
	88	}
	89	fV0ReaderV1->Init();
	90
	91	AliLog::SetGlobalLogLevel(AliLog::kInfo);
92
93	//connect input V0Reader
94	mgr->AddTask(fV0ReaderV1);
95	mgr->ConnectInput(fV0ReaderV1,0,cinput);
96
97	}
98
99	//================================================
100	//========= Add task to the ANALYSIS manager =====
101	//================================================
102	AliAnalysisTaskGammaConvV1 *task=NULL;
103	task= new AliAnalysisTaskGammaConvV1(Form("GammaConvV1_%i",trainConfig));
104	task->SetIsHeavyIon(isHeavyIon);
105	task->SetIsMC(isMC);
106	// Cut Numbers to use in Analysis
107	Int_t numberOfCuts = 1;
108
109	TString *eventCutArray = new TString[numberOfCuts];
110	TString *photonCutArray = new TString[numberOfCuts];
111	TString *mesonCutArray = new TString[numberOfCuts];
112
113	if (trainConfig == 1){
114	eventCutArray[ 0] = "0000012"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , only boxes
115	} else if (trainConfig == 2) {
116	eventCutArray[ 0] = "0000012"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD, V0AND , only boxes
117	} else if (trainConfig == 3) {
118	eventCutArray[ 0] = "0000012"; photonCutArray[ 0] = "002093260003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Gamma pp 2-76TeV , only boxes
119	} else if (trainConfig == 4) {
120	eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , only Minbias MC
121	} else if (trainConfig == 5) {
122	eventCutArray[ 0] = "0001011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD, V0AND
123	} else if (trainConfig == 6) {
124	eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "002093260003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Gamma pp 2-76TeV
125	} else if (trainConfig == 7) {
126	eventCutArray[ 0] = "0002011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , only Minbias MC
127	} else if (trainConfig == 8) {
128	eventCutArray[ 0] = "0003011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD, V0AND , only Minbias MC
129	} else if (trainConfig == 9) {
130	eventCutArray[ 0] = "0002012"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , only Boxes MC
131	} else if (trainConfig == 10) {
132	eventCutArray[ 0] = "0003012"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD, V0AND, only Boxes MC
133	} else if (trainConfig == 11) {
134	eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , all photon qualities
135	} else if (trainConfig == 12) {
136	eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "007093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , all photon qualities, min R = 35 cm
137	} else if (trainConfig == 13) {
138	eventCutArray[ 0] = "0002011"; photonCutArray[ 0] = "002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , all photon qualities
139	} else if (trainConfig == 14) {
140	eventCutArray[ 0] = "0002011"; photonCutArray[ 0] = "007093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , all photon qualities, min R = 35 cm
141	} else if (trainConfig == 15) {
142	eventCutArray[ 0] = "0000011"; photonCutArray[ 0] = "002092970028250400000"; mesonCutArray[0] = "01525065000000"; //standard cut LHC11h pp 2.76TeV
143	} else {
144	Error(Form("GammaConvV1_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration");
145	return;
146	}
147
148	TList *EventCutList = new TList();
149	TList *ConvCutList = new TList();
150	TList *MesonCutList = new TList();
151
152	TList *HeaderList = new TList();
153	TObjString *Header2 = new TObjString("BOX");
154	HeaderList->Add(Header2);
155
156	EventCutList->SetOwner(kTRUE);
157	AliConvEventCuts *analysisEventCuts = new AliConvEventCuts[numberOfCuts];
158	ConvCutList->SetOwner(kTRUE);
159	AliConversionPhotonCuts *analysisCuts = new AliConversionPhotonCuts[numberOfCuts];
160	MesonCutList->SetOwner(kTRUE);
161	AliConversionMesonCuts *analysisMesonCuts = new AliConversionMesonCuts[numberOfCuts];
162
163
164	for(Int_t i = 0; i<numberOfCuts; i++){
165	analysisEventCuts[i] = new AliConvEventCuts();
166	analysisEventCuts[i]->InitializeCutsFromCutString(eventCutArray[i].Data());
167	EventCutList->Add(analysisEventCuts[i]);
168	analysisEventCuts[i]->SetFillCutHistograms("",kFALSE);
169
170	analysisCuts[i] = new AliConversionPhotonCuts();
171	analysisCuts[i]->InitializeCutsFromCutString(photonCutArray[i].Data());
172	ConvCutList->Add(analysisCuts[i]);
173	analysisCuts[i]->SetFillCutHistograms("",kFALSE);
174
175	analysisMesonCuts[i] = new AliConversionMesonCuts();
176	analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data());
177	MesonCutList->Add(analysisMesonCuts[i]);
178	analysisMesonCuts[i]->SetFillCutHistograms("");
179
180	analysisEventCuts[i]->SetAcceptedHeader(HeaderList);
181	}
182
183	task->SetEventCutList(numberOfCuts,EventCutList);
184	task->SetConversionCutList(numberOfCuts,ConvCutList);
185	task->SetMesonCutList(numberOfCuts,MesonCutList);
186	task->SetMoveParticleAccordingToVertex(kTRUE);
187	task->SetDoMesonAnalysis(kTRUE);
188	task->SetDoMesonQA(enableQAMesonTask); //Attention new switch for Pi0 QA
189	task->SetDoPhotonQA(enableQAPhotonTask); //Attention new switch small for Photon QA
190
191	//connect containers
192	AliAnalysisDataContainer *coutput =
193	mgr->CreateContainer(Form("GammaConvV1_%i",trainConfig), TList::Class(),
194	AliAnalysisManager::kOutputContainer,Form("GammaConvV1_%i.root",trainConfig));
195
196	mgr->AddTask(task);
197	mgr->ConnectInput(task,0,cinput);
198	mgr->ConnectOutput(task,1,coutput);
199
200	return;
88202ecb	201
88202ecb	202	}