Univerxel/src/world/Universe.cpp

#include "Universe.hpp"

#include "../contouring/Dummy.hpp"
#include "../data/geometry/Sphere.hpp"
#include <Remotery.h>
#include <filesystem>
#include <fstream>

using namespace world;

Universe::Universe(const Universe::options &options): loadPool(4, options.folderPath), savePool(2, options.folderPath),
        contouring(std::make_shared<contouring::Dummy>()) {
    setOptions(options);
}
Universe::~Universe() {
    contouring = NULL;

    // Save all
    for(auto& pair: chunks) {
        savePool.push(pair);
    }
    if(savePool.size() > 0) {
        std::cout << "Saving..." << std::endl;
    }
    while(savePool.size() > 0) {
        std::this_thread::sleep_for(std::chrono::milliseconds(100));
    }
}

Universe::LoadPool::LoadPool(size_t count, const std::string& folderPath): folderPath(folderPath) {
    for (size_t i = 0; i < count; i++) {
        workers.push_back(std::thread([&] {
            while (running) {
                chunk_pos ctx;
                loadQueue.wait();
                if (loadQueue.pop(ctx)) {
                    //MAYBE: loadQueue.take to avoid duplicated work on fast move
                    rmt_ScopedCPUSample(ProcessLoad, 0);
                    const auto path = folderPath + '/' + std::to_string(ctx.x) + '.' +
                        std::to_string(ctx.y) + '.' + std::to_string(ctx.z) + ".map";
                    std::ifstream str(path);
                    if(str.good()) {
                        rmt_ScopedCPUSample(ProcessRead, 0);
                        loadedQueue.push({ctx, std::make_shared<Chunk>(str)});
                        if(str.bad()) {
                            std::cout << "Fail reading '" << path << "'" << std::endl;
                        }
                    } else {
                        rmt_ScopedCPUSample(ProcessGenerate, 0);
                        loadedQueue.push({ctx, std::make_shared<Chunk>(ctx, generator)});
                    }
                }
            }
        }));
    }
}
Universe::LoadPool::~LoadPool() {
    running = false;
    loadQueue.notify();

    for (auto &worker : workers) {
        if (worker.joinable())
            worker.join();
    }
}
inline void Universe::LoadPool::push(const chunk_pos &pos, int weight) { loadQueue.push(pos, weight); }
inline bool Universe::LoadPool::pop(robin_hood::pair<chunk_pos, std::shared_ptr<Chunk>> &out) { return loadedQueue.pop(out); }
inline size_t Universe::LoadPool::size() { return loadQueue.size(); }

Universe::SavePool::SavePool(size_t count, const std::string& folderPath): folderPath(folderPath) {
    std::filesystem::create_directories(folderPath);
    for (size_t i = 0; i < count; i++) {
        workers.push_back(std::thread([&] {
            while (running) {
                robin_hood::pair<chunk_pos, std::shared_ptr<Chunk>> ctx;
                queue.wait();
                if (queue.pop(ctx) && ctx.second->isModified()) {
                    //MAYBE: queue.take to avoid concurent write or duplicated work on fast move
                    rmt_ScopedCPUSample(ProcessSave, 0);
                    const auto path = folderPath + '/' + std::to_string(ctx.first.x) + '.' +
                        std::to_string(ctx.first.y) + '.' + std::to_string(ctx.first.z) + ".map";
                    std::ofstream ofs(path);
                    if(!ofs.good()) {
                        std::cout << "Fail opening '" << path << "' to save" << std::endl;
                        continue;
                    }
                    ctx.second->write(ofs);
                    ofs.flush();
                    ofs.close();
                    if(!ofs.good()) {
                        std::cout << "Fail closing '" << path << "' to save" << std::endl;
                    }
                }
            }
        }));
    }
}
Universe::SavePool::~SavePool() {
    running = false;
    queue.notify();

    for (auto &worker : workers) {
        if (worker.joinable())
            worker.join();
    }
}
inline void Universe::SavePool::push(const robin_hood::pair<chunk_pos, std::shared_ptr<Chunk>> &pos) { queue.push(pos); }
inline size_t Universe::SavePool::size() { return queue.size(); }

