python.TriggerAnalysisSFConfig Namespace Reference

Classes
class	TriggerAnalysisSFBlock

Functions
bool	is_mc_from (ConfigAccumulator config, Union[Campaign, Iterable[Campaign]] campaign_list)
bool	is_data_from (config, Union[int, Iterable[int]] data_year_list)
list	get_year_data (dict dictionary, int|str year)
list[int]	get_input_years (ConfigAccumulator config)
	trigger_set (config, triggerChainsPerYear, includeAllYearsPerRun)

Function Documentation

get_input_years()

list[int] get_input_years

(

ConfigAccumulator

config

)

Utility function to get the list of years that the input corresponds to

Definition at line 35 of file TriggerAnalysisSFConfig.py.

def get_input_years(config: ConfigAccumulator) -> list[int]:
    """
    Utility function to get the list of years that the input corresponds to
    """
    years = []
    if config.campaign() is Campaign.MC20a:
        years = [2015, 2016]
    elif is_data_from(config, 2015):
        years = [2015]
    elif is_data_from(config, 2016):
        years = [2016]
    elif config.campaign() is Campaign.MC20d or is_data_from(config, 2017):
        years = [2017]
    elif config.campaign() is Campaign.MC20e or is_data_from(config, 2018):
        years = [2018]
    elif config.campaign() is Campaign.MC23a or is_data_from(config, 2022):
        years = [2022]
    elif config.campaign() is Campaign.MC23d or is_data_from(config, 2023):
        years = [2023]
    elif config.campaign() is Campaign.MC23e or is_data_from(config, 2024):
        years = [2024]
    elif config.campaign() is Campaign.MC23g:
        years = [2025, 2026]
    elif is_data_from(config, 2025):
        years = [2025]
    elif is_data_from(config, 2026):
        years = [2026]
    else:
        raise ValueError('TriggerAnalysisConfig: unable to deduce data taking year for input file')
 
    return years
 
 

get_year_data()

list get_year_data	(	dict	dictionary,
		int | str	year )

Utility function to get the data for a specific year from a dictionary

Definition at line 28 of file TriggerAnalysisSFConfig.py.

def get_year_data(dictionary: dict, year: int | str) -> list:
    """
    Utility function to get the data for a specific year from a dictionary
    """
    return dictionary.get(int(year), dictionary.get(str(year), []))
 
 

is_data_from()

bool is_data_from	(		config,
		Union[int, Iterable[int]]	data_year_list )

Utility function to check whether the data type is data and whether the year is in desired list of years

Definition at line 20 of file TriggerAnalysisSFConfig.py.

def is_data_from(config, data_year_list: Union[int, Iterable[int]]) -> bool:
    """
    Utility function to check whether the data type is data and whether the year is in desired list of years
    """
    data_year_list = [data_year_list] if isinstance(data_year_list, int) else data_year_list
    return config.dataType() is DataType.Data and config.dataYear() in data_year_list
 
 

is_mc_from()

bool is_mc_from	(	ConfigAccumulator	config,
		Union[Campaign, Iterable[Campaign]]	campaign_list )

Utility function to check whether the data type is mc and whether the campaign is in desired list of campaigns
without causing an invalid FlugEnum comparison

Definition at line 11 of file TriggerAnalysisSFConfig.py.

def is_mc_from(config: ConfigAccumulator, campaign_list: Union[Campaign, Iterable[Campaign]]) -> bool:
    """
    Utility function to check whether the data type is mc and whether the campaign is in desired list of campaigns
    without causing an invalid FlugEnum comparison
    """
    campaign_list = [campaign_list] if isinstance(campaign_list, Campaign) else campaign_list
    return config.dataType() is not DataType.Data and config.campaign() in campaign_list
 
 

trigger_set()

trigger_set	(		config,
			triggerChainsPerYear,
			includeAllYearsPerRun )

Definition at line 68 of file TriggerAnalysisSFConfig.py.

def trigger_set(config, triggerChainsPerYear, includeAllYearsPerRun):
    triggers = set()
    if includeAllYearsPerRun:
        for year in triggerChainsPerYear:
            if not is_year_in_current_period(config, year):
                continue
            triggers.update(get_year_data(triggerChainsPerYear, year))
    else:
        for year in get_input_years(config):
            triggers.update(get_year_data(triggerChainsPerYear, year))
    return triggers