use num_traits::{One, Zero};
use crate::{CombineOperation, Operation};
pub trait Identity<T> {
fn is_identity(&self) -> bool;
fn identity(w_out: usize, w_in: usize) -> Self;
}
impl Identity<u64> for Operation<u64> {
fn is_identity(&self) -> bool {
match self {
Operation::AddConst(_, _, c) => c.is_zero(),
Operation::SubConst(_, _, c) => c.is_zero(),
Operation::MulConst(_, _, c) => c.is_one(),
_ => false,
}
}
fn identity(w_out: usize, w_in: usize) -> Self {
Self::AddConst(w_out, w_in, 0u64)
}
}
impl Identity<bool> for Operation<bool> {
fn is_identity(&self) -> bool {
match *self {
Operation::AddConst(_, _, c) => !c,
Operation::SubConst(_, _, c) => !c,
Operation::MulConst(_, _, c) => c,
_ => false,
}
}
fn identity(w_out: usize, w_in: usize) -> Self {
Self::AddConst(w_out, w_in, false)
}
}
impl Identity<u64> for CombineOperation {
fn is_identity(&self) -> bool {
match self {
Self::Z64(g) => g.is_identity(),
_ => false,
}
}
fn identity(w_out: usize, w_in: usize) -> Self {
Self::Z64(Operation::identity(w_out, w_in))
}
}
impl Identity<bool> for CombineOperation {
fn is_identity(&self) -> bool {
match self {
Self::GF2(g) => g.is_identity(),
_ => false,
}
}
fn identity(w_out: usize, w_in: usize) -> Self {
Self::GF2(Operation::identity(w_out, w_in))
}
}