helpers.CloneSequenceVisitor Class Reference

Inheritance diagram for helpers.CloneSequenceVisitor:

Public Member Functions
def	__init__ (self, process, label, postfix, removePostfix="", noClones=[], addToTask=False, verbose=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
	_verbose

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 155 of file helpers.py.

Constructor & Destructor Documentation

__init__()

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

Definition at line 158 of file helpers.py.

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

Member Function Documentation

__appendToTopSequence()

def helpers.CloneSequenceVisitor.__appendToTopSequence (   self,   visitee  )

private

Definition at line 204 of file helpers.py.

References helpers.CloneSequenceVisitor._clonedSequence.

Referenced by helpers.CloneSequenceVisitor.enter().

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

_newLabel()

def helpers.CloneSequenceVisitor._newLabel (   self,   label  )

private

Definition at line 196 of file helpers.py.

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

Referenced by helpers.CloneSequenceVisitor.enter(), and helpers.CloneTaskVisitor.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

clonedSequence()

def helpers.CloneSequenceVisitor.clonedSequence

(

self

)

Definition at line 190 of file helpers.py.

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

enter()

def helpers.CloneSequenceVisitor.enter	(		self,
			visitee
	)

Definition at line 171 of file helpers.py.

References helpers.CloneSequenceVisitor.__appendToTopSequence(), helpers.CloneSequenceVisitor._addToTask, helpers.CloneSequenceVisitor._moduleLabels, helpers.CloneSequenceVisitor._newLabel(), helpers.CloneSequenceVisitor._noClones, helpers.CloneSequenceVisitor._patAlgosToolsTask, helpers.CloneSequenceVisitor._process, PVValHelper.add(), and mps_setup.append.

    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)

leave()

def helpers.CloneSequenceVisitor.leave	(		self,
			visitee
	)

Definition at line 187 of file helpers.py.

    def leave(self, visitee):
        pass

Member Data Documentation

_addToTask

helpers.CloneSequenceVisitor._addToTask

private

Definition at line 163 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.enter().

_clonedSequence

helpers.CloneSequenceVisitor._clonedSequence

private

Definition at line 166 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.__appendToTopSequence().

_moduleLabels

helpers.CloneSequenceVisitor._moduleLabels

private

Definition at line 165 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.enter(), and helpers.CloneTaskVisitor.enter().

_noClones

helpers.CloneSequenceVisitor._noClones

private

Definition at line 162 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.enter(), and helpers.CloneTaskVisitor.enter().

_patAlgosToolsTask

helpers.CloneSequenceVisitor._patAlgosToolsTask

private

Definition at line 169 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.enter().

_postfix

helpers.CloneSequenceVisitor._postfix

private

Definition at line 160 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor._newLabel(), and helpers.CloneTaskVisitor._newLabel().

_process

helpers.CloneSequenceVisitor._process

private

Definition at line 159 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor.enter(), and helpers.CloneTaskVisitor.enter().

_removePostfix

helpers.CloneSequenceVisitor._removePostfix

private

Definition at line 161 of file helpers.py.

Referenced by helpers.CloneSequenceVisitor._newLabel(), and helpers.CloneTaskVisitor._newLabel().

_verbose

helpers.CloneSequenceVisitor._verbose

private

Definition at line 164 of file helpers.py.