arnaucube/ark-r1cs-std
1
0
Fork 0
mirror of https://github.com/arnaucube/ark-r1cs-std.git synced 2026-01-09 07:21:29 +01:00
Code Issues Projects Releases Wiki Activity

Apply suggestions from code review

Browse Source 
Co-authored-by: Pratyush Mishra <pratyushmishra@berkeley.edu>
This commit is contained in:
Nirvan Tyagi
2020-11-20 12:54:13 -05:00
committed by Pratyush Mishra
parent 0abb3a7dfe
commit 957ac48da4
2 changed files with 24 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

26
src/bits/uint.rs
View File

@@ -330,23 +330,25 @@ macro_rules! make_uint {
}
impl<ConstraintF: Field> CondSelectGadget<ConstraintF> for $name<ConstraintF> {
#[tracing::instrument(target = "r1cs")]
#[tracing::instrument(target = "r1cs", skip(cond, true_value, false_value))]
fn conditionally_select(
cond: &Boolean<ConstraintF>,
true_value: &Self,
false_value: &Self,
) -> Result<Self, SynthesisError> {
let selected_bits = true_value.bits.iter().zip(&false_value.bits)
.map(|(true_bit, false_bit)| {
cond.select(true_bit, false_bit)
}).collect::<Result<Vec<Boolean<ConstraintF>>, SynthesisError>>()?;
let selected_value = match (cond.value(), true_value.value(), false_value.value()) {
(Ok(true), Err(_), _) => None,
(Ok(true), Ok(v), _) => Some(v),
(Ok(false), _, Err(_)) => None,
(Ok(false), _, Ok(v)) => Some(v),
(Err(_), _, _) => None,
};
let selected_bits = true_value
.bits
.iter()
.zip(&false_value.bits)
.map(|(t, f)| cond.select(t, f))
.collect::<Result<Vec<_>, SynthesisError>>()?;
let selected_value = cond.value().ok().and_then(|cond| {
if cond {
true_value.value().ok()
} else {
false_value.value().ok()
}
});
Ok(Self {
bits: selected_bits,
value: selected_value,

20
src/bits/uint8.rs
View File

@@ -285,7 +285,7 @@ impl<ConstraintF: Field> EqGadget<ConstraintF> for UInt8<ConstraintF> {
}
impl<ConstraintF: Field> CondSelectGadget<ConstraintF> for UInt8<ConstraintF> {
#[tracing::instrument(target = "r1cs")]
#[tracing::instrument(target = "r1cs", skip(cond, true_value, false_value))]
fn conditionally_select(
cond: &Boolean<ConstraintF>,
true_value: &Self,
@@ -295,15 +295,15 @@ impl<ConstraintF: Field> CondSelectGadget<ConstraintF> for UInt8<ConstraintF> {
.bits
.iter()
.zip(&false_value.bits)
.map(|(true_bit, false_bit)| cond.select(true_bit, false_bit))
.collect::<Result<Vec<Boolean<ConstraintF>>, SynthesisError>>()?;
let selected_value = match (cond.value(), true_value.value(), false_value.value()) {
(Ok(true), Err(_), _) => None,
(Ok(true), Ok(v), _) => Some(v),
(Ok(false), _, Err(_)) => None,
(Ok(false), _, Ok(v)) => Some(v),
(Err(_), _, _) => None,
};
.map(|(t, f)| cond.select(t, f))
.collect::<Result<Vec<_>, SynthesisError>>()?;
let selected_value = cond.value().ok().and_then(|cond| {
if cond {
true_value.value().ok()
} else {
false_value.value().ok()
}
});
Ok(Self {
bits: selected_bits,
value: selected_value,