scripts/trackfactory folder#

A folder that contains scripts to plot tracks.

trackfactory.analyse_long_electron_clones.py script#

trackfactory/plot_crossing_tracks.py script#

trackfactory/plot_shared_electron_tracks.py script#

trackfactory.dfutils module#

scripts.trackfactory.dfutils.add_particle_information(truncated_path, df_hits_particles, particle_columns)[source]#

Add particle information to the dataframe of hits-particles.

Return type:

DataFrame

scripts.trackfactory.dfutils.compute_edge_counts(df_edges)[source]#
Return type:

DataFrame

scripts.trackfactory.dfutils.compute_n_particles_hit(df_hits_particles)[source]#
Return type:

DataFrame

scripts.trackfactory.dfutils.find_connected_particle_ids(particle_id, df_hits_particles)[source]#
Return type:

ndarray[Any, dtype[TypeVar(_ScalarType_co, bound= generic, covariant=True)]]

scripts.trackfactory.dfutils.get_df_edges(edge_indices, df_hits_particles)[source]#
Return type:

DataFrame

scripts.trackfactory.dfutils.get_df_hits_particles_from_batch(batch)[source]#

Load dataframe of hits-particles from a PyTorch batch.

Parameters:

batch (Data) – PyTorch Geometric data object

Return type:

DataFrame

Returns:

Dataframe of hits-particles, with columns particle_id and hit_idx, as long with the coordinates x, y and z, the plane number, and the hit_number within the particle and the particle length.

scripts.trackfactory.dfutils.no_shared_edges(hit_indices, df_edges)[source]#
scripts.trackfactory.dfutils.plot_connected_particle_ids(particle_ids, df_hits_particles, df_true_edges, n_plots_max, output_wpath, lhcb=False)[source]#
Return type:

int