use ark_std::rand::RngCore;
use core::num::NonZeroU32;
pub struct OptionalRng<R>(pub Option<R>);
impl<R: RngCore> RngCore for OptionalRng<R> {
#[inline]
fn next_u32(&mut self) -> u32 {
(&mut self.0)
.as_mut()
.map(|r| r.next_u32())
.expect("Rng was invoked in a non-hiding context")
}
#[inline]
fn next_u64(&mut self) -> u64 {
(&mut self.0)
.as_mut()
.map(|r| r.next_u64())
.expect("Rng was invoked in a non-hiding context")
}
#[inline]
fn fill_bytes(&mut self, dest: &mut [u8]) {
(&mut self.0)
.as_mut()
.map(|r| r.fill_bytes(dest))
.expect("Rng was invoked in a non-hiding context")
}
#[inline]
fn try_fill_bytes(&mut self, dest: &mut [u8]) -> Result<(), ark_std::rand::Error> {
match &mut self.0 {
Some(r) => r.try_fill_bytes(dest),
None => Err(NonZeroU32::new(ark_std::rand::Error::CUSTOM_START)
.unwrap()
.into()),
}
}
}
impl<R: RngCore> From<R> for OptionalRng<R> {
fn from(other: R) -> Self {
Self(Some(other))
}
}