From 7728e5e3276f225de8757cc40e6c46a02bfc3e7e Mon Sep 17 00:00:00 2001
From: Nicolas Quesada <nquesada@pop-os.localdomain>
Date: Mon, 27 Apr 2026 12:05:40 -0400
Subject: [PATCH 1/2] Adds printer

---
 src/phaseshift/bell_interferometer.py | 40 +++++++++++++++++++++++++++
 1 file changed, 40 insertions(+)

diff --git a/src/phaseshift/bell_interferometer.py b/src/phaseshift/bell_interferometer.py
index b25de47..b0450c7 100644
--- a/src/phaseshift/bell_interferometer.py
+++ b/src/phaseshift/bell_interferometer.py
@@ -31,6 +31,7 @@
 It was introduced in [this PR by Jake Bulmer and Yuan Yao](https://github.com/XanaduAI/strawberryfields/pull/584#issue-894649549).
 """
 
+from io import StringIO
 from collections import defaultdict
 from dataclasses import dataclass, field
 
@@ -444,3 +445,42 @@ def circuit_reconstruction(decomp: BellDecomp) -> NDArray[np.complex128]:
         U = phase_shifter(dim, mode, phi) @ U
 
     return U
+
+def circuit_printer(decomp, width=30, rounding=3):
+    """Print the circuit corresponding to the decomposition using qpic syntax.
+
+    Given a `BellDecomp` instance, this function prints the circuits in qpic format.
+    The `BellDecomp` instance can be obtained from the `bell_decomposition` function.
+
+    Args:
+        decomp (BellDecomp): The Bell decomposition containing the parameters of the circuit.
+
+    Returns:
+        ### NEED TO WRITE IT
+
+    Raises:
+        TypeError: If the input is not a BellDecomp instance.
+    """
+    output = StringIO()
+    dim = decomp.dim
+    # Apply the phase shifters at the input
+    for mode, phi in decomp.phi_input.items():
+        print(f"{mode} G ${round(phi/np.pi, rounding)}$ width={width}",file=output)
+
+    # Iterate through the layers of the circuit
+    for layer in range(dim):
+        # Apply the sMZIs in the layer
+        for mode in range(layer % 2, dim - 1, 2):
+            delta = round((decomp.delta[mode, layer] / np.pi) % 1, rounding)
+            sigma = round((decomp.sigma[mode, layer] / np.pi) % 1, rounding)
+            print(f"{mode} {mode+1} G $\\genfrac{{}}{{}}{{0pt}}{{}}{{{delta}}}{{{sigma}}}$ width={width}",file=output)
+        if (dim - 1, layer) in decomp.phi_edge:
+            phi_bottom = round((decomp.phi_edge[dim - 1, layer]/ np.pi) % 1, rounding)
+            print(f"{dim-1} G ${phi_bottom}$ width={width}",file=output)
+    # Apply the phase shifters at the output
+    for mode, phi in decomp.phi_output.items():
+        phi = round((phi/ np.pi) % 1, rounding)
+        print(f"{mode} G ${phi}$ width={width}",file=output)
+    captured_text = output.getvalue()
+    output.close()
+    return captured_text

From 8fe054a0e2ecf38dbbb32991d2050930b67af48e Mon Sep 17 00:00:00 2001
From: Nicolas Quesada <nquesada@pop-os.localdomain>
Date: Mon, 27 Apr 2026 12:06:35 -0400
Subject: [PATCH 2/2] Passes black

---
 src/phaseshift/bell_interferometer.py | 16 ++++++++++------
 1 file changed, 10 insertions(+), 6 deletions(-)

diff --git a/src/phaseshift/bell_interferometer.py b/src/phaseshift/bell_interferometer.py
index b0450c7..1387538 100644
--- a/src/phaseshift/bell_interferometer.py
+++ b/src/phaseshift/bell_interferometer.py
@@ -446,6 +446,7 @@ def circuit_reconstruction(decomp: BellDecomp) -> NDArray[np.complex128]:
 
     return U
 
+
 def circuit_printer(decomp, width=30, rounding=3):
     """Print the circuit corresponding to the decomposition using qpic syntax.
 
@@ -465,7 +466,7 @@ def circuit_printer(decomp, width=30, rounding=3):
     dim = decomp.dim
     # Apply the phase shifters at the input
     for mode, phi in decomp.phi_input.items():
-        print(f"{mode} G ${round(phi/np.pi, rounding)}$ width={width}",file=output)
+        print(f"{mode} G ${round(phi/np.pi, rounding)}$ width={width}", file=output)
 
     # Iterate through the layers of the circuit
     for layer in range(dim):
@@ -473,14 +474,17 @@ def circuit_printer(decomp, width=30, rounding=3):
         for mode in range(layer % 2, dim - 1, 2):
             delta = round((decomp.delta[mode, layer] / np.pi) % 1, rounding)
             sigma = round((decomp.sigma[mode, layer] / np.pi) % 1, rounding)
-            print(f"{mode} {mode+1} G $\\genfrac{{}}{{}}{{0pt}}{{}}{{{delta}}}{{{sigma}}}$ width={width}",file=output)
+            print(
+                f"{mode} {mode+1} G $\\genfrac{{}}{{}}{{0pt}}{{}}{{{delta}}}{{{sigma}}}$ width={width}",
+                file=output,
+            )
         if (dim - 1, layer) in decomp.phi_edge:
-            phi_bottom = round((decomp.phi_edge[dim - 1, layer]/ np.pi) % 1, rounding)
-            print(f"{dim-1} G ${phi_bottom}$ width={width}",file=output)
+            phi_bottom = round((decomp.phi_edge[dim - 1, layer] / np.pi) % 1, rounding)
+            print(f"{dim-1} G ${phi_bottom}$ width={width}", file=output)
     # Apply the phase shifters at the output
     for mode, phi in decomp.phi_output.items():
-        phi = round((phi/ np.pi) % 1, rounding)
-        print(f"{mode} G ${phi}$ width={width}",file=output)
+        phi = round((phi / np.pi) % 1, rounding)
+        print(f"{mode} G ${phi}$ width={width}", file=output)
     captured_text = output.getvalue()
     output.close()
     return captured_text