diff options
Diffstat (limited to 'src/openvic-simulation/dataloader/NodeTools.hpp')
| -rw-r--r-- | src/openvic-simulation/dataloader/NodeTools.hpp | 23 |
1 files changed, 23 insertions, 0 deletions
diff --git a/src/openvic-simulation/dataloader/NodeTools.hpp b/src/openvic-simulation/dataloader/NodeTools.hpp index 51bbfa9..5c22e11 100644 --- a/src/openvic-simulation/dataloader/NodeTools.hpp +++ b/src/openvic-simulation/dataloader/NodeTools.hpp @@ -41,6 +41,7 @@ namespace OpenVic { node_callback_t expect_list_and_length(length_callback_t length_callback, node_callback_t callback); node_callback_t expect_list_of_length(size_t length, node_callback_t callback); node_callback_t expect_list(node_callback_t callback); + node_callback_t expect_length(callback_t<size_t> callback); node_callback_t expect_dictionary_and_length(length_callback_t length_callback, key_value_callback_t callback); node_callback_t expect_dictionary(key_value_callback_t callback); @@ -136,6 +137,28 @@ namespace OpenVic { }; } + template<typename T> + requires requires(T& t) { + t += T {}; + } + callback_t<T> add_variable_callback(T& var) { + return [&var](T val) -> bool { + var += val; + return true; + }; + } + + template<typename T> + requires requires(T& t) { + t--; + } + node_callback_t decrement_callback(T& var, node_callback_t callback) { + return [&var, callback](ast::NodeCPtr node) -> bool { + var--; + return callback(node); + }; + } + template<typename I, typename T> requires(std::integral<I>, std::integral<T>) callback_t<I> _assign_variable_callback_int(const std::string_view name, T& var) { |