helpers.CloneSequenceVisitor Class Reference

Inheritance diagram for helpers.CloneSequenceVisitor:

Public Member Functions
def	__init__ (self, process, label, postfix, removePostfix="", noClones=[], addToTask=False)
def	clonedSequence (self)
def	enter (self, visitee)
def	leave (self, visitee)

Private Member Functions
def	__appendToTopSequence (self, visitee)
def	_newLabel (self, label)

Private Attributes
	_addToTask
	_clonedSequence
	_moduleLabels
	_noClones
	_patAlgosToolsTask
	_postfix
	_process
	_removePostfix

Detailed Description

Visitor that travels within a cms.Sequence, and returns a cloned version of the Sequence.
All modules and sequences are cloned and a postfix is added

Definition at line 154 of file helpers.py.

Constructor & Destructor Documentation

def helpers.CloneSequenceVisitor.__init__	(		self,
			process,
			label,
			postfix,
			removePostfix = "",
			noClones = [],
			addToTask = False
	)

Definition at line 157 of file helpers.py.

    def __init__(self, process, label, postfix, removePostfix="", noClones = [], addToTask = False):
        self._process = process
        self._postfix = postfix
        self._removePostfix = removePostfix
        self._noClones = noClones
        self._addToTask = addToTask
        self._moduleLabels = []
        self._clonedSequence = cms.Sequence()
        setattr(process, self._newLabel(label), self._clonedSequence)
        if addToTask:
            self._patAlgosToolsTask = getPatAlgosToolsTask(process)

Member Function Documentation

def helpers.CloneSequenceVisitor.__appendToTopSequence (   self,   visitee  )

private

Definition at line 202 of file helpers.py.

References helpers.CloneSequenceVisitor._clonedSequence.

Referenced by HiHelperTools.CloneSequenceVisitor.enter(), helpers.CloneSequenceVisitor.enter(), and HiHelperTools.CloneSequenceVisitor.leave().

    def __appendToTopSequence(self, visitee):
        self._clonedSequence += visitee

def helpers.CloneSequenceVisitor._newLabel (   self,   label  )

private

Definition at line 194 of file helpers.py.

References helpers.CloneSequenceVisitor._postfix, and helpers.CloneSequenceVisitor._removePostfix.

Referenced by helpers.CloneSequenceVisitor.enter().

    def _newLabel(self, label):
        if self._removePostfix != "":
            if label[-len(self._removePostfix):] == self._removePostfix:
                label = label[0:-len(self._removePostfix)]
            else:
                raise Exception("Tried to remove postfix %s from label %s, but it wasn't there" % (self._removePostfix, label))
        return label + self._postfix

def helpers.CloneSequenceVisitor.clonedSequence

(

self

)

Definition at line 188 of file helpers.py.

    def clonedSequence(self):
        for label in self._moduleLabels:
            massSearchReplaceAnyInputTag(self._clonedSequence, label, self._newLabel(label), moduleLabelOnly=True, verbose=False)
        self._moduleLabels = [] # prevent the InputTag replacement next time the 'clonedSequence' function is called.
        return self._clonedSequence

def helpers.CloneSequenceVisitor.enter	(		self,
			visitee
	)

Definition at line 169 of file helpers.py.

References helpers.CloneSequenceVisitor.__appendToTopSequence(), helpers.CloneSequenceVisitor._addToTask, helpers.CloneSequenceVisitor._moduleLabels, helpers.CloneSequenceVisitor._newLabel(), helpers.CloneSequenceVisitor._noClones, and helpers.CloneSequenceVisitor._process.

    def enter(self, visitee):
        if isinstance(visitee, cms._Module):
            label = visitee.label()
            newModule = None
            if label in self._noClones: #keep unchanged
                newModule = getattr(self._process, label)
            elif label in self._moduleLabels: # has the module already been cloned ?
                newModule = getattr(self._process, self._newLabel(label))
            else:
                self._moduleLabels.append(label)
                newModule = visitee.clone()
                setattr(self._process, self._newLabel(label), newModule)
                if self._addToTask:
                    self._patAlgosToolsTask.add(getattr(self._process, self._newLabel(label)))
            self.__appendToTopSequence(newModule)

def helpers.CloneSequenceVisitor.leave	(		self,
			visitee
	)

Definition at line 185 of file helpers.py.

    def leave(self, visitee):
        pass

Member Data Documentation

helpers.CloneSequenceVisitor._addToTask

private

Definition at line 162 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.enter().

helpers.CloneSequenceVisitor._clonedSequence

private

Definition at line 164 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.__appendToTopSequence(), and helpers.CloneSequenceVisitor.clonedSequence().

helpers.CloneSequenceVisitor._moduleLabels

private

Definition at line 163 of file helpers.py.

Referenced by HiHelperTools.CloneSequenceVisitor.clonedSequence(), helpers.CloneSequenceVisitor.clonedSequence(), HiHelperTools.CloneSequenceVisitor.enter(), and helpers.CloneSequenceVisitor.enter().

helpers.CloneSequenceVisitor._noClones

private

Definition at line 161 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.enter().

helpers.CloneSequenceVisitor._patAlgosToolsTask

private

Definition at line 167 of file helpers.py.

helpers.CloneSequenceVisitor._postfix

private

Definition at line 159 of file helpers.py.

Referenced by HiHelperTools.CloneSequenceVisitor.__appendToTopSequence(), helpers.CloneSequenceVisitor._newLabel(), HiHelperTools.CloneSequenceVisitor.clonedSequence(), and HiHelperTools.CloneSequenceVisitor.enter().

helpers.CloneSequenceVisitor._process

private

Definition at line 158 of file helpers.py.

Referenced by HiHelperTools.CloneSequenceVisitor.__appendToTopSequence(), HiHelperTools.CloneSequenceVisitor.enter(), and helpers.CloneSequenceVisitor.enter().

helpers.CloneSequenceVisitor._removePostfix

private

Definition at line 160 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor._newLabel().