Merge branch 'master' of ssh://bitbucket.org/czan/honours

1 files changed, 29 insertions, 12 deletions

diff --git a/impl/main.cpp b/impl/main.cpp
index 94351ec..1fed389 100644
--- a/impl/main.cpp
+++ b/impl/main.cpp
@@ -1,5 +1,6 @@
 #include "Log.hpp"
 #include <iostream>
+#include <iomanip>
 #include <vector>
 #include <sstream>
 #include "Complete.hpp"
@@ -9,6 +10,8 @@
 #include "MaxStrategy.hpp"
 #include "VariableAssignment.hpp"
 
+#include <ctime>
+
 extern "C" {
 #include "parser/EquationSystemParser.h"
 #include "parser/EquationSystemLexer.h"
@@ -134,44 +137,58 @@ int main (int argc, char* argv[]) {
 
   EquationSystemParser_equation_system_return ret = parser -> equation_system(parser);
 
+  std::cerr << "Parse complete." << std::endl;
+
   EquationSystem<ZBar> system;
   treeToSystem<ZBar>(ret.tree, system);
 
   log::debug << system << endl;
   system.indexMaxExpressions(); // make reverse-lookup O(1) instead of O(n)
 
+  std::cerr << "System created and indexed." << std::endl;
+
   Solver<ZBar> solver(system); // local *and* lazy. I love it!
+
+  timespec start, finish;
+  clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &start);
   if (variables.size() > 0) {
-    for (unsigned int i = 0, size = system.variableCount(); i < size; ++i) {
-      Variable<ZBar>& var = system.variable(i);
-      if (variables.find(var.name()) != variables.end())
-        cout << var.name() << " = " << solver.solve(var) << endl;
+    for (set<string>::iterator it = variables.begin(), ei = variables.end();
+         it != ei;
+         ++it) {
+      solver.solve(system.variable(*it));
     }
   } else {
     for (unsigned int i = 0, size = system.variableCount(); i < size; ++i) {
       Variable<ZBar>& var = system.variable(i);
-      cout << var.name() << " = " << solver.solve(var) << endl;
+      solver.solve(var);
     }
   }
+  clock_gettime(CLOCK_PROCESS_CPUTIME_ID, &finish);
 
-  /*
-  DynamicMaxStrategy<ZBar> strategy(system);
-  DynamicVariableAssignment<ZBar> rho(system, strategy);
-  strategy.setRho(rho);
+  std::cerr << "System solved." << std::endl;
 
   if (variables.size() > 0) {
     for (unsigned int i = 0, size = system.variableCount(); i < size; ++i) {
       Variable<ZBar>& var = system.variable(i);
       if (variables.find(var.name()) != variables.end())
-        cout << var.name() << " = " << rho[var] << endl;
+        cout << var.name() << " = " << solver.solve(var) << endl;
     }
   } else {
     for (unsigned int i = 0, size = system.variableCount(); i < size; ++i) {
       Variable<ZBar>& var = system.variable(i);
-      cout << var.name() << " = " << rho[var] << endl;
+      cout << var.name() << " = " << solver.solve(var) << endl;
     }
   }
-  */
+
+  timespec temp;
+  if ((finish.tv_nsec-start.tv_nsec)<0) {
+    temp.tv_sec = finish.tv_sec-start.tv_sec-1;
+    temp.tv_nsec = 1000000000+finish.tv_nsec-start.tv_nsec;
+  } else {
+    temp.tv_sec = finish.tv_sec-start.tv_sec;
+    temp.tv_nsec = finish.tv_nsec-start.tv_nsec;
+  }
+  cerr << "Time taken: " << temp.tv_sec << "." << setfill('0') << setw(9) << temp.tv_nsec << " seconds." << endl;
 
   parser -> free(parser);
   tokens -> free(tokens);