uzkge 0.1.2

App-specific PlonK with various gadgets & primitives
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57

use ark_ff::PrimeField;
use ark_std::rand::{CryptoRng, Rng, RngCore};

#[derive(Debug)]
pub struct Permutation<F: PrimeField>(Vec<Vec<F>>);

impl<F: PrimeField> Permutation<F> {
    pub fn rand<R: CryptoRng + RngCore>(prng: &mut R, n: usize) -> Self {
        let mut permutation_matrix = vec![vec![F::ZERO; n]; n];

        let mut remainder = (0..n).collect::<Vec<usize>>();

        for i in 0..n {
            let r = prng.gen_range(0..remainder.len());
            let index = remainder[r];
            remainder.remove(r);

            permutation_matrix[i][index] = F::ONE;
        }

        Permutation(permutation_matrix)
    }

    pub fn len(&self) -> usize {
        self.0.len()
    }

    pub fn get_matrix(&self) -> &Vec<Vec<F>> {
        &self.0
    }

    pub fn sanity_check(&self) {
        for matrixs in self.get_matrix().iter() {
            assert_eq!(F::ONE, matrixs.iter().sum())
        }

        (0..self.0.len()).for_each(|j| {
            let x = (0..self.0[j].len()).map(|i| self.get_matrix()[i][j]).sum();
            assert_eq!(F::ONE, x)
        })
    }
}

#[cfg(test)]
mod test {
    use ark_bn254::Fr;
    use ark_std::rand::SeedableRng;
    use rand_chacha::ChaChaRng;

    use super::Permutation;

    #[test]
    fn test_permutation() {
        let mut prng = ChaChaRng::from_entropy();
        Permutation::<Fr>::rand(&mut prng, 52).sanity_check();
    }
}