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
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144

use std::ops::Deref;

use pauli::{Pauli, PauliOperator};
use serde::{Deserialize, Serialize};
use sparse_bin_mat::{SparseBinSlice, SparseBinVec, SparseBinVecBase};

#[derive(Debug, PartialEq, Eq, Clone, Hash, Serialize, Deserialize)]
pub struct Css<X, Z = X> {
    pub x: X,
    pub z: Z,
}

impl<T> Css<T> {
    pub fn map<'a, F, S>(&'a self, func: F) -> Css<S>
    where
        F: Fn(&'a T) -> S,
    {
        Css {
            x: func(&self.x),
            z: func(&self.z),
        }
    }

    pub fn map_with_pauli<'a, F, S>(&'a self, func: F) -> Css<S>
    where
        F: Fn(&'a T, Pauli) -> S,
    {
        Css {
            x: func(&self.x, Pauli::X),
            z: func(&self.z, Pauli::Z),
        }
    }

    pub fn both<F>(&self, func: F) -> bool
    where
        F: Fn(&T) -> bool,
    {
        func(&self.x) && func(&self.z)
    }
}

impl<X, Z> Css<X, Z> {
    pub fn map_each<'a, F, G, XX, ZZ>(&'a self, funcs: Css<F, G>) -> Css<XX, ZZ>
    where
        F: Fn(&'a X) -> XX,
        G: Fn(&'a Z) -> ZZ,
    {
        Css {
            x: (funcs.x)(&self.x),
            z: (funcs.z)(&self.z),
        }
    }

    pub fn pair<XX, ZZ>(self, other: Css<XX, ZZ>) -> Css<(X, XX), (Z, ZZ)> {
        Css {
            x: (self.x, other.x),
            z: (self.z, other.z),
        }
    }

    pub fn combine_with<F, T>(self, func: F) -> T
    where
        F: Fn(X, Z) -> T,
    {
        func(self.x, self.z)
    }

    pub fn swap_xz(self) -> Css<Z, X> {
        Css {
            x: self.z,
            z: self.x,
        }
    }

    pub fn as_ref(&self) -> Css<&X, &Z> {
        Css {
            x: &self.x,
            z: &self.z,
        }
    }

    pub fn as_mut(&mut self) -> Css<&mut X, &mut Z> {
        Css {
            x: &mut self.x,
            z: &mut self.z,
        }
    }
}

pub type CssOperator = Css<SparseBinVec>;

impl<'a> From<&'a PauliOperator> for CssOperator {
    fn from(operator: &'a PauliOperator) -> Self {
        Self {
            x: SparseBinVec::new(operator.len(), operator.x_part().into_raw_positions()),
            z: SparseBinVec::new(operator.len(), operator.z_part().into_raw_positions()),
        }
    }
}

impl From<PauliOperator> for CssOperator {
    fn from(operator: PauliOperator) -> Self {
        Self::from(&operator)
    }
}

impl<'a> From<&'a CssOperator> for PauliOperator {
    fn from(operator: &'a CssOperator) -> Self {
        let x = PauliOperator::new(
            operator.x.len(),
            operator.x.non_trivial_positions().collect(),
            vec![Pauli::X; operator.x.weight()],
        );
        let z = PauliOperator::new(
            operator.z.len(),
            operator.z.non_trivial_positions().collect(),
            vec![Pauli::Z; operator.z.weight()],
        );
        &x * &z
    }
}

impl From<CssOperator> for PauliOperator {
    fn from(operator: CssOperator) -> Self {
        Self::from(&operator)
    }
}


pub type CssSyndrome<T = Vec<usize>> = Css<SparseBinVecBase<T>>;
pub type CssSyndromeView<'a> = Css<SparseBinSlice<'a>>;

impl<T> CssSyndrome<T>
where
    T: Deref<Target = [usize]>,
{
    pub fn is_trivial(&self) -> bool {
        self.both(|syndrome| syndrome.is_zero())
    }

    pub fn as_view(&self) -> CssSyndromeView {
        self.map(|syndrome| syndrome.as_view())
    }
}