CMS 3D CMS Logo

 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Properties Friends Macros Pages
List of all members | Public Member Functions | Private Attributes | Static Private Attributes
tauTools.AddTauCollection Class Reference
Inheritance diagram for tauTools.AddTauCollection:

Public Member Functions

def __call__
 
def __init__
 
def getDefaultParameters
 
def toolCode
 

Private Attributes

 _comment
 
 _parameters
 

Static Private Attributes

tuple _defaultParameters = dicttypes.SortedKeysDict()
 
string _label = 'addTauCollection'
 

Detailed Description

Add a new collection of taus. Takes the configuration from the
already configured standard tau collection as starting point;
replaces before calling addTauCollection will also affect the
new tau collections

Definition at line 364 of file tauTools.py.

Constructor & Destructor Documentation

def tauTools.AddTauCollection.__init__ (   self)

Definition at line 373 of file tauTools.py.

References metTools.AddMETCollection._defaultParameters, editorTools.UserCodeTool._defaultParameters, heavyIonTools.ConfigureHeavyIons._defaultParameters, coreTools.RunOnData._defaultParameters, Vispa.Plugins.ConfigEditor.ToolDataAccessor.ImportTool._defaultParameters, heavyIonTools.ProductionDefaults._defaultParameters, editorTools.ChangeSource._defaultParameters, cmsswVersionTools.PickRelValInputFiles._defaultParameters, coreTools.RemoveMCMatching._defaultParameters, heavyIonTools.SelectionDefaults._defaultParameters, heavyIonTools.DisbaleMonteCarloDeps._defaultParameters, jetTools.AddJetCollection._defaultParameters, tauTools.AddTauCollection._defaultParameters, jetTools.SwitchJetCollection._defaultParameters, jetTools.UpdateJetCollection._defaultParameters, jetTools.AddJetID._defaultParameters, jetTools.SetTagInfos._defaultParameters, ConfigToolBase.ConfigToolBase._defaultValue, GenObject.GenObject._defaultValue, root::helper::RootFunctionAdapter< F, args >.addParameter(), root::helper::RootFunctionHelper< F, args, Tag >.addParameter(), ParameterMgr.addParameter(), fit::RootMinuit< Function >.addParameter(), DropBoxMetadata::Parameters.addParameter(), PythonParameterSet.addParameter(), LutXml.addParameter(), ConfigToolBase.ConfigToolBase.addParameter(), and edm::ParameterSet.addParameter().

374  def __init__(self):
375  ConfigToolBase.__init__(self)
376  self.addParameter(self._defaultParameters, 'tauCollection',
377  self._defaultValue, 'Input tau collection', cms.InputTag)
378  self.addParameter(self._defaultParameters, 'algoLabel',
379  self._defaultValue, "label to indicate the tau algorithm (e.g.'hps')", str)
380  self.addParameter(self._defaultParameters, 'typeLabel',
381  self._defaultValue, "label to indicate the type of constituents (either 'PFTau' or 'Tau')", str)
382  self.addParameter(self._defaultParameters, 'doPFIsoDeposits',
383  True, "run sequence for computing particle-flow based IsoDeposits")
384  self.addParameter(self._defaultParameters, 'standardAlgo',
385  "hps", "standard algorithm label of the collection from which the clones " \
386  + "for the new tau collection will be taken from " \
387  + "(note that this tau collection has to be available in the event before hand)")
388  self.addParameter(self._defaultParameters, 'standardType',
389  "PFTau", "standard constituent type label of the collection from which the clones " \
390  + " for the new tau collection will be taken from "\
391  + "(note that this tau collection has to be available in the event before hand)")
393  self._parameters=copy.deepcopy(self._defaultParameters)
394  self._comment = ""

Member Function Documentation

def tauTools.AddTauCollection.__call__ (   self,
  process,
  tauCollection = None,
  algoLabel = None,
  typeLabel = None,
  doPFIsoDeposits = None,
  jetCorrLabel = None,
  standardAlgo = None,
  standardType = None 
)

Definition at line 405 of file tauTools.py.

