collectionMerger.collectionMerger Class Reference

Inheritance diagram for collectionMerger.collectionMerger:

Public Member Functions
def	__init__ (self, input, output, sortkey=lambda x:x.pt, reverse=True, selector=None, maxObjects=None)
def	analyze (self, event)
def	beginFile (self, inputFile, outputFile, inputTree, wrappedOutputTree)
def	beginJob (self)
def	endFile (self, inputFile, outputFile, inputTree, wrappedOutputTree)
def	endJob (self)
def	filterBranchNames (self, branches, collection)

Public Attributes
	branchType
	brlist_all
	brlist_sep
	input
	is_there
	maxObjects
	nInputs
	out
	output
	reverse
	selector
	sortkey

Detailed Description

Definition at line 21 of file collectionMerger.py.

Constructor & Destructor Documentation

__init__()

def collectionMerger.collectionMerger.__init__	(		self,
			input,
			output,
			sortkey = lambda x: x.pt,
			reverse = True,
			selector = None,
			maxObjects = None
	)

Definition at line 28 of file collectionMerger.py.

                 maxObjects=None):
        self.input = input
        self.output = output
        self.nInputs = len(self.input)
        self.sortkey = lambda obj_j_i1: sortkey(obj_j_i1[0])
        self.reverse = reverse
        # pass dict([(collection_name,lambda obj : selection(obj)])
        self.selector = [(selector[coll] if coll in selector else
                          (lambda x: True))
                         for coll in self.input] if selector else None
        # save only the first maxObjects objects passing the selection in the merged collection
        self.maxObjects = maxObjects
        self.branchType = {}
        pass

Member Function Documentation

analyze()

def collectionMerger.collectionMerger.analyze	(		self,
			event
	)

process event, return True (go to next module) or False (fail, go to next event)

Definition at line 104 of file collectionMerger.py.

References collectionMerger.collectionMerger.brlist_all, event.Event.input, WorkFlow.WorkFlow.input, collectionMerger.collectionMerger.input, RBCInput.input, MuonStub.input, personalPlayback.Playback.input, options.HLTProcessOptions.input, collectionMerger.collectionMerger.is_there, collectionMerger.collectionMerger.maxObjects, collectionMerger.collectionMerger.nInputs, L1TMuonOverlapParams::LayerInputNode.nInputs, exampleModule.exampleProducer.out, mhtjuProducerCpp.mhtjuProducerCpp.out, ProcMultiply.out, collectionMerger.collectionMerger.out, PhysicsTools::Calibration::ProcMultiply.out, BeamHaloAnalyzer.out, l1ct::Event.out, CSCFileDumper.output, collectionMerger.collectionMerger.output, MessageLogger.output, GeometryComparisonPlotter.output, edm::service::MessageLoggerDefaults::Destination.output, personalPlayback.Playback.output, options.HLTProcessOptions.output, PhysicsTools::MVAComputer.output, PhysicsTools::Calibration::MVAComputer.output, PhysicsTools::VarProcessor::ValueIterator.output, isotrackApplyRegressor.range, collectionMerger.collectionMerger.reverse, hgcalTriggerChains.HGCalTriggerChains.selector, RawDataSelector.selector, collectionMerger.collectionMerger.selector, BPHMultiSelectBase< T >::SelectElement.selector, BPHRecoBuilder::BPHRecoSource.selector, and collectionMerger.collectionMerger.sortkey.

    def analyze(self, event):
        """process event, return True (go to next module) or False (fail, go to next event)"""
        coll = [Collection(event, x) for x in self.input]
        objects = [(coll[j][i], j, i) for j in range(self.nInputs)
                   for i in range(len(coll[j]))]
        if self.selector:
            objects = [
                obj_j_i for obj_j_i in objects
                if self.selector[obj_j_i[1]](obj_j_i[0])
            ]
        objects.sort(key=self.sortkey, reverse=self.reverse)
        if self.maxObjects:
            objects = objects[:self.maxObjects]
        for bridx, br in enumerate(self.brlist_all):
            out = []
            for obj, j, i in objects:
                out.append(getattr(obj, br) if self.is_there[bridx][j] else 0)
            self.out.fillBranch("%s_%s" % (self.output, br), out)
        return True