void Universe::update(const camera_pos& pos, Universe::report& rep) {
    const chunk_pos newPos = glm::divide(pos, chunk_voxel_pos(CHUNK_LENGTH));
    const auto chunkChange = last_pos != newPos;
    last_pos = newPos;
    rmt_ScopedCPUSample(Universe, 0);

    // Update alive chunks
    {
        rmt_ScopedCPUSample(Update, 0);
        auto it = chunks.begin();
        while (it != chunks.end()) {
            if (glm::length2(last_pos - it->first) > keepDistance * keepDistance) {
                savePool.push(*it);
                it = chunks.erase(it);
            } else {
                if (const auto neighbors = it->second->update()) {
                    contouring->onUpdate(it->first, chunks, neighbors.value()); //TODO: get update update_type(simple(pos), complex)
                } else if (chunkChange) {
                    contouring->onNotify(it->first, chunks);
                }
                ++it;
            }
        }
    }
    rep.chunk_unload.push(savePool.size());
    {
        rmt_ScopedCPUSample(Contouring, 0);
        contouring->update(pos);
        //MAYBE: if(chunkChange) contouring->notify(chunks);
    }

    // Find missing chunks
    if(chunkChange) {
        rmt_ScopedCPUSample(ToLoad, 0);
        //NOTE: need dist so no easy sphere fill
        for (int x = -loadDistance; x <= loadDistance; x++) {
        for (int y = -loadDistance; y <= loadDistance; y++) {
        for (int z = -loadDistance; z <= loadDistance; z++) {
            const auto dist2 = x * x + y * y + z * z;
            if (dist2 <= loadDistance * loadDistance) {
                const chunk_pos p = last_pos + glm::ivec3(x, y, z);
                if (chunks.find(p) == chunks.end()) {
                    loadPool.push(p, -dist2);
                }
            }
        }}}
    }

    rep.chunk_load.push(loadPool.size());
    // Loaded chunks
    {
        rmt_ScopedCPUSample(Load, 0);
        robin_hood::pair<chunk_pos, std::shared_ptr<Chunk>> loaded;
        while (loadPool.pop(loaded)) {
            chunks.insert(loaded);
            contouring->onUpdate(loaded.first, chunks, Faces::All);
        }
    }
    rep.chunk_count.push(chunks.size());
}
void Universe::setOptions(const Universe::options& options) {
    loadDistance = options.loadDistance;
    keepDistance = options.keepDistance;
}

void Universe::setContouring(std::shared_ptr<contouring::Abstract> ct) {
    contouring = ct;
    last_pos = chunk_pos(INT_MAX); // trigger chunkChange on next update
}

std::optional<std::pair<voxel_pos, Voxel>> Universe::raycast(const Ray &ray) const {
    std::vector<voxel_pos> points;
    ray.grid(points);
    std::shared_ptr<Chunk> chunk = NULL;
    chunk_pos chunk_pos(INT_MAX);
    for(auto point: points) {
        const auto pos = glm::divide(point, glm::ivec3(CHUNK_LENGTH));
        if(pos != chunk_pos) {
            if(const auto& newChunk = at(pos)) {
                chunk = newChunk.value();
                chunk_pos = pos;
            }
        }
        if(chunk != NULL) {
            const auto voxel = chunk->getAt(glm::modulo(point, glm::uvec3(CHUNK_LENGTH)));
            if(voxel.Density > 0)
                return {{point, voxel}};
        }
    }
    return {};
}

std::optional<Item> Universe::set(const voxel_pos& pos, const Voxel& val) {
    const auto chunkPos = glm::divide(pos, glm::ivec3(CHUNK_LENGTH));
    if(const auto& chunk = at(chunkPos)) {
        return {chunk.value()->breakAt(glm::modulo(pos, glm::ivec3(CHUNK_LENGTH)), val)};
    } else {
        return {};
    }
}
ItemList Universe::setCube(const voxel_pos& pos, const Voxel& val, int radius) {
    ItemList list;
    for (int z = -radius; z <= radius; z++) {
    for (int y = -radius; y <= radius; y++) {
    for (int x = -radius; x <= radius; x++) {
        //TODO: list.pop(val)
        list.add(set(pos + glm::lvec3(x, y, z), val));
    }}}
    return list;
}