use snarkvm_circuit_network::Aleo;
use snarkvm_circuit_types::{environment::prelude::*, Boolean, Field, U8};
#[derive(Clone)]
pub struct HeaderLeaf<A: Aleo> {
index: U8<A>,
id: Field<A>,
}
impl<A: Aleo> HeaderLeaf<A> {
pub fn index(&self) -> &U8<A> {
&self.index
}
pub const fn id(&self) -> &Field<A> {
&self.id
}
}
impl<A: Aleo> Inject for HeaderLeaf<A> {
type Primitive = console::HeaderLeaf<A::Network>;
fn new(mode: Mode, leaf: Self::Primitive) -> Self {
Self { index: U8::new(mode, console::U8::new(leaf.index())), id: Field::new(mode, leaf.id()) }
}
}
impl<A: Aleo> Eject for HeaderLeaf<A> {
type Primitive = console::HeaderLeaf<A::Network>;
fn eject_mode(&self) -> Mode {
(&self.index, &self.id).eject_mode()
}
fn eject_value(&self) -> Self::Primitive {
Self::Primitive::new(*self.index.eject_value(), self.id.eject_value())
}
}
impl<A: Aleo> ToBits for HeaderLeaf<A> {
type Boolean = Boolean<A>;
fn to_bits_le(&self) -> Vec<Self::Boolean> {
let mut bits_le = self.index.to_bits_le();
bits_le.extend(self.id.to_bits_le());
bits_le
}
fn to_bits_be(&self) -> Vec<Self::Boolean> {
let mut bits_be = self.index.to_bits_be();
bits_be.extend(self.id.to_bits_be());
bits_be
}
}