Loading...
Searching...
No Matches
ChangeBackground.h
Go to the documentation of this file.
// Copyright Contributors to the OpenVDB Project
// SPDX-License-Identifier: MPL-2.0
//
/// @file ChangeBackground.h
///
/// @brief Efficient multi-threaded replacement of the background
/// values in tree.
///
/// @author Ken Museth
#ifndef OPENVDB_TOOLS_ChangeBACKGROUND_HAS_BEEN_INCLUDED
#define OPENVDB_TOOLS_ChangeBACKGROUND_HAS_BEEN_INCLUDED
#include <openvdb/math/Math.h> // for isNegative and negative
#include <openvdb/Types.h> // for Index typedef
#include <openvdb/tree/NodeManager.h>
#include <openvdb/openvdb.h>
namespace openvdb {
OPENVDB_USE_VERSION_NAMESPACE
namespace OPENVDB_VERSION_NAME {
namespace tools {
/// @brief Replace the background value in all the nodes of a tree.
/// @details The sign of the background value is preserved, and only
/// inactive values equal to the old background value are replaced.
///
/// @note If a LeafManager is used the cached leaf nodes are reused,
/// resulting in slightly better overall performance.
///
/// @param tree Tree (or LeafManager) that will have its background value changed
/// @param background the new background value
/// @param threaded enable or disable threading (threading is enabled by default)
/// @param grainSize used to control the threading granularity (default is 32)
template<typename TreeOrLeafManagerT>
void
changeBackground(
TreeOrLeafManagerT& tree,
const typename TreeOrLeafManagerT::ValueType& background,
bool threaded = true,
size_t grainSize = 32);
/// @brief Replace the background value in all the nodes of a floating-point tree
/// containing a symmetric narrow-band level set.
/// @details All inactive values will be set to +| @a halfWidth | if outside
/// and -| @a halfWidth | if inside, where @a halfWidth is half the width
/// of the symmetric narrow band.
///
/// @note This method is faster than changeBackground since it does not
/// perform tests to see if inactive values are equal to the old background value.
/// @note If a LeafManager is used the cached leaf nodes are reused,
/// resulting in slightly better overall performance.
///
/// @param tree Tree (or LeafManager) that will have its background value changed
/// @param halfWidth half of the width of the symmetric narrow band
/// @param threaded enable or disable threading (threading is enabled by default)
/// @param grainSize used to control the threading granularity (default is 32)
///
/// @throw ValueError if @a halfWidth is negative (as defined by math::isNegative)
template<typename TreeOrLeafManagerT>
void
changeLevelSetBackground(
TreeOrLeafManagerT& tree,
const typename TreeOrLeafManagerT::ValueType& halfWidth,
bool threaded = true,
size_t grainSize = 32);
/// @brief Replace the background values in all the nodes of a floating-point tree
/// containing a possibly asymmetric narrow-band level set.
/// @details All inactive values will be set to +| @a outsideWidth | if outside
/// and -| @a insideWidth | if inside, where @a outsideWidth is the outside
/// width of the narrow band and @a insideWidth is its inside width.
///
/// @note This method is faster than changeBackground since it does not
/// perform tests to see if inactive values are equal to the old background value.
/// @note If a LeafManager is used the cached leaf nodes are reused,
/// resulting in slightly better overall performance.
///
/// @param tree Tree (or LeafManager) that will have its background value changed
/// @param outsideWidth The width of the outside of the narrow band
/// @param insideWidth The width of the inside of the narrow band
/// @param threaded enable or disable threading (threading is enabled by default)
/// @param grainSize used to control the threading granularity (default is 32)
///
/// @throw ValueError if @a outsideWidth is negative or @a insideWidth is
/// not negative (as defined by math::isNegative)
template<typename TreeOrLeafManagerT>
void
changeAsymmetricLevelSetBackground(
TreeOrLeafManagerT& tree,
const typename TreeOrLeafManagerT::ValueType& outsideWidth,
const typename TreeOrLeafManagerT::ValueType& insideWidth,
bool threaded = true,
size_t grainSize = 32);
//////////////////////////////////////////////////////
// Replaces the background value in a Tree of any type.
template<typename TreeOrLeafManagerT>
class ChangeBackgroundOp
{
public:
typedef typename TreeOrLeafManagerT::ValueType ValueT;
typedef typename TreeOrLeafManagerT::RootNodeType RootT;
typedef typename TreeOrLeafManagerT::LeafNodeType LeafT;
ChangeBackgroundOp(const TreeOrLeafManagerT& tree, const ValueT& newValue)
: mOldValue(tree.root().background())
, mNewValue(newValue)
{
}
void operator()(RootT& root) const
{
for (typename RootT::ValueOffIter it = root.beginValueOff(); it; ++it) this->set(it);
root.setBackground(mNewValue, false);
}
void operator()(LeafT& node) const
{
for (typename LeafT::ValueOffIter it = node.beginValueOff(); it; ++it) this->set(it);
}
template<typename NodeT>
void operator()(NodeT& node) const
{
typename NodeT::NodeMaskType mask = node.getValueOffMask();
for (typename NodeT::ValueOnIter it(mask.beginOn(), &node); it; ++it) this->set(it);
}
private:
template<typename IterT>
inline void set(IterT& iter) const
{
if (math::isApproxEqual(*iter, mOldValue)) {
iter.setValue(mNewValue);
} else if (math::isApproxEqual(*iter, math::negative(mOldValue))) {
iter.setValue(math::negative(mNewValue));
}
}
const ValueT mOldValue, mNewValue;
};// ChangeBackgroundOp
// Replaces the background value in a Tree assumed to represent a
// level set. It is generally faster than ChangeBackgroundOp.
// Note that is follows the sign-convention that outside is positive
// and inside is negative!
template<typename TreeOrLeafManagerT>
class ChangeLevelSetBackgroundOp
{
public:
typedef typename TreeOrLeafManagerT::ValueType ValueT;
typedef typename TreeOrLeafManagerT::RootNodeType RootT;
typedef typename TreeOrLeafManagerT::LeafNodeType LeafT;
/// @brief Constructor for asymmetric narrow-bands
ChangeLevelSetBackgroundOp(const ValueT& outside, const ValueT& inside)
: mOutside(outside)
, mInside(inside)
{
if (math::isNegative(mOutside)) {
OPENVDB_THROW(ValueError,
"ChangeLevelSetBackgroundOp: the outside value cannot be negative!");
}
if (!math::isNegative(mInside)) {
OPENVDB_THROW(ValueError,
"ChangeLevelSetBackgroundOp: the inside value must be negative!");
}
}
void operator()(RootT& root) const
{
for (typename RootT::ValueOffIter it = root.beginValueOff(); it; ++it) this->set(it);
root.setBackground(mOutside, false);
}
void operator()(LeafT& node) const
{
for(typename LeafT::ValueOffIter it = node.beginValueOff(); it; ++it) this->set(it);
}
template<typename NodeT>
void operator()(NodeT& node) const
{
typedef typename NodeT::ValueOffIter IterT;
for (IterT it(node.getChildMask().beginOff(), &node); it; ++it) this->set(it);
}
private:
template<typename IterT>
inline void set(IterT& iter) const
{
//this is safe since we know ValueType is_floating_point
ValueT& v = const_cast<ValueT&>(*iter);
v = v < 0 ? mInside : mOutside;
}
const ValueT mOutside, mInside;
};// ChangeLevelSetBackgroundOp
template<typename TreeOrLeafManagerT>
void
changeBackground(
TreeOrLeafManagerT& tree,
const typename TreeOrLeafManagerT::ValueType& background,
bool threaded,
size_t grainSize)
{
tree::NodeManager<TreeOrLeafManagerT> linearTree(tree);
ChangeBackgroundOp<TreeOrLeafManagerT> op(tree, background);
linearTree.foreachTopDown(op, threaded, grainSize);
}
template<typename TreeOrLeafManagerT>
void
changeAsymmetricLevelSetBackground(
TreeOrLeafManagerT& tree,
const typename TreeOrLeafManagerT::ValueType& outsideValue,
const typename TreeOrLeafManagerT::ValueType& insideValue,
bool threaded,
size_t grainSize)
{
tree::NodeManager<TreeOrLeafManagerT> linearTree(tree);
ChangeLevelSetBackgroundOp<TreeOrLeafManagerT> op(outsideValue, insideValue);
linearTree.foreachTopDown(op, threaded, grainSize);
}
// If the narrow-band is symmetric only one background value is required
template<typename TreeOrLeafManagerT>
void
changeLevelSetBackground(
TreeOrLeafManagerT& tree,
const typename TreeOrLeafManagerT::ValueType& background,
bool threaded,
size_t grainSize)
{
changeAsymmetricLevelSetBackground(
tree, background, math::negative(background), threaded, grainSize);
}
////////////////////////////////////////
// Explicit Template Instantiation
#ifdef OPENVDB_USE_EXPLICIT_INSTANTIATION
#ifdef OPENVDB_INSTANTIATE_CHANGEBACKGROUND
#include <openvdb/util/ExplicitInstantiation.h>
#endif
#define _FUNCTION(TreeT) \
void changeBackground(TreeT&, const TreeT::ValueType&, bool, size_t)
OPENVDB_VOLUME_TREE_INSTANTIATE(_FUNCTION)
#undef _FUNCTION
#define _FUNCTION(TreeT) \
void changeBackground(tree::LeafManager<TreeT>&, const TreeT::ValueType&, bool, size_t)
OPENVDB_VOLUME_TREE_INSTANTIATE(_FUNCTION)
#undef _FUNCTION
#define _FUNCTION(TreeT) \
void changeLevelSetBackground(TreeT&, const TreeT::ValueType&, bool, size_t)
OPENVDB_REAL_TREE_INSTANTIATE(_FUNCTION)
#undef _FUNCTION
#define _FUNCTION(TreeT) \
void changeLevelSetBackground(tree::LeafManager<TreeT>&, const TreeT::ValueType&, bool, size_t)
OPENVDB_REAL_TREE_INSTANTIATE(_FUNCTION)
#undef _FUNCTION
#define _FUNCTION(TreeT) \
void changeAsymmetricLevelSetBackground(TreeT&, const TreeT::ValueType&, const TreeT::ValueType&, bool, size_t)
OPENVDB_REAL_TREE_INSTANTIATE(_FUNCTION)
#undef _FUNCTION
#define _FUNCTION(TreeT) \
void changeAsymmetricLevelSetBackground(tree::LeafManager<TreeT>&, const TreeT::ValueType&, const TreeT::ValueType&, bool, size_t)
OPENVDB_REAL_TREE_INSTANTIATE(_FUNCTION)
#undef _FUNCTION
#endif // OPENVDB_USE_EXPLICIT_INSTANTIATION
} // namespace tools
} // namespace OPENVDB_VERSION_NAME
} // namespace openvdb
#endif // OPENVDB_TOOLS_CHANGEBACKGROUND_HAS_BEEN_INCLUDED
ExplicitInstantiation.h
Math.h
General-purpose arithmetic and comparison routines, most of which accept arbitrary value types (or at...
openvdb::v10_0::ValueError
Definition Exceptions.h:65
openvdb::v10_0::tools::ChangeBackgroundOp
Definition ChangeBackground.h:106
openvdb::v10_0::tools::ChangeBackgroundOp::ChangeBackgroundOp
ChangeBackgroundOp(const TreeOrLeafManagerT &tree, const ValueT &newValue)
Definition ChangeBackground.h:113
openvdb::v10_0::tools::ChangeBackgroundOp::operator()
void operator()(NodeT &node) const
Definition ChangeBackground.h:128
openvdb::v10_0::tools::ChangeBackgroundOp::operator()
void operator()(RootT &root) const
Definition ChangeBackground.h:118
openvdb::v10_0::tools::ChangeBackgroundOp::ValueT
TreeOrLeafManagerT::ValueType ValueT
Definition ChangeBackground.h:108
openvdb::v10_0::tools::ChangeBackgroundOp::LeafT
TreeOrLeafManagerT::LeafNodeType LeafT
Definition ChangeBackground.h:110
openvdb::v10_0::tools::ChangeBackgroundOp::operator()
void operator()(LeafT &node) const
Definition ChangeBackground.h:123
openvdb::v10_0::tools::ChangeBackgroundOp::RootT
TreeOrLeafManagerT::RootNodeType RootT
Definition ChangeBackground.h:109
openvdb::v10_0::tools::ChangeLevelSetBackgroundOp
Definition ChangeBackground.h:154
openvdb::v10_0::tools::ChangeLevelSetBackgroundOp::operator()
void operator()(NodeT &node) const
Definition ChangeBackground.h:184
openvdb::v10_0::tools::ChangeLevelSetBackgroundOp::operator()
void operator()(RootT &root) const
Definition ChangeBackground.h:174
openvdb::v10_0::tools::ChangeLevelSetBackgroundOp::ValueT
TreeOrLeafManagerT::ValueType ValueT
Definition ChangeBackground.h:156
openvdb::v10_0::tools::ChangeLevelSetBackgroundOp::ChangeLevelSetBackgroundOp
ChangeLevelSetBackgroundOp(const ValueT &outside, const ValueT &inside)
Constructor for asymmetric narrow-bands.
Definition ChangeBackground.h:161
openvdb::v10_0::tools::ChangeLevelSetBackgroundOp::LeafT
TreeOrLeafManagerT::LeafNodeType LeafT
Definition ChangeBackground.h:158
openvdb::v10_0::tools::ChangeLevelSetBackgroundOp::operator()
void operator()(LeafT &node) const
Definition ChangeBackground.h:179
openvdb::v10_0::tools::ChangeLevelSetBackgroundOp::RootT
TreeOrLeafManagerT::RootNodeType RootT
Definition ChangeBackground.h:157
openvdb::v10_0::tree::NodeManager
To facilitate threading over the nodes of a tree, cache node pointers in linear arrays,...
Definition NodeManager.h:531
openvdb::v10_0::tree::NodeManager::foreachTopDown
void foreachTopDown(const NodeOp &op, bool threaded=true, size_t grainSize=1)
Definition NodeManager.h:631
openvdb::v10_0::math::isApproxEqual
bool isApproxEqual(const Type &a, const Type &b, const Type &tolerance)
Return true if a is equal to b to within the given tolerance.
Definition Math.h:406
openvdb::v10_0::math::isNegative
bool isNegative(const Type &x)
Return true if x is less than zero.
Definition Math.h:367
openvdb::v10_0::math::negative
T negative(const T &val)
Return the unary negation of the given value.
Definition Math.h:128
openvdb::v10_0::tools::changeAsymmetricLevelSetBackground
void changeAsymmetricLevelSetBackground(TreeOrLeafManagerT &tree, const typename TreeOrLeafManagerT::ValueType &outsideWidth, const typename TreeOrLeafManagerT::ValueType &insideWidth, bool threaded=true, size_t grainSize=32)
Replace the background values in all the nodes of a floating-point tree containing a possibly asymmet...
Definition ChangeBackground.h:218
openvdb::v10_0::tools::changeBackground
void changeBackground(TreeOrLeafManagerT &tree, const typename TreeOrLeafManagerT::ValueType &background, bool threaded=true, size_t grainSize=32)
Replace the background value in all the nodes of a tree.
Definition ChangeBackground.h:204
openvdb::v10_0::tools::changeLevelSetBackground
void changeLevelSetBackground(TreeOrLeafManagerT &tree, const typename TreeOrLeafManagerT::ValueType &halfWidth, bool threaded=true, size_t grainSize=32)
Replace the background value in all the nodes of a floating-point tree containing a symmetric narrow-...
Definition ChangeBackground.h:234
openvdb::v10_0::tree
Definition PointDataGrid.h:169
openvdb
Definition Exceptions.h:13
OPENVDB_THROW
#define OPENVDB_THROW(exception, message)
Definition Exceptions.h:74
Types.h
NodeManager.h
NodeManager produces linear arrays of all tree nodes allowing for efficient threading and bottom-up p...
openvdb.h
OPENVDB_VERSION_NAME
#define OPENVDB_VERSION_NAME
The version namespace name for this library version.
Definition version.h.in:121
OPENVDB_USE_VERSION_NAMESPACE
#define OPENVDB_USE_VERSION_NAMESPACE
Definition version.h.in:212
OPENVDB_REAL_TREE_INSTANTIATE
#define OPENVDB_REAL_TREE_INSTANTIATE(Function)
Definition version.h.in:157
OPENVDB_VOLUME_TREE_INSTANTIATE
#define OPENVDB_VOLUME_TREE_INSTANTIATE(Function)
Definition version.h.in:160