graphnet.data.extractors.prometheus.prometheus_extractor module

Parquet Extractor for conversion of simulation files from PROMETHEUS.

class graphnet.data.extractors.prometheus.prometheus_extractor.PrometheusExtractor(extractor_name, columns)[source]

Bases: Extractor

Class for extracting information from PROMETHEUS parquet files.

Contains functionality required to extract data from PROMETHEUS parquet files.

Construct PrometheusExtractor.

Parameters:

  • extractor_name (str) – Name of the PrometheusExtractor instance.

  • data (Used to keep track of the provenance of different)

  • columns (List[str])

:param : :param and to name tables to which this data is saved.: :type columns: List[str] :param columns: List of column names to extract from the table.

class graphnet.data.extractors.prometheus.prometheus_extractor.PrometheusTruthExtractor(table_name)[source]

Bases: PrometheusExtractor

Class for extracting event level truth from Prometheus parquet files.

This Extractor will “initial_state” i.e. neutrino truth.

Construct PrometheusTruthExtractor.

Parameters:

table_name (str, default: 'mc_truth') – Name of the table in the parquet files that contain event-level truth. Defaults to “mc_truth”.

class graphnet.data.extractors.prometheus.prometheus_extractor.PrometheusFeatureExtractor(table_name)[source]

Bases: PrometheusExtractor

Class for extracting pulses/photons from Prometheus parquet files.

Construct PrometheusFeatureExtractor.

Parameters:

table_name (str, default: 'photons') – Name of table in parquet files that contain the photons/pulses. Defaults to “photons”.