arnaucube
arnaucube/poulpyPublic Self-hosted repository
Added bytes_of to structs that can be created through the trait extension of Scratch
Jean-Philippe Bossuat jeanphilippe.bossuat@gmail.com
Wed, 28 May 2025 15:19:52 +0200
commit

f58d06ddf508b246ba1dcd0d9adffde14024d8a3

parent

187756a495db09b3fdc6724bcd8aae8e7db9be4c

8 files changed, 41 insertions(+), 0 deletions(-)

jump to
M core/src/automorphism.rs → core/src/automorphism.rs 
@@ -26,6 +26,10 @@ key: GLWESwitchingKey::alloc(module, basek, k, rows, rank, rank),
             p: 0,
         }
     }
+
+    pub fn bytes_of(module: &Module<FFT64>, basek: usize, k: usize, rows: usize, rank: usize) -> usize {
+        GLWESwitchingKey::<Vec<u8>, FFT64>::bytes_of(module, basek, k, rows, rank, rank)
+    }
 }
 
 impl<T, B: Backend> Infos for AutomorphismKey<T, B> {
M core/src/gglwe_ciphertext.rs → core/src/gglwe_ciphertext.rs 
@@ -27,6 +27,10 @@ basek: basek,
             k,
         }
     }
+
+    pub fn bytes_of(module: &Module<FFT64>, basek: usize, k: usize, rows: usize, rank_in: usize, rank_out: usize) -> usize {
+        module.bytes_of_mat_znx_dft(rows, rank_in, rank_out + 1, derive_size(basek, k))
+    }
 }
 
 impl<T, B: Backend> Infos for GGLWECiphertext<T, B> {
M core/src/ggsw_ciphertext.rs → core/src/ggsw_ciphertext.rs 
@@ -31,6 +31,10 @@ basek: basek,
             k: k,
         }
     }
+
+    pub fn bytes_of(module: &Module<FFT64>, basek: usize, k: usize, rows: usize, rank: usize) -> usize {
+        module.bytes_of_mat_znx_dft(rows, rank + 1, rank + 1, derive_size(basek, k))
+    }
 }
 
 impl<T, B: Backend> Infos for GGSWCiphertext<T, B> {
M core/src/glwe_ciphertext.rs → core/src/glwe_ciphertext.rs 
@@ -32,6 +32,10 @@ basek,
             k,
         }
     }
+
+    pub fn bytes_of(module: &Module<FFT64>, basek: usize, k: usize, rank: usize) -> usize {
+        module.bytes_of_vec_znx(rank + 1, derive_size(basek, k))
+    }
 }
 
 impl<T> Infos for GLWECiphertext<T> {
M core/src/glwe_ciphertext_fourier.rs → core/src/glwe_ciphertext_fourier.rs 
@@ -23,6 +23,10 @@ basek: basek,
             k: k,
         }
     }
+
+    pub fn bytes_of(module: &Module<B>, basek: usize, k: usize, rank: usize) -> usize {
+        module.bytes_of_vec_znx_dft(rank + 1, derive_size(basek, k))
+    }
 }
 
 impl<T, B: Backend> Infos for GLWECiphertextFourier<T, B> {
M core/src/keys.rs → core/src/keys.rs 
@@ -26,6 +26,10 @@ data: module.new_scalar_znx(rank),
             dist: SecretDistribution::NONE,
         }
     }
+
+    pub fn bytes_of(module: &Module<FFT64>, rank: usize) -> usize {
+        module.bytes_of_scalar_znx(rank + 1)
+    }
 }
 
 impl<DataSelf> SecretKey<DataSelf> {

@@ -89,6 +93,10 @@ data: module.new_scalar_znx_dft(rank),
             dist: SecretDistribution::NONE,
         }
     }
+
+    pub fn bytes_of(module: &Module<B>, rank: usize) -> usize {
+        module.bytes_of_scalar_znx_dft(rank + 1)
+    }
 }
 
 impl<D: AsRef<[u8]> + AsMut<[u8]>> SecretKeyFourier<D, FFT64> {

@@ -123,6 +131,10 @@ Self {
             data: GLWECiphertextFourier::alloc(module, basek, k, rank),
             dist: SecretDistribution::NONE,
         }
+    }
+
+    pub fn bytes_of(module: &Module<B>, basek: usize, k: usize, rank: usize) -> usize {
+        GLWECiphertextFourier::<Vec<u8>, B>::bytes_of(module, basek, k, rank)
     }
 }
M core/src/keyswitch_key.rs → core/src/keyswitch_key.rs 
@@ -17,6 +17,10 @@ GLWESwitchingKey(GGLWECiphertext::alloc(
             module, basek, k, rows, rank_in, rank_out,
         ))
     }
+
+    pub fn bytes_of(module: &Module<FFT64>, basek: usize, k: usize, rows: usize, rank_in: usize, rank_out: usize) -> usize {
+        GGLWECiphertext::<Vec<u8>, FFT64>::bytes_of(module, basek, k, rows, rank_in, rank_out)
+    }
 }
 
 impl<T, B: Backend> Infos for GLWESwitchingKey<T, B> {
M core/src/tensor_key.rs → core/src/tensor_key.rs 
@@ -20,6 +20,11 @@ keys.push(GLWESwitchingKey::alloc(module, basek, k, rows, 1, rank));
         });
         Self { keys: keys }
     }
+
+    pub fn bytes_of(module: &Module<FFT64>, basek: usize, k: usize, rows: usize, rank: usize) -> usize {
+        let pairs: usize = (((rank + 1) * rank) >> 1).max(1);
+        pairs * GLWESwitchingKey::<Vec<u8>, FFT64>::bytes_of(module, basek, k, rows, 1, rank)
+    }
 }
 
 impl<T, B: Backend> Infos for TensorKey<T, B> {