wip

poulpy-core/src/layouts/glwe_sk.rs

@@ -1,5 +1,5 @@
 use poulpy_hal::{
-    layouts::{Data, DataMut, DataRef, ReaderFrom, ScalarZnx, WriterTo, ZnxInfos, ZnxZero},
+    layouts::{Data, DataMut, DataRef, ReaderFrom, ScalarZnx, ScalarZnxToMut, ScalarZnxToRef, WriterTo, ZnxInfos, ZnxZero},
     source::Source,
 };
 
@@ -136,6 +136,32 @@ impl<D: DataMut> GLWESecret<D> {
     }
 }
 
+pub trait GLWESecretToMut {
+    fn to_mut(&mut self) -> GLWESecret<&mut [u8]>;
+}
+
+impl<D: DataMut> GLWESecretToMut for GLWESecret<D> {
+    fn to_mut(&mut self) -> GLWESecret<&mut [u8]> {
+        GLWESecret {
+            dist: self.dist,
+            data: self.data.to_mut(),
+        }
+    }
+}
+
+pub trait GLWESecretToRef {
+    fn to_ref(&self) -> GLWESecret<&[u8]>;
+}
+
+impl<D: DataRef> GLWESecretToRef for GLWESecret<D> {
+    fn to_ref(&self) -> GLWESecret<&[u8]> {
+        GLWESecret {
+            data: self.data.to_ref(),
+            dist: self.dist,
+        }
+    }
+}
+
 impl<D: DataMut> ReaderFrom for GLWESecret<D> {
     fn read_from<R: std::io::Read>(&mut self, reader: &mut R) -> std::io::Result<()> {
         match Distribution::read_from(reader) {