updated sampling traits

rlwe/src/keys.rs

@@ -1,82 +0,0 @@
-use crate::ciphertext::{Ciphertext, new_gadget_ciphertext};
-use crate::elem::{Elem, ElemCommon};
-use crate::encryptor::{encrypt_rlwe_sk, encrypt_rlwe_sk_tmp_bytes};
-use base2k::{Module, Scalar, ScalarZnxDft, ScalarZnxDftOps, VecZnx, MatZnxDft};
-use sampling::source::Source;
-
-pub struct SecretKey(pub Scalar);
-
-impl SecretKey {
-    pub fn new(module: &Module) -> Self {
-        SecretKey(Scalar::new(module.n()))
-    }
-
-    pub fn fill_ternary_prob(&mut self, prob: f64, source: &mut Source) {
-        self.0.fill_ternary_prob(prob, source);
-    }
-
-    pub fn fill_ternary_hw(&mut self, hw: usize, source: &mut Source) {
-        self.0.fill_ternary_hw(hw, source);
-    }
-
-    pub fn prepare(&self, module: &Module, sk_ppol: &mut ScalarZnxDft) {
-        module.scalar_znx_dft_prepare(sk_ppol, &self.0)
-    }
-}
-
-pub struct PublicKey(pub Elem<VecZnx>);
-
-impl PublicKey {
-    pub fn new(module: &Module, log_base2k: usize, log_q: usize) -> PublicKey {
-        PublicKey(Elem::<VecZnx>::new(module, log_base2k, log_q, 2))
-    }
-
-    pub fn gen_thread_safe(
-        &mut self,
-        module: &Module,
-        sk: &ScalarZnxDft,
-        xe: f64,
-        xa_source: &mut Source,
-        xe_source: &mut Source,
-        tmp_bytes: &mut [u8],
-    ) {
-        encrypt_rlwe_sk(
-            module,
-            &mut self.0,
-            None,
-            sk,
-            xa_source,
-            xe_source,
-            xe,
-            tmp_bytes,
-        );
-    }
-
-    pub fn gen_thread_safe_tmp_bytes(module: &Module, log_base2k: usize, log_q: usize) -> usize {
-        encrypt_rlwe_sk_tmp_bytes(module, log_base2k, log_q)
-    }
-}
-
-pub struct SwitchingKey(pub Ciphertext<MatZnxDft>);
-
-impl SwitchingKey {
-    pub fn new(module: &Module, log_base2k: usize, rows: usize, log_q: usize) -> SwitchingKey {
-        SwitchingKey(new_gadget_ciphertext(module, log_base2k, rows, log_q))
-    }
-
-    pub fn n(&self) -> usize {
-        self.0.n()
-    }
-
-    pub fn rows(&self) -> usize {
-        self.0.rows()
-    }
-
-    pub fn cols(&self) -> usize {
-        self.0.cols()
-    }
-
-    pub fn log_base2k(&self) -> usize {
-        self.0.log_base2k()
-    }
-}