# define modules using the syntax 'name = lambda : constructor' to avoid having them loaded when not needed

beginFile()

def collectionMerger.collectionMerger.beginFile	(		self,
			inputFile,
			outputFile,
			inputTree,
			wrappedOutputTree
	)

Definition at line 49 of file collectionMerger.py.

    def beginFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):

        # Find list of activated branches in input tree
        _brlist_in = inputTree.GetListOfBranches()
        branches_in = set(
            [_brlist_in.At(i) for i in range(_brlist_in.GetEntries())])
        branches_in = [
            x for x in branches_in if inputTree.GetBranchStatus(x.GetName())
        ]

        # Find list of activated branches in output tree
        _brlist_out = wrappedOutputTree._tree.GetListOfBranches()
        branches_out = set(
            [_brlist_out.At(i) for i in range(_brlist_out.GetEntries())])
        branches_out = [
            x for x in branches_out
            if wrappedOutputTree._tree.GetBranchStatus(x.GetName())
        ]

        # Use both
        branches = branches_in + branches_out

        # Only keep branches with right collection name
        self.brlist_sep = [
            self.filterBranchNames(branches, x) for x in self.input
        ]
        self.brlist_all = set(itertools.chain(*(self.brlist_sep)))

        self.is_there = np.zeros(shape=(len(self.brlist_all), self.nInputs),
                                 dtype=bool)
        for bridx, br in enumerate(self.brlist_all):
            for j in range(self.nInputs):
                if br in self.brlist_sep[j]:
                    self.is_there[bridx][j] = True

        # Create output branches
        self.out = wrappedOutputTree
        for br in self.brlist_all:
            self.out.branch("%s_%s" % (self.output, br),
                            _rootLeafType2rootBranchType[self.branchType[br]],
                            lenVar="n%s" % self.output)

beginJob()

def collectionMerger.collectionMerger.beginJob

(

self

)

Definition at line 43 of file collectionMerger.py.

    def beginJob(self):
        pass

endFile()

def collectionMerger.collectionMerger.endFile	(		self,
			inputFile,
			outputFile,
			inputTree,
			wrappedOutputTree
	)

Definition at line 91 of file collectionMerger.py.

    def endFile(self, inputFile, outputFile, inputTree, wrappedOutputTree):
        pass

endJob()

def collectionMerger.collectionMerger.endJob

(

self

)

Definition at line 46 of file collectionMerger.py.

    def endJob(self):
        pass

filterBranchNames()

def collectionMerger.collectionMerger.filterBranchNames	(		self,
			branches,
			collection
	)

Definition at line 94 of file collectionMerger.py.

    def filterBranchNames(self, branches, collection):
        out = []
        for br in branches:
            name = br.GetName()
            if not name.startswith(collection + '_'):
                continue
            out.append(name.replace(collection + '_', ''))
            self.branchType[out[-1]] = br.FindLeaf(br.GetName()).GetTypeName()
        return out

Member Data Documentation

branchType

collectionMerger.collectionMerger.branchType

Definition at line 40 of file collectionMerger.py.

brlist_all

collectionMerger.collectionMerger.brlist_all

Definition at line 75 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

brlist_sep

collectionMerger.collectionMerger.brlist_sep

Definition at line 72 of file collectionMerger.py.

input

collectionMerger.collectionMerger.input

Definition at line 29 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

is_there

collectionMerger.collectionMerger.is_there

Definition at line 77 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

maxObjects

collectionMerger.collectionMerger.maxObjects

Definition at line 39 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

nInputs

collectionMerger.collectionMerger.nInputs

Definition at line 31 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

out

collectionMerger.collectionMerger.out

Definition at line 85 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

output

collectionMerger.collectionMerger.output

Definition at line 30 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

reverse

collectionMerger.collectionMerger.reverse

Definition at line 33 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

selector

collectionMerger.collectionMerger.selector

Definition at line 35 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().

sortkey

collectionMerger.collectionMerger.sortkey

Definition at line 32 of file collectionMerger.py.

Referenced by collectionMerger.collectionMerger.analyze().