DiscreteBayesTree.h

Go to the documentation of this file.

/* ----------------------------------------------------------------------------
 
 * 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
 
 * -------------------------------------------------------------------------- */
 
#pragma once
 
#include <gtsam/discrete/DiscreteBayesNet.h>
#include <gtsam/discrete/DiscreteFactorGraph.h>
#include <gtsam/inference/BayesTree.h>
#include <gtsam/inference/Conditional.h>
#include <gtsam/inference/BayesTreeCliqueBase.h>
 
#include <string>
 
namespace gtsam {
 
// Forward declarations
class DiscreteConditional;
class VectorValues;
 
/* ************************************************************************* */
class GTSAM_EXPORT DiscreteBayesTreeClique
    : public BayesTreeCliqueBase<DiscreteBayesTreeClique, DiscreteFactorGraph> {
 public:
  typedef DiscreteBayesTreeClique This;
  typedef BayesTreeCliqueBase<DiscreteBayesTreeClique, DiscreteFactorGraph>
      Base;
  typedef boost::shared_ptr<This> shared_ptr;
  typedef boost::weak_ptr<This> weak_ptr;
  DiscreteBayesTreeClique() {}
  virtual ~DiscreteBayesTreeClique() {}
  DiscreteBayesTreeClique(
      const boost::shared_ptr<DiscreteConditional>& conditional)
      : Base(conditional) {}
 
  void printSignature(
      const std::string& s = "Clique: ",
      const KeyFormatter& formatter = DefaultKeyFormatter) const {
    conditional_->printSignature(s, formatter);
  }
 
  //** evaluate conditional probability of subtree for given Values */
  double evaluate(const DiscreteConditional::Values& values) const;
};
 
/* ************************************************************************* */
class GTSAM_EXPORT DiscreteBayesTree
    : public BayesTree<DiscreteBayesTreeClique> {
 private:
  typedef BayesTree<DiscreteBayesTreeClique> Base;
 
 public:
  typedef DiscreteBayesTree This;
  typedef boost::shared_ptr<This> shared_ptr;
 
  DiscreteBayesTree() {}
 
  bool equals(const This& other, double tol = 1e-9) const;
 
  //** evaluate probability for given Values */
  double evaluate(const DiscreteConditional::Values& values) const;
};
 
}  // namespace gtsam