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