dfir_lang/graph/

flat_graph_builder.rs

//! Build a flat graph from [`HfStatement`]s.

use std::borrow::Cow;
use std::collections::btree_map::Entry;
use std::collections::{BTreeMap, BTreeSet};

use itertools::Itertools;
use proc_macro2::Span;
use quote::ToTokens;
use syn::spanned::Spanned;
use syn::{Error, Ident, ItemUse};

use crate::diagnostic::{Diagnostic, Diagnostics, Level};
use crate::graph::ops::next_iteration::NEXT_ITERATION;
use crate::graph::ops::{FloType, Persistence, PortListSpec, RangeTrait};
use crate::graph::{
    DfirGraph, GraphEdgeId, GraphLoopId, GraphNode, GraphNodeId, HandoffKind, PortIndexValue,
    graph_algorithms,
};
use crate::parse::{DfirCode, DfirStatement, Operator, Pipeline};
use crate::pretty_span::PrettySpan;

#[derive(Clone, Debug)]
struct Ends {
    inn: Option<(PortIndexValue, GraphDet)>,
    out: Option<(PortIndexValue, GraphDet)>,
}

#[derive(Clone, Debug)]
enum GraphDet {
    Determined(GraphNodeId),
    Undetermined(Ident),
}

/// Variable name info for each ident, see [`FlatGraphBuilder::varname_ends`].
#[derive(Debug)]
struct VarnameInfo {
    /// What the variable name resolves to.
    pub ends: Ends,
    /// Set to true if the varname reference creates an illegal self-referential cycle.
    pub illegal_cycle: bool,
    /// Set to true once the in port is used. Used to track unused ports.
    pub inn_used: bool,
    /// Set to true once the out port is used. Used to track unused ports.
    pub out_used: bool,
}
impl VarnameInfo {
    pub fn new(ends: Ends) -> Self {
        Self {
            ends,
            illegal_cycle: false,
            inn_used: false,
            out_used: false,
        }
    }
}

/// Wraper around [`DfirGraph`] to build a flat graph from AST code.
#[derive(Debug, Default)]
pub struct FlatGraphBuilder {
    /// Spanned error/warning/etc diagnostics to emit.
    diagnostics: Diagnostics,

    /// [`DfirGraph`] being built.
    flat_graph: DfirGraph,
    /// Variable names, used as [`HfStatement::Named`] are added.
    varname_ends: BTreeMap<Ident, VarnameInfo>,
    /// Each (out -> inn) link inputted.
    links: Vec<Ends>,

    /// Use statements.
    uses: Vec<ItemUse>,

    /// If the flat graph is being loaded as a module, then two initial ModuleBoundary nodes are inserted into the graph. One
    /// for the input into the module and one for the output out of the module.
    module_boundary_nodes: Option<(GraphNodeId, GraphNodeId)>,
}

/// Output of [`FlatGraphBuilder::build`].
pub struct FlatGraphBuilderOutput {
    /// The flat DFIR graph.
    pub flat_graph: DfirGraph,
    /// Any `use` statements.
    pub uses: Vec<ItemUse>,
    /// Any emitted diagnostics (warnings, errors).
    pub diagnostics: Diagnostics,
}

impl FlatGraphBuilder {
    /// Create a new empty graph builder.
    pub fn new() -> Self {
        Default::default()
    }

    /// Convert the DFIR code AST into a graph builder.
    pub fn from_dfir(input: DfirCode) -> Self {
        let mut builder = Self::default();
        builder.add_dfir(input, None, None);
        builder
    }

    /// Build into an unpartitioned [`DfirGraph`], returning a struct containing the flat graph, any diagnostics, and
    /// other outputs.
    ///
    /// If any diagnostics are errors, `Err` is returned and the underlying graph is lost.
    pub fn build(mut self) -> Result<FlatGraphBuilderOutput, Diagnostics> {
        self.finalize_connect_operator_links();
        self.process_operator_errors();

        if self.diagnostics.has_error() {
            Err(self.diagnostics)
        } else {
            Ok(FlatGraphBuilderOutput {
                flat_graph: self.flat_graph,
                uses: self.uses,
                diagnostics: self.diagnostics,
            })
        }
    }

    /// Adds all [`DfirStatement`]s within the [`DfirCode`] to this [`DfirGraph`].
    ///
    /// Optional configuration:
    /// * In the given loop context `current_loop`.
    /// * With the given operator tag `operator_tag`.
    pub fn add_dfir(
        &mut self,
        dfir: DfirCode,
        current_loop: Option<GraphLoopId>,
        operator_tag: Option<&str>,
    ) {
        for stmt in dfir.statements {
            self.add_statement_internal(stmt, current_loop, operator_tag);
        }
    }

