gnn_tracking.preprocessing.exatrkx_cell_features#

Attributes#

cf_logger

Classes#

DetectorRotations

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

DetectorThicknesses

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

DetectorPixelSize

Adapted from ExaTrkX's preprocessing. See docstring above.

Functions#

get_logger([name, level])

Sets up global logger.

load_detector(→ tuple[pandas.DataFrame, dict])

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

preprocess_detector(→ dict)

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

determine_array_size(→ tuple[float, float, float])

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

cartesian_to_spherical(x, y, z)

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

theta_to_eta(theta)

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

get_all_local_angles(→ tuple[pandas.Series, ...)

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

get_all_rotated(hits, detector, l_u, l_v, l_w)

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

extract_dir_new(→ pandas.DataFrame)

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

augment_hit_features(hits, cells, detector_proc)

Adapted/copied from ExaTrkX's preprocessing. See docstring above.

Module Contents#

gnn_tracking.preprocessing.exatrkx_cell_features.get_logger(name='gnn-tracking', level=LOG_DEFAULT_LEVEL)#

Sets up global logger.

gnn_tracking.preprocessing.exatrkx_cell_features.cf_logger#
gnn_tracking.preprocessing.exatrkx_cell_features.load_detector(detector_path: pathlib.Path) tuple[pandas.DataFrame, dict]#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

WARNING: This might create a race condition with creating the preprocessed file.

gnn_tracking.preprocessing.exatrkx_cell_features.preprocess_detector(detector: pandas.DataFrame) dict#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

gnn_tracking.preprocessing.exatrkx_cell_features.determine_array_size(detector: pandas.DataFrame) tuple[float, float, float]#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

class gnn_tracking.preprocessing.exatrkx_cell_features.DetectorRotations(detector: pandas.DataFrame)#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

get_rotations()#
_init_rotation_array()#
_extract_all_rotations()#
_extract_rotation_matrix(mod)#

Extract the rotation matrix from module dataframe

class gnn_tracking.preprocessing.exatrkx_cell_features.DetectorThicknesses(detector: pandas.DataFrame)#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

get_thicknesses()#
_init_thickness_array()#
_extract_all_thicknesses()#
class gnn_tracking.preprocessing.exatrkx_cell_features.DetectorPixelSize(detector: pandas.DataFrame)#

Adapted from ExaTrkX’s preprocessing. See docstring above.

get_pixel_size() numpy.ndarray#
_init_size_array()#
_extract_all_size()#
gnn_tracking.preprocessing.exatrkx_cell_features.cartesian_to_spherical(x, y, z)#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

gnn_tracking.preprocessing.exatrkx_cell_features.theta_to_eta(theta)#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

gnn_tracking.preprocessing.exatrkx_cell_features.get_all_local_angles(hits: pandas.DataFrame, cells: pandas.DataFrame, detector: dict) tuple[pandas.Series, pandas.Series, pandas.Series]#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

gnn_tracking.preprocessing.exatrkx_cell_features.get_all_rotated(hits: pandas.DataFrame, detector: dict, l_u: pandas.Series, l_v: pandas.Series, l_w: pandas.Series)#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

gnn_tracking.preprocessing.exatrkx_cell_features.extract_dir_new(hits: pandas.DataFrame, cells: pandas.DataFrame, detector: dict) pandas.DataFrame#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.

gnn_tracking.preprocessing.exatrkx_cell_features.augment_hit_features(hits: pandas.DataFrame, cells: pandas.DataFrame, detector_proc: dict)#

Adapted/copied from ExaTrkX’s preprocessing. See docstring above.