1#![allow(non_snake_case)]
 2#![allow(non_upper_case_globals)]
 3#![allow(non_camel_case_types)]
 4#![allow(clippy::upper_case_acronyms)]
 5#![allow(dead_code)]
 6use ark_bn254::{constraints::GVar, Bn254, Fr, G1Projective as G1};
 7use ark_crypto_primitives::snark::SNARK;
 8use ark_groth16::{Groth16, ProvingKey, VerifyingKey as G16VerifierKey};
 9use ark_grumpkin::{constraints::GVar as GVar2, Projective as G2};
10use ark_poly_commit::kzg10::VerifierKey as KZGVerifierKey;
11use ark_std::Zero;
12use std::time::Instant;
13
14use folding_schemes::{
15    commitment::{
16        kzg::{ProverKey as KZGProverKey, KZG},
17        pedersen::Pedersen,
18        CommitmentScheme,
19    },
20    folding::nova::{
21        decider_eth_circuit::DeciderEthCircuit, get_r1cs, Nova, ProverParams, VerifierParams,
22    },
23    frontend::FCircuit,
24    transcript::poseidon::poseidon_canonical_config,
25    FoldingScheme,
26};
27
28// This method computes the Nova's Prover & Verifier parameters for the example.
29// Warning: this method is only for testing purposes. For a real world use case those parameters
30// should be generated carefully (both the PoseidonConfig and the PedersenParams).
31#[allow(clippy::type_complexity)]
32pub(crate) fn init_nova_ivc_params<FC: FCircuit<Fr>>(
33    F_circuit: FC,
34) -> (
35    ProverParams<G1, G2, KZG<'static, Bn254>, Pedersen<G2>>,
36    VerifierParams<G1, G2>,
37    KZGVerifierKey<Bn254>,
38) {
39    let mut rng = ark_std::test_rng();
40    let poseidon_config = poseidon_canonical_config::<Fr>();
41
42    // get the CM & CF_CM len
43    let (r1cs, cf_r1cs) = get_r1cs::<G1, GVar, G2, GVar2, FC>(&poseidon_config, F_circuit).unwrap();
44    let cs_len = r1cs.A.n_rows;
45    let cf_cs_len = cf_r1cs.A.n_rows;
46
47    // let (pedersen_params, _) = Pedersen::<G1>::setup(&mut rng, cf_len).unwrap();
48    let (kzg_pk, kzg_vk): (KZGProverKey<G1>, KZGVerifierKey<Bn254>) =
49        KZG::<Bn254>::setup(&mut rng, cs_len).unwrap();
50    let (cf_pedersen_params, _) = Pedersen::<G2>::setup(&mut rng, cf_cs_len).unwrap();
51
52    let fs_prover_params = ProverParams::<G1, G2, KZG<Bn254>, Pedersen<G2>> {
53        poseidon_config: poseidon_config.clone(),
54        cs_params: kzg_pk.clone(),
55        cf_cs_params: cf_pedersen_params,
56    };
57    let fs_verifier_params = VerifierParams::<G1, G2> {
58        poseidon_config: poseidon_config.clone(),
59        r1cs,
60        cf_r1cs,
61    };
62    (fs_prover_params, fs_verifier_params, kzg_vk)
63}
64
65/// Initializes Nova parameters and DeciderEth parameters. Only for test purposes.
66#[allow(clippy::type_complexity)]
67pub(crate) fn init_ivc_and_decider_params<FC: FCircuit<Fr>>(
68    f_circuit: FC,
69) -> (
70    ProverParams<G1, G2, KZG<'static, Bn254>, Pedersen<G2>>,
71    KZGVerifierKey<Bn254>,
72    ProvingKey<Bn254>,
73    G16VerifierKey<Bn254>,
74) {
75    let mut rng = rand::rngs::OsRng;
76    let start = Instant::now();
77    let (fs_prover_params, _, kzg_vk) = init_nova_ivc_params::<FC>(f_circuit.clone());
78    println!("generated Nova folding params: {:?}", start.elapsed());
79
80    pub type NOVA<FC> = Nova<G1, GVar, G2, GVar2, FC, KZG<'static, Bn254>, Pedersen<G2>>;
81    let z_0 = vec![Fr::zero(); f_circuit.state_len()];
82    let nova = NOVA::init(&fs_prover_params, f_circuit, z_0.clone()).unwrap();
83
84    let decider_circuit =
85        DeciderEthCircuit::<G1, GVar, G2, GVar2, KZG<Bn254>, Pedersen<G2>>::from_nova::<FC>(
86            nova.clone(),
87        )
88        .unwrap();
89    let start = Instant::now();
90    let (g16_pk, g16_vk) =
91        Groth16::<Bn254>::circuit_specific_setup(decider_circuit.clone(), &mut rng).unwrap();
92    println!(
93        "generated G16 (Decider circuit) params: {:?}",
94        start.elapsed()
95    );
96    (fs_prover_params, kzg_vk, g16_pk, g16_vk)
97}
98
99fn main() {}