    /// Add a single [`DfirStatement`] line to this [`DfirGraph`] in the root context.
    pub fn add_statement(&mut self, stmt: DfirStatement) {
        self.add_statement_internal(stmt, None, None);
    }

    /// Add a single [`DfirStatement`] line to this [`DfirGraph`] with given configuration.
    ///
    /// Optional configuration:
    /// * In the given loop context `current_loop`.
    /// * With the given operator tag `operator_tag`.
    fn add_statement_internal(
        &mut self,
        stmt: DfirStatement,
        current_loop: Option<GraphLoopId>,
        operator_tag: Option<&str>,
    ) {
        match stmt {
            DfirStatement::Use(yuse) => {
                self.uses.push(yuse);
            }
            DfirStatement::Named(named) => {
                let stmt_span = named.span();
                let ends = self.add_pipeline(
                    named.pipeline,
                    Some(&named.name),
                    current_loop,
                    operator_tag,
                );
                self.assign_varname_checked(named.name, stmt_span, ends);
            }
            DfirStatement::Pipeline(pipeline_stmt) => {
                let ends =
                    self.add_pipeline(pipeline_stmt.pipeline, None, current_loop, operator_tag);
                Self::helper_check_unused_port(&mut self.diagnostics, &ends, true);
                Self::helper_check_unused_port(&mut self.diagnostics, &ends, false);
            }
            DfirStatement::Loop(loop_statement) => {
                let inner_loop = self.flat_graph.insert_loop(current_loop);
                for stmt in loop_statement.statements {
                    self.add_statement_internal(stmt, Some(inner_loop), operator_tag);
                }
            }
        }
    }

    /// Programatically add an pipeline, optionally adding `pred_name` as a single predecessor and
    /// assigning it all to `asgn_name`.
    ///
    /// In DFIR syntax, equivalent to [`Self::add_statement`] of (if all names are supplied):
    /// ```text
    /// #asgn_name = #pred_name -> #pipeline;
    /// ```
    ///
    /// But with, optionally:
    /// * A `current_loop` to put the operator in.
    /// * An `operator_tag` to tag the operator with, for debugging/tracing.
    pub fn append_assign_pipeline(
        &mut self,
        asgn_name: Option<&Ident>,
        pred_name: Option<&Ident>,
        pipeline: Pipeline,
        current_loop: Option<GraphLoopId>,
        operator_tag: Option<&str>,
    ) {
        let span = pipeline.span();
        let mut ends = self.add_pipeline(pipeline, asgn_name, current_loop, operator_tag);

        // Connect `pred_name` if supplied.
        if let Some(pred_name) = pred_name {
            if let Some(pred_varname_info) = self.varname_ends.get(pred_name) {
                // Update ends for `asgn_name`.
                ends = self.connect_ends(pred_varname_info.ends.clone(), ends);
            } else {
                self.diagnostics.push(Diagnostic::spanned(
                    pred_name.span(),
                    Level::Error,
                    format!(
                        "Cannot find referenced name `{}`; name was never assigned.",
                        pred_name
                    ),
                ));
            }
        }

        // Assign `asgn_name` if supplied.
        if let Some(asgn_name) = asgn_name {
            self.assign_varname_checked(asgn_name.clone(), span, ends);
        }
    }
}

/// Internal methods.
impl FlatGraphBuilder {
    /// Assign a variable name to a pipeline, checking for conflicts.
    fn assign_varname_checked(&mut self, name: Ident, stmt_span: Span, ends: Ends) {
        match self.varname_ends.entry(name) {
            Entry::Vacant(vacant_entry) => {
                vacant_entry.insert(VarnameInfo::new(ends));
            }
            Entry::Occupied(occupied_entry) => {
                let prev_conflict = occupied_entry.key();
                self.diagnostics.push(Diagnostic::spanned(
                    prev_conflict.span(),
                    Level::Error,
                    format!(
                        "Existing assignment to `{}` conflicts with later assignment: {} (1/2)",
                        prev_conflict,
                        PrettySpan(stmt_span),
                    ),
                ));
                self.diagnostics.push(Diagnostic::spanned(
                    stmt_span,
                    Level::Error,
                    format!(
                        "Name assignment to `{}` conflicts with existing assignment: {} (2/2)",
                        prev_conflict,
                        PrettySpan(prev_conflict.span())
                    ),
                ));
            }
        }
    }

