Add ToConstraintField for bool (#35)

arkworks-rs · Nov 12, 2020 · 4a4851d · 4a4851d
1 parent d0e4077
commit 4a4851d
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 4 deletions.
diff --git a/ff/src/to_field_vec.rs b/ff/src/to_field_vec.rs
@@ -8,6 +8,16 @@ pub trait ToConstraintField<F: Field> {
     fn to_field_elements(&self) -> Option<Vec<F>>;
 }
 
+impl<F: Field> ToConstraintField<F> for bool {
+    fn to_field_elements(&self) -> Option<Vec<F>> {
+        if *self {
+            Some(vec![F::one()])
+        } else {
+            Some(vec![F::zero()])
+        }
+    }
+}
+
 impl<F: PrimeField> ToConstraintField<F> for F {
     fn to_field_elements(&self) -> Option<Vec<F>> {
         Some(vec![*self])

diff --git a/serialize/src/flags.rs b/serialize/src/flags.rs
@@ -57,10 +57,7 @@ impl SWFlags {
 
     #[inline]
     pub fn is_infinity(&self) -> bool {
-        match self {
-            SWFlags::Infinity => true,
-            _ => false,
-        }
+        matches!(self, SWFlags::Infinity)
     }
 
     #[inline]