1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42

use quil_rs::instruction::{Qubit, QubitPlaceholder};

use rigetti_pyo3::{
    impl_compare, impl_hash, impl_repr, py_wrap_type, py_wrap_union_enum,
    pyo3::{
        pymethods,
        types::{PyLong, PyString},
        Py,
    },
};

use crate::{impl_eq, impl_to_quil};

py_wrap_union_enum! {
    #[derive(Debug, Eq, Hash, PartialEq)]
    PyQubit(Qubit) as "Qubit" {
        fixed: Fixed => Py<PyLong>,
        variable: Variable => Py<PyString>,
        placeholder: Placeholder => PyQubitPlaceholder
    }
}
impl_repr!(PyQubit);
impl_to_quil!(PyQubit);
impl_hash!(PyQubit);
impl_eq!(PyQubit);

py_wrap_type! {
    #[pyo3(subclass)]
    #[derive(Debug, PartialEq, Eq, PartialOrd, Ord)]
    PyQubitPlaceholder(QubitPlaceholder) as "QubitPlaceholder"
}
impl_repr!(PyQubitPlaceholder);
impl_hash!(PyQubitPlaceholder);
impl_compare!(PyQubitPlaceholder);

#[pymethods]
impl PyQubitPlaceholder {
    #[new]
    fn new() -> Self {
        Self(QubitPlaceholder::default())
    }
}