def main(configFileName, nFiles, nEvents, debugLevel=0):
f = open(configFileName, 'r')
config = yaml.load(f)
f.close()
physSel = ExtractTriggerSelection(config["trigger"])
print("Trigger selection is {}".format(physSel))
ROOT.gSystem.Load("libCGAL")
ROOT.AliTrackContainer.SetDefTrackCutsPeriod(config["run_period"])
mode = None
if config["mode"] == "AOD":
mode = helperFunctions.AnaMode.AOD
elif config["mode"] == "ESD":
mode = helperFunctions.AnaMode.ESD
else:
print "Error: mode has to be either ESD or AOD!"
exit(1)
print("{0} analysis chosen.".format(config["mode"]))
print(
"Setting local analysis for {0} files from list {1} max events = {2}".
format(nFiles, config["file_list"], nEvents))
# Analysis manager
mgr = ROOT.AliAnalysisManager("TrackingQA")
helperFunctions.LoadMacros()
if mode is helperFunctions.AnaMode.AOD:
helperFunctions.AddAODHandler()
elif mode is helperFunctions.AnaMode.ESD:
helperFunctions.AddESDHandler()
if config["MC"]:
helperFunctions.AddMCHandler()
if mode is helperFunctions.AnaMode.ESD:
task = helperFunctions.AddTaskCDBConnect()
# task.SetFallBackToRaw(True)
# Physics selection task
if not config["MC"]: ROOT.AddTaskPhysicsSelection()
if mode is helperFunctions.AnaMode.ESD and config["MC"]:
ROOT.AddTaskMCTrackSelector("mcparticles", False, False, -1, False)
if config["cent_type"] == "new":
ROOT.AddTaskMultSelection(False)
elif config["cent_type"] == "old":
print("Not configured for old centrality framework!")
return
if config["MC"]:
print "Running on a MC production"
else:
print "Running on data"
trackQAtask = ROOT.AliEmcalTrackingQATask.AddTaskTrackingQA(config["MC"])
trackQAtask.SelectCollisionCandidates(physSel)
tasks = mgr.GetTasks()
for task in tasks:
if isinstance(task, ROOT.AliAnalysisTaskEmcalLight):
task.SetIsPythia(config["is_pythia"])
if config["is_pythia"]: task.SetMaxMinimumBiasPtHard(5)
if config["run_period"] == "LHC15o":
task.SetSwitchOffLHC15oFaultyBranches(True)
if config["beam_type"] == "pp":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcalLight.kpp)
elif config["beam_type"] == "PbPb":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcalLight.kAA)
elif config["beam_type"] == "pPb":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcalLight.kpA)
if config["cent_type"] == "new":
task.SetCentralityEstimation(
ROOT.AliAnalysisTaskEmcalLight.kNewCentrality)
elif config["cent_type"] == "old":
task.SetCentralityEstimation(
ROOT.AliAnalysisTaskEmcalLight.kOldCentrality)
else:
task.SetCentralityEstimation(
ROOT.AliAnalysisTaskEmcalLight.kNoCentrality)
if isinstance(task, ROOT.AliAnalysisTaskEmcal):
if config["beam_type"] == "pp":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcal.kpp)
elif config["beam_type"] == "PbPb":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcal.kAA)
task.SetUseNewCentralityEstimation(True)
res = mgr.InitAnalysis()
if not res:
print "Error initializing the analysis!"
exit(1)
mgr.PrintStatus()
outFile = ROOT.TFile("train.root", "RECREATE")
outFile.cd()
mgr.Write()
outFile.Close()
chain = None
if mode is helperFunctions.AnaMode.AOD:
ROOT.gROOT.LoadMacro(
"$ALICE_PHYSICS/PWG/EMCAL/macros/CreateAODChain.C")
chain = ROOT.CreateAODChain(config["file_list"], nFiles, 0, False)
elif mode is helperFunctions.AnaMode.ESD:
ROOT.gROOT.LoadMacro(
"$ALICE_PHYSICS/PWG/EMCAL/macros/CreateESDChain.C")
chain = ROOT.CreateESDChain(config["file_list"], nFiles, 0, False)
if debugLevel == 0:
mgr.SetUseProgressBar(1, 250)
else:
# To have more debug info
mgr.AddClassDebug("AliEmcalTrackingQATask", ROOT.AliLog.kDebug + 100)
mgr.AddClassDebug("AliEmcalContainer", ROOT.AliLog.kDebug + 100)
mgr.AddClassDebug("AliMCParticleContainer", ROOT.AliLog.kDebug + 100)
mgr.AddClassDebug("AliEmcalMCTrackSelector", ROOT.AliLog.kDebug + 100)
mgr.SetDebugLevel(debugLevel)
# start analysis
print "Starting Analysis..."
mgr.StartAnalysis("local", chain, nEvents)
def main(configFileName, nFiles, nEvents, d2h, doRecLevel, doSignalOnly, doMCTruth, doWrongPID, doResponse, noInclusiveJets, efficiency, selectGen, taskName="JetDmesonAna", debugLevel=0):
f = open(configFileName, 'r')
config = yaml.load(f)
f.close()
physSel = ExtractTriggerSelection(config["trigger"])
print("Trigger selection is {}".format(physSel))
ROOT.gSystem.Load("libCGAL")
ROOT.AliTrackContainer.SetDefTrackCutsPeriod(config["run_period"])
mode = None
if config["mode"] == "AOD":
mode = helperFunctions.AnaMode.AOD
elif config["mode"] == "ESD":
mode = helperFunctions.AnaMode.ESD
else:
print "Error: mode has to be either ESD or AOD!"
exit(1)
print("{0} analysis chosen.".format(config["mode"]))
print("Setting local analysis for {0} files from list {1} max events = {2}".format(nFiles, config["file_list"], nEvents))
# Analysis manager
mgr = ROOT.AliAnalysisManager(taskName)
helperFunctions.LoadMacros()
if mode is helperFunctions.AnaMode.AOD:
helperFunctions.AddAODHandler()
elif mode is helperFunctions.AnaMode.ESD:
helperFunctions.AddESDHandler()
# task = helperFunctions.AddTaskCDBConnect()
# task.SetFallBackToRaw(True)
# Physics selection task
if not config["MC"]:
ROOT.AddTaskPhysicsSelection()
if config["cent_type"] == "new":
ROOT.AddTaskMultSelection(False)
elif config["cent_type"] == "old":
print("Not configured for old centrality framework!")
return
if config["full_jets"]:
helperFunctions.PrepareEMCAL("userQAconfiguration.yaml")
if config["MC"]:
print "Running on a MC production"
else:
print "Running on data"
# PID response
helperFunctions.AddTaskPIDResponse(config["MC"], True, True, config["reco_pass"])
if config["full_jets"]:
pSpectraTask = ROOT.AddTaskEmcalJetQA("usedefault", "usedefault", "usedefault")
pSpectraTask.SetNeedEmcalGeom(True)
else:
pSpectraTask = ROOT.AddTaskEmcalJetQA("usedefault", "", "")
pSpectraTask.SetNeedEmcalGeom(False)
pSpectraTask.SelectCollisionCandidates(physSel)
pSpectraTask.SetPtBin(1, 150)
if config["beam_type"] == "PbPb": pSpectraTask.SetCentRange(0, 90)
if config["ue_sub"]:
pKtChJetTask = ROOT.AddTaskEmcalJet("usedefault", "", ROOT.AliJetContainer.kt_algorithm, 0.4, ROOT.AliJetContainer.kChargedJet, 0.15, 0., 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pKtChJetTask.SelectCollisionCandidates(physSel)
pChJetTask = ROOT.AddTaskEmcalJet("usedefault", "", ROOT.AliJetContainer.antikt_algorithm, 0.4, ROOT.AliJetContainer.kChargedJet, 0.15, 0., 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pChJetTask.SelectCollisionCandidates(physSel)
pRhoTask = ROOT.AliAnalysisTaskRhoDev.AddTaskRhoDev("usedefault", 0.15, "", 0.30, "Rho", 0.4, ROOT.AliEmcalJet.kTPCfid, ROOT.AliJetContainer.kChargedJet, True)
pRhoTask.SelectCollisionCandidates(physSel)
pRhoTask.SetVzRange(-10, 10)
pRhoTask.SetEventSelectionAfterRun(True)
jetCont = ROOT.AliJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.kt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, pRhoTask.GetParticleContainer("tracks"), 0)
jetCont.SetJetPtCut(1)
jetCont.SetJetAcceptanceType(ROOT.AliEmcalJet.kTPCfid)
jetCont.SetName("Signal")
pRhoTask.AdoptJetContainer(jetCont)
if config["beam_type"] == "pp":
pRhoTask.SetRhoSparse(True)
elif config["beam_type"] == "PbPb":
pRhoTask.SetRhoSparse(False)
elif config["beam_type"] == "pPb":
pRhoTask.SetRhoSparse(True)
pRhoExclLeadJetsTask = ROOT.AliAnalysisTaskRhoDev.AddTaskRhoDev("usedefault", 0.15, "", 0.30, "RhoExclLeadJets", 0.4, ROOT.AliEmcalJet.kTPCfid, ROOT.AliJetContainer.kChargedJet, True)
pRhoExclLeadJetsTask.SelectCollisionCandidates(physSel)
pRhoExclLeadJetsTask.SetVzRange(-10, 10)
pRhoExclLeadJetsTask.SetEventSelectionAfterRun(True)
pRhoExclLeadJetsTask.SetExcludeLeadJets(2)
jetCont = ROOT.AliJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.kt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, pRhoExclLeadJetsTask.GetParticleContainer("tracks"), 0)
jetCont.SetJetPtCut(1)
jetCont.SetJetAcceptanceType(ROOT.AliEmcalJet.kTPCfid)
jetCont.SetName("Signal")
pRhoExclLeadJetsTask.AdoptJetContainer(jetCont)
if config["beam_type"] == "pp":
pRhoExclLeadJetsTask.SetRhoSparse(True)
elif config["beam_type"] == "PbPb":
pRhoExclLeadJetsTask.SetRhoSparse(False)
elif config["beam_type"] == "pPb":
pRhoExclLeadJetsTask.SetRhoSparse(True)
pRhoTransTask = ROOT.AliAnalysisTaskRhoTransDev.AddTaskRhoTransDev("usedefault", 0.15, "", 0.30, "RhoTrans", 0.4, ROOT.AliEmcalJet.kTPCfid, ROOT.AliJetContainer.kChargedJet, True)
pRhoTransTask.SelectCollisionCandidates(physSel)
pRhoTransTask.SetVzRange(-10, 10)
pRhoTransTask.SetEventSelectionAfterRun(True)
pUEstudies = ROOT.AliAnalysisTaskJetUEStudies.AddTaskJetUEStudies("usedefault", "", 0.15, 0.30)
jetCont = pUEstudies.AddJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.antikt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, ROOT.AliJetContainer.kTPCfid, "tracks", "")
jetCont.SetRhoName("Rho")
pUEstudies.AddAltRho("RhoTrans")
pUEstudies.AddAltRho("RhoExclLeadJets")
if config["MC"]:
pGenKtChJetTask = ROOT.AddTaskEmcalJet("mcparticles", "", ROOT.AliJetContainer.kt_algorithm, 0.4, ROOT.AliJetContainer.kChargedJet, 0., 0., 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pGenKtChJetTask.SelectCollisionCandidates(physSel)
pGenChJetTask = ROOT.AddTaskEmcalJet("mcparticles", "", ROOT.AliJetContainer.antikt_algorithm, 0.4, ROOT.AliJetContainer.kChargedJet, 0., 0., 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pGenChJetTask.SelectCollisionCandidates(physSel)
pGenRhoTask = ROOT.AliAnalysisTaskRhoDev.AddTaskRhoDev("mcparticles", 0, "", 0, "RhoGen", 0.4, ROOT.AliEmcalJet.kTPCfid, ROOT.AliJetContainer.kChargedJet, True)
pGenRhoTask.SelectCollisionCandidates(physSel)
pGenRhoTask.SetVzRange(-10, 10)
pGenRhoTask.SetEventSelectionAfterRun(True)
jetCont = ROOT.AliJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.kt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, pGenRhoTask.GetParticleContainer("mcparticles"), 0)
jetCont.SetJetPtCut(1)
jetCont.SetJetAcceptanceType(ROOT.AliEmcalJet.kTPCfid)
jetCont.SetName("Signal")
pGenRhoTask.AdoptJetContainer(jetCont)
if config["beam_type"] == "pp":
pGenRhoTask.SetRhoSparse(True)
elif config["beam_type"] == "PbPb":
pGenRhoTask.SetRhoSparse(False)
elif config["beam_type"] == "pPb":
pGenRhoTask.SetRhoSparse(True)
pGenRhoExclLeadJetsTask = ROOT.AliAnalysisTaskRhoDev.AddTaskRhoDev("mcparticles", 0, "", 0, "RhoExclLeadJetsGen", 0.4, ROOT.AliEmcalJet.kTPCfid, ROOT.AliJetContainer.kChargedJet, True)
pGenRhoExclLeadJetsTask.SelectCollisionCandidates(physSel)
pGenRhoExclLeadJetsTask.SetVzRange(-10, 10)
pGenRhoExclLeadJetsTask.SetEventSelectionAfterRun(True)
pGenRhoExclLeadJetsTask.SetExcludeLeadJets(2)
jetCont = ROOT.AliJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.kt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, pGenRhoExclLeadJetsTask.GetParticleContainer("mcparticles"), 0)
jetCont.SetJetPtCut(1)
jetCont.SetJetAcceptanceType(ROOT.AliEmcalJet.kTPCfid)
jetCont.SetName("Signal")
pGenRhoExclLeadJetsTask.AdoptJetContainer(jetCont)
if config["beam_type"] == "pp":
pGenRhoExclLeadJetsTask.SetRhoSparse(True)
elif config["beam_type"] == "PbPb":
pGenRhoExclLeadJetsTask.SetRhoSparse(False)
elif config["beam_type"] == "pPb":
pGenRhoExclLeadJetsTask.SetRhoSparse(True)
pRhoGenTransTask = ROOT.AliAnalysisTaskRhoTransDev.AddTaskRhoTransDev("mcparticles", 0, "", 0, "RhoTransGen", 0.4, ROOT.AliEmcalJet.kTPCfid, ROOT.AliJetContainer.kChargedJet, True)
pRhoGenTransTask.SelectCollisionCandidates(physSel)
pRhoGenTransTask.SetVzRange(-10, 10)
pRhoGenTransTask.SetEventSelectionAfterRun(True)
pUEstudies = ROOT.AliAnalysisTaskJetUEStudies.AddTaskJetUEStudies("mcparticles", "", 0., 0., "gen")
jetCont = pUEstudies.AddJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.antikt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, ROOT.AliJetContainer.kTPCfid, "mcparticles", "")
jetCont.SetRhoName("RhoGen")
pUEstudies.AddAltRho("RhoTransGen")
pUEstudies.AddAltRho("RhoExclLeadJetsGen")
pUEstudies.SetMaxMomentumThridJet(15)
pUEstudies.SetBackToBackJetPtFraction(0.8)
if not noInclusiveJets:
# Charged jet analysis
if config["charged_jets"]:
pChJetTask = ROOT.AddTaskEmcalJet("usedefault", "", ROOT.AliJetContainer.antikt_algorithm, 0.4, ROOT.AliJetContainer.kChargedJet, 0.15, 0., 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pChJetTask.SelectCollisionCandidates(physSel)
# pChJetTask = ROOT.AddTaskEmcalJet("usedefault", "", 1, 0.6, ROOT.AliJetContainer.kChargedJet, 0.15, 0., 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
# pChJetTask.SelectCollisionCandidates(physSel)
if config["MC"]:
pGenChJetTask = ROOT.AddTaskEmcalJet("mcparticles", "", ROOT.AliJetContainer.antikt_algorithm, 0.4, ROOT.AliJetContainer.kChargedJet, 0., 0., 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pGenChJetTask.SelectCollisionCandidates(physSel)
# Full jet analysis
if config["full_jets"]:
pJetTask = ROOT.AddTaskEmcalJet("usedefault", "usedefault", ROOT.AliJetContainer.antikt_algorithm, 0.2, ROOT.AliJetContainer.kFullJet, 0.15, 0.30, 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pJetTask.SelectCollisionCandidates(physSel)
# pJetTask = ROOT.AddTaskEmcalJet("usedefault", "usedefault", 1, 0.4, ROOT.AliJetContainer.kFullJet, 0.15, 0.30, 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
# pJetTask.SelectCollisionCandidates(physSel)
if config["MC"]:
pGenJetTask = ROOT.AddTaskEmcalJet("mcparticles", "", ROOT.AliJetContainer.antikt_algorithm, 0.4, ROOT.AliJetContainer.kFullJet, 0., 0., 0.005, ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pGenJetTask.SelectCollisionCandidates(physSel)
ROOT.AliAnalysisManager.SetCommonFileName("AnalysisResults_jets.root")
if config["full_jets"]:
if config["ue_sub"]:
anaType = ROOT.AliAnalysisTaskEmcalJetTreeBase.kJetPbPb
rhoName = "Rho"
rhoGenName = "RhoGen"
else:
anaType = ROOT.AliAnalysisTaskEmcalJetTreeBase.kJetPP
rhoName = ""
rhoGenName = ""
pJetSpectraTask = ROOT.AliAnalysisTaskEmcalJetTreeBase.AddTaskEmcalJetTree("usedefault", "usedefault", 0.15, 0.30, anaType)
pJetSpectraTask.SetNeedEmcalGeom(True)
else:
if config["ue_sub"]:
anaType = ROOT.AliAnalysisTaskEmcalJetTreeBase.kJetPbPbCharged
rhoName = "Rho"
rhoGenName = "RhoGen"
else:
anaType = ROOT.AliAnalysisTaskEmcalJetTreeBase.kJetPPCharged
rhoName = ""
rhoGenName = ""
pJetSpectraTask = ROOT.AliAnalysisTaskEmcalJetTreeBase.AddTaskEmcalJetTree("usedefault", "", 0.15, 0.30, anaType)
pJetSpectraTask.SetNeedEmcalGeom(False)
ROOT.AliAnalysisManager.SetCommonFileName("AnalysisResults.root")
if config["MC"]:
partCont = pJetSpectraTask.AddParticleContainer("mcparticles")
partCont.SetMinPt(0)
pJetSpectraTask.SelectCollisionCandidates(physSel)
if config["beam_type"] == "PbPb": pJetSpectraTask.SetCentRange(0, 90)
if config["charged_jets"]:
jetCont = pJetSpectraTask.AddJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.antikt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, ROOT.AliJetContainer.kTPCfid, "tracks", "")
jetCont.SetRhoName(rhoName)
# jetCont = pJetSpectraTask.AddJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.antikt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.6, ROOT.AliJetContainer.kTPCfid)
# jetCont.SetRhoName(rhoName)
if config["MC"]:
jetCont = pJetSpectraTask.AddJetContainer(ROOT.AliJetContainer.kChargedJet, ROOT.AliJetContainer.antikt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, ROOT.AliJetContainer.kTPCfid, "mcparticles", "")
jetCont.SetRhoName(rhoGenName)
if config["full_jets"]:
jetCont = pJetSpectraTask.AddJetContainer(ROOT.AliJetContainer.kFullJet, ROOT.AliJetContainer.antikt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.2, ROOT.AliJetContainer.kEMCALfid, "tracks", "caloClusters")
jetCont.SetRhoName(rhoName)
# jetCont = pJetSpectraTask.AddJetContainer(ROOT.AliJetContainer.kFullJet, ROOT.AliJetContainer.antikt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, ROOT.AliJetContainer.kEMCALfid)
# jetCont.SetRhoName(rhoName)
if config["MC"]:
jetCont = pJetSpectraTask.AddJetContainer(ROOT.AliJetContainer.kFullJet, ROOT.AliJetContainer.antikt_algorithm, ROOT.AliJetContainer.pt_scheme, 0.4, ROOT.AliJetContainer.kTPCfid, "mcparticles", "")
jetCont.SetRhoName(rhoGenName)
# ROOT.AddTaskCleanupVertexingHF()
if d2h: ROOT.AddTaskD0Mass(0, config["MC"], False, False, 0, 0, 0, 0, "Loose", config["rdhf_cuts_dzero"])
if config["cent_bins"]:
for cmin, cnmax in zip(config["cent_bins"][:-1], config["cent_bins"][1:]):
pDMesonJetsTask = AddDMesonJetTask(mgr, config, doRecLevel, doSignalOnly, doMCTruth, doWrongPID, doResponse, cmin, cnmax)
pDMesonJetsTask.SelectCollisionCandidates(physSel)
pDMesonJetsTask.SetTrackEfficiency(efficiency)
else:
pDMesonJetsTask = AddDMesonJetTask(mgr, config, doRecLevel, doSignalOnly, doMCTruth, doWrongPID, doResponse, None, None)
pDMesonJetsTask.SelectCollisionCandidates(physSel)
pDMesonJetsTask.SetTrackEfficiency(efficiency)
tasks = mgr.GetTasks()
for task in tasks:
if isinstance(task, ROOT.AliAnalysisTaskEmcalLight):
task.SetIsPythia(config["is_pythia"])
if selectGen: pDMesonJetsTask.SelectGeneratorName(selectGen)
if config["is_pythia"]: task.SetMaxMinimumBiasPtHard(5)
if config["run_period"] == "LHC15o": task.SetSwitchOffLHC15oFaultyBranches(True)
if config["beam_type"] == "pp":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcalLight.kpp)
elif config["beam_type"] == "PbPb":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcalLight.kAA)
elif config["beam_type"] == "pPb":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcalLight.kpA)
if config["cent_type"] == "new":
task.SetCentralityEstimation(ROOT.AliAnalysisTaskEmcalLight.kNewCentrality)
elif config["cent_type"] == "old":
task.SetCentralityEstimation(ROOT.AliAnalysisTaskEmcalLight.kOldCentrality)
else:
task.SetCentralityEstimation(ROOT.AliAnalysisTaskEmcalLight.kNoCentrality)
if isinstance(task, ROOT.AliAnalysisTaskEmcal):
if config["beam_type"] == "pp":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcal.kpp)
elif config["beam_type"] == "PbPb":
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcal.kAA)
task.SetUseNewCentralityEstimation(True)
res = mgr.InitAnalysis()
if not res:
print "Error initializing the analysis!"
exit(1)
mgr.PrintStatus()
outFile = ROOT.TFile("train.root", "RECREATE")
outFile.cd()
mgr.Write()
outFile.Close()
chain = None
if mode is helperFunctions.AnaMode.AOD:
ROOT.gROOT.LoadMacro("$ALICE_PHYSICS/PWG/EMCAL/macros/CreateAODChain.C")
chain = ROOT.CreateAODChain(config["file_list"], nFiles, 0, False , "AliAOD.VertexingHF.root")
elif mode is helperFunctions.AnaMode.ESD:
ROOT.gROOT.LoadMacro("$ALICE_PHYSICS/PWG/EMCAL/macros/CreateESDChain.C");
chain = ROOT.CreateESDChain(config["file_list"], nFiles, 0, False)
if debugLevel == 0:
mgr.SetUseProgressBar(1, 250)
else:
# To have more debug info
mgr.AddClassDebug("AliAnalysisTaskDmesonJets", ROOT.AliLog.kDebug + 100)
mgr.AddClassDebug("AliAnalysisTaskDmesonJets::AnalysisEngine", ROOT.AliLog.kDebug + 100)
mgr.SetDebugLevel(debugLevel)
# start analysis
print "Starting Analysis..."
mgr.StartAnalysis("local", chain, nEvents)
#!/usr/bin/env python3
# Run Analysis script
import ROOT
# Include
ROOT.gInterpreter.ProcessLine(".include $ROOTSYS/include");
ROOT.gInterpreter.ProcessLine(".include $ALICE_ROOT/include");
ROOT.gInterpreter.ProcessLine(".include $ALICE_PHYSICS/include");
# Analysis Manager
mgr = ROOT.AliAnalysisManager("JpsiJetAnalysis");
# Input handler
aodH = ROOT.AliAODInputHandler();
mgr.SetInputEventHandler(aodH);
# Output handler
aodOutputH = ROOT.AliAODHandler();
aodOutputH.SetOutputFileName("AliAOD.root");
mgr.SetOutputEventHandler(aodOutputH);
ROOT.gInterpreter.ExecuteMacro("$ALICE_PHYSICS/OADB/macros/AddTaskPhysicsSelection.C");
ROOT.gInterpreter.ExecuteMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDResponse.C");
ROOT.gInterpreter.ExecuteMacro("$ALICE_ROOT/ANALYSIS/macros/AddTaskPIDqa.C");
# Start analysis
if( not mgr.InitAnalysis()):
exit()
mgr.PrintStatus();
mgr.SetUseProgressBar(1, 25);
def CreateAnalysisManager():
mgr = ROOT.AliAnalysisManager("MGR")
mgr.SetNSysInfo(1) # switch on syswatch
mgr.SetCommonFileName("AnalysisResults.root")
return mgr
def main(config):
ROOT.gSystem.Load("libCGAL")
ROOT.gInterpreter.ProcessLine(
os.path.expandvars(
'.L $ALICE_ROOT/include/AliEMCALTriggerConstants.h'))
physSel = 0
ROOT.AliTrackContainer.SetDefTrackCutsPeriod(config["run_period"])
mode = None
if config["mode"] == "AOD":
mode = helperFunctions.AnaMode.AOD
elif config["mode"] == "ESD":
mode = helperFunctions.AnaMode.ESD
else:
print "Error: mode has to be either ESD or AOD!"
exit(1)
print("{0} analysis chosen.".format(config["mode"]))
print(
"Setting local analysis for {0} files from list {1}, max events = {2}".
format(config["n_files"], config["file_list"], config["n_events"]))
# AliVEvent::kINT7, AliVEvent::kMB, AliVEvent::kCentral, AliVEvent::kSemiCentral
# AliVEvent::kEMCEGA, AliVEvent::kEMCEJ
# Analysis manager
mgr = ROOT.AliAnalysisManager(config["task_name"])
helperFunctions.LoadMacros()
if mode is helperFunctions.AnaMode.AOD:
helperFunctions.AddAODHandler()
elif mode is helperFunctions.AnaMode.ESD:
helperFunctions.AddESDHandler()
# CDB connect
pCDBConnect = ROOT.AliTaskCDBconnect("CDBconnect", "cvmfs://", 0)
mgr.AddTask(pCDBConnect)
cinput1 = mgr.GetCommonInputContainer()
mgr.ConnectInput(pCDBConnect, 0, cinput1)
pCDBConnect.SetFallBackToRaw(True)
# Physics selection task
if mode is helperFunctions.AnaMode.ESD and physSel:
ROOT.AddTaskPhysicsSelection()
if config["run_period"] == "LHC15o":
ROOT.AddTaskMultSelection(False)
# EMCal prep
if config["cluster_qa"] or config["trigger_qa"]:
helperFunctions.PrepareEMCAL("userQAconfiguration.yaml")
# helperFunctions.PrepareEMCAL_old(physSel, True, True, config["cluster_qa"] or config["full_jets"] or config["neutral_jets"], config["full_jets"] or config["neutral_jets"], config["full_jets"] or config["neutral_jets"])
# Trigger QA
if config["trigger_qa"]:
pTriggerMakerTask = ROOT.AddTaskEmcalTriggerMakerNew("EmcalTriggers")
pTriggerMakerTask.SelectCollisionCandidates(physSel)
pTriggerMakerTask.GetTriggerMaker().SetFastORandCellThresholds(0, 0, 0)
if config.has_key("bad_fastor"):
pTriggerMakerTask.GetTriggerMaker().ReadFastORBadChannelFromFile(
config["bad_fastor"])
if config["run_period"] == "LHC16q":
pTriggerMakerTask.GetTriggerMaker().ConfigureForPP2015()
elif config["run_period"] == "LHC16b" or config[
"run_period"] == "LHC16c" or config[
"run_period"] == "LHC16d" or config[
"run_period"] == "LHC16e" or config[
"run_period"] == "LHC16f":
pTriggerMakerTask.GetTriggerMaker().ConfigureForPP2015()
elif config["run_period"] == "LHC15o":
pTriggerMakerTask.GetTriggerMaker().ConfigureForPbPb2015()
elif config["run_period"] == "LHC15j":
pTriggerMakerTask.GetTriggerMaker().ConfigureForPP2015()
pTriggerMakerTask.SetUseL0Amplitudes(True)
elif config["run_period"].startswith("LHC13"):
pTriggerMakerTask.GetTriggerMaker().ConfigureForPPb2013()
elif config["run_period"].startswith("LHC12"):
pTriggerMakerTask.GetTriggerMaker().ConfigureForPP2012()
elif config["run_period"] == "LHC11h":
pTriggerMakerTask.GetTriggerMaker().ConfigureForPbPb2011()
elif config["run_period"].startswith(
"LHC11") and config["run_period"] != "LHC11h":
pTriggerMakerTask.GetTriggerMaker().ConfigureForPP2011()
# Charged jet analysis
if config["charged_jets"]:
pChJetTask = ROOT.AddTaskEmcalJet(
"usedefault", "", ROOT.AliJetContainer.antikt_algorithm, 0.4,
ROOT.AliJetContainer.kChargedJet, 0.15, 0., 0.1,
ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pChJetTask.SelectCollisionCandidates(physSel)
pChJetTask = ROOT.AddTaskEmcalJet(
"usedefault", "", ROOT.AliJetContainer.antikt_algorithm, 0.6,
ROOT.AliJetContainer.kChargedJet, 0.15, 0., 0.1,
ROOT.AliJetContainer.pt_scheme, "Jet", 0., False, False)
pChJetTask.SelectCollisionCandidates(physSel)
# Full jet analysis
if config["full_jets"]:
pJetTask = ROOT.AddTaskEmcalJet("usedefault", "usedefault",
ROOT.AliJetContainer.antikt_algorithm,
0.2, ROOT.AliJetContainer.kFullJet,
0.15, 0.30, 0.1,
ROOT.AliJetContainer.pt_scheme, "Jet",
0., False, False)
pJetTask.SelectCollisionCandidates(physSel)
pJetTask = ROOT.AddTaskEmcalJet("usedefault", "usedefault",
ROOT.AliJetContainer.antikt_algorithm,
0.4, ROOT.AliJetContainer.kFullJet,
0.15, 0.30, 0.1,
ROOT.AliJetContainer.pt_scheme, "Jet",
0., False, False)
pJetTask.SelectCollisionCandidates(physSel)
# Neutral jet analysis
if config["neutral_jets"]:
pJetTask = ROOT.AddTaskEmcalJet("", "usedefault",
ROOT.AliJetContainer.antikt_algorithm,
0.2, ROOT.AliJetContainer.kNeutralJet,
0.15, 0.30, 0.1,
ROOT.AliJetContainer.pt_scheme, "Jet",
0., False, False)
pJetTask.SelectCollisionCandidates(physSel)
pJetTask = ROOT.AddTaskEmcalJet("", "usedefault",
ROOT.AliJetContainer.antikt_algorithm,
0.4, ROOT.AliJetContainer.kNeutralJet,
0.15, 0.30, 0.1,
ROOT.AliJetContainer.pt_scheme, "Jet",
0., False, False)
pJetTask.SelectCollisionCandidates(physSel)
for trigger in config["triggers"]:
AddTriggerQATasks(config, trigger, physSel)
if config["run_period"] == "LHC11h" or config["run_period"].startswith(
"LHC13") or config["run_period"] == "LHC15o":
isPP = False
else:
isPP = True
tasks = mgr.GetTasks()
for task in tasks:
if isinstance(task, ROOT.AliAnalysisTaskEmcal) or isinstance(
task, ROOT.AliAnalysisTaskEmcalLight):
if isPP:
task.SetForceBeamType(ROOT.AliAnalysisTaskEmcal.kpp)
task.SetVzRange(-999, -999)
res = mgr.InitAnalysis()
if not res:
print "Error initializing the analysis!"
exit(1)
mgr.PrintStatus()
outFile = ROOT.TFile("train.root", "RECREATE")
outFile.cd()
mgr.Write()
outFile.Close()
chain = None
if mode is helperFunctions.AnaMode.AOD:
ROOT.gROOT.LoadMacro(
"$ALICE_PHYSICS/PWG/EMCAL/macros/CreateAODChain.C")
chain = ROOT.CreateAODChain(config["file_list"], config["n_files"], 0,
False)
elif mode is helperFunctions.AnaMode.ESD:
ROOT.gROOT.LoadMacro(
"$ALICE_PHYSICS/PWG/EMCAL/macros/CreateESDChain.C")
chain = ROOT.CreateESDChain(config["file_list"], config["n_files"], 0,
False)
if config["debug_level"] == 0:
mgr.SetUseProgressBar(1, 250)
else:
mgr.SetUseProgressBar(0, 0)
mgr.SetDebugLevel(config["debug_level"])
# To have more debug info
# pMgr->AddClassDebug("AliEmcalClusTrackMatcherTask", AliLog::kDebug+100);
# start analysis
print "Starting Analysis..."
mgr.StartAnalysis("local", chain, config["n_events"])
