input_network_encoder.cpp

#include "input_network_encoder.h"

#include "scene_synchronizer.h"

void InputNetworkEncoder::_bind_methods() {
	ClassDB::bind_method(D_METHOD("register_input", "name", "default_value", "type", "compression_level"), &InputNetworkEncoder::register_input);
	ClassDB::bind_method(D_METHOD("find_input_id", "name"), &InputNetworkEncoder::find_input_id);
	ClassDB::bind_method(D_METHOD("encode", "inputs", "buffer"), &InputNetworkEncoder::script_encode);
	ClassDB::bind_method(D_METHOD("decode", "buffer"), &InputNetworkEncoder::script_decode);
	ClassDB::bind_method(D_METHOD("get_defaults"), &InputNetworkEncoder::script_get_defaults);
	ClassDB::bind_method(D_METHOD("are_different", "buffer_a", "buffer_b"), &InputNetworkEncoder::script_are_different);
	ClassDB::bind_method(D_METHOD("count_size", "buffer"), &InputNetworkEncoder::script_count_size);
}

uint32_t InputNetworkEncoder::register_input(
		const StringName &p_name,
		const Variant &p_default_value,
		DataBuffer::DataType p_type,
		DataBuffer::CompressionLevel p_compression_level,
		real_t p_comparison_floating_point_precision) {
	switch (p_type) {
		case DataBuffer::DATA_TYPE_BOOL:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::BOOL, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `BOOL` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_INT:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::INT, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `INT` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_REAL:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::FLOAT, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `REAL` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_POSITIVE_UNIT_REAL:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::FLOAT, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `REAL` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_UNIT_REAL:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::FLOAT, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `REAL` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_VECTOR2:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::VECTOR2, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `Vector2` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR2:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::VECTOR2, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `Vector2` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_VECTOR3:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::VECTOR3, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `Vector3` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR3:
			ERR_FAIL_COND_V_MSG(p_default_value.get_type() != Variant::VECTOR3, UINT32_MAX, "The moveset initialization failed for" + p_name + " the specified data type is `Vector3` but the default parameter is " + itos(p_default_value.get_type()));
			break;
		case DataBuffer::DATA_TYPE_VARIANT:
			/* No need to check variant, anything is accepted at this point.*/
			break;
	};

	const uint32_t index = input_info.size();

	input_info.resize(input_info.size() + 1);
	input_info[index].name = p_name;
	input_info[index].default_value = p_default_value;
	input_info[index].data_type = p_type;
	input_info[index].compression_level = p_compression_level;
	input_info[index].comparison_floating_point_precision = p_comparison_floating_point_precision;

	return index;
}

uint32_t InputNetworkEncoder::find_input_id(const StringName &p_name) const {
	for (uint32_t i = 0; i < input_info.size(); i += 1) {
		if (input_info[i].name == p_name) {
			return i;
		}
	}
	return INDEX_NONE;
}

const LocalVector<NetworkedInputInfo> &InputNetworkEncoder::get_input_info() const {
	return input_info;
}

