use std::marker::PhantomData;
use ff::PrimeField;
use bellpepper_core::{ConstraintSystem, Index, LinearCombination, SynthesisError, Variable};
#[allow(clippy::upper_case_acronyms)]
#[derive(Debug)]
pub struct BenchCS<Scalar: PrimeField> {
inputs: usize,
aux: usize,
a: usize,
b: usize,
c: usize,
_e: PhantomData<Scalar>,
}
impl<Scalar: PrimeField> BenchCS<Scalar> {
pub fn new() -> Self {
BenchCS::default()
}
pub fn num_constraints(&self) -> usize {
self.a
}
pub fn num_inputs(&self) -> usize {
self.inputs
}
}
impl<Scalar: PrimeField> Default for BenchCS<Scalar> {
fn default() -> Self {
BenchCS {
inputs: 1,
aux: 0,
a: 0,
b: 0,
c: 0,
_e: PhantomData,
}
}
}
impl<Scalar: PrimeField> ConstraintSystem<Scalar> for BenchCS<Scalar> {
type Root = Self;
fn new() -> Self {
BenchCS::default()
}
fn alloc<F, A, AR>(&mut self, _: A, _f: F) -> Result<Variable, SynthesisError>
where
F: FnOnce() -> Result<Scalar, SynthesisError>,
A: FnOnce() -> AR,
AR: Into<String>,
{
self.aux += 1;
Ok(Variable::new_unchecked(Index::Aux(self.aux - 1)))
}
fn alloc_input<F, A, AR>(&mut self, _: A, _f: F) -> Result<Variable, SynthesisError>
where
F: FnOnce() -> Result<Scalar, SynthesisError>,
A: FnOnce() -> AR,
AR: Into<String>,
{
self.inputs += 1;
Ok(Variable::new_unchecked(Index::Input(self.inputs - 1)))
}
fn enforce<A, AR, LA, LB, LC>(&mut self, _: A, _a: LA, _b: LB, _c: LC)
where
A: FnOnce() -> AR,
AR: Into<String>,
LA: FnOnce(LinearCombination<Scalar>) -> LinearCombination<Scalar>,
LB: FnOnce(LinearCombination<Scalar>) -> LinearCombination<Scalar>,
LC: FnOnce(LinearCombination<Scalar>) -> LinearCombination<Scalar>,
{
self.a += 1;
self.b += 1;
self.c += 1;
}
fn push_namespace<NR, N>(&mut self, _: N)
where
NR: Into<String>,
N: FnOnce() -> NR,
{
}
fn pop_namespace(&mut self) {}
fn get_root(&mut self) -> &mut Self::Root {
self
}
}