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 9 of file diffTrackingNtuple.py.

References 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 24 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 70 of file diffTrackingNtuple.py.

References inOrder(), and outOfOrder().

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 45 of file diffTrackingNtuple.py.

References body().

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 90 of file diffTrackingNtuple.py.

diffTrackingNtuple.default

Definition at line 94 of file diffTrackingNtuple.py.

diffTrackingNtuple.description

Definition at line 85 of file diffTrackingNtuple.py.

diffTrackingNtuple.help

Definition at line 87 of file diffTrackingNtuple.py.

diffTrackingNtuple.int

Definition at line 94 of file diffTrackingNtuple.py.

diffTrackingNtuple.opts

Definition at line 101 of file diffTrackingNtuple.py.

diffTrackingNtuple.parser

Definition at line 85 of file diffTrackingNtuple.py.

diffTrackingNtuple.type

Definition at line 86 of file diffTrackingNtuple.py.