python.cmstools Namespace Reference

Classes
class	cmserror
class	Event
class	EventBranch
class	EventTree

Functions
def	all (container)
	workaround iterator generators for ROOT classes More...
def	createBranchBuffer (branch)
	auto branch types (Chris Jones) More...
def	loop (begin, end)

Detailed Description

Python helper tools for CMS FWLite

benedikt.hegner@cern.ch

Function Documentation

all()

def python.cmstools.all

(

container

)

workaround iterator generators for ROOT classes

Definition at line 25 of file cmstools.py.

References FastTimerService_cff.range.

Referenced by pat::EventHypothesis.all(), HepMCValidationHelper.allStatus1(), HepMCValidationHelper.allStatus2(), HepMCValidationHelper.allStatus3(), HepMCValidationHelper.allVisibleParticles(), SiPixelPhase1GeometryDebug.analyze(), HGcalHitIdCheck.analyze(), LayerCollector.barrelLayers(), AlignmentMonitorSurvey.book(), DQMGenericTnPClient.calculateEfficiency(), dqmTnP::AbstractFitter.calculateEfficiency(), HGCalCLUEAlgoT< TILE >.calculateLocalDensity(), CastorHardcodeCalibrations.CastorHardcodeCalibrations(), SummaryOutputProducer::TriggerSummary.changed(), combineBTagCalibrationData.check_csv_data(), GlobalMuonRefitter.checkMuonHits(), Phase2L1GMT::ROITempAssociator.clean(), CmsShowMain.CmsShowMain(), HcalDbASCIIIO.createObject< HcalFrontEndMap >(), HcalDbASCIIIO.createObject< HcalSiPMCharacteristics >(), EgHLTOfflineClient.createTrigTagProbeEffHists(), EgHLTOfflineClient.createTrigTagProbeEffHists2Leg(), TemplatedJetBProbabilityComputer< Container, Base >.discriminator(), EwkElecDQM.dqmEndRun(), HepMCValidationHelper.findFSRPhotons(), FWDetailViewManager.findViewersFor(), dqmTnP::GaussianPlusLinearFitter.fit(), dqmTnP::VoigtianPlusExponentialFitter.fit(), GlobalTrajectoryBuilderBase.fixTEC(), LayerCollector.forwardLayers(), SummaryOutputProducer::TriggerSummary.gained(), models.generate(), HepMCValidationHelper.genMet(), l1tVertexFinder::AnalysisSettings.genPdgIds(), run_AlCaRecoTriggerBitsUpdateWorkflow.get_iovs(), tools.get_iovs(), tools.get_tags(), SiStripDetCabling.getAllDetectorsContiguousIds(), G2GainsValidator.getExpressGT(), GlobalMuonRefitter.getFirstHits(), dataset.Dataset.getrunnumberfromfilename(), dataset.getrunnumbersfromfile(), helperFunctions.getTagsMap(), HcalHardcodeCalibrations.HcalHardcodeCalibrations(), AlignmentParameterStore.hierarchyConstraints(), conddblib.Connection.is_valid(), MuonTrackResidualAnalyzer.isInTheAcceptance(), MuonTrackAnalyzer.isInTheAcceptance(), o2olib.O2OJobMgr.listJobs(), conddblib.listObject(), HGCalDDDConstants.locateCell(), SummaryOutputProducer::TriggerSummary.lost(), LA_Filler_Fitter.make_and_fit_symmchi2(), tmtt::Histos.makeEfficiencyPlot(), EgHLTOfflineClient.makeEffMonElem2Leg(), EgHLTOfflineClient.makeEffMonElemFromPassAndAll(), SiStripHitEffFromCalibTree.makeSummary(), AlignableTracker.merge(), mergeLHE.DefaultLHEMerger.merge_headers(), mergeLHE.DefaultLHEMerger.merge_init_blocks(), pat::TriggerEvent.pathModules(), MuonSeedOrcaPatternRecognition.produce(), utils_v2.recursive_search_online(), HepMCValidationHelper.removeIsolatedLeptons(), plotBaryCentre_VS_BeamSpot.Run(), checkBTagCalibrationConsistency.run_check_data(), SiStripConfigDb.runs(), utils_v2.search_on_disk(), CSCTrackConfigSelector.select(), Phase2L1GMT::ROITempAssociator.selectLayerBX(), GlobalTrajectoryBuilderBase.selectTrackerHits(), DQMImplNet< DQMNet::Object >.sendObjectListToPeer(), DQMImplNet< DQMNet::Object >.sendObjectListToPeers(), heppy::IsolationComputer.setPackedCandidates(), TTDTC.size(), data_formats_tests.data_formats_tests.test_dicts_to_orm_objects(), data_formats_tests.data_formats_tests.test_orm_objects_to_dicts(), data_sources_tests.data_sources_tests.test_type_all_iovs(), data_sources_tests.data_sources_tests.test_type_all_tags(), TrimmedVertexFinder.vertices(), ConfigurableTrimmedVertexFinder.vertices(), and dataset.DatasetBase.writefilelist_hippy().

def all(container):

  # loop over ROOT::TTree and similar
  if hasattr(container,'GetEntries'):
    try:
      entries = container.GetEntries()
      for entry in range(entries):
        yield entry
    except:
        raise cmserror("Looping of %s failed" %container) 

  # loop over std::vectors and similar
  elif hasattr(container, 'size'):
    try:
      entries = container.size()
      for entry in range(entries):
        yield container[entry]
    except:
      pass

 # loop over containers with begin and end iterators

createBranchBuffer()

def python.cmstools.createBranchBuffer

(

branch

)

auto branch types (Chris Jones)

Definition at line 53 of file cmstools.py.

References if().

def createBranchBuffer(branch):
    reColons = re.compile(r'::')
    reCloseTemplate =re.compile(r'>')
    reOpenTemplate =re.compile(r'<')
    branchType = ROOT.branchToClass(branch)
    #buffer = eval ('ROOT.'+reColons.sub(".",reOpenTemplate.sub("(ROOT.",reCloseTemplate.sub(")",branchType.GetName())))+'()')
    buffer = ROOT.MakeRootClass(branchType.GetName()) ()
    if( branch.GetName()[-1] != '.') and (branch.GetName()!="EventAuxiliary"):
        branch.SetAddress(buffer)
    else:
        branch.SetAddress(ROOT.AddressOf(buffer))
    return buffer

loop()

def python.cmstools.loop	(		begin,
			end
	)

Convert a pair of C++ iterators into a python generator

Definition at line 46 of file cmstools.py.

def loop(begin, end):
    """Convert a pair of C++ iterators into a python generator"""
    while (begin != end):
        yield begin.__deref__()  #*b
        begin.__preinc__()       #++b