gnn_tracking.preprocessing.exatrkx_cell_features

Module Contents

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

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.

Attributes

cf_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.