diff --git a/gtsam/discrete/DiscreteMarginals.cpp b/gtsam/discrete/DiscreteMarginals.cpp index d7cc79a5c..74da37e73 100644 --- a/gtsam/discrete/DiscreteMarginals.cpp +++ b/gtsam/discrete/DiscreteMarginals.cpp @@ -55,4 +55,12 @@ Vector DiscreteMarginals::marginalProbabilities(const DiscreteKey& key) const { return vResult; } +/* ************************************************************************* */ +void DiscreteMarginals::print( + const std::string& s = "", + const KeyFormatter formatter = DefaultKeyFormatter) const { + std::cout << (s.empty() ? "Discrete Marginals of:" : s + " ") << std::endl; + bayesTree_.print("", formatter); +} + } /* namespace gtsam */ diff --git a/gtsam/discrete/DiscreteMarginals.h b/gtsam/discrete/DiscreteMarginals.h index fd6c93fe4..be2e17bcb 100644 --- a/gtsam/discrete/DiscreteMarginals.h +++ b/gtsam/discrete/DiscreteMarginals.h @@ -52,6 +52,10 @@ class DiscreteMarginals { * @return Vector of marginal probabilities */ Vector marginalProbabilities(const DiscreteKey& key) const; + + /// Print details + void print(const std::string& s = "", + const KeyFormatter formatter = DefaultKeyFormatter) const; }; } /* namespace gtsam */