higgs_dna.scripts.postprocessing package
Subpackages
Submodules
higgs_dna.scripts.postprocessing.convert_parquet_to_root module
higgs_dna.scripts.postprocessing.merge_parquet module
- higgs_dna.scripts.postprocessing.merge_parquet.extract_filter(dataset, additionalConditionTuple)[source]
higgs_dna.scripts.postprocessing.prepare_output_file module
- higgs_dna.scripts.postprocessing.prepare_output_file.MKDIRP(dirpath, verbose=False, dry_run=False)[source]
- higgs_dna.scripts.postprocessing.prepare_output_file.decompose_string(input_string, era_flag=False)[source]
Decomposes the input string into process, mass, and era components based on underscores.
- Args:
input_string (str): The string to be decomposed. era_flag (bool): If True, include the era in the output. If False, exclude the era.
- Returns:
str: The formatted string in the style process + _ + mass (+ _ + era if era_flag is True).
higgs_dna.scripts.postprocessing.prepare_root_for_tree2ws_diff module
- higgs_dna.scripts.postprocessing.prepare_root_for_tree2ws_diff.get_trees_in_directory(file_path, directory_name)[source]
Retrieves the list of trees and the original directory from a ROOT file.
Parameters: - file_path (str): Path to the ROOT file. - directory_name (str): Name of the directory containing TTrees.
Returns: - original_trees (list): List of tree names. - original_directory (uproot.reading.ReadOnlyDirectory): Original directory object.
- higgs_dna.scripts.postprocessing.prepare_root_for_tree2ws_diff.process_root_file(file_path, output_directory)[source]
Processes the original TTrees, applies fiducial region selection, and saves the selected data to the output directory.
Parameters: - file_path (str): Path to the input ROOT file. - output_directory (uproot.write.TDirectory): Output directory for saving processed TTrees.