-
Notifications
You must be signed in to change notification settings - Fork 0
/
my_function_map.h
105 lines (85 loc) · 3.3 KB
/
my_function_map.h
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
#ifndef MY_FUNCTION_MAP_H
#define MY_FUNCTION_MAP_H
#include <any>
#include <assert.h>
#include <functional>
#include <iostream>
#include <map>
#include <memory>
#include <string>
#include <utility>
class Function {
public:
virtual std::any call(const std::vector<std::any>& args) = 0;
};
template<typename ReturnType, typename... Args>
class CallableFunction : public Function {
public:
CallableFunction(ReturnType(*func)(Args...)) : func_(func) {}
std::any call(const std::vector<std::any>& args) override {
assert(sizeof...(Args) == args.size());
try {
std::tuple<Args...> arguments;
fill_tuple(arguments, args);
return std::apply([this](auto&&... args) { return func_(args ...); }, arguments);
}
catch (std::bad_any_cast& bac) {
std::cerr << "Error : Bad Any Cast. Returning Nullptr.\n";
return std::any(nullptr);
}
}
private:
ReturnType(*func_)(Args...);
template<std::size_t... I>
void fill_tuple(std::tuple<Args...>& t, const std::vector<std::any>& args, std::index_sequence<I...>) {
((std::get<I>(t) = std::any_cast<typename std::tuple_element<I, std::tuple<Args...>>::type>(args[I])), ...);
}
void fill_tuple(std::tuple<Args...>& t, const std::vector<std::any>& args) {
fill_tuple(t, args, std::index_sequence_for<Args...>{});
}
};
template<typename ClassType, typename ReturnType, typename... Args>
class CallableClassFunction : public Function {
public:
CallableClassFunction(ReturnType(ClassType::* func)(Args...), ClassType* obj)
: func_(func), obj_(obj) {}
std::any call(const std::vector<std::any>& args) override {
assert(sizeof...(Args) == args.size());
try {
std::tuple<Args...> arguments;
fill_tuple(arguments, args);
return std::apply([this](auto&&... args){ return (obj_->*func_)(args...); }, arguments);
}
catch (std::bad_any_cast& bac) {
std::cerr << "Error : Bad Any Cast. Returning Nullptr.\n";
return std::any(nullptr);
}
}
private:
ReturnType(ClassType::* func_)(Args...);
ClassType* obj_;
template<std::size_t... I>
void fill_tuple(std::tuple<Args...>& t, const std::vector<std::any>& args, std::index_sequence<I...>) {
((std::get<I>(t) = std::any_cast<typename std::tuple_element<I, std::tuple<Args...>>::type>(args[I])), ...);
}
void fill_tuple(std::tuple<Args...>& t, const std::vector<std::any>& args) {
fill_tuple(t, args, std::index_sequence_for<Args...>{});
}
};
class MyFunctionMap {
public:
template<typename ReturnType, typename... Args>
void bindFunction(const std::string& name, ReturnType(*func)(Args...)) {
functions[name] = std::make_shared<CallableFunction<ReturnType, Args...>>(func);
}
template<typename ClassType, typename ReturnType, typename... Args>
void bindClassFunction(const std::string& name, ReturnType(ClassType::* func)(Args...), ClassType* obj) {
functions[name] = std::make_shared<CallableClassFunction<ClassType, ReturnType, Args...>>(func, obj);
}
std::shared_ptr<Function> getFunction(const std::string& name) {
return functions[name];
}
private:
std::map<std::string, std::shared_ptr<Function>> functions;
};
#endif