Moved to inst file
parent
b95bc712f4
commit
d34c1808a8
|
@ -123,6 +123,62 @@ void ClusterTree<BAYESTREE, GRAPH>::Cluster::print(const std::string& s,
|
||||||
PrintKeyVector(orderedFrontalKeys);
|
PrintKeyVector(orderedFrontalKeys);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
/* ************************************************************************* */
|
||||||
|
template<class BAYESTREE, class GRAPH>
|
||||||
|
void ClusterTree<BAYESTREE, GRAPH>::Cluster::mergeChildren(
|
||||||
|
const std::vector<bool>& merge) {
|
||||||
|
gttic(Cluster::mergeChildren);
|
||||||
|
size_t nrChildren = children.size();
|
||||||
|
|
||||||
|
// Count how many keys, factors and children we'll end up with
|
||||||
|
size_t nrKeys = orderedFrontalKeys.size();
|
||||||
|
size_t nrFactors = factors.size();
|
||||||
|
size_t nrNewChildren = 0;
|
||||||
|
// Loop over children
|
||||||
|
for (size_t i = 0; i < nrChildren; ++i) {
|
||||||
|
if (merge[i]) {
|
||||||
|
// Get a reference to the i, adjusting the index to account for children
|
||||||
|
// previously merged and removed from the i list.
|
||||||
|
sharedNode child = children[i];
|
||||||
|
nrKeys += child->orderedFrontalKeys.size();
|
||||||
|
nrFactors += child->factors.size();
|
||||||
|
nrNewChildren += child->children.size();
|
||||||
|
} else {
|
||||||
|
nrNewChildren += 1; // we keep the child
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// now reserve space, and really merge
|
||||||
|
orderedFrontalKeys.reserve(nrKeys);
|
||||||
|
factors.reserve(nrFactors);
|
||||||
|
typename Node::Children newChildren;
|
||||||
|
newChildren.reserve(nrNewChildren);
|
||||||
|
// Loop over newChildren
|
||||||
|
for (size_t i = 0; i < nrChildren; ++i) {
|
||||||
|
// Check if we should merge the i^th child
|
||||||
|
sharedNode child = children[i];
|
||||||
|
if (merge[i]) {
|
||||||
|
// Get a reference to the i, adjusting the index to account for newChildren
|
||||||
|
// previously merged and removed from the i list.
|
||||||
|
// Merge keys. For efficiency, we add keys in reverse order at end, calling reverse after..
|
||||||
|
orderedFrontalKeys.insert(orderedFrontalKeys.end(),
|
||||||
|
child->orderedFrontalKeys.rbegin(), child->orderedFrontalKeys.rend());
|
||||||
|
// Merge keys, factors, and children.
|
||||||
|
factors.insert(factors.end(), child->factors.begin(),
|
||||||
|
child->factors.end());
|
||||||
|
newChildren.insert(newChildren.end(), child->children.begin(),
|
||||||
|
child->children.end());
|
||||||
|
// Increment problem size
|
||||||
|
problemSize_ = std::max(problemSize_, child->problemSize_);
|
||||||
|
// Increment number of frontal variables
|
||||||
|
} else {
|
||||||
|
newChildren.push_back(child); // we keep the child
|
||||||
|
}
|
||||||
|
}
|
||||||
|
children = newChildren;
|
||||||
|
std::reverse(orderedFrontalKeys.begin(), orderedFrontalKeys.end());
|
||||||
|
}
|
||||||
|
|
||||||
/* ************************************************************************* */
|
/* ************************************************************************* */
|
||||||
template<class BAYESTREE, class GRAPH>
|
template<class BAYESTREE, class GRAPH>
|
||||||
void ClusterTree<BAYESTREE, GRAPH>::print(const std::string& s,
|
void ClusterTree<BAYESTREE, GRAPH>::print(const std::string& s,
|
||||||
|
|
|
@ -55,64 +55,12 @@ public:
|
||||||
return problemSize_;
|
return problemSize_;
|
||||||
}
|
}
|
||||||
|
|
||||||
/** print this node */
|
/// print this node
|
||||||
void print(const std::string& s = "", const KeyFormatter& keyFormatter =
|
void print(const std::string& s = "", const KeyFormatter& keyFormatter =
|
||||||
DefaultKeyFormatter) const;
|
DefaultKeyFormatter) const;
|
||||||
|
|
||||||
void mergeChildren(const std::vector<bool>& merge) {
|
/// Merge all children for which bit is set into this node
|
||||||
gttic(merge_children);
|
void mergeChildren(const std::vector<bool>& merge);
|
||||||
size_t nrChildren = children.size();
|
|
||||||
|
|
||||||
// Count how many keys, factors and children we'll end up with
|
|
||||||
size_t nrKeys = this->orderedFrontalKeys.size();
|
|
||||||
size_t nrFactors = this->factors.size();
|
|
||||||
size_t nrNewChildren = 0;
|
|
||||||
// Loop over children
|
|
||||||
for (size_t i = 0; i < nrChildren; ++i) {
|
|
||||||
if (merge[i]) {
|
|
||||||
// Get a reference to the i, adjusting the index to account for children
|
|
||||||
// previously merged and removed from the i list.
|
|
||||||
sharedNode child = this->children[i];
|
|
||||||
nrKeys += child->orderedFrontalKeys.size();
|
|
||||||
nrFactors += child->factors.size();
|
|
||||||
nrNewChildren += child->children.size();
|
|
||||||
} else {
|
|
||||||
nrNewChildren += 1; // we keep the child
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// now reserve space, and really merge
|
|
||||||
this->orderedFrontalKeys.reserve(nrKeys);
|
|
||||||
this->factors.reserve(nrFactors);
|
|
||||||
typename Node::Children newChildren;
|
|
||||||
newChildren.reserve(nrNewChildren);
|
|
||||||
// Loop over newChildren
|
|
||||||
for (size_t i = 0; i < nrChildren; ++i) {
|
|
||||||
// Check if we should merge the i^th child
|
|
||||||
sharedNode child = this->children[i];
|
|
||||||
if (merge[i]) {
|
|
||||||
// Get a reference to the i, adjusting the index to account for newChildren
|
|
||||||
// previously merged and removed from the i list.
|
|
||||||
// Merge keys. For efficiency, we add keys in reverse order at end, calling reverse after..
|
|
||||||
this->orderedFrontalKeys.insert(this->orderedFrontalKeys.end(),
|
|
||||||
child->orderedFrontalKeys.rbegin(),
|
|
||||||
child->orderedFrontalKeys.rend());
|
|
||||||
// Merge keys, factors, and children.
|
|
||||||
this->factors.insert(this->factors.end(), child->factors.begin(),
|
|
||||||
child->factors.end());
|
|
||||||
newChildren.insert(newChildren.end(), child->children.begin(),
|
|
||||||
child->children.end());
|
|
||||||
// Increment problem size
|
|
||||||
problemSize_ = std::max(problemSize_, child->problemSize_);
|
|
||||||
// Increment number of frontal variables
|
|
||||||
} else {
|
|
||||||
newChildren.push_back(child); // we keep the child
|
|
||||||
}
|
|
||||||
}
|
|
||||||
this->children = newChildren;
|
|
||||||
std::reverse(this->orderedFrontalKeys.begin(),
|
|
||||||
this->orderedFrontalKeys.end());
|
|
||||||
}
|
|
||||||
};
|
};
|
||||||
|
|
||||||
typedef boost::shared_ptr<Cluster> sharedCluster; ///< Shared pointer to Cluster
|
typedef boost::shared_ptr<Cluster> sharedCluster; ///< Shared pointer to Cluster
|
||||||
|
|
Loading…
Reference in New Issue