logos-co
/
nomos-node
mirror of
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Refactor evaluations into method

This commit is contained in:
Daniel Sanchez Quiros 2024-03-30 13:29:46 +01:00
parent a619bb362c
commit 38fe0e8171
1 changed files with 22 additions and 20 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

42
nomos-da/kzgrs/src/common.rs
View File

@ -23,20 +23,12 @@ pub enum KzgRsError {
ChunkSizeTooBig(usize),
}
fn bytes_to_polynomial<const CHUNK_SIZE: usize>(
fn bytes_to_evaluations<const CHUNK_SIZE: usize>(
data: &[u8],
domain: GeneralEvaluationDomain<Fr>,
) -> Result<DensePolynomial<Fr>, KzgRsError> {
if CHUNK_SIZE >= 32 {
return Err(KzgRsError::ChunkSizeTooBig(CHUNK_SIZE));
}
if data.len() % CHUNK_SIZE != 0 {
return Err(KzgRsError::UnpaddedDataError {
expected_modulus: CHUNK_SIZE,
current_size: data.len(),
});
}
let coefficients = Evaluations::from_vec_and_domain(
) -> Evaluations<Fr> {
assert!((data.len() % CHUNK_SIZE).is_zero());
Evaluations::from_vec_and_domain(
data.chunks(CHUNK_SIZE)
.map(|e| {
let mut buff = [0u8; 32];
@ -49,13 +41,24 @@ fn bytes_to_polynomial<const CHUNK_SIZE: usize>(
.collect(),
domain,
)
.interpolate()
.coeffs
.into_iter()
.take(data.len() / CHUNK_SIZE);
Ok(DensePolynomial::from_coefficients_vec(
coefficients.collect(),
))
}
pub fn bytes_to_polynomial<const CHUNK_SIZE: usize>(
data: &[u8],
domain: GeneralEvaluationDomain<Fr>,
) -> Result<DensePolynomial<Fr>, KzgRsError> {
if CHUNK_SIZE >= 32 {
return Err(KzgRsError::ChunkSizeTooBig(CHUNK_SIZE));
}
if data.len() % CHUNK_SIZE != 0 {
return Err(KzgRsError::UnpaddedDataError {
expected_modulus: CHUNK_SIZE,
current_size: data.len(),
});
}
let evals = bytes_to_evaluations::<CHUNK_SIZE>(data, domain);
let coefficients = evals.interpolate();
Ok(coefficients)
}
#[cfg(test)]
@ -78,7 +81,6 @@ mod test {
let mut rng = thread_rng();
bytes.try_fill(&mut rng).unwrap();
let poly = bytes_to_polynomial::<31>(&bytes, *DOMAIN).unwrap();
assert_eq!(poly.degree(), N - 1);
for (i, e) in (0..100).map(|i| 2u32.pow(i)).enumerate() {
let eval_point = Fr::from(e);
let point = poly.evaluate(&eval_point);