void InputNetworkEncoder::encode(const LocalVector<Variant> &p_input, DataBuffer &r_buffer) const {
	for (uint32_t i = 0; i < input_info.size(); i += 1) {
		const NetworkedInputInfo &info = input_info[i];

		const bool is_default =
				// If the input exist into the array.
				i >= p_input.size() ||
				// Use default if the variable type is different.
				info.default_value.get_type() != p_input[i].get_type() ||
				// Use default if the variable value is equal to default.
				info.default_value == p_input[i];

		if (info.default_value.get_type() != Variant::BOOL) {
			r_buffer.add_bool(is_default);

			if (!is_default) {
				const Variant &pending_input = p_input[i];
				switch (info.data_type) {
					case DataBuffer::DATA_TYPE_BOOL:
						CRASH_NOW_MSG("Boolean are handled differently. Thanks to the above IF this condition never occurs.");
						break;
					case DataBuffer::DATA_TYPE_INT:
						r_buffer.add_int(pending_input.operator int(), info.compression_level);
						break;
					case DataBuffer::DATA_TYPE_REAL:
						r_buffer.add_real(pending_input.operator real_t(), info.compression_level);
						break;
					case DataBuffer::DATA_TYPE_POSITIVE_UNIT_REAL:
						r_buffer.add_positive_unit_real(pending_input.operator real_t(), info.compression_level);
						break;
					case DataBuffer::DATA_TYPE_UNIT_REAL:
						r_buffer.add_unit_real(pending_input.operator real_t(), info.compression_level);
						break;
					case DataBuffer::DATA_TYPE_VECTOR2:
						_ALLOW_DISCARD_ r_buffer.add_vector2(pending_input.operator Vector2(), info.compression_level);
						break;
					case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR2:
						_ALLOW_DISCARD_ r_buffer.add_normalized_vector2(pending_input.operator Vector2(), info.compression_level);
						break;
					case DataBuffer::DATA_TYPE_VECTOR3:
						_ALLOW_DISCARD_ r_buffer.add_vector3(pending_input.operator Vector3(), info.compression_level);
						break;
					case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR3:
						_ALLOW_DISCARD_ r_buffer.add_normalized_vector3(pending_input.operator Vector3(), info.compression_level);
						break;
					case DataBuffer::DATA_TYPE_VARIANT:
						_ALLOW_DISCARD_ r_buffer.add_variant(pending_input);
						break;
				};
			}
		} else {
			// If the data is bool no need to set the default.
			if (!is_default) {
				r_buffer.add_bool(p_input[i].operator bool());
			} else {
				r_buffer.add_bool(info.default_value.operator bool());
			}
		}
	}
}

void InputNetworkEncoder::decode(DataBuffer &p_buffer, LocalVector<Variant> &r_inputs) const {
	if (r_inputs.size() < input_info.size()) {
		r_inputs.resize(input_info.size());
	}

	for (uint32_t i = 0; i < input_info.size(); i += 1) {
		const NetworkedInputInfo &info = input_info[i];

		const bool is_bool = info.default_value.get_type() == Variant::BOOL;

		bool is_default = false;
		if (is_bool == false) {
			is_default = p_buffer.read_bool();
		}

		if (is_default) {
			r_inputs[i] = info.default_value;
		} else {
			switch (info.data_type) {
				case DataBuffer::DATA_TYPE_BOOL:
					r_inputs[i] = p_buffer.read_bool();
					break;
				case DataBuffer::DATA_TYPE_INT:
					r_inputs[i] = p_buffer.read_int(info.compression_level);
					break;
				case DataBuffer::DATA_TYPE_REAL:
					r_inputs[i] = p_buffer.read_real(info.compression_level);
					break;
				case DataBuffer::DATA_TYPE_POSITIVE_UNIT_REAL:
					r_inputs[i] = p_buffer.read_positive_unit_real(info.compression_level);
					break;
				case DataBuffer::DATA_TYPE_UNIT_REAL:
					r_inputs[i] = p_buffer.read_unit_real(info.compression_level);
					break;
				case DataBuffer::DATA_TYPE_VECTOR2:
					r_inputs[i] = p_buffer.read_vector2(info.compression_level);
					break;
				case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR2:
					r_inputs[i] = p_buffer.read_normalized_vector2(info.compression_level);
					break;
				case DataBuffer::DATA_TYPE_VECTOR3:
					r_inputs[i] = p_buffer.read_vector3(info.compression_level);
					break;
				case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR3:
					r_inputs[i] = p_buffer.read_normalized_vector3(info.compression_level);
					break;
				case DataBuffer::DATA_TYPE_VARIANT:
					r_inputs[i] = p_buffer.read_variant();
					break;
			};
		}
	}
}

