12 #ifndef OPENVDB_TOOLS_LEVELSETSPHERE_HAS_BEEN_INCLUDED
13 #define OPENVDB_TOOLS_LEVELSETSPHERE_HAS_BEEN_INCLUDED
23 #include <type_traits>
25 #include <tbb/enumerable_thread_specific.h>
26 #include <tbb/parallel_for.h>
27 #include <tbb/parallel_reduce.h>
28 #include <tbb/blocked_range.h>
50 template<
typename Gr
idType,
typename InterruptT>
51 typename GridType::Ptr
54 InterruptT* interrupt =
nullptr,
bool threaded =
true);
69 template<
typename Gr
idType>
70 typename GridType::Ptr
74 return createLevelSetSphere<GridType, util::NullInterrupter>(radius,center,voxelSize,halfWidth,
nullptr,threaded);
87 template<
typename Gr
idT,
typename InterruptT = util::NullInterrupter>
91 using TreeT =
typename GridT::TreeType;
92 using ValueT =
typename GridT::ValueType;
94 static_assert(std::is_floating_point<ValueT>::value,
95 "level set grids must have scalar, floating-point value types");
108 : mRadius(radius), mCenter(center), mInterrupt(interrupt)
120 mGrid = createLevelSet<GridT>(voxelSize, halfWidth);
121 this->rasterSphere(voxelSize, halfWidth, threaded);
127 void rasterSphere(ValueT dx, ValueT w,
bool threaded)
133 const ValueT r0 = mRadius/dx, rmax = r0 + w;
136 if (r0 < 1.5f)
return;
139 const Vec3T c(mCenter[0]/dx, mCenter[1]/dx, mCenter[2]/dx);
147 typename GridT::Accessor accessor = mGrid->getAccessor();
149 if (mInterrupt) mInterrupt->start(
"Generating level set of sphere");
151 tbb::enumerable_thread_specific<TreeT> pool(mGrid->tree());
153 auto kernel = [&](
const tbb::blocked_range<int>& r) {
155 int &i = ijk[0], &j = ijk[1], &k = ijk[2], m=1;
156 TreeT &tree = pool.local();
157 typename GridT::Accessor acc(tree);
159 for (i = r.begin(); i != r.end(); ++i) {
162 for (j = jmin; j <= jmax; ++j) {
163 const auto x2y2 =
math::Pow2(ValueT(j) - c[1]) + x2;
164 for (k = kmin; k <= kmax; k += m) {
170 acc.setValue(ijk, dx*v);
184 tbb::parallel_for(tbb::blocked_range<int>(imin, imax, 128), kernel);
185 using RangeT = tbb::blocked_range<typename tbb::enumerable_thread_specific<TreeT>::iterator>;
189 Op(TreeT &tree) : mDelete(false), mTree(&tree) {}
190 Op(
const Op& other,
tbb::split) : mDelete(true), mTree(new TreeT(other.mTree->background())) {}
191 ~Op() {
if (mDelete)
delete mTree; }
192 void operator()(
const RangeT &r) {
for (
auto i=r.begin(); i!=r.end(); ++i) this->merge(*i);}
193 void join(Op &other) { this->merge(*(other.mTree)); }
195 } op( mGrid->tree() );
196 tbb::parallel_reduce(RangeT(pool.begin(), pool.end(), 4), op);
198 kernel(tbb::blocked_range<int>(imin, imax));
205 if (mInterrupt) mInterrupt->end();
208 const ValueT mRadius;
210 InterruptT* mInterrupt;
211 typename GridT::Ptr mGrid;
218 template<
typename Gr
idType,
typename InterruptT>
219 typename GridType::Ptr
221 float halfWidth, InterruptT* interrupt,
bool threaded)
224 static_assert(std::is_floating_point<typename GridType::ValueType>::value,
225 "level set grids must have scalar, floating-point value types");
227 using ValueT =
typename GridType::ValueType;
229 return factory.
getLevelSet(ValueT(voxelSize), ValueT(halfWidth), threaded);
238 #ifdef OPENVDB_USE_EXPLICIT_INSTANTIATION
240 #ifdef OPENVDB_INSTANTIATE_LEVELSETSPHERE
244 #define _FUNCTION(TreeT) \
245 Grid<TreeT>::Ptr createLevelSetSphere<Grid<TreeT>>(float, const openvdb::Vec3f&, float, float, \
246 util::NullInterrupter*, bool)
General-purpose arithmetic and comparison routines, most of which accept arbitrary value types (or at...
Propagate the signs of distance values from the active voxels in the narrow band to the inactive valu...
Definition: Exceptions.h:65
Signed (x, y, z) 32-bit integer coordinates.
Definition: Coord.h:25
Coord Abs(const Coord &xyz)
Definition: Coord.h:517
float Sqrt(float x)
Return the square root of a floating-point value.
Definition: Math.h:761
int Floor(float x)
Return the floor of x.
Definition: Math.h:848
int Ceil(float x)
Return the ceiling of x.
Definition: Math.h:856
Type Pow2(Type x)
Return x2.
Definition: Math.h:548
void split(ContainerT &out, const std::string &in, const char delim)
Definition: Name.h:43
bool wasInterrupted(T *i, int percent=-1)
Definition: NullInterrupter.h:49
static const Real LEVEL_SET_HALF_WIDTH
Definition: Types.h:461
@ GRID_LEVEL_SET
Definition: Types.h:455
@ MERGE_ACTIVE_STATES
Definition: Types.h:507
Definition: Exceptions.h:13
#define OPENVDB_THROW(exception, message)
Definition: Exceptions.h:74
#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_REAL_TREE_INSTANTIATE(Function)
Definition: version.h.in:157