References metTools.AddMETCollection._defaultParameters, editorTools.UserCodeTool._defaultParameters, heavyIonTools.ConfigureHeavyIons._defaultParameters, coreTools.RunOnData._defaultParameters, Vispa.Plugins.ConfigEditor.ToolDataAccessor.ImportTool._defaultParameters, heavyIonTools.ProductionDefaults._defaultParameters, editorTools.ChangeSource._defaultParameters, cmsswVersionTools.PickRelValInputFiles._defaultParameters, coreTools.RemoveMCMatching._defaultParameters, heavyIonTools.SelectionDefaults._defaultParameters, heavyIonTools.DisbaleMonteCarloDeps._defaultParameters, jetTools.AddJetCollection._defaultParameters, tauTools.AddTauCollection._defaultParameters, jetTools.SwitchJetCollection._defaultParameters, jetTools.UpdateJetCollection._defaultParameters, jetTools.AddJetID._defaultParameters, jetTools.SetTagInfos._defaultParameters, SignedImpactParameter3D.apply(), gen::PhotosInterfaceBase.apply(), SignedTransverseImpactParameter.apply(), SignedDecayLength3D.apply(), HcalDeterministicFit.apply(), MatrixReader.MatrixReader.apply, DummyOOTPileupCorrection.apply(), gen::PhotosppInterface.apply(), SurveyParameters.apply(), FWPSetCellEditor.apply(), gen::PhotosInterface.apply(), FWGUIEventFilter.apply(), DeviationsFromFileSensor2D.apply(), BeamSpotAlignmentParameters.apply(), RigidBodyAlignmentParameters.apply(), MatRepSparse< T, D1, D2, S, F >.apply(), ConfigToolBase.ConfigToolBase.apply(), AlignmentParameters.apply(), OOTPileupCorrData.apply(), BowedSurfaceAlignmentParameters.apply(), Vispa.Plugins.ConfigEditor.ToolDataAccessor.ImportTool.apply(), Vispa.Plugins.ConfigEditor.ToolDataAccessor.ApplyTool.apply(), AbsOOTPileupCorrection.apply(), TwoBowedSurfacesAlignmentParameters.apply(), Vispa.Gui.BoxContentDialog.BoxContentDialog.apply(), reco::modules::TrackerTrackHitFilter::Rule.apply(), PulseShapeFitOOTPileupCorrection.apply(), Vispa.Plugins.ConfigEditor.ToolDialog.ToolDialog.apply(), reco::Add< T >.apply(), reco::Sub< T >.apply(), reco::Mul< T >.apply(), reco::Div< T >.apply(), cmsswVersionTools.PickRelValInputFiles.apply(), reco::And< bool >.apply(), reco::Or< bool >.apply(), reco::Less< T >.apply(), reco::LessEqual< T >.apply(), reco::More< T >.apply(), reco::MoreEqual< T >.apply(), reco::Equal< T >.apply(), ArrayND< Accumulator >.apply(), ArrayND< Numeric, Len2, Dim2 >.apply(), npstat::ArrayND< Numeric >.apply(), ArrayND< Numeric, Len, Dim >.apply(), npstat::ArrayND< Numeric, StackLen, StackDim >.apply(), npstat::ArrayND< Num1, Len1, Dim1 >.apply(), ArrayND< Num2, Len2, Dim2 >.apply(), Config.ProcessModifier.apply(), ora::OraMainTable.setParameter(), ora::IMainTable.setParameter(), PedeReader.setParameter(), ora::PoolMainTable.setParameter(), QIE8Simulator.setParameter(), AbsElectronicODERHS.setParameter(), IntegratedCalibrationBase.setParameter(), SiStripLorentzAngleCalibration.setParameter(), SiPixelLorentzAngleCalibration.setParameter(), SiStripBackplaneCalibration.setParameter(), fit::RootMinuit< Function >.setParameter(), and ConfigToolBase.ConfigToolBase.setParameter().

Referenced by tauTools.AddTauCollection.getDefaultParameters().

406  standardType = None):
407 
408  if tauCollection is None:
409  tauCollection = self._defaultParameters['tauCollection'].value
410  if algoLabel is None:
411  algoLabel = self._defaultParameters['algoLabel'].value
412  if typeLabel is None:
413  typeLabel = self._defaultParameters['typeLabel'].value
414  if doPFIsoDeposits is None:
415  doPFIsoDeposits = self._defaultParameters['doPFIsoDeposits'].value
416  if standardAlgo is None:
417  standardAlgo = self._defaultParameters['standardAlgo'].value
418  if standardType is None:
419  standardType = self._defaultParameters['standardType'].value
420 
421  self.setParameter('tauCollection', tauCollection)
422  self.setParameter('algoLabel', algoLabel)
423  self.setParameter('typeLabel', typeLabel)
424  self.setParameter('doPFIsoDeposits', doPFIsoDeposits)
425  self.setParameter('standardAlgo', standardAlgo)
426  self.setParameter('standardType', standardType)
427 
428  self.apply(process)
def tauTools.AddTauCollection.getDefaultParameters (   self)

