gammaloop_api/

integrand_info.rs

use std::fmt;

use color_eyre::Result;
use eyre::{eyre, Context};
use gammalooprs::{
    cff::esurface::EsurfaceExistenceStatus,
    graph::{threshold_counterterms::ThresholdCountertermVariant, FeynmanGraph, Graph},
    integrands::process::{ActiveF64Backend, LmbMultiChannelingSetup, ParamBuilder},
    model::Model,
    processes::{
        Amplitude, CrossSection, CrossSectionCut, CutId, ProcessCollection,
        ThresholdCountertermAssociation, ThresholdCountertermMetadataRegistry,
        ThresholdCountertermMultiplierMetadata, ThresholdCountertermStatus,
    },
    settings::{global::FrozenCompilationMode, runtime::ParameterizationSettings, RuntimeSettings},
    utils::F,
    DependentMomentaConstructor,
};
use linnet::half_edge::involution::{EdgeVec, Orientation};
use schemars::JsonSchema;
use serde::{Deserialize, Serialize};

use crate::state::State;

#[derive(Debug, Clone, Copy, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum IntegrandKind {
    Amplitude,
    CrossSection,
}

impl fmt::Display for IntegrandKind {
    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
        match self {
            Self::Amplitude => f.write_str("amplitude"),
            Self::CrossSection => f.write_str("cross section"),
        }
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandGraphInfo {
    pub graph_id: usize,
    pub name: String,
    pub is_master: bool,
    /// Compact, explicit graph directives, flattened only for display/API consumption.
    ///
    /// Autogenerated legacy associations are deliberately absent. An explicit threshold with no
    /// counterterm blocks contributes one `implicit_default` row because that shorthand is part of
    /// the graph document's semantics.
    #[serde(default, skip_serializing_if = "Vec::is_empty")]
    pub threshold_counterterm_directives: Vec<IntegrandThresholdCountertermDirectiveInfo>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub threshold_counterterms: Option<ThresholdCountertermMetadataRegistry>,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandThresholdCountertermDirectiveInfo {
    pub cut_edge_ids: Vec<usize>,
    pub threshold_edge_ids: Vec<usize>,
    pub name: String,
    pub implicit_default: bool,
    pub requested_subspace: Option<Vec<usize>>,
    pub requested_parent_lmb: Option<Vec<usize>>,
    pub disabled: bool,
    pub multiplier: Option<ThresholdCountertermMultiplierMetadata>,
}

impl IntegrandThresholdCountertermDirectiveInfo {
    fn explicit_for_graph(graph: &Graph) -> Vec<Self> {
        if graph.threshold_counterterms.autogenerated {
            return Vec::new();
        }

        graph
            .threshold_counterterms
            .cuts
            .iter()
            .flat_map(|cut| {
                cut.thresholds.iter().flat_map(move |threshold| {
                    let implicit_default = ThresholdCountertermVariant {
                        name: Some("default".to_string()),
                        group_id: None,
                        subspace: None,
                        parent_lmb: None,
                        disable: false,
                        multiplier: None,
                    };
                    let variants = if threshold.counterterms.is_empty() {
                        std::slice::from_ref(&implicit_default)
                    } else {
                        &threshold.counterterms
                    };

                    variants
                        .iter()
                        .map(move |variant| Self {
                            cut_edge_ids: cut.edges.iter().map(|edge| edge.0).collect(),
                            threshold_edge_ids: threshold.edges.iter().map(|edge| edge.0).collect(),
                            name: variant
                                .name
                                .clone()
                                .unwrap_or_else(|| "default".to_string()),
                            implicit_default: threshold.counterterms.is_empty(),
                            requested_subspace: variant
                                .subspace
                                .as_ref()
                                .map(|edges| edges.iter().map(|edge| edge.0).collect()),
                            requested_parent_lmb: variant
                                .parent_lmb
                                .as_ref()
                                .map(|edges| edges.iter().map(|edge| edge.0).collect()),
                            disabled: variant.disable,
                            multiplier: variant.multiplier.as_ref().map(|multiplier| {
                                ThresholdCountertermMultiplierMetadata {
                                    expression: multiplier.expression.clone(),
                                    function_map: graph
                                        .threshold_counterterms
                                        .function_map
                                        .iter()
                                        .chain(&multiplier.function_map)
                                        .map(|(key, value)| (key.clone(), value.clone()))
                                        .collect(),
                                    symmetrize: multiplier.symmetrize,
                                    opaque_derivatives: multiplier.opaque_derivatives,
                                }
                            }),
                        })
                        .collect::<Vec<_>>()
                })
            })
            .collect()
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandOrientationInfo {
    pub orientation_id: usize,
    pub signature: Vec<i8>,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandLoopMomentumBasisInfo {
    pub basis_id: usize,
    pub channel_id: Option<usize>,
    pub edge_ids: Vec<usize>,
    pub matches_generation_basis: bool,
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandCutInfo {
    pub cut_id: usize,
    pub edge_ids: Vec<usize>,
    pub raising_power: usize,
    pub left_thresholds: Vec<IntegrandCutThresholdInfo>,
    pub right_thresholds: Vec<IntegrandCutThresholdInfo>,
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum IntegrandThresholdStatus {
    NoRadialDependence,
    AlwaysPinched,
    CanBecomePinched,
    ProvenNonExisting,
    PotentiallyExisting,
}

impl IntegrandThresholdStatus {
    pub fn as_str(self) -> &'static str {
        match self {
            Self::NoRadialDependence => "no_radial_dependence",
            Self::AlwaysPinched => "always_pinched",
            Self::CanBecomePinched => "can_become_pinched",
            Self::ProvenNonExisting => "proven_non_existing",
            Self::PotentiallyExisting => "potentially_existing",
        }
    }

    pub fn is_currently_viable(self) -> bool {
        matches!(self, Self::CanBecomePinched | Self::PotentiallyExisting)
    }

    pub fn can_become_pinched(self) -> bool {
        self == Self::CanBecomePinched
    }
}

impl From<ThresholdCountertermStatus> for IntegrandThresholdStatus {
    fn from(value: ThresholdCountertermStatus) -> Self {
        match value {
            ThresholdCountertermStatus::NoRadialDependence => Self::NoRadialDependence,
            ThresholdCountertermStatus::AlwaysPinched => Self::AlwaysPinched,
            ThresholdCountertermStatus::CanBecomePinched => Self::CanBecomePinched,
            ThresholdCountertermStatus::ProvenNonExisting => Self::ProvenNonExisting,
            ThresholdCountertermStatus::PotentiallyExisting => Self::PotentiallyExisting,
        }
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandCutThresholdInfo {
    pub esurface_id: usize,
    pub status: IntegrandThresholdStatus,
    pub cut_boundary_edge_ids: Vec<usize>,
    pub threshold_boundary_edge_ids: Vec<usize>,
    pub invariant_bound_is_applicable: bool,
}

impl IntegrandCutThresholdInfo {
    fn from_association(
        association: &ThresholdCountertermAssociation,
        graph: &Graph,
        cut: &CrossSectionCut,
        model: &Model,
        param_builder: &ParamBuilder,
        settings: &RuntimeSettings,
    ) -> Self {
        Self {
            esurface_id: association.topological_threshold_id.0,
            status: association
                .classify_for_model(
                    graph,
                    cut,
                    model,
                    param_builder,
                    settings,
                    settings.subtraction.esurface_existence_threshold,
                )
                .into(),
            cut_boundary_edge_ids: association
                .cut_boundary_edges
                .iter()
                .map(|edge_id| edge_id.0)
                .collect(),
            threshold_boundary_edge_ids: association
                .threshold_boundary_edges
                .iter()
                .map(|edge_id| edge_id.0)
                .collect(),
            invariant_bound_is_applicable: association.invariant_bound_is_applicable,
        }
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandActiveThresholdCutInfo {
    pub cut_id: usize,
    pub can_become_pinched: bool,
}

#[derive(Debug, Clone, Copy, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
#[serde(rename_all = "snake_case")]
pub enum IntegrandEsurfaceClassification {
    NonExisting,
    Pinched,
    Existing,
}

impl IntegrandEsurfaceClassification {
    pub fn as_str(self) -> &'static str {
        match self {
            Self::NonExisting => "non_existing",
            Self::Pinched => "pinched",
            Self::Existing => "existing",
        }
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandThresholdEsurfaceInfo {
    pub esurface_id: usize,
    pub representative_graph_id: usize,
    pub edge_ids: Vec<usize>,
    pub classification: Option<IntegrandEsurfaceClassification>,
    pub active_cuts: Vec<IntegrandActiveThresholdCutInfo>,
}

impl IntegrandCutInfo {
    fn from_cross_section_cut(
        graph_term: &gammalooprs::integrands::process::cross_section::CrossSectionGraphTerm,
        cut_id: CutId,
        cut: &CrossSectionCut,
        raising_power: usize,
        model: &Model,
        param_builder: &ParamBuilder,
        settings: &RuntimeSettings,
    ) -> Self {
        let associations = &graph_term.cut_threshold_associations[cut_id];
        let threshold_info = |association| {
            IntegrandCutThresholdInfo::from_association(
                association,
                &graph_term.graph,
                cut,
                model,
                param_builder,
                settings,
            )
        };

        Self {
            // Retain `CutId` for internal indexing and convert only at the API boundary.
            cut_id: usize::from(cut_id),
            edge_ids: cut_edge_ids(&graph_term.graph, cut),
            raising_power,
            left_thresholds: associations.left.iter().map(&threshold_info).collect(),
            right_thresholds: associations.right.iter().map(threshold_info).collect(),
        }
    }

    fn active_threshold_cut(&self, esurface_id: usize) -> Option<IntegrandActiveThresholdCutInfo> {
        let mut is_currently_viable = false;
        let mut can_become_pinched = false;

        for threshold in self
            .left_thresholds
            .iter()
            .chain(self.right_thresholds.iter())
            .filter(|threshold| threshold.esurface_id == esurface_id)
        {
            is_currently_viable |= threshold.status.is_currently_viable();
            can_become_pinched |= threshold.status.can_become_pinched();
        }

        is_currently_viable.then_some(IntegrandActiveThresholdCutInfo {
            cut_id: self.cut_id,
            can_become_pinched,
        })
    }
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandGraphGroupInfo {
    pub group_id: usize,
    pub graphs: Vec<IntegrandGraphInfo>,
    pub orientation_edge_ids: Vec<usize>,
    pub orientations: Vec<IntegrandOrientationInfo>,
    pub loop_momentum_bases: Vec<IntegrandLoopMomentumBasisInfo>,
    pub threshold_esurface_ids: Vec<usize>,
    pub threshold_esurfaces: Vec<IntegrandThresholdEsurfaceInfo>,
    pub cuts: Vec<IntegrandCutInfo>,
}

fn threshold_esurface_edge_ids(
    esurfaces: &gammalooprs::cff::esurface::EsurfaceCollection,
    esurface_id: usize,
) -> Vec<usize> {
    esurfaces
        .iter()
        .nth(esurface_id)
        .expect("threshold esurface id should resolve in collection")
        .energies
        .iter()
        .map(|edge_id| edge_id.0)
        .collect()
}

#[derive(Debug, Clone, Serialize, Deserialize, JsonSchema, PartialEq, Eq)]
pub struct IntegrandInfo {
    /// Numeric identifier of the process that owns the integrand.
    pub process_id: usize,
    /// Persistent folder name of the owning process.
    pub process_name: String,
    /// Canonical name of the resolved integrand.
    pub integrand_name: String,
    /// Whether the integrand represents an amplitude or cross section.
    pub kind: IntegrandKind,
    /// Compilation mode frozen into the generated integrand.
    pub generation_compilation: FrozenCompilationMode,
    /// Floating-point evaluator backend currently active for the integrand.
    pub active_f64_backend: ActiveF64Backend,
    /// Number of amplitude graphs or cross-section supergraphs.
    pub graph_count: usize,
    /// Number of graph groups in the generated integrand.
    pub graph_group_count: usize,
    /// Bincode size of the saved integrand record, excluding its compiled evaluator.
    pub record_size_bytes: usize,
    /// Per-group graphs, orientations, loop-momentum bases, thresholds, and cuts.
    pub graph_groups: Vec<IntegrandGraphGroupInfo>,
}

pub(crate) fn collect_integrand_info(
    state: &State,
    process_id: usize,
    integrand_name: &str,
) -> Result<IntegrandInfo> {
    let process = &state.process_list.processes[process_id];
    let resolved = process.get_integrand(integrand_name)?;
    let generated = resolved.require_generated()?;
    let generation_compilation = generated.frozen_compilation().clone();
    let active_f64_backend = generated.active_f64_backend();

    match (generated, &process.collection) {
        (
            gammalooprs::integrands::process::ProcessIntegrand::Amplitude(integrand),
            ProcessCollection::Amplitudes(amplitudes),
        ) => {
            let amplitude = amplitudes.get(&resolved.canonical_name).ok_or_else(|| {
                eyre!(
                    "Could not resolve amplitude record '{}' in process #{} ({})",
                    resolved.canonical_name,
                    process.definition.process_id,
                    process.definition.folder_name
                )
            })?;
            let model = state.resolve_model_for_integrand(
                process.definition.process_id,
                &resolved.canonical_name,
            )?;
            Ok(IntegrandInfo {
                process_id: process.definition.process_id,
                process_name: process.definition.folder_name.clone(),
                integrand_name: resolved.canonical_name,
                kind: IntegrandKind::Amplitude,
                generation_compilation: generation_compilation.clone(),
                active_f64_backend,
                graph_count: amplitude.graphs.len(),
                graph_group_count: amplitude.graph_group_structure.len(),
                record_size_bytes: integrand_record_size_from_amplitude(amplitude)?,
                graph_groups: amplitude_graph_groups(integrand, &model)?,
            })
        }
        (
            gammalooprs::integrands::process::ProcessIntegrand::CrossSection(integrand),
            ProcessCollection::CrossSections(cross_sections),
        ) => {
            let cross_section = cross_sections
                .get(&resolved.canonical_name)
                .ok_or_else(|| {
                    eyre!(
                        "Could not resolve cross-section record '{}' in process #{} ({})",
                        resolved.canonical_name,
                        process.definition.process_id,
                        process.definition.folder_name
                    )
                })?;
            let model = state.resolve_model_for_integrand(
                process.definition.process_id,
                &resolved.canonical_name,
            )?;
            Ok(IntegrandInfo {
                process_id: process.definition.process_id,
                process_name: process.definition.folder_name.clone(),
                integrand_name: resolved.canonical_name,
                kind: IntegrandKind::CrossSection,
                generation_compilation,
                active_f64_backend,
                graph_count: cross_section.supergraphs.len(),
                graph_group_count: cross_section.graph_group_structure.len(),
                record_size_bytes: integrand_record_size_from_cross_section(cross_section)?,
                graph_groups: cross_section_graph_groups(integrand, &model)?,
            })
        }
        _ => Err(eyre!(
            "Process/integrand type mismatch for '{}' in process #{} ({})",
            resolved.canonical_name,
            process.definition.process_id,
            process.definition.folder_name
        )),
    }
}

fn orientation_signature(orientation: &EdgeVec<Orientation>) -> Vec<i8> {
    orientation
        .iter()
        .map(|(_, orientation)| match *orientation {
            Orientation::Default => 1,
            Orientation::Reversed => -1,
            Orientation::Undirected => 0,
        })
        .collect()
}

fn orientation_edge_ids(orientation: &EdgeVec<Orientation>) -> Vec<usize> {
    orientation.iter().map(|(edge_id, _)| edge_id.0).collect()
}

fn cut_edge_ids(graph: &Graph, cut: &CrossSectionCut) -> Vec<usize> {
    cut.cut
        .iter_edges(&graph.underlying)
        .map(|(_, edge)| {
            graph
                .underlying
                .iter_edges()
                .find_map(|(_, edge_id, edge_data)| {
                    (edge_data.data.name == edge.data.name).then_some(edge_id.0)
                })
                .expect("cut edge should resolve in graph")
        })
        .collect()
}

fn cut_raising_powers(
    graph: &gammalooprs::integrands::process::cross_section::CrossSectionGraphTerm,
) -> typed_index_collections::TiVec<CutId, usize> {
    let mut raising_powers: typed_index_collections::TiVec<CutId, usize> =
        vec![1; graph.cuts.len()].into();
    for cut_group in graph.cut_group_data.cut_groups.iter() {
        let raising_power = cut_group.related_esurface_group.max_occurence;
        for cut_id in &cut_group.cuts {
            raising_powers[*cut_id] = raising_power;
        }
    }
    raising_powers
}

fn sampling_channel_ids(
    lmbs: &typed_index_collections::TiVec<
        gammalooprs::graph::LmbIndex,
        gammalooprs::graph::LoopMomentumBasis,
    >,
    multi_channeling_setup: &LmbMultiChannelingSetup,
    graph_name: &str,
    parameterization_settings: &ParameterizationSettings,
) -> Result<Vec<Option<usize>>> {
    let mut channel_ids = vec![None; lmbs.len()];
    for channel_id in
        multi_channeling_setup.sampling_channel_ids(graph_name, parameterization_settings)?
    {
        if let Some(lmb_index) = multi_channeling_setup.sampling_channel_lmb_id(
            channel_id,
            graph_name,
            parameterization_settings,
        )? {
            channel_ids[usize::from(lmb_index)] = Some(channel_id.index());
        }
    }
    Ok(channel_ids)
}

fn amplitude_graph_groups(
    integrand: &gammalooprs::integrands::process::amplitude::AmplitudeIntegrand,
    model: &Model,
) -> Result<Vec<IntegrandGraphGroupInfo>> {
    let parameterization_settings = integrand
        .settings
        .sampling
        .get_parameterization_settings()
        .unwrap_or_default();
    let external_momenta = integrand
        .settings
        .kinematics
        .externals
        .get_dependent_externals::<f64>(DependentMomentaConstructor::Amplitude(
            &integrand.data.external_signature,
        ))
        .context("While constructing amplitude external momenta for integrand information")?;
    let e_cm = F(integrand.settings.kinematics.e_cm);
    let existence_tolerance = F(integrand.settings.subtraction.esurface_existence_threshold);
    let real_mass_vectors = integrand
        .data
        .graph_terms
        .iter()
        .map(|graph_term| graph_term.graph.get_real_mass_vector::<f64>(model))
        .collect::<Vec<_>>();
    integrand
        .data
        .graph_group_structure
        .iter_enumerated()
        .map(|(group_id, group)| {
            let master_graph_id = group
                .into_iter()
                .next()
                .expect("graph group should not be empty");
            let master_graph = &integrand.data.graph_terms[master_graph_id];
            let channel_ids = sampling_channel_ids(
                &master_graph.lmbs,
                &master_graph.multi_channeling_setup,
                &master_graph.graph.name,
                &parameterization_settings,
            )?;
            // Amplitude overlap centers are solved from the canonical union across all members
            // of the graph group. Report that same union rather than the master graph's local
            // raised-surface numbering.
            let threshold_esurfaces = integrand.data.group_derived_data[group_id]
                .esurface_map
                .iter_enumerated()
                .filter_map(|(group_esurface_id, raised_esurface_map)| {
                    let (representative_graph_id, raised_esurface_id) = raised_esurface_map
                        .iter_enumerated()
                        .filter_map(|(graph_group_position, raised_esurface_id)| {
                            raised_esurface_id.map(|raised_esurface_id| {
                                let graph_id = group[graph_group_position];
                                (graph_id, raised_esurface_id)
                            })
                        })
                        .find(|(graph_id, raised_esurface_id)| {
                            integrand.data.graph_terms[*graph_id]
                                .threshold_counterterm
                                .generated_mask[*raised_esurface_id]
                        })?;
                    let graph_term = &integrand.data.graph_terms[representative_graph_id];
                    let local_esurface_id =
                        graph_term.threshold_counterterm.raised_data.raised_groups
                            [raised_esurface_id]
                            .esurface_ids[0];

                    let mut classification = IntegrandEsurfaceClassification::NonExisting;
                    for (graph_group_position, raised_esurface_id) in raised_esurface_map
                        .iter_enumerated()
                        .filter_map(|(graph_group_position, raised_esurface_id)| {
                            raised_esurface_id.map(|raised_esurface_id| {
                                (graph_group_position, raised_esurface_id)
                            })
                        })
                    {
                        let graph_id = group[graph_group_position];
                        let candidate = &integrand.data.graph_terms[graph_id];
                        let candidate_esurface_id =
                            candidate.threshold_counterterm.raised_data.raised_groups
                                [raised_esurface_id]
                                .esurface_ids[0];
                        let candidate_status = candidate.esurfaces[candidate_esurface_id]
                            .existence_status(
                                &external_momenta,
                                &candidate.graph.loop_momentum_basis,
                                &real_mass_vectors[graph_id],
                                &e_cm,
                                &existence_tolerance,
                            );
                        match candidate_status {
                            EsurfaceExistenceStatus::Existing => {
                                classification = IntegrandEsurfaceClassification::Existing;
                                break;
                            }
                            EsurfaceExistenceStatus::Pinched => {
                                classification = IntegrandEsurfaceClassification::Pinched;
                            }
                            EsurfaceExistenceStatus::NonExisting => {}
                        }
                    }

                    Some(IntegrandThresholdEsurfaceInfo {
                        esurface_id: group_esurface_id.0,
                        representative_graph_id,
                        edge_ids: threshold_esurface_edge_ids(
                            &graph_term.esurfaces,
                            local_esurface_id.0,
                        ),
                        classification: Some(classification),
                        active_cuts: Vec::new(),
                    })
                })
                .collect::<Vec<_>>();
            let threshold_esurface_ids = threshold_esurfaces
                .iter()
                .map(|threshold| threshold.esurface_id)
                .collect();
            Ok(IntegrandGraphGroupInfo {
                group_id: usize::from(group_id),
                graphs: group
                    .into_iter()
                    .map(|graph_id| {
                        let graph_term = &integrand.data.graph_terms[graph_id];
                        IntegrandGraphInfo {
                            graph_id,
                            name: graph_term.graph.name.clone(),
                            is_master: graph_id == master_graph_id,
                            threshold_counterterm_directives:
                                IntegrandThresholdCountertermDirectiveInfo::explicit_for_graph(
                                    &graph_term.graph,
                                ),
                            threshold_counterterms: graph_term
                                .threshold_counterterm_metadata()
                                .cloned(),
                        }
                    })
                    .collect(),
                orientation_edge_ids: master_graph
                    .orientations
                    .first()
                    .map(orientation_edge_ids)
                    .unwrap_or_default(),
                orientations: master_graph
                    .orientations
                    .iter()
                    .enumerate()
                    .map(|(orientation_id, orientation)| IntegrandOrientationInfo {
                        orientation_id,
                        signature: orientation_signature(orientation),
                    })
                    .collect(),
                loop_momentum_bases: master_graph
                    .lmbs
                    .iter_enumerated()
                    .map(|(basis_id, lmb)| IntegrandLoopMomentumBasisInfo {
                        basis_id: usize::from(basis_id),
                        channel_id: channel_ids[usize::from(basis_id)],
                        edge_ids: lmb.loop_edges.iter().map(|edge_id| edge_id.0).collect(),
                        matches_generation_basis: lmb.loop_edges.len()
                            == master_graph.graph.loop_momentum_basis.loop_edges.len()
                            && lmb.loop_edges.iter().all(|edge_id| {
                                master_graph
                                    .graph
                                    .loop_momentum_basis
                                    .loop_edges
                                    .contains(edge_id)
                            }),
                    })
                    .collect(),
                threshold_esurface_ids,
                threshold_esurfaces,
                cuts: Vec::new(),
            })
        })
        .collect()
}

fn cross_section_graph_groups(
    integrand: &gammalooprs::integrands::process::cross_section::CrossSectionIntegrand,
    model: &Model,
) -> Result<Vec<IntegrandGraphGroupInfo>> {
    let parameterization_settings = integrand
        .settings
        .sampling
        .get_parameterization_settings()
        .unwrap_or_default();
    integrand
        .data
        .graph_group_structure
        .iter()
        .enumerate()
        .map(|(group_id, group)| {
            let master_graph_id = group
                .into_iter()
                .next()
                .expect("graph group should not be empty");
            let master_graph = &integrand.data.graph_terms[master_graph_id];
            let mut active_model_param_builder: ParamBuilder =
                master_graph.graph.param_builder.clone();
            active_model_param_builder.update_model_values(model)?;
            let channel_ids = sampling_channel_ids(
                &master_graph.lmbs,
                &master_graph.multi_channeling_setup,
                &master_graph.graph.name,
                &parameterization_settings,
            )?;
            let cut_raising_powers = cut_raising_powers(master_graph);

            let cuts = master_graph
                .cuts
                .iter_enumerated()
                .map(|(cut_id, cut)| {
                    IntegrandCutInfo::from_cross_section_cut(
                        master_graph,
                        cut_id,
                        cut,
                        cut_raising_powers[cut_id],
                        model,
                        &active_model_param_builder,
                        &integrand.settings,
                    )
                })
                .collect::<Vec<_>>();

            let threshold_esurfaces = master_graph
                .topological_threshold_esurfaces
                .iter_enumerated()
                .map(|(topological_threshold_id, esurface)| {
                    let esurface_id = topological_threshold_id.0;
                    let active_cuts = cuts
                        .iter()
                        .filter_map(|cut| cut.active_threshold_cut(esurface_id))
                        .collect();

                    IntegrandThresholdEsurfaceInfo {
                        esurface_id,
                        representative_graph_id: master_graph_id,
                        edge_ids: esurface.energies.iter().map(|edge_id| edge_id.0).collect(),
                        classification: None,
                        active_cuts,
                    }
                })
                .collect::<Vec<_>>();
            let threshold_esurface_ids = threshold_esurfaces
                .iter()
                .map(|threshold| threshold.esurface_id)
                .collect();

            Ok(IntegrandGraphGroupInfo {
                group_id,
                graphs: group
                    .into_iter()
                    .map(|graph_id| {
                        let graph_term = &integrand.data.graph_terms[graph_id];
                        IntegrandGraphInfo {
                            graph_id,
                            name: graph_term.graph.name.clone(),
                            is_master: graph_id == master_graph_id,
                            threshold_counterterm_directives:
                                IntegrandThresholdCountertermDirectiveInfo::explicit_for_graph(
                                    &graph_term.graph,
                                ),
                            threshold_counterterms: graph_term
                                .threshold_counterterm_metadata()
                                .cloned(),
                        }
                    })
                    .collect(),
                orientation_edge_ids: master_graph
                    .orientations
                    .first()
                    .map(orientation_edge_ids)
                    .unwrap_or_default(),
                orientations: master_graph
                    .orientations
                    .iter()
                    .enumerate()
                    .map(|(orientation_id, orientation)| IntegrandOrientationInfo {
                        orientation_id,
                        signature: orientation_signature(orientation),
                    })
                    .collect(),
                loop_momentum_bases: master_graph
                    .lmbs
                    .iter_enumerated()
                    .map(|(basis_id, lmb)| IntegrandLoopMomentumBasisInfo {
                        basis_id: usize::from(basis_id),
                        channel_id: channel_ids[usize::from(basis_id)],
                        edge_ids: lmb.loop_edges.iter().map(|edge_id| edge_id.0).collect(),
                        matches_generation_basis: lmb.loop_edges.len()
                            == master_graph.graph.loop_momentum_basis.loop_edges.len()
                            && lmb.loop_edges.iter().all(|edge_id| {
                                master_graph
                                    .graph
                                    .loop_momentum_basis
                                    .loop_edges
                                    .contains(edge_id)
                            }),
                    })
                    .collect(),
                threshold_esurface_ids,
                threshold_esurfaces,
                cuts,
            })
        })
        .collect()
}

fn integrand_record_size_from_amplitude(amplitude: &Amplitude) -> Result<usize> {
    let mut record = amplitude.clone();
    record.integrand = None;
    let encoded =
        bincode::encode_to_vec(&record, bincode::config::standard()).with_context(|| {
            format!(
                "While serializing amplitude '{}' for integrand info",
                amplitude.name
            )
        })?;
    Ok(encoded.len())
}

fn integrand_record_size_from_cross_section(cross_section: &CrossSection) -> Result<usize> {
    let mut record = cross_section.clone();
    record.integrand = None;
    let encoded =
        bincode::encode_to_vec(&record, bincode::config::standard()).with_context(|| {
            format!(
                "While serializing cross-section '{}' for integrand info",
                cross_section.name
            )
        })?;
    Ok(encoded.len())
}