From 29f207edf9751ba4ca6a5c02b7152e2e0451e32b Mon Sep 17 00:00:00 2001
From: Linden <65407488+thelindat@users.noreply.github.com>
Date: Mon, 10 Mar 2025 14:27:35 +1100
Subject: [PATCH] feat(grid): add filtering to getNearbyEntries

---
 imports/grid/shared.lua | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/imports/grid/shared.lua b/imports/grid/shared.lua
index 9355f59..8300ee6 100644
--- a/imports/grid/shared.lua
+++ b/imports/grid/shared.lua
@@ -81,8 +81,9 @@ function lib.grid.getCell(point)
 end
 
 ---@param point vector
+---@param filter? fun(entry: GridEntry): boolean
 ---@return GridEntry[]
-function lib.grid.getNearbyEntries(point)
+function lib.grid.getNearbyEntries(point, filter)
     local minX, maxX, minY, maxY = getGridDimensions(point, xDelta, yDelta)
 
     if lastNearbyEntries.minX == minX and
@@ -107,7 +108,7 @@ function lib.grid.getNearbyEntries(point)
                 for j = 1, #cell do
                     local entry = cell[j]
 
-                    if not entrySet[entry] then
+                    if not entrySet[entry] and (not filter or filter(entry)) then
                         n = n + 1
                         entrySet[entry] = true
                         entries[n] = entry