wip adding automorphism on AutomorphismKey

base2k/src/scalar_znx.rs

@@ -130,7 +130,7 @@ pub trait ScalarZnxOps {
         A: ScalarZnxToRef;
 
     /// Applies the automorphism X^i -> X^ik on the selected column of `a`.
-    fn vec_znx_automorphism_inplace<A>(&self, k: i64, a: &mut A, a_col: usize)
+    fn scalar_znx_automorphism_inplace<A>(&self, k: i64, a: &mut A, a_col: usize)
     where
         A: ScalarZnxToMut;
 }
@@ -162,7 +162,7 @@ impl<B: Backend> ScalarZnxOps for Module<B> {
         }
     }
 
-    fn vec_znx_automorphism_inplace<A>(&self, k: i64, a: &mut A, a_col: usize)
+    fn scalar_znx_automorphism_inplace<A>(&self, k: i64, a: &mut A, a_col: usize)
     where
         A: ScalarZnxToMut,
     {

base2k/src/vec_znx_big.rs

@@ -1,6 +1,6 @@
 use crate::ffi::vec_znx_big;
 use crate::znx_base::{ZnxInfos, ZnxView};
-use crate::{Backend, DataView, DataViewMut, FFT64, Module, ZnxSliceSize, ZnxViewMut, ZnxZero, alloc_aligned};
+use crate::{alloc_aligned, Backend, DataView, DataViewMut, Module, VecZnx, ZnxSliceSize, ZnxViewMut, ZnxZero, FFT64};
 use std::fmt;
 use std::marker::PhantomData;
 
@@ -97,7 +97,18 @@ impl<D, B: Backend> VecZnxBig<D, B> {
 impl<D> VecZnxBig<D, FFT64>
 where
     VecZnxBig<D, FFT64>: VecZnxBigToMut<FFT64> + ZnxInfos,
-{
+{   
+    // Consumes the VecZnxBig to return a VecZnx. 
+    // Useful when no normalization is needed.
+    pub fn to_vec_znx_small(self) -> VecZnx<D>{
+        VecZnx{
+            data: self.data,
+            n: self.n,
+            cols: self.cols,
+            size: self.size,
+        }
+    }
+
     /// Extracts the a_col-th column of 'a' and stores it on the self_col-th column [Self].
     pub fn extract_column<C>(&mut self, self_col: usize, a: &VecZnxBig<C, FFT64>, a_col: usize)
     where

base2k/src/vec_znx_big_ops.rs

@@ -147,6 +147,7 @@ pub trait VecZnxBigOps<BACKEND: Backend> {
     fn vec_znx_big_automorphism_inplace<A>(&self, k: i64, a: &mut A, a_col: usize)
     where
         A: VecZnxBigToMut<BACKEND>;
+
 }
 
 pub trait VecZnxBigScratch {
@@ -169,6 +170,7 @@ impl<B: Backend> VecZnxBigAlloc<B> for Module<B> {
 }
 
 impl VecZnxBigOps<FFT64> for Module<FFT64> {
+
     fn vec_znx_big_add<R, A, B>(&self, res: &mut R, res_col: usize, a: &A, a_col: usize, b: &B, b_col: usize)
     where
         R: VecZnxBigToMut<FFT64>,