56 lines
1.5 KiB
C
56 lines
1.5 KiB
C
|
/* ----------------------------------------------------------------------------
|
||
|
|
|
||
|
|
* GTSAM Copyright 2010, Georgia Tech Research Corporation,
|
||
|
|
* Atlanta, Georgia 30332-0415
|
||
|
|
* All Rights Reserved
|
||
|
|
* Authors: Frank Dellaert, et al. (see THANKS for the full author list)
|
||
|
|
|
||
|
|
* See LICENSE for the license information
|
||
|
|
|
||
|
|
* -------------------------------------------------------------------------- */
|
||
|
|
|
||
|
|
/**
|
||
|
|
* @file BayesNet.h
|
||
|
|
* @brief Bayes network
|
||
|
|
* @author Frank Dellaert
|
||
|
|
* @author Richard Roberts
|
||
|
|
*/
|
||
|
|
|
||
|
|
#pragma once
|
||
|
|
|
||
|
|
#include <gtsam/inference/FactorGraph-inst.h>
|
||
|
|
#include <gtsam/inference/BayesNet.h>
|
||
|
|
|
||
|
|
#include <boost/foreach.hpp>
|
||
|
|
#include <fstream>
|
||
|
|
|
||
|
|
namespace gtsam {
|
||
|
|
|
||
|
|
/* ************************************************************************* */
|
||
|
|
template<class CONDITIONAL>
|
||
|
|
void BayesNet<CONDITIONAL>::print(const std::string& s, const KeyFormatter& formatter) const
|
||
|
|
{
|
||
|
|
Base::print(s, formatter);
|
||
|
|
}
|
||
|
|
|
||
|
|
/* ************************************************************************* */
|
||
|
|
template<class CONDITIONAL>
|
||
|
|
void BayesNet<CONDITIONAL>::saveGraph(const std::string &s, const KeyFormatter& keyFormatter) const
|
||
|
|
{
|
||
|
|
std::ofstream of(s.c_str());
|
||
|
|
of << "digraph G{\n";
|
||
|
|
|
||
|
|
BOOST_REVERSE_FOREACH(const sharedConditional& conditional, *this) {
|
||
|
|
typename CONDITIONAL::Frontals frontals = conditional->frontals();
|
||
|
|
Key me = frontals.front();
|
||
|
|
typename CONDITIONAL::Parents parents = conditional->parents();
|
||
|
|
BOOST_FOREACH(Key p, parents)
|
||
|
|
of << p << "->" << me << std::endl;
|
||
|
|
}
|
||
|
|
|
||
|
|
of << "}";
|
||
|
|
of.close();
|
||
|
|
}
|
||
|
|
|
||
|
|
}
|