Compression integration (draft) [JOLT-191]

jolt-core/src/poly/commitment/commitment_scheme.rs

@@ -13,6 +13,15 @@ pub trait CommitmentScheme: Clone + Sync + Send + 'static {
     type Field: JoltField + Sized;
     type ProverSetup: Clone + Sync + Send + Debug + CanonicalSerialize + CanonicalDeserialize;
     type VerifierSetup: Clone + Sync + Send + Debug + CanonicalSerialize + CanonicalDeserialize;
+    type CompressedCommitment: Default
+        + Debug
+        + Sync
+        + Send
+        + PartialEq
+        + CanonicalSerialize
+        + CanonicalDeserialize
+        + AppendToTranscript
+        + Clone;
     type Commitment: Default
         + Debug
         + Sync

jolt-core/src/poly/commitment/dory/commitment_scheme.rs

@@ -3,8 +3,8 @@
 use super::dory_globals::DoryGlobals;
 use super::jolt_dory_routines::{JoltG1Routines, JoltG2Routines};
 use super::wrappers::{
-    jolt_to_ark, ArkDoryProof, ArkFr, ArkG1, ArkGT, ArkworksProverSetup, ArkworksVerifierSetup,
-    JoltToDoryTranscript, BN254,
+    jolt_to_ark, ArkDoryProof, ArkFr, ArkG1, ArkGT, ArkGTCompressed, ArkworksProverSetup,
+    ArkworksVerifierSetup, JoltToDoryTranscript, BN254,
 };
 use crate::{
     field::JoltField,
@@ -26,7 +26,9 @@ use std::borrow::Borrow;
 use tracing::trace_span;
 
 #[derive(Clone)]
-pub struct DoryCommitmentScheme;
+pub struct DoryCommitmentScheme {
+    compression_enabled: bool,
+}
 
 impl CommitmentScheme for DoryCommitmentScheme {
     type Field = ark_bn254::Fr;
@@ -36,6 +38,7 @@ impl CommitmentScheme for DoryCommitmentScheme {
     type Proof = ArkDoryProof;
     type BatchedProof = Vec<ArkDoryProof>;
     type OpeningProofHint = Vec<ArkG1>;
+    type CompressedCommitment = ArkGTCompressed;
 
     fn setup_prover(max_num_vars: usize) -> Self::ProverSetup {
         let _span = trace_span!("DoryCommitmentScheme::setup_prover").entered();

jolt-core/src/poly/commitment/dory/wrappers.rs

@@ -23,7 +23,8 @@
 use rayon::prelude::*;
 
 pub use dory::backends::arkworks::{
-    ArkDoryProof, ArkFr, ArkG1, ArkG2, ArkGT, ArkworksProverSetup, ArkworksVerifierSetup, BN254,
+    ArkDoryProof, ArkFr, ArkG1, ArkG2, ArkGT, ArkGTCompressed, ArkworksProverSetup,
+    ArkworksVerifierSetup, BN254,
 };
 
 pub type JoltFieldWrapper = ArkFr;
@@ -40,6 +41,18 @@
     }
 }
 
+impl AppendToTranscript for ArkGTCompressed {
+    fn append_to_transcript<S: Transcript>(&self, transcript: &mut S) {
+        transcript.append_serializable(self);
+    }
+}
+
+impl AppendToTranscript for &ArkGTCompressed {
+    fn append_to_transcript<S: Transcript>(&self, transcript: &mut S) {
+        transcript.append_serializable(*self);
+    }
+}
+
 impl AppendToTranscript for ArkDoryProof {
     fn append_to_transcript<S: Transcript>(&self, transcript: &mut S) {
         transcript.append_serializable(self);

jolt-core/src/poly/commitment/hyperkzg.rs

@@ -423,6 +423,7 @@ where
     type Proof = HyperKZGProof<P>;
     type BatchedProof = HyperKZGProof<P>;
     type OpeningProofHint = ();
+    type CompressedCommitment = Self::Commitment;
 
     fn setup_prover(max_num_vars: usize) -> Self::ProverSetup {
         HyperKZGSRS(Arc::new(SRS::setup(

jolt-core/src/poly/commitment/mock.rs

@@ -44,6 +44,7 @@ where
     type Proof = MockProof<F>;
     type BatchedProof = MockProof<F>;
     type OpeningProofHint = ();
+    type CompressedCommitment = Self::Commitment;
 
     fn setup_prover(_num_vars: usize) -> Self::ProverSetup {}