29#ifndef OPENVDB_TOOLS_VALUETRANSFORMER_HAS_BEEN_INCLUDED
30#define OPENVDB_TOOLS_VALUETRANSFORMER_HAS_BEEN_INCLUDED
33#include <tbb/parallel_for.h>
34#include <tbb/parallel_reduce.h>
88template<
typename IterT,
typename XformOp>
89inline void foreach(
const IterT& iter, XformOp& op,
90 bool threaded =
true,
bool shareOp =
true);
92template<
typename IterT,
typename XformOp>
93inline void foreach(
const IterT& iter,
const XformOp& op,
94 bool threaded =
true,
bool shareOp =
true);
137template<
typename InIterT,
typename OutGr
idT,
typename XformOp>
139 XformOp& op,
bool threaded =
true,
bool shareOp =
true,
140 MergePolicy merge = MERGE_ACTIVE_STATES);
142template<
typename InIterT,
typename OutGr
idT,
typename XformOp>
144 const XformOp& op,
bool threaded =
true,
bool shareOp =
true,
145 MergePolicy merge = MERGE_ACTIVE_STATES);
192template<
typename IterT,
typename XformOp>
193inline void accumulate(
const IterT& iter, XformOp& op,
bool threaded =
true);
201template<
typename TreeT>
202void setValueOnMin(TreeT& tree,
const Coord& xyz,
const typename TreeT::ValueType&
value);
209template<
typename TreeT>
210void setValueOnMax(TreeT& tree,
const Coord& xyz,
const typename TreeT::ValueType&
value);
217template<
typename TreeT>
218void setValueOnSum(TreeT& tree,
const Coord& xyz,
const typename TreeT::ValueType&
value);
225template<
typename TreeT>
234template<
typename ValueType>
237 MinOp(
const ValueType& v): val(v) {}
238 inline void operator()(ValueType& v)
const { v = std::min<ValueType>(v, val); }
241template<
typename ValueType>
244 MaxOp(
const ValueType& v): val(v) {}
245 inline void operator()(ValueType& v)
const { v = std::max<ValueType>(v, val); }
248template<
typename ValueType>
251 SumOp(
const ValueType& v): val(v) {}
263template<
typename ValueType>
281template<
typename TreeT>
289template<
typename TreeT>
297template<
typename TreeT>
305template<
typename TreeT>
318template<
typename IterT,
typename OpT>
330 tbb::parallel_for(range, *
this);
344template<
typename IterT,
typename OpT>
355 mIter(other.mIter), mOp(*other.mOrigOp), mOrigOp(other.mOrigOp) {}
361 tbb::parallel_for(range, *
this);
372 OpT
const *
const mOrigOp;
378template<
typename IterT,
typename XformOp>
380foreach(
const IterT& iter, XformOp& op,
bool threaded,
bool shared)
387 Processor proc(iter, op);
388 proc.process(threaded);
392template<
typename IterT,
typename XformOp>
394foreach(
const IterT& iter,
const XformOp& op,
bool threaded,
bool )
407template<
typename InIterT,
typename OutTreeT,
typename OpT>
408class SharedOpTransformer
411 using InTreeT =
typename InIterT::TreeT;
413 using OutValueT =
typename OutTreeT::ValueType;
415 SharedOpTransformer(
const InIterT& inIter, OutTreeT& outTree, OpT& op,
MergePolicy merge):
418 mInputTree(inIter.getTree()),
419 mOutputTree(&outTree),
423 if (
static_cast<const void*
>(mInputTree) ==
static_cast<void*
>(mOutputTree)) {
425 " to transform a grid in place");
430 SharedOpTransformer(SharedOpTransformer& other, tbb::split):
432 mInputIter(other.mInputIter),
433 mInputTree(other.mInputTree),
434 mOutputTree(new OutTreeT(zeroVal<OutValueT>())),
436 mMergePolicy(other.mMergePolicy)
439 ~SharedOpTransformer()
445 mOutputTree =
nullptr;
449 void process(
bool threaded =
true)
451 if (!mInputTree || !mOutputTree)
return;
453 IterRange range(mInputIter);
458 tbb::parallel_reduce(range, *
this);
465 void operator()(IterRange& range)
const
467 if (!mOutputTree)
return;
468 typename tree::ValueAccessor<OutTreeT> outAccessor(*mOutputTree);
469 for ( ; range; ++range) {
470 mOp(range.iterator(), outAccessor);
474 void join(
const SharedOpTransformer& other)
476 if (mOutputTree && other.mOutputTree) {
477 mOutputTree->merge(*other.mOutputTree, mMergePolicy);
484 const InTreeT* mInputTree;
485 OutTreeT* mOutputTree;
491template<
typename InIterT,
typename OutTreeT,
typename OpT>
492class CopyableOpTransformer
495 using InTreeT =
typename InIterT::TreeT;
496 using IterRange =
typename tree::IteratorRange<InIterT>;
497 using OutValueT =
typename OutTreeT::ValueType;
499 CopyableOpTransformer(
const InIterT& inIter, OutTreeT& outTree,
500 const OpT& op, MergePolicy merge):
503 mInputTree(inIter.getTree()),
504 mOutputTree(&outTree),
509 if (
static_cast<const void*
>(mInputTree) ==
static_cast<void*
>(mOutputTree)) {
511 " to transform a grid in place");
517 CopyableOpTransformer(CopyableOpTransformer& other, tbb::split):
519 mInputIter(other.mInputIter),
520 mInputTree(other.mInputTree),
521 mOutputTree(new OutTreeT(
zeroVal<OutValueT>())),
523 mOrigOp(other.mOrigOp),
524 mMergePolicy(other.mMergePolicy)
527 ~CopyableOpTransformer()
533 mOutputTree =
nullptr;
537 void process(
bool threaded =
true)
539 if (!mInputTree || !mOutputTree)
return;
541 IterRange range(mInputIter);
546 tbb::parallel_reduce(range, *
this);
553 void operator()(IterRange& range)
555 if (!mOutputTree)
return;
556 typename tree::ValueAccessor<OutTreeT> outAccessor(*mOutputTree);
557 for ( ; range; ++range) {
558 mOp(range.iterator(), outAccessor);
562 void join(
const CopyableOpTransformer& other)
564 if (mOutputTree && other.mOutputTree) {
565 mOutputTree->merge(*other.mOutputTree, mMergePolicy);
572 const InTreeT* mInputTree;
573 OutTreeT* mOutputTree;
575 OpT
const *
const mOrigOp;
585template<
typename InIterT,
typename OutGr
idT,
typename XformOp>
591 using OutTreeT =
typename Adapter::TreeType;
593 using Processor =
typename valxform::SharedOpTransformer<InIterT, OutTreeT, XformOp>;
594 Processor proc(inIter, Adapter::tree(outGrid), op, merge);
595 proc.process(threaded);
597 using Processor =
typename valxform::CopyableOpTransformer<InIterT, OutTreeT, XformOp>;
598 Processor proc(inIter, Adapter::tree(outGrid), op, merge);
599 proc.process(threaded);
603template<
typename InIterT,
typename OutGr
idT,
typename XformOp>
609 using OutTreeT =
typename Adapter::TreeType;
611 using Processor =
typename valxform::SharedOpTransformer<InIterT, OutTreeT, const XformOp>;
612 Processor proc(inIter, Adapter::tree(outGrid), op, merge);
613 proc.process(threaded);
622template<
typename IterT,
typename OpT>
632 OpAccumulator(
const IterT& iter, OpT& op):
641 OpAccumulator(OpAccumulator& other, tbb::split):
644 mOp(new OpT(*other.mOrigOp)),
645 mOrigOp(other.mOrigOp)
648 ~OpAccumulator() {
if (mIsRoot)
delete mOrigOp;
else delete mOp; }
650 void process(
bool threaded =
true)
652 IterRange range(mIter);
654 tbb::parallel_reduce(range, *
this);
660 void operator()(IterRange& r) {
for ( ; r; ++r) (*mOp)(r.iterator()); }
662 void join(OpAccumulator& other) { mOp->join(*other.mOp); }
668 OpT
const *
const mOrigOp;
677template<
typename IterT,
typename XformOp>
679accumulate(
const IterT& iter, XformOp& op,
bool threaded)
681 typename valxform::OpAccumulator<IterT, XformOp> proc(iter, op);
682 proc.process(threaded);
691#ifdef OPENVDB_USE_EXPLICIT_INSTANTIATION
693#ifdef OPENVDB_INSTANTIATE_VALUETRANSFORMER
697#define _FUNCTION(TreeT) \
698 void setValueOnMin(TreeT&, const Coord&, const TreeT::ValueType&)
702#define _FUNCTION(TreeT) \
703 void setValueOnMax(TreeT&, const Coord&, const TreeT::ValueType&)
707#define _FUNCTION(TreeT) \
708 void setValueOnSum(TreeT&, const Coord&, const TreeT::ValueType&)
712#define _FUNCTION(TreeT) \
713 void setValueOnMult(TreeT&, const Coord&, const TreeT::ValueType&)
ValueT value
Definition GridBuilder.h:1290
Signed (x, y, z) 32-bit integer coordinates.
Definition Coord.h:25
Definition TreeIterator.h:1303
#define OPENVDB_LOG_INFO(message)
Log an info message of the form 'someVar << "some text" << ...'.
Definition logging.h:254
PromoteType< ValueT >::Highest accumulate(const PointDataTreeT &points, const std::string &attribute, const FilterT &filter=NullFilter())
Evaluates the total value of a point attribute.
Definition PointStatistics.h:729
T zeroVal()
Return the value of type T that corresponds to zero.
Definition Math.h:70
MergePolicy
Definition Types.h:467
Definition Exceptions.h:13
This adapter allows code that is templated on a Tree type to accept either a Tree type or a Grid type...
Definition Grid.h:1060
#define OPENVDB_VERSION_NAME
The version namespace name for this library version.
Definition version.h.in:121
#define OPENVDB_USE_VERSION_NAMESPACE
Definition version.h.in:212
#define OPENVDB_VOLUME_TREE_INSTANTIATE(Function)
Definition version.h.in:160