mirror of
https://github.com/logos-blockchain/logos-blockchain-specs.git
synced 2026-01-07 15:43:07 +00:00
09c9b7e4ec
* Implement generator polynomial and rs encoding * Implement encode/decode+test using fft. Non-working * Use lagrange for interpolation * Remove fft, use evaluations instead * Move and rename kzg and rs test modules * Update docs
56 lines
1.8 KiB
Python
56 lines
1.8 KiB
Python
from typing import Sequence, List
|
|
|
|
import scipy.interpolate
|
|
from eth2spec.deneb.mainnet import BLSFieldElement
|
|
from eth2spec.eip7594.mainnet import interpolate_polynomialcoeff
|
|
from .common import G1, BLS_MODULUS
|
|
from .poly import Polynomial
|
|
|
|
ExtendedData = Sequence[BLSFieldElement]
|
|
|
|
|
|
def encode(polynomial: Polynomial, factor: int, roots_of_unity: Sequence[BLSFieldElement]) -> ExtendedData:
|
|
"""
|
|
Encode a polynomial extending to the given factor
|
|
Parameters:
|
|
polynomial: Polynomial to be encoded
|
|
factor: Encoding factor
|
|
roots_of_unity: Powers of 2 sequence
|
|
|
|
Returns:
|
|
list: Extended data set
|
|
"""
|
|
assert factor >= 2
|
|
assert len(polynomial)*factor <= len(roots_of_unity)
|
|
return [polynomial.eval(e) for e in roots_of_unity[:len(polynomial)*factor]]
|
|
|
|
|
|
def __interpolate(evaluations: List[int], roots_of_unity: List[int]) -> List[int]:
|
|
"""
|
|
Lagrange interpolation
|
|
|
|
Parameters:
|
|
evaluations: List of evaluations
|
|
roots_of_unity: Powers of 2 sequence
|
|
|
|
Returns:
|
|
list: Coefficients of the interpolated polynomial
|
|
"""
|
|
return list(map(int, interpolate_polynomialcoeff(roots_of_unity[:len(evaluations)], evaluations)))
|
|
|
|
|
|
def decode(encoded: ExtendedData, roots_of_unity: Sequence[BLSFieldElement], original_len: int) -> Polynomial:
|
|
"""
|
|
Decode a polynomial from an extended data-set and the roots of unity, cap to original length
|
|
|
|
Parameters:
|
|
encoded: Extended data set
|
|
roots_of_unity: Powers of 2 sequence
|
|
original_len: Original length of the encoded polynomial
|
|
|
|
Returns:
|
|
Polynomial: original polynomial
|
|
"""
|
|
coefs = __interpolate(list(map(int, encoded)), list(map(int, roots_of_unity)))[:original_len]
|
|
return Polynomial([int(c) for c in coefs], BLS_MODULUS)
|