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