diffTrackingNtuple Namespace Reference

Functions
def	body (opts, ev1, ev2, printTrack)
def	inOrder (opts, ntpl1, ntpl2, args, kwargs)
def	main (opts)
def	outOfOrder (opts, ntpl1, ntpl2, args, kwargs)

Variables
	action
	default
	description
	help
	int
	opts
	parser
	type

Function Documentation

def diffTrackingNtuple.body	(		opts,
			ev1,
			ev2,
			printTrack
	)

Definition at line 10 of file diffTrackingNtuple.py.

References edm.print(), and str.

Referenced by inOrder(), and outOfOrder().

def body(opts, ev1, ev2, printTrack):
    print(ev1.eventIdStr())

    tracks1 = ev1.tracks()
    tracks2 = ev2.tracks()

    singleTrack = (opts.track is not None)
    if singleTrack:
        tracks1 = [tracks1[opts.track]]

    diff = ntuple.diffTrackListsGeneric(printTrack, tracks1, tracks2, ignoreAdditionalLst2=singleTrack)
    if diff.hasDifference():
        print(str(diff))
        print()

def diffTrackingNtuple.inOrder	(		opts,
			ntpl1,
			ntpl2,
			args,
			kwargs
	)

Definition at line 25 of file diffTrackingNtuple.py.

References body().

Referenced by main().

def inOrder(opts, ntpl1, ntpl2, *args, **kwargs):
    if opts.entry is not None:
        ev1 = ntpl1.getEvent(opts.entry)
        ev2 = ntpl2.getEvent(opts.entry)

        if ev1.eventId() != ev2.eventId():
            raise Exception("Events are out of order, entry %d file1 has %s and file %s. Use --outOfOrder option instead." % (ev1.entry(), ev1.eventIdStr(), ev2.eventIdStr()))

        body(opts, ev1, ev2, *args, **kwargs)
        return

    for i, (ev1, ev2) in enumerate(itertools.izip(ntpl1, ntpl2)):
        if opts.maxEvents >= 0 and i >= opts.maxEvents:
            break

        if ev1.eventId() != ev2.eventId():
            raise Exception("Events are out of order, entry %d file1 has %s and file %s. Use --outOfOrder option instead." % (ev1.entry(), ev1.eventIdStr(), ev2.eventIdStr()))

        body(opts, ev1, ev2, *args, **kwargs)

def diffTrackingNtuple.main

(

opts

)

Definition at line 71 of file diffTrackingNtuple.py.

References inOrder(), outOfOrder(), and edm.print().

def main(opts):
    ntpl1 = ntuple.TrackingNtuple(opts.file1)
    ntpl2 = ntuple.TrackingNtuple(opts.file2)

    print("--- %s" % opts.file1)
    print("+++ %s" % opts.file2)

    printTrack = ntuple.TrackPrinter(trackingParticlePrinter=ntuple.TrackingParticlePrinter(parentage=False), diffForTwiki=opts.twiki)

    if opts.outOfOrder:
        outOfOrder(opts, ntpl1, ntpl2, printTrack)
    else:
        inOrder(opts, ntpl1, ntpl2, printTrack)

def diffTrackingNtuple.outOfOrder	(		opts,
			ntpl1,
			ntpl2,
			args,
			kwargs
	)

Definition at line 46 of file diffTrackingNtuple.py.

References body(), and edm.print().

Referenced by main().

def outOfOrder(opts, ntpl1, ntpl2, *args, **kwargs):
    if opts.entry is not None:
        raise Exception("--entry does not make sense with --outOfOrder")

    events2 = collections.OrderedDict()
    for ev2 in ntpl2:
        events2[ev2.eventIdStr()] = ev2.entry()

    for i, ev1 in enumerate(ntpl1):
        if opts.maxEvents >= 0 and i >= opts.maxEvents:
            break

        if not ev1.eventId() in events2:
            print("-", ev1.eventIdStr())
            continue

        ev2 = ntpl2.getEvent(events2[ev1.eventIdStr()])
        events2.remove(ev1.eventId())

        body(opts, ev1, ev2, *args, **kwargs)


    for eventIdStr in events2.iterkeys():
        print("+", eventIdStr)

Variable Documentation

diffTrackingNtuple.action

Definition at line 91 of file diffTrackingNtuple.py.

diffTrackingNtuple.default

Definition at line 95 of file diffTrackingNtuple.py.

diffTrackingNtuple.description

Definition at line 86 of file diffTrackingNtuple.py.

diffTrackingNtuple.help

Definition at line 88 of file diffTrackingNtuple.py.

diffTrackingNtuple.int

Definition at line 95 of file diffTrackingNtuple.py.

diffTrackingNtuple.opts

Definition at line 102 of file diffTrackingNtuple.py.

diffTrackingNtuple.parser

Definition at line 86 of file diffTrackingNtuple.py.

diffTrackingNtuple.type

Definition at line 87 of file diffTrackingNtuple.py.