82 def makeAlgs (self, config) :
83
84 sysService = config.createService( 'CP::SystematicsSvc', 'SystematicsSvc' )
85
86
87 streamName = self.streamName or config.defaultHistogramStream()
88
89
90 if self.runSystematics is False:
91 runSystematics = self.runSystematics
92 elif config.noSystematics() is not None:
93
94
95 self.runSystematics = not config.noSystematics()
96 runSystematics = self.runSystematics
97 else:
98 runSystematics = True
99
100
101 config._noSystematics = not runSystematics
102
103 if runSystematics:
104 sysService.sigmaRecommended = 1
105 if config.dataType() is DataType.Data:
106
107 self.onlySystematicsCategories = ['JER']
108 if self.onlySystematicsCategories is not None:
109
110 requested_categories =
set()
111 for category_str in self.onlySystematicsCategories:
112 try:
113 category_enum = SystematicsCategories[category_str.upper()]
114 requested_categories |= category_enum.value
115 except KeyError:
116 raise ValueError(f"Invalid systematics category passed to option 'onlySystematicsCategories': {category_str}. Must be one of {', '.join(category.name for category in SystematicsCategories)}")
117
118 if len(requested_categories):
119 sysService.systematicsRegex = "^(?=.*(" + "|".join(sorted(requested_categories)) + ")|$).*"
120 if self.filterSystematics is not None:
121 sysService.systematicsRegex = self.filterSystematics
122 config.createService( 'CP::SelectionNameSvc', 'SelectionNameSvc')
123
124 if self.systematicsHistogram is not None:
125
126 allSysDumper = config.createAlgorithm( 'CP::SysListDumperAlg', 'SystematicsPrinter' )
127 allSysDumper.histogramName = self.systematicsHistogram
128 allSysDumper.RootStreamName = streamName
129
130 if self.separateWeightSystematics:
131
132 weightSysDumper = config.createAlgorithm( 'CP::SysListDumperAlg', 'OnlyWeightSystematicsPrinter' )
133 weightSysDumper.histogramName = f"{self.systematicsHistogram}OnlyWeights"
134 weightSysDumper.systematicsRegex = "^(GEN_|EL_EFF_|MUON_EFF_|PH_EFF_|TAUS_TRUEHADTAU_EFF_|FT_EFF_|JET_.*JvtEfficiency_|PRW_).*"
135 weightSysDumper.RootStreamName = streamName
136
137 if self.metadataHistogram:
138
139 if not config.flags:
140 raise ValueError ("Writing out the metadata histogram requires to pass config flags")
141 metadataHistAlg = config.createAlgorithm( 'CP::MetadataHistAlg', 'MetadataHistAlg' )
142 metadataHistAlg.histogramName = self.metadataHistogram
143 metadataHistAlg.dataType = str(config.dataType().value)
144 metadataHistAlg.campaign = str(config.dataYear()) if config.dataType() is DataType.Data else str(config.campaign().value)
145 metadataHistAlg.mcChannelNumber = str(config.dsid())
146 metadataHistAlg.RootStreamName = streamName
147 if config.dataType() is DataType.Data:
148 etag = "unavailable"
149 else:
150 from AthenaConfiguration.AutoConfigFlags import GetFileMD
151 metadata = GetFileMD(config.flags.Input.Files)
152 amiTags = metadata.get("AMITag", "not found!")
153 etag = str(amiTags.split("_")[0])
154 metadataHistAlg.etag = etag
155
156 if self.enableExpertMode:
157
158 warnings.simplefilter('ignore', ExpertModeWarning)
159
160 log = logging.getLogger('CommonServices')
161 bold = "\033[1m"
162 red = "\033[91m"
163 yellow = "\033[93m"
164 reset = "\033[0m"
165 log.warning(red +r"""
166 ________ _______ ______ _____ _______ __ __ ____ _____ ______ ______ _ _ ____ _ ______ _____
167 | ____\ \ / / __ \| ____| __ \__ __| | \/ |/ __ \| __ \| ____| | ____| \ | | /\ | _ \| | | ____| __ \
168 | |__ \ V /| |__) | |__ | |__) | | | | \ / | | | | | | | |__ | |__ | \| | / \ | |_) | | | |__ | | | |
169 | __| > < | ___/| __| | _ / | | | |\/| | | | | | | | __| | __| | . ` | / /\ \ | _ <| | | __| | | | |
170 | |____ / . \| | | |____| | \ \ | | | | | | |__| | |__| | |____ | |____| |\ |/ ____ \| |_) | |____| |____| |__| |
171 |______/_/ \_\_| |______|_| \_\ |_| |_| |_|\____/|_____/|______| |______|_| \_/_/ \_\____/|______|______|_____/
172
173"""
174 +reset)
175 log.warning(f"{bold}{yellow}These settings are not recommended for analysis. Make sure you know what you're doing, or disable them with `enableExpertMode: False` in `CommonServices`.{reset}")
176
177 if self.setupONNX:
178 config.createService('AthOnnx::OnnxRuntimeSvc', 'OnnxRuntimeSvc')
179 config.setContainerMeta ('EventInfo', 'OnnxEnabled', self.setupONNX)
180
181@groupBlocks
1.17.0