portgraph 0.16.1

//! Views into non-hierarchical parts of `PortView`s.

use std::borrow::Cow;
use std::collections::{BTreeSet, HashMap};

use delegate::delegate;
use itertools::{Either, Itertools};
use thiserror::Error;

use crate::algorithms::{ConvexChecker, TopoConvexChecker};
use crate::boundary::{Boundary, HasBoundary};
use crate::{
    Direction, LinkError, LinkMut, LinkView, MultiView, NodeIndex, PortIndex, PortOffset, PortView,
};

/// View into a subgraph of a PortView.
///
/// The subgraph is given by boundary edges that define where one "enters" the
/// subgraph (incoming boundary edge) and one "leaves" it (outgoing boundary
/// edges).
///
/// A subgraph is well-defined if the incoming/outgoing boundary edges partition
/// the edges between the children of the root node into three sets:
///  - boundary edges: either incoming boundary or outgoing boundary edges,
///  - interior edges: edges such that all the successor edges are either
///    outgoing boundary edges or interior edges AND all the predecessor edges
///    are either incoming boundary edges or interior edges,
///  - exterior edges: edges such that all the successor edges are either
///    incoming boundary edges or exterior edges AND all the predecessor edges
///    are either outgoing boundary edges or exterior edges.
///
/// Then the subgraph is made of the interior edges and contains all nodes that
/// are
///  - adjacent to an interior edge, or
///  - are the target of an incoming boundary edge AND the source of an outgoing boundary edge.
///
/// An intuitive way of looking at this definition is to imagine that the
/// boundary edges form a wall around the subgraph, and the subgraph is given
/// by all nodes and edges that can be reached from within without crossing the
/// wall. The [Direction] of edges (incoming/outgoing) defines which side of
/// the wall is inside, and which is outside.
///
/// Note that if the graph contains multiple connected components, there may be
/// components none of whose edges are in the boundary. The definition above is
/// consistent with such a component being either in or outside the subgraph.
/// The [Self::new_subgraph] constructor offers only limited flexibility in
/// defining such subgraphs and [Self::with_nodes] may need to be used instead.
///
/// If an invalid subgraph is defined, then behaviour is undefined.
///
/// If any graph method is called with a node or port that is not in the subgraph,
/// the behaviour is unspecified.
#[derive(Debug, Clone, PartialEq)]
pub struct Subgraph<G: PortView> {
    /// The base graph.
    graph: G,
    /// All the nodes included in the subgraph.
    nodes: BTreeSet<NodeIndex<G::NodeIndexBase>>,
    /// The ordered list of incoming and outgoing ports in the boundary.
    boundary: Boundary<G::PortIndexBase>,
}

impl<G> Subgraph<G>
where
    G: LinkView,
{
    /// Create a new subgraph view of `graph`.
    ///
    /// ### Arguments
    ///
    /// - `graph`: the graph to take a subgraph from,
    /// - `boundary`: the boundary ports. Incoming ports are incoming boundary edges,
    ///   and outgoing ports are outgoing boundary edges.
    ///
    /// This initialisation is linear in the size of the subgraph.
    ///
    /// If both incoming and outgoing boundary edges are empty, the subgraph is taken
    /// to be the entire graph (i.e. all components); otherwise, the subgraph
    /// contains only (parts of) those components mentioned in the boundary.
    /// (Specifically, where the boundary includes at least one edge, or disconnected
    /// port. To create a subgraph containing the whole of a component without any
    /// disconnected ports, use [`Self::with_nodes`].)
    pub fn new_subgraph(graph: G, boundary: Boundary<G::PortIndexBase>) -> Self {
        let nodes = boundary.internal_nodes(&graph).collect();
        Self {
            graph,
            nodes,
            boundary,
        }
    }

    /// Create a new subgraph of `graph` containing only the given nodes.
    ///
    /// The boundary of the subgraph is defined by all ports of the given nodes
    /// that have edges to other (i.e. external) nodes, in the order `nodes` are given.
    ///
    /// See [`Subgraph::new_subgraph`] for a method that takes an explicit port boundary.
    pub fn with_nodes(
        graph: G,
        nodes: impl IntoIterator<Item = NodeIndex<G::NodeIndexBase>>,
    ) -> Self {
        let ordered_nodes = nodes.into_iter().collect_vec();
        let nodes = ordered_nodes.iter().copied().collect();
        let boundary = collect_boundary_ports(&graph, ordered_nodes, &nodes);
        Self {
            graph,
            nodes,
            boundary,
        }
    }

    /// Whether the subgraph is convex.
    #[inline]
    pub fn is_convex(&self) -> bool {
        if self.node_count() <= 1 {
            return true;
        }
        let checker = TopoConvexChecker::new(&self.graph);
        self.is_convex_with_checker(&checker)
    }

    /// Whether the subgraph is convex, using a pre-existing checker.
    #[inline]
    pub fn is_convex_with_checker(
        &self,
        checker: &impl ConvexChecker<NodeIndexBase = G::NodeIndexBase, PortIndexBase = G::PortIndexBase>,
    ) -> bool {
        if self.node_count() <= 1 {
            return true;
        }
        checker.is_convex(
            self.nodes_iter(),
            self.boundary.input_indices().iter().copied(),
            self.boundary.output_indices().iter().copied(),
        )
    }

    /// Utility function to filter out links that are not in the subgraph.
    #[inline(always)]
    fn contains_link(&self, (from, to): (G::LinkEndpoint, G::LinkEndpoint)) -> bool {
        self.contains_endpoint(from) && self.contains_endpoint(to)
    }

    /// Utility function to filter out link endpoints that are not in the subgraph.
    #[inline(always)]
    fn contains_endpoint(&self, e: G::LinkEndpoint) -> bool {
        self.contains_port(e.into())
    }
}

