run_xAOD_ElectronPhotonFourMomentumCorrection Namespace Reference

Functions
	get_electron_collection (tree)
	get_photon_collection (tree)
	xAOD_particle_generator (tree, collection_getter, newevent_function=None, endevent_function=None, event_numbers=None, min_pt=None, min_abseta=None, max_abseta=None)
	main (filename, **args)

Variables
	level
	INFO
	format
	IgnoreCommandLineOptions
	xAOD_photon_generator = partial(xAOD_particle_generator, collection_getter=get_photon_collection)
	xAOD_electron_generator = partial(xAOD_particle_generator, collection_getter=get_electron_collection)
	parser
	type
	str
	help
	int
	default
	float
	action
	args = parser.parse_args()

Function Documentation

get_electron_collection()

run_xAOD_ElectronPhotonFourMomentumCorrection.get_electron_collection

(

tree

)

Definition at line 16 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

def get_electron_collection(tree):
    try:
        collection = tree.Electrons
    except AttributeError:
        collection = tree.ElectronCollection
    return collection
 
 

get_photon_collection()

run_xAOD_ElectronPhotonFourMomentumCorrection.get_photon_collection

(

tree

)

Definition at line 24 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

def get_photon_collection(tree):
    try:
        collection = tree.Photons
    except AttributeError:
        collection = tree.PhotonCollection
    return collection
 
 

main()

run_xAOD_ElectronPhotonFourMomentumCorrection.main	(		filename,
		**	args )

Definition at line 71 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

def main(filename, **args):
    logging.debug("initializing xAOD")
    if (not ROOT.xAOD.Init().isSuccess()):
        raise IOError("Failed xAOD.Init()")
 
 
    tree = None
    if ".txt" in filename:
        logging.debug("filename is a list of files")
        chain = ROOT.TChain(args['tree_name'])
        for line in islice(open(filename), 10):
            logging.debug("adding %s", line.strip())
            chain.Add(line.strip())
        tree = ROOT.xAOD.MakeTransientTree(chain)
    else:
        logging.debug("opening file %s", filename)
        f = ROOT.TFile.Open(filename)
        if not f:
            logging.error("problem opening file %s", filename)
        tree = ROOT.xAOD.MakeTransientTree(f, args['tree_name'])
    if not tree:
        logging.warning("cannot find tree in the file")
        f.Print()
        return
 
    logging.info("input has %d entries", tree.GetEntries())
 
    logging.debug("initializing tool")
    tool = ROOT.CP.EgammaCalibrationAndSmearingTool("tool")
    tool.setProperty("ESModel", args["esmodel"]).ignore()
    tool.setProperty("decorrelationModel", args["decorrelation"]).ignore()
    if args["no_smearing"]:
        tool.setProperty("int")("doSmearing", 0).ignore()
    if args['debug']:
        tool.msg().setLevel(0)
    if args['use_afii'] is not None:
        tool.setProperty("bool")("useFastSim", bool(args['use_afii'])).ignore()
 
    tool.initialize()
 
    if args['variation'] is not None:
        logging.info("applying systematic variation %s", args['variation'])
        sys_set = ROOT.CP.SystematicSet()
        sys_set.insert(ROOT.CP.SystematicVariation(args['variation'], args['variation_sigma']))
        tool.applySystematicVariation(sys_set)
 
    logging.debug("creating output tree")
    fout = ROOT.TFile("output.root", "recreate")
    tree_out = ROOT.TNtuple(args["tree_name"], args["tree_name"], "eventNumber:eta:phi:true_e:pdgId:e:xAOD_e:raw_e:raw_ps")
 
    logging.debug("looping on electron container")
 
    def newevent_function():
        logging.debug("executing new event function")
        tool.beginInputFile()
        tool.beginEvent()
 
    def endevent_function():
        logging.debug("executing end event function")
        tool.endInputFile()
 
    generator = xAOD_photon_generator(tree, newevent_function = newevent_function,
                                      endevent_function = endevent_function,
                                      min_pt=args['min_pt'],
                                      min_abseta=args['min_abseta'], max_abseta=args['max_abseta'],
                                      event_numbers=args['event_number'])
 
    _ = ROOT.xAOD.TruthParticle_v1    # this is needed to run the next line, don't know why, dictionary stuff...
    get_truth_particle = ROOT.xAOD.TruthHelpers.getTruthParticle
 
    for particle in islice(generator, None, args['nparticles']):
        ei = tree.EventInfo
        logging.debug(" === new photon eventNumber|eta|phi|e| = |%d|%.2f|%.2f|%.2f|", ei.eventNumber(), particle.eta(), particle.phi(), particle.e())
        if not particle.caloCluster():
            logging.warning("particle has no calo cluster")
            continue
        logging.debug("                |rawe0|raweAcc|TileGap3| = |%.2f|%.2f|%.2f|",
                      particle.caloCluster().energyBE(0),
                      particle.caloCluster().energyBE(1) + particle.caloCluster().energyBE(2) + particle.caloCluster().energyBE(3),
                      particle.caloCluster().eSample(17))
        xAOD_energy = particle.e()
        cluster = particle.caloCluster()
        raw_e = cluster.energyBE(1) + cluster.energyBE(2) + cluster.energyBE(3)
        raw_ps = cluster.energyBE(0)
        true_particle = get_truth_particle(particle)
        true_e = true_particle.e() if true_particle else 0
        pdgId = true_particle.pdgId() if true_particle else 0
        calibrated_energy = tool.getEnergy(particle)
        tree_out.Fill(ei.eventNumber(), cluster.eta(), cluster.phi(),
                      true_e, pdgId, calibrated_energy, xAOD_energy, raw_e, raw_ps)
        if math.isnan(calibrated_energy) or math.isnan(calibrated_energy) or calibrated_energy < 1:
            print("==>", particle.author(), particle.eta(), particle.phi(), xAOD_energy, calibrated_energy)
 
 
    logging.info("%d events written", tree_out.GetEntries())
 
    tree_out.Write()
    fout.Close()
 