void InputNetworkEncoder::reset_inputs_to_defaults(LocalVector<Variant> &r_input) const {
	const uint32_t size = r_input.size() < input_info.size() ? r_input.size() : input_info.size();

	for (uint32_t i = 0; i < size; i += 1) {
		r_input[i] = input_info[i].default_value;
	}
}

bool InputNetworkEncoder::are_different(DataBuffer &p_buffer_A, DataBuffer &p_buffer_B) const {
	for (uint32_t i = 0; i < input_info.size(); i += 1) {
		const NetworkedInputInfo &info = input_info[i];

		const bool is_bool = info.default_value.get_type() == Variant::BOOL;

		bool is_default_A = false;
		bool is_default_B = false;
		if (is_bool == false) {
			is_default_A = p_buffer_A.read_bool();
			is_default_B = p_buffer_B.read_bool();
		}

		bool are_equals = true;
		if (is_default_A && is_default_B) {
			are_equals = true;
		} else {
			switch (info.data_type) {
				case DataBuffer::DATA_TYPE_BOOL:
					are_equals = p_buffer_A.read_bool() == p_buffer_B.read_bool();
					break;
				case DataBuffer::DATA_TYPE_INT:
					are_equals = Math::is_equal_approx(p_buffer_A.read_int(info.compression_level), p_buffer_B.read_int(info.compression_level), info.comparison_floating_point_precision);
					break;
				case DataBuffer::DATA_TYPE_REAL:
					are_equals = Math::is_equal_approx(static_cast<real_t>(p_buffer_A.read_real(info.compression_level)), static_cast<real_t>(p_buffer_B.read_real(info.compression_level)), info.comparison_floating_point_precision);
					break;
				case DataBuffer::DATA_TYPE_POSITIVE_UNIT_REAL:
					are_equals = Math::is_equal_approx(p_buffer_A.read_positive_unit_real(info.compression_level), p_buffer_B.read_positive_unit_real(info.compression_level), info.comparison_floating_point_precision);
					break;
				case DataBuffer::DATA_TYPE_UNIT_REAL:
					are_equals = Math::is_equal_approx(p_buffer_A.read_unit_real(info.compression_level), p_buffer_B.read_unit_real(info.compression_level), info.comparison_floating_point_precision);
					break;
				case DataBuffer::DATA_TYPE_VECTOR2:
					are_equals = SceneSynchronizer::compare(p_buffer_A.read_vector2(info.compression_level), p_buffer_B.read_vector2(info.compression_level), info.comparison_floating_point_precision);
					break;
				case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR2:
					are_equals = SceneSynchronizer::compare(p_buffer_A.read_normalized_vector2(info.compression_level), p_buffer_B.read_normalized_vector2(info.compression_level), info.comparison_floating_point_precision);
					break;
				case DataBuffer::DATA_TYPE_VECTOR3:
					are_equals = SceneSynchronizer::compare(p_buffer_A.read_vector3(info.compression_level), p_buffer_B.read_vector3(info.compression_level), info.comparison_floating_point_precision);
					break;
				case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR3:
					are_equals = SceneSynchronizer::compare(p_buffer_A.read_normalized_vector3(info.compression_level), p_buffer_B.read_normalized_vector3(info.compression_level), info.comparison_floating_point_precision);
					break;
				case DataBuffer::DATA_TYPE_VARIANT:
					are_equals = SceneSynchronizer::compare(p_buffer_A.read_variant(), p_buffer_B.read_variant(), info.comparison_floating_point_precision);
					break;
			};
		}

		if (!are_equals) {
			return true;
		}
	}

	return false;
}