/// Collect all the boundary input and output ports of a set of nodes.
///
/// Ports that connect nodes in the set are not included.
fn collect_boundary_ports<G>(
    graph: &G,
    ordered_nodes: impl IntoIterator<Item = NodeIndex<G::NodeIndexBase>>,
    node_set: &BTreeSet<NodeIndex<G::NodeIndexBase>>,
) -> Boundary<G::PortIndexBase>
where
    G: LinkView,
{
    let mut inputs = Vec::new();
    let mut outputs = Vec::new();

    let has_external_link = |p: &PortIndex<G::PortIndexBase>| -> bool {
        graph.port_links(*p).any(|(_, lnk)| {
            let neighbour = graph.port_node(lnk).expect("Linked port belongs to a node");
            !node_set.contains(&neighbour)
        })
    };

    for node in ordered_nodes.into_iter() {
        for p in graph.inputs(node).filter(has_external_link) {
            inputs.push(p);
        }
        for p in graph.outputs(node).filter(has_external_link) {
            outputs.push(p);
        }
    }

    Boundary::new(inputs, outputs)
}

impl<G> PortView for Subgraph<G>
where
    G: PortView,
{
    type NodeIndexBase = G::NodeIndexBase;
    type PortIndexBase = G::PortIndexBase;
    type PortOffsetBase = G::PortOffsetBase;

    #[inline(always)]
    fn contains_node(&'_ self, node: NodeIndex<Self::NodeIndexBase>) -> bool {
        self.nodes.contains(&node)
    }

    #[inline(always)]
    fn contains_port(&self, port: PortIndex<Self::PortIndexBase>) -> bool {
        let Some(node) = self.graph.port_node(port) else {
            return false;
        };
        self.contains_node(node)
    }

    #[inline]
    fn is_empty(&self) -> bool {
        self.nodes.is_empty()
    }

    #[inline]
    fn node_count(&self) -> usize {
        self.nodes.len()
    }

    #[inline]
    fn port_count(&self) -> usize {
        self.ports_iter().count()
    }

    #[inline]
    fn nodes_iter(&self) -> impl Iterator<Item = NodeIndex<Self::NodeIndexBase>> + Clone {
        self.nodes.iter().copied()
    }

    #[inline]
    fn ports_iter(&self) -> impl Iterator<Item = PortIndex<Self::PortIndexBase>> + Clone {
        self.nodes.iter().flat_map(|&n| self.graph.all_ports(n))
    }

    #[inline]
    fn node_capacity(&self) -> usize {
        self.graph.node_capacity() - self.graph.node_count() + self.node_count()
    }

    #[inline]
    fn port_capacity(&self) -> usize {
        self.graph.port_capacity() - self.graph.port_count() + self.port_count()
    }

    delegate! {
        to self.graph {
            fn port_direction(&self, port: impl Into<PortIndex<Self::PortIndexBase>>) -> Option<Direction>;
            fn port_node(&self, port: impl Into<PortIndex<Self::PortIndexBase>>) -> Option<NodeIndex<Self::NodeIndexBase>>;
            fn port_offset(&self, port: impl Into<PortIndex<Self::PortIndexBase>>) -> Option<PortOffset<Self::PortOffsetBase>>;
            fn port_index(&self, node: NodeIndex<Self::NodeIndexBase>, offset: PortOffset<Self::PortOffsetBase>) -> Option<PortIndex<Self::PortIndexBase>>;
            fn ports(&self, node: NodeIndex<Self::NodeIndexBase>, direction: Direction) -> impl Iterator<Item = PortIndex<Self::PortIndexBase>> + Clone;
            fn all_ports(&self, node: NodeIndex<Self::NodeIndexBase>) -> impl Iterator<Item = PortIndex<Self::PortIndexBase>> + Clone;
            fn input(&self, node: NodeIndex<Self::NodeIndexBase>, offset: usize) -> Option<PortIndex<Self::PortIndexBase>>;
            fn output(&self, node: NodeIndex<Self::NodeIndexBase>, offset: usize) -> Option<PortIndex<Self::PortIndexBase>>;
            fn num_ports(&self, node: NodeIndex<Self::NodeIndexBase>, direction: Direction) -> usize;
            fn port_offsets(&self, node: NodeIndex<Self::NodeIndexBase>, direction: Direction) -> impl Iterator<Item = PortOffset<Self::PortOffsetBase>> + Clone;
            fn all_port_offsets(&self, node: NodeIndex<Self::NodeIndexBase>) -> impl Iterator<Item = PortOffset<Self::PortOffsetBase>> + Clone;
            fn node_port_capacity(&self, node: NodeIndex<Self::NodeIndexBase>) -> usize;
        }
    }
}