xAOD_particle_generator()

run_xAOD_ElectronPhotonFourMomentumCorrection.xAOD_particle_generator	(		tree,
			collection_getter,
			newevent_function = None,
			endevent_function = None,
			event_numbers = None,
			min_pt = None,
			min_abseta = None,
			max_abseta = None )

Definition at line 32 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

                            min_pt=None, min_abseta=None, max_abseta=None):
    if event_numbers is None:
        event_numbers = []
    elif type(event_numbers) is int:
        event_numbers = [event_numbers]
 
    for ievent in range(tree.GetEntries()):
        tree.GetEntry(ievent)
        ei = tree.EventInfo
        event_number = ei.eventNumber()
        if event_numbers:
            if event_number not in event_numbers:
                continue
        logging.debug("=== event number %d ievent = %d", event_number, ievent)
        if newevent_function is not None:
            newevent_function()
 
        collection = collection_getter(tree)
 
        for i in range(collection.size()):
            p = collection.at(i)
            if min_pt is not None and p.pt() < min_pt:
                continue
            if min_abseta is not None and abs(p.eta()) < min_abseta:
                continue
            if max_abseta is not None and abs(p.eta()) > max_abseta:
                continue
            yield p
 
        if endevent_function is not None:
            endevent_function()
 
 

Variable Documentation

action

run_xAOD_ElectronPhotonFourMomentumCorrection.action

Definition at line 183 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

args

run_xAOD_ElectronPhotonFourMomentumCorrection.args = parser.parse_args()

Definition at line 192 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

default

run_xAOD_ElectronPhotonFourMomentumCorrection.default

Definition at line 178 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

float

run_xAOD_ElectronPhotonFourMomentumCorrection.float

Definition at line 179 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

format

run_xAOD_ElectronPhotonFourMomentumCorrection.format

Definition at line 11 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

help

run_xAOD_ElectronPhotonFourMomentumCorrection.help

Definition at line 176 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

IgnoreCommandLineOptions

run_xAOD_ElectronPhotonFourMomentumCorrection.IgnoreCommandLineOptions

Definition at line 13 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

INFO

run_xAOD_ElectronPhotonFourMomentumCorrection.INFO

Definition at line 11 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

int

run_xAOD_ElectronPhotonFourMomentumCorrection.int

Definition at line 177 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

level

run_xAOD_ElectronPhotonFourMomentumCorrection.level

Definition at line 11 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

parser

run_xAOD_ElectronPhotonFourMomentumCorrection.parser

Initial value:

=  argparse.ArgumentParser(description='Run on xAOD and dump calibrated energy for electron and photons',
                                     formatter_class=argparse.RawTextHelpFormatter,
                                     epilog='example: ./run_xAOD_ElectronPhotonFourMomentumCorrection.py root://eosatlas.cern.ch//eos/atlas/user/t/turra/Hgamgam_20.7/mc15_13TeV:AOD.07922786._000014.pool.root.1')

Definition at line 173 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

str

run_xAOD_ElectronPhotonFourMomentumCorrection.str

Definition at line 176 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

type

run_xAOD_ElectronPhotonFourMomentumCorrection.type

Definition at line 176 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

xAOD_electron_generator

run_xAOD_ElectronPhotonFourMomentumCorrection.xAOD_electron_generator = partial(xAOD_particle_generator, collection_getter=get_electron_collection)

Definition at line 68 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.

xAOD_photon_generator

run_xAOD_ElectronPhotonFourMomentumCorrection.xAOD_photon_generator = partial(xAOD_particle_generator, collection_getter=get_photon_collection)

Definition at line 67 of file run_xAOD_ElectronPhotonFourMomentumCorrection.py.