uint32_t InputNetworkEncoder::count_size(DataBuffer &p_buffer) const {
	int size = 0;
	for (uint32_t i = 0; i < input_info.size(); i += 1) {
		const NetworkedInputInfo &info = input_info[i];

		const bool is_bool = info.default_value.get_type() == Variant::BOOL;

		if (is_bool == false) {
			size += p_buffer.read_bool_size();
		}

		switch (info.data_type) {
			case DataBuffer::DATA_TYPE_BOOL:
				size += p_buffer.read_bool_size();
				break;
			case DataBuffer::DATA_TYPE_INT:
				size += p_buffer.read_int_size(info.compression_level);
				break;
			case DataBuffer::DATA_TYPE_REAL:
				size += p_buffer.read_real_size(info.compression_level);
				break;
			case DataBuffer::DATA_TYPE_POSITIVE_UNIT_REAL:
				size += p_buffer.read_positive_unit_real_size(info.compression_level);
				break;
			case DataBuffer::DATA_TYPE_UNIT_REAL:
				size += p_buffer.read_unit_real_size(info.compression_level);
				break;
			case DataBuffer::DATA_TYPE_VECTOR2:
				size += p_buffer.read_vector2_size(info.compression_level);
				break;
			case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR2:
				size += p_buffer.read_normalized_vector2_size(info.compression_level);
				break;
			case DataBuffer::DATA_TYPE_VECTOR3:
				size += p_buffer.read_vector3_size(info.compression_level);
				break;
			case DataBuffer::DATA_TYPE_NORMALIZED_VECTOR3:
				size += p_buffer.read_normalized_vector3_size(info.compression_level);
				break;
			case DataBuffer::DATA_TYPE_VARIANT:
				size += p_buffer.read_variant_size();
				break;
		};
	}

	return size;
}

void InputNetworkEncoder::script_encode(const Array &p_inputs, Object *r_buffer) const {
	ERR_FAIL_COND(r_buffer == nullptr);
	DataBuffer *db = Object::cast_to<DataBuffer>(r_buffer);
	ERR_FAIL_COND(db == nullptr);

	LocalVector<Variant> inputs;
	inputs.resize(p_inputs.size());
	for (int i = 0; i < p_inputs.size(); i += 1) {
		inputs[i] = p_inputs[i];
	}

	encode(inputs, *db);
}

Array InputNetworkEncoder::script_decode(Object *p_buffer) const {
	ERR_FAIL_COND_V(p_buffer == nullptr, Array());
	DataBuffer *db = Object::cast_to<DataBuffer>(p_buffer);
	ERR_FAIL_COND_V(db == nullptr, Array());

	LocalVector<Variant> inputs;
	decode(*db, inputs);

	Array out;
	out.resize(inputs.size());
	for (uint32_t i = 0; i < inputs.size(); i += 1) {
		out[i] = inputs[i];
	}
	return out;
}

Array InputNetworkEncoder::script_get_defaults() const {
	LocalVector<Variant> inputs;
	inputs.resize(input_info.size());

	reset_inputs_to_defaults(inputs);

	Array out;
	out.resize(inputs.size());
	for (uint32_t i = 0; i < inputs.size(); i += 1) {
		out[i] = inputs[i];
	}
	return out;
}

bool InputNetworkEncoder::script_are_different(Object *p_buffer_A, Object *p_buffer_B) const {
	ERR_FAIL_COND_V(p_buffer_A == nullptr, true);
	DataBuffer *db_A = Object::cast_to<DataBuffer>(p_buffer_A);
	ERR_FAIL_COND_V(db_A == nullptr, false);

	ERR_FAIL_COND_V(p_buffer_B == nullptr, true);
	DataBuffer *db_B = Object::cast_to<DataBuffer>(p_buffer_B);
	ERR_FAIL_COND_V(db_B == nullptr, true);

	return are_different(*db_A, *db_B);
}

uint32_t InputNetworkEncoder::script_count_size(Object *p_buffer) const {
	ERR_FAIL_COND_V(p_buffer == nullptr, 0);
	DataBuffer *db = Object::cast_to<DataBuffer>(p_buffer);
	ERR_FAIL_COND_V(db == nullptr, 0);

	return count_size(*db);
}