github/plonkathon
1
1
Fork 0
mirror of https://github.com/0xPARC/plonkathon.git synced 2026-01-08 21:28:02 -05:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5 from franklynwang/main

Browse Source 
support add/sub scalar for monomial representation
This commit is contained in:
ying tong
2023-01-28 17:30:22 -05:00
committed by GitHub
parent 3d358d8172 acf6b7c323
commit 954c48f913
1 changed files with 21 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

29
poly.py
View File

@@ -31,10 +31,16 @@ class Polynomial:
)
else:
assert isinstance(other, Scalar)
return Polynomial(
[x + other for x in self.values],
self.basis,
)
if self.basis == Basis.LAGRANGE:
return Polynomial(
[x + other for x in self.values],
self.basis,
)
else:
return Polynomial(
[self.values[0] + other] + self.values[1:],
self.basis
)
def __sub__(self, other):
if isinstance(other, Polynomial):
@@ -47,10 +53,17 @@ class Polynomial:
)
else:
assert isinstance(other, Scalar)
return Polynomial(
[x - other for x in self.values],
self.basis,
)
if self.basis == Basis.LAGRANGE:
return Polynomial(
[x - other for x in self.values],
self.basis,
)
else:
return Polynomial(
[self.values[0] - other] + self.values[1:],
self.basis
)
def __mul__(self, other):
if isinstance(other, Polynomial):