BKE_volume_grid_process.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2023 Blender Foundation
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once

 
#ifdef WITH_OPENVDB
 
#  include <openvdb/openvdb.h>
 
#  include "BKE_volume_grid.hh"
#  include "BKE_volume_openvdb.hh"
 
#  include "BLI_function_ref.hh"
#  include "BLI_generic_pointer.hh"
#  include "BLI_generic_span.hh"
#  include "BLI_index_mask_fwd.hh"
 
namespace blender::bke::volume_grid {
 
using LeafNodeMask = openvdb::util::NodeMask<3u>;
 
using GetVoxelsFn = FunctionRef<void(MutableSpan<openvdb::Coord> r_voxels)>;
using ProcessLeafFn = FunctionRef<void(const LeafNodeMask &leaf_node_mask,
                                       const openvdb::CoordBBox &leaf_bbox,
                                       GetVoxelsFn get_voxels_fn)>;
using ProcessTilesFn = FunctionRef<void(Span<openvdb::CoordBBox> tiles)>;
using ProcessVoxelsFn = FunctionRef<void(Span<openvdb::Coord> voxels)>;

void parallel_grid_topology_tasks(const openvdb::MaskTree &mask_tree,
                                  ProcessLeafFn process_leaf_fn,
                                  ProcessVoxelsFn process_voxels_fn,
                                  ProcessTilesFn process_tiles_fn);
 
template<typename GridT>
constexpr bool is_supported_grid_type = is_same_any_v<GridT,
                                                      openvdb::FloatGrid,
                                                      openvdb::Vec3fGrid,
                                                      openvdb::BoolGrid,
                                                      openvdb::Int32Grid,
                                                      openvdb::Vec4fGrid>;
 
template<typename Fn> inline void to_typed_grid(const openvdb::GridBase &grid_base, Fn &&fn)
{
  const VolumeGridType grid_type = get_type(grid_base);
  BKE_volume_grid_type_to_static_type(grid_type, [&](auto type_tag) {
    using GridT = typename decltype(type_tag)::type;
    if constexpr (is_supported_grid_type<GridT>) {
      fn(static_cast<const GridT &>(grid_base));
    }
    else {
      BLI_assert_unreachable();
    }
  });
}
 
template<typename Fn> inline void to_typed_grid(openvdb::GridBase &grid_base, Fn &&fn)
{
  const VolumeGridType grid_type = get_type(grid_base);
  BKE_volume_grid_type_to_static_type(grid_type, [&](auto type_tag) {
    using GridT = typename decltype(type_tag)::type;
    if constexpr (is_supported_grid_type<GridT>) {
      fn(static_cast<GridT &>(grid_base));
    }
    else {
      BLI_assert_unreachable();
    }
  });
}

openvdb::GridBase::Ptr create_grid_with_topology(const openvdb::MaskTree &topology,
                                                 const openvdb::math::Transform &transform,
                                                 const VolumeGridType grid_type);

void set_grid_values(openvdb::GridBase &grid_base, GSpan values, Span<openvdb::Coord> voxels);

void set_tile_values(openvdb::GridBase &grid_base, GSpan values, Span<openvdb::CoordBBox> tiles);

void set_mask_leaf_buffer_from_bools(openvdb::BoolGrid &grid,
                                     Span<bool> values,
                                     const IndexMask &index_mask,
                                     Span<openvdb::Coord> voxels);
 
void set_grid_background(openvdb::GridBase &grid_base, const GPointer value);

void prune_inactive(openvdb::GridBase &grid_base);
 
}  // namespace blender::bke::volume_grid

 
#endif /* WITH_OPENVDB */