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>;
}
pub trait Group<E: Engine>
pub trait Group<E: Engine>: Copy
{
fn group_mul_assign(&mut self, &E, scalar: &E::Fr);
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 m: u64,
@ -46,12 +46,12 @@ impl<E: Engine> EvaluationDomain<E> {
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);
self._fft(e, v, &self.omegainv);
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);
}
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 {
let mut r = 0;
@ -118,12 +118,12 @@ impl<E: Engine> EvaluationDomain<E> {
while k < self.m {
let mut w = E::Fr::one(e);
for j in 0..m {
let mut t = w;
t.mul_assign(e, &a[(k+j+m) as usize]);
let mut t = a[(k+j+m) as usize];
t.group_mul_assign(e, &w);
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) as usize].add_assign(e, &t);
a[(k+j) as usize].group_add_assign(e, &t);
w.mul_assign(e, &w_m);
}