    /// Helper: Add a pipeline, i.e. `a -> b -> c`. Return the input and output [`Ends`] for it.
    fn add_pipeline(
        &mut self,
        pipeline: Pipeline,
        current_varname: Option<&Ident>,
        current_loop: Option<GraphLoopId>,
        operator_tag: Option<&str>,
    ) -> Ends {
        match pipeline {
            Pipeline::Paren(ported_pipeline_paren) => {
                let (inn_port, pipeline_paren, out_port) =
                    PortIndexValue::from_ported(ported_pipeline_paren);
                let og_ends = self.add_pipeline(
                    *pipeline_paren.pipeline,
                    current_varname,
                    current_loop,
                    operator_tag,
                );
                Self::helper_combine_ends(&mut self.diagnostics, og_ends, inn_port, out_port)
            }
            Pipeline::Name(pipeline_name) => {
                let (inn_port, ident, out_port) = PortIndexValue::from_ported(pipeline_name);

                // Mingwei: We could lookup non-forward references immediately, but easier to just
                // have one consistent code path: `GraphDet::Undetermined`.
                Ends {
                    inn: Some((inn_port, GraphDet::Undetermined(ident.clone()))),
                    out: Some((out_port, GraphDet::Undetermined(ident))),
                }
            }
            Pipeline::ModuleBoundary(pipeline_name) => {
                let Some((input_node, output_node)) = self.module_boundary_nodes else {
                    self.diagnostics.push(
                        Error::new(
                            pipeline_name.span(),
                            "`mod` is only usable inside of a module.",
                        )
                        .into(),
                    );

                    return Ends {
                        inn: None,
                        out: None,
                    };
                };

                let (inn_port, _, out_port) = PortIndexValue::from_ported(pipeline_name);

                Ends {
                    inn: Some((inn_port, GraphDet::Determined(output_node))),
                    out: Some((out_port, GraphDet::Determined(input_node))),
                }
            }
            Pipeline::Link(pipeline_link) => {
                // Add the nested LHS and RHS of this link.
                let lhs_ends = self.add_pipeline(
                    *pipeline_link.lhs,
                    current_varname,
                    current_loop,
                    operator_tag,
                );
                let rhs_ends = self.add_pipeline(
                    *pipeline_link.rhs,
                    current_varname,
                    current_loop,
                    operator_tag,
                );

                self.connect_ends(lhs_ends, rhs_ends)
            }
            Pipeline::Operator(operator) => {
                let op_span = Some(operator.span());
                let (node_id, ends) =
                    self.add_operator(current_varname, current_loop, operator, op_span);
                if let Some(operator_tag) = operator_tag {
                    self.flat_graph
                        .set_operator_tag(node_id, operator_tag.to_owned());
                }
                ends
            }
        }
    }

    /// Connects two [`Ends`] together. Returns the outer [`Ends`] for the connection.
    ///
    /// Links the inner ends together by adding it to `self.links`.
    fn connect_ends(&mut self, lhs_ends: Ends, rhs_ends: Ends) -> Ends {
        // Outer (first and last) ends.
        let outer_ends = Ends {
            inn: lhs_ends.inn,
            out: rhs_ends.out,
        };
        // Inner (link) ends.
        let link_ends = Ends {
            out: lhs_ends.out,
            inn: rhs_ends.inn,
        };
        self.links.push(link_ends);
        outer_ends
    }

    /// Adds an operator to the graph, returning its [`GraphNodeId`] the input and output [`Ends`] for it.
    fn add_operator(
        &mut self,
        current_varname: Option<&Ident>,
        current_loop: Option<GraphLoopId>,
        operator: Operator,
        op_span: Option<Span>,
    ) -> (GraphNodeId, Ends) {
        let node_id = self.flat_graph.insert_node(
            GraphNode::Operator(operator),
            current_varname.cloned(),
            current_loop,
        );
        let ends = Ends {
            inn: Some((
                PortIndexValue::Elided(op_span),
                GraphDet::Determined(node_id),
            )),
            out: Some((
                PortIndexValue::Elided(op_span),
                GraphDet::Determined(node_id),
            )),
        };
        (node_id, ends)
    }

    /// Connects operator links as a final building step. Processes all the links stored in
    /// `self.links` and actually puts them into the graph.
    fn finalize_connect_operator_links(&mut self) {
        // `->` edges
        for Ends { out, inn } in std::mem::take(&mut self.links) {
            let out_opt = self.helper_resolve_name(out, false);
            let inn_opt = self.helper_resolve_name(inn, true);
            // `None` already have errors in `self.diagnostics`.
            if let (Some((out_port, out_node)), Some((inn_port, inn_node))) = (out_opt, inn_opt) {
                let _ = self.finalize_connect_operators(out_port, out_node, inn_port, inn_node);
            }
        }

        // Resolve the singleton references for each node.
        for node_id in self.flat_graph.node_ids().collect::<Vec<_>>() {
            if let GraphNode::Operator(operator) = self.flat_graph.node(node_id) {
                let singletons_referenced = operator
                    .singletons_referenced
                    .clone()
                    .into_iter()
                    .map(|singleton_ref| {
                        let port_det = self
                            .varname_ends
                            .get(&singleton_ref)
                            .filter(|varname_info| !varname_info.illegal_cycle)
                            .map(|varname_info| &varname_info.ends)
                            .and_then(|ends| ends.out.as_ref())
                            .cloned();
                        if let Some((_port, node_id)) = self.helper_resolve_name(port_det, false) {
                            Some(node_id)
                        } else {
                            self.diagnostics.push(Diagnostic::spanned(
                                singleton_ref.span(),
                                Level::Error,
                                format!(
                                    "Cannot find referenced name `{}`; name was never assigned.",
                                    singleton_ref
                                ),
                            ));
                            None
                        }
                    })
                    .collect();

                self.flat_graph
                    .set_node_singleton_references(node_id, singletons_referenced);
            }
        }
    }