impl<G> LinkView for Subgraph<G>
where
    G: LinkView,
{
    type LinkEndpoint = G::LinkEndpoint;

    fn get_connections(
        &self,
        from: NodeIndex<Self::NodeIndexBase>,
        to: NodeIndex<Self::NodeIndexBase>,
    ) -> impl Iterator<Item = (Self::LinkEndpoint, Self::LinkEndpoint)> + Clone {
        if self.nodes.contains(&from) && self.nodes.contains(&to) {
            Either::Left(self.graph.get_connections(from, to))
        } else {
            Either::Right(std::iter::empty())
        }
    }

    fn port_links(
        &self,
        port: PortIndex<Self::PortIndexBase>,
    ) -> impl Iterator<Item = (Self::LinkEndpoint, Self::LinkEndpoint)> + Clone {
        self.graph
            .port_links(port)
            .filter(|&lnk| self.contains_link(lnk))
    }

    fn links(
        &self,
        node: NodeIndex<Self::NodeIndexBase>,
        direction: Direction,
    ) -> impl Iterator<Item = (Self::LinkEndpoint, Self::LinkEndpoint)> + Clone {
        self.graph
            .links(node, direction)
            .filter(|&lnk| self.contains_link(lnk))
    }

    fn all_links(
        &self,
        node: NodeIndex<Self::NodeIndexBase>,
    ) -> impl Iterator<Item = (Self::LinkEndpoint, Self::LinkEndpoint)> + Clone {
        self.graph
            .all_links(node)
            .filter(|&lnk| self.contains_link(lnk))
    }

    fn neighbours(
        &self,
        node: NodeIndex<Self::NodeIndexBase>,
        direction: Direction,
    ) -> impl Iterator<Item = NodeIndex<Self::NodeIndexBase>> + Clone {
        self.graph
            .neighbours(node, direction)
            .filter(|&n| self.contains_node(n))
    }

    fn all_neighbours(
        &self,
        node: NodeIndex<Self::NodeIndexBase>,
    ) -> impl Iterator<Item = NodeIndex<Self::NodeIndexBase>> + Clone {
        self.graph
            .all_neighbours(node)
            .filter(|&n| self.contains_node(n))
    }

    fn link_count(&self) -> usize {
        self.nodes_iter()
            .flat_map(|node| self.links(node, Direction::Outgoing))
            .count()
    }
}

impl<G> MultiView for Subgraph<G>
where
    G: MultiView,
{
    fn subport_link(&self, subport: Self::LinkEndpoint) -> Option<Self::LinkEndpoint> {
        self.graph
            .subport_link(subport)
            .filter(|&p| self.contains_endpoint(p))
    }

    delegate! {
        to self.graph {
            fn subports(&self, node: NodeIndex<Self::NodeIndexBase>, direction: Direction) -> impl Iterator<Item = Self::LinkEndpoint> + Clone;
            fn all_subports(&self, node: NodeIndex<Self::NodeIndexBase>) -> impl Iterator<Item = Self::LinkEndpoint> + Clone;
        }
    }
}

/// An error from [Subgraph::copy_in_parent]
#[derive(Clone, Debug, PartialEq, Eq, Error)]
#[non_exhaustive]
pub enum CopySubgraphError<P: crate::index::IndexBase = u32> {
    /// Tried to copy an edge crossing a subgraph boundary
    /// when the containing graph is not a [`MultiMut`](crate::MultiMut)
    #[error("Cannot copy edge between external port {external:?} and (copy of) internal port {internal:?}")]
    #[allow(missing_docs)]
    CantCopyBoundary {
        internal: PortIndex<P>,
        external: PortIndex<P>,
    },
}

