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