use super::{FermionOperator, FermionProduct, OperateOnFermions};
use crate::mappings::JordanWignerFermionToSpin;
use crate::spins::{DecoherenceOperator, SpinLindbladNoiseOperator};
use crate::{
ModeIndex, OperateOnDensityMatrix, OperateOnModes, StruqtureError,
StruqtureVersionSerializable, MINIMUM_STRUQTURE_VERSION,
};
use itertools::Itertools;
use qoqo_calculator::{CalculatorComplex, CalculatorFloat};
use serde::{Deserialize, Serialize};
use std::fmt::{self, Write};
use std::iter::{FromIterator, IntoIterator};
use std::ops;
#[cfg(feature = "indexed_map_iterators")]
use indexmap::map::{Entry, Iter, Keys, Values};
#[cfg(feature = "indexed_map_iterators")]
use indexmap::IndexMap;
#[cfg(not(feature = "indexed_map_iterators"))]
use std::collections::hash_map::{Entry, Iter, Keys, Values};
#[cfg(not(feature = "indexed_map_iterators"))]
use std::collections::HashMap;
#[derive(Debug, Clone, PartialEq, Deserialize, Serialize)]
#[serde(from = "FermionLindbladNoiseOperatorSerialize")]
#[serde(into = "FermionLindbladNoiseOperatorSerialize")]
pub struct FermionLindbladNoiseOperator {
#[cfg(feature = "indexed_map_iterators")]
internal_map: IndexMap<(FermionProduct, FermionProduct), CalculatorComplex>,
#[cfg(not(feature = "indexed_map_iterators"))]
internal_map: HashMap<(FermionProduct, FermionProduct), CalculatorComplex>,
}
impl crate::MinSupportedVersion for FermionLindbladNoiseOperator {}
#[cfg(feature = "json_schema")]
impl schemars::JsonSchema for FermionLindbladNoiseOperator {
fn schema_name() -> std::borrow::Cow<'static, str> {
"FermionLindbladNoiseOperator".into()
}
fn json_schema(generator: &mut schemars::SchemaGenerator) -> schemars::Schema {
<FermionLindbladNoiseOperatorSerialize>::json_schema(generator)
}
}
#[derive(Debug, Clone, PartialEq, Deserialize, Serialize)]
#[cfg_attr(feature = "json_schema", derive(schemars::JsonSchema))]
#[cfg_attr(feature = "json_schema", schemars(deny_unknown_fields))]
struct FermionLindbladNoiseOperatorSerialize {
items: Vec<(
FermionProduct,
FermionProduct,
CalculatorFloat,
CalculatorFloat,
)>,
_struqture_version: StruqtureVersionSerializable,
}
impl From<FermionLindbladNoiseOperatorSerialize> for FermionLindbladNoiseOperator {
fn from(value: FermionLindbladNoiseOperatorSerialize) -> Self {
let new_noise_op: FermionLindbladNoiseOperator = value
.items
.into_iter()
.map(|(left, right, real, imag)| {
((left, right), CalculatorComplex { re: real, im: imag })
})
.collect();
new_noise_op
}
}
impl From<FermionLindbladNoiseOperator> for FermionLindbladNoiseOperatorSerialize {
fn from(value: FermionLindbladNoiseOperator) -> Self {
let new_noise_op: Vec<(
FermionProduct,
FermionProduct,
CalculatorFloat,
CalculatorFloat,
)> = value
.into_iter()
.map(|((left, right), val)| (left, right, val.re, val.im))
.collect();
let current_version = StruqtureVersionSerializable {
major_version: MINIMUM_STRUQTURE_VERSION.0,
minor_version: MINIMUM_STRUQTURE_VERSION.1,
};
Self {
items: new_noise_op,
_struqture_version: current_version,
}
}
}
impl<'a> OperateOnDensityMatrix<'a> for FermionLindbladNoiseOperator {
type Index = (FermionProduct, FermionProduct);
type Value = CalculatorComplex;
type IteratorType = Iter<'a, (FermionProduct, FermionProduct), CalculatorComplex>;
type KeyIteratorType = Keys<'a, (FermionProduct, FermionProduct), CalculatorComplex>;
type ValueIteratorType = Values<'a, (FermionProduct, FermionProduct), CalculatorComplex>;
fn get(&self, key: &Self::Index) -> &Self::Value {
match self.internal_map.get(key) {
Some(value) => value,
None => &CalculatorComplex::ZERO,
}
}
fn iter(&'a self) -> Self::IteratorType {
self.internal_map.iter()
}
fn keys(&'a self) -> Self::KeyIteratorType {
self.internal_map.keys()
}
fn values(&'a self) -> Self::ValueIteratorType {
self.internal_map.values()
}
#[cfg(feature = "indexed_map_iterators")]
fn remove(&mut self, key: &Self::Index) -> Option<Self::Value> {
self.internal_map.shift_remove(key)
}
#[cfg(not(feature = "indexed_map_iterators"))]
fn remove(&mut self, key: &Self::Index) -> Option<Self::Value> {
self.internal_map.remove(key)
}
fn empty_clone(&self, capacity: Option<usize>) -> Self {
match capacity {
Some(cap) => Self::with_capacity(cap),
None => Self::new(),
}
}
fn set(
&mut self,
key: Self::Index,
value: Self::Value,
) -> Result<Option<Self::Value>, StruqtureError> {
if key.0 == FermionProduct::new([], [])? || key.1 == FermionProduct::new([], [])? {
return Err(StruqtureError::InvalidLindbladTerms);
}
if value != CalculatorComplex::ZERO {
Ok(self.internal_map.insert(key, value))
} else {
match self.internal_map.entry(key) {
#[cfg(feature = "indexed_map_iterators")]
Entry::Occupied(val) => Ok(Some(val.shift_remove())),
#[cfg(not(feature = "indexed_map_iterators"))]
Entry::Occupied(val) => Ok(Some(val.remove())),
Entry::Vacant(_) => Ok(None),
}
}
}
}
impl<'a> OperateOnModes<'a> for FermionLindbladNoiseOperator {
fn current_number_modes(&'a self) -> usize {
let mut max_mode: usize = 0;
if !self.is_empty() {
for key in self.keys() {
let maxk = key
.0
.current_number_modes()
.max(key.1.current_number_modes());
if maxk > max_mode {
max_mode = maxk;
}
}
}
max_mode
}
fn number_modes(&'a self) -> usize {
self.current_number_modes()
}
}
impl OperateOnFermions<'_> for FermionLindbladNoiseOperator {}
impl Default for FermionLindbladNoiseOperator {
fn default() -> Self {
Self::new()
}
}
impl FermionLindbladNoiseOperator {
pub fn new() -> Self {
FermionLindbladNoiseOperator {
#[cfg(not(feature = "indexed_map_iterators"))]
internal_map: HashMap::new(),
#[cfg(feature = "indexed_map_iterators")]
internal_map: IndexMap::new(),
}
}
pub fn with_capacity(capacity: usize) -> Self {
FermionLindbladNoiseOperator {
#[cfg(not(feature = "indexed_map_iterators"))]
internal_map: HashMap::with_capacity(capacity),
#[cfg(feature = "indexed_map_iterators")]
internal_map: IndexMap::with_capacity(capacity),
}
}
pub fn add_noise_from_full_operators(
&mut self,
left: &FermionOperator,
right: &FermionOperator,
value: CalculatorComplex,
) -> Result<(), StruqtureError> {
if left.is_empty() || right.is_empty() {
return Err(StruqtureError::InvalidLindbladTerms);
}
for ((fermion_product_left, value_left), (fermion_product_right, value_right)) in
left.iter().cartesian_product(right.iter())
{
if !(*fermion_product_left == FermionProduct::new([], [])?
|| *fermion_product_right == FermionProduct::new([], [])?)
{
let value_complex = value_right.conj() * value_left;
self.add_operator_product(
(fermion_product_left.clone(), fermion_product_right.clone()),
value_complex * value.clone(),
)?;
}
}
Ok(())
}
pub fn separate_into_n_terms(
&self,
number_creators_annihilators_left: (usize, usize),
number_creators_annihilators_right: (usize, usize),
) -> Result<(Self, Self), StruqtureError> {
let mut separated = Self::default();
let mut remainder = Self::default();
for ((prod_l, prod_r), val) in self.iter() {
if (prod_l.creators().len(), prod_l.annihilators().len())
== number_creators_annihilators_left
&& (prod_r.creators().len(), prod_r.annihilators().len())
== number_creators_annihilators_right
{
separated.add_operator_product((prod_l.clone(), prod_r.clone()), val.clone())?;
} else {
remainder.add_operator_product((prod_l.clone(), prod_r.clone()), val.clone())?;
}
}
Ok((separated, remainder))
}
}
impl ops::Neg for FermionLindbladNoiseOperator {
type Output = FermionLindbladNoiseOperator;
fn neg(self) -> Self {
#[cfg(not(feature = "indexed_map_iterators"))]
let mut internal = HashMap::with_capacity(self.len());
#[cfg(feature = "indexed_map_iterators")]
let mut internal = IndexMap::with_capacity(self.len());
for (key, val) in self {
internal.insert(key.clone(), val.neg());
}
FermionLindbladNoiseOperator {
internal_map: internal,
}
}
}
impl<T, V> ops::Add<T> for FermionLindbladNoiseOperator
where
T: IntoIterator<Item = ((FermionProduct, FermionProduct), V)>,
V: Into<CalculatorComplex>,
{
type Output = Self;
fn add(mut self, other: T) -> Self {
for (key, value) in other.into_iter() {
self.add_operator_product(key.clone(), Into::<CalculatorComplex>::into(value))
.expect("Internal bug in add_operator_product");
}
self
}
}
impl<T, V> ops::Sub<T> for FermionLindbladNoiseOperator
where
T: IntoIterator<Item = ((FermionProduct, FermionProduct), V)>,
V: Into<CalculatorComplex>,
{
type Output = Self;
fn sub(mut self, other: T) -> Self {
for (key, value) in other.into_iter() {
self.add_operator_product(key.clone(), Into::<CalculatorComplex>::into(value) * -1.0)
.expect("Internal bug in add_operator_product");
}
self
}
}
impl<T> ops::Mul<T> for FermionLindbladNoiseOperator
where
T: Into<CalculatorComplex>,
{
type Output = Self;
fn mul(self, other: T) -> Self {
let other_cc = Into::<CalculatorComplex>::into(other);
#[cfg(not(feature = "indexed_map_iterators"))]
let mut internal = HashMap::with_capacity(self.len());
#[cfg(feature = "indexed_map_iterators")]
let mut internal = IndexMap::with_capacity(self.len());
for (key, val) in self {
internal.insert(key, val * other_cc.clone());
}
FermionLindbladNoiseOperator {
internal_map: internal,
}
}
}
impl IntoIterator for FermionLindbladNoiseOperator {
type Item = ((FermionProduct, FermionProduct), CalculatorComplex);
#[cfg(not(feature = "indexed_map_iterators"))]
type IntoIter =
std::collections::hash_map::IntoIter<(FermionProduct, FermionProduct), CalculatorComplex>;
#[cfg(feature = "indexed_map_iterators")]
type IntoIter = indexmap::map::IntoIter<(FermionProduct, FermionProduct), CalculatorComplex>;
fn into_iter(self) -> Self::IntoIter {
self.internal_map.into_iter()
}
}
impl<'a> IntoIterator for &'a FermionLindbladNoiseOperator {
type Item = (&'a (FermionProduct, FermionProduct), &'a CalculatorComplex);
type IntoIter = Iter<'a, (FermionProduct, FermionProduct), CalculatorComplex>;
fn into_iter(self) -> Self::IntoIter {
self.internal_map.iter()
}
}
impl FromIterator<((FermionProduct, FermionProduct), CalculatorComplex)>
for FermionLindbladNoiseOperator
{
fn from_iter<I: IntoIterator<Item = ((FermionProduct, FermionProduct), CalculatorComplex)>>(
iter: I,
) -> Self {
let mut slno = FermionLindbladNoiseOperator::new();
for (pair, cc) in iter {
slno.add_operator_product(pair, cc)
.expect("Internal bug in add_operator_product");
}
slno
}
}
impl Extend<((FermionProduct, FermionProduct), CalculatorComplex)>
for FermionLindbladNoiseOperator
{
fn extend<I: IntoIterator<Item = ((FermionProduct, FermionProduct), CalculatorComplex)>>(
&mut self,
iter: I,
) {
for (pair, cc) in iter {
self.add_operator_product(pair, cc)
.expect("Internal bug in add_operator_product");
}
}
}
impl fmt::Display for FermionLindbladNoiseOperator {
fn fmt(&self, f: &mut std::fmt::Formatter<'_>) -> std::fmt::Result {
let mut output = "FermionLindbladNoiseOperator{\n".to_string();
for (key, val) in self.iter() {
writeln!(output, "({}, {}): {},", key.0, key.1, val)?;
}
output.push('}');
write!(f, "{output}")
}
}
impl JordanWignerFermionToSpin for FermionLindbladNoiseOperator {
type Output = SpinLindbladNoiseOperator;
fn jordan_wigner(&self) -> Self::Output {
let mut out = SpinLindbladNoiseOperator::new();
for key in self.keys() {
let decoherence_operator_left = DecoherenceOperator::from(key.0.jordan_wigner());
let decoherence_operator_right = DecoherenceOperator::from(key.1.jordan_wigner());
out.add_noise_from_full_operators(
&decoherence_operator_left,
&decoherence_operator_right,
self.get(key).into(),
)
.expect("Internal bug in add_noise_from_full_operators");
}
out
}
}
#[cfg(test)]
mod test {
use super::*;
use serde_test::{assert_tokens, Configure, Token};
#[test]
fn so_from_sos() {
let pp: FermionProduct = FermionProduct::new([0], [0]).unwrap();
let sos = FermionLindbladNoiseOperatorSerialize {
items: vec![(pp.clone(), pp.clone(), 0.5.into(), 0.0.into())],
_struqture_version: StruqtureVersionSerializable {
major_version: 1,
minor_version: 0,
},
};
let mut so = FermionLindbladNoiseOperator::new();
so.set((pp.clone(), pp), CalculatorComplex::from(0.5))
.unwrap();
assert_eq!(FermionLindbladNoiseOperator::from(sos.clone()), so);
assert_eq!(FermionLindbladNoiseOperatorSerialize::from(so), sos);
}
#[test]
fn clone_partial_eq() {
let pp: FermionProduct = FermionProduct::new([0], [0]).unwrap();
let sos = FermionLindbladNoiseOperatorSerialize {
items: vec![(pp.clone(), pp, 0.5.into(), 0.0.into())],
_struqture_version: StruqtureVersionSerializable {
major_version: 1,
minor_version: 0,
},
};
assert_eq!(sos.clone(), sos);
let pp_1: FermionProduct = FermionProduct::new([0], [0]).unwrap();
let sos_1 = FermionLindbladNoiseOperatorSerialize {
items: vec![(pp_1.clone(), pp_1, 0.5.into(), 0.0.into())],
_struqture_version: StruqtureVersionSerializable {
major_version: 1,
minor_version: 0,
},
};
let pp_2: FermionProduct = FermionProduct::new([0], [1]).unwrap();
let sos_2 = FermionLindbladNoiseOperatorSerialize {
items: vec![(pp_2.clone(), pp_2, 0.5.into(), 0.0.into())],
_struqture_version: StruqtureVersionSerializable {
major_version: 1,
minor_version: 0,
},
};
assert!(sos_1 == sos);
assert!(sos == sos_1);
assert!(sos_2 != sos);
assert!(sos != sos_2);
}
#[test]
fn debug() {
let pp: FermionProduct = FermionProduct::new([0], [0]).unwrap();
let sos = FermionLindbladNoiseOperatorSerialize {
items: vec![(pp.clone(), pp, 0.5.into(), 0.0.into())],
_struqture_version: StruqtureVersionSerializable {
major_version: 1,
minor_version: 0,
},
};
assert_eq!(
format!("{sos:?}"),
"FermionLindbladNoiseOperatorSerialize { items: [(FermionProduct { creators: [0], annihilators: [0] }, FermionProduct { creators: [0], annihilators: [0] }, Float(0.5), Float(0.0))], _struqture_version: StruqtureVersionSerializable { major_version: 1, minor_version: 0 } }"
);
}
#[test]
fn serde_readable() {
let pp: FermionProduct = FermionProduct::new([0], [0]).unwrap();
let sos = FermionLindbladNoiseOperatorSerialize {
items: vec![(pp.clone(), pp, 0.5.into(), 0.0.into())],
_struqture_version: StruqtureVersionSerializable {
major_version: 1,
minor_version: 0,
},
};
assert_tokens(
&sos.readable(),
&[
Token::Struct {
name: "FermionLindbladNoiseOperatorSerialize",
len: 2,
},
Token::Str("items"),
Token::Seq { len: Some(1) },
Token::Tuple { len: 4 },
Token::Str("c0a0"),
Token::Str("c0a0"),
Token::F64(0.5),
Token::F64(0.0),
Token::TupleEnd,
Token::SeqEnd,
Token::Str("_struqture_version"),
Token::Struct {
name: "StruqtureVersionSerializable",
len: 2,
},
Token::Str("major_version"),
Token::U32(1),
Token::Str("minor_version"),
Token::U32(0),
Token::StructEnd,
Token::StructEnd,
],
);
}
#[test]
fn serde_compact() {
let pp: FermionProduct = FermionProduct::new([0], [0]).unwrap();
let sos = FermionLindbladNoiseOperatorSerialize {
items: vec![(pp.clone(), pp, 0.5.into(), 0.0.into())],
_struqture_version: StruqtureVersionSerializable {
major_version: 1,
minor_version: 0,
},
};
assert_tokens(
&sos.compact(),
&[
Token::Struct {
name: "FermionLindbladNoiseOperatorSerialize",
len: 2,
},
Token::Str("items"),
Token::Seq { len: Some(1) },
Token::Tuple { len: 4 },
Token::Tuple { len: 2 },
Token::Seq { len: Some(1) },
Token::U64(0),
Token::SeqEnd,
Token::Seq { len: Some(1) },
Token::U64(0),
Token::SeqEnd,
Token::TupleEnd,
Token::Tuple { len: 2 },
Token::Seq { len: Some(1) },
Token::U64(0),
Token::SeqEnd,
Token::Seq { len: Some(1) },
Token::U64(0),
Token::SeqEnd,
Token::TupleEnd,
Token::NewtypeVariant {
name: "CalculatorFloat",
variant: "Float",
},
Token::F64(0.5),
Token::NewtypeVariant {
name: "CalculatorFloat",
variant: "Float",
},
Token::F64(0.0),
Token::TupleEnd,
Token::SeqEnd,
Token::Str("_struqture_version"),
Token::Struct {
name: "StruqtureVersionSerializable",
len: 2,
},
Token::Str("major_version"),
Token::U32(1),
Token::Str("minor_version"),
Token::U32(0),
Token::StructEnd,
Token::StructEnd,
],
);
}
}