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

void runAOD(Bool_t isMC = 0)
{

//gEnv->SetValue("XSec.GSI.DelegProxy", "2");
//TProof::Open("pverstee@alice-caf.cern.ch");

   gSystem->Load("libTree");
   gSystem->Load("libGeom");
   gSystem->Load("libVMC");
   gSystem->Load("libPhysics");
   gSystem->Load("libSTEERBase");
   gSystem->Load("libESD");
   gSystem->Load("libAOD");
   gSystem->Load("libANALYSIS");
   gSystem->Load("libANALYSISalice");
   gSystem->Load("libANALYSIS");
   gSystem->Load("libANALYSISalice");
   gROOT->LoadMacro("$ALICE_ROOT/PWG0/CreateESDChain.C");

   ifstream flist;
   flist.open("filelist.txt");
   TChain * chain = new TChain("aodTree");
   TString line;
   while (line.ReadLine(flist))
   {
      gSystem->ExpandPathName(line);
      chain->AddFile(line.Data());
      cout << "Adding file " << line.Data() << endl;
   }
//  gProof->UploadPackage("AF-v4-19-04-AN");
//  gProof->EnablePackage("AF-v4-19-04-AN");
   gSystem->AddIncludePath("-I$ALICE_ROOT/include");
   gStyle->SetPalette(1);


   AliAnalysisManager *mgr = new AliAnalysisManager("testAnalysis");
   AliAODInputHandler* aodH = new AliAODInputHandler();
   mgr->SetInputEventHandler(aodH);

   gROOT->LoadMacro("AliSpectraAODTrackCuts.cxx+g");
   gROOT->LoadMacro("AliSpectraAODVertexCuts.cxx+g");
   gROOT->LoadMacro("AliSpectraAODHistoManager.cxx+g");
   gROOT->LoadMacro("AliAnalysisTaskSpectraAOD.cxx+g");

   gROOT->LoadMacro("$(ALICE_ROOT)/ANALYSIS/macros/AddTaskPIDResponse.C");
   AliAnalysisTask * taskPID = AddTaskPIDResponse();
   mgr->AddTask(taskPID);


   AliAnalysisTaskSpectraAOD *task = new AliAnalysisTaskSpectraAOD("TaskAODExercise");
   task->SetIsMC(isMC);
   mgr->AddTask(task);

   // Set the cuts
   AliSpectraAODVertexCuts * vcuts = new AliSpectraAODVertexCuts("Vertex Cuts");
   AliSpectraAODTrackCuts  * tcuts = new AliSpectraAODTrackCuts("Tracks Cuts");
   tcuts->SetTrackType(6);
   tcuts->SetEta(1.);
   task->SetVertexCuts(vcuts);
   task->SetTrackCuts(tcuts);


   AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
   AliAnalysisDataContainer *coutputpt1 = mgr->CreateContainer("chistpt", AliSpectraAODHistoManager::Class(),  AliAnalysisManager::kOutputContainer, "Pt.AOD.1.root");
   AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer("cvcutpt", AliSpectraAODVertexCuts::Class(),    AliAnalysisManager::kOutputContainer, "Pt.AOD.1.root");
   AliAnalysisDataContainer *coutputpt3 = mgr->CreateContainer("ctcutpt", AliSpectraAODTrackCuts::Class(),     AliAnalysisManager::kOutputContainer, "Pt.AOD.1.root");

   mgr->ConnectInput(task, 0, cinput);
   mgr->ConnectOutput(task, 1, coutputpt1);
   mgr->ConnectOutput(task, 2, coutputpt2);
   mgr->ConnectOutput(task, 3, coutputpt3);
   mgr->SetDebugLevel(2);

   if (!mgr->InitAnalysis()) return;
   mgr->PrintStatus();
   mgr->StartAnalysis("local", chain, 100);
}
Commit	Line	Data
c88234ad	1	void runAOD(Bool_t isMC = 0)
	2	{
	3
	4	//gEnv->SetValue("XSec.GSI.DelegProxy", "2");
	5	//TProof::Open("pverstee@alice-caf.cern.ch");
	6
4070f709	7	gSystem->Load("libTree");
	8	gSystem->Load("libGeom");
	9	gSystem->Load("libVMC");
	10	gSystem->Load("libPhysics");
	11	gSystem->Load("libSTEERBase");
	12	gSystem->Load("libESD");
	13	gSystem->Load("libAOD");
	14	gSystem->Load("libANALYSIS");
	15	gSystem->Load("libANALYSISalice");
c88234ad	16	gSystem->Load("libANALYSIS");
	17	gSystem->Load("libANALYSISalice");
	18	gROOT->LoadMacro("$ALICE_ROOT/PWG0/CreateESDChain.C");
	19
	20	ifstream flist;
	21	flist.open("filelist.txt");
	22	TChain * chain = new TChain("aodTree");
	23	TString line;
	24	while (line.ReadLine(flist))
	25	{
	26	gSystem->ExpandPathName(line);
	27	chain->AddFile(line.Data());
	28	cout << "Adding file " << line.Data() << endl;
	29	}
	30	// gProof->UploadPackage("AF-v4-19-04-AN");
	31	// gProof->EnablePackage("AF-v4-19-04-AN");
	32	gSystem->AddIncludePath("-I$ALICE_ROOT/include");
	33	gStyle->SetPalette(1);
	34
	35
	36
	37	AliAnalysisManager *mgr = new AliAnalysisManager("testAnalysis");
	38	AliAODInputHandler* aodH = new AliAODInputHandler();
	39	mgr->SetInputEventHandler(aodH);
	40
	41	gROOT->LoadMacro("AliSpectraAODTrackCuts.cxx+g");
	42	gROOT->LoadMacro("AliSpectraAODVertexCuts.cxx+g");
	43	gROOT->LoadMacro("AliSpectraAODHistoManager.cxx+g");
	44	gROOT->LoadMacro("AliAnalysisTaskSpectraAOD.cxx+g");
	45
	46	gROOT->LoadMacro("$(ALICE_ROOT)/ANALYSIS/macros/AddTaskPIDResponse.C");
	47	AliAnalysisTask * taskPID = AddTaskPIDResponse();
	48	mgr->AddTask(taskPID);
	49
	50
	51	AliAnalysisTaskSpectraAOD *task = new AliAnalysisTaskSpectraAOD("TaskAODExercise");
	52	task->SetIsMC(isMC);
	53	mgr->AddTask(task);
	54
	55	// Set the cuts
	56	AliSpectraAODVertexCuts * vcuts = new AliSpectraAODVertexCuts("Vertex Cuts");
	57	AliSpectraAODTrackCuts * tcuts = new AliSpectraAODTrackCuts("Tracks Cuts");
	58	tcuts->SetTrackType(6);
	59	tcuts->SetEta(1.);
	60	task->SetVertexCuts(vcuts);
	61	task->SetTrackCuts(tcuts);
	62
	63
	64	AliAnalysisDataContainer *cinput = mgr->GetCommonInputContainer();
	65	AliAnalysisDataContainer *coutputpt1 = mgr->CreateContainer("chistpt", AliSpectraAODHistoManager::Class(), AliAnalysisManager::kOutputContainer, "Pt.AOD.1.root");
	66	AliAnalysisDataContainer *coutputpt2 = mgr->CreateContainer("cvcutpt", AliSpectraAODVertexCuts::Class(), AliAnalysisManager::kOutputContainer, "Pt.AOD.1.root");
	67	AliAnalysisDataContainer *coutputpt3 = mgr->CreateContainer("ctcutpt", AliSpectraAODTrackCuts::Class(), AliAnalysisManager::kOutputContainer, "Pt.AOD.1.root");
	68
	69	mgr->ConnectInput(task, 0, cinput);
	70	mgr->ConnectOutput(task, 1, coutputpt1);
	71	mgr->ConnectOutput(task, 2, coutputpt2);
	72	mgr->ConnectOutput(task, 3, coutputpt3);
	73	mgr->SetDebugLevel(2);
	74
	75	if (!mgr->InitAnalysis()) return;
	76	mgr->PrintStatus();
	77	mgr->StartAnalysis("local", chain, 100);
	78	}