127 lines
4.2 KiB
C++
127 lines
4.2 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 HybridGaussianISAM.h
|
|
* @date March 31, 2022
|
|
* @author Fan Jiang
|
|
* @author Frank Dellaert
|
|
* @author Varun Agrawal
|
|
*/
|
|
|
|
#include <gtsam/base/treeTraversal-inst.h>
|
|
#include <gtsam/hybrid/HybridBayesTree.h>
|
|
#include <gtsam/hybrid/HybridGaussianFactorGraph.h>
|
|
#include <gtsam/hybrid/HybridGaussianISAM.h>
|
|
#include <gtsam/inference/ISAM-inst.h>
|
|
#include <gtsam/inference/Key.h>
|
|
|
|
#include <iterator>
|
|
|
|
namespace gtsam {
|
|
|
|
// Instantiate base class
|
|
// template class ISAM<HybridBayesTree>;
|
|
|
|
/* ************************************************************************* */
|
|
HybridGaussianISAM::HybridGaussianISAM() {}
|
|
|
|
/* ************************************************************************* */
|
|
HybridGaussianISAM::HybridGaussianISAM(const HybridBayesTree& bayesTree)
|
|
: Base(bayesTree) {}
|
|
|
|
/* ************************************************************************* */
|
|
Ordering HybridGaussianISAM::GetOrdering(
|
|
HybridGaussianFactorGraph& factors,
|
|
const HybridGaussianFactorGraph& newFactors) {
|
|
// Get all the discrete keys from the factors
|
|
const KeySet allDiscrete = factors.discreteKeySet();
|
|
|
|
// Create KeyVector with continuous keys followed by discrete keys.
|
|
KeyVector newKeysDiscreteLast;
|
|
const KeySet newFactorKeys = newFactors.keys();
|
|
// Insert continuous keys first.
|
|
for (auto& k : newFactorKeys) {
|
|
if (!allDiscrete.exists(k)) {
|
|
newKeysDiscreteLast.push_back(k);
|
|
}
|
|
}
|
|
// Insert discrete keys at the end
|
|
std::copy(allDiscrete.begin(), allDiscrete.end(),
|
|
std::back_inserter(newKeysDiscreteLast));
|
|
|
|
const VariableIndex index(factors);
|
|
|
|
// Get an ordering where the new keys are eliminated last
|
|
Ordering ordering = Ordering::ColamdConstrainedLast(
|
|
index, KeyVector(newKeysDiscreteLast.begin(), newKeysDiscreteLast.end()),
|
|
true);
|
|
return ordering;
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void HybridGaussianISAM::updateInternal(
|
|
const HybridGaussianFactorGraph& newFactors,
|
|
HybridBayesTree::Cliques* orphans,
|
|
const std::optional<size_t>& maxNrLeaves,
|
|
const std::optional<Ordering>& ordering,
|
|
const HybridBayesTree::Eliminate& function) {
|
|
// Remove the contaminated part of the Bayes tree
|
|
BayesNetType bn;
|
|
const KeySet newFactorKeys = newFactors.keys();
|
|
if (!this->empty()) {
|
|
KeyVector keyVector(newFactorKeys.begin(), newFactorKeys.end());
|
|
this->removeTop(keyVector, &bn, orphans);
|
|
}
|
|
|
|
// Add the removed top and the new factors
|
|
HybridGaussianFactorGraph factors;
|
|
factors += bn;
|
|
factors += newFactors;
|
|
|
|
// Add the orphaned subtrees
|
|
for (const sharedClique& orphan : *orphans) {
|
|
factors += std::make_shared<BayesTreeOrphanWrapper<Node>>(orphan);
|
|
}
|
|
|
|
const VariableIndex index(factors);
|
|
Ordering elimination_ordering;
|
|
if (ordering) {
|
|
elimination_ordering = *ordering;
|
|
} else {
|
|
elimination_ordering = GetOrdering(factors, newFactors);
|
|
}
|
|
|
|
// eliminate all factors (top, added, orphans) into a new Bayes tree
|
|
HybridBayesTree::shared_ptr bayesTree =
|
|
factors.eliminateMultifrontal(elimination_ordering, function, std::cref(index));
|
|
|
|
if (maxNrLeaves) {
|
|
bayesTree->prune(*maxNrLeaves);
|
|
}
|
|
|
|
// Re-add into Bayes tree data structures
|
|
this->roots_.insert(this->roots_.end(), bayesTree->roots().begin(),
|
|
bayesTree->roots().end());
|
|
this->nodes_.insert(bayesTree->nodes().begin(), bayesTree->nodes().end());
|
|
}
|
|
|
|
/* ************************************************************************* */
|
|
void HybridGaussianISAM::update(const HybridGaussianFactorGraph& newFactors,
|
|
const std::optional<size_t>& maxNrLeaves,
|
|
const std::optional<Ordering>& ordering,
|
|
const HybridBayesTree::Eliminate& function) {
|
|
Cliques orphans;
|
|
this->updateInternal(newFactors, &orphans, maxNrLeaves, ordering, function);
|
|
}
|
|
|
|
} // namespace gtsam
|