[u/mrichter/AliRoot.git] / PWGJE / EMCALJetTasks / macros / AddTaskChargedJetsPA.C

AliAnalysisTaskChargedJetsPA* AddTaskChargedJetsPA(
  Double_t            jetRadius               = 0.4,
  Int_t               trigger                 = AliVEvent::kINT7,
  Bool_t              isMC                    = kFALSE,
  Int_t               ptHardBin               = -1,
  Double_t            randomConeR             = 0.4,
  Double_t            trackBgrdConeR          = 0.6,
  const char*         containerSuffix         = "",
  const char*         usedTracks              = "PicoTracks",
  const char*         centralityType          = "V0A",
  Double_t            trackEtaWindow          = 0.9,
  Double_t            minJetPt                = 5.0, // signal jet min pt
  Double_t            minBackgroundJetPt      = 0.0, // background jet min pt
  Double_t            dijetLeadingMinPt       = 10.0,
  Double_t            dijetMaxAngleDev        = 10.0,
  Int_t               numberOfPtHardBins      = 0,
  const char*         externalMacro           = NULL,
  Bool_t              useVertexCut            = kTRUE,
  Bool_t              usePileUpCut            = kTRUE,
  Bool_t              isEMCalTrain            = kFALSE,
  Bool_t              calculateExternalRho    = kFALSE,
  Bool_t              analyzeDeprecatedBackgrounds = kTRUE
)
{
  // #### Detect the demanded trigger with its readable name
  TString triggerName(Form("Trigger_%i", trigger));
  if (trigger == AliVEvent::kAnyINT)
    triggerName = "kAnyINT";
  else if (trigger == AliVEvent::kAny)
    triggerName = "kAny";
  else if(trigger == AliVEvent::kINT7)
    triggerName = "kINT7";
  else if(trigger == AliVEvent::kMB)
    triggerName = "kMB";
  else if(trigger == AliVEvent::kEMC7)
    triggerName = "kEMC7";
  else if(trigger == AliVEvent::kEMCEJE)
    triggerName = "kEMCEJE";
  else if(trigger == AliVEvent::kEMCEGA)
    triggerName = "kEMCEGA";

  // #### Define manager and data container names
  AliAnalysisManager *manager = AliAnalysisManager::GetAnalysisManager();
  if (!manager) {
    ::Error("AddTaskChargedJetsPA", "No analysis manager to connect to.");
    return NULL;
  }

  TString stringPtHard("");
  TString containerNameSuffix("");

  if (ptHardBin!=-1)
    stringPtHard = Form("_PtHard_%d",ptHardBin);
  if (strcmp(containerSuffix,""))
    containerNameSuffix = Form("_%s", containerSuffix);

  TString myContName("");
  if(isMC)
    myContName = Form("AnalysisR0%2.0f_%s_MC%s%s", jetRadius*100, triggerName.Data(), stringPtHard.Data(), containerNameSuffix.Data());
  else
    myContName = Form("AnalysisR0%2.0f_%s%s%s", jetRadius*100, triggerName.Data(), stringPtHard.Data(), containerNameSuffix.Data());

  // #### Add necessary jet finder tasks
  gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJet.C");
  AliEmcalJetTask* jetFinderTask = AddTaskEmcalJet(usedTracks,"",1,jetRadius,1,0.150,0.300); // anti-kt
  AliEmcalJetTask* jetFinderTaskKT = AddTaskEmcalJet(usedTracks,"",0,jetRadius,1,0.150,0.300); // kt
  
  if(jetRadius < 0.1)
  {
    jetFinderTask->SetMinJetArea(0.0);
    jetFinderTaskKT->SetMinJetArea(0.0);
    jetFinderTask->SetMinJetPt(0.15);
    jetFinderTaskKT->SetMinJetPt(0.15);
    jetFinderTask->SetGhostArea(0.001);
    jetFinderTaskKT->SetGhostArea(0.001);
  }

  if(minBackgroundJetPt == -1.0)
  {
    if(analyzeDeprecatedBackgrounds)
      minBackgroundJetPt = 0.0;
    else
      minBackgroundJetPt = 0.15;
  }

  jetFinderTaskKT->SetMinJetPt(minBackgroundJetPt);

  // #### Define extern rho task
  if(calculateExternalRho)
  {
    TString myRhoName("");
    if(isMC)
      myRhoName = Form("RhoR0%2.0f_%s_MC%s",jetRadius*100,triggerName.Data(), stringPtHard.Data());
    else
      myRhoName = Form("RhoR0%2.0f_%s%s",jetRadius*100,triggerName.Data(), stringPtHard.Data());

    contRhoHistos = manager->CreateContainer(myRhoName.Data(), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s:ChargedJetsPA", AliAnalysisManager::GetCommonFileName()));
    AliAnalysisTaskRhoSparse *rhotask = new AliAnalysisTaskRhoSparse(myRhoName.Data(), kTRUE);
    rhotask->SetAnaType(AliAnalysisTaskEmcal::kTPC);
    rhotask->SetJetsName(jetFinderTaskKT->GetName());
    rhotask->SetSigJetsName(jetFinderTask->GetName());
    rhotask->SetTracksName(usedTracks);
    rhotask->SetRhoName(myRhoName.Data());
    rhotask->SetJetAreaCut(0);
    rhotask->SetJetPtCut(0.150);
    rhotask->SetJetRadius(jetRadius);
    rhotask->SetRhoCMS(kTRUE);
    rhotask->SelectCollisionCandidates(trigger);
    manager->AddTask(rhotask);
    manager->ConnectInput(rhotask, 0, manager->GetCommonInputContainer());
    manager->ConnectOutput(rhotask, 1, contRhoHistos);

  }

  // #### Define analysis task
  AliAnalysisTaskChargedJetsPA *task = NULL;
  contHistos = manager->CreateContainer(myContName.Data(), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s:ChargedJetsPA", AliAnalysisManager::GetCommonFileName()));
  task = new AliAnalysisTaskChargedJetsPA(Form("AnalysisPA_%s_%s", jetFinderTask->GetName(), triggerName.Data()), usedTracks, jetFinderTask->GetName(),jetFinderTaskKT->GetName());

  // #### Task preferences
  task->SetAcceptanceWindows(trackEtaWindow, jetRadius, jetRadius);
  task->SetAnalyzeQA(kTRUE);
  task->SetAnalyzeBackground(kTRUE);
  task->SetAnalyzeDeprecatedBackgrounds(analyzeDeprecatedBackgrounds);
  task->SetUsePileUpCut(usePileUpCut);
  task->SetUseVertexCut(useVertexCut);
  task->SetSignalJetMinPt(minJetPt);
  task->SetSignalJetMinArea(0.6*jetRadius*jetRadius*TMath::Pi());
  task->SetDijetLeadingMinPt(dijetLeadingMinPt);
  task->SetDijetMaxAngleDeviation(dijetMaxAngleDev);
  task->SetRandConeRadius(randomConeR);
  task->SetBackgroundJetMinPt(minBackgroundJetPt);
  task->SetTRBackgroundConeRadius(trackBgrdConeR);
  task->SelectCollisionCandidates(trigger);
  task->SetCentralityType(centralityType);
  task->SetUsePtHardBin(ptHardBin);
  if(calculateExternalRho)
    task->SetExternalRhoTaskName(myRhoName.Data());

  if(numberOfPtHardBins)
    task->SetNumberOfPtHardBins(numberOfPtHardBins);

  // #### Add analysis task
  manager->AddTask(task);
  manager->ConnectInput(task, 0, manager->GetCommonInputContainer());
  manager->ConnectOutput(task, 1, contHistos);

  if(isEMCalTrain)
    RequestMemory(task,200*1024);

  // #### Do some nasty piggybacking on demand
  if (externalMacro)
    gROOT->Macro(externalMacro);

  return task;
}
Commit	Line	Data
8628b70c	1	AliAnalysisTaskChargedJetsPA* AddTaskChargedJetsPA(
	2	Double_t jetRadius = 0.4,
	3	Int_t trigger = AliVEvent::kINT7,
	4	Bool_t isMC = kFALSE,
b95d0245	5	Int_t ptHardBin = -1,
8628b70c	6	Double_t randomConeR = 0.4,
8628b70c	7	Double_t trackBgrdConeR = 0.6,
c2b0976c	8	const char* containerSuffix = "",
8628b70c	9	const char* usedTracks = "PicoTracks",
12eb4ac1	10	const char* centralityType = "V0A",
8628b70c	11	Double_t trackEtaWindow = 0.9,
8628b70c	12	Double_t minJetPt = 5.0, // signal jet min pt
c154ba90	13	Double_t minBackgroundJetPt = 0.0, // background jet min pt
8628b70c	14	Double_t dijetLeadingMinPt = 10.0,
8628b70c	15	Double_t dijetMaxAngleDev = 10.0,
12eb4ac1	16	Int_t numberOfPtHardBins = 0,
b95d0245	17	const char* externalMacro = NULL,
06b41974	18	Bool_t useVertexCut = kTRUE,
0cb49d99	19	Bool_t usePileUpCut = kTRUE,
06b41974	20	Bool_t isEMCalTrain = kFALSE,
0cb49d99	21	Bool_t calculateExternalRho = kFALSE,
bf8f2a1c	22	Bool_t analyzeDeprecatedBackgrounds = kTRUE
8628b70c	23	)
	24	{
	25	// #### Detect the demanded trigger with its readable name
	26	TString triggerName(Form("Trigger_%i", trigger));
	27	if (trigger == AliVEvent::kAnyINT)
	28	triggerName = "kAnyINT";
	29	else if (trigger == AliVEvent::kAny)
	30	triggerName = "kAny";
	31	else if(trigger == AliVEvent::kINT7)
	32	triggerName = "kINT7";
efb9b161	33	else if(trigger == AliVEvent::kMB)
efb9b161	34	triggerName = "kMB";
8628b70c	35	else if(trigger == AliVEvent::kEMC7)
	36	triggerName = "kEMC7";
	37	else if(trigger == AliVEvent::kEMCEJE)
	38	triggerName = "kEMCEJE";
	39	else if(trigger == AliVEvent::kEMCEGA)
	40	triggerName = "kEMCEGA";
	41
	42	// #### Define manager and data container names
8628b70c	43	AliAnalysisManager *manager = AliAnalysisManager::GetAnalysisManager();
	44	if (!manager) {
	45	::Error("AddTaskChargedJetsPA", "No analysis manager to connect to.");
	46	return NULL;
	47	}
b95d0245	48
b95d0245	49	TString stringPtHard("");
c2b0976c	50	TString containerNameSuffix("");
c2b0976c	51
b95d0245	52	if (ptHardBin!=-1)
b95d0245	53	stringPtHard = Form("_PtHard_%d",ptHardBin);
c2b0976c	54	if (strcmp(containerSuffix,""))
	55	containerNameSuffix = Form("_%s", containerSuffix);
	56
8628b70c	57	TString myContName("");
8628b70c	58	if(isMC)
c2b0976c	59	myContName = Form("AnalysisR0%2.0f_%s_MC%s%s", jetRadius*100, triggerName.Data(), stringPtHard.Data(), containerNameSuffix.Data());
8628b70c	60	else
c2b0976c	61	myContName = Form("AnalysisR0%2.0f_%s%s%s", jetRadius*100, triggerName.Data(), stringPtHard.Data(), containerNameSuffix.Data());
8628b70c	62
	63	// #### Add necessary jet finder tasks
	64	gROOT->LoadMacro("$ALICE_ROOT/PWGJE/EMCALJetTasks/macros/AddTaskEmcalJet.C");
0983dd5b	65	AliEmcalJetTask* jetFinderTask = AddTaskEmcalJet(usedTracks,"",1,jetRadius,1,0.150,0.300); // anti-kt
8628b70c	66	AliEmcalJetTask* jetFinderTaskKT = AddTaskEmcalJet(usedTracks,"",0,jetRadius,1,0.150,0.300); // kt
c2b0976c	67
	68	if(jetRadius < 0.1)
	69	{
	70	jetFinderTask->SetMinJetArea(0.0);
	71	jetFinderTaskKT->SetMinJetArea(0.0);
	72	jetFinderTask->SetMinJetPt(0.15);
	73	jetFinderTaskKT->SetMinJetPt(0.15);
	74	jetFinderTask->SetGhostArea(0.001);
	75	jetFinderTaskKT->SetGhostArea(0.001);
	76	}
8628b70c	77
c2b0976c	78	if(minBackgroundJetPt == -1.0)
0cb49d99	79	{
	80	if(analyzeDeprecatedBackgrounds)
	81	minBackgroundJetPt = 0.0;
	82	else
	83	minBackgroundJetPt = 0.15;
	84	}
	85
b95d0245	86	jetFinderTaskKT->SetMinJetPt(minBackgroundJetPt);
06b41974	87
	88	// #### Define extern rho task
	89	if(calculateExternalRho)
	90	{
	91	TString myRhoName("");
	92	if(isMC)
	93	myRhoName = Form("RhoR0%2.0f_%s_MC%s",jetRadius*100,triggerName.Data(), stringPtHard.Data());
	94	else
	95	myRhoName = Form("RhoR0%2.0f_%s%s",jetRadius*100,triggerName.Data(), stringPtHard.Data());
	96
	97	contRhoHistos = manager->CreateContainer(myRhoName.Data(), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s:ChargedJetsPA", AliAnalysisManager::GetCommonFileName()));
	98	AliAnalysisTaskRhoSparse *rhotask = new AliAnalysisTaskRhoSparse(myRhoName.Data(), kTRUE);
	99	rhotask->SetAnaType(AliAnalysisTaskEmcal::kTPC);
	100	rhotask->SetJetsName(jetFinderTaskKT->GetName());
	101	rhotask->SetSigJetsName(jetFinderTask->GetName());
	102	rhotask->SetTracksName(usedTracks);
	103	rhotask->SetRhoName(myRhoName.Data());
	104	rhotask->SetJetAreaCut(0);
	105	rhotask->SetJetPtCut(0.150);
	106	rhotask->SetJetRadius(jetRadius);
	107	rhotask->SetRhoCMS(kTRUE);
	108	rhotask->SelectCollisionCandidates(trigger);
	109	manager->AddTask(rhotask);
	110	manager->ConnectInput(rhotask, 0, manager->GetCommonInputContainer());
	111	manager->ConnectOutput(rhotask, 1, contRhoHistos);
	112
	113	}
	114
8628b70c	115	// #### Define analysis task
8628b70c	116	AliAnalysisTaskChargedJetsPA *task = NULL;
7cfa86a3	117	contHistos = manager->CreateContainer(myContName.Data(), TList::Class(), AliAnalysisManager::kOutputContainer, Form("%s:ChargedJetsPA", AliAnalysisManager::GetCommonFileName()));
12eb4ac1	118	task = new AliAnalysisTaskChargedJetsPA(Form("AnalysisPA_%s_%s", jetFinderTask->GetName(), triggerName.Data()), usedTracks, jetFinderTask->GetName(),jetFinderTaskKT->GetName());
8628b70c	119
8628b70c	120	// #### Task preferences
06b41974	121	task->SetAcceptanceWindows(trackEtaWindow, jetRadius, jetRadius);
b95d0245	122	task->SetAnalyzeQA(kTRUE);
b95d0245	123	task->SetAnalyzeBackground(kTRUE);
0cb49d99	124	task->SetAnalyzeDeprecatedBackgrounds(analyzeDeprecatedBackgrounds);
06b41974	125	task->SetUsePileUpCut(usePileUpCut);
06b41974	126	task->SetUseVertexCut(useVertexCut);
8628b70c	127	task->SetSignalJetMinPt(minJetPt);
	128	task->SetSignalJetMinArea(0.6jetRadiusjetRadius*TMath::Pi());
	129	task->SetDijetLeadingMinPt(dijetLeadingMinPt);
	130	task->SetDijetMaxAngleDeviation(dijetMaxAngleDev);
	131	task->SetRandConeRadius(randomConeR);
b95d0245	132	task->SetBackgroundJetMinPt(minBackgroundJetPt);
12eb4ac1	133	task->SetTRBackgroundConeRadius(trackBgrdConeR);
8628b70c	134	task->SelectCollisionCandidates(trigger);
12eb4ac1	135	task->SetCentralityType(centralityType);
b95d0245	136	task->SetUsePtHardBin(ptHardBin);
06b41974	137	if(calculateExternalRho)
06b41974	138	task->SetExternalRhoTaskName(myRhoName.Data());
b95d0245	139
8628b70c	140	if(numberOfPtHardBins)
	141	task->SetNumberOfPtHardBins(numberOfPtHardBins);
	142
	143	// #### Add analysis task
	144	manager->AddTask(task);
	145	manager->ConnectInput(task, 0, manager->GetCommonInputContainer());
	146	manager->ConnectOutput(task, 1, contHistos);
12eb4ac1	147
b95d0245	148	if(isEMCalTrain)
	149	RequestMemory(task,200*1024);
	150
12eb4ac1	151	// #### Do some nasty piggybacking on demand
12eb4ac1	152	if (externalMacro)
6bb6522e	153	gROOT->Macro(externalMacro);
12eb4ac1	154
8628b70c	155	return task;
8628b70c	156	}