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

void AddTask_GammaConvV1_PbPb2(  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");
      
   // ================== 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 cutnumber = "0000000002084000002200000000"; 
   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;
      }

      // Set AnalysisCut Number
      AliConversionCuts *fCuts=NULL;
      if(cutnumber!=""){
         fCuts= new AliConversionCuts(cutnumber.Data(),cutnumber.Data());
         fCuts->SetPreSelectionCutFlag(kTRUE);
         if(fCuts->InitializeCutsFromCutString(cutnumber.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(0);
   task->SetIsMC(isMC);
   // Cut Numbers to use in Analysis
   Int_t numberOfCuts = 1;

   TString *cutarray = new TString[numberOfCuts];
   TString *mesonCutArray = new TString[numberOfCuts];

   if (trainConfig == 1){ 
      cutarray[ 0] = "0000012002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , only boxes
   } else if (trainConfig == 2) { 
      cutarray[ 0] = "0001012002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD, V0AND , only boxes
   } else if (trainConfig == 3) { 
      cutarray[ 0] = "0000012002093260003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Gamma pp 2-76TeV , only boxes
   } else if (trainConfig == 4) { 
      cutarray[ 0] = "0000011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , only Minbias MC
   } else if (trainConfig == 5) { 
      cutarray[ 0] = "0001011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD, V0AND
   } else if (trainConfig == 6) { 
      cutarray[ 0] = "0000011002093260003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Gamma pp 2-76TeV
   } else if (trainConfig == 7) {    
      cutarray[ 0] = "0002011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , only Minbias MC
   } else if (trainConfig == 8) {    
      cutarray[ 0] = "0003011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD, V0AND , only Minbias MC
   } else if (trainConfig == 9) {    
      cutarray[ 0] = "0002012002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , only Boxes MC
   } else if (trainConfig == 10) { 
      cutarray[ 0] = "0003012002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD, V0AND, only Boxes MC
   } else if (trainConfig == 11) { 
      cutarray[ 0] = "0000011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , all photon qualities
   } else if (trainConfig == 12) { 
      cutarray[ 0] = "0000011007093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , all photon qualities, min R = 35 cm
   } else if (trainConfig == 13) { 
      cutarray[ 0] = "0002011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , all photon qualities
   } else if (trainConfig == 14) { 
      cutarray[ 0] = "0002011007093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , all photon qualities, min R = 35 cm
   } else if (trainConfig == 15) { 
      cutarray[ 0] = "00000010020929700282504000000"; 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 *ConvCutList = new TList();
   TList *MesonCutList = new TList();

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

   ConvCutList->SetOwner(kTRUE);
   AliConversionCuts **analysisCuts = new AliConversionCuts*[numberOfCuts];
   MesonCutList->SetOwner(kTRUE);
   AliConversionMesonCuts **analysisMesonCuts = new AliConversionMesonCuts*[numberOfCuts];

   for(Int_t i = 0; i<numberOfCuts; i++){
      analysisCuts[i] = new AliConversionCuts();
      analysisCuts[i]->InitializeCutsFromCutString(cutarray[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("");
      analysisCuts[i]->SetAcceptedHeader(HeaderList);
   }

   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
88202ecb	1	void AddTask_GammaConvV1_PbPb2( 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	TString cutnumberAODBranch = "0000000060084001001500000"
	7	) {
	8
	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");
	27
	28	// ================== GetAnalysisManager ===============================
	29	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	30	if (!mgr) {
	31	Error(Form("AddTask_GammaConvV1_%i",trainConfig), "No analysis manager found.");
	32	return ;
	33	}
	34
	35	// ================== GetInputEventHandler =============================
	36	AliVEventHandler *inputHandler=mgr->GetInputEventHandler();
	37
	38	//========= Add PID Reponse to ANALYSIS manager ====
	39	if(!(AliPIDResponse*)mgr->GetTask("PIDResponseTask")){
	40	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
	41	AddTaskPIDResponse(isMC);
	42	}
	43
	44	//========= Set Cutnumber for V0Reader ================================
	45	TString cutnumber = "0000000002084000002200000000";
	46	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	47
	48	//========= Add V0 Reader to ANALYSIS manager if not yet existent =====
	49	if( !(AliV0ReaderV1*)mgr->GetTask("V0ReaderV1") ){
	50	AliV0ReaderV1 *fV0ReaderV1 = new AliV0ReaderV1("V0ReaderV1");
	51
	52	fV0ReaderV1->SetUseOwnXYZCalculation(kTRUE);
	53	fV0ReaderV1->SetCreateAODs(kFALSE);// AOD Output
	54	fV0ReaderV1->SetUseAODConversionPhoton(kTRUE);
	55
	56	if (!mgr) {
	57	Error("AddTask_V0ReaderV1", "No analysis manager found.");
	58	return;
	59	}
	60
	61	// Set AnalysisCut Number
	62	AliConversionCuts *fCuts=NULL;
	63	if(cutnumber!=""){
	64	fCuts= new AliConversionCuts(cutnumber.Data(),cutnumber.Data());
65	fCuts->SetPreSelectionCutFlag(kTRUE);
66	if(fCuts->InitializeCutsFromCutString(cutnumber.Data())){
67	fV0ReaderV1->SetConversionCuts(fCuts);
68	fCuts->SetFillCutHistograms("",kTRUE);
69	}
70	}
71	if(inputHandler->IsA()==AliAODInputHandler::Class()){
72	// AOD mode
73	fV0ReaderV1->SetDeltaAODBranchName(Form("GammaConv_%s_gamma",cutnumberAODBranch.Data()));
74	}
75	fV0ReaderV1->Init();
76
77	AliLog::SetGlobalLogLevel(AliLog::kInfo);
78
79	//connect input V0Reader
80	mgr->AddTask(fV0ReaderV1);
81	mgr->ConnectInput(fV0ReaderV1,0,cinput);
82
83	}
84
85	//================================================
86	//========= Add task to the ANALYSIS manager =====
87	//================================================
88	AliAnalysisTaskGammaConvV1 *task=NULL;
89	task= new AliAnalysisTaskGammaConvV1(Form("GammaConvV1_%i",trainConfig));
90	task->SetIsHeavyIon(0);
91	task->SetIsMC(isMC);
92	// Cut Numbers to use in Analysis
93	Int_t numberOfCuts = 1;
94
95	TString *cutarray = new TString[numberOfCuts];
96	TString *mesonCutArray = new TString[numberOfCuts];
97
98	if (trainConfig == 1){
99	cutarray[ 0] = "0000012002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , only boxes
100	} else if (trainConfig == 2) {
101	cutarray[ 0] = "0001012002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD, V0AND , only boxes
102	} else if (trainConfig == 3) {
103	cutarray[ 0] = "0000012002093260003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Gamma pp 2-76TeV , only boxes
104	} else if (trainConfig == 4) {
105	cutarray[ 0] = "0000011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , only Minbias MC
106	} else if (trainConfig == 5) {
107	cutarray[ 0] = "0001011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD, V0AND
108	} else if (trainConfig == 6) {
109	cutarray[ 0] = "0000011002093260003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Gamma pp 2-76TeV
110	} else if (trainConfig == 7) {
111	cutarray[ 0] = "0002011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , only Minbias MC
112	} else if (trainConfig == 8) {
113	cutarray[ 0] = "0003011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD, V0AND , only Minbias MC
114	} else if (trainConfig == 9) {
115	cutarray[ 0] = "0002012002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , only Boxes MC
116	} else if (trainConfig == 10) {
117	cutarray[ 0] = "0003012002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD, V0AND, only Boxes MC
118	} else if (trainConfig == 11) {
119	cutarray[ 0] = "0000011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , all photon qualities
120	} else if (trainConfig == 12) {
121	cutarray[ 0] = "0000011007093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV without SDD , all photon qualities, min R = 35 cm
122	} else if (trainConfig == 13) {
123	cutarray[ 0] = "0002011002093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , all photon qualities
124	} else if (trainConfig == 14) {
125	cutarray[ 0] = "0002011007093663003800000000"; mesonCutArray[0] = "01631031009000"; //standard cut Pi0 pp 2.76TeV with SDD , all photon qualities, min R = 35 cm
126	} else if (trainConfig == 15) {
127	cutarray[ 0] = "00000010020929700282504000000"; mesonCutArray[0] = "01525065000000"; //standard cut LHC11h pp 2.76TeV
128	} else {
129	Error(Form("GammaConvV1_%i",trainConfig), "wrong trainConfig variable no cuts have been specified for the configuration");
130	return;
131	}
132
133	TList *ConvCutList = new TList();
134	TList *MesonCutList = new TList();
135
136	TList *HeaderList = new TList();
137	TObjString *Header1 = new TObjString("BOX");
138	HeaderList->Add(Header2);
139
140	ConvCutList->SetOwner(kTRUE);
141	AliConversionCuts *analysisCuts = new AliConversionCuts[numberOfCuts];
142	MesonCutList->SetOwner(kTRUE);
143	AliConversionMesonCuts *analysisMesonCuts = new AliConversionMesonCuts[numberOfCuts];
144
145	for(Int_t i = 0; i<numberOfCuts; i++){
146	analysisCuts[i] = new AliConversionCuts();
147	analysisCuts[i]->InitializeCutsFromCutString(cutarray[i].Data());
148	ConvCutList->Add(analysisCuts[i]);
149
150	analysisCuts[i]->SetFillCutHistograms("",kFALSE);
151	analysisMesonCuts[i] = new AliConversionMesonCuts();
152	analysisMesonCuts[i]->InitializeCutsFromCutString(mesonCutArray[i].Data());
153	MesonCutList->Add(analysisMesonCuts[i]);
154	analysisMesonCuts[i]->SetFillCutHistograms("");
155	analysisCuts[i]->SetAcceptedHeader(HeaderList);
156	}
157
158	task->SetConversionCutList(numberOfCuts,ConvCutList);
159	task->SetMesonCutList(numberOfCuts,MesonCutList);
160	task->SetMoveParticleAccordingToVertex(kTRUE);
161	task->SetDoMesonAnalysis(kTRUE);
162	task->SetDoMesonQA(enableQAMesonTask); //Attention new switch for Pi0 QA
163	task->SetDoPhotonQA(enableQAPhotonTask); //Attention new switch small for Photon QA
164
165	//connect containers
166	AliAnalysisDataContainer *coutput =
167	mgr->CreateContainer(Form("GammaConvV1_%i",trainConfig), TList::Class(),
168	AliAnalysisManager::kOutputContainer,Form("GammaConvV1_%i.root",trainConfig));
169
170	mgr->AddTask(task);
171	mgr->ConnectInput(task,0,cinput);
172	mgr->ConnectOutput(task,1,coutput);
173
174	return;
175
176	}