feat: Add a FuncTypeBase::io method (#1458)

Small helper method, should be enough to close #1209.
CQCL · Aug 21, 2024 · c7cd840 · c7cd840
1 parent a61f4df
commit c7cd840
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/hugr-core/src/types/signature.rs b/hugr-core/src/types/signature.rs
@@ -98,6 +98,12 @@ impl<RV: MaybeRV> FuncTypeBase<RV> {
         &self.output
     }
 
+    #[inline]
+    /// Returns a tuple with the input and output rows of the function.
+    pub fn io(&self) -> (&TypeRowBase<RV>, &TypeRowBase<RV>) {
+        (&self.input, &self.output)
+    }
+
     pub(super) fn validate(
         &self,
         extension_registry: &ExtensionRegistry,
@@ -310,5 +316,6 @@ mod test {
 
         assert_eq!(f_type.input_types(), &[Type::UNIT]);
         assert_eq!(f_type.output_types(), &[USIZE_T]);
+        assert_eq!(f_type.io(), (&type_row![Type::UNIT], &type_row![USIZE_T]));
     }
 }