operation.hpp 7.83 KB
Newer Older
Paul's avatar
Paul committed
1
#ifndef RTG_GUARD_RTGLIB_OPERAND_HPP
Paul's avatar
Paul committed
2
#define RTG_GUARD_RTGLIB_OPERAND_HPP
Paul's avatar
Paul committed
3

Paul's avatar
Paul committed
4
#include <string>
Paul's avatar
Paul committed
5
#include <functional>
Paul's avatar
Paul committed
6
7
8
#include <memory>
#include <type_traits>
#include <utility>
Paul's avatar
Paul committed
9
#include <rtg/shape.hpp>
Paul's avatar
Paul committed
10
#include <rtg/argument.hpp>
Paul's avatar
Paul committed
11
#include <rtg/context.hpp>
Paul's avatar
Paul committed
12
13
14

namespace rtg {

Paul's avatar
Paul committed
15
16
17
18
19
20
21
22
23
24
namespace operation_stream {

template <class T>
auto operator<<(std::ostream& os, const T& x) -> decltype(os << x.name())
{
    return os << x.name();
}

} // namespace operation_stream

Paul's avatar
Paul committed
25
/*
Paul's avatar
Paul committed
26
27
 * Type-erased interface for:
 *
Paul's avatar
Paul committed
28
 * struct operation
Paul's avatar
Paul committed
29
 * {
Paul's avatar
Paul committed
30
31
 *      std::string name() const;
 *      shape compute_shape(std::vector<shape> input) const;
Paul's avatar
Paul committed
32
 *      argument compute(context& ctx,shape output,std::vector<argument> input) const;
Paul's avatar
Paul committed
33
 *     friend std::ostream & operator<<(std::ostream & os,const operation & op) ;
Paul's avatar
Paul committed
34
35
36
 * };
 *
 */
Paul's avatar
Paul committed
37

Paul's avatar
Paul committed
38
struct operation
Paul's avatar
Paul committed
39
{
Paul's avatar
Paul committed
40
    // Constructors
Paul's avatar
Paul committed
41
    operation() = default;
Paul's avatar
Paul committed
42

Paul's avatar
Paul committed
43
    template <typename PrivateDetailTypeErasedT>
Paul's avatar
Paul committed
44
    operation(PrivateDetailTypeErasedT value)
Paul's avatar
Paul committed
45
46
47
48
        : private_detail_te_handle_mem_var(
              std::make_shared<private_detail_te_handle_type<
                  typename std::remove_reference<PrivateDetailTypeErasedT>::type>>(
                  std::forward<PrivateDetailTypeErasedT>(value)))
Paul's avatar
Paul committed
49
50
51
52
    {
    }

    // Assignment
Paul's avatar
Paul committed
53
    template <typename PrivateDetailTypeErasedT>
Paul's avatar
Paul committed
54
    operation& operator=(PrivateDetailTypeErasedT value)
Paul's avatar
Paul committed
55
    {
Paul's avatar
Paul committed
56
57
58
59
60
        if(private_detail_te_handle_mem_var.unique())
            *private_detail_te_handle_mem_var = std::forward<PrivateDetailTypeErasedT>(value);
        else if(!private_detail_te_handle_mem_var)
            private_detail_te_handle_mem_var = std::make_shared<PrivateDetailTypeErasedT>(
                std::forward<PrivateDetailTypeErasedT>(value));
Paul's avatar
Paul committed
61
62
63
        return *this;
    }

Paul's avatar
Paul committed
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
    // Cast
    template <typename PrivateDetailTypeErasedT>
    PrivateDetailTypeErasedT* any_cast()
    {
        return private_detail_te_get_handle().type() == typeid(PrivateDetailTypeErasedT)
                   ? std::addressof(static_cast<private_detail_te_handle_type<
                                        typename std::remove_cv<PrivateDetailTypeErasedT>::type>&>(
                                        private_detail_te_get_handle())
                                        .private_detail_te_value)
                   : nullptr;
    }

    template <typename PrivateDetailTypeErasedT>
    const typename std::remove_cv<PrivateDetailTypeErasedT>::type* any_cast() const
    {
        return private_detail_te_get_handle().type() == typeid(PrivateDetailTypeErasedT)
                   ? std::addressof(static_cast<const private_detail_te_handle_type<
                                        typename std::remove_cv<PrivateDetailTypeErasedT>::type>&>(
                                        private_detail_te_get_handle())
                                        .private_detail_te_value)
                   : nullptr;
    }

Paul's avatar
Paul committed
87
88
    std::string name() const
    {
Paul's avatar
Paul committed
89
90
        assert((*this).private_detail_te_handle_mem_var);
        return (*this).private_detail_te_get_handle().name();
Paul's avatar
Paul committed
91
92
93
94
    }

    shape compute_shape(std::vector<shape> input) const
    {
Paul's avatar
Paul committed
95
96
        assert((*this).private_detail_te_handle_mem_var);
        return (*this).private_detail_te_get_handle().compute_shape(std::move(input));
Paul's avatar
Paul committed
97
98
    }

Paul's avatar
Paul committed
99
    argument compute(context& ctx, shape output, std::vector<argument> input) const
Paul's avatar
Paul committed
100
    {
Paul's avatar
Paul committed
101
        assert((*this).private_detail_te_handle_mem_var);
Paul's avatar
Paul committed
102
103
        return (*this).private_detail_te_get_handle().compute(
            ctx, std::move(output), std::move(input));
Paul's avatar
Paul committed
104
105
106
107
108
109
    }

    friend std::ostream& operator<<(std::ostream& os, const operation& op)
    {
        assert(op.private_detail_te_handle_mem_var);
        return op.private_detail_te_get_handle().operator_shift_left(os);
Paul's avatar
Paul committed
110
111
112
    }

    private:
Paul's avatar
Paul committed
113
    struct private_detail_te_handle_base_type
Paul's avatar
Paul committed
114
    {
Paul's avatar
Paul committed
115
116
        virtual ~private_detail_te_handle_base_type() {}
        virtual std::shared_ptr<private_detail_te_handle_base_type> clone() const = 0;
Paul's avatar
Paul committed
117
        virtual const std::type_info& type() const                                = 0;
Paul's avatar
Paul committed
118

Paul's avatar
Paul committed
119
120
121
122
        virtual std::string name() const                                                        = 0;
        virtual shape compute_shape(std::vector<shape> input) const                             = 0;
        virtual argument compute(context& ctx, shape output, std::vector<argument> input) const = 0;
        virtual std::ostream& operator_shift_left(std::ostream& os) const                       = 0;
Paul's avatar
Paul committed
123
124
    };

Paul's avatar
Paul committed
125
126
    template <typename PrivateDetailTypeErasedT>
    struct private_detail_te_handle_type : private_detail_te_handle_base_type
Paul's avatar
Paul committed
127
    {
Paul's avatar
Paul committed
128
129
130
131
132
133
        template <typename PrivateDetailTypeErasedU = PrivateDetailTypeErasedT>
        private_detail_te_handle_type(
            PrivateDetailTypeErasedT value,
            typename std::enable_if<std::is_reference<PrivateDetailTypeErasedU>::value>::type* =
                nullptr)
            : private_detail_te_value(value)
Paul's avatar
Paul committed
134
135
136
        {
        }

Paul's avatar
Paul committed
137
138
139
140
141
142
        template <typename PrivateDetailTypeErasedU = PrivateDetailTypeErasedT>
        private_detail_te_handle_type(
            PrivateDetailTypeErasedT value,
            typename std::enable_if<!std::is_reference<PrivateDetailTypeErasedU>::value,
                                    int>::type* = nullptr) noexcept
            : private_detail_te_value(std::move(value))
Paul's avatar
Paul committed
143
144
145
        {
        }

Paul's avatar
Paul committed
146
        std::shared_ptr<private_detail_te_handle_base_type> clone() const override
Paul's avatar
Paul committed
147
        {
Paul's avatar
Paul committed
148
            return std::make_shared<private_detail_te_handle_type>(private_detail_te_value);
Paul's avatar
Paul committed
149
150
        }

Paul's avatar
Paul committed
151
        const std::type_info& type() const override { return typeid(private_detail_te_value); }
Paul's avatar
Paul committed
152

Paul's avatar
Paul committed
153
        std::string name() const override { return private_detail_te_value.name(); }
Paul's avatar
Paul committed
154

Paul's avatar
Paul committed
155
        shape compute_shape(std::vector<shape> input) const override
Paul's avatar
Paul committed
156
        {
Paul's avatar
Paul committed
157

Paul's avatar
Paul committed
158
            return private_detail_te_value.compute_shape(std::move(input));
Paul's avatar
Paul committed
159
160
        }

Paul's avatar
Paul committed
161
        argument compute(context& ctx, shape output, std::vector<argument> input) const override
Paul's avatar
Paul committed
162
        {
Paul's avatar
Paul committed
163

Paul's avatar
Paul committed
164
            return private_detail_te_value.compute(ctx, std::move(output), std::move(input));
Paul's avatar
Paul committed
165
166
        }

Paul's avatar
Paul committed
167
168
        std::ostream& operator_shift_left(std::ostream& os) const override
        {
Paul's avatar
Paul committed
169
            using rtg::operation_stream::operator<<;
Paul's avatar
Paul committed
170
171
172
            return os << private_detail_te_value;
        }

Paul's avatar
Paul committed
173
        PrivateDetailTypeErasedT private_detail_te_value;
Paul's avatar
Paul committed
174
175
    };

Paul's avatar
Paul committed
176
177
178
    template <typename PrivateDetailTypeErasedT>
    struct private_detail_te_handle_type<std::reference_wrapper<PrivateDetailTypeErasedT>>
        : private_detail_te_handle_type<PrivateDetailTypeErasedT&>
Paul's avatar
Paul committed
179
    {
Paul's avatar
Paul committed
180
181
        private_detail_te_handle_type(std::reference_wrapper<PrivateDetailTypeErasedT> ref)
            : private_detail_te_handle_type<PrivateDetailTypeErasedT&>(ref.get())
Paul's avatar
Paul committed
182
183
184
185
        {
        }
    };

Paul's avatar
Paul committed
186
187
188
189
    const private_detail_te_handle_base_type& private_detail_te_get_handle() const
    {
        return *private_detail_te_handle_mem_var;
    }
Paul's avatar
Paul committed
190

Paul's avatar
Paul committed
191
    private_detail_te_handle_base_type& private_detail_te_get_handle()
Paul's avatar
Paul committed
192
    {
Paul's avatar
Paul committed
193
194
195
        if(!private_detail_te_handle_mem_var.unique())
            private_detail_te_handle_mem_var = private_detail_te_handle_mem_var->clone();
        return *private_detail_te_handle_mem_var;
Paul's avatar
Paul committed
196
197
    }

Paul's avatar
Paul committed
198
    std::shared_ptr<private_detail_te_handle_base_type> private_detail_te_handle_mem_var;
Paul's avatar
Paul committed
199
200
};

Paul's avatar
Paul committed
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
template <typename ValueType>
inline const ValueType* any_cast(const operation* x)
{
    return x->any_cast<ValueType>();
}

template <typename ValueType>
inline ValueType* any_cast(operation* x)
{
    return x->any_cast<ValueType>();
}

template <typename ValueType>
inline ValueType& any_cast(operation& x)
{
Paul's avatar
Paul committed
216
    auto* y = x.any_cast<typename std::remove_reference<ValueType>::type>();
Paul's avatar
Paul committed
217
218
219
220
221
222
223
224
    if(y == nullptr)
        throw std::bad_cast();
    return *y;
}

template <typename ValueType>
inline const ValueType& any_cast(const operation& x)
{
Paul's avatar
Paul committed
225
    const auto* y = x.any_cast<typename std::remove_reference<ValueType>::type>();
Paul's avatar
Paul committed
226
227
228
229
230
    if(y == nullptr)
        throw std::bad_cast();
    return *y;
}

Paul's avatar
Paul committed
231
} // namespace rtg
Paul's avatar
Paul committed
232
233

#endif