add update function for commitment type

o1-labs · Feb 10, 2025 · 40851bd · 40851bd
1 parent b83f4e2
commit 40851bd
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/saffron/src/commitment.rs b/saffron/src/commitment.rs
@@ -12,6 +12,7 @@ use poly_commitment::{
 use rayon::prelude::*;
 use serde::{Deserialize, Serialize};
 use serde_with::serde_as;
+use std::ops::Add;
 use tracing::instrument;
 
 #[serde_as]
@@ -36,6 +37,14 @@ impl<G: KimchiCurve> Commitment<G> {
             folded,
         }
     }
+
+    pub fn update<EFqSponge>(&self, diff: Vec<PolyComm<G>>, sponge: &mut EFqSponge) -> Self
+    where
+        EFqSponge: FqSponge<G::BaseField, G, G::ScalarField>,
+    {
+        let new_chunks = self.chunks.iter().zip(diff).map(|(g, d)| g.add(&d));
+        Self::from_chunks(new_chunks.collect(), sponge)
+    }
 }
 
 #[instrument(skip_all, level = "debug")]