impl<G> Subgraph<G>
where
    G: LinkMut,
{
    /// Copies all the nodes and edges in this subgraph into the parent graph.
    ///
    /// If there are any boundary edges, these will also be copied but keeping
    /// the same *external* end port - this will fail unless the underlying graph
    /// is a [`MultiMut`]
    ///
    /// # Returns
    ///
    ///   A map from node indices within this subgraph, to the indices of the
    ///   newly-created nodes in the parent graph (they are not in this subgraph).
    ///
    /// # Errors
    ///
    /// - [`CopySubgraphError::CantCopyBoundary`] if there is a boundary edge and
    ///   the underlying graph is not a [`MultiMut`]. In this case the underlying
    ///   graph's nodes and edges will be unchanged, but capacity may have changed.
    ///
    /// [`MultiMut`]: crate::MultiMut
    #[allow(clippy::type_complexity)]
    pub fn copy_in_parent(
        &mut self,
    ) -> Result<
        HashMap<NodeIndex<G::NodeIndexBase>, NodeIndex<G::NodeIndexBase>>,
        CopySubgraphError<G::PortIndexBase>,
    > {
        self.graph.reserve(self.node_count(), self.port_count());
        let g = &mut self.graph;
        let node_map = self
            .nodes
            .iter()
            .map(|node| (*node, g.add_node(g.num_inputs(*node), g.num_outputs(*node))))
            .collect::<HashMap<_, _>>();
        for (node, new_node) in node_map.iter() {
            for (node_p, other_p) in self.graph.all_links(*node).collect::<Vec<_>>() {
                let new_node_p = self
                    .graph
                    .port_index(*new_node, self.graph.port_offset(node_p).unwrap())
                    .unwrap();
                match node_map.get(&self.graph.port_node(other_p).unwrap()) {
                    Some(new_other) => {
                        // Internal edge. Add once only, not once per end. (`all_links` reports self-cycles only once.)
                        if new_other < new_node {
                            continue;
                        }
                        let offset = self.graph.port_offset(other_p).unwrap();
                        let new_other_p = self.graph.port_index(*new_other, offset).unwrap();
                        self.graph
                            .link_ports(new_node_p, new_other_p)
                            .expect("Copying known-good edge");
                    }
                    None => {
                        // Boundary edge. Keep same external endpoint
                        match self.graph.link_ports(new_node_p, other_p.into()) {
                            Err(LinkError::AlreadyLinked { port }) => {
                                // Must undo insertion
                                for n in node_map.values() {
                                    self.graph.remove_node(*n);
                                }
                                return Err(CopySubgraphError::CantCopyBoundary {
                                    external: port,
                                    internal: node_p.into(),
                                });
                            }
                            Err(e) => panic!("Unexpected error copying boundary edge: {e}"),
                            Ok(_) => (),
                        }
                    }
                }
            }
        }
        Ok(node_map)
    }
}

impl<G: PortView> HasBoundary<G::PortIndexBase> for Subgraph<G> {
    fn port_boundary(&self) -> Cow<'_, Boundary<G::PortIndexBase>> {
        Cow::Borrowed(&self.boundary)
    }
}

#[cfg(test)]
mod tests {
    use std::collections::HashSet;

    use itertools::Itertools;
    use rstest::rstest;

    use crate::LinkMut;
    type SubportIndex = crate::multiportgraph::SubportIndex<u32>;
    use crate::PortMut;

    use super::*;

    type PortGraph = crate::PortGraph<u32, u32, u16>;
    type MultiPortGraph = crate::MultiPortGraph<u32, u32, u16>;
    type NodeIndex = crate::NodeIndex<u32>;
    type PortIndex = crate::PortIndex<u32>;

    /// Create the following graph
    ///
    ///  ┌─────┐┌─┐
    ///  │0    ││3│
    ///  └┬───┬┘└┬┘
    ///   │   │  │
    ///  ┌▽─┐┌▽─┐│
    ///  │1 ││2 ││
    ///  └┬┬┘└┬┬┘│
    ///   ││ ┌┘│ │
    ///   │└─│┐│┌┘
    ///   │┌─┘│││
    ///  ┌▽▽┐┌▽▽▽┐
    ///  │5 ││4  │
    ///  └──┘└───┘
    fn graph() -> PortGraph {
        let mut graph = PortGraph::new();
        let n0 = graph.add_node(0, 2);
        let n1 = graph.add_node(1, 2);
        let n2 = graph.add_node(1, 2);
        let n3 = graph.add_node(0, 1);
        let n4 = graph.add_node(3, 0);
        let n5 = graph.add_node(2, 0);
        graph.link_nodes(n0, 0, n1, 0).unwrap();
        graph.link_nodes(n0, 1, n2, 0).unwrap();
        graph.link_nodes(n3, 0, n4, 0).unwrap();
        graph.link_nodes(n1, 0, n4, 1).unwrap();
        graph.link_nodes(n2, 1, n4, 2).unwrap();
        graph.link_nodes(n1, 1, n5, 0).unwrap();
        graph.link_nodes(n2, 0, n5, 1).unwrap();
        graph
    }

    #[test]
    fn test_traverse_subgraph_single_node() {
        let graph = graph();
        let (_, n1, _, _, _, _) = (0..6).map(NodeIndex::new).collect_tuple().unwrap();

        // Define the incoming and outgoing boundary edges
        let boundary = Boundary::new(graph.inputs(n1), graph.outputs(n1));

        // Traverse the subgraph
        let nodes: HashSet<_> = boundary.internal_nodes(&graph).collect();

        // Check that the correct nodes and ports were found
        assert_eq!(nodes, [n1].into_iter().collect());
        assert_eq!(boundary.num_ports(), 3);
    }

