feat: Add array linearisation pass

Author: mark-koch

Cargo.lock

@@ -1,10 +1,12 @@
 use std::sync::Arc;
 
+use crate::std_extensions::collections::array::op_builder::GenericArrayOpBuilder;
 use crate::{
+    builder::{BuildError, Dataflow},
     extension::{ExtensionId, SignatureError, TypeDef},
     ops::constant::ValueName,
     types::{CustomType, Type, TypeArg, TypeName},
-    Extension,
+    Extension, Wire,
 };
 
 /// Trait capturing a concrete array implementation in an extension.
@@ -90,4 +92,28 @@ pub trait ArrayKind:
     ) -> Result<Type, SignatureError> {
         Self::instantiate_ty(Self::type_def(), size, element_ty)
     }
+
+    /// Adds a operation to a dataflow graph that clones an array of copyable values.
+    ///
+    /// The default implementation uses the array clone operation.
+    fn build_clone<D: Dataflow>(
+        builder: &mut D,
+        elem_ty: Type,
+        size: u64,
+        arr: Wire,
+    ) -> Result<(Wire, Wire), BuildError> {
+        builder.add_generic_array_clone::<Self>(elem_ty, size, arr)
+    }
+
+    /// Adds a operation to a dataflow graph that clones an array of copyable values.
+    ///
+    /// The default implementation uses the array clone operation.
+    fn build_discard<D: Dataflow>(
+        builder: &mut D,
+        elem_ty: Type,
+        size: u64,
+        arr: Wire,
+    ) -> Result<(), BuildError> {
+        builder.add_generic_array_discard::<Self>(elem_ty, size, arr)
+    }
 }

hugr-core/src/std_extensions/collections/array/array_kind.rs

@@ -51,6 +51,24 @@ impl ArrayKind for ValueArray {
     fn type_def() -> &'static TypeDef {
         EXTENSION.get_type(&VALUE_ARRAY_TYPENAME).unwrap()
     }
+
+    fn build_clone<D: Dataflow>(
+        _builder: &mut D,
+        _elem_ty: Type,
+        _size: u64,
+        arr: Wire,
+    ) -> Result<(Wire, Wire), BuildError> {
+        Ok((arr, arr))
+    }
+
+    fn build_discard<D: Dataflow>(
+        _builder: &mut D,
+        _elem_ty: Type,
+        _size: u64,
+        _arr: Wire,
+    ) -> Result<(), BuildError> {
+        Ok(())
+    }
 }
 
 /// Value array operation definitions.

hugr-core/src/std_extensions/collections/value_array.rs

@@ -25,6 +25,7 @@ lazy_static = { workspace = true }
 paste = { workspace = true }
 thiserror = { workspace = true }
 petgraph = { workspace = true }
+strum = { workspace = true }
 
 [features]
 extension_inference = ["hugr-core/extension_inference"]

hugr-passes/Cargo.toml

@@ -11,6 +11,8 @@ mod dead_funcs;
 pub use dead_funcs::{remove_dead_funcs, RemoveDeadFuncsError, RemoveDeadFuncsPass};
 pub mod force_order;
 mod half_node;
+pub mod linearize_array;
+pub use linearize_array::LinearizeArrayPass;
 pub mod lower;
 pub mod merge_bbs;
 mod monomorphize;