diff --git a/examples/main.rs b/examples/main.rs index 25cb23d..355d0cb 100644 --- a/examples/main.rs +++ b/examples/main.rs @@ -5,9 +5,9 @@ use math::dft::ntt::Table; fn main() { // Example usage of `Prime` - let q_base: u64 = 0x1fffffffffe00001; // Example prime base + let q_base: u64 = 65537; // Example prime base let q_power: usize = 1; // Example power - let mut prime_instance: Prime = Prime::::new(q_base, q_power); + let prime_instance: Prime = Prime::::new(q_base, q_power); // Display the fields of `Prime` to verify println!("Prime instance created:"); @@ -15,7 +15,7 @@ fn main() { println!("q_base: {}", prime_instance.q_base()); println!("q_power: {}", prime_instance.q_power()); - let n: u64 = 1024; + let n: u64 = 32; let nth_root: u64 = n<<1; let ntt_table: Table = Table::::new(prime_instance, nth_root); @@ -38,4 +38,15 @@ fn main() { let r : Ring = Ring::::new(n as usize, q_base, q_power); + let mut p0: math::poly::Poly = r.new_poly(); + let mut p1: math::poly::Poly = r.new_poly(); + + for i in 0..p0.n(){ + p0.0[i] = i as u64 + } + + r.automorphism(p0, (2*r.n-1) as u64, &mut p1); + + println!("{:?}", p1); + } \ No newline at end of file diff --git a/src/poly.rs b/src/poly.rs index d1e4d92..1cf8541 100644 --- a/src/poly.rs +++ b/src/poly.rs @@ -1,5 +1,6 @@ pub mod poly; +#[derive(Clone, Debug, PartialEq, Eq)] pub struct Poly(pub Vec); impl Poly{ diff --git a/src/ring/impl_u64/automorphism.rs b/src/ring/impl_u64/automorphism.rs index c6dcb60..c78b2ce 100644 --- a/src/ring/impl_u64/automorphism.rs +++ b/src/ring/impl_u64/automorphism.rs @@ -34,10 +34,10 @@ impl Ring{ let a_vec: &_ = &a.0; for i in 0..n{ - let i_in: u64 = i as u64 * gal_el; - let i_out: u64 = i_in & mask; - let sign: u64 = (i_in>>log_n) & 1; - b_vec[i_out as usize] = a_vec[i_in as usize] * (sign^1) | (q - a_vec[i_in as usize]) * sign + let gal_el_i: u64 = i as u64 * gal_el; + let i_out: u64 = gal_el_i & mask; + let sign: u64 = (gal_el_i>>log_n) & 1; + b_vec[i_out as usize] = a_vec[i] * (sign^1) | (q - a_vec[i]) * sign } } } \ No newline at end of file