    #[test]
    fn test_traverse_subgraph_all_but_one_edge() {
        let graph = graph();
        let (_, n1, _, _, n4, _) = (0..6).map(NodeIndex::new).collect_tuple().unwrap();

        // Define both ends of the `1 -> 4` edge as boundary, effectively selecting the whole graph.
        let boundary = Boundary::new(
            [graph.output(n1, 0).unwrap()],
            [graph.input(n4, 1).unwrap()],
        );

        // Traverse the subgraph
        let nodes: HashSet<_> = boundary.internal_nodes(&graph).collect();

        // Check that the correct nodes and ports were found
        assert_eq!(nodes, graph.nodes_iter().collect());
    }

    #[test]
    fn test_traverse_subgraph_basic() {
        let graph = graph();
        let (_, n1, n2, _, _, n5) = (0..6).map(NodeIndex::new).collect_tuple().unwrap();

        // Define the incoming and outgoing boundary edges
        let incoming = [graph.inputs(n1), graph.inputs(n2)].into_iter().flatten();
        let outgoing = [graph.output(n1, 0).unwrap(), graph.output(n2, 1).unwrap()];

        // Traverse the subgraph
        let nodes: HashSet<_> = Boundary::new(incoming, outgoing)
            .internal_nodes(&graph)
            .collect();

        // Check that the correct nodes and ports were found
        assert_eq!(nodes, [n1, n2, n5].into_iter().collect());
    }

    #[test]
    fn test_traverse_subgraph_almost_complete() {
        let graph = graph();
        let (n0, n1, n2, n3, n4, n5) = (0..6).map(NodeIndex::new).collect_tuple().unwrap();

        // Define the incoming and outgoing boundary edges
        let incoming = [
            graph.input(n1, 0).unwrap(),
            graph.input(n4, 2).unwrap(),
            graph.input(n5, 1).unwrap(),
        ];
        let outgoing = [
            graph.output(n0, 0).unwrap(),
            graph.output(n2, 1).unwrap(),
            graph.output(n2, 0).unwrap(),
        ];
        let boundary = Boundary::new(incoming, outgoing);

        // Traverse the subgraph
        let nodes: HashSet<_> = boundary.internal_nodes(&graph).collect();

        // Check that the correct nodes and ports were found
        assert_eq!(nodes, [n0, n1, n2, n3, n4, n5].into_iter().collect());
    }

    #[test]
    fn test_traverse_subgraph_complete() {
        let graph = graph();

        // Traverse the subgraph
        let nodes: HashSet<_> = Boundary::new([], []).internal_nodes(&graph).collect();

        assert_eq!(nodes, graph.nodes_iter().collect());
    }

    #[test]
    fn test_with_nodes() {
        let graph = graph();
        let (_, n1, n2, _, n4, _) = (0..6).map(NodeIndex::new).collect_tuple().unwrap();

        let boundary = Boundary::new(
            [
                graph.input(n1, 0).unwrap(),
                graph.input(n2, 0).unwrap(),
                graph.input(n4, 0).unwrap(),
            ],
            [graph.output(n1, 1).unwrap(), graph.output(n2, 0).unwrap()],
        );
        let from_boundary = Subgraph::new_subgraph(&graph, boundary);

        let from_nodes = Subgraph::with_nodes(&graph, [n1, n2, n4]);

        assert_eq!(from_boundary, from_nodes);
    }

