Better print

release/4.3a0
Frank dellaert 2020-07-11 13:16:35 -04:00
parent 7dba3023d6
commit 550dc377e3
1 changed files with 16 additions and 4 deletions

View File

@ -27,6 +27,7 @@
#include <algorithm> #include <algorithm>
#include <random> #include <random>
#include <stdexcept> #include <stdexcept>
#include <string>
#include <vector> #include <vector>
using namespace std; using namespace std;
@ -66,10 +67,21 @@ DiscreteConditional::DiscreteConditional(const Signature& signature)
BaseConditional(1) {} BaseConditional(1) {}
/* ******************************************************************************** */ /* ******************************************************************************** */
void DiscreteConditional::print(const std::string& s, void DiscreteConditional::print(const string& s,
const KeyFormatter& formatter) const { const KeyFormatter& formatter) const {
std::cout << s << std::endl; cout << s << " P( ";
Potentials::print(s); for (const_iterator it = beginFrontals(); it != endFrontals(); ++it) {
cout << formatter(*it) << " ";
}
if (nrParents()) {
cout << "| ";
for (const_iterator it = beginParents(); it != endParents(); ++it) {
cout << formatter(*it) << " ";
}
}
cout << ")";
Potentials::print("");
cout << endl;
} }
/* ******************************************************************************** */ /* ******************************************************************************** */