update BDD ciphertext types + API for GLWEToLWE

poulpy-schemes/src/tfhe/bdd_arithmetic/tests/test_suite/mod.rs

@@ -17,6 +17,11 @@ pub use and::*;
 pub use ggsw_blind_rotations::*;
 pub use glwe_blind_rotation::*;
 pub use or::*;
+use poulpy_hal::{
+    api::{ModuleNew, ScratchOwnedAlloc, ScratchOwnedBorrow},
+    layouts::{Backend, Module, Scratch, ScratchOwned},
+    source::Source,
+};
 pub use prepare::*;
 pub use sll::*;
 pub use slt::*;
@@ -26,15 +31,96 @@ pub use srl::*;
 pub use sub::*;
 pub use xor::*;
 
-use poulpy_core::layouts::{
-    Base2K, Degree, Dnum, Dsize, GGSWLayout, GLWEAutomorphismKeyLayout, GLWELayout, GLWETensorKeyLayout, GLWEToLWEKeyLayout,
-    Rank, TorusPrecision,
+use poulpy_core::{
+    ScratchTakeCore,
+    layouts::{
+        Base2K, Degree, Dnum, Dsize, GGSWLayout, GLWEAutomorphismKeyLayout, GLWELayout, GLWESecret, GLWESecretPrepared,
+        GLWESecretPreparedFactory, GLWETensorKeyLayout, GLWEToLWEKeyLayout, LWESecret, Rank, TorusPrecision,
+    },
 };
 
 use crate::tfhe::{
-    bdd_arithmetic::BDDKeyLayout, blind_rotation::BlindRotationKeyLayout, circuit_bootstrapping::CircuitBootstrappingKeyLayout,
+    bdd_arithmetic::{BDDKey, BDDKeyEncryptSk, BDDKeyLayout, BDDKeyPrepared, BDDKeyPreparedFactory},
+    blind_rotation::{
+        BlindRotationAlgo, BlindRotationKey, BlindRotationKeyFactory, BlindRotationKeyLayout, BlindRotationKeyPreparedFactory,
+    },
+    circuit_bootstrapping::CircuitBootstrappingKeyLayout,
 };
 
+pub struct TestContext<BRA: BlindRotationAlgo, BE: Backend> {
+    pub module: Module<BE>,
+    pub sk_glwe: GLWESecretPrepared<Vec<u8>, BE>,
+    pub sk_lwe: LWESecret<Vec<u8>>,
+    pub bdd_key: BDDKeyPrepared<Vec<u8>, BRA, BE>,
+}
+
+impl<BRA: BlindRotationAlgo, BE: Backend> Default for TestContext<BRA, BE>
+where
+    Module<BE>: ModuleNew<BE>
+        + BDDKeyEncryptSk<BRA, BE>
+        + GLWESecretPreparedFactory<BE>
+        + BlindRotationKeyPreparedFactory<BRA, BE>
+        + BDDKeyPreparedFactory<BRA, BE>,
+    BlindRotationKey<Vec<u8>, BRA>: BlindRotationKeyFactory<BRA>,
+    ScratchOwned<BE>: ScratchOwnedAlloc<BE> + ScratchOwnedBorrow<BE>,
+    Scratch<BE>: ScratchTakeCore<BE>,
+{
+    fn default() -> Self {
+        Self::new()
+    }
+}
+
+impl<BRA: BlindRotationAlgo, BE: Backend> TestContext<BRA, BE> {
+    pub fn new() -> Self
+    where
+        Module<BE>: ModuleNew<BE>
+            + BDDKeyEncryptSk<BRA, BE>
+            + GLWESecretPreparedFactory<BE>
+            + BlindRotationKeyPreparedFactory<BRA, BE>
+            + BDDKeyPreparedFactory<BRA, BE>,
+        BlindRotationKey<Vec<u8>, BRA>: BlindRotationKeyFactory<BRA>,
+        ScratchOwned<BE>: ScratchOwnedAlloc<BE> + ScratchOwnedBorrow<BE>,
+        Scratch<BE>: ScratchTakeCore<BE>,
+    {
+        let module: Module<BE> = Module::<BE>::new(TEST_N_GLWE as u64);
+
+        let mut source_xs: Source = Source::new([1u8; 32]);
+        let mut source_xa: Source = Source::new([2u8; 32]);
+        let mut source_xe: Source = Source::new([3u8; 32]);
+
+        let mut scratch: ScratchOwned<BE> = ScratchOwned::alloc(1 << 22);
+
+        let mut sk_glwe: GLWESecret<Vec<u8>> = GLWESecret::alloc(TEST_N_GLWE.into(), TEST_RANK.into());
+        sk_glwe.fill_ternary_prob(0.5, &mut source_xs);
+        let mut sk_glwe_prep: GLWESecretPrepared<Vec<u8>, BE> = GLWESecretPrepared::alloc(&module, TEST_RANK.into());
+        sk_glwe_prep.prepare(&module, &sk_glwe);
+
+        let n_lwe: u32 = TEST_N_LWE;
+        let block_size: u32 = TEST_BLOCK_SIZE;
+        let mut sk_lwe: LWESecret<Vec<u8>> = LWESecret::alloc(n_lwe.into());
+        sk_lwe.fill_binary_block(block_size as usize, &mut source_xs);
+        let bdd_key_infos: BDDKeyLayout = TEST_BDD_KEY_LAYOUT;
+        let mut bdd_key: BDDKey<Vec<u8>, BRA> = BDDKey::alloc_from_infos(&bdd_key_infos);
+        bdd_key.encrypt_sk(
+            &module,
+            &sk_lwe,
+            &sk_glwe,
+            &mut source_xa,
+            &mut source_xe,
+            scratch.borrow(),
+        );
+        let mut bdd_key_prepared: BDDKeyPrepared<Vec<u8>, BRA, BE> = BDDKeyPrepared::alloc_from_infos(&module, &bdd_key_infos);
+        bdd_key_prepared.prepare(&module, &bdd_key, scratch.borrow());
+
+        TestContext {
+            bdd_key: bdd_key_prepared,
+            sk_glwe: sk_glwe_prep,
+            sk_lwe,
+            module,
+        }
+    }
+}
+
 pub(crate) const TEST_N_GLWE: u32 = 512;
 pub(crate) const TEST_N_LWE: u32 = 77;
 pub(crate) const TEST_BASE2K: u32 = 13;