242 lines
7.5 KiB
C++
242 lines
7.5 KiB
C++
#include "WorldRegions.hpp"
|
|
|
|
#include <cstring>
|
|
|
|
#include "util/data_io.hpp"
|
|
|
|
#define REGION_FORMAT_MAGIC ".VOXREG"
|
|
|
|
static fs::path get_region_filename(int x, int z) {
|
|
return fs::path(std::to_string(x) + "_" + std::to_string(z) + ".bin");
|
|
}
|
|
|
|
/// @brief Read missing chunks data (null pointers) from region file
|
|
static void fetch_chunks(WorldRegion* region, int x, int z, regfile* file) {
|
|
auto* chunks = region->getChunks();
|
|
auto sizes = region->getSizes();
|
|
|
|
for (size_t i = 0; i < REGION_CHUNKS_COUNT; i++) {
|
|
int chunk_x = (i % REGION_SIZE) + x * REGION_SIZE;
|
|
int chunk_z = (i / REGION_SIZE) + z * REGION_SIZE;
|
|
if (chunks[i] == nullptr) {
|
|
chunks[i] = RegionsLayer::readChunkData(
|
|
chunk_x, chunk_z, sizes[i][0], sizes[i][1], file);
|
|
}
|
|
}
|
|
}
|
|
|
|
regfile::regfile(fs::path filename) : file(std::move(filename)) {
|
|
if (file.length() < REGION_HEADER_SIZE)
|
|
throw std::runtime_error("incomplete region file header");
|
|
char header[REGION_HEADER_SIZE];
|
|
file.read(header, REGION_HEADER_SIZE);
|
|
|
|
// avoid of use strcmp_s
|
|
if (std::string(header, std::strlen(REGION_FORMAT_MAGIC)) !=
|
|
REGION_FORMAT_MAGIC) {
|
|
throw std::runtime_error("invalid region file magic number");
|
|
}
|
|
version = header[8];
|
|
if (static_cast<uint>(version) > REGION_FORMAT_VERSION) {
|
|
throw illegal_region_format(
|
|
"region format " + std::to_string(version) + " is not supported"
|
|
);
|
|
}
|
|
}
|
|
|
|
std::unique_ptr<ubyte[]> regfile::read(int index, uint32_t& size, uint32_t& srcSize) {
|
|
size_t file_size = file.length();
|
|
size_t table_offset = file_size - REGION_CHUNKS_COUNT * 4;
|
|
|
|
uint32_t buff32;
|
|
file.seekg(table_offset + index * 4);
|
|
file.read(reinterpret_cast<char*>(&buff32), 4);
|
|
uint32_t offset = dataio::le2h(buff32);
|
|
if (offset == 0) {
|
|
return nullptr;
|
|
}
|
|
|
|
file.seekg(offset);
|
|
file.read(reinterpret_cast<char*>(&buff32), 4);
|
|
size = dataio::le2h(buff32);
|
|
file.read(reinterpret_cast<char*>(&buff32), 4);
|
|
srcSize = dataio::le2h(buff32);
|
|
|
|
auto data = std::make_unique<ubyte[]>(size);
|
|
file.read(reinterpret_cast<char*>(data.get()), size);
|
|
return data;
|
|
}
|
|
|
|
|
|
void RegionsLayer::closeRegFile(glm::ivec2 coord) {
|
|
openRegFiles.erase(coord);
|
|
regFilesCv.notify_one();
|
|
}
|
|
|
|
regfile_ptr RegionsLayer::useRegFile(glm::ivec2 coord) {
|
|
auto* file = openRegFiles[coord].get();
|
|
file->inUse = true;
|
|
return regfile_ptr(file, ®FilesCv);
|
|
}
|
|
|
|
// Marks regfile as used and unmarks when shared_ptr dies
|
|
regfile_ptr RegionsLayer::getRegFile(glm::ivec2 coord, bool create) {
|
|
{
|
|
std::lock_guard lock(regFilesMutex);
|
|
const auto found = openRegFiles.find(coord);
|
|
if (found != openRegFiles.end()) {
|
|
if (found->second->inUse) {
|
|
throw std::runtime_error("regfile is currently in use");
|
|
}
|
|
return useRegFile(found->first);
|
|
}
|
|
}
|
|
if (create) {
|
|
return createRegFile(coord);
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
regfile_ptr RegionsLayer::createRegFile(glm::ivec2 coord) {
|
|
auto file = folder / get_region_filename(coord[0], coord[1]);
|
|
if (!fs::exists(file)) {
|
|
return nullptr;
|
|
}
|
|
if (openRegFiles.size() == MAX_OPEN_REGION_FILES) {
|
|
std::unique_lock lock(regFilesMutex);
|
|
while (true) {
|
|
bool closed = false;
|
|
// FIXME: bad choosing algorithm
|
|
for (auto& entry : openRegFiles) {
|
|
if (!entry.second->inUse) {
|
|
closeRegFile(entry.first);
|
|
closed = true;
|
|
break;
|
|
}
|
|
}
|
|
if (closed) {
|
|
break;
|
|
}
|
|
// notified when any regfile gets out of use or closed
|
|
regFilesCv.wait(lock);
|
|
}
|
|
openRegFiles[coord] = std::make_unique<regfile>(file);
|
|
return useRegFile(coord);
|
|
} else {
|
|
std::lock_guard lock(regFilesMutex);
|
|
openRegFiles[coord] = std::make_unique<regfile>(file);
|
|
return useRegFile(coord);
|
|
}
|
|
}
|
|
|
|
WorldRegion* RegionsLayer::getRegion(int x, int z) {
|
|
std::lock_guard lock(mapMutex);
|
|
auto found = regions.find({x, z});
|
|
if (found == regions.end()) {
|
|
return nullptr;
|
|
}
|
|
return found->second.get();
|
|
}
|
|
|
|
fs::path RegionsLayer::getRegionFilePath(int x, int z) const {
|
|
return folder / get_region_filename(x, z);
|
|
}
|
|
|
|
WorldRegion* RegionsLayer::getOrCreateRegion(int x, int z) {
|
|
if (auto region = getRegion(x, z)) {
|
|
return region;
|
|
}
|
|
std::lock_guard lock(mapMutex);
|
|
auto region_ptr = std::make_unique<WorldRegion>();
|
|
auto region = region_ptr.get();
|
|
regions[{x, z}] = std::move(region_ptr);
|
|
return region;
|
|
}
|
|
|
|
ubyte* RegionsLayer::getData(int x, int z, uint32_t& size, uint32_t& srcSize) {
|
|
int regionX, regionZ, localX, localZ;
|
|
calc_reg_coords(x, z, regionX, regionZ, localX, localZ);
|
|
|
|
WorldRegion* region = getOrCreateRegion(regionX, regionZ);
|
|
ubyte* data = region->getChunkData(localX, localZ);
|
|
if (data == nullptr) {
|
|
auto regfile = getRegFile({regionX, regionZ});
|
|
if (regfile != nullptr) {
|
|
auto dataptr = readChunkData(x, z, size, srcSize, regfile.get());
|
|
if (dataptr) {
|
|
data = dataptr.get();
|
|
region->put(localX, localZ, std::move(dataptr), size, srcSize);
|
|
}
|
|
}
|
|
}
|
|
if (data != nullptr) {
|
|
auto sizevec = region->getChunkDataSize(localX, localZ);
|
|
size = sizevec[0];
|
|
srcSize = sizevec[1];
|
|
return data;
|
|
}
|
|
return nullptr;
|
|
}
|
|
|
|
void RegionsLayer::writeRegion(int x, int z, WorldRegion* entry) {
|
|
fs::path filename = folder / get_region_filename(x, z);
|
|
|
|
glm::ivec2 regcoord(x, z);
|
|
if (auto regfile = getRegFile(regcoord, false)) {
|
|
fetch_chunks(entry, x, z, regfile.get());
|
|
|
|
std::lock_guard lock(regFilesMutex);
|
|
regfile.reset();
|
|
closeRegFile(regcoord);
|
|
}
|
|
|
|
char header[REGION_HEADER_SIZE] = REGION_FORMAT_MAGIC;
|
|
header[8] = REGION_FORMAT_VERSION;
|
|
header[9] = static_cast<ubyte>(compression); // FIXME
|
|
std::ofstream file(filename, std::ios::out | std::ios::binary);
|
|
file.write(header, REGION_HEADER_SIZE);
|
|
|
|
size_t offset = REGION_HEADER_SIZE;
|
|
uint32_t intbuf;
|
|
uint offsets[REGION_CHUNKS_COUNT] {};
|
|
|
|
auto region = entry->getChunks();
|
|
auto sizes = entry->getSizes();
|
|
|
|
for (size_t i = 0; i < REGION_CHUNKS_COUNT; i++) {
|
|
ubyte* chunk = region[i].get();
|
|
if (chunk == nullptr) {
|
|
continue;
|
|
}
|
|
offsets[i] = offset;
|
|
|
|
auto sizevec = sizes[i];
|
|
uint32_t compressedSize = sizevec[0];
|
|
uint32_t srcSize = sizevec[1];
|
|
|
|
intbuf = dataio::h2le(compressedSize);
|
|
file.write(reinterpret_cast<const char*>(&intbuf), 4);
|
|
offset += 4;
|
|
|
|
intbuf = dataio::h2le(srcSize);
|
|
file.write(reinterpret_cast<const char*>(&intbuf), 4);
|
|
offset += 4;
|
|
|
|
file.write(reinterpret_cast<const char*>(chunk), compressedSize);
|
|
offset += compressedSize;
|
|
}
|
|
for (size_t i = 0; i < REGION_CHUNKS_COUNT; i++) {
|
|
intbuf = dataio::h2le(offsets[i]);
|
|
file.write(reinterpret_cast<const char*>(&intbuf), 4);
|
|
}
|
|
}
|
|
|
|
std::unique_ptr<ubyte[]> RegionsLayer::readChunkData(
|
|
int x, int z, uint32_t& size, uint32_t& srcSize, regfile* rfile
|
|
) {
|
|
int regionX, regionZ, localX, localZ;
|
|
calc_reg_coords(x, z, regionX, regionZ, localX, localZ);
|
|
int chunkIndex = localZ * REGION_SIZE + localX;
|
|
return rfile->read(chunkIndex, size, srcSize);
|
|
}
|