trying to add spatial grid. please send help
This commit is contained in:
@@ -99,37 +99,72 @@ public:
|
|||||||
|
|
||||||
};
|
};
|
||||||
|
|
||||||
class SpatialCell2Leaf {
|
class SpatialGrid {
|
||||||
private:
|
private:
|
||||||
static constexpr float SIZE = 10;
|
float cellSize;
|
||||||
Vec2 origin;
|
std::unordered_map<Vec2, std::unordered_set<size_t>, Vec2::Hash> grid;
|
||||||
std::vector<size_t> ids;
|
|
||||||
size_t id; //this leaf nodes id
|
|
||||||
public:
|
public:
|
||||||
SpatialCell2Leaf(const Vec2& origin, size_t id) : origin(origin), id(id) {}
|
SpatialGrid(float cellSize = 2.0f) : cellSize(cellSize) {}
|
||||||
void insertID(size_t id, Vec2& pos) {
|
|
||||||
//if (pos.distance(origin) < size) {
|
Vec2 worldToGrid(const Vec2& worldPos) const {
|
||||||
ids.push_back(id);
|
return (worldPos / cellSize).floor();
|
||||||
//}
|
|
||||||
}
|
}
|
||||||
void safeInsertID(size_t id, Vec2& pos) {
|
|
||||||
if (pos.distance(origin) < SIZE) {
|
void insert(size_t id, const Vec2& pos) {
|
||||||
ids.push_back(id);
|
Vec2 gridPos = worldToGrid(pos);
|
||||||
|
grid[gridPos].insert(id);
|
||||||
|
}
|
||||||
|
|
||||||
|
void remove(size_t id, const Vec2& pos) {
|
||||||
|
Vec2 gridPos = worldToGrid(pos);
|
||||||
|
auto cellIt = grid.find(gridPos);
|
||||||
|
if (cellIt != grid.end()) {
|
||||||
|
cellIt->second.erase(id);
|
||||||
|
if (cellIt->second.empty()) {
|
||||||
|
grid.erase(cellIt);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
std::vector<size_t> getIds() {
|
|
||||||
return ids;
|
|
||||||
}
|
}
|
||||||
const Vec2& getOrigin() const { return origin; }
|
|
||||||
|
void update(size_t id, const Vec2& oldPos, const Vec2& newPos) {
|
||||||
|
Vec2 oldGridPos = worldToGrid(oldPos);
|
||||||
|
Vec2 newGridPos = worldToGrid(newPos);
|
||||||
|
|
||||||
|
if (oldGridPos != newGridPos) {
|
||||||
|
remove(id, oldPos);
|
||||||
|
insert(id, newPos);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
std::vector<size_t> queryRange(const Vec2& center, float radius) const {
|
||||||
|
std::vector<size_t> results;
|
||||||
|
float radiusSq = radius * radius;
|
||||||
|
|
||||||
|
// Calculate grid bounds for the query
|
||||||
|
Vec2 minGrid = worldToGrid(center - Vec2(radius, radius));
|
||||||
|
Vec2 maxGrid = worldToGrid(center + Vec2(radius, radius));
|
||||||
|
|
||||||
|
// Check all relevant grid cells
|
||||||
|
for (int x = minGrid.x; x <= maxGrid.x; ++x) {
|
||||||
|
for (int y = minGrid.y; y <= maxGrid.y; ++y) {
|
||||||
|
Vec2 gridPos(x, y);
|
||||||
|
auto cellIt = grid.find(gridPos);
|
||||||
|
if (cellIt != grid.end()) {
|
||||||
|
for (size_t id : cellIt->second) {
|
||||||
|
results.push_back(id);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return results;
|
||||||
|
}
|
||||||
|
|
||||||
|
void clear() {
|
||||||
|
grid.clear();
|
||||||
|
}
|
||||||
};
|
};
|
||||||
|
|
||||||
class SpatialGrid2 {
|
|
||||||
private:
|
|
||||||
std::unordered_map<Vec2, SpatialCell2Leaf, Vec2::Hash> cells;
|
|
||||||
public:
|
|
||||||
|
|
||||||
}
|
|
||||||
|
|
||||||
class Grid2 {
|
class Grid2 {
|
||||||
private:
|
private:
|
||||||
//all positions
|
//all positions
|
||||||
@@ -151,7 +186,8 @@ private:
|
|||||||
float neighborRadius = 1.0f;
|
float neighborRadius = 1.0f;
|
||||||
|
|
||||||
//TODO: spatial map
|
//TODO: spatial map
|
||||||
std::unordered_map<Vec2, std::vector<size_t>, Vec2::Hash> spatialtable;
|
std::unordered_map<Vec2, reverselookupassistantclasscausecppisdumb, Vec2::Hash> spatialTable;
|
||||||
|
float spatSize = 2.0f;
|
||||||
public:
|
public:
|
||||||
//get position from id
|
//get position from id
|
||||||
Vec2 getPositionID(size_t id) const {
|
Vec2 getPositionID(size_t id) const {
|
||||||
@@ -169,6 +205,12 @@ public:
|
|||||||
return getPositionVec(Vec2(x,y), radius);
|
return getPositionVec(Vec2(x,y), radius);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
size_t getIDFromSpatPOS(Vec2 pos) {
|
||||||
|
Vec2 spat2pos = (pos / spatSize).floor();
|
||||||
|
reverselookupassistantclasscausecppisdumb spatids = spatialTable.at(spat2pos);
|
||||||
|
return spatids.at(pos);
|
||||||
|
}
|
||||||
|
|
||||||
//get all id in region
|
//get all id in region
|
||||||
std::vector<size_t> getPositionVecRegion(Vec2 pos, float radius = 1.0f) {
|
std::vector<size_t> getPositionVecRegion(Vec2 pos, float radius = 1.0f) {
|
||||||
float searchRadius = (radius == 0.0f) ? std::numeric_limits<float>::epsilon() : radius;
|
float searchRadius = (radius == 0.0f) ? std::numeric_limits<float>::epsilon() : radius;
|
||||||
|
|||||||
Reference in New Issue
Block a user