Inheritance diagram for HiHelperTools.CloneSequenceVisitor:

Public Member Functions
def	__init__ (self, process, label, postfix)

def	clonedSequence (self)

def	enter (self, visitee)

def	leave (self, visitee)

Private Member Functions
def	__appendToTopSequence (self, visitee)

Private Attributes
	_moduleLabels

	_postfix

	_process

	_sequenceLabels

	_sequenceStack

	_waitForSequenceToClose

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 55 of file HiHelperTools.py.

Constructor & Destructor Documentation

def HiHelperTools.CloneSequenceVisitor.__init__	(	self,
		process,
		label,
		postfix
	)

Definition at line 58 of file HiHelperTools.py.

     def __init__(self, process, label, postfix):
         self._process = process
         self._postfix = postfix
         self._sequenceStack = [label]
         self._moduleLabels = []
         self._sequenceLabels = []
         self._waitForSequenceToClose = None # modules will only be cloned or added if this is None
 

Member Function Documentation

def HiHelperTools.CloneSequenceVisitor.__appendToTopSequence	(	self,
		visitee
	)

private

Definition at line 104 of file HiHelperTools.py.

References HiHelperTools.CloneSequenceVisitor._postfix, helpers.CloneSequenceVisitor._postfix, HiHelperTools.CloneSequenceVisitor._process, helpers.CloneSequenceVisitor._process, HiHelperTools.CloneSequenceVisitor._sequenceStack, and label.

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

     def __appendToTopSequence(self, visitee):#this is darn ugly because empty cms.Sequences are not supported
         if isinstance(self._sequenceStack[-1], str):#we have the name of an empty sequence on the stack. create it!
             oldSequenceLabel = self._sequenceStack.pop()
             newSequenceLabel = oldSequenceLabel + self._postfix
             self._sequenceStack.append(cms.Sequence(visitee))
             if hasattr(self._process, newSequenceLabel):
                 raise Exception("Cloning the sequence "+self._sequenceStack[-1].label()+" would overwrite existing object." )
             setattr(self._process, newSequenceLabel, self._sequenceStack[-1])
             self._sequenceLabels.append(oldSequenceLabel)
         else:
             self._sequenceStack[-1] += visitee
         

def HiHelperTools.CloneSequenceVisitor.clonedSequence ( self )

Definition at line 96 of file HiHelperTools.py.

References HiHelperTools.CloneSequenceVisitor._moduleLabels, helpers.CloneSequenceVisitor._moduleLabels, HiHelperTools.CloneSequenceVisitor._postfix, helpers.CloneSequenceVisitor._postfix, HiHelperTools.CloneSequenceVisitor._sequenceStack, and MassReplace.massSearchReplaceAnyInputTag().

     def clonedSequence(self):
         if not len(self._sequenceStack) == 1:
             raise Exception("someting went wrong, the sequence stack looks like: %s"%self._sequenceStack)
         for label in self._moduleLabels:
             massSearchReplaceAnyInputTag(self._sequenceStack[-1], label, label+self._postfix, moduleLabelOnly=True, verbose=False)
         self._moduleLabels = [] #prevent the InputTag replacement next time this is called.
         return self._sequenceStack[-1]
 

def HiHelperTools.CloneSequenceVisitor.enter	(	self,
		visitee
	)

Definition at line 66 of file HiHelperTools.py.

     def enter(self,visitee):
         if not self._waitForSequenceToClose is None:
             return #we are in a already cloned sequence
         if isinstance(visitee,cms._Module):
             label = visitee.label()
             newModule = None
             if label in self._moduleLabels:
                 newModule = getattr(self._process, label+self._postfix)
             else:
                 self._moduleLabels.append(label)
                 
                 newModule = visitee.clone()
                 setattr(self._process, label+self._postfix, newModule)
             self.__appendToTopSequence(newModule)
 
         if isinstance(visitee,cms.Sequence):
             if visitee.label() in self._sequenceLabels: # is the sequence allready cloned?
                 self._waitForSequenceToClose = visitee.label()
                 self._sequenceStack.append(  getattr(self._process, visitee.label()+self._postfix) )
             else:
                 self._sequenceStack.append(visitee.label())#save desired label as placeholder until we have a module to create the sequence
 

def HiHelperTools.CloneSequenceVisitor.leave	(	self,
		visitee
	)

Definition at line 88 of file HiHelperTools.py.

References HiHelperTools.CloneSequenceVisitor.__appendToTopSequence(), helpers.CloneSequenceVisitor.__appendToTopSequence(), HiHelperTools.CloneSequenceVisitor._sequenceStack, and HiHelperTools.CloneSequenceVisitor._waitForSequenceToClose.

     def leave(self,visitee):
         if isinstance(visitee,cms.Sequence):
             if self._waitForSequenceToClose == visitee.label():
                 self._waitForSequenceToClose = None
             if not isinstance(self._sequenceStack[-1], cms.Sequence):
                 raise Exception("empty Sequence encountered during cloneing. sequnece stack: %s"%self._sequenceStack)
             self.__appendToTopSequence( self._sequenceStack.pop() )
 