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