    #[test]
    fn test_properties() {
        let graph = graph();
        let (n0, n1, n2, _n3, n4, _n5) = (0..6).map(NodeIndex::new).collect_tuple().unwrap();
        let subgraph = Subgraph::with_nodes(&graph, [n1, n2, n4]);

        let n1_o0 = subgraph.output(n1, 0).unwrap();
        let n2_o1 = subgraph.output(n2, 1).unwrap();
        let n4_i1 = subgraph.input(n4, 1).unwrap();
        let n4_i2 = subgraph.input(n4, 2).unwrap();

        assert!(!subgraph.is_empty());
        assert_eq!(subgraph.node_count(), 3);
        assert_eq!(subgraph.node_capacity(), graph.node_capacity() - 3);
        assert_eq!(subgraph.port_count(), 9);
        assert_eq!(subgraph.port_capacity(), graph.port_capacity() - 5);
        assert_eq!(
            subgraph.node_port_capacity(n1),
            graph.node_port_capacity(n1)
        );
        assert_eq!(
            subgraph.port_offsets(n1, Direction::Outgoing).collect_vec(),
            graph.port_offsets(n1, Direction::Outgoing).collect_vec()
        );

        assert!(!subgraph.contains_node(n0));
        assert!(subgraph.contains_node(n1));
        assert!(!subgraph.contains_port(graph.output(n0, 0).unwrap()));
        assert!(subgraph.contains_port(n1_o0));

        assert_eq!(subgraph.inputs(n1).count(), 1);
        assert_eq!(subgraph.outputs(n1).count(), 2);
        assert_eq!(subgraph.num_ports(n1, Direction::Incoming), 1);
        assert_eq!(subgraph.num_ports(n1, Direction::Outgoing), 2);
        assert_eq!(subgraph.all_ports(n1).count(), 3);
        assert_eq!(subgraph.all_port_offsets(n1).count(), 3);

        let inputs = subgraph
            .inputs(n1)
            .enumerate()
            .map(|(i, port)| (i, port, Direction::Incoming));
        let outputs = subgraph
            .outputs(n1)
            .enumerate()
            .map(|(i, port)| (i, port, Direction::Outgoing));
        for (i, port, dir) in inputs.chain(outputs) {
            let offset = PortOffset::new(dir, i);
            assert_eq!(subgraph.port_direction(port), Some(dir));
            assert_eq!(subgraph.port_offset(port), Some(offset));
            assert_eq!(subgraph.port_node(port), Some(n1));
            assert_eq!(subgraph.port_index(n1, offset), Some(port));
        }

        // Global iterators
        let nodes = subgraph.nodes_iter().collect_vec();
        assert_eq!(nodes.as_slice(), [n1, n2, n4]);

        let ports = subgraph.ports_iter().collect_vec();
        assert_eq!(ports.len(), subgraph.port_count());

        // Links
        assert!(subgraph.connected(n1, n4));
        assert_eq!(subgraph.link_count(), 2);
        assert_eq!(
            subgraph.output_neighbours(n1).collect_vec().as_slice(),
            [n4]
        );
        assert_eq!(
            subgraph.output_links(n1).collect_vec().as_slice(),
            [(n1_o0, n4_i1)]
        );
        assert_eq!(
            subgraph.port_links(n1_o0).collect_vec().as_slice(),
            [(n1_o0, n4_i1)]
        );
        assert_eq!(
            subgraph.all_links(n4).collect_vec().as_slice(),
            [(n4_i1, n1_o0), (n4_i2, n2_o1),]
        );
        assert_eq!(
            subgraph.get_connections(n1, n4).collect_vec().as_slice(),
            [(n1_o0, n4_i1)]
        );
        assert_eq!(subgraph.get_connections(n0, n1).count(), 0);
        assert_eq!(
            subgraph.all_neighbours(n4).collect_vec().as_slice(),
            [n1, n2]
        );

        // Multiports
        let multigraph = MultiPortGraph::from(graph);
        let subgraph = Subgraph::with_nodes(&multigraph, [n1, n2, n4]);
        let n1_o0 = SubportIndex::new_unique(n1_o0);
        assert_eq!(
            subgraph.all_subports(n1).collect_vec(),
            multigraph.all_subports(n1).collect_vec()
        );
        assert_eq!(
            subgraph.subports(n4, Direction::Incoming).collect_vec(),
            multigraph.subports(n4, Direction::Incoming).collect_vec()
        );
        assert_eq!(subgraph.subport_link(n1_o0), multigraph.subport_link(n1_o0));
    }

    #[test]
    fn test_is_convex() {
        let graph = graph();
        let (n0, n1, n2, _, n4, n5) = (0..6).map(NodeIndex::new).collect_tuple().unwrap();

        let boundary = Boundary::new(graph.inputs(n1), graph.outputs(n1));
        let subg = Subgraph::new_subgraph(&graph, boundary);
        assert!(subg.is_convex());

        let boundary = Boundary::new(
            [graph.input(n4, 1).unwrap()],
            [graph.output(n1, 0).unwrap()],
        );
        let subg = Subgraph::new_subgraph(&graph, boundary);
        assert!(!subg.is_convex());

        // Check the short-circuited case
        let subg = Subgraph::with_nodes(&graph, [n0]);
        assert!(subg.is_convex());
        assert!(subg.is_convex_with_checker(&TopoConvexChecker::new(&graph)));

        // Define the incoming and outgoing boundary edges
        let incoming = [
            graph.input(n1, 0).unwrap(),
            graph.input(n4, 2).unwrap(),
            graph.input(n5, 1).unwrap(),
        ];
        let outgoing = [
            graph.output(n0, 0).unwrap(),
            graph.output(n2, 1).unwrap(),
            graph.output(n2, 0).unwrap(),
        ];
        let boundary = Boundary::new(incoming, outgoing);
        let subg = Subgraph::new_subgraph(&graph, boundary);
        assert!(!subg.is_convex());
    }

    #[test]
    fn test_is_convex_line() {
        let mut graph: PortGraph = PortGraph::new();
        let n0 = graph.add_node(0, 1);
        let n1 = graph.add_node(1, 1);
        let n2 = graph.add_node(1, 0);
        graph.link_nodes(n0, 0, n1, 0).unwrap();
        graph.link_nodes(n1, 0, n2, 0).unwrap();

        let boundary = Boundary::from_ports(
            &graph,
            [graph.output(n0, 0).unwrap(), graph.input(n2, 0).unwrap()],
        );
        let subg = Subgraph::new_subgraph(&graph, boundary);
        assert_eq!(subg.nodes_iter().collect_vec(), [n0, n2]);
        assert!(!subg.is_convex());
    }

