add check_proof_multi to KZG

primitives/src/kzg.rs

@@ -304,6 +304,22 @@ impl KZG {
 		self.fs.max_width
 	}
 
+	pub fn get_expanded_roots_of_unity_at(&self, i: usize) -> FsFr {
+		self.fs.get_expanded_roots_of_unity_at(i)
+	}
+
+	pub fn check_proof_multi(
+		&self,
+		commitment: &KZGCommitment,
+		i: usize,
+		values: &[FsFr],
+		proof: &KZGProof,
+	) -> Result<bool, String> {
+		let x = self.fs.get_expanded_roots_of_unity_at(i);
+		self.ks
+			.check_proof_multi(&commitment.0,&proof.0, &x, values ,FIELD_ELEMENTS_PER_BLOB)
+	}
+
 	pub fn compute_proof(&self, poly: &FsPoly, point_index: usize) -> Result<KZGProof, String> {
 		let x = self.ks.get_expanded_roots_of_unity_at(point_index as usize);
 		self.ks.compute_proof_single(poly, &x).map(KZGProof)

primitives/src/segment.rs

@@ -40,13 +40,11 @@ impl Segment {
 		let domain_stride = kzg.max_width() / (2 * FIELD_ELEMENTS_PER_BLOB);
 		let chunk_count = FIELD_ELEMENTS_PER_BLOB / SEGMENT_LENGTH;
 		let domain_pos = Self::reverse_bits_limited(chunk_count, self.position.x as usize);
-		let x = kzg.fs.get_expanded_roots_of_unity_at(domain_pos * domain_stride);
-		kzg.ks.check_proof_multi(
-			&commitment.0,
-			&self.proof,
-			&x,
+		kzg.check_proof_multi(
+			&commitment,
+			domain_pos * domain_stride,
 			&self.content,
-			FIELD_ELEMENTS_PER_BLOB,
+			&self.proof,
 		)
 	}