spatc16 (16x16 spatial subgrid for fixed size optimizations)

tests/g2chromatic.cpp

@@ -8,9 +8,9 @@
 #include "../util/timing_decorator.cpp"
 
 struct AnimationConfig {
-    int width = 512;
-    int height = 512;
-    int totalFrames = 240;
+    int width = 1024;
+    int height = 1024;
+    int totalFrames = 480;
     float fps = 30.0f;
     int numSeeds = 1;
 };
@@ -99,7 +99,7 @@ std::vector<uint8_t> convertFrameToBGR(Grid2& grid, const AnimationConfig& confi
     TIME_FUNCTION;
     int frameWidth, frameHeight;
     std::vector<int> bgrData;
-    grid.getGridRegionAsBGR(Vec2(0,0),Vec2(512,512), frameWidth, frameHeight, bgrData);
+    grid.getGridRegionAsBGR(Vec2(0,0),Vec2(config.width,config.height), frameWidth, frameHeight, bgrData);
     //grid.getGridRegionAsRGB(0.0f,0.0f,512.0f,512.0f,frameWidth,frameHeight,rgbData);
     
     std::vector<uint8_t> bgrFrame(frameWidth * frameHeight * 3);

tests/g2chromaticsimd.cpp

@@ -9,8 +9,8 @@
 #include "../util/timing_decorator.cpp"
 
 struct AnimationConfig {
-    int width = 512;
-    int height = 512;
+    int width = 1024;
+    int height = 1024;
     int totalFrames = 240;
     float fps = 30.0f;
     int numSeeds = 1;
@@ -223,9 +223,7 @@ std::vector<uint8_t> convertFrameToBGR(Grid2& grid, const AnimationConfig& confi
 }
 
 
-std::vector<std::vector<uint8_t>> createAnimationFrames(Grid2& grid, 
-                                                       const SeedDataSoA& seeds,
-                                                       const AnimationConfig& config) {
+std::vector<std::vector<uint8_t>> createAnimationFrames(Grid2& grid, const SeedDataSoA& seeds, const AnimationConfig& config) {
     TIME_FUNCTION;
     std::vector<std::vector<uint8_t>> frames;
     
@@ -241,10 +239,9 @@ std::vector<std::vector<uint8_t>> createAnimationFrames(Grid2& grid,
     
     return frames;
 }
-void printSuccessMessage(const std::string& filename, 
-                        const std::vector<Vec2>& seedPoints,
-                        const std::vector<Vec4>& seedColors,
-                        const AnimationConfig& config) {
+
+void printSuccessMessage(const std::string& filename, const std::vector<Vec2>& seedPoints,
+                            const std::vector<Vec4>& seedColors, const AnimationConfig& config) {
     std::cout << "\nSuccessfully saved chromatic transformation animation to: " << filename << std::endl;
     std::cout << "Video details:" << std::endl;
     std::cout << "  - Dimensions: " << config.width << " x " << config.height << std::endl;

util/grid/grid2fast.hpp

@@ -0,0 +1,316 @@
+#ifndef FIXED_SPATIAL_GRID_2_HPP
+#define FIXED_SPATIAL_GRID_2_HPP
+
+#include "../vectorlogic/vec2.hpp"
+#include <vector>
+#include <unordered_map>
+#include <array>
+#include <algorithm>
+#include <cmath>
+
+class Grid2Fast {
+private:
+    struct Cell {
+        std::vector<size_t> objectIds;
+        
+        void add(size_t id) {
+            objectIds.push_back(id);
+        }
+        
+        void remove(size_t id) {
+            auto it = std::find(objectIds.begin(), objectIds.end(), id);
+            if (it != objectIds.end()) {
+                objectIds.erase(it);
+            }
+        }
+        
+        bool contains(size_t id) const {
+            return std::find(objectIds.begin(), objectIds.end(), id) != objectIds.end();
+        }
+        
+        void clear() {
+            objectIds.clear();
+        }
+        
+        size_t size() const {
+            return objectIds.size();
+        }
+        
+        bool empty() const {
+            return objectIds.empty();
+        }
+    };
+
+    // Fixed grid dimensions
+    int gridWidth, gridHeight;
+    float cellSize;
+    Vec2 worldMin, worldMax;
+    
+    // 2D grid storage
+    std::vector<Cell> grid;
+    std::unordered_map<size_t, std::pair<int, int>> objectToCell;
+    
+    // Helper methods
+    inline int toIndex(int x, int y) const {
+        return y * gridWidth + x;
+    }
+    
+    inline bool isValidCell(int x, int y) const {
+        return x >= 0 && x < gridWidth && y >= 0 && y < gridHeight;
+    }
+
+public:
+    Grid2Fast(const Vec2& minCorner, const Vec2& maxCorner, float cellSize)
+        : cellSize(cellSize), worldMin(minCorner), worldMax(maxCorner) {
+        
+        // Calculate grid dimensions
+        float worldWidth = maxCorner.x - minCorner.x;
+        float worldHeight = maxCorner.y - minCorner.y;
+        
+        gridWidth = static_cast<int>(std::ceil(worldWidth / cellSize));
+        gridHeight = static_cast<int>(std::ceil(worldHeight / cellSize));
+        
+        // Initialize grid with empty cells
+        grid.resize(gridWidth * gridHeight);
+    }
+    
+    Grid2Fast(float minX, float minY, float maxX, float maxY, float cellSize)
+        : Grid2Fast(Vec2(minX, minY), Vec2(maxX, maxY), cellSize) {}
+    
+    // Convert world position to grid coordinates
+    std::pair<int, int> worldToGrid(const Vec2& pos) const {
+        int x = static_cast<int>((pos.x - worldMin.x) / cellSize);
+        int y = static_cast<int>((pos.y - worldMin.y) / cellSize);
+        
+        // Clamp to grid boundaries
+        x = std::clamp(x, 0, gridWidth - 1);
+        y = std::clamp(y, 0, gridHeight - 1);
+        
+        return {x, y};
+    }
+    
+    // Convert grid coordinates to world position (center of cell)
+    Vec2 gridToWorld(int gridX, int gridY) const {
+        float x = worldMin.x + (gridX + 0.5f) * cellSize;
+        float y = worldMin.y + (gridY + 0.5f) * cellSize;
+        return Vec2(x, y);
+    }
+    
+    // Add object to spatial grid
+    bool addObject(size_t id, const Vec2& position) {
+        auto [gridX, gridY] = worldToGrid(position);
+        
+        if (!isValidCell(gridX, gridY)) {
+            return false; // Object outside grid bounds
+        }
+        
+        int index = toIndex(gridX, gridY);
+        grid[index].add(id);
+        objectToCell[id] = {gridX, gridY};
+        return true;
+    }
+    
+    // Remove object from spatial grid
+    bool removeObject(size_t id) {
+        auto it = objectToCell.find(id);
+        if (it == objectToCell.end()) {
+            return false;
+        }
+        
+        auto [gridX, gridY] = it->second;
+        if (isValidCell(gridX, gridY)) {
+            int index = toIndex(gridX, gridY);
+            grid[index].remove(id);
+        }
+        
+        objectToCell.erase(it);
+        return true;
+    }
+    
+    // Update object position
+    bool updateObject(size_t id, const Vec2& oldPos, const Vec2& newPos) {
+        auto oldCell = worldToGrid(oldPos);
+        auto newCell = worldToGrid(newPos);
+        
+        if (oldCell == newCell) {
+            // Same cell, no update needed
+            objectToCell[id] = newCell;
+            return true;
+        }
+        
+        // Remove from old cell
+        auto [oldX, oldY] = oldCell;
+        if (isValidCell(oldX, oldY)) {
+            int oldIndex = toIndex(oldX, oldY);
+            grid[oldIndex].remove(id);
+        }
+        
+        // Add to new cell
+        auto [newX, newY] = newCell;
+        if (!isValidCell(newX, newY)) {
+            // Object moved outside grid, remove completely
+            objectToCell.erase(id);
+            return false;
+        }
+        
+        int newIndex = toIndex(newX, newY);
+        grid[newIndex].add(id);
+        objectToCell[id] = newCell;
+        return true;
+    }
+    
+    // Get objects in radius (optimized using grid)
+    std::vector<size_t> getObjectsInRadius(const Vec2& position, float radius) const {
+        std::vector<size_t> result;
+        
+        if (radius <= 0.0f) {
+            return getObjectsAt(position);
+        }
+        
+        Vec2 minPos(position.x - radius, position.y - radius);
+        Vec2 maxPos(position.x + radius, position.y + radius);
+        
+        auto minCell = worldToGrid(minPos);
+        auto maxCell = worldToGrid(maxPos);
+        
+        float radiusSq = radius * radius;
+        
+        // Check only relevant cells
+        for (int y = minCell.second; y <= maxCell.second; ++y) {
+            for (int x = minCell.first; x <= maxCell.first; ++x) {
+                if (!isValidCell(x, y)) continue;
+                
+                int index = toIndex(x, y);
+                const Cell& cell = grid[index];
+                
+                for (size_t id : cell.objectIds) {
+                    // We need external position data for distance check
+                    // This assumes the caller will filter results based on actual positions
+                    result.push_back(id);
+                }
+            }
+        }
+        
+        return result;
+    }
+    
+    // Get objects at exact position
+    std::vector<size_t> getObjectsAt(const Vec2& position) const {
+        auto [gridX, gridY] = worldToGrid(position);
+        
+        if (!isValidCell(gridX, gridY)) {
+            return {};
+        }
+        
+        int index = toIndex(gridX, gridY);
+        return grid[index].objectIds; // Return copy
+    }
+    
+    // Get objects in rectangular region
+    std::vector<size_t> getObjectsInRegion(const Vec2& minCorner, const Vec2& maxCorner) const {
+        std::vector<size_t> result;
+        
+        auto minCell = worldToGrid(minCorner);
+        auto maxCell = worldToGrid(maxCorner);
+        
+        for (int y = minCell.second; y <= maxCell.second; ++y) {
+            for (int x = minCell.first; x <= maxCell.first; ++x) {
+                if (!isValidCell(x, y)) continue;
+                
+                int index = toIndex(x, y);
+                const Cell& cell = grid[index];
+                
+                // Add all objects from these cells
+                // Note: This may include objects outside the exact region due to cell granularity
+                // Caller should filter based on actual positions if precise region is needed
+                result.insert(result.end(), cell.objectIds.begin(), cell.objectIds.end());
+            }
+        }
+        
+        return result;
+    }
+    
+    // Get all objects in the grid
+    std::vector<size_t> getAllObjects() const {
+        std::vector<size_t> result;
+        
+        for (const auto& pair : objectToCell) {
+            result.push_back(pair.first);
+        }
+        
+        return result;
+    }
+    
+    // Get cell information
+    const Cell& getCell(int x, int y) const {
+        static Cell emptyCell;
+        if (!isValidCell(x, y)) {
+            return emptyCell;
+        }
+        return grid[toIndex(x, y)];
+    }
+    
+    const Cell& getCellAtWorldPos(const Vec2& pos) const {
+        auto [x, y] = worldToGrid(pos);
+        return getCell(x, y);
+    }
+    
+    // Statistics
+    size_t getTotalObjectCount() const {
+        return objectToCell.size();
+    }
+    
+    size_t getNonEmptyCellCount() const {
+        size_t count = 0;
+        for (const auto& cell : grid) {
+            if (!cell.empty()) {
+                ++count;
+            }
+        }
+        return count;
+    }
+    
+    size_t getMaxObjectsPerCell() const {
+        size_t maxCount = 0;
+        for (const auto& cell : grid) {
+            maxCount = std::max(maxCount, cell.size());
+        }
+        return maxCount;
+    }
+    
+    float getAverageObjectsPerCell() const {
+        if (grid.empty()) return 0.0f;
+        return static_cast<float>(objectToCell.size()) / grid.size();
+    }
+    
+    // Grid properties
+    int getGridWidth() const { return gridWidth; }
+    int getGridHeight() const { return gridHeight; }
+    float getCellSize() const { return cellSize; }
+    Vec2 getWorldMin() const { return worldMin; }
+    Vec2 getWorldMax() const { return worldMax; }
+    
+    // Clear all objects
+    void clear() {
+        for (auto& cell : grid) {
+            cell.clear();
+        }
+        objectToCell.clear();
+    }
+    
+    // Check if object exists in grid
+    bool contains(size_t id) const {
+        return objectToCell.find(id) != objectToCell.end();
+    }
+    
+    // Get cell coordinates for object
+    std::pair<int, int> getObjectCell(size_t id) const {
+        auto it = objectToCell.find(id);
+        if (it != objectToCell.end()) {
+            return it->second;
+        }
+        return {-1, -1};
+    }
+};
+
+#endif

util/grid/spatc16.hpp

@@ -0,0 +1,233 @@
+#ifndef SPATIAL_CELL_16X16_HPP
+#define SPATIAL_CELL_16X16_HPP
+
+#include "../vectorlogic/vec2.hpp"
+#include "../vectorlogic/vec4.hpp"
+#include <vector>
+#include <unordered_map>
+#include <unordered_set>
+#include <algorithm>
+#include <memory>
+
+class SpatialCell16x16 {
+private:
+    static constexpr int CELL_SIZE = 16;
+    static constexpr int TOTAL_CELLS = CELL_SIZE * CELL_SIZE;
+    
+    // Store objects in the cell
+    std::unordered_map<size_t, Vec2> positions;
+    std::unordered_map<size_t, Vec4> colors;
+    std::unordered_map<size_t, float> sizes;
+    
+    // Bounds of this cell in world coordinates
+    Vec2 worldMin, worldMax;
+    float worldCellSize; // Size of each pixel in world coordinates
+    
+public:
+    SpatialCell16x16(const Vec2& minCorner, const Vec2& maxCorner) 
+        : worldMin(minCorner), worldMax(maxCorner) {
+        
+        // Calculate world size per cell pixel
+        worldCellSize = std::max(
+            (worldMax.x - worldMin.x) / CELL_SIZE,
+            (worldMax.y - worldMin.y) / CELL_SIZE
+        );
+    }
+    
+    // Convert world position to cell coordinates [0,15]
+    std::pair<int, int> worldToCell(const Vec2& worldPos) const {
+        float localX = (worldPos.x - worldMin.x) / (worldMax.x - worldMin.x);
+        float localY = (worldPos.y - worldMin.y) / (worldMax.y - worldMin.y);
+        
+        int cellX = static_cast<int>(localX * CELL_SIZE);
+        int cellY = static_cast<int>(localY * CELL_SIZE);
+        
+        // Clamp to valid range
+        cellX = std::clamp(cellX, 0, CELL_SIZE - 1);
+        cellY = std::clamp(cellY, 0, CELL_SIZE - 1);
+        
+        return {cellX, cellY};
+    }
+    
+    // Convert cell coordinates to linear index
+    int cellToIndex(int x, int y) const {
+        return y * CELL_SIZE + x;
+    }
+    
+    // Convert linear index to cell coordinates
+    std::pair<int, int> indexToCell(int index) const {
+        return {index % CELL_SIZE, index / CELL_SIZE};
+    }
+    
+    // Convert cell coordinates to world position (center of cell)
+    Vec2 cellToWorld(int x, int y) const {
+        float worldX = worldMin.x + (x + 0.5f) * worldCellSize;
+        float worldY = worldMin.y + (y + 0.5f) * worldCellSize;
+        return Vec2(worldX, worldY);
+    }
+    
+    // Add object to the spatial cell
+    bool addObject(size_t id, const Vec2& position, const Vec4& color, float size = 1.0f) {
+        if (!contains(position)) {
+            return false;
+        }
+        
+        positions[id] = position;
+        colors[id] = color;
+        sizes[id] = size;
+        
+        return true;
+    }
+    
+    // Check if world position is within this cell's bounds
+    bool contains(const Vec2& worldPos) const {
+        return worldPos.x >= worldMin.x && worldPos.x <= worldMax.x &&
+               worldPos.y >= worldMin.y && worldPos.y <= worldMax.y;
+    }
+    
+    // Update object position
+    void updateObject(size_t id, const Vec2& oldPos, const Vec2& newPos) {
+        if (!hasObject(id)) return;
+        
+        positions[id] = newPos;
+    }
+    
+    // Remove object
+    void removeObject(size_t id) {
+        if (!hasObject(id)) return;
+        
+        positions.erase(id);
+        colors.erase(id);
+        sizes.erase(id);
+    }
+    
+    // Check if object exists
+    bool hasObject(size_t id) const {
+        return positions.find(id) != positions.end();
+    }
+    
+    // Get object data
+    Vec2 getPosition(size_t id) const {
+        auto it = positions.find(id);
+        return it != positions.end() ? it->second : Vec2();
+    }
+    
+    Vec4 getColor(size_t id) const {
+        auto it = colors.find(id);
+        return it != colors.end() ? it->second : Vec4();
+    }
+    
+    float getSize(size_t id) const {
+        auto it = sizes.find(id);
+        return it != sizes.end() ? it->second : 1.0f;
+    }
+    
+    // Set object data
+    void setPosition(size_t id, const Vec2& position) {
+        if (hasObject(id)) {
+            positions[id] = position;
+        }
+    }
+    
+    void setColor(size_t id, const Vec4& color) {
+        colors[id] = color;
+    }
+    
+    void setSize(size_t id, float size) {
+        if (hasObject(id)) {
+            sizes[id] = size;
+        }
+    }
+    
+    // Spatial queries
+    std::vector<size_t> getObjectsAt(const Vec2& position) const {
+        std::vector<size_t> result;
+        
+        // Check all objects since we don't have spatial indexing
+        for (const auto& pair : positions) {
+            size_t id = pair.first;
+            const Vec2& objPos = pair.second;
+            float size = sizes.at(id);
+            
+            // Check if position is within object bounds
+            if (position.x >= objPos.x - size * 0.5f && position.x <= objPos.x + size * 0.5f &&
+                position.y >= objPos.y - size * 0.5f && position.y <= objPos.y + size * 0.5f) {
+                result.push_back(id);
+            }
+        }
+        
+        return result;
+    }
+    
+    std::vector<size_t> getObjectsInRadius(const Vec2& center, float radius) const {
+        std::vector<size_t> result;
+        float radius_sq = radius * radius;
+        
+        // Check all objects since we don't have spatial indexing
+        for (const auto& pair : positions) {
+            size_t id = pair.first;
+            const Vec2& pos = pair.second;
+            
+            float dx = pos.x - center.x;
+            float dy = pos.y - center.y;
+            if (dx * dx + dy * dy <= radius_sq) {
+                result.push_back(id);
+            }
+        }
+        
+        return result;
+    }
+    
+    std::vector<size_t> getObjectsInRegion(const Vec2& minCorner, const Vec2& maxCorner) const {
+        std::vector<size_t> result;
+        
+        // Check all objects since we don't have spatial indexing
+        for (const auto& pair : positions) {
+            size_t id = pair.first;
+            const Vec2& pos = pair.second;
+            
+            if (pos.x >= minCorner.x && pos.x <= maxCorner.x &&
+                pos.y >= minCorner.y && pos.y <= maxCorner.y) {
+                result.push_back(id);
+            }
+        }
+        
+        return result;
+    }
+    
+    // Get all object IDs
+    std::vector<size_t> getAllObjectIds() const {
+        std::vector<size_t> ids;
+        ids.reserve(positions.size());
+        for (const auto& pair : positions) {
+            ids.push_back(pair.first);
+        }
+        return ids;
+    }
+    
+    // Get cell statistics
+    size_t getObjectCount() const { return positions.size(); }
+    size_t getNonEmptyCellCount() const {
+        // Since we removed cellBuckets, return 1 if there are objects, 0 otherwise
+        return positions.empty() ? 0 : 1;
+    }
+    
+    // Get bounds
+    Vec2 getMinCorner() const { return worldMin; }
+    Vec2 getMaxCorner() const { return worldMax; }
+    
+    // Clear all objects
+    void clear() {
+        positions.clear();
+        colors.clear();
+        sizes.clear();
+    }
+
+private:
+    // Spatial indexing is no longer used
+    void updateSpatialIndex(size_t id, const Vec2& oldPos, const Vec2& newPos) {
+        // Empty implementation since we removed spatial indexing
+    }
+};
+
+#endif