    #[test]
    fn test_disconnected_components() {
        let mut graph: PortGraph = PortGraph::new();
        let n0 = graph.add_node(0, 1);
        let n1 = graph.add_node(1, 0);
        graph.link_nodes(n0, 0, n1, 0).unwrap();
        let n2 = graph.add_node(0, 1);
        let n3 = graph.add_node(1, 1);
        graph.link_nodes(n2, 0, n3, 0).unwrap();

        let subg_nodes = |b| Subgraph::new_subgraph(&graph, b).nodes_iter().collect_vec();

        // No edges -> all components
        let b = Boundary::from_ports(&graph, []);
        let nodes = subg_nodes(b.clone());
        assert_eq!(nodes, [n0, n1, n2, n3]);
        assert_eq!(Subgraph::with_nodes(&graph, nodes).boundary, b);

        // Edge in only one component -> just (part of) that component
        let b = Boundary::from_ports(&graph, [graph.output(n0, 0).unwrap()]);
        let nodes = subg_nodes(b.clone());
        assert_eq!(nodes, [n0]);
        assert_eq!(Subgraph::with_nodes(&graph, nodes).boundary, b);

        // Edges in two components -> relevant parts of each component
        let b = Boundary::from_ports(
            &graph,
            [graph.output(n0, 0).unwrap(), graph.input(n3, 0).unwrap()],
        );
        let nodes = subg_nodes(b.clone());
        assert_eq!(nodes, [n0, n3]);
        assert_eq!(Subgraph::with_nodes(&graph, nodes).boundary, b);

        // Disconnected port -> includes whole component:
        let b = Boundary::from_ports(&graph, [graph.output(n3, 0).unwrap()]);
        assert_eq!(subg_nodes(b), [n2, n3]);
        // However the graph with those nodes includes the unconnected port as interior,
        // so has an empty boundary:
        assert_eq!(
            Subgraph::with_nodes(&graph, [n2, n3]).boundary,
            Boundary::from_ports(&graph, [])
        );
        // (The subgraph cannot be recreated from said boundary,
        //    see https://github.com/Quantinuum/portgraph/issues/179.)
    }

    #[test]
    fn test_copy_in_parent() {
        let mut graph: PortGraph = PortGraph::new();
        // First component: n0 -> n1 + cycle
        let n0 = graph.add_node(0, 1);
        let n1 = graph.add_node(2, 1);
        graph.link_nodes(n0, 0, n1, 0).unwrap();
        graph.link_nodes(n1, 0, n1, 1).unwrap();
        // Second component: n2 -> n3
        let n2 = graph.add_node(0, 1);
        let n3 = graph.add_node(1, 0);
        graph.link_nodes(n2, 0, n3, 0).unwrap();
        let backup = graph.clone();

        let mut subg = Subgraph::with_nodes(&mut graph, [n0, n1]);
        let mut node_map = subg.copy_in_parent().unwrap();
        assert_eq!(subg.nodes_iter().collect_vec(), vec![n0, n1]);
        assert_eq!(graph.node_count(), 6);
        let n0_copy = node_map.remove(&n0).unwrap();
        let n1_copy = node_map.remove(&n1).unwrap();
        assert!(node_map.is_empty()); // No other keys
        assert_eq!(
            graph.input_links(n1_copy).collect_vec(),
            vec![
                (
                    graph.input(n1_copy, 0).unwrap(),
                    graph.output(n0_copy, 0).unwrap()
                ),
                (
                    graph.input(n1_copy, 1).unwrap(),
                    graph.output(n1_copy, 0).unwrap()
                )
            ]
        );
        assert_same_for_nodes(&graph, &backup, backup.nodes_iter()); // Rest of graph unchanged
    }

    fn assert_same_for_nodes<A, B>(a: A, b: B, nodes: impl IntoIterator<Item = NodeIndex>)
    where
        A: LinkView<NodeIndexBase = u32, PortIndexBase = u32>,
        B: LinkView<NodeIndexBase = u32, PortIndexBase = u32>,
        PortIndex: From<A::LinkEndpoint> + From<B::LinkEndpoint>,
    {
        for node in nodes {
            assert_eq!(a.num_inputs(node), b.num_inputs(node));
            assert_eq!(a.num_outputs(node), b.num_outputs(node));
            for (a_link, b_link) in a.all_links(node).zip_eq(b.all_links(node)) {
                assert_eq!(PortIndex::from(a_link.0), PortIndex::from(b_link.0));
                assert_eq!(PortIndex::from(a_link.1), PortIndex::from(b_link.1));
            }
        }
    }