    /// Recursively resolve a variable name. For handling forward (and backward) name references
    /// after all names have been assigned.
    /// Returns `None` if the name is not resolvable, either because it was never assigned or
    /// because it contains a self-referential cycle.
    ///
    /// `is_in` set to `true` means the _input_ side will be returned. `false` means the _output_ side will be returned.
    fn helper_resolve_name(
        &mut self,
        mut port_det: Option<(PortIndexValue, GraphDet)>,
        is_in: bool,
    ) -> Option<(PortIndexValue, GraphNodeId)> {
        const BACKUP_RECURSION_LIMIT: usize = 1024;

        let mut names = Vec::new();
        for _ in 0..BACKUP_RECURSION_LIMIT {
            match port_det? {
                (port, GraphDet::Determined(node_id)) => {
                    return Some((port, node_id));
                }
                (port, GraphDet::Undetermined(ident)) => {
                    let Some(varname_info) = self.varname_ends.get_mut(&ident) else {
                        self.diagnostics.push(Diagnostic::spanned(
                            ident.span(),
                            Level::Error,
                            format!("Cannot find name `{}`; name was never assigned.", ident),
                        ));
                        return None;
                    };
                    // Check for a self-referential cycle.
                    let cycle_found = names.contains(&ident);
                    if !cycle_found {
                        names.push(ident);
                    };
                    if cycle_found || varname_info.illegal_cycle {
                        let len = names.len();
                        for (i, name) in names.into_iter().enumerate() {
                            self.diagnostics.push(Diagnostic::spanned(
                                name.span(),
                                Level::Error,
                                format!(
                                    "Name `{}` forms or references an illegal self-referential cycle ({}/{}).",
                                    name,
                                    i + 1,
                                    len
                                ),
                            ));
                            // Set value as `Err(())` to trigger `name_ends_result.is_err()`
                            // diagnostics above if the name is referenced in the future.
                            self.varname_ends.get_mut(&name).unwrap().illegal_cycle = true;
                        }
                        return None;
                    }

                    // No self-cycle.
                    let prev = if is_in {
                        varname_info.inn_used = true;
                        &varname_info.ends.inn
                    } else {
                        varname_info.out_used = true;
                        &varname_info.ends.out
                    };
                    port_det = Self::helper_combine_end(
                        &mut self.diagnostics,
                        prev.clone(),
                        port,
                        if is_in { "input" } else { "output" },
                    );
                }
            }
        }
        self.diagnostics.push(Diagnostic::spanned(
            Span::call_site(),
            Level::Error,
            format!(
                "Reached the recursion limit {} while resolving names. This is either a dfir bug or you have an absurdly long chain of names: `{}`.",
                BACKUP_RECURSION_LIMIT,
                names.iter().map(ToString::to_string).collect::<Vec<_>>().join("` -> `"),
            )
        ));
        None
    }

    /// Connect two operators on the given port indexes.
    fn finalize_connect_operators(
        &mut self,
        src_port: PortIndexValue,
        src: GraphNodeId,
        dst_port: PortIndexValue,
        dst: GraphNodeId,
    ) -> GraphEdgeId {
        {
            /// Helper to emit conflicts when a port is used twice.
            fn emit_conflict(
                inout: &str,
                old: &PortIndexValue,
                new: &PortIndexValue,
                diagnostics: &mut Diagnostics,
            ) {
                // TODO(mingwei): Use `MultiSpan` once `proc_macro2` supports it.
                diagnostics.push(Diagnostic::spanned(
                    old.span(),
                    Level::Error,
                    format!(
                        "{} connection conflicts with below ({}) (1/2)",
                        inout,
                        PrettySpan(new.span()),
                    ),
                ));
                diagnostics.push(Diagnostic::spanned(
                    new.span(),
                    Level::Error,
                    format!(
                        "{} connection conflicts with above ({}) (2/2)",
                        inout,
                        PrettySpan(old.span()),
                    ),
                ));
            }

            // Handle src's successor port conflicts:
            if src_port.is_specified() {
                for conflicting_port in self
                    .flat_graph
                    .node_successor_edges(src)
                    .map(|edge_id| self.flat_graph.edge_ports(edge_id).0)
                    .filter(|&port| port == &src_port)
                {
                    emit_conflict("Output", conflicting_port, &src_port, &mut self.diagnostics);
                }
            }

            // Handle dst's predecessor port conflicts:
            if dst_port.is_specified() {
                for conflicting_port in self
                    .flat_graph
                    .node_predecessor_edges(dst)
                    .map(|edge_id| self.flat_graph.edge_ports(edge_id).1)
                    .filter(|&port| port == &dst_port)
                {
                    emit_conflict("Input", conflicting_port, &dst_port, &mut self.diagnostics);
                }
            }
        }
        self.flat_graph.insert_edge(src, src_port, dst, dst_port)
    }

