helpers.CloneTaskVisitor Class Reference

Inheritance diagram for helpers.CloneTaskVisitor:

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

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

Private Attributes
	_clonedTask
	_moduleLabels
	_noClones
	_postfix
	_process
	_removePostfix
	_verbose

Detailed Description

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

Definition at line 437 of file helpers.py.

Constructor & Destructor Documentation

__init__()

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

Definition at line 440 of file helpers.py.

    def __init__(self, process, label, postfix, removePostfix="", noClones = [], verbose = False):
        self._process = process
        self._postfix = postfix
        self._removePostfix = removePostfix
        self._noClones = noClones
        self._verbose = verbose
        self._moduleLabels = []
        self._clonedTask = cms.Task()
        setattr(process, self._newLabel(label), self._clonedTask)

Member Function Documentation

__appendToTopTask()

def helpers.CloneTaskVisitor.__appendToTopTask (   self,   visitee  )

private

Definition at line 481 of file helpers.py.

References helpers.CloneTaskVisitor._clonedTask, and PVValHelper.add().

Referenced by helpers.CloneTaskVisitor.enter().

    def __appendToTopTask(self, visitee):
        self._clonedTask.add(visitee)

_newLabel()

def helpers.CloneTaskVisitor._newLabel (   self,   label  )

private

Definition at line 473 of file helpers.py.

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

Referenced by 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

clonedTask()

def helpers.CloneTaskVisitor.clonedTask

(

self

)

Definition at line 467 of file helpers.py.

    def clonedTask(self):#FIXME: can the following be used for Task?
        for label in self._moduleLabels:
            massSearchReplaceAnyInputTag(self._clonedTask, label, self._newLabel(label), moduleLabelOnly=True, verbose=self._verbose)
        self._moduleLabels = [] # prevent the InputTag replacement next time the 'clonedTask' function is called.
        return self._clonedTask

enter()

def helpers.CloneTaskVisitor.enter	(		self,
			visitee
	)

Definition at line 450 of file helpers.py.

References helpers.CloneTaskVisitor.__appendToTopTask(), helpers.CloneSequenceVisitor._moduleLabels, helpers.CloneTaskVisitor._moduleLabels, helpers.CloneSequenceVisitor._newLabel(), helpers.CloneTaskVisitor._newLabel(), helpers.CloneSequenceVisitor._noClones, helpers.CloneTaskVisitor._noClones, helpers.CloneSequenceVisitor._process, helpers.CloneTaskVisitor._process, 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)
            self.__appendToTopTask(newModule)

leave()

def helpers.CloneTaskVisitor.leave	(		self,
			visitee
	)

Definition at line 464 of file helpers.py.

    def leave(self, visitee):
        pass

Member Data Documentation

_clonedTask

helpers.CloneTaskVisitor._clonedTask

private

Definition at line 447 of file helpers.py.

Referenced by helpers.CloneTaskVisitor.__appendToTopTask().

_moduleLabels

helpers.CloneTaskVisitor._moduleLabels

private

Definition at line 446 of file helpers.py.

Referenced by helpers.CloneTaskVisitor.enter().

_noClones

helpers.CloneTaskVisitor._noClones

private

Definition at line 444 of file helpers.py.

Referenced by helpers.CloneTaskVisitor.enter().

_postfix

helpers.CloneTaskVisitor._postfix

private

Definition at line 442 of file helpers.py.

Referenced by helpers.CloneTaskVisitor._newLabel().

_process

helpers.CloneTaskVisitor._process

private

Definition at line 441 of file helpers.py.

Referenced by helpers.CloneTaskVisitor.enter().

_removePostfix

helpers.CloneTaskVisitor._removePostfix

private

Definition at line 443 of file helpers.py.

Referenced by helpers.CloneTaskVisitor._newLabel().

_verbose

helpers.CloneTaskVisitor._verbose

private

Definition at line 445 of file helpers.py.