|
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
|
import itertools
import FWCore.ParameterSet.Config as cms
def producers_by_type(process, *types):
"Find all EDProducers in the Process that are instances of the given C++ type."
switches = (module for module in (getattr(switchproducer, case) \
for switchproducer in process._Process__switchproducers.values() \
for case in switchproducer.parameterNames_()) \
if isinstance(module, cms.EDProducer))
return (module for module in itertools.chain(process._Process__producers.values(), switches) \
if module._TypedParameterizable__type in types)
def filters_by_type(process, *types):
"Find all EDFilters in the Process that are instances of the given C++ type."
return (filter for filter in process._Process__filters.values() if filter._TypedParameterizable__type in types)
def analyzers_by_type(process, *types):
"Find all EDAnalyzers in the Process that are instances of the given C++ type."
return (analyzer for analyzer in process._Process__analyzers.values() if analyzer._TypedParameterizable__type in types)
def esproducers_by_type(process, *types):
"Find all ESProducers in the Process that are instances of the given C++ type."
return (module for module in process._Process__esproducers.values() if module._TypedParameterizable__type in types)
def modules_by_type(process, *types):
"Find all modiles or other components in the Process that are instances of the given C++ type."
switches = (module for module in (getattr(switchproducer, case) \
for switchproducer in process._Process__switchproducers.values() \
for case in switchproducer.parameterNames_()))
return (module for module in itertools.chain(process.__dict__.values(), switches) \
if hasattr(module, '_TypedParameterizable__type') and module._TypedParameterizable__type in types)
def insert_modules_before(process, target, *modules):
"Add the `modules` before the `target` in any Sequence, Paths or EndPath that contains the latter."
for sequence in itertools.chain(
process._Process__sequences.values(),
process._Process__paths.values(),
process._Process__endpaths.values()
):
try:
position = sequence.index(target)
except ValueError:
continue
else:
for module in reversed(modules):
sequence.insert(position, module)
def insert_modules_after(process, target, *modules):
"Add the `modules` after the `target` in any Sequence, Paths or EndPath that contains the latter."
for sequence in itertools.chain(
process._Process__sequences.values(),
process._Process__paths.values(),
process._Process__endpaths.values()
):
try:
position = sequence.index(target)
except ValueError:
continue
else:
for module in reversed(modules):
sequence.insert(position+1, module)
# logic from Modifier.toModify from FWCore/ParameterSet/python/Config.py
def replace_with(fromObj, toObj):
"""Replace one object with a different one of the same type.
This function replaces the contents of `fromObj` object with those of `toObj`,
so all references ot it remain valid.
"""
if not isinstance(toObj, type(fromObj)):
raise TypeError('replaceWith requires both arguments to be the same type')
if isinstance(toObj, cms._ModuleSequenceType):
fromObj._seq = toObj._seq
elif isinstance(toObj, cms._Parameterizable):
# delete the old items, in case `toObj` is not a complete superset of `fromObj`
for p in fromObj.parameterNames_():
delattr(fromObj, p)
for p in toObj.parameterNames_():
setattr(fromObj, p, getattr(toObj, p))
if isinstance(toObj, cms._TypedParameterizable):
fromObj._TypedParameterizable__type = toObj._TypedParameterizable__type
else:
raise TypeError('replaceWith does not work with "%s" objects' % str(type(fromObj)))
# update or set the HLT prescale for the given path to the given value, in all the prescale columns
def set_prescale(process, path, prescale):
columns = len(process.PrescaleService.lvl1Labels.value())
prescales = cms.vuint32([prescale] * columns)
for pset in process.PrescaleService.prescaleTable:
if pset.pathName.value() == path:
pset.prescales = prescales
break
else:
process.PrescaleService.prescaleTable.append(cms.PSet(
pathName = cms.string(path),
prescales = prescales
))
|