Update quread/gates.py

quread/gates.py

@@ -69,4 +69,49 @@ SINGLE_QUBIT_GATES = {
     "Tdg": Tdg,
     "SQRTX": SQRTX,
     "SQRTZ": SQRTZ,
-}
+}
+
+
+def single_qubit_gate_matrix(gate: str) -> np.ndarray:
+    """
+    Returns a 2x2 unitary matrix for supported single-qubit gate labels.
+
+    Supports UI labels too:
+    - "S†" / "Sdg"
+    - "T†" / "Tdg"
+    - "√X" / "SQRTX"
+    - "√Z" / "SQRTZ"
+    - "RX(π)", "RX(π/2)" (and similarly RY/RZ)
+    """
+    g = gate.strip()
+
+    # Normalize common UI labels
+    if g in ("S†", "Sdg"):
+        g = "Sdg"
+    if g in ("T†", "Tdg"):
+        g = "Tdg"
+    if g in ("√X", "SX"):
+        g = "SQRTX"
+    if g in ("√Z", "SZ"):
+        g = "SQRTZ"
+    if g in ("I†", "Idg"):
+        g = "I"
+
+    # Fixed-angle rotation labels (from your palette)
+    if g == "RX(π)":
+        return rx(np.pi)
+    if g == "RX(π/2)":
+        return rx(np.pi / 2)
+    if g == "RY(π)":
+        return ry(np.pi)
+    if g == "RY(π/2)":
+        return ry(np.pi / 2)
+    if g == "RZ(π)":
+        return rz(np.pi)
+    if g == "RZ(π/2)":
+        return rz(np.pi / 2)
+
+    if g in SINGLE_QUBIT_GATES:
+        return SINGLE_QUBIT_GATES[g]
+
+    raise ValueError(f"Unsupported gate: {gate}")