    #[rstest]
    #[case(Direction::Incoming)]
    #[case(Direction::Outgoing)]
    fn test_copy_in_parent_bad_boundary(#[case] edge: Direction) {
        let mut graph: PortGraph = PortGraph::new();
        let n0 = graph.add_node(0, 1);
        let n1 = graph.add_node(1, 1);
        let n2 = graph.add_node(1, 0);
        graph.link_nodes(n0, 0, n1, 0).unwrap();
        graph.link_nodes(n1, 0, n2, 0).unwrap();
        let backup = graph.clone();

        let (subg_nodes, internal, external) = if edge == Direction::Incoming {
            (
                [n1, n2],
                graph.input(n1, 0).unwrap(),
                graph.output(n0, 0).unwrap(),
            )
        } else {
            (
                [n0, n1],
                graph.output(n1, 0).unwrap(),
                graph.input(n2, 0).unwrap(),
            )
        };

        let mut subg = Subgraph::with_nodes(&mut graph, subg_nodes);

        assert_eq!(
            subg.copy_in_parent(),
            Err(CopySubgraphError::CantCopyBoundary { internal, external })
        );
        assert_eq!(
            graph.nodes_iter().collect_vec(),
            backup.nodes_iter().collect_vec()
        );
        assert_same_for_nodes(&graph, &backup, backup.nodes_iter());
    }

    #[test]
    fn test_copy_in_parent_multi_input() {
        let mut graph: MultiPortGraph = MultiPortGraph::new();
        let n0 = graph.add_node(0, 1);
        let n1 = graph.add_node(1, 1);
        let n2 = graph.add_node(1, 0);
        graph.link_nodes(n0, 0, n1, 0).unwrap();
        graph.link_nodes(n1, 0, n2, 0).unwrap();

        let backup = graph.clone();

        let subg_nodes = [n1, n2]; // Edge n0 -> n1 is incoming
        let mut subg = Subgraph::with_nodes(&mut graph, subg_nodes);
        let mut node_map = subg.copy_in_parent().unwrap();

        let in_edge = |n| graph.inputs(n).exactly_one().ok().unwrap();
        let out_edge = |n| graph.outputs(n).exactly_one().ok().unwrap();

        assert_eq!(graph.node_count(), 5);
        assert_eq!(node_map.keys().copied().sorted().collect_vec(), subg_nodes);
        let n1_copy = node_map.remove(&n1).unwrap();
        let n2_copy = *node_map.values().exactly_one().unwrap();
        assert_same_for_nodes(&graph, &backup, subg_nodes);
        let (sp2, sp1) = graph.all_links(n2_copy).exactly_one().ok().unwrap();
        // Check the copied graph is correct. Its internal edge is n1 -> n2.
        assert_eq!(sp2.port(), in_edge(n2_copy));
        assert_eq!(sp1.port(), out_edge(n1_copy));

        let multiport = out_edge(n0);
        assert_eq!(
            graph
                .all_links(n0)
                .map(|(sp1, sp2)| (sp1.port(), sp2.port()))
                .collect_vec(),
            [(multiport, in_edge(n1)), (multiport, in_edge(n1_copy))]
        );
    }

    #[test]
    fn test_copy_in_parent_multi_output() {
        let mut graph: MultiPortGraph = MultiPortGraph::new();
        let n0 = graph.add_node(0, 1);
        let n1 = graph.add_node(1, 1);
        let n2 = graph.add_node(1, 0);
        graph.link_nodes(n0, 0, n1, 0).unwrap();
        graph.link_nodes(n1, 0, n2, 0).unwrap();

        let backup = graph.clone();

        let subg_nodes = [n0, n1]; // Edge n1 -> n2 is outgoing
        let mut subg = Subgraph::with_nodes(&mut graph, subg_nodes);
        let mut node_map = subg.copy_in_parent().unwrap();

        let out_edge = |n| graph.outputs(n).exactly_one().ok().unwrap();
        let in_edge = |n| graph.inputs(n).exactly_one().ok().unwrap();

        assert_eq!(graph.node_count(), 5);
        assert_eq!(node_map.keys().copied().sorted().collect_vec(), subg_nodes);
        let n1_copy = node_map.remove(&n1).unwrap();
        let n0_copy = *node_map.values().exactly_one().unwrap();
        assert_same_for_nodes(&graph, &backup, subg_nodes);
        let (sp0, sp1) = graph.all_links(n0_copy).exactly_one().ok().unwrap();
        // Check the copied graph is correct. Its internal edge is n0 -> n1.
        assert_eq!(sp0.port(), out_edge(n0_copy));
        assert_eq!(sp1.port(), in_edge(n1_copy));

        let multiport = in_edge(n2);
        assert_eq!(
            graph
                .all_links(n2)
                .map(|(sp1, sp2)| (sp1.port(), sp2.port()))
                .collect_vec(),
            [(multiport, out_edge(n1)), (multiport, out_edge(n1_copy))]
        );
    }
}