godot_voxel/streams/vox_loader.cpp

#include "vox_loader.h"
#include "../meshers/cubes/voxel_color_palette.h"
#include "../storage/voxel_buffer.h"
#include "vox_data.h"

Error VoxelVoxLoader::load_from_file(String fpath, Ref<VoxelBuffer> p_voxels, Ref<VoxelColorPalette> palette) {
	ERR_FAIL_COND_V(p_voxels.is_null(), ERR_INVALID_PARAMETER);
	VoxelBufferInternal &voxels = p_voxels->get_buffer();

	vox::Data data;
	Error load_err = data.load_from_file(fpath);
	ERR_FAIL_COND_V(load_err != OK, load_err);

	const vox::Model &model = data.get_model(0);

	const VoxelBufferInternal::ChannelId channel = VoxelBufferInternal::CHANNEL_COLOR;
	Span<const Color8> src_palette = to_span_const(data.get_palette());
	const VoxelBufferInternal::Depth depth = voxels.get_channel_depth(VoxelBufferInternal::CHANNEL_COLOR);

	Span<uint8_t> dst_raw;
	voxels.create(model.size);
	voxels.decompress_channel(channel);
	CRASH_COND(!voxels.get_channel_raw(channel, dst_raw));

	if (palette.is_valid()) {
		for (size_t i = 0; i < src_palette.size(); ++i) {
			palette->set_color8(i, src_palette[i]);
		}

		switch (depth) {
			case VoxelBufferInternal::DEPTH_8_BIT: {
				memcpy(dst_raw.data(), model.color_indexes.data(), model.color_indexes.size());
			} break;

			case VoxelBufferInternal::DEPTH_16_BIT: {
				Span<uint16_t> dst = dst_raw.reinterpret_cast_to<uint16_t>();
				for (size_t i = 0; i < dst.size(); ++i) {
					dst[i] = model.color_indexes[i];
				}
			} break;

			default:
				ERR_FAIL_V_MSG(ERR_INVALID_PARAMETER, "Unsupported depth");
				break;
		}

	} else {
		switch (depth) {
			case VoxelBufferInternal::DEPTH_8_BIT: {
				for (size_t i = 0; i < dst_raw.size(); ++i) {
					const uint8_t ci = model.color_indexes[i];
					dst_raw[i] = src_palette[ci].to_u8();
				}
			} break;

			case VoxelBufferInternal::DEPTH_16_BIT: {
				Span<uint16_t> dst = dst_raw.reinterpret_cast_to<uint16_t>();
				for (size_t i = 0; i < dst.size(); ++i) {
					const uint8_t ci = model.color_indexes[i];
					dst[i] = src_palette[ci].to_u16();
				}
			} break;

			default:
				ERR_FAIL_V_MSG(ERR_INVALID_PARAMETER, "Unsupported depth");
				break;
		}
	}

	return load_err;
}

void VoxelVoxLoader::_bind_methods() {
	ClassDB::bind_method(D_METHOD("load_from_file", "fpath", "voxels"), &VoxelVoxLoader::load_from_file);
}