tornado-packages/phase2-bn254
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Allow evaluation domain to act over all groups.

Browse Source
This commit is contained in:
Sean Bowe 2017-05-07 18:25:49 -06:00
parent 1434ad7b28
commit 750131f0a1
2 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
src/curves/mod.rs
View File

@ -46,7 +46,7 @@ pub trait Engine: Sized + Clone
fn batch_baseexp<G: Curve<Self>, S: AsRef<[Self::Fr]>>(&self, table: &WindowTable<Self, G, Vec<G>>, scalars: S) -> Vec<G::Affine>; fn batch_baseexp<G: Curve<Self>, S: AsRef<[Self::Fr]>>(&self, table: &WindowTable<Self, G, Vec<G>>, scalars: S) -> Vec<G::Affine>;
} }
pub trait Group<E: Engine> pub trait Group<E: Engine>: Copy
{ {
fn group_mul_assign(&mut self, &E, scalar: &E::Fr); fn group_mul_assign(&mut self, &E, scalar: &E::Fr);
fn group_add_assign(&mut self, &E, other: &Self); fn group_add_assign(&mut self, &E, other: &Self);

18
src/groth16/domain.rs
View File

@ -1,4 +1,4 @@
use curves::{Engine, Field, SnarkField, PrimeField}; use curves::{Engine, Field, SnarkField, PrimeField, Group};
pub struct EvaluationDomain<E: Engine> { pub struct EvaluationDomain<E: Engine> {
pub m: u64, pub m: u64,
@ -46,12 +46,12 @@ impl<E: Engine> EvaluationDomain<E> {
tmp tmp
} }
pub fn ifft(&self, e: &E, v: &mut [E::Fr]) pub fn ifft<T: Group<E>>(&self, e: &E, v: &mut [T])
{ {
assert!(v.len() == self.m as usize); assert!(v.len() == self.m as usize);
self._fft(e, v, &self.omegainv); self._fft(e, v, &self.omegainv);
for v in v { for v in v {
v.mul_assign(e, &self.minv); v.group_mul_assign(e, &self.minv);
} }
} }
@ -84,12 +84,12 @@ impl<E: Engine> EvaluationDomain<E> {
} }
} }
pub fn fft(&self, e: &E, a: &mut [E::Fr]) pub fn fft<T: Group<E>>(&self, e: &E, a: &mut [T])
{ {
self._fft(e, a, &self.omega); self._fft(e, a, &self.omega);
} }
fn _fft(&self, e: &E, a: &mut [E::Fr], omega: &E::Fr) fn _fft<T: Group<E>>(&self, e: &E, a: &mut [T], omega: &E::Fr)
{ {
fn bitreverse(mut n: usize, l: u64) -> usize { fn bitreverse(mut n: usize, l: u64) -> usize {
let mut r = 0; let mut r = 0;
@ -118,12 +118,12 @@ impl<E: Engine> EvaluationDomain<E> {
while k < self.m { while k < self.m {
let mut w = E::Fr::one(e); let mut w = E::Fr::one(e);
for j in 0..m { for j in 0..m {
let mut t = w; let mut t = a[(k+j+m) as usize];
t.mul_assign(e, &a[(k+j+m) as usize]); t.group_mul_assign(e, &w);
let mut tmp = a[(k+j) as usize]; let mut tmp = a[(k+j) as usize];
tmp.sub_assign(e, &t); tmp.group_sub_assign(e, &t);
a[(k+j+m) as usize] = tmp; a[(k+j+m) as usize] = tmp;
a[(k+j) as usize].add_assign(e, &t); a[(k+j) as usize].group_add_assign(e, &t);
w.mul_assign(e, &w_m); w.mul_assign(e, &w_m);
} }