blob: 36369211bfb6cf6fc6b72241e7d1267a3789dc74 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
|
/* SPDX-License-Identifier: GPL-2.0-or-later */
#pragma once
#include <tuple>
#include "BLI_virtual_array.hh"
namespace blender {
struct SingleInputTagBase {
};
template<typename T> struct SingleInputTag : public SingleInputTagBase {
};
struct SingleOutputTagBase {
};
template<typename T> struct SingleOutputTag : public SingleOutputTagBase {
};
template<typename T> struct ParamType {
};
template<typename T> struct ParamType<SingleInputTag<T>> {
using type = VArray<T>;
};
template<typename T> struct ParamType<SingleOutputTag<T>> {
using type = MutableSpan<T>;
};
template<typename Fn, typename... Args> class ArrayDevirtualizer {
private:
using TagsTuple = std::tuple<Args...>;
Fn fn_;
IndexMask mask_;
std::tuple<const typename ParamType<Args>::type *...> params_;
std::array<bool, sizeof...(Args)> varray_is_span_;
std::array<bool, sizeof...(Args)> varray_is_single_;
public:
ArrayDevirtualizer(Fn fn, const IndexMask *mask, const typename ParamType<Args>::type *...params)
: fn_(std::move(fn)), mask_(*mask), params_{params...}
{
this->init(std::make_index_sequence<sizeof...(Args)>{});
}
void execute_fallback()
{
this->execute_fallback_impl(std::make_index_sequence<sizeof...(Args)>{});
}
private:
template<size_t... I> void init(std::index_sequence<I...> /* indices */)
{
varray_is_span_.fill(false);
varray_is_single_.fill(false);
(this->init_param<I>(), ...);
}
template<size_t I> void init_param()
{
using ParamTag = std::tuple_element_t<I, TagsTuple>;
if constexpr (std::is_base_of_v<SingleInputTagBase, ParamTag>) {
const typename ParamType<ParamTag>::type *varray = std::get<I>(params_);
varray_is_span_[I] = varray->is_span();
varray_is_single_[I] = varray->is_single();
}
}
template<size_t... I> void execute_fallback_impl(std::index_sequence<I...> /* indices */)
{
fn_(mask_, mask_, *std::get<I>(params_)...);
}
};
} // namespace blender
|