wip rlwe + some bug fixes in base2k

rlwe/src/plaintext.rs

@@ -1,20 +1,35 @@
 use crate::ciphertext::Ciphertext;
 use crate::elem::Elem;
-use base2k::VecZnx;
+use crate::parameters::Parameters;
+use base2k::{Module, VecZnx};
 
 pub struct Plaintext(pub Elem);
 
-/*
 impl Parameters {
     pub fn new_plaintext(&self, log_q: usize) -> Plaintext {
-        Plaintext(self.new_elem(0, log_q))
+        Plaintext::new(self.module(), self.log_base2k(), log_q, self.log_scale())
+    }
+
+    pub fn bytes_of_plaintext(&self, log_q: usize) -> usize {
+        Elem::bytes_of(self.module(), self.log_base2k(), log_q, 0)
+    }
+
+    pub fn plaintext_from_bytes(&self, log_q: usize, bytes: &mut [u8]) -> Plaintext {
+        Plaintext(self.elem_from_bytes(log_q, 0, bytes))
     }
 }
-*/
 
 impl Plaintext {
-    pub fn new(n: usize, log_base2k: usize, log_q: usize) -> Self {
-        Self(Elem::new(n, log_base2k, log_q, 0))
+    pub fn new(module: &Module, log_base2k: usize, log_q: usize, log_scale: usize) -> Self {
+        Self(Elem::new(module, log_base2k, log_q, 0, log_scale))
+    }
+
+    pub fn bytes_of(module: &Module, log_base2k: usize, log_q: usize) -> usize {
+        Elem::bytes_of(module, log_base2k, log_q, 0)
+    }
+
+    pub fn from_bytes(module: &Module, log_base2k: usize, log_q: usize, bytes: &mut [u8]) -> Self {
+        Self(Elem::from_bytes(module, log_base2k, log_q, 0, bytes))
     }
 
     pub fn n(&self) -> usize {
@@ -45,6 +60,10 @@ impl Plaintext {
         self.0.log_base2k()
     }
 
+    pub fn log_scale(&self) -> usize {
+        self.0.log_scale()
+    }
+
     pub fn as_ciphertext(&self) -> Ciphertext {
         unsafe { Ciphertext(std::ptr::read(&self.0)) }
     }