Feat/more sims

Cargo.toml

@@ -24,6 +24,7 @@ pyo3 = "0.23"
 rand = "0.8"
 rand_chacha = "0.3"
 rand_xoshiro = "0.6"
+num-complex = "0.4"
 
 pecos-core = { version = "0.1.1", path = "crates/pecos-core" }
 pecos-qsim = { version = "0.1.1", path = "crates/pecos-qsim" }

crates/pecos-python/Cargo.toml

@@ -17,6 +17,8 @@ crate-type = ["cdylib"]
 [dependencies]
 pyo3 = { workspace=true, features = ["extension-module"] }
 pecos = { workspace = true }
+num-complex = { workspace = true }
+rand = { workspace = true }
 
 [lints]
 workspace = true

crates/pecos-python/src/lib.rs

@@ -10,13 +10,17 @@
 // or implied. See the License for the specific language governing permissions and limitations under
 // the License.
 
-mod sparse_sim;
-use sparse_sim::SparseSim;
+mod sparse_stab_bindings;
+mod state_vec_bindings;
+
+use sparse_stab_bindings::SparseSim;
+use state_vec_bindings::RsStateVec;
 
 use pyo3::prelude::*;
 
 #[pymodule]
 fn _pecos_rslib(_py: Python<'_>, m: &Bound<'_, PyModule>) -> PyResult<()> {
     m.add_class::<SparseSim>()?;
+    m.add_class::<RsStateVec>()?;
     Ok(())
 }

crates/pecos-python/src/sparse_sim.rs → crates/pecos-python/src/sparse_stab_bindings.rs

@@ -133,9 +133,48 @@ impl SparseSim {
                 self.inner.szdg(location);
                 Ok(None)
             }
-            "MZ" | "MX" | "MY" | "MZForced" | "PZ" | "PX" | "PY" | "PZForced" | "PnZ" | "PnX"
-            | "PnY" => {
-                let (result, _) = match symbol {
+            "PZ" => {
+                self.inner.pz(location);
+                Ok(None)
+            }
+            "PX" => {
+                self.inner.px(location);
+                Ok(None)
+            }
+            "PY" => {
+                self.inner.py(location);
+                Ok(None)
+            }
+            "PnZ" => {
+                self.inner.pnz(location);
+                Ok(None)
+            }
+            "PnX" => {
+                self.inner.pnx(location);
+                Ok(None)
+            }
+            "PnY" => {
+                self.inner.pny(location);
+                Ok(None)
+            }
+            "PZForced" => {
+                let forced_value = params
+                    .ok_or_else(|| {
+                        PyErr::new::<pyo3::exceptions::PyValueError, _>("PZForced requires params")
+                    })?
+                    .get_item("forced_outcome")?
+                    .ok_or_else(|| {
+                        PyErr::new::<pyo3::exceptions::PyValueError, _>(
+                            "PZForced requires a 'forced_outcome' parameter",
+                        )
+                    })?
+                    .call_method0("__bool__")?
+                    .extract::<bool>()?;
+                self.inner.pz_forced(location, forced_value);
+                Ok(None)
+            }
+            "MZ" | "MX" | "MY" | "MZForced" => {
+                let result = match symbol {
                     "MZ" => self.inner.mz(location),
                     "MX" => self.inner.mx(location),
                     "MY" => self.inner.my(location),
@@ -156,32 +195,9 @@ impl SparseSim {
                             .extract::<bool>()?;
                         self.inner.mz_forced(location, forced_value)
                     }
-                    "PZ" => self.inner.pz(location),
-                    "PX" => self.inner.px(location),
-                    "PY" => self.inner.py(location),
-                    "PZForced" => {
-                        let forced_value = params
-                            .ok_or_else(|| {
-                                PyErr::new::<pyo3::exceptions::PyValueError, _>(
-                                    "PZForced requires params",
-                                )
-                            })?
-                            .get_item("forced_outcome")?
-                            .ok_or_else(|| {
-                                PyErr::new::<pyo3::exceptions::PyValueError, _>(
-                                    "PZForced requires a 'forced_outcome' parameter",
-                                )
-                            })?
-                            .call_method0("__bool__")?
-                            .extract::<bool>()?;
-                        self.inner.pz_forced(location, forced_value)
-                    }
-                    "PnZ" => self.inner.pnz(location),
-                    "PnX" => self.inner.pnx(location),
-                    "PnY" => self.inner.pny(location),
                     _ => unreachable!(),
                 };
-                Ok(Some(u8::from(result)))
+                Ok(Some(u8::from(result.outcome)))
             }
             _ => Err(PyErr::new::<pyo3::exceptions::PyValueError, _>(
                 "Unsupported single-qubit gate",
@@ -247,7 +263,7 @@ impl SparseSim {
                 Ok(None)
             }
             "G2" => {
-                self.inner.g2(q1, q2);
+                self.inner.g(q1, q2);
                 Ok(None)
             }
             _ => Err(PyErr::new::<pyo3::exceptions::PyValueError, _>(