1use base2k::{
  2    Encoding, Free, Infos, Matrix3D, Module, VecZnx, VecZnxBig, VecZnxDft, VecZnxOps, VmpPMat,
  3    VmpPMatOps, FFT64,
  4};
  5use std::cmp::min;
  6
  7fn main() {
  8    let log_n = 5;
  9    let n = 1 << log_n;
 10
 11    let module: Module = Module::new::<FFT64>(n);
 12    let log_base2k: usize = 15;
 13    let limbs: usize = 5;
 14    let log_k: usize = log_base2k * limbs - 5;
 15
 16    let rows: usize = limbs;
 17    let cols: usize = limbs + 1;
 18
 19    // Maximum size of the byte scratch needed
 20    let tmp_bytes: usize = module.vmp_prepare_tmp_bytes(rows, cols)
 21        | module.vmp_apply_dft_tmp_bytes(limbs, limbs, rows, cols);
 22
 23    let mut buf: Vec<u8> = vec![0; tmp_bytes];
 24
 25    let mut a_values: Vec<i64> = vec![i64::default(); n];
 26    a_values[1] = (1 << log_base2k) + 1;
 27
 28    let mut a: VecZnx = module.new_vec_znx(limbs);
 29    a.encode_vec_i64(log_base2k, log_k, &a_values, 32);
 30    a.normalize(log_base2k, &mut buf);
 31
 32    a.print_limbs(a.limbs(), n);
 33    println!();
 34
 35    let mut vecznx: Vec<VecZnx> = Vec::new();
 36    (0..rows).for_each(|_| {
 37        vecznx.push(module.new_vec_znx(cols));
 38    });
 39
 40    (0..rows).for_each(|i| {
 41        vecznx[i].data[i * n + 1] = 1 as i64;
 42    });
 43
 44    let mut vmp_pmat: VmpPMat = module.new_vmp_pmat(rows, cols);
 45    module.vmp_prepare_dblptr(&mut vmp_pmat, &vecznx, &mut buf);
 46
 47    let mut c_dft: VecZnxDft = module.new_vec_znx_dft(cols);
 48    module.vmp_apply_dft(&mut c_dft, &a, &vmp_pmat, &mut buf);
 49
 50    let mut c_big: VecZnxBig = c_dft.as_vec_znx_big();
 51    module.vec_znx_idft_tmp_a(&mut c_big, &mut c_dft, cols);
 52
 53    let mut res: VecZnx = module.new_vec_znx(cols);
 54    module.vec_znx_big_normalize(log_base2k, &mut res, &c_big, &mut buf);
 55
 56    let mut values_res: Vec<i64> = vec![i64::default(); n];
 57    res.decode_vec_i64(log_base2k, log_k, &mut values_res);
 58
 59    res.print_limbs(res.limbs(), n);
 60
 61    module.free();
 62    c_dft.free();
 63    vmp_pmat.free();
 64
 65    //println!("{:?}", values_res)
 66}
 67
 68/*
 69
 70use base2k::{
 71    Encoding, Free, Infos, Matrix3D, Module, VecZnx, VecZnxBig, VecZnxDft, VecZnxOps, VmpPMat,
 72    VmpPMatOps, FFT64,
 73};
 74use std::cmp::min;
 75
 76fn main() {
 77    use base2k::{Module, FFT64, Matrix3D, VmpPMat, VmpPMatOps, VecZnx, VecZnxOps, Free};
 78    use std::cmp::min;
 79
 80    let n: usize = 32;
 81    let module: Module = Module::new::<FFT64>(n);
 82    let rows: usize = 5;
 83    let cols: usize = 6;
 84
 85    let mut vecznx: Vec<VecZnx>= Vec::new();
 86    (0..rows).for_each(|_|{
 87        vecznx.push(module.new_vec_znx(cols));
 88    });
 89
 90    (0..rows).for_each(|i|{
 91        vecznx[i].data[i*n] = 1 as i64;
 92        vecznx[i].print_limbs(cols, n);
 93    });
 94
 95    let mut buf: Vec<u8> = vec![u8::default(); module.vmp_prepare_tmp_bytes(rows, cols)];
 96
 97    let mut vmp_pmat: VmpPMat = module.new_vmp_pmat(rows, cols);
 98
 99    println!("123");
100
101    module.vmp_prepare_dblptr(&mut vmp_pmat, &vecznx, &mut buf);
102
103
104    module.vmp_apply_dft(c, a, b, buf);
105
106    vmp_pmat.free();
107    module.free();
108}
109
110*/