Definition at line 395 of file tauTools.py.

References tauTools.AddTauCollection.__call__(), metTools.AddMETCollection._defaultParameters, editorTools.UserCodeTool._defaultParameters, heavyIonTools.ConfigureHeavyIons._defaultParameters, coreTools.RunOnData._defaultParameters, Vispa.Plugins.ConfigEditor.ToolDataAccessor.ImportTool._defaultParameters, heavyIonTools.ProductionDefaults._defaultParameters, editorTools.ChangeSource._defaultParameters, cmsswVersionTools.PickRelValInputFiles._defaultParameters, coreTools.RemoveMCMatching._defaultParameters, heavyIonTools.SelectionDefaults._defaultParameters, heavyIonTools.DisbaleMonteCarloDeps._defaultParameters, jetTools.AddJetCollection._defaultParameters, tauTools.AddTauCollection._defaultParameters, jetTools.SwitchJetCollection._defaultParameters, jetTools.UpdateJetCollection._defaultParameters, jetTools.AddJetID._defaultParameters, and jetTools.SetTagInfos._defaultParameters.

396  def getDefaultParameters(self):
397  return self._defaultParameters
def tauTools.AddTauCollection.toolCode (   self,
  process 
)

Definition at line 429 of file tauTools.py.

References editorTools.UserCodeTool._parameters, heavyIonTools.ConfigureHeavyIons._parameters, coreTools.RunOnData._parameters, metTools.AddMETCollection._parameters, Vispa.Plugins.ConfigEditor.ConfigDataAccessor.ConfigFolder._parameters, Vispa.Plugins.ConfigEditor.ToolDataAccessor.ImportTool._parameters, heavyIonTools.ProductionDefaults._parameters, ConfigToolBase.ConfigToolBase._parameters, editorTools.ChangeSource._parameters, TAbsFitConstraint._parameters, TAbsFitParticle._parameters, coreTools.RemoveMCMatching._parameters, cmsswVersionTools.PickRelValInputFiles._parameters, heavyIonTools.SelectionDefaults._parameters, heavyIonTools.DisbaleMonteCarloDeps._parameters, jetTools.AddJetCollection._parameters, tauTools.AddTauCollection._parameters, jetTools.SwitchJetCollection._parameters, jetTools.UpdateJetCollection._parameters, jetTools.AddJetID._parameters, jetTools.SetTagInfos._parameters, clone(), python.rootplot.root2matplotlib.replace(), tauTools.switchToCaloTau(), and tauTools.switchToPFTauByType().

