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/BayesTreeCliqueBase.h>

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() {}
    DiscreteBayesTreeClique(const boost::shared_ptr<DiscreteConditional>& conditional) : Base(conditional) {}
  };

  /* ************************************************************************* */
  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;
  };

}