BKE_volume_grid_fields.hh

Go to the documentation of this file.

/* SPDX-FileCopyrightText: 2025 Blender Authors
 *
 * SPDX-License-Identifier: GPL-2.0-or-later */
 
#pragma once
 
#include "FN_field.hh"
 
#include "BLI_math_basis_types.hh"
 
#ifdef WITH_OPENVDB
 
#  include "openvdb_fwd.hh"
 
namespace blender::bke {

class VoxelFieldContext : public fn::FieldContext {
 private:
  const openvdb::math::Transform &transform_;
  Span<openvdb::Coord> voxels_;
 
 public:
  VoxelFieldContext(const openvdb::math::Transform &transform, Span<openvdb::Coord> voxels);
 
  GVArray get_varray_for_input(const fn::FieldInput &field_input,
                               const IndexMask &mask,
                               ResourceScope &scope) const override;
 
  Span<openvdb::Coord> voxels() const
  {
    return voxels_;
  }
};

class TilesFieldContext : public fn::FieldContext {
 private:
  const openvdb::math::Transform &transform_;
  Span<openvdb::CoordBBox> tiles_;
 
 public:
  TilesFieldContext(const openvdb::math::Transform &transform,
                    const Span<openvdb::CoordBBox> tiles);
 
  GVArray get_varray_for_input(const fn::FieldInput &field_input,
                               const IndexMask &mask,
                               ResourceScope &scope) const override;
 
  Span<openvdb::CoordBBox> tiles() const
  {
    return tiles_;
  }
};
 
class VoxelCoordinateFieldInput : public fn::FieldInput {
 private:
  math::Axis axis_;
 
 public:
  VoxelCoordinateFieldInput(math::Axis axis);
 
  GVArray get_varray_for_context(const fn::FieldContext &context,
                                 const IndexMask &mask,
                                 ResourceScope &scope) const override;
};
 
class VoxelExtentFieldInput : public fn::FieldInput {
 private:
  math::Axis axis_;
 
 public:
  VoxelExtentFieldInput(math::Axis axis);
 
  GVArray get_varray_for_context(const fn::FieldContext &context,
                                 const IndexMask &mask,
                                 ResourceScope &scope) const override;
};
 
class IsTileFieldInput : public fn::FieldInput {
 public:
  IsTileFieldInput();
 
  GVArray get_varray_for_context(const fn::FieldContext &context,
                                 const IndexMask &mask,
                                 ResourceScope &scope) const override;
};
 
}  // namespace blender::bke
 
#endif