poulpy/base2k/src/scalar_vector_product.rs
base2k/src/scalar_vector_product.rsRaw
1use crate::bindings::{new_svp_ppol, svp_apply_dft, svp_prepare};
2use crate::scalar::Scalar;
3use crate::{Module, SvpPPol, VecZnx, VecZnxDft};
4
5impl Module {
6 // Prepares a scalar polynomial (1 limb) for a scalar x vector product.
7 // Method will panic if a.limbs() != 1.
8 pub fn svp_prepare(&self, svp_ppol: &mut SvpPPol, a: &Scalar) {
9 unsafe { svp_prepare(self.0, svp_ppol.0, a.as_ptr()) }
10 }
11
12 // Allocates a scalar-vector-product prepared-poly (VecZnxBig).
13 pub fn svp_new_ppol(&self) -> SvpPPol {
14 unsafe { SvpPPol(new_svp_ppol(self.0)) }
15 }
16
17 // Applies a scalar x vector product: res <- a (ppol) x b
18 pub fn svp_apply_dft(&self, c: &mut VecZnxDft, a: &SvpPPol, b: &VecZnx) {
19 let limbs: u64 = b.limbs() as u64;
20 assert!(
21 c.limbs() as u64 >= limbs,
22 "invalid c_vector: c_vector.limbs()={} < b.limbs()={}",
23 c.limbs(),
24 limbs
25 );
26 unsafe { svp_apply_dft(self.0, c.0, limbs, a.0, b.as_ptr(), limbs, b.n() as u64) }
27 }
28}