9aade6bd61
OpenVDB will throw an exception when trying to prune a SDF/Level-Set grid with a negative background value. The geometry nodes in this case are constructing a nominal SDF grid from a generic "density" volume cube. The negative background value results from processing the background value the same way as all other grid values in the new field+grid math feature (`blender::nodes::process_background`). Since any grid can be constructed with a negative background value we have to check and correct this before calling the OpenVDB prune function. Pull Request: https://projects.blender.org/blender/blender/pulls/146837
44 lines
1.4 KiB
C++
44 lines
1.4 KiB
C++
/* SPDX-FileCopyrightText: 2024 Blender Authors
|
|
*
|
|
* SPDX-License-Identifier: GPL-2.0-or-later */
|
|
|
|
#ifdef WITH_OPENVDB
|
|
|
|
# include "BKE_volume_grid.hh"
|
|
|
|
# include "GEO_volume_grid_resample.hh"
|
|
|
|
# include <openvdb/openvdb.h>
|
|
# include <openvdb/tools/GridTransformer.h>
|
|
|
|
namespace blender::geometry {
|
|
|
|
openvdb::FloatGrid &resample_sdf_grid_if_necessary(bke::VolumeGrid<float> &volume_grid,
|
|
bke::VolumeTreeAccessToken &tree_token,
|
|
const openvdb::math::Transform &transform,
|
|
std::shared_ptr<openvdb::FloatGrid> &storage)
|
|
{
|
|
const openvdb::FloatGrid &grid = volume_grid.grid(tree_token);
|
|
if (grid.transform() == transform) {
|
|
return volume_grid.grid_for_write(tree_token);
|
|
}
|
|
|
|
storage = openvdb::FloatGrid::create();
|
|
storage->setTransform(transform.copy());
|
|
|
|
/* TODO: Using #doResampleToMatch when the transform is affine and non-scaled may be faster. */
|
|
openvdb::tools::resampleToMatch<openvdb::tools::BoxSampler>(grid, *storage);
|
|
/* Ensure valid background value for level set grids, otherwise pruning will throw an exception.
|
|
*/
|
|
if (storage->background() < 0.0f) {
|
|
storage->tree().root().setBackground(0.0f, true);
|
|
}
|
|
openvdb::tools::pruneLevelSet(storage->tree());
|
|
|
|
return *storage;
|
|
}
|
|
|
|
} // namespace blender::geometry
|
|
|
|
#endif
|