430  def toolCode(self, process):
431  tauCollection = self._parameters['tauCollection'].value
432  algoLabel = self._parameters['algoLabel'].value
433  typeLabel = self._parameters['typeLabel'].value
434  doPFIsoDeposits = self._parameters['doPFIsoDeposits'].value
435  standardAlgo = self._parameters['standardAlgo'].value
436  standardType = self._parameters['standardType'].value
437 
438  ## disable computation of particle-flow based IsoDeposits
439  ## in case tau is of CaloTau type
440  if typeLabel == 'Tau':
441  print "NO PF Isolation will be computed for CaloTau (this could be improved later)"
442  doPFIsoDeposits = False
443 
444  ## create old module label from standardAlgo
445  ## and standardType and return
446  def oldLabel(prefix = ''):
447  if prefix == '':
448  return "patTaus"
449  else:
450  return prefix + "PatTaus"
451 
452  ## capitalize first character of appended part
453  ## when creating new module label
454  ## (giving e.g. "patTausCaloRecoTau")
455  def capitalize(label):
456  return label[0].capitalize() + label[1:]
457 
458  ## create new module label from old module
459  ## label and return
460  def newLabel(oldLabel):
461  newLabel = oldLabel
462  if ( oldLabel.find(standardAlgo) >= 0 and oldLabel.find(standardType) >= 0 ):
463  oldLabel = oldLabel.replace(standardAlgo, algoLabel).replace(standardType, typeLabel)
464  else:
465  oldLabel = oldLabel + capitalize(algoLabel + typeLabel)
466  return oldLabel
467 
468  ## clone module and add it to the patDefaultSequence
469  def addClone(hook, **replaceStatements):
470  ## create a clone of the hook with corresponding
471  ## parameter replacements
472  newModule = getattr(process, hook).clone(**replaceStatements)
473 
474  ## clone module for computing particle-flow IsoDeposits
475  def addPFIsoDepositClone(hook, **replaceStatements):
476  newModule = getattr(process, hook).clone(**replaceStatements)
477  newModuleIsoDepositExtractor = getattr(newModule, "ExtractorPSet")
478  setattr(newModuleIsoDepositExtractor, "tauSource", getattr(newModule, "src"))
479 
480  ## add a clone of patTaus
481  addClone(oldLabel(), tauSource = tauCollection)
482 
483  ## add a clone of selectedPatTaus
484  addClone(oldLabel('selected'), src = cms.InputTag(newLabel(oldLabel())))
485 
486  ## add a clone of cleanPatTaus
487  addClone(oldLabel('clean'), src=cms.InputTag(newLabel(oldLabel('selected'))))
488 
489  ## get attributes of new module
490  newTaus = getattr(process, newLabel(oldLabel()))
491 
492  ## add a clone of gen tau matching
493  addClone('tauMatch', src = tauCollection)
494  addClone('tauGenJetMatch', src = tauCollection)
495 
496  ## add a clone of IsoDeposits computed based on particle-flow
497  if doPFIsoDeposits:
498  addPFIsoDepositClone('tauIsoDepositPFCandidates', src = tauCollection)
499  addPFIsoDepositClone('tauIsoDepositPFChargedHadrons', src = tauCollection)
500  addPFIsoDepositClone('tauIsoDepositPFNeutralHadrons', src = tauCollection)
501  addPFIsoDepositClone('tauIsoDepositPFGammas', src = tauCollection)
502 
503  ## fix label for input tag
504  def fixInputTag(x):
505  x.setModuleLabel(newLabel(x.moduleLabel))
506 
507  ## provide patTau inputs with individual labels
508  fixInputTag(newTaus.genParticleMatch)
509  fixInputTag(newTaus.genJetMatch)
510  fixInputTag(newTaus.isoDeposits.pfAllParticles)
511  fixInputTag(newTaus.isoDeposits.pfNeutralHadron)
512  fixInputTag(newTaus.isoDeposits.pfChargedHadron)
513  fixInputTag(newTaus.isoDeposits.pfGamma)
514  fixInputTag(newTaus.userIsolation.pfAllParticles.src)
515  fixInputTag(newTaus.userIsolation.pfNeutralHadron.src)
516  fixInputTag(newTaus.userIsolation.pfChargedHadron.src)
517  fixInputTag(newTaus.userIsolation.pfGamma.src)
518 
519  ## set discriminators
520  ## (using switchTauCollection functions)
521  oldTaus = getattr(process, oldLabel())
522  if typeLabel == 'Tau':
523  switchToCaloTau(process,
524  tauSource = getattr(newTaus, "tauSource"),
525  patTauLabel = capitalize(algoLabel + typeLabel))
526  else:
527  switchToPFTauByType(process, pfTauType = algoLabel + typeLabel,
528  tauSource = getattr(newTaus, "tauSource"),
529  patTauLabel = capitalize(algoLabel + typeLabel))
530 
531 addTauCollection=AddTauCollection()
def switchToCaloTau
Definition: tauTools.py:20
TEveGeoShape * clone(const TEveElement *element, TEveElement *parent)
Definition: eve_macros.cc:135
def switchToPFTauByType
Definition: tauTools.py:351

Member Data Documentation

tauTools.AddTauCollection._comment
private

Definition at line 393 of file tauTools.py.

Referenced by runJetUncertainties.RunJetUncertainties.__init__(), and runMETCorrectionsAndUncertainties.RunMETCorrectionsAndUncertainties.__init__().

tuple tauTools.AddTauCollection._defaultParameters = dicttypes.SortedKeysDict()
staticprivate

Definition at line 372 of file tauTools.py.

