python.DQPostProcessingAlg.AthInputModule Class Reference

Inheritance diagram for python.DQPostProcessingAlg.AthInputModule:

Collaboration diagram for python.DQPostProcessingAlg.AthInputModule:

Public Member Functions
def	__init__ (self)
None	configure (self, Mapping[str, Any] options)
None	setSelectors (self, Collection[Pattern] selectors)
Generator[HistObject, None, None]	iterate (self, dryrun)
Iterator[HistObject]	__iter__ (self)
Iterable[HistObject]	warmup (self)

Public Attributes
	source
	classwarnings
	selectors
	entries
	prefix
	cachednames
	matchednames

Private Member Functions
def	_getklass (self, k)

Detailed Description

Definition at line 135 of file DQPostProcessingAlg.py.

Constructor & Destructor Documentation

__init__()

def python.DQPostProcessingAlg.AthInputModule.__init__

(

self

)

Definition at line 136 of file DQPostProcessingAlg.py.

    def __init__(self):
        self.source = None
        self.classwarnings = set()
        self.selectors = None
        self.entries = {}
 

Member Function Documentation

__iter__()

Iterator[HistObject] python.DQPostProcessingAlg.AthInputModule.__iter__

(

self

)

Definition at line 220 of file DQPostProcessingAlg.py.

    def __iter__(self) -> Iterator[HistObject]:
        return self.iterate(dryrun=False)
 

_getklass()

def python.DQPostProcessingAlg.AthInputModule._getklass (   self,   k  )

private

Definition at line 161 of file DQPostProcessingAlg.py.

    def _getklass(self, k):
        import ROOT
        hsvc = self.source.hsvc
        if hsvc.existsHist(k):
            hptr = ROOT.MakeNullPointer(ROOT.TH1)
            if hsvc.getHist(k, hptr).isSuccess():
                klass = getattr(ROOT, hptr.ClassName())
                return klass
        return None
 

configure()

None python.DQPostProcessingAlg.AthInputModule.configure	(		self,
		Mapping[str, Any]	options
	)

Configure this module. Potential elements of "options":
source: should be a ROOT-openable filename or URL.
prefix: directory path to search under. Returned histogram names
    will not include this.

Definition at line 142 of file DQPostProcessingAlg.py.

    def configure(self, options: Mapping[str, Any]) -> None:
        """
        Configure this module. Potential elements of "options":
        source: should be a ROOT-openable filename or URL.
        prefix: directory path to search under. Returned histogram names
            will not include this.
        """
        if 'source' not in options:
            raise ValueError("Must specify 'source' as an "
                             "option to AthInputModule")
        self.source = options['source']
        self.prefix = options.get('prefix', '/')
        self.cachednames = set()
        self.matchednames = {}
 

iterate()

Generator[HistObject, None, None] python.DQPostProcessingAlg.AthInputModule.iterate	(		self,
			dryrun
	)

Iterate over all histograms in THistSvc 

Definition at line 171 of file DQPostProcessingAlg.py.

    def iterate(self, dryrun) -> Generator[HistObject, None, None]:
        """ Iterate over all histograms in THistSvc """
        import ROOT
        log = self.source.msg
        specprefix = self.prefix % { 'run': self.source._run }
        log.debug(f'Would like to match {specprefix}')
        hsvc = self.source.hsvc
 
        # check if we have new histograms; if so, check against selectors to see if we're interested
        # weirdly unpythonic code to avoid memory leak in iterators pre-ROOT 6.26
        histnames = hsvc.getHists()
        currenthists = set(str(histnames[_]) for _ in range(len(histnames)))
        for k in currenthists - self.cachednames:
            # log.info(f'We have ... ? {k}')
            if not k.startswith(specprefix):
                continue
            shortk = k.replace(specprefix, '', 1)
            if self.selectors is not None:
                if not any(_.match(shortk) for _ in self.selectors):
                    continue
            self.matchednames[k] = None
        self.cachednames.update(currenthists)
        log.debug(f'We now have {len(self.cachednames)} entries in our cache, of {len(currenthists)} total plots')
        log.debug(f'There are {len(self.matchednames)} matches to be considered')
 
        # postprocess only matched histograms
        for k, klass in self.matchednames.items():
            if dryrun:
                yield HistObject(k.replace(specprefix, '', 1), None)
                
            log.debug(f'THistSvc input trying to read {k}')
            if klass is None:
                klass = self._getklass(k)
                self.matchednames[k] = klass
            hptr = ROOT.MakeNullPointer(klass)
            if hsvc.getHist(k, hptr).isSuccess():
                log.debug(f'THistSvc input read {k} as {type(hptr)}')
                obj = hptr
                ROOT.SetOwnership(obj, False) # no NOT attempt to GC histograms read from THistSvc
                if k in self.entries:
                    if obj.GetEntries() == self.entries[k]:
                        continue
                    self.entries[k] = obj.GetEntries()
                yield HistObject(k.replace(specprefix, '', 1), obj)
            else:
                log.error(f'Cannot read {k}')
            
        log.debug('Done on input side')
 

setSelectors()

None python.DQPostProcessingAlg.AthInputModule.setSelectors	(		self,
		Collection[Pattern]	selectors
	)

Do more later 

Definition at line 157 of file DQPostProcessingAlg.py.

    def setSelectors(self, selectors: Collection[Pattern]) -> None:
        """ Do more later """
        self.selectors = selectors
 

warmup()

Iterable[HistObject] python.DQPostProcessingAlg.AthInputModule.warmup

(

self

)

Definition at line 223 of file DQPostProcessingAlg.py.

    def warmup(self) -> Iterable[HistObject]:
        return self.iterate(dryrun=True)
 

Member Data Documentation

cachednames

python.DQPostProcessingAlg.AthInputModule.cachednames

Definition at line 154 of file DQPostProcessingAlg.py.

classwarnings

python.DQPostProcessingAlg.AthInputModule.classwarnings

Definition at line 138 of file DQPostProcessingAlg.py.

entries

python.DQPostProcessingAlg.AthInputModule.entries

Definition at line 140 of file DQPostProcessingAlg.py.

matchednames

python.DQPostProcessingAlg.AthInputModule.matchednames

Definition at line 155 of file DQPostProcessingAlg.py.

prefix

python.DQPostProcessingAlg.AthInputModule.prefix

Definition at line 153 of file DQPostProcessingAlg.py.

selectors

python.DQPostProcessingAlg.AthInputModule.selectors

Definition at line 139 of file DQPostProcessingAlg.py.

source

python.DQPostProcessingAlg.AthInputModule.source

Definition at line 137 of file DQPostProcessingAlg.py.

---

The documentation for this class was generated from the following file:

• DQPostProcessingAlg.py