67 lines
2.2 KiB
C++
67 lines
2.2 KiB
C++
#ifndef GEOPRO_RENDER_SOURCE_BRICKPAGER_HPP
|
||
#define GEOPRO_RENDER_SOURCE_BRICKPAGER_HPP
|
||
|
||
#include <cstddef>
|
||
#include <cstdint>
|
||
#include <list>
|
||
#include <unordered_map>
|
||
#include <vector>
|
||
|
||
namespace geopro::data {
|
||
class ChunkedVolumeStore;
|
||
}
|
||
|
||
namespace geopro::render {
|
||
|
||
// 缓存键：完整 brick 标识（含 level）。
|
||
struct BrickId {
|
||
int level = 0, bx = 0, by = 0, bz = 0;
|
||
bool operator==(const BrickId& o) const noexcept {
|
||
return level == o.level && bx == o.bx && by == o.by && bz == o.bz;
|
||
}
|
||
};
|
||
|
||
// 内存恒定的 brick LRU 缓存：任意时刻驻留 ≤ budgetBricks 个解压块，
|
||
// 与体总大小无关。这是 C 方案核外渲染的内存控制核心。
|
||
class BrickPager {
|
||
public:
|
||
BrickPager(const geopro::data::ChunkedVolumeStore& store,
|
||
std::size_t budgetBricks);
|
||
|
||
// 载入 visible 中缺失的块（经 store.readBrick 解压），按请求顺序更新 LRU；
|
||
// 之后淘汰最久未用的，直到 residentCount() <= budget。
|
||
// 若 visible.size() > budget，保留最近请求（visible 末尾）的 budget 个。
|
||
void requestVisible(const std::vector<BrickId>& visible);
|
||
|
||
// 命中返回该块解压数据指针（数据由 pager 持有，下次淘汰前有效）；未驻留返回
|
||
// nullptr。get 不改动 LRU（保持 const 语义），驻留集仅由 requestVisible 决定。
|
||
const std::vector<std::int16_t>* get(const BrickId& id) const;
|
||
|
||
std::size_t residentCount() const { return cache_.size(); }
|
||
std::size_t budget() const { return budget_; }
|
||
|
||
private:
|
||
struct Entry {
|
||
std::vector<std::int16_t> data;
|
||
std::list<BrickId>::iterator lruIt; // 指向 lru_ 中本块位置
|
||
};
|
||
|
||
struct BrickIdHash {
|
||
std::size_t operator()(const BrickId& id) const noexcept;
|
||
};
|
||
|
||
// 把 id 移到 MRU 端（lru_ 头部）。
|
||
void touch(const BrickId& id);
|
||
// 从 LRU 端淘汰直到 residentCount() <= budget_。
|
||
void evictToBudget();
|
||
|
||
const geopro::data::ChunkedVolumeStore& store_;
|
||
std::size_t budget_;
|
||
std::list<BrickId> lru_; // front = 最近使用(MRU)，back = 最久未用(LRU)
|
||
std::unordered_map<BrickId, Entry, BrickIdHash> cache_;
|
||
};
|
||
|
||
} // namespace geopro::render
|
||
|
||
#endif // GEOPRO_RENDER_SOURCE_BRICKPAGER_HPP
|