Referenced by HiCoreTools.RestrictInputToAOD.__call__(), trackTools.MakeAODTrackCandidates.__call__(), runJetUncertainties.RunJetUncertainties.__call__(), HiCoreTools.RemoveMCMatching.__call__(), trackTools.MakePATTrackCandidates.__call__(), trigTools.SwitchOnTrigger.__call__(), HiCoreTools.RemoveAllPATObjectsBut.__call__(), HiCoreTools.RemoveSpecificPATObjects.__call__(), trigTools.SwitchOnTriggerStandAlone.__call__(), trackTools.MakeTrackCandidates.__call__(), trigTools.SwitchOnTriggerMatching.__call__(), HiCoreTools.RemoveCleaning.__call__(), HiCoreTools.AddCleaning.__call__(), tauTools.AddTauCollection.__call__(), trigTools.SwitchOnTriggerMatchingStandAlone.__call__(), trigTools.SwitchOnTriggerMatchEmbedding.__call__(), trackTools.MakeAODTrackCandidates.__init__(), HiCoreTools.RestrictInputToAOD.__init__(), runJetUncertainties.RunJetUncertainties.__init__(), runMETCorrectionsAndUncertainties.RunMETCorrectionsAndUncertainties.__init__(), HiCoreTools.RemoveMCMatching.__init__(), trackTools.MakePATTrackCandidates.__init__(), trigTools.SwitchOnTrigger.__init__(), HiCoreTools.RemoveAllPATObjectsBut.__init__(), HiCoreTools.RemoveSpecificPATObjects.__init__(), trigTools.SwitchOnTriggerStandAlone.__init__(), trackTools.MakeTrackCandidates.__init__(), trigTools.SwitchOnTriggerMatching.__init__(), HiCoreTools.RemoveCleaning.__init__(), tauTools.AddTauCollection.__init__(), HiCoreTools.AddCleaning.__init__(), trigTools.SwitchOnTriggerMatchingStandAlone.__init__(), trigTools.SwitchOnTriggerMatchEmbedding.__init__(), ConfigToolBase.ConfigToolBase.dumpPython(), ConfigToolBase.ConfigToolBase.getAllowedValues(), HiCoreTools.RestrictInputToAOD.getDefaultParameters(), trackTools.MakeAODTrackCandidates.getDefaultParameters(), HiCoreTools.RemoveMCMatching.getDefaultParameters(), trackTools.MakePATTrackCandidates.getDefaultParameters(), trigTools.SwitchOnTrigger.getDefaultParameters(), HiCoreTools.RemoveAllPATObjectsBut.getDefaultParameters(), HiCoreTools.RemoveSpecificPATObjects.getDefaultParameters(), trigTools.SwitchOnTriggerStandAlone.getDefaultParameters(), trackTools.MakeTrackCandidates.getDefaultParameters(), trigTools.SwitchOnTriggerMatching.getDefaultParameters(), HiCoreTools.RemoveCleaning.getDefaultParameters(), HiCoreTools.AddCleaning.getDefaultParameters(), tauTools.AddTauCollection.getDefaultParameters(), trigTools.SwitchOnTriggerMatchingStandAlone.getDefaultParameters(), trigTools.SwitchOnTriggerMatchEmbedding.getDefaultParameters(), ConfigToolBase.ConfigToolBase.reset(), and ConfigToolBase.ConfigToolBase.setParameter().

string tauTools.AddTauCollection._label = 'addTauCollection'
staticprivate

Definition at line 371 of file tauTools.py.

Referenced by validation.SimpleSample.digest(), validation.SimpleSample.label(), trigTools.SwitchOnTrigger.toolCode(), trigTools.SwitchOnTriggerStandAlone.toolCode(), trigTools.SwitchOnTriggerMatching.toolCode(), trigTools.SwitchOnTriggerMatchingStandAlone.toolCode(), and trigTools.SwitchOnTriggerMatchEmbedding.toolCode().

tauTools.AddTauCollection._parameters
private

Definition at line 392 of file tauTools.py.

Referenced by runJetUncertainties.RunJetUncertainties.__init__(), runMETCorrectionsAndUncertainties.RunMETCorrectionsAndUncertainties.__init__(), HiCoreTools.RestrictInputToAOD.toolCode(), trackTools.MakeAODTrackCandidates.toolCode(), runJetUncertainties.RunJetUncertainties.toolCode(), HiCoreTools.RemoveMCMatching.toolCode(), trigTools.SwitchOnTrigger.toolCode(), trackTools.MakePATTrackCandidates.toolCode(), HiCoreTools.RemoveAllPATObjectsBut.toolCode(), HiCoreTools.RemoveSpecificPATObjects.toolCode(), trigTools.SwitchOnTriggerStandAlone.toolCode(), trackTools.MakeTrackCandidates.toolCode(), HiCoreTools.RemoveCleaning.toolCode(), trigTools.SwitchOnTriggerMatching.toolCode(), HiCoreTools.AddCleaning.toolCode(), tauTools.AddTauCollection.toolCode(), trigTools.SwitchOnTriggerMatchingStandAlone.toolCode(), and trigTools.SwitchOnTriggerMatchEmbedding.toolCode().