[u/mrichter/AliRoot.git] / PWGLF / SPECTRA / PiKaPr / TestAOD / runAODProof.C

void runAODProof(Int_t c=2, const char * proofMode = "full")
{ //1 data AOD049
  //2 MC AOD048
  //3 data AOD086
  //4 MC AOD090
  
  gEnv->SetValue("XSec.GSI.DelegProxy", "2");
  
  gSystem->Load("libTree.so");
   gSystem->Load("libGeom.so");
   gSystem->Load("libVMC.so");
   gSystem->Load("libPhysics.so");
   gSystem->Load("libSTEERBase.so");
   gSystem->Load("libESD.so");
   gSystem->Load("libAOD.so");
   gSystem->Load("libANALYSIS.so");
   gSystem->Load("libOADB.so");
   gSystem->Load("libANALYSISalice.so");
   gSystem->AddIncludePath("-I$ALICE_ROOT/include");


   AliAnalysisAlien * handler = new AliAnalysisAlien("test");
   handler->SetOverwriteMode();
   handler->SetRunMode(proofMode);
   handler->SetProofReset(0);
   //handler->SetROOTVersion("v5-33-02a");
   //handler->SetAliROOTVersion("v5-03-11-AN");
   handler->SetAliROOTVersion("v5-03-16-AN");
   
   //handler->SetNproofWorkers();
   handler->SetNproofWorkersPerSlave(1);
   handler->SetProofCluster(Form("%s@alice-caf.cern.ch", gSystem->Getenv("CAFUSER")));
   //handler->SetProofCluster(Form("%s@skaf.saske.sk",gSystem->Getenv("CAFUSER")));
   
   // Set handler for Real DATA:
   if (c == 1){
     handler->SetProofDataSet("/alice/data/LHC10h_000138653_p2_AOD049#aodTree");
     //handler->SetProofDataSet("/alice/data/LHC10h_000138653_p2_AOD049#aodTree|/alice/data/LHC10h_000138662_p2_AOD049#aodTree|/alice/data/LHC10h_000138666_p2_AOD049#aodTree|/alice/data/LHC10h_000138795_p2_AOD049#aodTree|/alice/data/LHC10h_000139107_p2_AOD049#aodTree|/alice/data/LHC10h_000139110_p2_AOD049#aodTree");
   }
   if (c == 2){
     handler->SetProofDataSet("/alice/sim/LHC11a10a_000138653_AOD048#aodTree|/alice/sim/LHC11a10a_000138662_AOD048#aodTree|/alice/sim/LHC11a10a_000138666_AOD048#aodTree|/alice/sim/LHC11a10a_000138795_AOD048#aodTree|/alice/sim/LHC11a10a_000139107_AOD048#aodTree|/alice/sim/LHC11a10a_000139110_AOD048#aodTree");      
   }
   if (c == 3){
     handler->SetProofDataSet("/default/lmilano/LHC10h_000138653_AOD086_p2#aodTree|/default/lmilano/LHC10h_000138662_AOD086_p2#aodTree|/default/lmilano/LHC10h_000138666_AOD086_p2#aodTree|/default/lmilano/LHC10h_000139107_AOD086_p2#aodTree");      
   }
   if (c == 4){
     handler->SetProofDataSet("/default/lmilano/LHC11a10a_bis_138653_AOD090#aodTree|/default/lmilano/LHC11a10a_bis_138662_AOD090#aodTree|/default/lmilano/LHC11a10a_bis_138666_AOD090#aodTree|/default/lmilano/LHC11a10a_bis_139107_AOD090#aodTree");      
   }
   
   handler->SetAliRootMode("default");
   handler->SetAdditionalLibs("AliSpectraAODHistoManager.cxx AliSpectraAODHistoManager.h AliSpectraAODEventCuts.cxx AliSpectraAODEventCuts.h AliSpectraAODTrackCuts.cxx AliSpectraAODTrackCuts.h AliAnalysisTaskSpectraAOD.cxx AliAnalysisTaskSpectraAOD.h");
   handler->SetAnalysisSource("AliSpectraAODHistoManager.cxx+ AliSpectraAODEventCuts.cxx+ AliSpectraAODTrackCuts.cxx+ AliAnalysisTaskSpectraAOD.cxx+");
   handler->SetFileForTestMode("filelist.txt"); // list of local files for testing
   //  handler->SetAliRootMode("");
   handler->SetClearPackages();
   
   AliAnalysisManager *mgr = new AliAnalysisManager("testAnalysis");
   mgr->SetGridHandler(handler);
   AliAODInputHandler* aodH = new AliAODInputHandler();
   mgr->SetInputEventHandler(aodH);
   
   gROOT->LoadMacro("AliSpectraAODTrackCuts.cxx+g");
   gROOT->LoadMacro("AliSpectraAODEventCuts.cxx+g");
   gROOT->LoadMacro("AliSpectraAODHistoManager.cxx+g");
   gROOT->LoadMacro("AliAnalysisTaskSpectraAOD.cxx+g");
   
   // Add PID task
   gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
   Bool_t isMC = kFALSE;
   if (c == 2 || c == 4) isMC = kTRUE;   
   AliAnalysisTask * taskPID = AddTaskPIDResponse(isMC);
   mgr->AddTask(taskPID);
   
   //LOOP OVER SELECTION
   Double_t CentCutMin[4]={0,0,20,20};
   Double_t CentCutMax[4]={100,5,40,40};
   Double_t QvecCutMin[4]={0,0,0,3};
   Double_t QvecCutMax[4]={100,100,2,100};
   // Double_t CentCutMin[4]={5,10,20,30};
   // Double_t CentCutMax[4]={10,20,30,40};
   // Double_t QvecCutMin[4]={0,0,0,0};
   // Double_t QvecCutMax[4]={100,100,100,100};
   AliSpectraAODPID * pid = new AliSpectraAODPID(AODPIDType_t::kNSigmaTPCTOF);
   pid->SetNSigmaCut(5.);

   for(Int_t iCut=1;iCut<2;iCut++){
     AliAnalysisTaskSpectraAOD *task = new AliAnalysisTaskSpectraAOD("TaskAODExercise");
     mgr->AddTask(task);
     //physics selection
     task->SelectCollisionCandidates();     
     // set pid object
     task->SetPID(pid);
     // Set the cuts
     AliSpectraAODEventCuts * vcuts = new AliSpectraAODEventCuts("Event Cuts");
     AliSpectraAODTrackCuts  * tcuts = new AliSpectraAODTrackCuts("Track Cuts");
     if(c==1 || c==2)tcuts->SetTrackType(5); //AOD 046 & 047. Standard Cut with loose DCA
     if(c==3 || c==4)tcuts->SetTrackType(10); //AOD 086 & 090. Standard Raa cut
     tcuts->SetEta(.8);
     //tcuts->SetDCA(.1);
     tcuts->SetPt(5);
     tcuts->SetPtTOFMatching(0.6);   
     tcuts->SetQvecMin(QvecCutMin[iCut]);   
     tcuts->SetQvecMax(QvecCutMax[iCut]);    
     vcuts->SetCentralityCutMax(CentCutMax[iCut]);  
     vcuts->SetCentralityCutMin(CentCutMin[iCut]);
     task->SetEventCuts(vcuts);
     task->SetTrackCuts(tcuts);
     task->SetYCut(.5);
     vcuts->PrintCuts();
     tcuts->PrintCuts();
     
     // check for MC or real data
     if (c == 2 || c == 4)
       {
	 task->SetIsMC(kTRUE);
	 AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	 AliAnalysisDataContainer *coutputpt1 = mgr->CreateContainer(Form("chistpt%d",iCut), AliSpectraAODHistoManager::Class(),  AliAnalysisManager::kOutputContainer, 
								     Form("Pt.AOD.1._MC_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	 AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer(Form("cvcutpt%d",iCut), AliSpectraAODEventCuts::Class(),    AliAnalysisManager::kOutputContainer, 
								     Form("Pt.AOD.1._MC_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	 AliAnalysisDataContainer *coutputpt3 = mgr->CreateContainer(Form("ctcutpt%d",iCut), AliSpectraAODTrackCuts::Class(),     AliAnalysisManager::kOutputContainer, 
								     Form("Pt.AOD.1._MC_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	 AliAnalysisDataContainer *coutputpt4 = mgr->CreateContainer(Form("cpidpt%d",iCut),  AliSpectraAODPID::Class(),     AliAnalysisManager::kOutputContainer, 
								     Form("Pt.AOD.1._MC_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
       }
     if (c == 1 || c==3)
       {
	 AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	 AliAnalysisDataContainer *coutputpt1 = mgr->CreateContainer(Form("chistpt%d",iCut), AliSpectraAODHistoManager::Class(),  AliAnalysisManager::kOutputContainer, 
								     Form("Pt.AOD.1._data_ptcut_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	 AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer(Form("cvcutpt%d",iCut), AliSpectraAODEventCuts::Class(),    AliAnalysisManager::kOutputContainer, 
								     Form("Pt.AOD.1._data_ptcut_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	 AliAnalysisDataContainer *coutputpt3 = mgr->CreateContainer(Form("ctcutpt%d",iCut), AliSpectraAODTrackCuts::Class(),     AliAnalysisManager::kOutputContainer, 
								     Form("Pt.AOD.1._data_ptcut_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	 AliAnalysisDataContainer *coutputpt4 = mgr->CreateContainer(Form("cpidpt%d",iCut),  AliSpectraAODPID::Class(),     AliAnalysisManager::kOutputContainer, 
								     Form("Pt.AOD.1._data_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	 
       }
     mgr->ConnectInput(task, 0, cinput);
     mgr->ConnectOutput(task, 1, coutputpt1);
     mgr->ConnectOutput(task, 2, coutputpt2);
     mgr->ConnectOutput(task, 3, coutputpt3);
     mgr->ConnectOutput(task, 4, coutputpt4);
   }
   mgr->SetDebugLevel(2);
   
   if (!mgr->InitAnalysis()) return;
   mgr->PrintStatus();
   mgr->StartAnalysis("proof");
}
Commit	Line	Data
acef4f19	1	void runAODProof(Int_t c=2, const char * proofMode = "full")
ed06106e	2	{ //1 data AOD049
	3	//2 MC AOD048
	4	//3 data AOD086
	5	//4 MC AOD090
	6
	7	gEnv->SetValue("XSec.GSI.DelegProxy", "2");
	8
	9	gSystem->Load("libTree.so");
c88234ad	10	gSystem->Load("libGeom.so");
	11	gSystem->Load("libVMC.so");
	12	gSystem->Load("libPhysics.so");
	13	gSystem->Load("libSTEERBase.so");
	14	gSystem->Load("libESD.so");
	15	gSystem->Load("libAOD.so");
	16	gSystem->Load("libANALYSIS.so");
	17	gSystem->Load("libOADB.so");
	18	gSystem->Load("libANALYSISalice.so");
	19	gSystem->AddIncludePath("-I$ALICE_ROOT/include");
	20
	21
	22	AliAnalysisAlien * handler = new AliAnalysisAlien("test");
	23	handler->SetOverwriteMode();
	24	handler->SetRunMode(proofMode);
	25	handler->SetProofReset(0);
823864bf	26	//handler->SetROOTVersion("v5-33-02a");
823864bf	27	//handler->SetAliROOTVersion("v5-03-11-AN");
ed06106e	28	handler->SetAliROOTVersion("v5-03-16-AN");
6522a8cb	29
ed06106e	30	//handler->SetNproofWorkers();
ed06106e	31	handler->SetNproofWorkersPerSlave(1);
c88234ad	32	handler->SetProofCluster(Form("%s@alice-caf.cern.ch", gSystem->Getenv("CAFUSER")));
6522a8cb	33	//handler->SetProofCluster(Form("%s@skaf.saske.sk",gSystem->Getenv("CAFUSER")));
6522a8cb	34
823864bf	35	// Set handler for Real DATA:
ed06106e	36	if (c == 1){
	37	handler->SetProofDataSet("/alice/data/LHC10h_000138653_p2_AOD049#aodTree");
	38	//handler->SetProofDataSet("/alice/data/LHC10h_000138653_p2_AOD049#aodTree\|/alice/data/LHC10h_000138662_p2_AOD049#aodTree\|/alice/data/LHC10h_000138666_p2_AOD049#aodTree\|/alice/data/LHC10h_000138795_p2_AOD049#aodTree\|/alice/data/LHC10h_000139107_p2_AOD049#aodTree\|/alice/data/LHC10h_000139110_p2_AOD049#aodTree");
	39	}
	40	if (c == 2){
	41	handler->SetProofDataSet("/alice/sim/LHC11a10a_000138653_AOD048#aodTree\|/alice/sim/LHC11a10a_000138662_AOD048#aodTree\|/alice/sim/LHC11a10a_000138666_AOD048#aodTree\|/alice/sim/LHC11a10a_000138795_AOD048#aodTree\|/alice/sim/LHC11a10a_000139107_AOD048#aodTree\|/alice/sim/LHC11a10a_000139110_AOD048#aodTree");
	42	}
	43	if (c == 3){
	44	handler->SetProofDataSet("/default/lmilano/LHC10h_000138653_AOD086_p2#aodTree\|/default/lmilano/LHC10h_000138662_AOD086_p2#aodTree\|/default/lmilano/LHC10h_000138666_AOD086_p2#aodTree\|/default/lmilano/LHC10h_000139107_AOD086_p2#aodTree");
	45	}
	46	if (c == 4){
	47	handler->SetProofDataSet("/default/lmilano/LHC11a10a_bis_138653_AOD090#aodTree\|/default/lmilano/LHC11a10a_bis_138662_AOD090#aodTree\|/default/lmilano/LHC11a10a_bis_138666_AOD090#aodTree\|/default/lmilano/LHC11a10a_bis_139107_AOD090#aodTree");
	48	}
	49
c88234ad	50	handler->SetAliRootMode("default");
	51	handler->SetAdditionalLibs("AliSpectraAODHistoManager.cxx AliSpectraAODHistoManager.h AliSpectraAODEventCuts.cxx AliSpectraAODEventCuts.h AliSpectraAODTrackCuts.cxx AliSpectraAODTrackCuts.h AliAnalysisTaskSpectraAOD.cxx AliAnalysisTaskSpectraAOD.h");
	52	handler->SetAnalysisSource("AliSpectraAODHistoManager.cxx+ AliSpectraAODEventCuts.cxx+ AliSpectraAODTrackCuts.cxx+ AliAnalysisTaskSpectraAOD.cxx+");
	53	handler->SetFileForTestMode("filelist.txt"); // list of local files for testing
c88234ad	54	// handler->SetAliRootMode("");
c88234ad	55	handler->SetClearPackages();
ed06106e	56
c88234ad	57	AliAnalysisManager *mgr = new AliAnalysisManager("testAnalysis");
	58	mgr->SetGridHandler(handler);
	59	AliAODInputHandler* aodH = new AliAODInputHandler();
	60	mgr->SetInputEventHandler(aodH);
ed06106e	61
c88234ad	62	gROOT->LoadMacro("AliSpectraAODTrackCuts.cxx+g");
	63	gROOT->LoadMacro("AliSpectraAODEventCuts.cxx+g");
	64	gROOT->LoadMacro("AliSpectraAODHistoManager.cxx+g");
	65	gROOT->LoadMacro("AliAnalysisTaskSpectraAOD.cxx+g");
ed06106e	66
c88234ad	67	// Add PID task
c88234ad	68	gROOT->LoadMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
823864bf	69	Bool_t isMC = kFALSE;
ed06106e	70	if (c == 2 \|\| c == 4) isMC = kTRUE;
823864bf	71	AliAnalysisTask * taskPID = AddTaskPIDResponse(isMC);
c88234ad	72	mgr->AddTask(taskPID);
ed06106e	73
823864bf	74	//LOOP OVER SELECTION
	75	Double_t CentCutMin[4]={0,0,20,20};
	76	Double_t CentCutMax[4]={100,5,40,40};
	77	Double_t QvecCutMin[4]={0,0,0,3};
	78	Double_t QvecCutMax[4]={100,100,2,100};
624a6bb0	79	// Double_t CentCutMin[4]={5,10,20,30};
	80	// Double_t CentCutMax[4]={10,20,30,40};
	81	// Double_t QvecCutMin[4]={0,0,0,0};
	82	// Double_t QvecCutMax[4]={100,100,100,100};
e8b0fe64	83	AliSpectraAODPID * pid = new AliSpectraAODPID(AODPIDType_t::kNSigmaTPCTOF);
	84	pid->SetNSigmaCut(5.);
	85
6522a8cb	86	for(Int_t iCut=1;iCut<2;iCut++){
823864bf	87	AliAnalysisTaskSpectraAOD *task = new AliAnalysisTaskSpectraAOD("TaskAODExercise");
	88	mgr->AddTask(task);
	89	//physics selection
	90	task->SelectCollisionCandidates();
e8b0fe64	91	// set pid object
e8b0fe64	92	task->SetPID(pid);
823864bf	93	// Set the cuts
	94	AliSpectraAODEventCuts * vcuts = new AliSpectraAODEventCuts("Event Cuts");
	95	AliSpectraAODTrackCuts * tcuts = new AliSpectraAODTrackCuts("Track Cuts");
ed06106e	96	if(c==1 \|\| c==2)tcuts->SetTrackType(5); //AOD 046 & 047. Standard Cut with loose DCA
ed06106e	97	if(c==3 \|\| c==4)tcuts->SetTrackType(10); //AOD 086 & 090. Standard Raa cut
823864bf	98	tcuts->SetEta(.8);
823864bf	99	//tcuts->SetDCA(.1);
624a6bb0	100	tcuts->SetPt(5);
823864bf	101	tcuts->SetPtTOFMatching(0.6);
	102	tcuts->SetQvecMin(QvecCutMin[iCut]);
	103	tcuts->SetQvecMax(QvecCutMax[iCut]);
	104	vcuts->SetCentralityCutMax(CentCutMax[iCut]);
	105	vcuts->SetCentralityCutMin(CentCutMin[iCut]);
	106	task->SetEventCuts(vcuts);
	107	task->SetTrackCuts(tcuts);
823864bf	108	task->SetYCut(.5);
	109	vcuts->PrintCuts();
	110	tcuts->PrintCuts();
	111
	112	// check for MC or real data
ed06106e	113	if (c == 2 \|\| c == 4)
823864bf	114	{
	115	task->SetIsMC(kTRUE);
	116	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	117	AliAnalysisDataContainer *coutputpt1 = mgr->CreateContainer(Form("chistpt%d",iCut), AliSpectraAODHistoManager::Class(), AliAnalysisManager::kOutputContainer,
	118	Form("Pt.AOD.1._MC_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	119	AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer(Form("cvcutpt%d",iCut), AliSpectraAODEventCuts::Class(), AliAnalysisManager::kOutputContainer,
	120	Form("Pt.AOD.1._MC_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	121	AliAnalysisDataContainer *coutputpt3 = mgr->CreateContainer(Form("ctcutpt%d",iCut), AliSpectraAODTrackCuts::Class(), AliAnalysisManager::kOutputContainer,
	122	Form("Pt.AOD.1._MC_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
e8b0fe64	123	AliAnalysisDataContainer *coutputpt4 = mgr->CreateContainer(Form("cpidpt%d",iCut), AliSpectraAODPID::Class(), AliAnalysisManager::kOutputContainer,
e8b0fe64	124	Form("Pt.AOD.1._MC_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
823864bf	125	}
ed06106e	126	if (c == 1 \|\| c==3)
823864bf	127	{
	128	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	129	AliAnalysisDataContainer *coutputpt1 = mgr->CreateContainer(Form("chistpt%d",iCut), AliSpectraAODHistoManager::Class(), AliAnalysisManager::kOutputContainer,
	130	Form("Pt.AOD.1._data_ptcut_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	131	AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer(Form("cvcutpt%d",iCut), AliSpectraAODEventCuts::Class(), AliAnalysisManager::kOutputContainer,
	132	Form("Pt.AOD.1._data_ptcut_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
	133	AliAnalysisDataContainer *coutputpt3 = mgr->CreateContainer(Form("ctcutpt%d",iCut), AliSpectraAODTrackCuts::Class(), AliAnalysisManager::kOutputContainer,
	134	Form("Pt.AOD.1._data_ptcut_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
e8b0fe64	135	AliAnalysisDataContainer *coutputpt4 = mgr->CreateContainer(Form("cpidpt%d",iCut), AliSpectraAODPID::Class(), AliAnalysisManager::kOutputContainer,
e8b0fe64	136	Form("Pt.AOD.1._data_Cent%.0fto%.0f_QVec%.1fto%.1f.root",CentCutMin[iCut],CentCutMax[iCut],QvecCutMin[iCut],QvecCutMax[iCut]));
823864bf	137
	138	}
	139	mgr->ConnectInput(task, 0, cinput);
	140	mgr->ConnectOutput(task, 1, coutputpt1);
	141	mgr->ConnectOutput(task, 2, coutputpt2);
	142	mgr->ConnectOutput(task, 3, coutputpt3);
e8b0fe64	143	mgr->ConnectOutput(task, 4, coutputpt4);
823864bf	144	}
c88234ad	145	mgr->SetDebugLevel(2);
823864bf	146
c88234ad	147	if (!mgr->InitAnalysis()) return;
c88234ad	148	mgr->PrintStatus();
ed06106e	149	mgr->StartAnalysis("proof");
c88234ad	150	}