trackingPlots.TrackingIterationOrder Class Reference

Public Member Functions
def	__call__ (self, tdirectory, labels)
def	__init__ (self)

Private Member Functions
def	_findOrder (self, f)

Private Attributes
	_cache

Detailed Description

Definition at line 1761 of file trackingPlots.py.

Constructor & Destructor Documentation

__init__()

def trackingPlots.TrackingIterationOrder.__init__

(

self

)

Definition at line 1762 of file trackingPlots.py.

    def __init__(self):
        self._cache = {}

Member Function Documentation

__call__()

def trackingPlots.TrackingIterationOrder.__call__	(		self,
			tdirectory,
			labels
	)

Definition at line 1776 of file trackingPlots.py.

References DTCache< T, Coll >._cache, tkal_create_file_lists.FileListCreator._cache, L1MuDTTrackFinder._cache, datamodel.Collection._cache, L1MuBMTrackFinder._cache, DTTrig._cache, trackingPlots.TimePerEventPlot._cache, trackingPlots.TrackingIterationOrder._cache, trackingPlots.TrackingIterationOrder._findOrder(), and isotrackApplyRegressor.range.

    def __call__(self, tdirectory, labels):
        ret = list(range(0, len(labels)))
        f = tdirectory.GetFile()
        if not f:
            return ret

        if not f.GetName() in self._cache:
            r = self._findOrder(f)
            if r is None:
                return ret
            self._cache[f.GetName()] = r
        order = self._cache[f.GetName()]

        # O(N^2) I know, but we're talking about O(10) elements...
        orderIndices = []
        for l in order:
            try:
                orderIndices.append(labels.index(l))
            except ValueError:
                pass
        ret = []
        for i, l in enumerate(labels):
            if l in order:
                try:
                    found = orderIndices.index(i)
                    if found == 0:
                        ret.append(i)
                    else:
                        ret.append(orderIndices[0])
                except ValueError:
                    ret.append(orderIndices[0])
                orderIndices.pop(0)
            else:
                ret.append(i)
        return ret

_findOrder()

def trackingPlots.TrackingIterationOrder._findOrder (   self,   f  )

private

Definition at line 1765 of file trackingPlots.py.

References isotrackApplyRegressor.range, and python.rootplot.root2matplotlib.replace().

Referenced by trackingPlots.TrackingIterationOrder.__call__().

    def _findOrder(self, f):
        h = f.Get(_trackingIterationOrderHistogram)
        if not h:
            return None
        xaxis = h.GetXaxis()
        def _edit(s):
            # remove "Tracks" from the track producer name to get the iteration name
            # muonSeeded iterations do not have "Step" in the producer name, so add it here
            return s.replace("Tracks", "").replace("muonSeeded", "muonSeededStep")
        return [_edit(xaxis.GetBinLabel(i)) for i in range(1, h.GetNbinsX()+1)]

Member Data Documentation

_cache

trackingPlots.TrackingIterationOrder._cache

private

Definition at line 1763 of file trackingPlots.py.

Referenced by trackingPlots.TrackingIterationOrder.__call__().