Add serialize for rational.

dpsa4fl · Aug 29, 2023 · fcdc568 · fcdc568
1 parent bd41119
commit fcdc568
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 2 deletions.
diff --git a/Cargo.lock b/Cargo.lock
diff --git a/Cargo.toml b/Cargo.toml
@@ -24,7 +24,7 @@ hmac = { version = "0.12.1", optional = true }
 num-bigint = { version = "0.4.3", optional = true, features = ["rand", "serde"] }
 num-integer = { version = "0.1.45", optional = true }
 num-iter = { version = "0.1.43", optional = true }
-num-rational = { version = "0.4.1", optional = true }
+num-rational = { version = "0.4.1", optional = true, features = ["serde"] }
 num-traits = { version = "0.2.15", optional = true }
 rand = { version = "0.8", optional = true }
 rand_core = "0.6.4"

diff --git a/src/dp.rs b/src/dp.rs
@@ -16,6 +16,7 @@
 //!
 use num_bigint::{BigInt, BigUint, TryFromBigIntError};
 use num_rational::{BigRational, Ratio};
+use serde::{Serialize, Deserialize};
 
 /// Errors propagated by methods in this module.
 #[derive(Debug, thiserror::Error)]
@@ -38,7 +39,7 @@ pub enum DpError {
 
 /// Positive arbitrary precision rational number to represent DP and noise distribution parameters in
 /// protocol messages and manipulate them without rounding errors.
-#[derive(Clone, Debug)]
+#[derive(Clone, Debug, Serialize, Deserialize, PartialEq, Eq)]
 pub struct Rational(Ratio<BigUint>);
 
 impl Rational {

diff --git a/src/flp/types.rs b/src/flp/types.rs
@@ -2,6 +2,8 @@
 
 //! A collection of [`Type`] implementations.
 
+
+#[cfg(feature = "experimental")]
 use crate::dp::distributions::ZCdpDiscreteGaussian;
 use crate::field::{FftFriendlyFieldElement, FieldElementWithIntegerExt};
 use crate::flp::gadgets::{BlindPolyEval, Mul, ParallelSumGadget, PolyEval};
@@ -11,6 +13,7 @@ use std::convert::TryInto;
 use std::fmt::{self, Debug};
 use std::marker::PhantomData;
 
+#[cfg(feature = "experimental")]
 use super::TypeWithNoise;
 /// The counter data type. Each measurement is `0` or `1` and the aggregate result is the sum of the measurements (i.e., the total number of `1s`).
 #[derive(Clone, PartialEq, Eq)]
@@ -39,6 +42,7 @@ impl<F: FftFriendlyFieldElement> Default for Count<F> {
     }
 }
 
+#[cfg(feature = "experimental")]
 impl<F: FftFriendlyFieldElement> TypeWithNoise<ZCdpDiscreteGaussian> for Count<F> {}
 
 impl<F: FftFriendlyFieldElement> Type for Count<F> {
@@ -144,6 +148,7 @@ impl<F: FftFriendlyFieldElement> Sum<F> {
     }
 }
 
+#[cfg(feature = "experimental")]
 impl<F: FftFriendlyFieldElement> TypeWithNoise<ZCdpDiscreteGaussian> for Sum<F> {}
 
 impl<F: FftFriendlyFieldElement> Type for Sum<F> {
@@ -245,6 +250,7 @@ impl<F: FftFriendlyFieldElement> Average<F> {
     }
 }
 
+#[cfg(feature = "experimental")]
 impl<F: FftFriendlyFieldElement> TypeWithNoise<ZCdpDiscreteGaussian> for Average<F> {}
 
 impl<F: FftFriendlyFieldElement> Type for Average<F> {
@@ -353,6 +359,7 @@ impl<F: FftFriendlyFieldElement> Histogram<F> {
     }
 }
 
+#[cfg(feature = "experimental")]
 impl<F: FftFriendlyFieldElement> TypeWithNoise<ZCdpDiscreteGaussian> for Histogram<F> {}
 
 impl<F: FftFriendlyFieldElement> Type for Histogram<F> {
@@ -531,6 +538,7 @@ impl<F: FftFriendlyFieldElement, S> Clone for SumVec<F, S> {
 }
 
 
+#[cfg(feature = "experimental")]
 impl<F, S> TypeWithNoise<ZCdpDiscreteGaussian> for SumVec<F, S>
 where
     F: FftFriendlyFieldElement,