94 lines
2.1 KiB
C++
94 lines
2.1 KiB
C++
/* ----------------------------------------------------------------------------
|
|
|
|
* 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
|
|
|
|
* -------------------------------------------------------------------------- */
|
|
|
|
/**
|
|
* @file Constraint.h
|
|
* @date May 15, 2012
|
|
* @author Frank Dellaert
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <gtsam_unstable/base/dllexport.h>
|
|
#include <gtsam/discrete/DiscreteFactor.h>
|
|
#include <boost/assign.hpp>
|
|
|
|
namespace gtsam {
|
|
|
|
class Domain;
|
|
|
|
/**
|
|
* Base class for discrete probabilistic factors
|
|
* The most general one is the derived DecisionTreeFactor
|
|
*/
|
|
class Constraint : public DiscreteFactor {
|
|
|
|
public:
|
|
|
|
typedef boost::shared_ptr<Constraint> shared_ptr;
|
|
|
|
protected:
|
|
|
|
/// Construct n-way factor
|
|
Constraint(const std::vector<Key>& js) :
|
|
DiscreteFactor(js) {
|
|
}
|
|
|
|
/// Construct unary factor
|
|
Constraint(Key j) :
|
|
DiscreteFactor(boost::assign::cref_list_of<1>(j)) {
|
|
}
|
|
|
|
/// Construct binary factor
|
|
Constraint(Key j1, Key j2) :
|
|
DiscreteFactor(boost::assign::cref_list_of<2>(j1)(j2)) {
|
|
}
|
|
|
|
/// construct from container
|
|
template<class KeyIterator>
|
|
Constraint(KeyIterator beginKey, KeyIterator endKey) :
|
|
DiscreteFactor(beginKey, endKey) {
|
|
}
|
|
|
|
public:
|
|
|
|
/// @name Standard Constructors
|
|
/// @{
|
|
|
|
/// Default constructor for I/O
|
|
Constraint();
|
|
|
|
/// Virtual destructor
|
|
virtual ~Constraint() {}
|
|
|
|
/// @}
|
|
/// @name Standard Interface
|
|
/// @{
|
|
|
|
/*
|
|
* Ensure Arc-consistency
|
|
* @param j domain to be checked
|
|
* @param domains all other domains
|
|
*/
|
|
virtual bool ensureArcConsistency(size_t j, std::vector<Domain>& domains) const = 0;
|
|
|
|
/// Partially apply known values
|
|
virtual shared_ptr partiallyApply(const Values&) const = 0;
|
|
|
|
|
|
/// Partially apply known values, domain version
|
|
virtual shared_ptr partiallyApply(const std::vector<Domain>&) const = 0;
|
|
/// @}
|
|
};
|
|
// DiscreteFactor
|
|
|
|
}// namespace gtsam
|