This repository has been archived on 2023-08-20. You can view files and clone it, but cannot push or open issues or pull requests.
yap-6.3/packages/CLPBN/horus/ElimGraph.cpp

247 lines
5.8 KiB
C++
Raw Normal View History

2013-02-07 20:09:10 +00:00
#include <iostream>
2012-05-23 14:56:01 +01:00
#include <fstream>
#include "ElimGraph.h"
2013-02-07 23:53:13 +00:00
namespace Horus {
2013-02-07 23:53:13 +00:00
ElimHeuristic ElimGraph::elimHeuristic_ = minNeighborsEh;
2012-05-23 14:56:01 +01:00
2013-02-07 13:37:15 +00:00
ElimGraph::ElimGraph (const std::vector<Factor*>& factors)
2012-05-23 14:56:01 +01:00
{
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < factors.size(); i++) {
2012-12-27 12:54:58 +00:00
if (factors[i]) {
const VarIds& args = factors[i]->arguments();
for (size_t j = 0; j < args.size() - 1; j++) {
EgNode* n1 = getEgNode (args[j]);
if (!n1) {
n1 = new EgNode (args[j], factors[i]->range (j));
addNode (n1);
2012-05-23 14:56:01 +01:00
}
2012-12-27 12:54:58 +00:00
for (size_t k = j + 1; k < args.size(); k++) {
EgNode* n2 = getEgNode (args[k]);
if (!n2) {
n2 = new EgNode (args[k], factors[i]->range (k));
addNode (n2);
}
if (!neighbors (n1, n2)) {
addEdge (n1, n2);
}
2012-12-20 23:19:10 +00:00
}
2012-05-23 14:56:01 +01:00
}
2012-12-27 12:54:58 +00:00
if (args.size() == 1 && !getEgNode (args[0])) {
addNode (new EgNode (args[0], factors[i]->range (0)));
2012-05-23 14:56:01 +01:00
}
}
}
}
ElimGraph::~ElimGraph (void)
{
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < nodes_.size(); i++) {
2012-12-17 18:39:42 +00:00
delete nodes_[i];
2012-05-23 14:56:01 +01:00
}
}
VarIds
2012-12-27 12:54:58 +00:00
ElimGraph::getEliminatingOrder (const VarIds& excludedVids)
2012-05-23 14:56:01 +01:00
{
VarIds elimOrder;
unmarked_.reserve (nodes_.size());
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < nodes_.size(); i++) {
if (Util::contains (excludedVids, nodes_[i]->varId()) == false) {
2012-05-23 14:56:01 +01:00
unmarked_.insert (nodes_[i]);
}
}
2012-12-27 12:54:58 +00:00
size_t nrVarsToEliminate = nodes_.size() - excludedVids.size();
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < nrVarsToEliminate; i++) {
2012-05-23 14:56:01 +01:00
EgNode* node = getLowestCostNode();
unmarked_.remove (node);
const EGNeighs& neighs = node->neighbors();
2012-05-24 22:55:20 +01:00
for (size_t j = 0; j < neighs.size(); j++) {
2012-05-23 14:56:01 +01:00
neighs[j]->removeNeighbor (node);
}
elimOrder.push_back (node->varId());
connectAllNeighbors (node);
}
return elimOrder;
}
void
ElimGraph::print (void) const
{
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < nodes_.size(); i++) {
2013-02-07 13:37:15 +00:00
std::cout << "node " << nodes_[i]->label() << " neighs:" ;
2012-05-23 14:56:01 +01:00
EGNeighs neighs = nodes_[i]->neighbors();
2012-05-24 22:55:20 +01:00
for (size_t j = 0; j < neighs.size(); j++) {
2013-02-07 13:37:15 +00:00
std::cout << " " << neighs[j]->label();
2012-05-23 14:56:01 +01:00
}
2013-02-07 13:37:15 +00:00
std::cout << std::endl;
2012-12-20 23:19:10 +00:00
}
2012-05-23 14:56:01 +01:00
}
void
ElimGraph::exportToGraphViz (
const char* fileName,
bool showNeighborless,
const VarIds& highlightVarIds) const
{
2013-02-07 13:37:15 +00:00
std::ofstream out (fileName);
2012-05-23 14:56:01 +01:00
if (!out.is_open()) {
2013-02-07 13:37:15 +00:00
std::cerr << "Error: couldn't open file '" << fileName << "'." ;
std::cerr << std::endl;
return;
2012-05-23 14:56:01 +01:00
}
2013-02-07 13:37:15 +00:00
out << "strict graph {" << std::endl;
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < nodes_.size(); i++) {
2012-12-27 12:54:58 +00:00
if (showNeighborless || nodes_[i]->neighbors().empty() == false) {
2013-02-07 13:37:15 +00:00
out << '"' << nodes_[i]->label() << '"' << std::endl;
2012-05-23 14:56:01 +01:00
}
}
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < highlightVarIds.size(); i++) {
2012-05-23 14:56:01 +01:00
EgNode* node =getEgNode (highlightVarIds[i]);
if (node) {
out << '"' << node->label() << '"' ;
2013-02-07 13:37:15 +00:00
out << " [shape=box3d]" << std::endl;
2012-05-23 14:56:01 +01:00
} else {
2013-02-07 13:37:15 +00:00
std::cerr << "Error: invalid variable id: " ;
std::cerr << highlightVarIds[i] << "." ;
std::cerr << std::endl;
exit (EXIT_FAILURE);
2012-05-23 14:56:01 +01:00
}
}
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < nodes_.size(); i++) {
2012-05-23 14:56:01 +01:00
EGNeighs neighs = nodes_[i]->neighbors();
2012-05-24 22:55:20 +01:00
for (size_t j = 0; j < neighs.size(); j++) {
2012-05-23 14:56:01 +01:00
out << '"' << nodes_[i]->label() << '"' << " -- " ;
2013-02-07 13:37:15 +00:00
out << '"' << neighs[j]->label() << '"' << std::endl;
2012-05-23 14:56:01 +01:00
}
}
2013-02-07 13:37:15 +00:00
out << "}" << std::endl;
2012-05-23 14:56:01 +01:00
out.close();
}
VarIds
ElimGraph::getEliminationOrder (
2012-05-29 17:12:57 +01:00
const Factors& factors,
2012-05-23 14:56:01 +01:00
VarIds excludedVids)
{
if (elimHeuristic_ == ElimHeuristic::sequentialEh) {
2012-05-29 17:12:57 +01:00
VarIds allVids;
Factors::const_iterator first = factors.begin();
Factors::const_iterator end = factors.end();
for (; first != end; ++first) {
Util::addToVector (allVids, (*first)->arguments());
2012-05-29 17:12:57 +01:00
}
TinySet<VarId> elimOrder (allVids);
2012-05-29 17:19:49 +01:00
elimOrder -= TinySet<VarId> (excludedVids);
2012-05-29 17:12:57 +01:00
return elimOrder.elements();
}
2012-05-23 14:56:01 +01:00
ElimGraph graph (factors);
return graph.getEliminatingOrder (excludedVids);
}
void
ElimGraph::addNode (EgNode* n)
{
nodes_.push_back (n);
n->setIndex (nodes_.size() - 1);
2013-02-07 13:37:15 +00:00
varMap_.insert (std::make_pair (n->varId(), n));
2012-05-23 14:56:01 +01:00
}
EgNode*
ElimGraph::getEgNode (VarId vid) const
{
2013-02-07 13:37:15 +00:00
std::unordered_map<VarId, EgNode*>::const_iterator it;
2012-05-23 14:56:01 +01:00
it = varMap_.find (vid);
return (it != varMap_.end()) ? it->second : 0;
}
EgNode*
ElimGraph::getLowestCostNode (void) const
{
EgNode* bestNode = 0;
unsigned minCost = Util::maxUnsigned();
2012-05-23 14:56:01 +01:00
EGNeighs::const_iterator it;
switch (elimHeuristic_) {
case minNeighborsEh: {
2012-05-23 14:56:01 +01:00
for (it = unmarked_.begin(); it != unmarked_.end(); ++ it) {
2012-07-02 22:53:44 +01:00
unsigned cost = getNeighborsCost (*it);
2012-05-23 14:56:01 +01:00
if (cost < minCost) {
bestNode = *it;
minCost = cost;
}
}}
break;
case minWeightEh: {
2012-07-02 22:53:44 +01:00
for (it = unmarked_.begin(); it != unmarked_.end(); ++ it) {
unsigned cost = getWeightCost (*it);
if (cost < minCost) {
bestNode = *it;
minCost = cost;
}
}}
2012-05-23 14:56:01 +01:00
break;
case minFillEh: {
2012-07-02 22:53:44 +01:00
for (it = unmarked_.begin(); it != unmarked_.end(); ++ it) {
unsigned cost = getFillCost (*it);
if (cost < minCost) {
bestNode = *it;
minCost = cost;
}
}}
2012-05-23 14:56:01 +01:00
break;
case weightedMinFillEh: {
2012-07-02 22:53:44 +01:00
for (it = unmarked_.begin(); it != unmarked_.end(); ++ it) {
unsigned cost = getWeightedFillCost (*it);
if (cost < minCost) {
bestNode = *it;
minCost = cost;
}
}}
2012-05-23 14:56:01 +01:00
break;
default:
assert (false);
}
assert (bestNode);
return bestNode;
}
void
ElimGraph::connectAllNeighbors (const EgNode* n)
{
const EGNeighs& neighs = n->neighbors();
if (neighs.size() > 0) {
2012-05-24 22:55:20 +01:00
for (size_t i = 0; i < neighs.size() - 1; i++) {
for (size_t j = i + 1; j < neighs.size(); j++) {
2012-12-27 12:54:58 +00:00
if (!neighbors (neighs[i], neighs[j])) {
2012-05-23 14:56:01 +01:00
addEdge (neighs[i], neighs[j]);
}
}
}
}
}
} // namespace Horus
2013-02-07 23:53:13 +00:00