1use base2k::{Encoding, FFT64, SvpPPolOps};
 2use rlwe::{
 3    ciphertext::Ciphertext,
 4    decryptor::{Decryptor, decrypt_rlwe_thread_safe_tmp_byte},
 5    encryptor::{EncryptorSk, encrypt_rlwe_sk_tmp_bytes},
 6    keys::SecretKey,
 7    parameters::{Parameters, ParametersLiteral},
 8    plaintext::Plaintext,
 9};
10use sampling::source::{Source, new_seed};
11
12fn main() {
13    let params_lit: ParametersLiteral = ParametersLiteral {
14        log_n: 10,
15        log_q: 54,
16        log_p: 0,
17        log_base2k: 17,
18        log_scale: 20,
19        xe: 3.2,
20        xs: 128,
21    };
22
23    let params: Parameters = Parameters::new::<FFT64>(&params_lit);
24
25    let mut tmp_bytes: Vec<u8> = vec![
26        0u8;
27        params.decrypt_rlwe_thread_safe_tmp_byte(params.log_q())
28            | params.encrypt_rlwe_sk_tmp_bytes(params.log_q())
29    ];
30
31    let mut source: Source = Source::new([0; 32]);
32
33    let mut sk0: SecretKey = SecretKey::new(params.module());
34    let mut sk1: SecretKey = SecretKey::new(params.module());
35
36    sk0.fill_ternary_hw(params.xs(), &mut source);
37    sk1.fill_ternary_hw(params.xs(), &mut source);
38
39    let mut want = vec![i64::default(); params.n()];
40
41    want.iter_mut().enumerate().for_each(|(i, x)| *x = i as i64);
42
43    let mut pt: Plaintext = params.new_plaintext(params.log_q());
44
45    let log_base2k = pt.log_base2k();
46
47    let log_k: usize = params.log_q() - 20;
48
49    pt.0.value[0].encode_vec_i64(log_base2k, log_k, &want, 32);
50    pt.0.value[0].normalize(log_base2k, &mut tmp_bytes);
51
52    println!("log_k: {}", log_k);
53    pt.0.value[0].print_limbs(pt.limbs(), 16);
54
55    let mut ct: Ciphertext = params.new_ciphertext(params.log_q());
56
57    let mut source_xe: Source = Source::new(new_seed());
58    let mut source_xa: Source = Source::new(new_seed());
59
60    let mut sk_svp_ppol: base2k::SvpPPol = params.module().svp_new_ppol();
61    params.module().svp_prepare(&mut sk_svp_ppol, &sk0.0);
62
63    params.encrypt_rlwe_sk_thread_safe(
64        &mut ct,
65        Some(&pt),
66        &sk_svp_ppol,
67        &mut source_xa,
68        &mut source_xe,
69        &mut tmp_bytes,
70    );
71
72    params.decrypt_rlwe_thread_safe(&mut pt, &ct, &sk_svp_ppol, &mut tmp_bytes);
73
74    pt.0.value[0].print_limbs(pt.limbs(), 16);
75
76    let mut have = vec![i64::default(); params.n()];
77
78    println!("pt: {}", log_k);
79    pt.0.value[0].decode_vec_i64(pt.log_base2k(), log_k, &mut have);
80
81    println!("want: {:?}", &want[..16]);
82    println!("have: {:?}", &have[..16]);
83}