    /// Process operators and emit operator errors.
    fn process_operator_errors(&mut self) {
        self.make_operator_instances();
        self.check_operator_errors();
        self.warn_unused_port_indexing();
        self.check_loop_errors();
    }

    /// Make `OperatorInstance`s for each operator node.
    fn make_operator_instances(&mut self) {
        self.flat_graph
            .insert_node_op_insts_all(&mut self.diagnostics);
    }

    /// Validates that operators have valid number of inputs, outputs, & arguments.
    /// Adds errors (and warnings) to `self.diagnostics`.
    fn check_operator_errors(&mut self) {
        /// Returns true if an error was found.
        fn emit_arity_error(
            op_span: Span,
            op_name: &str,
            is_in: bool,
            is_hard: bool,
            degree: usize,
            range: &dyn RangeTrait<usize>,
            diagnostics: &mut Diagnostics,
        ) -> bool {
            let message = format!(
                "`{}` {} have {} {}, actually has {}.",
                op_name,
                if is_hard { "must" } else { "should" },
                range.human_string(),
                if is_in { "input(s)" } else { "output(s)" },
                degree,
            );
            let out_of_range = !range.contains(&degree);
            if out_of_range {
                diagnostics.push(Diagnostic::spanned(
                    op_span,
                    if is_hard {
                        Level::Error
                    } else {
                        Level::Warning
                    },
                    message,
                ));
            }
            out_of_range
        }

        for (node_id, node) in self.flat_graph.nodes() {
            match node {
                GraphNode::Operator(operator) => {
                    let Some(op_inst) = self.flat_graph.node_op_inst(node_id) else {
                        // Error already emitted by `insert_node_op_insts_all`.
                        continue;
                    };
                    let op_constraints = op_inst.op_constraints;
                    let op_name = operator.name_string();

                    // Check number of args
                    if op_constraints.num_args != operator.args.len() {
                        self.diagnostics.push(Diagnostic::spanned(
                            operator.span(),
                            Level::Error,
                            format!(
                                "`{}` expects {} argument(s), received {}.",
                                op_name,
                                op_constraints.num_args,
                                operator.args.len()
                            ),
                        ));
                    }

                    // Check input/output (port) arity
                    let inn_degree = self.flat_graph.node_degree_in(node_id);
                    let _ = emit_arity_error(
                        operator.span(),
                        &op_name,
                        true,
                        true,
                        inn_degree,
                        op_constraints.hard_range_inn,
                        &mut self.diagnostics,
                    ) || emit_arity_error(
                        operator.span(),
                        &op_name,
                        true,
                        false,
                        inn_degree,
                        op_constraints.soft_range_inn,
                        &mut self.diagnostics,
                    );

                    let out_degree = self.flat_graph.node_degree_out(node_id);
                    let _ = emit_arity_error(
                        operator.span(),
                        &op_name,
                        false,
                        true,
                        out_degree,
                        op_constraints.hard_range_out,
                        &mut self.diagnostics,
                    ) || emit_arity_error(
                        operator.span(),
                        &op_name,
                        false,
                        false,
                        out_degree,
                        op_constraints.soft_range_out,
                        &mut self.diagnostics,
                    );

                    fn emit_port_error<'a>(
                        op_span: Span,
                        op_name: &str,
                        expected_ports_fn: Option<fn() -> PortListSpec>,
                        actual_ports_iter: impl Iterator<Item = &'a PortIndexValue>,
                        input_output: &'static str,
                        diagnostics: &mut Diagnostics,
                    ) {
                        let Some(expected_ports_fn) = expected_ports_fn else {
                            return;
                        };
                        let PortListSpec::Fixed(expected_ports) = (expected_ports_fn)() else {
                            // Separate check inside of `demux` special case.
                            return;
                        };
                        let expected_ports: Vec<_> = expected_ports.into_iter().collect();

                        // Reject unexpected ports.
                        let ports: BTreeSet<_> = actual_ports_iter
                            // Use `inspect` before collecting into `BTreeSet` to ensure we get
                            // both error messages on duplicated port names.
                            .inspect(|actual_port_iv| {
                                // For each actually used port `port_index_value`, check if it is expected.
                                let is_expected = expected_ports.iter().any(|port_index| {
                                    actual_port_iv == &&port_index.clone().into()
                                });
                                // If it is not expected, emit a diagnostic error.
                                if !is_expected {
                                    diagnostics.push(Diagnostic::spanned(
                                        actual_port_iv.span(),
                                        Level::Error,
                                        format!(
                                            "`{}` received unexpected {} port: {}. Expected one of: `{}`",
                                            op_name,
                                            input_output,
                                            actual_port_iv.as_error_message_string(),
                                            Itertools::intersperse(
                                                expected_ports
                                                    .iter()
                                                    .map(|port| port.to_token_stream().to_string())
                                                    .map(Cow::Owned),
                                                Cow::Borrowed("`, `"),
                                            ).collect::<String>()
                                        ),
                                    ))
                                }
                            })
                            .collect();

                        // List missing expected ports.
                        let missing: Vec<_> = expected_ports
                            .into_iter()
                            .filter_map(|expected_port| {
                                let tokens = expected_port.to_token_stream();
                                if !ports.contains(&&expected_port.into()) {
                                    Some(tokens)
                                } else {
                                    None
                                }
                            })
                            .collect();
                        if !missing.is_empty() {
                            diagnostics.push(Diagnostic::spanned(
                                op_span,
                                Level::Error,
                                format!(
                                    "`{}` missing expected {} port(s): `{}`.",
                                    op_name,
                                    input_output,
                                    Itertools::intersperse(
                                        missing.into_iter().map(|port| Cow::Owned(
                                            port.to_token_stream().to_string()
                                        )),
                                        Cow::Borrowed("`, `")
                                    )
                                    .collect::<String>()
                                ),
                            ));
                        }
                    }

                    emit_port_error(
                        operator.span(),
                        &op_name,
                        op_constraints.ports_inn,
                        self.flat_graph
                            .node_predecessor_edges(node_id)
                            .map(|edge_id| self.flat_graph.edge_ports(edge_id).1),
                        "input",
                        &mut self.diagnostics,
                    );
                    emit_port_error(
                        operator.span(),
                        &op_name,
                        op_constraints.ports_out,
                        self.flat_graph
                            .node_successor_edges(node_id)
                            .map(|edge_id| self.flat_graph.edge_ports(edge_id).0),
                        "output",
                        &mut self.diagnostics,
                    );

                    // Check that singleton references actually reference valid targets.
                    {
                        let singletons_resolved =
                            self.flat_graph.node_singleton_references(node_id);
                        for (singleton_node_id, singleton_ident) in singletons_resolved
                            .iter()
                            .zip_eq(&*operator.singletons_referenced)
                        {
                            let &Some(singleton_node_id) = singleton_node_id else {
                                // Error already emitted by `connect_operator_links`, "Cannot find referenced name...".
                                continue;
                            };
                            // HandoffKind::Option nodes are valid singleton reference targets.
                            if matches!(
                                self.flat_graph.node(singleton_node_id),
                                GraphNode::Handoff {
                                    kind: HandoffKind::Option,
                                    ..
                                }
                            ) {
                                continue;
                            }
                            let Some(ref_op_inst) = self.flat_graph.node_op_inst(singleton_node_id)
                            else {
                                // Error already emitted by `insert_node_op_insts_all`.
                                continue;
                            };
                            let ref_op_constraints = ref_op_inst.op_constraints;
                            if !ref_op_constraints.has_singleton_output {
                                self.diagnostics.push(Diagnostic::spanned(
                                    singleton_ident.span(),
                                    Level::Error,
                                    format!(
                                        "Cannot reference operator `{}`. Only operators with singleton state can be referenced.",
                                        ref_op_constraints.name,
                                    ),
                                ));
                            }
                        }
                    }
                }
                GraphNode::Handoff { kind, src_span, .. } => {
                    // Validate arity: handoff must have exactly 1 input and 1 output.
                    let op_name = match kind {
                        HandoffKind::Vec => "handoff",
                        HandoffKind::Option => "singleton",
                    };
                    let inn_degree = self.flat_graph.node_degree_in(node_id);
                    emit_arity_error(
                        *src_span,
                        op_name,
                        true,
                        true,
                        inn_degree,
                        &(1..=1),
                        &mut self.diagnostics,
                    );
                    let out_degree = self.flat_graph.node_degree_out(node_id);
                    let out_degree_range = match kind {
                        HandoffKind::Vec => 1..=1,
                        // `singleton()` may be no-output, only by ref. In the future this will also apply to vec.
                        HandoffKind::Option => 0..=1,
                    };
                    emit_arity_error(
                        *src_span,
                        op_name,
                        false,
                        true,
                        out_degree,
                        &out_degree_range,
                        &mut self.diagnostics,
                    );
                }
                GraphNode::ModuleBoundary { .. } => {
                    // Module boundaries don't require any checking.
                }
            }
        }
    }

    /// Warns about unused port indexing referenced in [`Self::varname_ends`].
    /// https://github.com/hydro-project/hydro/issues/1108
    fn warn_unused_port_indexing(&mut self) {
        for (_ident, varname_info) in self.varname_ends.iter() {
            if !varname_info.inn_used {
                Self::helper_check_unused_port(&mut self.diagnostics, &varname_info.ends, true);
            }
            if !varname_info.out_used {
                Self::helper_check_unused_port(&mut self.diagnostics, &varname_info.ends, false);
            }
        }
    }

    /// Emit a warning to `diagnostics` for an unused port (i.e. if the port is specified for
    /// reason).
    fn helper_check_unused_port(diagnostics: &mut Diagnostics, ends: &Ends, is_in: bool) {
        let port = if is_in { &ends.inn } else { &ends.out };
        if let Some((port, _)) = port
            && port.is_specified()
        {
            diagnostics.push(Diagnostic::spanned(
                port.span(),
                Level::Error,
                format!(
                    "{} port index is unused. (Is the port on the correct side?)",
                    if is_in { "Input" } else { "Output" },
                ),
            ));
        }
    }

    /// Helper function.
    /// Combine the port indexing information for indexing wrapped around a name.
    /// Because the name may already have indexing, this may introduce double indexing (i.e. `[0][0]my_var[0][0]`)
    /// which would be an error.
    fn helper_combine_ends(
        diagnostics: &mut Diagnostics,
        og_ends: Ends,
        inn_port: PortIndexValue,
        out_port: PortIndexValue,
    ) -> Ends {
        Ends {
            inn: Self::helper_combine_end(diagnostics, og_ends.inn, inn_port, "input"),
            out: Self::helper_combine_end(diagnostics, og_ends.out, out_port, "output"),
        }
    }

    /// Helper function.
    /// Combine the port indexing info for one input or output.
    fn helper_combine_end(
        diagnostics: &mut Diagnostics,
        og: Option<(PortIndexValue, GraphDet)>,
        other: PortIndexValue,
        input_output: &'static str,
    ) -> Option<(PortIndexValue, GraphDet)> {
        // TODO(mingwei): minification pass over this code?

        let other_span = other.span();

        let (og_port, og_node) = og?;
        match og_port.combine(other) {
            Ok(combined_port) => Some((combined_port, og_node)),
            Err(og_port) => {
                // TODO(mingwei): Use `MultiSpan` once `proc_macro2` supports it.
                diagnostics.push(Diagnostic::spanned(
                    og_port.span(),
                    Level::Error,
                    format!(
                        "Indexing on {} is overwritten below ({}) (1/2).",
                        input_output,
                        PrettySpan(other_span),
                    ),
                ));
                diagnostics.push(Diagnostic::spanned(
                    other_span,
                    Level::Error,
                    format!(
                        "Cannot index on already-indexed {}, previously indexed above ({}) (2/2).",
                        input_output,
                        PrettySpan(og_port.span()),
                    ),
                ));
                // When errored, just use original and ignore OTHER port to minimize
                // noisy/extra diagnostics.
                Some((og_port, og_node))
            }
        }
    }

    /// Check for loop context-related errors.
    fn check_loop_errors(&mut self) {
        for (node_id, node) in self.flat_graph.nodes() {
            let Some(op_inst) = self.flat_graph.node_op_inst(node_id) else {
                continue;
            };
            let loop_opt = self.flat_graph.node_loop(node_id);

            // Ensure no `'tick` or `'static` persistences are used WITHIN a loop context.
            // Ensure no `'loop` persistences are used OUTSIDE a loop context.
            for persistence in &op_inst.generics.persistence_args {
                let span = op_inst.generics.generic_args.span();
                match (loop_opt, persistence) {
                    (Some(_loop_id), p @ (Persistence::Tick | Persistence::Static)) => {
                        self.diagnostics.push(Diagnostic::spanned(
                            span,
                            Level::Error,
                            format!(
                                "Operator uses `'{}` persistence, which is not allowed within a `loop {{ ... }}` context.",
                                p.to_str_lowercase(),
                            ),
                        ));
                    }
                    (None, p @ (Persistence::None | Persistence::Loop)) => {
                        self.diagnostics.push(Diagnostic::spanned(
                            span,
                            Level::Error,
                            format!(
                                "Operator uses `'{}` persistence, but is not within a `loop {{ ... }}` context.",
                                p.to_str_lowercase(),
                            ),
                        ));
                    }
                    _ => {}
                }
            }

            // All inputs must be declared in the root block.
            if let (Some(_loop_id), Some(FloType::Source)) =
                (loop_opt, op_inst.op_constraints.flo_type)
            {
                self.diagnostics.push(Diagnostic::spanned(
                    node.span(),
                    Level::Error,
                    format!(
                        "Source operator `{}(...)` must be at the root level, not within any `loop {{ ... }}` contexts.",
                        op_inst.op_constraints.name
                    )
                ));
            }
        }

        // Check windowing and un-windowing operators, for loop inputs and outputs respectively.
        for (_edge_id, (pred_id, node_id)) in self.flat_graph.edges() {
            let Some(op_inst) = self.flat_graph.node_op_inst(node_id) else {
                continue;
            };
            let flo_type = &op_inst.op_constraints.flo_type;

            let pred_loop_id = self.flat_graph.node_loop(pred_id);
            let loop_id = self.flat_graph.node_loop(node_id);

            let span = self.flat_graph.node(node_id).span();

            let (is_input, is_output) = {
                let parent_pred_loop_id =
                    pred_loop_id.and_then(|lid| self.flat_graph.loop_parent(lid));
                let parent_loop_id = loop_id.and_then(|lid| self.flat_graph.loop_parent(lid));
                let is_same = pred_loop_id == loop_id;
                let is_input = !is_same && parent_loop_id == pred_loop_id;
                let is_output = !is_same && parent_pred_loop_id == loop_id;
                if !(is_input || is_output || is_same) {
                    self.diagnostics.push(Diagnostic::spanned(
                        span,
                        Level::Error,
                        "Operator input edge may not cross multiple loop contexts.",
                    ));
                    continue;
                }
                (is_input, is_output)
            };

            match flo_type {
                None => {
                    if is_input {
                        self.diagnostics.push(Diagnostic::spanned(
                            span,
                            Level::Error,
                            format!(
                                "Operator `{}(...)` entering a loop context must be a windowing operator, but is not.",
                                op_inst.op_constraints.name
                            )
                        ));
                    }
                    if is_output {
                        self.diagnostics.push(Diagnostic::spanned(
                            span,
                            Level::Error,
                            format!(
                                "Operator `{}(...)` exiting a loop context must be an un-windowing operator, but is not.",
                                op_inst.op_constraints.name
                            )
                        ));
                    }
                }
                Some(FloType::Windowing) => {
                    if !is_input {
                        self.diagnostics.push(Diagnostic::spanned(
                            span,
                            Level::Error,
                            format!(
                                "Windowing operator `{}(...)` must be the first input operator into a `loop {{ ... }} context.",
                                op_inst.op_constraints.name
                            )
                        ));
                    }
                }
                Some(FloType::Unwindowing) => {
                    if !is_output {
                        self.diagnostics.push(Diagnostic::spanned(
                            span,
                            Level::Error,
                            format!(
                                "Un-windowing operator `{}(...)` must be the first output operator after exiting a `loop {{ ... }} context.",
                                op_inst.op_constraints.name
                            )
                        ));
                    }
                }
                Some(FloType::NextIteration) => {
                    // Must be in a loop context.
                    if loop_id.is_none() {
                        self.diagnostics.push(Diagnostic::spanned(
                            span,
                            Level::Error,
                            format!(
                                "Operator `{}(...)` must be within a `loop {{ ... }}` context.",
                                op_inst.op_constraints.name
                            ),
                        ));
                    }
                }
                Some(FloType::Source) => {
                    // Handled above.
                }
            }
        }

        // Must be a DAG (excluding `next_iteration()` operators).
        // TODO(mingwei): Nested loop blocks should count as a single node.
        // But this doesn't cause any correctness issues because the nested loops are also DAGs.
        for (loop_id, loop_nodes) in self.flat_graph.loops() {
            // Filter out `next_iteration()` operators.
            let filter_next_iteration = |&node_id: &GraphNodeId| {
                self.flat_graph
                    .node_op_inst(node_id)
                    .map(|op_inst| Some(FloType::NextIteration) != op_inst.op_constraints.flo_type)
                    .unwrap_or(true)
            };

            let topo_sort_result = graph_algorithms::topo_sort(
                loop_nodes.iter().copied().filter(filter_next_iteration),
                |dst| {
                    self.flat_graph
                        .node_predecessor_nodes(dst)
                        .filter(|&src| Some(loop_id) == self.flat_graph.node_loop(src))
                        .filter(filter_next_iteration)
                },
            );
            if let Err(cycle) = topo_sort_result {
                let len = cycle.len();
                for (i, node_id) in cycle.into_iter().enumerate() {
                    let span = self.flat_graph.node(node_id).span();
                    self.diagnostics.push(Diagnostic::spanned(
                        span,
                        Level::Error,
                        format!(
                            "Operator forms an illegal cycle within a `loop {{ ... }}` block. Use `{}()` to pass data across loop iterations. ({}/{})",
                            NEXT_ITERATION.name,
                            i + 1,
                            len,
                        ),
                    ));
                }
            }
        }
    }
}