ok, second try

packages/CLPBN/horus/HorusCli.cpp

@@ -0,0 +1,188 @@
+#include <cstdlib>
+
+#include <iostream>
+#include <sstream>
+
+#include "FactorGraph.h"
+#include "VarElimSolver.h"
+#include "BpSolver.h"
+#include "CbpSolver.h"
+
+using namespace std;
+
+int readHorusFlags (int, const char* []);
+void readFactorGraph (FactorGraph&, const char*);
+VarIds readQueryAndEvidence (FactorGraph&, int, const char* [], int);
+
+void runSolver (const FactorGraph&, const VarIds&);
+
+const string USAGE = "usage: ./hcli [HORUS_FLAG=VALUE] \
+NETWORK_FILE [VARIABLE | OBSERVED_VARIABLE=EVIDENCE] ..." ;
+
+
+int
+main (int argc, const char* argv[])
+{
+  if (argc <= 1) {
+    cerr << "error: no graphical model specified" << endl;
+    cerr << USAGE << endl;
+    exit (0);
+  }
+  int idx = readHorusFlags (argc, argv);
+  FactorGraph fg;
+  readFactorGraph (fg, argv[idx]);
+  VarIds queryIds = readQueryAndEvidence (fg, argc, argv, idx + 1);
+  runSolver (fg, queryIds);
+  return 0;
+}
+
+
+
+int
+readHorusFlags (int argc, const char* argv[])
+{
+  int i = 1;
+  for (; i < argc; i++) {
+    const string& arg = argv[i];
+    size_t pos = arg.find ('=');
+    if (pos == std::string::npos) {
+      return i;
+    }
+    string leftArg  = arg.substr (0, pos);
+    string rightArg = arg.substr (pos + 1);
+    if (leftArg.empty()) {
+      cerr << "error: missing left argument" << endl;
+      cerr << USAGE << endl;
+      exit (0);
+    }
+    if (rightArg.empty()) {
+      cerr << "error: missing right argument" << endl;
+      cerr << USAGE << endl;
+      exit (0);
+    }
+    Util::setHorusFlag (leftArg, rightArg);
+  }
+  return i + 1;
+}
+
+
+
+void
+readFactorGraph (FactorGraph& fg, const char* s)
+{
+  string fileName (s);
+  string extension = fileName.substr (fileName.find_last_of ('.') + 1);
+  if (extension == "uai") {
+    fg.readFromUaiFormat (fileName.c_str());
+  } else if (extension == "fg") {
+    fg.readFromLibDaiFormat (fileName.c_str());
+  } else {
+    cerr << "error: the graphical model must be defined either " ; 
+    cerr << "in a UAI or libDAI file" << endl;
+    exit (0);
+  }
+}
+
+
+
+VarIds
+readQueryAndEvidence (
+    FactorGraph& fg,
+    int argc,
+    const char* argv[],
+    int start)
+{
+  VarIds queryIds;
+  for (int i = start; i < argc; i++) {
+    const string& arg = argv[i];
+    if (arg.find ('=') == std::string::npos) {
+      if (Util::isInteger (arg) == false) {
+        cerr << "error: `" << arg << "' " ;
+        cerr << "is not a variable id" ;
+        cerr << endl;
+        exit (0);
+      }
+      VarId vid = Util::stringToUnsigned (arg);
+      VarNode* queryVar = fg.getVarNode (vid);
+      if (queryVar == false) {
+        cerr << "error: unknow variable with id " ;
+        cerr << "`" << vid << "'"  << endl;
+        exit (0);
+      }
+      queryIds.push_back (vid);
+    } else {
+      size_t pos = arg.find ('=');
+      string leftArg  = arg.substr (0, pos);
+      string rightArg = arg.substr (pos + 1);
+      if (leftArg.empty()) {
+        cerr << "error: missing left argument" << endl;
+        cerr << USAGE << endl;
+        exit (0);
+      }
+      if (Util::isInteger (leftArg) == false) {
+        cerr << "error: `" << leftArg << "' " ;
+        cerr << "is not a variable id" << endl ;
+        exit (0);
+        continue;
+      }
+      VarId vid = Util::stringToUnsigned (leftArg);
+      VarNode* observedVar = fg.getVarNode (vid);
+      if (observedVar == false) {
+        cerr << "error: unknow variable with id " ;
+        cerr << "`" << vid << "'"  << endl;
+        exit (0);
+      }
+      if (rightArg.empty()) {
+        cerr << "error: missing right argument" << endl;
+        cerr << USAGE << endl;
+        exit (0);
+      }
+      if (Util::isInteger (rightArg) == false) {
+        cerr << "error: `" << rightArg << "' " ;
+        cerr << "is not a state index" << endl ;
+        exit (0);
+      }
+      unsigned stateIdx = Util::stringToUnsigned (rightArg);
+      if (observedVar->isValidState (stateIdx) == false) {
+        cerr << "error: `" << stateIdx << "' " ;
+        cerr << "is not a valid state index for variable with id " ;
+        cerr << "`" << vid << "'"  << endl;
+        exit (0);
+      }
+      observedVar->setEvidence (stateIdx);
+    }
+  }
+  return queryIds;
+}
+
+
+
+void
+runSolver (const FactorGraph& fg, const VarIds& queryIds)
+{
+  Solver* solver = 0;
+  switch (Globals::infAlgorithm) {
+    case InfAlgorithms::VE:
+      solver = new VarElimSolver (fg);
+      break;
+    case InfAlgorithms::BP:
+      solver = new BpSolver (fg);
+      break;
+    case InfAlgorithms::CBP:
+      solver = new CbpSolver (fg);
+      break;
+    default:
+      assert (false);
+  }
+  if (Globals::verbosity > 0) {
+    solver->printSolverFlags();
+    cout << endl;
+  }
+  if (queryIds.size() == 0) {
+    solver->printAllPosterioris();
+  } else {
+    solver->printAnswer (queryIds);
+  }
+  delete solver;
+}
+