Allow clearing of SolverModel.

erdos-project · Oct 23, 2023 · 8b2b7f7 · 8b2b7f7
1 parent 9d1527f
commit 8b2b7f7
Show file tree

Hide file tree

Showing 3 changed files with 13 additions and 0 deletions.
diff --git a/schedulers/tetrisched/include/tetrisched/SolverModel.hpp b/schedulers/tetrisched/include/tetrisched/SolverModel.hpp
@@ -375,6 +375,9 @@ class SolverModelT {
   /// Throws an error if the model was not solved first.
   T getObjectiveValue() const;
 
+  /// Clears all the Variables and Constraints from this Model.
+  void clear();
+
   /// All the Solver implementations should be a friend of the SolverModel.
   /// This allows Solver implementations to construct the model to pass
   /// back to the user.

diff --git a/schedulers/tetrisched/src/Scheduler.cpp b/schedulers/tetrisched/src/Scheduler.cpp
@@ -32,6 +32,9 @@ Scheduler::Scheduler(Time discretization, SolverBackendType solverBackend)
 
 void Scheduler::registerSTRL(ExpressionPtr expression,
                              Partitions availablePartitions, Time currentTime) {
+  // Clear the previously saved expressions in the SolverModel.
+  // solverModel->clear();
+
   // Check if the expression is an objective function.
   if (expression->getType() != ExpressionType::EXPR_OBJECTIVE) {
     throw exceptions::ExpressionConstructionException(

diff --git a/schedulers/tetrisched/src/SolverModel.cpp b/schedulers/tetrisched/src/SolverModel.cpp
@@ -404,4 +404,11 @@ template <typename T>
 T SolverModelT<T>::getObjectiveValue() const {
   return objectiveFunction->getValue();
 }
+
+template <typename T>
+void SolverModelT<T>::clear() {
+  variables.clear();
+  constraints.clear();
+  objectiveFunction.reset();
+}
 }  // namespace tetrisched