C++ LP analysis: add LP debugging helper

Dump a LP solution to a stream for inspection.

2 files changed, 71 insertions, 0 deletions

diff --git a/native/include/linprog/io.h b/native/include/linprog/io.h
index 1e9a3cc..1e8885f 100644
--- a/native/include/linprog/io.h
+++ b/native/include/linprog/io.h
@@ -4,7 +4,17 @@
 #include <ostream>
 
 #include "linprog/model.h"
+#include "linprog/solver.h"
+#include "lp_common.h"
 
 std::ostream& operator<<(std::ostream &os, const LinearExpression &exp);
 
+void dump_lp_solution(
+        VarMapper& vars,
+        const ResourceSharingInfo& info,
+        const TaskInfo& ti,
+        const Solution& solution,
+        std::ostream& out,
+        bool show_zeros = false);
+
 #endif
diff --git a/native/src/linprog/io.cpp b/native/src/linprog/io.cpp
index 1b5d503..14eeca5 100644
--- a/native/src/linprog/io.cpp
+++ b/native/src/linprog/io.cpp
@@ -1,5 +1,6 @@
 #include <iostream>
 
+#include "linprog/solver.h"
 #include "linprog/io.h"
 
 std::ostream& operator<<(std::ostream &os, const LinearExpression &exp)
@@ -20,3 +21,63 @@ std::ostream& operator<<(std::ostream &os, const LinearExpression &exp)
 
         return os;
 }
+
+void dump_lp_solution(
+        VarMapper& vars,
+        const ResourceSharingInfo& info,
+        const TaskInfo& ti,
+        const Solution& solution,
+        std::ostream& out,
+        bool show_zeros)
+{
+        foreach_task_except(info.get_tasks(), ti, tx)
+        {
+                unsigned int t = tx->get_id();
+
+                out << "T" << t << " part=" << tx->get_cluster() << std::endl;
+
+                foreach(tx->get_requests(), request)
+                {
+                        unsigned int q = request->get_resource_id();
+
+                        out << "  res=" << q
+                            << "  L=" << request->get_request_length()
+                            << std::endl;
+
+                        foreach_request_instance(*request, ti, v)
+                        {
+                                unsigned int var_id;
+                                bool newline = false;
+
+                                var_id = vars.lookup(t, q, v, BLOCKING_DIRECT);
+                                if (solution.get_value(var_id) || show_zeros)
+                                {
+                                        out << "    XD_" << t  << "_" << q << "_" << v
+                                            << "=" << solution.get_value(var_id);
+                                        newline = true;
+                                }
+
+                                var_id = vars.lookup(t, q, v, BLOCKING_INDIRECT);
+                                if (solution.get_value(var_id) || show_zeros)
+                                {
+                                        out << "    XI_" << t  << "_" << q << "_" << v
+                                            << "=" << solution.get_value(var_id);
+                                        newline = true;
+                                }
+
+
+                                var_id = vars.lookup(t, q, v, BLOCKING_PREEMPT);
+                                if (solution.get_value(var_id) || show_zeros)
+                                {
+                                        out << "    XP_" << t  << "_" << q << "_" << v
+                                            << "=" << solution.get_value(var_id);
+                                        newline = true;
+                                }
+
+                                if (newline)
+                                        out << std::endl;
+                        }
+                }
+        }
+}
+