diff options
Diffstat (limited to 'source/blender/blenlib/BLI_array_utils.hh')
-rw-r--r-- | source/blender/blenlib/BLI_array_utils.hh | 75 |
1 files changed, 75 insertions, 0 deletions
diff --git a/source/blender/blenlib/BLI_array_utils.hh b/source/blender/blenlib/BLI_array_utils.hh index cf2f948b0b4..95b3bde10f4 100644 --- a/source/blender/blenlib/BLI_array_utils.hh +++ b/source/blender/blenlib/BLI_array_utils.hh @@ -6,6 +6,7 @@ #include "BLI_generic_virtual_array.hh" #include "BLI_index_mask.hh" #include "BLI_task.hh" +#include "BLI_virtual_array.hh" namespace blender::array_utils { @@ -25,6 +26,7 @@ inline void copy(const Span<T> src, MutableSpan<T> dst, const int64_t grain_size = 4096) { + BLI_assert(src.size() == dst.size()); threading::parallel_for(selection.index_range(), grain_size, [&](const IndexRange range) { for (const int64_t index : selection.slice(range)) { dst[index] = src[index]; @@ -32,4 +34,77 @@ inline void copy(const Span<T> src, }); } +/** + * Fill the destination span by gathering indexed values from the `src` array. + */ +void gather(const GVArray &src, IndexMask indices, GMutableSpan dst, int64_t grain_size = 4096); + +/** + * Fill the destination span by gathering indexed values from the `src` array. + */ +template<typename T> +inline void gather(const VArray<T> &src, + const IndexMask indices, + MutableSpan<T> dst, + const int64_t grain_size = 4096) +{ + BLI_assert(indices.size() == dst.size()); + threading::parallel_for(indices.index_range(), grain_size, [&](const IndexRange range) { + src.materialize_compressed_to_uninitialized(indices.slice(range), dst.slice(range).data()); + }); +} + +/** + * Fill the destination span by gathering indexed values from the `src` array. + */ +template<typename T, typename IndexT> +inline void gather(const Span<T> src, + const IndexMask indices, + MutableSpan<T> dst, + const int64_t grain_size = 4096) +{ + BLI_assert(indices.size() == dst.size()); + threading::parallel_for(indices.index_range(), grain_size, [&](const IndexRange range) { + for (const int64_t i : range) { + dst[i] = src[indices[i]]; + } + }); +} + +/** + * Fill the destination span by gathering indexed values from the `src` array. + */ +template<typename T, typename IndexT> +inline void gather(const Span<T> src, + const Span<IndexT> indices, + MutableSpan<T> dst, + const int64_t grain_size = 4096) +{ + BLI_assert(indices.size() == dst.size()); + threading::parallel_for(indices.index_range(), grain_size, [&](const IndexRange range) { + for (const int64_t i : range) { + dst[i] = src[indices[i]]; + } + }); +} + +/** + * Fill the destination span by gathering indexed values from the `src` array. + */ +template<typename T, typename IndexT> +inline void gather(const VArray<T> &src, + const Span<IndexT> indices, + MutableSpan<T> dst, + const int64_t grain_size = 4096) +{ + BLI_assert(indices.size() == dst.size()); + devirtualize_varray(src, [&](const auto &src) { + threading::parallel_for(indices.index_range(), grain_size, [&](const IndexRange range) { + for (const int64_t i : range) { + dst[i] = src[indices[i]]; + } + }); + }); +} + } // namespace blender::array_utils |