Source code for analyzer.modules.common.save_arrays

from analyzer.core.analysis_modules import AnalyzerModule
from analyzer.core.columns import Column
from attrs import define
from analyzer.core.results import SavedColumns
from analyzer.core.analysis_modules import MetadataExpr


@define


class SaveCols(AnalyzerModule):


    save_name: str



    to_save: dict[str, str]



    remap_hlt_prefix: bool = True




    def run(self, columns, params):
        ret = {}
        era_trigger_names = columns.metadata["era"]["trigger_names"]
        for name, col in self.to_save.items():
            real_name = col
            if real_name.startswith("HLTMAP:"):
                real_name = "HLT." + era_trigger_names[real_name[7:]]
            ret[name] = columns[real_name]
        return columns, [SavedColumns(self.save_name, ret)]




    def inputs(self, metadata):
        era_trigger_names = metadata["era"]["trigger_names"]
        ret = []
        for col in self.to_save.values():
            real_name = col
            if real_name.startswith("HLTMAP:"):
                real_name = "HLT." + era_trigger_names[real_name[7:]]
            ret.append(Column(real_name))
        return ret




    def outputs(self, metadata):
        return []