Query whether a port is linear in command

CQCL · Sep 21, 2023 · dad511c · dad511c
1 parent 65426a7
commit dad511c
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/src/circuit/command.rs b/src/circuit/command.rs
@@ -146,6 +146,14 @@ impl<'circ, Circ: Circuit> Command<'circ, Circ> {
             .find(|(cu, _, _)| *cu == unit)
             .map(|(_, port, _)| port)
     }
+
+    /// Returns whether the port is a linear port.
+    #[inline]
+    pub fn is_linear_port(&self, port: Port) -> bool {
+        self.optype()
+            .port_kind(port)
+            .map_or(false, |kind| kind.is_linear())
+    }
 }
 
 impl<'a, 'circ, Circ: Circuit> UnitLabeller for &'a Command<'circ, Circ> {