PulseStudio/VoxelEngine
PulseStudio/VoxelEngine
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases 1 Wiki Activity

add metadata to chunk data used in world.get/set_chunk_data

Browse Source
This commit is contained in:
MihailRis 2025-01-12 01:47:07 +03:00
parent 077de25acc
commit c5fa6935a6
1 changed files with 41 additions and 59 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

100
src/logic/scripting/lua/libs/libworld.cpp
View File

@ -7,6 +7,7 @@
#include "coders/compression.hpp"
#include "coders/gzip.hpp"
#include "coders/json.hpp"
#include "coders/byte_utils.hpp"
#include "coders/rle.hpp"
#include "engine/Engine.hpp"
#include "files/engine_paths.hpp"
@ -123,39 +124,29 @@ static int l_get_generator(lua::State* L) {
return lua::pushstring(L, require_world_info().generator);
}
static std::vector<ubyte> prepare_chunk_data(const Chunk& chunk, bool compress) {
static std::vector<ubyte> prepare_chunk_data(const Chunk& chunk) {
auto data = chunk.encode();
std::vector<ubyte> chunkData;
if (compress) {
// world.get_chunk_data is only available in the main Lua state
static util::Buffer<ubyte> rleBuffer;
if (rleBuffer.size() < CHUNK_DATA_LEN * 2) {
rleBuffer = util::Buffer<ubyte>(CHUNK_DATA_LEN * 2);
}
size_t rleCompressedSize =
extrle::encode16(data.get(), CHUNK_DATA_LEN, rleBuffer.data());
const auto gzipCompressedData = gzip::compress(
rleBuffer.data(), rleCompressedSize
);
auto tmp = dataio::h2le(rleCompressedSize);
chunkData.reserve(gzipCompressedData.size() + sizeof(tmp));
chunkData.insert(
chunkData.begin() + 0, (char*)&tmp, ((char*)&tmp) + sizeof(tmp)
);
chunkData.insert(
chunkData.begin() + sizeof(tmp),
gzipCompressedData.data(),
gzipCompressedData.data() + gzipCompressedData.size()
);
} else {
chunkData.reserve(CHUNK_DATA_LEN);
chunkData.insert(
chunkData.begin(), data.get(), data.get() + CHUNK_DATA_LEN
);
/// world.get_chunk_data is only available in the main Lua state
static util::Buffer<ubyte> rleBuffer;
if (rleBuffer.size() < CHUNK_DATA_LEN * 2) {
rleBuffer = util::Buffer<ubyte>(CHUNK_DATA_LEN * 2);
}
return chunkData;
size_t rleCompressedSize =
extrle::encode16(data.get(), CHUNK_DATA_LEN, rleBuffer.data());
const auto gzipCompressedData = gzip::compress(
rleBuffer.data(), rleCompressedSize
);
auto metadataBytes = chunk.blocksMetadata.serialize();
ByteBuilder builder(2 + 8 + gzipCompressedData.size() + metadataBytes.size());
builder.putInt16(0); // header
builder.putInt32(gzipCompressedData.size());
builder.put(gzipCompressedData.data(), gzipCompressedData.size());
builder.putInt32(metadataBytes.size());
builder.put(metadataBytes.data(), metadataBytes.size());
return builder.build();
}
static int l_get_chunk_data(lua::State* L) {
@ -166,11 +157,7 @@ static int l_get_chunk_data(lua::State* L) {
lua::pushnil(L);
return 0;
}
bool compress = true;
if (lua::gettop(L) >= 3) {
compress = lua::toboolean(L, 3);
}
auto chunkData = prepare_chunk_data(*chunk, compress);
auto chunkData = prepare_chunk_data(*chunk);
return lua::newuserdata<lua::LuaBytearray>(L, std::move(chunkData));
}
@ -178,33 +165,29 @@ static int l_set_chunk_data(lua::State* L) {
int x = static_cast<int>(lua::tointeger(L, 1));
int y = static_cast<int>(lua::tointeger(L, 2));
auto buffer = lua::touserdata<lua::LuaBytearray>(L, 3);
bool isCompressed = true;
if (lua::gettop(L) >= 4) {
isCompressed = lua::toboolean(L, 4);
}
auto chunk = level->chunks->getChunk(x, y);
if (chunk == nullptr) {
return 0;
}
if (isCompressed) {
std::vector<ubyte>& rawData = buffer->data();
size_t gzipDecompressedSize = dataio::le2h(*(size_t*)(rawData.data()));
auto rleData = compression::decompress(
rawData.data() + sizeof(gzipDecompressedSize),
buffer->data().size() - sizeof(gzipDecompressedSize),
gzipDecompressedSize,
compression::Method::GZIP
);
auto data = compression::decompress(
rleData.get(),
gzipDecompressedSize,
CHUNK_DATA_LEN,
compression::Method::EXTRLE16
);
chunk->decode(data.get());
} else {
chunk->decode(buffer->data().data());
}
std::vector<ubyte>& rawData = buffer->data();
ByteReader reader(rawData.data(), rawData.size());
uint16_t header = reader.getInt16();
size_t gzipCompressedSize = reader.getInt32();
auto rleData = gzip::decompress(reader.pointer(), gzipCompressedSize);
reader.skip(gzipCompressedSize);
/// world.get_chunk_data is only available in the main Lua state
static util::Buffer<ubyte> voxelData (CHUNK_DATA_LEN);
extrle::decode16(rleData.data(), rleData.size(), voxelData.data());
chunk->decode(voxelData.data());
size_t metadataSize = reader.getInt32();
chunk->blocksMetadata.deserialize(reader.pointer(), metadataSize);
reader.skip(metadataSize);
chunk->setModifiedAndUnsaved();
chunk->updateHeights();
@ -233,7 +216,6 @@ static int l_set_chunk_data(lua::State* L) {
}
}
}
return lua::pushboolean(L, true);
}