Cursor Agentand西风逍遥游
410fa7895b
Provide 2D/3D broad-phase structures for map/scene culling queries, export them through eve.Spatial, and document the new module. Co-authored-by: 西风逍遥游 <sunxfancy@gmail.com>
54 lines
1.9 KiB
C++
54 lines
1.9 KiB
C++
#pragma once
|
|
|
|
#include "spatial/Bounds.h"
|
|
#include "spatial/QueryIds.h"
|
|
|
|
#include <unordered_map>
|
|
#include <unordered_set>
|
|
#include <vector>
|
|
|
|
namespace eve::spatial {
|
|
|
|
/** Uniform-grid spatial hash for 3D AABB / sphere queries. */
|
|
class SpatialHash3D {
|
|
public:
|
|
explicit SpatialHash3D(float cellSize = 64.f);
|
|
~SpatialHash3D() = default;
|
|
|
|
SpatialHash3D(const SpatialHash3D &) = delete;
|
|
SpatialHash3D &operator=(const SpatialHash3D &) = delete;
|
|
|
|
void clear();
|
|
void setCellSize(float cellSize);
|
|
float getCellSize() const { return cellSize_; }
|
|
|
|
bool insert(int id, float minX, float minY, float minZ, float maxX, float maxY, float maxZ);
|
|
bool remove(int id);
|
|
bool update(int id, float minX, float minY, float minZ, float maxX, float maxY, float maxZ);
|
|
bool contains(int id) const;
|
|
int getCount() const { return static_cast<int>(items_.size()); }
|
|
|
|
int queryPoint(float x, float y, float z);
|
|
int queryAABB(float minX, float minY, float minZ, float maxX, float maxY, float maxZ);
|
|
int querySphere(float cx, float cy, float cz, float radius);
|
|
|
|
int getResultCount() const { return results_.getCount(); }
|
|
int getResultId(int index) const { return results_.getId(index); }
|
|
|
|
private:
|
|
void cellRange(const AABB3 &b, int &minCX, int &minCY, int &minCZ, int &maxCX, int &maxCY,
|
|
int &maxCZ) const;
|
|
void insertCells(int id, const AABB3 &b);
|
|
void eraseCells(int id, const AABB3 &b);
|
|
void queryCells(int minCX, int minCY, int minCZ, int maxCX, int maxCY, int maxCZ,
|
|
const AABB3 *box, float cx, float cy, float cz, float radius, bool useSphere,
|
|
bool usePoint);
|
|
|
|
float cellSize_ = 64.f;
|
|
std::unordered_map<int, AABB3> items_;
|
|
std::unordered_map<uint64_t, std::vector<int>> cells_;
|
|
QueryIds results_;
|
|
};
|
|
|
|
} // namespace eve::spatial
|