/**************************************************************************/ /* gdscript_utility_functions.cpp */ /**************************************************************************/ /* This file is part of: */ /* GODOT ENGINE */ /* https://godotengine.org */ /**************************************************************************/ /* Copyright (c) 2014-present Godot Engine contributors (see AUTHORS.md). */ /* Copyright (c) 2007-2014 Juan Linietsky, Ariel Manzur. */ /* */ /* Permission is hereby granted, free of charge, to any person obtaining */ /* a copy of this software and associated documentation files (the */ /* "Software"), to deal in the Software without restriction, including */ /* without limitation the rights to use, copy, modify, merge, publish, */ /* distribute, sublicense, and/or sell copies of the Software, and to */ /* permit persons to whom the Software is furnished to do so, subject to */ /* the following conditions: */ /* */ /* The above copyright notice and this permission notice shall be */ /* included in all copies or substantial portions of the Software. */ /* */ /* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, */ /* EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF */ /* MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. */ /* IN NO EVENT SHALL THE AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY */ /* CLAIM, DAMAGES OR OTHER LIABILITY, WHETHER IN AN ACTION OF CONTRACT, */ /* TORT OR OTHERWISE, ARISING FROM, OUT OF OR IN CONNECTION WITH THE */ /* SOFTWARE OR THE USE OR OTHER DEALINGS IN THE SOFTWARE. */ /**************************************************************************/ #include "gdscript_utility_functions.h" #include "gdscript.h" #include "core/io/resource_loader.h" #include "core/object/class_db.h" #include "core/object/object.h" #include "core/templates/oa_hash_map.h" #include "core/templates/vector.h" #include "core/variant/typed_array.h" #ifdef DEBUG_ENABLED #define DEBUG_VALIDATE_ARG_COUNT(m_min_count, m_max_count) \ if (unlikely(p_arg_count < m_min_count)) { \ *r_ret = Variant(); \ r_error.error = Callable::CallError::CALL_ERROR_TOO_FEW_ARGUMENTS; \ r_error.expected = m_min_count; \ return; \ } \ if (unlikely(p_arg_count > m_max_count)) { \ *r_ret = Variant(); \ r_error.error = Callable::CallError::CALL_ERROR_TOO_MANY_ARGUMENTS; \ r_error.expected = m_max_count; \ return; \ } #define DEBUG_VALIDATE_ARG_TYPE(m_arg, m_type) \ if (unlikely(!Variant::can_convert_strict(p_args[m_arg]->get_type(), m_type))) { \ *r_ret = Variant(); \ r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT; \ r_error.argument = m_arg; \ r_error.expected = m_type; \ return; \ } #define DEBUG_VALIDATE_ARG_CUSTOM(m_arg, m_type, m_cond, m_msg) \ if (unlikely(m_cond)) { \ *r_ret = m_msg; \ r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT; \ r_error.argument = m_arg; \ r_error.expected = m_type; \ return; \ } #else // !DEBUG_ENABLED #define DEBUG_VALIDATE_ARG_COUNT(m_min_count, m_max_count) #define DEBUG_VALIDATE_ARG_TYPE(m_arg, m_type) #define DEBUG_VALIDATE_ARG_CUSTOM(m_arg, m_type, m_cond, m_msg) #endif // DEBUG_ENABLED #define VALIDATE_ARG_CUSTOM(m_arg, m_type, m_cond, m_msg) \ if (unlikely(m_cond)) { \ *r_ret = m_msg; \ r_error.error = Callable::CallError::CALL_ERROR_INVALID_ARGUMENT; \ r_error.argument = m_arg; \ r_error.expected = m_type; \ return; \ } #define GDFUNC_FAIL_COND_MSG(m_cond, m_msg) \ if (unlikely(m_cond)) { \ *r_ret = m_msg; \ r_error.error = Callable::CallError::CALL_ERROR_INVALID_METHOD; \ return; \ } struct GDScriptUtilityFunctionsDefinitions { #ifndef DISABLE_DEPRECATED static inline void convert(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) { DEBUG_VALIDATE_ARG_COUNT(2, 2); DEBUG_VALIDATE_ARG_TYPE(1, Variant::INT); int type = *p_args[1]; DEBUG_VALIDATE_ARG_CUSTOM(1, Variant::INT, type < 0 || type >= Variant::VARIANT_MAX, RTR("Invalid type argument to convert(), use TYPE_* constants.")); Variant::construct(Variant::Type(type), *r_ret, p_args, 1, r_error); } #endif // DISABLE_DEPRECATED static inline void type_exists(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) { DEBUG_VALIDATE_ARG_COUNT(1, 1); DEBUG_VALIDATE_ARG_TYPE(0, Variant::STRING_NAME); *r_ret = ClassDB::class_exists(*p_args[0]); } static inline void _char(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) { DEBUG_VALIDATE_ARG_COUNT(1, 1); DEBUG_VALIDATE_ARG_TYPE(0, Variant::INT); char32_t result[2] = { *p_args[0], 0 }; *r_ret = String(result); } static inline void range(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) { DEBUG_VALIDATE_ARG_COUNT(1, 3); switch (p_arg_count) { case 1: { DEBUG_VALIDATE_ARG_TYPE(0, Variant::INT); int count = *p_args[0]; Array arr; if (count <= 0) { *r_ret = arr; return; } Error err = arr.resize(count); GDFUNC_FAIL_COND_MSG(err != OK, RTR("Cannot resize array.")); for (int i = 0; i < count; i++) { arr[i] = i; } *r_ret = arr; } break; case 2: { DEBUG_VALIDATE_ARG_TYPE(0, Variant::INT); DEBUG_VALIDATE_ARG_TYPE(1, Variant::INT); int from = *p_args[0]; int to = *p_args[1]; Array arr; if (from >= to) { *r_ret = arr; return; } Error err = arr.resize(to - from); GDFUNC_FAIL_COND_MSG(err != OK, RTR("Cannot resize array.")); for (int i = from; i < to; i++) { arr[i - from] = i; } *r_ret = arr; } break; case 3: { DEBUG_VALIDATE_ARG_TYPE(0, Variant::INT); DEBUG_VALIDATE_ARG_TYPE(1, Variant::INT); DEBUG_VALIDATE_ARG_TYPE(2, Variant::INT); int from = *p_args[0]; int to = *p_args[1]; int incr = *p_args[2]; VALIDATE_ARG_CUSTOM(2, Variant::INT, incr == 0, RTR("Step argument is zero!")); Array arr; if (from >= to && incr > 0) { *r_ret = arr; return; } if (from <= to && incr < 0) { *r_ret = arr; return; } // Calculate how many. int count = 0; if (incr > 0) { count = Math::division_round_up(to - from, incr); } else { count = Math::division_round_up(from - to, -incr); } Error err = arr.resize(count); GDFUNC_FAIL_COND_MSG(err != OK, RTR("Cannot resize array.")); if (incr > 0) { int idx = 0; for (int i = from; i < to; i += incr) { arr[idx++] = i; } } else { int idx = 0; for (int i = from; i > to; i += incr) { arr[idx++] = i; } } *r_ret = arr; } break; } } static inline void load(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) { DEBUG_VALIDATE_ARG_COUNT(1, 1); DEBUG_VALIDATE_ARG_TYPE(0, Variant::STRING); *r_ret = ResourceLoader::load(*p_args[0]); } static inline void inst_to_dict(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) { DEBUG_VALIDATE_ARG_COUNT(1, 1); DEBUG_VALIDATE_ARG_TYPE(0, Variant::OBJECT); if (p_args[0]->get_type() == Variant::NIL) { *r_ret = Variant(); return; } Object *obj = *p_args[0]; if (!obj) { *r_ret = Variant(); return; } VALIDATE_ARG_CUSTOM(0, Variant::OBJECT, !obj->get_script_instance() || obj->get_script_instance()->get_language() != GDScriptLanguage::get_singleton(), RTR("Not a script with an instance.")); GDScriptInstance *inst = static_cast(obj->get_script_instance()); Ref base = inst->get_script(); VALIDATE_ARG_CUSTOM(0, Variant::OBJECT, base.is_null(), RTR("Not based on a script.")); GDScript *p = base.ptr(); String path = p->get_script_path(); Vector sname; while (p->_owner) { sname.push_back(p->local_name); p = p->_owner; } sname.reverse(); VALIDATE_ARG_CUSTOM(0, Variant::OBJECT, !path.is_resource_file(), RTR("Not based on a resource file.")); NodePath cp(sname, Vector(), false); Dictionary d; d["@subpath"] = cp; d["@path"] = path; for (const KeyValue &E : base->member_indices) { if (!d.has(E.key)) { d[E.key] = inst->members[E.value.index]; } } *r_ret = d; } static inline void dict_to_inst(Variant *r_ret, const Variant **p_args, int p_arg_count, Callable::CallError &r_error) { DEBUG_VALIDATE_ARG_COUNT(1, 1); DEBUG_VALIDATE_ARG_TYPE(0, Variant::DICTIONARY); Dictionary d = *p_args[0]; VALIDATE_ARG_CUSTOM(0, Variant::DICTIONARY, !d.has("@path"), RTR("Invalid instance dictionary format (missing @path).")); Ref