use crate::{traits::FiatShamirRng, Vec};
use snarkvm_fields::PrimeField;
use snarkvm_gadgets::{
bits::Boolean,
fields::FpGadget,
integers::uint::UInt8,
nonnative::{params::OptimizationType, NonNativeFieldVar},
};
use snarkvm_r1cs::{ConstraintSystem, SynthesisError};
pub trait FiatShamirRngVar<TargetField: PrimeField, BaseField: PrimeField, PFS: FiatShamirRng<TargetField, BaseField>>:
Clone
{
fn new<CS: ConstraintSystem<BaseField>>(cs: CS) -> Self;
fn constant<CS: ConstraintSystem<BaseField>>(cs: CS, pfs: &PFS) -> Self;
fn absorb_nonnative_field_elements<CS: ConstraintSystem<BaseField>>(
&mut self,
cs: CS,
elems: &[NonNativeFieldVar<TargetField, BaseField>],
ty: OptimizationType,
) -> Result<(), SynthesisError>;
fn absorb_native_field_elements<CS: ConstraintSystem<BaseField>>(
&mut self,
cs: CS,
elems: &[FpGadget<BaseField>],
) -> Result<(), SynthesisError>;
fn absorb_bytes<CS: ConstraintSystem<BaseField>>(&mut self, cs: CS, elems: &[UInt8]) -> Result<(), SynthesisError>;
fn squeeze_native_field_elements<CS: ConstraintSystem<BaseField>>(
&mut self,
cs: CS,
num: usize,
) -> Result<Vec<FpGadget<BaseField>>, SynthesisError>;
fn squeeze_field_elements<CS: ConstraintSystem<BaseField>>(
&mut self,
cs: CS,
num: usize,
) -> Result<Vec<NonNativeFieldVar<TargetField, BaseField>>, SynthesisError>;
#[allow(clippy::type_complexity)]
fn squeeze_field_elements_and_bits<CS: ConstraintSystem<BaseField>>(
&mut self,
cs: CS,
num: usize,
) -> Result<(Vec<NonNativeFieldVar<TargetField, BaseField>>, Vec<Vec<Boolean>>), SynthesisError>;
fn squeeze_128_bits_field_elements<CS: ConstraintSystem<BaseField>>(
&mut self,
cs: CS,
num: usize,
) -> Result<Vec<NonNativeFieldVar<TargetField, BaseField>>, SynthesisError>;
#[allow(clippy::type_complexity)]
fn squeeze_128_bits_field_elements_and_bits<CS: ConstraintSystem<BaseField>>(
&mut self,
cs: CS,
num: usize,
) -> Result<(Vec<NonNativeFieldVar<TargetField, BaseField>>, Vec<Vec<Boolean>>), SynthesisError>;
}