Merge pull request #13 from pluflou/lastvalequal

added LastValEqual component class
cvxgrp · Apr 14, 2023 · dbf2126 · dbf2126
2 parents fb301bc + 86d0349
commit dbf2126
Show file tree

Hide file tree

Showing 2 changed files with 27 additions and 0 deletions.
diff --git a/gfosd/components/__init__.py b/gfosd/components/__init__.py
@@ -8,6 +8,7 @@
 from gfosd.components.finite_set import FiniteSet, Boolean
 from gfosd.components.aggregate import Aggregate
 from gfosd.components.equality_constraints import (FirstValEqual,
+                                                   LastValEqual,
                                                    AverageEqual,
                                                    NoCurvature,
                                                    NoSlope)
diff --git a/gfosd/components/equality_constraints.py b/gfosd/components/equality_constraints.py
@@ -27,7 +27,33 @@ def _make_B(self):
     def _make_c(self):
         self._c = np.concatenate([np.atleast_1d(self._c[0]),
                                   [self._first_val]])
+
+class LastValEqual(GraphComponent):
+    def __init__(self, value=0, *args, **kwargs):
+        self._last_val = value
+        super().__init__(*args, **kwargs)
+        # always retain helper variable
+        self._has_helpers = True
 
+    def _make_A(self):
+        super()._make_A()
+        super()._make_B()
+        super()._make_c()
+        self._A = sp.bmat([
+           [self._A.tocsr()[-1]],
+            [sp.dok_matrix((1, self._A.shape[1]))]
+        ])
+
+    def _make_B(self):
+        self._B = sp.bmat([
+           [self._B.tocsr()[-1]],
+            [sp.coo_matrix(([1], ([0], [self._B.shape[1]-1])), shape=(1, self._B.shape[1]))]
+        ])
+
+    def _make_c(self):
+        self._c = np.concatenate([np.atleast_1d(self._c[-1]),
+                                  [self._last_val]])
+
 class AverageEqual(GraphComponent):
     def __init__(self, value=0, period=None, *args, **kwargs):
         self._avg_val = value