From f5b21dacb81cb2d8a3a23ab455dab2c0bb983a17 Mon Sep 17 00:00:00 2001
From: Michael Schneeberger <michael.schneeberger@fhnw.ch>
Date: Wed, 10 Aug 2022 10:09:32 +0200
Subject: add inverse property to truncate expression
---
polymatrix/expression/impl/truncateexprimpl.py | 1 +
polymatrix/expression/init/inittruncateexpr.py | 7 ++++++-
polymatrix/expression/mixins/truncateexprmixin.py | 8 ++++++--
3 files changed, 13 insertions(+), 3 deletions(-)
diff --git a/polymatrix/expression/impl/truncateexprimpl.py b/polymatrix/expression/impl/truncateexprimpl.py
index e4d10ae..5bd45f3 100644
--- a/polymatrix/expression/impl/truncateexprimpl.py
+++ b/polymatrix/expression/impl/truncateexprimpl.py
@@ -8,3 +8,4 @@ class TruncateExprImpl(TruncateExpr):
underlying: ExpressionBaseMixin
variables: ExpressionBaseMixin
degrees: tuple[int]
+ inverse: bool
diff --git a/polymatrix/expression/init/inittruncateexpr.py b/polymatrix/expression/init/inittruncateexpr.py
index 060ebaf..4582bcb 100644
--- a/polymatrix/expression/init/inittruncateexpr.py
+++ b/polymatrix/expression/init/inittruncateexpr.py
@@ -6,12 +6,17 @@ def init_truncate_expr(
underlying: ExpressionBaseMixin,
variables: ExpressionBaseMixin,
degrees: tuple[int],
+ inverse: bool = None,
):
if isinstance(degrees, int):
degrees = (degrees,)
+ if inverse is None:
+ inverse = False
+
return TruncateExprImpl(
underlying=underlying,
variables=variables,
degrees=degrees,
-)
+ inverse=inverse,
+ )
diff --git a/polymatrix/expression/mixins/truncateexprmixin.py b/polymatrix/expression/mixins/truncateexprmixin.py
index 2421f48..42f192e 100644
--- a/polymatrix/expression/mixins/truncateexprmixin.py
+++ b/polymatrix/expression/mixins/truncateexprmixin.py
@@ -8,7 +8,6 @@ from polymatrix.expressionstate.expressionstate import ExpressionState
from polymatrix.expression.utils.getvariableindices import get_variable_indices
-# replace by filter operation?
class TruncateExprMixin(ExpressionBaseMixin):
@property
@abc.abstractmethod
@@ -25,6 +24,11 @@ class TruncateExprMixin(ExpressionBaseMixin):
def degrees(self) -> tuple[int]:
...
+ @property
+ @abc.abstractmethod
+ def inverse(self) -> bool:
+ ...
+
# overwrites abstract method of `ExpressionBaseMixin`
def apply(
self,
@@ -49,7 +53,7 @@ class TruncateExprMixin(ExpressionBaseMixin):
degree = sum((count for var_idx, count in monomial if var_idx in variable_indices))
- if degree in self.degrees:
+ if (degree in self.degrees) is not self.inverse:
terms_row_col[monomial] = value
terms[row, col] = terms_row_col
--
cgit v1.2.1