Source code for lib.unit_functions

from srcs.utils import get_project_root

import glob
import inquirer
import numpy as np
import matplotlib
matplotlib.use('Qt5Agg')
from matplotlib import pyplot as plt
from itertools import product
from rich import print as rprint

from lib.io_functions import read_yaml_file

root = get_project_root()


[docs]def calibrate_charges(my_runs, info, user_input, debug=False):
    # Check if the calibration file exists
    if glob.glob(f'{root}/{info["OUT_PATH"][0]}/analysis/calibration.yml') != []:
        calibration_info = read_yaml_file(
        "calibration",
        path=f'{root}/{info["OUT_PATH"][0]}/analysis/',
        debug=user_input["debug"],
        )

        rprint("Calibration info: ", calibration_info)
        for run, ch in product(
            np.asarray(user_input["runs"]).astype(str),
            np.asarray(user_input["channels"]).astype(str),
        ):
            q = [
                inquirer.List(
                    "OV",
                    message=f"select gain value according to run {run} - ch {ch}",
                    choices=calibration_info[my_runs[run][ch]["Label"]].keys(),
                    default="MidOV",
                )
            ]
            ov_gain = inquirer.prompt(q)["OV"]
            variables = list(my_runs[run][ch].keys())
            for var in variables:
                if "Charge" in var and "Dict" not in var:
                    my_runs[run][ch][var + "PE"] = (
                        my_runs[run][ch][var]
                        / calibration_info[my_runs[run][ch]["Label"]][ov_gain]
                    )
    else:
        rprint("\n[red][ERROR] Calibration file not found, change VisConfig.yml PE settings to [/red]False!\n")

    return my_runs


[docs]def get_run_units(my_runs, debug=False):
    """Computes and store in a dictionary the units of each variable.
    
    :param my_runs: dictionary containing the data
    :type my_runs: dict
    :param debug: boolean to print debug messages, defaults to False
    :type debug: bool, optional
    
    :return: None
    """

    if debug:
        rprint("Getting units...")
    for run, ch in product(my_runs["NRun"], my_runs["NChannel"]):
        keys = my_runs[run][ch].keys()
        aux_dict = {}
        for key in keys:
            aux_dict[key] = get_unit(key, debug)
        my_runs[run][ch]["UnitsDict"] = aux_dict


[docs]def get_unit(key, debug=False):
    unit = "a.u."
    if "Amp" in key or "Ped" in key or "ADC" in key or "PreTrigger" in key:
        unit = "ADC"
    if "Time" in key or "Sampling" in key:
        unit = "ticks"
    if "Charge" in key and "Ana" in key and "PE" not in key:
        unit = "ADC x ticks"
    if "PE" in key and "Ana" in key:
        unit = "PE"
    if "Charge" in key and "Gauss" in key:
        unit = "PE"

    return unit
SCINT DOCUMENTATION

Source code for lib.unit_functions
