10 #ifndef OPENVDB_TOOLS_LEVEL_SET_ADVECT_HAS_BEEN_INCLUDED
11 #define OPENVDB_TOOLS_LEVEL_SET_ADVECT_HAS_BEEN_INCLUDED
13 #include <tbb/parallel_for.h>
14 #include <tbb/parallel_reduce.h>
71 template<
typename GridT,
72 typename FieldT = EnrightField<typename GridT::ValueType>,
73 typename InterruptT = util::NullInterrupter>
87 mTracker(grid, interrupt), mField(field),
89 mTemporalScheme(math::
TVD_RK2) {}
152 Advect(
const Advect& other);
154 virtual ~Advect() {
if (mIsMaster) this->clearField(); }
161 if (mTask) mTask(const_cast<Advect*>(
this), r);
162 else OPENVDB_THROW(ValueError,
"task is undefined - don\'t call this method directly");
165 void cook(
const char* msg,
size_t swapBuffer = 0);
171 this->sample<false>(
r, t0, t1);
175 this->sample<true>(
r, t0, t1);
180 template <
int Nominator,
int Denominator>
191 typename std::function<void (Advect*, const LeafRange&)> mTask;
192 const bool mIsMaster;
195 template<math::BiasedGradientScheme SpatialScheme>
216 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
220 switch (mSpatialScheme) {
222 return this->advect1<math::FIRST_BIAS >(time0, time1);
224 return this->advect1<math::SECOND_BIAS >(time0, time1);
226 return this->advect1<math::THIRD_BIAS >(time0, time1);
228 return this->advect1<math::WENO5_BIAS >(time0, time1);
230 return this->advect1<math::HJWENO5_BIAS>(time0, time1);
232 OPENVDB_THROW(ValueError,
"Spatial difference scheme not supported!");
238 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
239 template<math::BiasedGradientScheme SpatialScheme>
243 switch (mTemporalScheme) {
245 return this->advect2<SpatialScheme, math::TVD_RK1>(time0, time1);
247 return this->advect2<SpatialScheme, math::TVD_RK2>(time0, time1);
249 return this->advect2<SpatialScheme, math::TVD_RK3>(time0, time1);
251 OPENVDB_THROW(ValueError,
"Temporal integration scheme not supported!");
257 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
258 template<math::BiasedGradientScheme SpatialScheme, math::TemporalIntegrationScheme TemporalScheme>
260 LevelSetAdvection<GridT, FieldT, InterruptT>::advect2(
ValueType time0,
ValueType time1)
262 const math::Transform&
trans = mTracker.grid().transform();
264 return this->advect3<SpatialScheme, TemporalScheme, math::UniformScaleMap>(time0, time1);
266 return this->advect3<SpatialScheme, TemporalScheme, math::UniformScaleTranslateMap>(
269 return this->advect3<SpatialScheme, TemporalScheme, math::UnitaryMap >(time0, time1);
271 return this->advect3<SpatialScheme, TemporalScheme, math::TranslationMap>(time0, time1);
279 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
285 LevelSetAdvection<GridT, FieldT, InterruptT>::advect3(
ValueType time0,
ValueType time1)
287 Advect<MapT, SpatialScheme, TemporalScheme> tmp(*
this);
288 return tmp.advect(time0, time1);
295 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
301 LevelSetAdvection<GridT, FieldT, InterruptT>::
302 Advect<MapT, SpatialScheme, TemporalScheme>::
303 Advect(LevelSetAdvection& parent)
307 , mMap(parent.mTracker.grid().
transform().template constMap<MapT>().
get())
314 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
320 LevelSetAdvection<GridT, FieldT, InterruptT>::
321 Advect<MapT, SpatialScheme, TemporalScheme>::
322 Advect(
const Advect& other)
323 : mParent(other.mParent)
324 , mVelocity(other.mVelocity)
325 , mOffsets(other.mOffsets)
333 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
343 namespace ph = std::placeholders;
348 const bool isForward = time0 < time1;
349 while ((isForward ? time0<time1 : time0>time1) && mParent.mTracker.checkInterrupter()) {
352 mParent.mTracker.leafs().rebuildAuxBuffers(TemporalScheme ==
math::TVD_RK3 ? 2 : 1);
355 const ValueType dt = this->sampleField(time0, time1);
359 switch(TemporalScheme) {
363 mTask = std::bind(&Advect::euler01, ph::_1, ph::_2, dt);
366 this->cook(
"Advecting level set using TVD_RK1", 1);
371 mTask = std::bind(&Advect::euler01, ph::_1, ph::_2, dt);
374 this->cook(
"Advecting level set using TVD_RK1 (step 1 of 2)", 1);
378 mTask = std::bind(&Advect::euler12, ph::_1, ph::_2, dt);
381 this->cook(
"Advecting level set using TVD_RK1 (step 2 of 2)", 1);
386 mTask = std::bind(&Advect::euler01, ph::_1, ph::_2, dt);
389 this->cook(
"Advecting level set using TVD_RK3 (step 1 of 3)", 1);
393 mTask = std::bind(&Advect::euler34, ph::_1, ph::_2, dt);
396 this->cook(
"Advecting level set using TVD_RK3 (step 2 of 3)", 2);
400 mTask = std::bind(&Advect::euler13, ph::_1, ph::_2, dt);
403 this->cook(
"Advecting level set using TVD_RK3 (step 3 of 3)", 2);
406 OPENVDB_THROW(ValueError,
"Temporal integration scheme not supported!");
410 time0 += isForward ? dt : -dt;
412 mParent.mTracker.leafs().removeAuxBuffers();
415 mParent.mTracker.track();
421 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
427 LevelSetAdvection<GridT, FieldT, InterruptT>::
428 Advect<MapT, SpatialScheme, TemporalScheme>::
431 namespace ph = std::placeholders;
433 const int grainSize = mParent.mTracker.getGrainSize();
434 const size_t leafCount = mParent.mTracker.leafs().leafCount();
438 size_t size=0, voxelCount=mParent.mTracker.leafs().getPrefixSum(mOffsets, size, grainSize);
441 if (mParent.mField.transform() == mParent.mTracker.grid().transform()) {
442 mTask = std::bind(&Advect::sampleAligned, ph::_1, ph::_2, time0, time1);
444 mTask = std::bind(&Advect::sampleXformed, ph::_1, ph::_2, time0, time1);
446 OPENVDB_ASSERT(voxelCount == mParent.mTracker.grid().activeVoxelCount());
447 mVelocity =
new VectorType[ voxelCount ];
448 this->cook(
"Sampling advection field");
452 VectorType*
v = mVelocity;
453 for (
size_t i = 0; i < voxelCount; ++i, ++
v) {
460 TemporalScheme == math::
TVD_RK2 ? ValueType(0.9) :
461 ValueType(1.0))/math::
Sqrt(ValueType(3.0));
462 const ValueType dt =
math::Abs(time1 - time0), dx = mParent.mTracker.voxelSize();
467 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
472 template<
bool Aligned>
474 LevelSetAdvection<GridT, FieldT, InterruptT>::
475 Advect<MapT, SpatialScheme, TemporalScheme>::
476 sample(
const LeafRange&
range, ValueType time0, ValueType time1)
478 const bool isForward = time0 < time1;
479 using VoxelIterT =
typename LeafType::ValueOnCIter;
480 const MapT& map = *mMap;
481 const FieldT field( mParent.mField );
482 mParent.mTracker.checkInterrupter();
483 for (
typename LeafRange::Iterator leafIter = range.begin(); leafIter; ++leafIter) {
484 VectorType* vel = mVelocity + mOffsets[ leafIter.pos() ];
485 for (VoxelIterT iter = leafIter->cbeginValueOn(); iter; ++iter, ++vel) {
487 const VectorType v = Aligned ? field(iter.getCoord(), time0) :
488 field(map.applyMap(iter.getCoord().asVec3d()), time0);
489 *vel = isForward ? v : -
v;
496 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
502 LevelSetAdvection<GridT, FieldT, InterruptT>::
503 Advect<MapT, SpatialScheme, TemporalScheme>::
513 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
519 LevelSetAdvection<GridT, FieldT, InterruptT>::
520 Advect<MapT, SpatialScheme, TemporalScheme>::
521 cook(
const char* msg,
size_t swapBuffer)
523 mParent.mTracker.startInterrupter( msg );
525 const int grainSize = mParent.mTracker.getGrainSize();
526 const LeafRange range = mParent.mTracker.leafs().leafRange(grainSize);
530 mParent.mTracker.leafs().swapLeafBuffer(swapBuffer, grainSize == 0);
532 mParent.mTracker.endInterrupter();
538 template<
typename Gr
idT,
typename FieldT,
typename InterruptT>
543 template <
int Nominator,
int Denominator>
545 LevelSetAdvection<GridT, FieldT, InterruptT>::
546 Advect<MapT, SpatialScheme, TemporalScheme>::
547 euler(
const LeafRange& range, ValueType dt,
Index phiBuffer,
Index resultBuffer)
549 using SchemeT = math::BIAS_SCHEME<SpatialScheme>;
550 using StencilT =
typename SchemeT::template ISStencil<GridType>::StencilType;
551 using VoxelIterT =
typename LeafType::ValueOnCIter;
552 using GradT = math::GradientBiased<MapT, SpatialScheme>;
557 mParent.mTracker.checkInterrupter();
558 const MapT& map = *mMap;
559 StencilT
stencil(mParent.mTracker.grid());
560 for (
typename LeafRange::Iterator leafIter = range.begin(); leafIter; ++leafIter) {
561 const VectorType* vel = mVelocity + mOffsets[ leafIter.pos() ];
562 const ValueType* phi = leafIter.buffer(phiBuffer).data();
563 ValueType*
result = leafIter.buffer(resultBuffer).data();
564 for (VoxelIterT voxelIter = leafIter->cbeginValueOn(); voxelIter; ++voxelIter, ++vel) {
565 const Index i = voxelIter.pos();
569 result[i] = Nominator ? Alpha * phi[i] + Beta * a :
a;
580 #ifdef OPENVDB_USE_EXPLICIT_INSTANTIATION
582 #ifdef OPENVDB_INSTANTIATE_LEVELSETADVECT
589 #endif // OPENVDB_USE_EXPLICIT_INSTANTIATION
596 #endif // OPENVDB_TOOLS_LEVEL_SET_ADVECT_HAS_BEEN_INCLUDED
TemporalIntegrationScheme
Temporal integration schemes.
GLsizei const GLfloat * value
GLboolean GLboolean GLboolean GLboolean a
#define OPENVDB_USE_VERSION_NAMESPACE
**But if you need a result
BiasedGradientScheme
Biased Gradients are limited to non-centered differences.
#define OPENVDB_ASSERT(X)
#define OPENVDB_INSTANTIATE_CLASS
Coord Abs(const Coord &xyz)
GA_API const UT_StringHolder trans
OIIO_UTIL_API void parallel_for(int32_t begin, int32_t end, function_view< void(int32_t)> task, paropt opt=0)
float Sqrt(float x)
Return the square root of a floating-point value.
auto get(const UT_ARTIterator< T > &it) -> decltype(it.key())
Performs multi-threaded interface tracking of narrow band level sets. This is the building-block for ...
bool isApproxZero(const Type &x)
Return true if x is equal to zero to within the default floating-point comparison tolerance...
GA_API const UT_StringHolder transform
static Name mapType()
Return UnitaryMap.
Defines two simple wrapper classes for advection velocity fields as well as VelocitySampler and Veloc...
const Type & Min(const Type &a, const Type &b)
Return the minimum of two values.
GLint GLfloat GLint stencil
__hostdev__ float Sqrt(float x)
Return the square root of a floating-point value.
GA_API const UT_StringHolder Alpha
#define OPENVDB_VERSION_NAME
The version namespace name for this library version.
const Type & Max(const Type &a, const Type &b)
Return the maximum of two values.
bool isZero(const Type &x)
Return true if x is exactly equal to zero.
#define OPENVDB_THROW(exception, message)