printTrackingNtuple Namespace Reference

Functions
def	findEvent (ntpl, event)
def	main (opts)

Variables
	description
	help
	opts
	parser
	type

Function Documentation

def printTrackingNtuple.findEvent	(		ntpl,
			event
	)

Definition at line 9 of file printTrackingNtuple.py.

References createfilelist.int.

Referenced by main().

def findEvent(ntpl, event):
    eventId = event.split(":")
    if len(eventId) != 3:
        raise Exception("Malformed eventId %s, is not run:lumi:event" % eventId)
    eventId = (int(eventId[0]), int(eventId[1]), int(eventId[2]))

    for ev in ntpl:
        if ev.eventId() == eventId:
            return ev

    raise Exception("Did not find event %s from file %s" % (eventId, ntpl.file().GetPath()))

def printTrackingNtuple.main

(

opts

)

Definition at line 21 of file printTrackingNtuple.py.

References cmsPerfStripChart.dict, findEvent(), and edm.print().

def main(opts):
    if opts.track is None and opts.trackingParticle is None and opts.seed is None and opts.pixelHit is None and opts.stripHit is None:
        return

    ntpl = ntuple.TrackingNtuple(opts.file)

    if opts.entry is not None:
        event = ntpl.getEvent(opts.entry)
        print(event.eventIdStr())
    elif opts.event is not None:
        event = findEvent(ntpl, opts.event)
        print("Entry %d" % event.entry())

    hasHits = ntpl.hasHits()
    hasSeeds = ntpl.hasSeeds()

    if not hasSeeds and opts.seed is not None:
        print("Ntuple %s does not have seeds saved!" % opts.file)
        return
    if not hasHits and (opts.pixelHit is not None or opts.stripHit is not None):
        print("Ntuple %s does not have hits saved!" % opts.file)
        return

    seedArgs = dict(hits=hasHits, bestMatchingTrackingParticle=hasHits)
    trackArgs = dict(hits=hasHits, bestMatchingTrackingParticle=hasHits)
    tpArgs = dict(hits=hasHits, bestMatchingTrack=hasHits)
    if not hasSeeds:
        trackArgs["seedPrinter"] = None
        tpArgs["seedPrinter"] = None
    elif not hasHits:
        trackArgs["seedPrinter"] = ntuple.SeedPrinter(**seedArgs)
        tpArgs["seedPrinter"] = ntuple.SeedPrinter(**seedArgs)

    printSeed = ntuple.SeedPrinter(trackingParticles=True, trackingParticlePrinter=ntuple.TrackingParticlePrinter(**tpArgs), **seedArgs)
    printTrack = ntuple.TrackPrinter(trackingParticlePrinter=ntuple.TrackingParticlePrinter(**tpArgs), **trackArgs)
    printTrackingParticle = ntuple.TrackingParticlePrinter(trackPrinter=ntuple.TrackPrinter(**trackArgs), **tpArgs)

    if opts.track is not None:
        trks = event.tracks()
        if opts.track >= len(trks):
            print("You requested track %d, but this event has only %d tracks" % (opts.track, len(trks)))
            return
        trk = trks[opts.track]
        printTrack(trk)

    if opts.trackingParticle is not None:
        tps = event.trackingParticles()
        if opts.trackingParticle >= len(tps):
            print("You requested TrackingParticle %d, but this event has ony %d TrackingParticles" % (opts.trackingParticle, len(tps)))
            return
        tp = tps[opts.trackingParticle]
        printTrackingParticle(tp)

    if opts.seed is not None:
        seeds = event.seeds()
        if opts.seedIteration is not None:
            algo = getattr(ntuple.Algo, opts.seedIteration)
            if opts.seed >= seeds.nSeedsForAlgo(algo):
                print("You requested seed %d for algo %s, but this event has only %d seeds for that algo" % (opts.seed, opts.seedIteration, seeds.nSeedsForAlgo(algo)))
                return
            seed = seeds.seedForAlgo(algo, opts.seed)
        else:
            if opts.seed >= len(seeds):
                print("You requested seed %d, but this event has only %d seeds" % (opts.seed, len(seeds)))
                return
            seed = seeds[opts.seed]
        printSeed(seed)

    if opts.pixelHit is not None:
        hits = event.pixelHits()
        if opts.pixelHit >= len(hits):
            print("You requested pixel hit %d, but this event has only %d pixel hits" % (opts.pixelHit, len(hits)))
            return

        hit = hits[opts.pixelHit]
        print("Pixel hit %d tracks" % opts.pixelHit)
        for t in hit.tracks():
            printTrack(t)
        if hasSeeds:
            print("Pixel hit %d seeds" % opts.pixelHit)
            for s in hit.seeds():
                printSeed(s)

    if opts.stripHit is not None:
        hits = event.stripHits()
        if opts.stripHit >= len(hits):
            print("You requested strip hit %d, but this event has only %d strip hits" % (opts.stripHit, len(hits)))
            return
        hit = hits[opts.stripHit]
        print("Strip hit %d tracks" % opts.stripHit)
        for t in hit.tracks():
            printTrack(t)
        if hasSeeds:
            print("Strip hit %d seeds" % opts.stripHit)
            for s in hit.seeds():
                printSeed(s)

Variable Documentation

printTrackingNtuple.description

Definition at line 120 of file printTrackingNtuple.py.

printTrackingNtuple.help

Definition at line 122 of file printTrackingNtuple.py.

printTrackingNtuple.opts

Definition at line 142 of file printTrackingNtuple.py.

printTrackingNtuple.parser

Definition at line 120 of file printTrackingNtuple.py.

printTrackingNtuple.type

Definition at line 121 of file printTrackingNtuple.py.