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 8 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 20 of file printTrackingNtuple.py.

References cmsPerfStripChart.dict, and findEvent().

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:
        trk = event.tracks()[opts.track]
        printTrack(trk)

    if opts.trackingParticle is not None:
        tp = event.trackingParticles()[opts.trackingParticle]
        printTrackingParticle(tp)

    if opts.seed is not None:
        seeds = event.seeds()
        if opts.seedIteration is not None:
            seed = seeds.seedForAlgo(getattr(ntuple.Algo, opts.seedIteration), opts.seed)
        else:
            seed = seeds[opts.seed]
        printSeed(seed)

    if opts.pixelHit is not None:
        hit = event.pixelHits()[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 t in hit.seeds():
                printSeed(s)

    if opts.stripHit is not None:
        hit = event.stripHits()[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 t in hit.seeds():
                printSeed(s)

Variable Documentation

printTrackingNtuple.description

Definition at line 95 of file printTrackingNtuple.py.

printTrackingNtuple.help

Definition at line 97 of file printTrackingNtuple.py.

printTrackingNtuple.opts

Definition at line 117 of file printTrackingNtuple.py.

printTrackingNtuple.parser

Definition at line 95 of file printTrackingNtuple.py.

printTrackingNtuple.type

Definition at line 96 of file printTrackingNtuple.py.