compression.rs

use super::Prover;
use crate::{
    config::layer_config_path,
    io::{load_snark, write_snark},
    utils::gen_rng,
};
use anyhow::{anyhow, Result};
use compression::{utils::ce_snark_to_snark, CompressionCircuit};
use rand::Rng;
use snark_verifier_sdk::Snark;
use ce_snark_verifier_sdk::halo2::aggregation::AggregationConfigParams;
use std::env;

impl<'params> Prover<'params> {
    pub fn gen_comp_snark(
        &mut self,
        id: &str,
        has_accumulator: bool,
        degree: u32,
        mut rng: impl Rng + Send,
        prev_snark: Snark,
    ) -> Result<Snark> {
        env::set_var("COMPRESSION_CONFIG", layer_config_path(id));
        let circuit =
            CompressionCircuit::new(AggregationConfigParams::from_path(layer_config_path(id)), self.params(degree), prev_snark, has_accumulator, &mut rng)
                .map_err(|err| anyhow!("Failed to construct compression circuit: {err:?}"))?;
        let ce_snark = self.gen_snark_ce(id, degree, &mut rng, circuit, "gen_comp_snark")?;
        Ok(ce_snark_to_snark(ce_snark))
    }

    pub fn load_or_gen_comp_snark(
        &mut self,
        name: &str,
        id: &str,
        has_accumulator: bool,
        degree: u32,
        prev_snark: Snark,
        output_dir: Option<&str>,
    ) -> Result<Snark> {
        let file_path = format!(
            "{}/compression_snark_{}_{}.json",
            output_dir.unwrap_or_default(),
            id,
            name
        );

        match output_dir.and_then(|_| load_snark(&file_path).ok().flatten()) {
            Some(snark) => Ok(snark),
            None => {
                let rng = gen_rng();
                let result = self.gen_comp_snark(id, has_accumulator, degree, rng, prev_snark);
                if let (Some(_), Ok(snark)) = (output_dir, &result) {
                    write_snark(&file_path, snark);
                }

                result
            }
        }
    }
}