Use foldl and function to implement condition with list of measurements

Author: david-pl

src/bloqade/squin/cirq/lowering.py

@@ -173,8 +173,28 @@ def visit_ClassicallyControlledOperation(
             measurement_outcome = state.current_frame.defs[key]
 
             if measurement_outcome.type.is_subseteq(ilist.IListType):
-                # TODO: need to represent Any(measurement_outcome) here
-                raise NotImplementedError("TODO")
+                # NOTE: there is currently no convenient ilist.any method, so we need to use foldl
+                # with a simple function that just does an or
+
+                def bool_op_or(x: bool, y: bool) -> bool:
+                    return x or y
+
+                f_code = state.current_frame.push(
+                    lowering.Python(self.dialects).python_function(bool_op_or)
+                )
+                fn = ir.Method(
+                    mod=None,
+                    py_func=bool_op_or,
+                    sym_name="bool_op_or",
+                    arg_names=[],
+                    dialects=self.dialects,
+                    code=f_code,
+                )
+                f_const = state.current_frame.push(py.constant.Constant(fn))
+                init_val = state.current_frame.push(py.Constant(False)).result
+                condition = state.current_frame.push(
+                    ilist.Foldl(f_const.result, measurement_outcome, init=init_val)
+                ).result
             else:
                 condition = measurement_outcome
 

test/squin/cirq/test_cirq_to_squin.py

@@ -196,13 +196,12 @@ def test_classical_control_register():
 
 
 def test_multiple_classical_controls(run_sim: bool = False):
-    # TODO: test combination of list & single measurement here
     q = cirq.LineQubit.range(2)
     q2 = cirq.GridQubit(0, 1)
     circuit = cirq.Circuit(
         cirq.H(q[0]),
         cirq.H(q2),
-        cirq.measure(q[0], key="test"),
+        cirq.measure(q, key="test"),
         cirq.measure(q2),
         cirq.X(q[1]).with_classical_controls("test", "q(0, 1)"),
         cirq.measure(q[1]),