use std::io;
use crate::{
AtomicConfig, Config, Context, ContextUnit, Element, Polynomial, Preamble, Scalar, Source,
};
#[derive(Debug, Default, Clone, PartialEq, Eq, Hash)]
pub struct Constraint {
id: usize,
polynomial: Polynomial,
source: Source,
}
impl Constraint {
pub const fn new(id: usize, polynomial: Polynomial, source: Source) -> Self {
Self {
id,
polynomial,
source,
}
}
pub const fn id(&self) -> usize {
self.id
}
pub const fn polynomial(&self) -> &Polynomial {
&self.polynomial
}
pub const fn source(&self) -> &Source {
&self.source
}
pub const fn is_ok(&self) -> bool {
self.polynomial.is_ok()
}
}
impl Element for Constraint {
type Config = Config;
fn zeroed() -> Self {
Self::default()
}
fn len(config: &Self::Config) -> usize {
usize::len(&AtomicConfig)
+ Scalar::len(config)
+ Polynomial::len(config)
+ Source::len(config)
}
fn to_buffer(&self, config: &Self::Config, context: &mut ContextUnit, buf: &mut [u8]) {
let buf = self.id.encode(&AtomicConfig, context, buf);
let buf = self.polynomial.encode(config, context, buf);
let _ = self.source.encode(config, context, buf);
}
fn try_from_buffer_in_place<S>(
&mut self,
config: &Self::Config,
context: &mut Context<S>,
buf: &[u8],
) -> io::Result<()>
where
S: io::Read + io::Seek,
{
Self::validate_buffer_len(config, buf.len())?;
let buf = self.id.try_decode_in_place(&AtomicConfig, context, buf)?;
let buf = self.polynomial.try_decode_in_place(config, context, buf)?;
let _ = self.source.try_decode_in_place(config, context, buf)?;
Ok(())
}
fn validate(&self, preamble: &Preamble) -> io::Result<()> {
self.id.validate(preamble)?;
self.polynomial.validate(preamble)?;
self.source.validate(preamble)?;
Ok(())
}
}
#[test]
fn validate_works() {
Constraint::zeroed()
.validate(&Default::default())
.expect("default config validate should pass");
}