Add mpp_state{_unsigned} methods to stim.Tableau.to_circuit

Author: Strilanc

src/stim/circuit/gate_target.cc

@@ -18,6 +18,13 @@
 
 using namespace stim;
 
+GateTarget GateTarget::pauli_xz(uint32_t qubit, bool x, bool z, bool inverted) {
+    if (qubit != (qubit & TARGET_VALUE_MASK)) {
+        throw std::invalid_argument("qubit target larger than " + std::to_string(TARGET_VALUE_MASK));
+    }
+    return {qubit | (TARGET_INVERTED_BIT * inverted) | (TARGET_PAULI_X_BIT * x) | (TARGET_PAULI_Z_BIT * z)};
+}
+
 GateTarget GateTarget::x(uint32_t qubit, bool inverted) {
     if (qubit != (qubit & TARGET_VALUE_MASK)) {
         throw std::invalid_argument("qubit target larger than " + std::to_string(TARGET_VALUE_MASK));

src/stim/circuit/gate_target.h

@@ -39,6 +39,7 @@ struct GateTarget {
     static GateTarget x(uint32_t qubit, bool inverted = false);
     static GateTarget y(uint32_t qubit, bool inverted = false);
     static GateTarget z(uint32_t qubit, bool inverted = false);
+    static GateTarget pauli_xz(uint32_t qubit, bool x, bool z, bool inverted = false);
     static GateTarget qubit(uint32_t qubit, bool inverted = false);
     static GateTarget rec(int32_t lookback);
     static GateTarget sweep_bit(uint32_t index);

src/stim/gates/gates.test.cc

@@ -79,8 +79,8 @@ std::pair<std::vector<PauliString<W>>, std::vector<PauliString<W>>> circuit_outp
     }
     TableauSimulator<W> sim1(INDEPENDENT_TEST_RNG(), circuit.count_qubits(), -1);
     TableauSimulator<W> sim2(INDEPENDENT_TEST_RNG(), circuit.count_qubits(), +1);
-    sim1.expand_do_circuit(circuit);
-    sim2.expand_do_circuit(circuit);
+    sim1.safe_do_circuit(circuit);
+    sim2.safe_do_circuit(circuit);
     return {sim1.canonical_stabilizers(), sim2.canonical_stabilizers()};
 }
 

src/stim/simulators/graph_simulator.test.cc

@@ -16,8 +16,8 @@ void expect_graph_circuit_is_equivalent(const Circuit &circuit) {
     Circuit converted = sim.to_circuit();
     TableauSimulator<64> sim1(std::mt19937_64{}, n);
     TableauSimulator<64> sim2(std::mt19937_64{}, n);
-    sim1.expand_do_circuit(circuit);
-    sim2.expand_do_circuit(converted);
+    sim1.safe_do_circuit(circuit);
+    sim2.safe_do_circuit(converted);
     auto s1 = sim1.canonical_stabilizers();
     auto s2 = sim2.canonical_stabilizers();
     if (s1 != s2) {
@@ -35,9 +35,9 @@ void expect_graph_sim_effect_matches_tableau_sim(const GraphSimulator &state, co
 
     TableauSimulator<64> tableau_sim1(std::mt19937_64{}, n);
     TableauSimulator<64> tableau_sim2(std::mt19937_64{}, n);
-    tableau_sim1.expand_do_circuit(state.to_circuit());
-    tableau_sim1.expand_do_circuit(effect);
-    tableau_sim2.expand_do_circuit(state_after_effect.to_circuit());
+    tableau_sim1.safe_do_circuit(state.to_circuit());
+    tableau_sim1.safe_do_circuit(effect);
+    tableau_sim2.safe_do_circuit(state_after_effect.to_circuit());
     auto s1 = tableau_sim1.canonical_stabilizers();
     auto s2 = tableau_sim2.canonical_stabilizers();
     if (s1 != s2) {
@@ -240,8 +240,8 @@ TEST(graph_simulator, do_complementation) {
 
         TableauSimulator<64> tableau_sim1(std::mt19937_64{}, 8);
         TableauSimulator<64> tableau_sim2(std::mt19937_64{}, 8);
-        tableau_sim1.expand_do_circuit(sim.to_circuit());
-        tableau_sim2.expand_do_circuit(sim2.to_circuit());
+        tableau_sim1.safe_do_circuit(sim.to_circuit());
+        tableau_sim2.safe_do_circuit(sim2.to_circuit());
         auto s1 = tableau_sim1.canonical_stabilizers();
         auto s2 = tableau_sim2.canonical_stabilizers();
         if (s1 != s2) {

src/stim/simulators/tableau_simulator.h

@@ -96,7 +96,7 @@ struct TableauSimulator {
     /// Runs all of the operations in the given circuit.
     ///
     /// Automatically expands the tableau simulator's state, if needed.
-    void expand_do_circuit(const Circuit &circuit, uint64_t reps = 1);
+    void safe_do_circuit(const Circuit &circuit, uint64_t reps = 1);
     void do_operation_ensure_size(const CircuitInstruction &operation);
 
     void apply_tableau(const Tableau<W> &tableau, const std::vector<size_t> &targets);

src/stim/simulators/tableau_simulator.inl

@@ -1107,7 +1107,7 @@ void TableauSimulator<W>::do_Z(const CircuitInstruction &target_data) {
 template <size_t W>
 simd_bits<W> TableauSimulator<W>::sample_circuit(const Circuit &circuit, std::mt19937_64 &rng, int8_t sign_bias) {
     TableauSimulator<W> sim(std::move(rng), circuit.count_qubits(), sign_bias);
-    sim.expand_do_circuit(circuit);
+    sim.safe_do_circuit(circuit);
 
     const std::vector<bool> &v = sim.measurement_record.storage;
     simd_bits<W> result(v.size());
@@ -1347,7 +1347,7 @@ void TableauSimulator<W>::collapse_isolate_qubit_z(size_t target, TableauTranspo
 }
 
 template <size_t W>
-void TableauSimulator<W>::expand_do_circuit(const Circuit &circuit, uint64_t reps) {
+void TableauSimulator<W>::safe_do_circuit(const Circuit &circuit, uint64_t reps) {
     ensure_large_enough_for_qubits(circuit.count_qubits());
     for (uint64_t k = 0; k < reps; k++) {
         circuit.for_each_operation([&](const CircuitInstruction &op) {

src/stim/simulators/tableau_simulator.pybind.cc

@@ -29,10 +29,10 @@ using namespace stim_pybind;
 template <size_t W>
 void do_obj(TableauSimulator<W> &self, const pybind11::object &obj) {
     if (pybind11::isinstance<Circuit>(obj)) {
-        self.expand_do_circuit(pybind11::cast<Circuit>(obj));
+        self.safe_do_circuit(pybind11::cast<Circuit>(obj));
     } else if (pybind11::isinstance<CircuitRepeatBlock>(obj)) {
         const CircuitRepeatBlock &block = pybind11::cast<CircuitRepeatBlock>(obj);
-        self.expand_do_circuit(block.body, block.repeat_count);
+        self.safe_do_circuit(block.body, block.repeat_count);
     } else if (pybind11::isinstance<FlexPauliString>(obj)) {
         const FlexPauliString &pauli_string = pybind11::cast<FlexPauliString>(obj);
         self.ensure_large_enough_for_qubits(pauli_string.value.num_qubits);
@@ -474,7 +474,7 @@ void stim_pybind::pybind_tableau_simulator_methods(
     c.def(
         "do_circuit",
         [](TableauSimulator<MAX_BITWORD_WIDTH> &self, const Circuit &circuit) {
-            self.expand_do_circuit(circuit);
+            self.safe_do_circuit(circuit);
         },
         pybind11::arg("circuit"),
         clean_doc_string(R"DOC(