[u/mrichter/AliRoot.git] / PWGDQ / dielectron / macrosLMEE / AddTask_Asako_LMEEPbPb2011AOD.C

AliAnalysisTask *AddTask_Asako_LMEEPbPb2011AOD(Bool_t runAll=kFALSE,Bool_t setMC=kFALSE,Bool_t getFromAlien=kFALSE, Bool_t PIDbaseline=kFALSE){

  Bool_t bESDANA=kFALSE; //Autodetect via InputHandler
  //get the current analysis manager
  AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
  if (!mgr) {
	Error("AddTask_Asako_LMEEPbPb2011AOD", "No analysis manager found.");
	return 0;
  }


//  create task and add it to the manager
//	gSystem->AddIncludePath("$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE");


  TString configBasePath("$TRAIN_ROOT/cbaumann_dielectron/");
  TString trainRoot=gSystem->Getenv("TRAIN_ROOT");

  if (trainRoot.IsNull()) configBasePath= "$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE/";


  if (getFromAlien &&
      (!gSystem->Exec("alien_cp alien:///alice/cern.ch/user/c/cbaumann/PWGDQ/dielectron/macrosLMEE/ConfigLMEEPbPb2011AOD.C .")) &&
      (!gSystem->Exec("alien_cp alien:///alice/cern.ch/user/c/cbaumann/PWGDQ/dielectron/macrosLMEE/LMEECutLibAOD.C ."))
     ) {
        configBasePath=Form("%s/",gSystem->pwd());
  }

  TString configFile("ConfigAsakoLMEEPbPb2011AOD.C");
  TString configLMEECutLib("LMEECutLibAsako.C");

  TString configFilePath(configBasePath+configFile);
  TString configLMEECutLibPath(configBasePath+configLMEECutLib);

  //AOD Usage currently tested with separate task, to be merged
  if (mgr->GetInputEventHandler()->IsA()==AliAODInputHandler::Class()){
	::Info("AddTaskLMEEPbPb2011", "no dedicated AOD configuration");
  }
  else if (mgr->GetInputEventHandler()->IsA()==AliESDInputHandler::Class()){
	::Info("AddTaskLMEEPbPb2011AOD","switching on ESD specific code");
	bESDANA=kTRUE;
  }


  //Do we have an MC handler?
  Bool_t hasMC=setMC;
  if (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0)
	hasMC=kTRUE;


  if (!gROOT->GetListOfGlobalFunctions()->FindObject(configLMEECutLib.Data()))
	gROOT->LoadMacro(configLMEECutLibPath.Data());
  if (!gROOT->GetListOfGlobalFunctions()->FindObject(configFile.Data()))
	gROOT->LoadMacro(configFilePath.Data());


  LMEECutLibAsako* cutlib = new LMEECutLibAsako();
  AliAnalysisTaskMultiDielectron *task=new AliAnalysisTaskMultiDielectron("MultiDiEData");
  if (!hasMC) task->UsePhysicsSelection();
  task->SetTriggerMask(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
//  task->SelectCollisionCandidates(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
//  task->SetRejectPileup();
  task->SelectCollisionCandidates(AliVEvent::kAny);  
  task->SetEventFilter(cutlib->GetEventCuts(LMEECutLibAsako::kPbPb2011TPCandTOF)); //
	

  //load dielectron configuration file

  //add dielectron analysis with different cuts to the task

  
  AliDielectron *lowmass1=ConfigAsakoLMEEPbPb2011AOD(1,hasMC,bESDANA);
  task->AddDielectron(lowmass1);
  printf("add: %s\n",lowmass1->GetName());
  

  AliDielectron *lowmass2=ConfigAsakoLMEEPbPb2011AOD(2,hasMC,bESDANA);
  task->AddDielectron(lowmass2);
  printf("add: %s\n",lowmass2->GetName());
  

  AliDielectron *lowmass3=ConfigAsakoLMEEPbPb2011AOD(3,hasMC,bESDANA);
  task->AddDielectron(lowmass3);
  printf("add: %s\n",lowmass3->GetName());
  

  AliDielectron *lowmass4=ConfigAsakoLMEEPbPb2011AOD(4,hasMC,bESDANA);
  task->AddDielectron(lowmass4);
  printf("add: %s\n",lowmass4->GetName());
  

  AliDielectron *lowmass5=ConfigAsakoLMEEPbPb2011AOD(5,hasMC,bESDANA);
  task->AddDielectron(lowmass5);
  printf("add: %s\n",lowmass5->GetName());
  
   AliDielectron *lowmass6=ConfigAsakoLMEEPbPb2011AOD(6,hasMC,bESDANA);
   task->AddDielectron(lowmass6);
   printf("add: %s\n",lowmass6->GetName());
  
   AliDielectron *lowmass7=ConfigAsakoLMEEPbPb2011AOD(7,hasMC,bESDANA);
   task->AddDielectron(lowmass7);
  printf("add: %s\n",lowmass7->GetName());
  
   AliDielectron *lowmass8=ConfigAsakoLMEEPbPb2011AOD(8,hasMC,bESDANA);
   task->AddDielectron(lowmass8);
   printf("add: %s\n",lowmass8->GetName());
  
  AliDielectron *lowmass9=ConfigAsakoLMEEPbPb2011AOD(9,hasMC,bESDANA);
  task->AddDielectron(lowmass9);
  printf("add: %s\n",lowmass9->GetName());

  AliDielectron *lowmass10=ConfigAsakoLMEEPbPb2011AOD(10,hasMC,bESDANA);
  task->AddDielectron(lowmass10);
  printf("add: %s\n",lowmass9->GetName());


AliDielectron *lowmass11=ConfigAsakoLMEEPbPb2011AOD(11,hasMC,bESDANA);
  task->AddDielectron(lowmass11);
  printf("add: %s\n",lowmass11->GetName());
  

  AliDielectron *lowmass12=ConfigAsakoLMEEPbPb2011AOD(12,hasMC,bESDANA);
  task->AddDielectron(lowmass12);
  printf("add: %s\n",lowmass12->GetName());
  

  AliDielectron *lowmass13=ConfigAsakoLMEEPbPb2011AOD(13,hasMC,bESDANA);
  task->AddDielectron(lowmass13);
  printf("add: %s\n",lowmass13->GetName());
  
  //if (PIDbaseline) {
  //AliDielectron *lowmass7=ConfigLMEEPbPb2011AOD(7,hasMC,bESDANA);
  //	task->AddDielectron(lowmass7);
  //	printf("add: %s\n",lowmass7->GetName());
  //}

  mgr->AddTask(task);

  //create output container
  AliAnalysisDataContainer *coutput1 =
	mgr->CreateContainer("asako_LMEEPbPb2011_tree",
		TTree::Class(),
		AliAnalysisManager::kExchangeContainer,
		"LMEEoutput.root");

  AliAnalysisDataContainer *cOutputHist1 =
	mgr->CreateContainer("asako_LMEEPbPb2011_out",
		TList::Class(),
		AliAnalysisManager::kOutputContainer,
		"LMEEoutput.root");

  /*  AliAnalysisDataContainer *cOutputHist2 =
	  mgr->CreateContainer("cbaumann_lowmass_CF",
	  TList::Class(),
	  AliAnalysisManager::kOutputContainer,
	  "cbaumann_lowmass_CF.root");
	  */
  AliAnalysisDataContainer *cOutputHist2 =
	mgr->CreateContainer("asako_LMEEPbPb2011_CF",
		TList::Class(),
		AliAnalysisManager::kOutputContainer,
		"LMEEoutput.root");

  AliAnalysisDataContainer *cOutputHist3 =
	mgr->CreateContainer("asako_EventStatPbPb2011",
		TH1D::Class(),
		AliAnalysisManager::kOutputContainer,
		"LMEEoutput.root");


  mgr->ConnectInput(task,  0, mgr->GetCommonInputContainer());
  mgr->ConnectOutput(task, 0, coutput1 );
  mgr->ConnectOutput(task, 1, cOutputHist1);
  mgr->ConnectOutput(task, 2, cOutputHist2);
  mgr->ConnectOutput(task, 3, cOutputHist3);

  return task;
}
Commit	Line	Data
a65a7e70	1	AliAnalysisTask *AddTask_Asako_LMEEPbPb2011AOD(Bool_t runAll=kFALSE,Bool_t setMC=kFALSE,Bool_t getFromAlien=kFALSE, Bool_t PIDbaseline=kFALSE){
	2
	3	Bool_t bESDANA=kFALSE; //Autodetect via InputHandler
	4	//get the current analysis manager
	5	AliAnalysisManager *mgr = AliAnalysisManager::GetAnalysisManager();
	6	if (!mgr) {
	7	Error("AddTask_Asako_LMEEPbPb2011AOD", "No analysis manager found.");
	8	return 0;
	9	}
	10
	11
	12	// create task and add it to the manager
	13	// gSystem->AddIncludePath("$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE");
	14
	15
	16	TString configBasePath("$TRAIN_ROOT/cbaumann_dielectron/");
	17	TString trainRoot=gSystem->Getenv("TRAIN_ROOT");
	18
	19	if (trainRoot.IsNull()) configBasePath= "$ALICE_ROOT/PWGDQ/dielectron/macrosLMEE/";
	20
	21
	22	if (getFromAlien &&
	23	(!gSystem->Exec("alien_cp alien:///alice/cern.ch/user/c/cbaumann/PWGDQ/dielectron/macrosLMEE/ConfigLMEEPbPb2011AOD.C .")) &&
	24	(!gSystem->Exec("alien_cp alien:///alice/cern.ch/user/c/cbaumann/PWGDQ/dielectron/macrosLMEE/LMEECutLibAOD.C ."))
	25	) {
	26	configBasePath=Form("%s/",gSystem->pwd());
	27	}
	28
	29	TString configFile("ConfigAsakoLMEEPbPb2011AOD.C");
	30	TString configLMEECutLib("LMEECutLibAsako.C");
	31
	32	TString configFilePath(configBasePath+configFile);
	33	TString configLMEECutLibPath(configBasePath+configLMEECutLib);
	34
	35	//AOD Usage currently tested with separate task, to be merged
	36	if (mgr->GetInputEventHandler()->IsA()==AliAODInputHandler::Class()){
	37	::Info("AddTaskLMEEPbPb2011", "no dedicated AOD configuration");
	38	}
	39	else if (mgr->GetInputEventHandler()->IsA()==AliESDInputHandler::Class()){
	40	::Info("AddTaskLMEEPbPb2011AOD","switching on ESD specific code");
	41	bESDANA=kTRUE;
	42	}
	43
	44
	45	//Do we have an MC handler?
	46	Bool_t hasMC=setMC;
	47	if (AliAnalysisManager::GetAnalysisManager()->GetMCtruthEventHandler()!=0x0)
	48	hasMC=kTRUE;
	49
	50
	51
	52	if (!gROOT->GetListOfGlobalFunctions()->FindObject(configLMEECutLib.Data()))
	53	gROOT->LoadMacro(configLMEECutLibPath.Data());
	54	if (!gROOT->GetListOfGlobalFunctions()->FindObject(configFile.Data()))
	55	gROOT->LoadMacro(configFilePath.Data());
	56
	57
	58	LMEECutLibAsako* cutlib = new LMEECutLibAsako();
	59	AliAnalysisTaskMultiDielectron *task=new AliAnalysisTaskMultiDielectron("MultiDiEData");
	60	if (!hasMC) task->UsePhysicsSelection();
	61	task->SetTriggerMask(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
	62	// task->SelectCollisionCandidates(AliVEvent::kMB+AliVEvent::kCentral+AliVEvent::kSemiCentral);
	63	// task->SetRejectPileup();
	64	task->SelectCollisionCandidates(AliVEvent::kAny);
65	task->SetEventFilter(cutlib->GetEventCuts(LMEECutLibAsako::kPbPb2011TPCandTOF)); //
66
67
68	//load dielectron configuration file
69
70	//add dielectron analysis with different cuts to the task
71
72
73	AliDielectron *lowmass1=ConfigAsakoLMEEPbPb2011AOD(1,hasMC,bESDANA);
74	task->AddDielectron(lowmass1);
75	printf("add: %s\n",lowmass1->GetName());
76
77
78	AliDielectron *lowmass2=ConfigAsakoLMEEPbPb2011AOD(2,hasMC,bESDANA);
79	task->AddDielectron(lowmass2);
80	printf("add: %s\n",lowmass2->GetName());
81
82
83	AliDielectron *lowmass3=ConfigAsakoLMEEPbPb2011AOD(3,hasMC,bESDANA);
84	task->AddDielectron(lowmass3);
85	printf("add: %s\n",lowmass3->GetName());
86
87
88	AliDielectron *lowmass4=ConfigAsakoLMEEPbPb2011AOD(4,hasMC,bESDANA);
89	task->AddDielectron(lowmass4);
90	printf("add: %s\n",lowmass4->GetName());
91
92
93
94	AliDielectron *lowmass5=ConfigAsakoLMEEPbPb2011AOD(5,hasMC,bESDANA);
95	task->AddDielectron(lowmass5);
96	printf("add: %s\n",lowmass5->GetName());
97
98	AliDielectron *lowmass6=ConfigAsakoLMEEPbPb2011AOD(6,hasMC,bESDANA);
99	task->AddDielectron(lowmass6);
100	printf("add: %s\n",lowmass6->GetName());
101
102	AliDielectron *lowmass7=ConfigAsakoLMEEPbPb2011AOD(7,hasMC,bESDANA);
103	task->AddDielectron(lowmass7);
104	printf("add: %s\n",lowmass7->GetName());
105
106	AliDielectron *lowmass8=ConfigAsakoLMEEPbPb2011AOD(8,hasMC,bESDANA);
107	task->AddDielectron(lowmass8);
108	printf("add: %s\n",lowmass8->GetName());
109
110	AliDielectron *lowmass9=ConfigAsakoLMEEPbPb2011AOD(9,hasMC,bESDANA);
111	task->AddDielectron(lowmass9);
112	printf("add: %s\n",lowmass9->GetName());
113
114	AliDielectron *lowmass10=ConfigAsakoLMEEPbPb2011AOD(10,hasMC,bESDANA);
115	task->AddDielectron(lowmass10);
116	printf("add: %s\n",lowmass9->GetName());
117
118
119	AliDielectron *lowmass11=ConfigAsakoLMEEPbPb2011AOD(11,hasMC,bESDANA);
120	task->AddDielectron(lowmass11);
121	printf("add: %s\n",lowmass11->GetName());
122
123
124	AliDielectron *lowmass12=ConfigAsakoLMEEPbPb2011AOD(12,hasMC,bESDANA);
125	task->AddDielectron(lowmass12);
126	printf("add: %s\n",lowmass12->GetName());
127
128
129	AliDielectron *lowmass13=ConfigAsakoLMEEPbPb2011AOD(13,hasMC,bESDANA);
130	task->AddDielectron(lowmass13);
131	printf("add: %s\n",lowmass13->GetName());
132
133	//if (PIDbaseline) {
134	//AliDielectron *lowmass7=ConfigLMEEPbPb2011AOD(7,hasMC,bESDANA);
135	// task->AddDielectron(lowmass7);
136	// printf("add: %s\n",lowmass7->GetName());
137	//}
138
139	mgr->AddTask(task);
140
141	//create output container
142	AliAnalysisDataContainer *coutput1 =
143	mgr->CreateContainer("asako_LMEEPbPb2011_tree",
144	TTree::Class(),
145	AliAnalysisManager::kExchangeContainer,
146	"LMEEoutput.root");
147
148	AliAnalysisDataContainer *cOutputHist1 =
149	mgr->CreateContainer("asako_LMEEPbPb2011_out",
150	TList::Class(),
151	AliAnalysisManager::kOutputContainer,
152	"LMEEoutput.root");
153
154	/* AliAnalysisDataContainer *cOutputHist2 =
155	mgr->CreateContainer("cbaumann_lowmass_CF",
156	TList::Class(),
157	AliAnalysisManager::kOutputContainer,
158	"cbaumann_lowmass_CF.root");
159	*/
160	AliAnalysisDataContainer *cOutputHist2 =
161	mgr->CreateContainer("asako_LMEEPbPb2011_CF",
162	TList::Class(),
163	AliAnalysisManager::kOutputContainer,
164	"LMEEoutput.root");
165
166	AliAnalysisDataContainer *cOutputHist3 =
167	mgr->CreateContainer("asako_EventStatPbPb2011",
168	TH1D::Class(),
169	AliAnalysisManager::kOutputContainer,
170	"LMEEoutput.root");
171
172
173	mgr->ConnectInput(task, 0, mgr->GetCommonInputContainer());
174	mgr->ConnectOutput(task, 0, coutput1 );
175	mgr->ConnectOutput(task, 1, cOutputHist1);
176	mgr->ConnectOutput(task, 2, cOutputHist2);
177	mgr->ConnectOutput(task, 3, cOutputHist3);
178
179	return task;
180	}