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