#include <gvox/gvox.h>
#include <cstdlib>
#include <bit>
#include <array>
#include <vector>
struct RunLengthEncodingUserState {
GvoxRegionRange range{};
std::vector<uint32_t> voxels;
std::vector<uint8_t> channels;
size_t offset{};
};
extern "C" void gvox_serialize_adapter_gvox_run_length_encoding_create(GvoxAdapterContext *ctx, void const * ) {
auto *user_state_ptr = malloc(sizeof(RunLengthEncodingUserState));
new (user_state_ptr) RunLengthEncodingUserState();
gvox_adapter_set_user_pointer(ctx, user_state_ptr);
}
extern "C" void gvox_serialize_adapter_gvox_run_length_encoding_destroy(GvoxAdapterContext *ctx) {
auto &user_state = *static_cast<RunLengthEncodingUserState *>(gvox_adapter_get_user_pointer(ctx));
user_state.~RunLengthEncodingUserState();
free(&user_state);
}
extern "C" void gvox_serialize_adapter_gvox_run_length_encoding_blit_begin(GvoxBlitContext *blit_ctx, GvoxAdapterContext *ctx, GvoxRegionRange const *range, uint32_t channel_flags) {
auto &user_state = *static_cast<RunLengthEncodingUserState *>(gvox_adapter_get_user_pointer(ctx));
user_state.offset = 0;
user_state.range = *range;
auto magic = std::bit_cast<uint32_t>(std::array<char, 4>{'r', 'l', 'e', '\0'});
gvox_output_write(blit_ctx, user_state.offset, sizeof(uint32_t), &magic);
user_state.offset += sizeof(magic);
gvox_output_write(blit_ctx, user_state.offset, sizeof(*range), range);
user_state.offset += sizeof(*range);
gvox_output_write(blit_ctx, user_state.offset, sizeof(channel_flags), &channel_flags);
user_state.offset += sizeof(channel_flags);
user_state.channels.resize(static_cast<size_t>(std::popcount(channel_flags)));
uint32_t next_channel = 0;
for (uint8_t channel_i = 0; channel_i < 32; ++channel_i) {
if ((channel_flags & (1u << channel_i)) != 0) {
user_state.channels[next_channel] = channel_i;
++next_channel;
}
}
user_state.voxels.resize(user_state.channels.size() * range->extent.x * range->extent.y * range->extent.z);
}
extern "C" void gvox_serialize_adapter_gvox_run_length_encoding_blit_end(GvoxBlitContext *blit_ctx, GvoxAdapterContext *ctx) {
auto &user_state = *static_cast<RunLengthEncodingUserState *>(gvox_adapter_get_user_pointer(ctx));
auto output = std::vector<uint32_t>{};
output.resize(user_state.range.extent.x * user_state.range.extent.y);
for (uint32_t yi = 0; yi < user_state.range.extent.y; ++yi) {
for (uint32_t xi = 0; xi < user_state.range.extent.x; ++xi) {
auto pointer_index = xi + yi * user_state.range.extent.x;
auto column = std::vector<uint32_t>{};
auto index = (xi + yi * user_state.range.extent.x + 0 * user_state.range.extent.x * user_state.range.extent.y) * user_state.channels.size();
for (uint32_t ci = 0; ci < user_state.channels.size(); ++ci) {
column.push_back(user_state.voxels[index + ci]);
}
column.push_back(1);
for (uint32_t zi = 1; zi < user_state.range.extent.z; ++zi) {
index = (xi + yi * user_state.range.extent.x + zi * user_state.range.extent.x * user_state.range.extent.y) * user_state.channels.size();
auto col_back_index = column.size() / (user_state.channels.size() + 1) - 1;
bool matched = true;
for (uint32_t ci = 0; ci < user_state.channels.size(); ++ci) {
auto const &test_val = user_state.voxels[index + ci];
auto const &last_val = column[col_back_index * (user_state.channels.size() + 1) + ci];
if (test_val != last_val) {
matched = false;
}
}
if (matched) {
column[col_back_index * (user_state.channels.size() + 1) + user_state.channels.size()] += 1;
} else {
for (uint32_t ci = 0; ci < user_state.channels.size(); ++ci) {
column.push_back(user_state.voxels[index + ci]);
}
column.push_back(1);
}
}
auto out_index = output.size();
output[pointer_index] = static_cast<uint32_t>(out_index);
output.insert(output.end(), column.begin(), column.end());
}
}
gvox_output_write(blit_ctx, user_state.offset, output.size() * sizeof(output[0]), output.data());
}
static void handle_region(RunLengthEncodingUserState &user_state, GvoxRegionRange const *range, auto user_func) {
for (uint32_t zi = 0; zi < range->extent.z; ++zi) {
for (uint32_t yi = 0; yi < range->extent.y; ++yi) {
for (uint32_t xi = 0; xi < range->extent.x; ++xi) {
auto const pos = GvoxOffset3D{
static_cast<int32_t>(xi) + range->offset.x,
static_cast<int32_t>(yi) + range->offset.y,
static_cast<int32_t>(zi) + range->offset.z,
};
if (pos.x < user_state.range.offset.x ||
pos.y < user_state.range.offset.y ||
pos.z < user_state.range.offset.z ||
pos.x >= user_state.range.offset.x + static_cast<int32_t>(user_state.range.extent.x) ||
pos.y >= user_state.range.offset.y + static_cast<int32_t>(user_state.range.extent.y) ||
pos.z >= user_state.range.offset.z + static_cast<int32_t>(user_state.range.extent.z)) {
continue;
}
auto output_rel_x = static_cast<size_t>(pos.x - user_state.range.offset.x);
auto output_rel_y = static_cast<size_t>(pos.y - user_state.range.offset.y);
auto output_rel_z = static_cast<size_t>(pos.z - user_state.range.offset.z);
for (uint32_t channel_i = 0; channel_i < user_state.channels.size(); ++channel_i) {
auto output_index = static_cast<size_t>(output_rel_x + output_rel_y * user_state.range.extent.x + output_rel_z * user_state.range.extent.x * user_state.range.extent.y) * user_state.channels.size() + channel_i;
user_func(channel_i, output_index, pos);
}
}
}
}
}
extern "C" void gvox_serialize_adapter_gvox_run_length_encoding_serialize_region(GvoxBlitContext *blit_ctx, GvoxAdapterContext *ctx, GvoxRegionRange const *range, uint32_t ) {
auto &user_state = *static_cast<RunLengthEncodingUserState *>(gvox_adapter_get_user_pointer(ctx));
handle_region(
user_state, range,
[blit_ctx, &user_state](uint32_t channel_i, size_t output_index, GvoxOffset3D const &pos) {
auto const sample_range = GvoxRegionRange{
.offset = pos,
.extent = GvoxExtent3D{1, 1, 1},
};
auto region = gvox_load_region_range(blit_ctx, &sample_range, 1u << user_state.channels[channel_i]);
auto sample = gvox_sample_region(blit_ctx, ®ion, &pos, user_state.channels[channel_i]);
if (sample.is_present == 0u) {
sample.data = 0u;
}
user_state.voxels[output_index] = sample.data;
gvox_unload_region_range(blit_ctx, ®ion, &sample_range);
});
}
extern "C" void gvox_serialize_adapter_gvox_run_length_encoding_receive_region(GvoxBlitContext *blit_ctx, GvoxAdapterContext *ctx, GvoxRegion const *region) {
auto &user_state = *static_cast<RunLengthEncodingUserState *>(gvox_adapter_get_user_pointer(ctx));
handle_region(
user_state, ®ion->range,
[blit_ctx, region, &user_state](uint32_t channel_i, size_t output_index, GvoxOffset3D const &pos) {
auto sample = gvox_sample_region(blit_ctx, region, &pos, user_state.channels[channel_i]);
if (sample.is_present != 0u) {
user_state.voxels[output_index] = sample.data;
}
});
}