poulpy_core/layouts/
lwe_to_glwe_ksk.rs

1use std::fmt;
2
3use poulpy_hal::{
4    layouts::{Data, DataMut, DataRef, FillUniform, MatZnx, ReaderFrom, Reset, WriterTo},
5    source::Source,
6};
7
8use crate::layouts::{GGLWESwitchingKey, Infos};
9
10#[derive(PartialEq, Eq, Clone)]
11pub struct LWEToGLWESwitchingKey<D: Data>(pub(crate) GGLWESwitchingKey<D>);
12
13impl<D: DataRef> fmt::Debug for LWEToGLWESwitchingKey<D> {
14    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
15        write!(f, "{}", self)
16    }
17}
18
19impl<D: DataMut> FillUniform for LWEToGLWESwitchingKey<D> {
20    fn fill_uniform(&mut self, log_bound: usize, source: &mut Source) {
21        self.0.fill_uniform(log_bound, source);
22    }
23}
24
25impl<D: DataMut> Reset for LWEToGLWESwitchingKey<D> {
26    fn reset(&mut self) {
27        self.0.reset();
28    }
29}
30
31impl<D: DataRef> fmt::Display for LWEToGLWESwitchingKey<D> {
32    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
33        write!(f, "(LWEToGLWESwitchingKey) {}", self.0)
34    }
35}
36
37impl<D: Data> Infos for LWEToGLWESwitchingKey<D> {
38    type Inner = MatZnx<D>;
39
40    fn inner(&self) -> &Self::Inner {
41        self.0.inner()
42    }
43
44    fn basek(&self) -> usize {
45        self.0.basek()
46    }
47
48    fn k(&self) -> usize {
49        self.0.k()
50    }
51}
52
53impl<D: Data> LWEToGLWESwitchingKey<D> {
54    pub fn digits(&self) -> usize {
55        self.0.digits()
56    }
57
58    pub fn rank(&self) -> usize {
59        self.0.rank()
60    }
61
62    pub fn rank_in(&self) -> usize {
63        self.0.rank_in()
64    }
65
66    pub fn rank_out(&self) -> usize {
67        self.0.rank_out()
68    }
69}
70
71impl<D: DataMut> ReaderFrom for LWEToGLWESwitchingKey<D> {
72    fn read_from<R: std::io::Read>(&mut self, reader: &mut R) -> std::io::Result<()> {
73        self.0.read_from(reader)
74    }
75}
76
77impl<D: DataRef> WriterTo for LWEToGLWESwitchingKey<D> {
78    fn write_to<W: std::io::Write>(&self, writer: &mut W) -> std::io::Result<()> {
79        self.0.write_to(writer)
80    }
81}
82
83impl LWEToGLWESwitchingKey<Vec<u8>> {
84    pub fn alloc(n: usize, basek: usize, k: usize, rows: usize, rank_out: usize) -> Self {
85        Self(GGLWESwitchingKey::alloc(n, basek, k, rows, 1, 1, rank_out))
86    }
87}
poulpy_core/layouts/lwe_to_glwe_ksk.rs

poulpy_core/layouts/
lwe_to_glwe_ksk.rs
