diff --git a/src/game/g_grid.pas b/src/game/g_grid.pas
index aeeba590205eed25b63aaf4ecab8fcd708609222..8ff01a629340b582ad58c0cb40046f8b80c251a3 100644 (file)
--- a/src/game/g_grid.pas
+++ b/src/game/g_grid.pas
* You should have received a copy of the GNU General Public License
* along with this program. If not, see <http://www.gnu.org/licenses/>.
*)
+// universal spatial grid
{$INCLUDE ../shared/a_modes.inc}
+{$DEFINE grid_use_buckets}
unit g_grid;
interface
-uses e_log;
-
-const
- GridDefaultTileSize = 32;
-
-type
- GridQueryCB = function (obj: TObject; tag: Integer): Boolean is nested; // return `true` to stop
type
- TBodyGrid = class;
+ TBodyProxyId = Integer;
- TBodyProxy = Integer;
+ generic TBodyGridBase<ITP> = class(TObject)
+ public
+ type TGridQueryCB = function (obj: ITP; tag: Integer): Boolean is nested; // return `true` to stop
- PBodyProxyRec = ^TBodyProxyRec;
- TBodyProxyRec = record
private
- mX, mY, mWidth, mHeight: Integer; // aabb
- mQueryMark: DWord; // was this object visited at this query?
- mObj: TObject;
- //mGrid: TBodyGrid;
- mTag: Integer;
- nextLink: TBodyProxy; // next free or nothing
+ const
+ GridDefaultTileSize = 32;
+ GridCellBucketSize = 8; // WARNING! can't be less than 2!
private
- procedure setup (aX, aY, aWidth, aHeight: Integer; aObj: TObject; aTag: Integer);
-
- public
- //constructor Create (aGrid: TBodyGrid; aX, aY, aWidth, aHeight: Integer; aObj: TObject; aTag: Integer);
- //destructor Destroy (); override;
-
- property x: Integer read mX;
- property y: Integer read mY;
- property width: Integer read mWidth;
- property height: Integer read mHeight;
- property obj: TObject read mObj;
- property tag: Integer read mTag;
- //property grid: TBodyGrid read mGrid;
- end;
+ type
+ PBodyProxyRec = ^TBodyProxyRec;
+ TBodyProxyRec = record
+ private
+ mX, mY, mWidth, mHeight: Integer; // aabb
+ mQueryMark: DWord; // was this object visited at this query?
+ mObj: ITP;
+ mTag: Integer;
+ nextLink: TBodyProxyId; // next free or nothing
+
+ private
+ procedure setup (aX, aY, aWidth, aHeight: Integer; aObj: ITP; aTag: Integer);
+ end;
- PGridCell = ^TGridCell;
- TGridCell = record
- body: Integer;
- next: Integer; // in this cell; index in mCells
- end;
+ PGridCell = ^TGridCell;
+ TGridCell = record
+ {$IFDEF grid_use_buckets}
+ bodies: array [0..GridCellBucketSize-1] of Integer; // -1: end of list
+ {$ELSE}
+ body: Integer;
+ {$ENDIF}
+ next: Integer; // in this cell; index in mCells
+ end;
- GridInternalCB = function (grida: Integer): Boolean is nested; // return `true` to stop
+ TGridInternalCB = function (grida: Integer): Boolean of object; // return `true` to stop
- TBodyGrid = class(TObject)
private
mTileSize: Integer;
mMinX, mMinY: Integer; // so grids can start at any origin
mGrid: array of Integer; // mWidth*mHeight, index in mCells
mCells: array of TGridCell; // cell pool
mFreeCell: Integer; // first free cell index or -1
- mLastQuery: Integer;
+ mLastQuery: DWord;
mUsedCells: Integer;
mProxies: array of TBodyProxyRec;
- mProxyFree: TBodyProxy; // free
+ mProxyFree: TBodyProxyId; // free
mProxyCount: Integer; // currently used
mProxyMaxCount: Integer;
+ mUData: TBodyProxyId; // for inserter/remover
+ mTagMask: Integer; // for iterator
+ mItCB: TGridQueryCB; // for iterator
+
private
function allocCell: Integer;
procedure freeCell (idx: Integer); // `next` is simply overwritten
- function allocProxy (aX, aY, aWidth, aHeight: Integer; aObj: TObject; aTag: Integer): TBodyProxy;
- procedure freeProxy (body: TBodyProxy);
+ function allocProxy (aX, aY, aWidth, aHeight: Integer; aObj: ITP; aTag: Integer): TBodyProxyId;
+ procedure freeProxy (body: TBodyProxyId);
+
+ procedure insert (body: TBodyProxyId);
+ procedure remove (body: TBodyProxyId);
- procedure insert (body: TBodyProxy);
- procedure remove (body: TBodyProxy);
+ function forGridRect (x, y, w, h: Integer; cb: TGridInternalCB): Boolean;
- function forGridRect (x, y, w, h: Integer; cb: GridInternalCB): Boolean;
+ function inserter (grida: Integer): Boolean;
+ function remover (grida: Integer): Boolean;
+ function iterator (grida: Integer): Boolean;
public
constructor Create (aMinPixX, aMinPixY, aPixWidth, aPixHeight: Integer; aTileSize: Integer=GridDefaultTileSize);
destructor Destroy (); override;
- function insertBody (aObj: TObject; ax, ay, aWidth, aHeight: Integer; aTag: Integer=0): TBodyProxy;
- procedure removeBody (aObj: TBodyProxy); // WARNING! this WILL destroy proxy!
+ function insertBody (aObj: ITP; ax, ay, aWidth, aHeight: Integer; aTag: Integer=0): TBodyProxyId;
+ procedure removeBody (aObj: TBodyProxyId); // WARNING! this WILL destroy proxy!
- procedure moveBody (body: TBodyProxy; dx, dy: Integer);
- procedure resizeBody (body: TBodyProxy; sx, sy: Integer);
- procedure moveResizeBody (body: TBodyProxy; dx, dy, sx, sy: Integer);
+ procedure moveBody (body: TBodyProxyId; dx, dy: Integer);
+ procedure resizeBody (body: TBodyProxyId; sx, sy: Integer);
+ procedure moveResizeBody (body: TBodyProxyId; dx, dy, sx, sy: Integer);
- function forEachInAABB (x, y, w, h: Integer; cb: GridQueryCB): Boolean;
-
- function getProxyForBody (aObj: TObject; x, y, w, h: Integer): TBodyProxy;
+ function forEachInAABB (x, y, w, h: Integer; cb: TGridQueryCB; tagmask: Integer=-1): Boolean;
procedure dumpStats ();
end;
-type
- TBinaryHeapLessFn = function (a, b: TObject): Boolean;
-
- TBinaryHeapObj = class(TObject)
- private
- elem: array of TObject;
- elemUsed: Integer;
- lessfn: TBinaryHeapLessFn;
-
- private
- procedure heapify (root: Integer);
-
- public
- constructor Create (alessfn: TBinaryHeapLessFn);
- destructor Destroy (); override;
-
- procedure clear ();
-
- procedure insert (val: TObject);
-
- function front (): TObject;
- procedure popFront ();
-
- property count: Integer read elemUsed;
- end;
-
-
implementation
uses
- SysUtils;
+ SysUtils, e_log;
// ////////////////////////////////////////////////////////////////////////// //
-(*
-constructor TBodyProxy.Create (aGrid: TBodyGrid; aX, aY, aWidth, aHeight: Integer; aObj: TObject; aTag: Integer);
-begin
- mGrid := aGrid;
- setup(aX, aY, aWidth, aHeight, aObj, aTag);
-end;
-
-
-destructor TBodyProxy.Destroy ();
-begin
- inherited;
-end;
-*)
-
-
-procedure TBodyProxyRec.setup (aX, aY, aWidth, aHeight: Integer; aObj: TObject; aTag: Integer);
+procedure TBodyGridBase.TBodyProxyRec.setup (aX, aY, aWidth, aHeight: Integer; aObj: ITP; aTag: Integer);
begin
mX := aX;
mY := aY;
// ////////////////////////////////////////////////////////////////////////// //
-constructor TBodyGrid.Create (aMinPixX, aMinPixY, aPixWidth, aPixHeight: Integer; aTileSize: Integer=GridDefaultTileSize);
+constructor TBodyGridBase.Create (aMinPixX, aMinPixY, aPixWidth, aPixHeight: Integer; aTileSize: Integer=GridDefaultTileSize);
var
idx: Integer;
begin
// init free list
for idx := 0 to High(mCells) do
begin
+ {$IFDEF grid_use_buckets}
+ mCells[idx].bodies[0] := -1;
+ {$ELSE}
mCells[idx].body := -1;
+ {$ENDIF}
mCells[idx].next := idx+1;
end;
mCells[High(mCells)].next := -1; // last cell
mProxyFree := 0;
mProxyCount := 0;
mProxyMaxCount := 0;
+ mUData := 0;
+ mTagMask := -1;
+ mItCB := nil;
e_WriteLog(Format('created grid with size: %dx%d (tile size: %d); pix: %dx%d', [mWidth, mHeight, mTileSize, mWidth*mTileSize, mHeight*mTileSize]), MSG_NOTIFY);
end;
-destructor TBodyGrid.Destroy ();
-var
- px: TBodyProxy;
+destructor TBodyGridBase.Destroy ();
begin
mCells := nil;
mGrid := nil;
end;
-procedure TBodyGrid.dumpStats ();
+procedure TBodyGridBase.dumpStats ();
var
idx, mcb, cidx, cnt: Integer;
begin
end;
-function TBodyGrid.allocCell: Integer;
+function TBodyGridBase.allocCell: Integer;
var
idx: Integer;
begin
SetLength(mCells, mFreeCell+32768); // arbitrary number
for idx := mFreeCell to High(mCells) do
begin
+ {$IFDEF grid_use_buckets}
+ mCells[idx].bodies[0] := -1;
+ {$ELSE}
mCells[idx].body := -1;
+ {$ENDIF}
mCells[idx].next := idx+1;
end;
mCells[High(mCells)].next := -1; // last cell
end;
-procedure TBodyGrid.freeCell (idx: Integer);
+procedure TBodyGridBase.freeCell (idx: Integer);
begin
if (idx >= 0) and (idx < High(mCells)) then
begin
- if mCells[idx].body = -1 then exit; // the thing that should not be
- mCells[idx].body := -1;
+ //if mCells[idx].body = -1 then exit; // the thing that should not be
+ //mCells[idx].body := -1;
mCells[idx].next := mFreeCell;
mFreeCell := idx;
Dec(mUsedCells);
end;
-function TBodyGrid.allocProxy (aX, aY, aWidth, aHeight: Integer; aObj: TObject; aTag: Integer): TBodyProxy;
+function TBodyGridBase.allocProxy (aX, aY, aWidth, aHeight: Integer; aObj: ITP; aTag: Integer): TBodyProxyId;
var
olen, idx: Integer;
px: PBodyProxyRec;
if (mProxyMaxCount < mProxyCount) then mProxyMaxCount := mProxyCount;
end;
-procedure TBodyGrid.freeProxy (body: TBodyProxy);
+procedure TBodyGridBase.freeProxy (body: TBodyProxyId);
begin
if (body < 0) or (body > High(mProxies)) then exit; // just in case
if (mProxyCount = 0) then raise Exception.Create('wutafuuuuu in grid (no allocated proxies, what i should free now?)');
end;
-function TBodyGrid.forGridRect (x, y, w, h: Integer; cb: GridInternalCB): Boolean;
+function TBodyGridBase.forGridRect (x, y, w, h: Integer; cb: TGridInternalCB): Boolean;
var
gx, gy: Integer;
begin
end;
-procedure TBodyGrid.insert (body: TBodyProxy);
-
- function inserter (grida: Integer): Boolean;
- var
- cidx: Integer;
+function TBodyGridBase.inserter (grida: Integer): Boolean;
+var
+ cidx: Integer;
+ pc: PInteger;
+ {$IFDEF grid_use_buckets}
+ pi: PGridCell;
+ f: Integer;
+ {$ENDIF}
+begin
+ result := false; // never stop
+ // add body to the given grid cell
+ pc := @mGrid[grida];
+ {$IFDEF grid_use_buckets}
+ if (pc^ <> -1) then
begin
- result := false; // never stop
- // add body to the given grid cell
- cidx := allocCell();
- //e_WriteLog(Format(' 01: allocated cell for grid coords (%d,%d), body coords:(%d,%d): #%d', [gx, gy, dx, dy, cidx]), MSG_NOTIFY);
- mCells[cidx].body := body;
- mCells[cidx].next := mGrid[grida];
- mGrid[grida] := cidx;
+ pi := @mCells[pc^];
+ f := 0;
+ for f := 0 to High(TGridCell.bodies) do
+ begin
+ if (pi.bodies[f] = -1) then
+ begin
+ // can add here
+ pi.bodies[f] := mUData;
+ if (f+1 < Length(TGridCell.bodies)) then pi.bodies[f+1] := -1;
+ exit;
+ end;
+ end;
end;
+ // either no room, or no cell at all
+ cidx := allocCell();
+ mCells[cidx].bodies[0] := mUData;
+ mCells[cidx].bodies[1] := -1;
+ mCells[cidx].next := pc^;
+ pc^ := cidx;
+ {$ELSE}
+ cidx := allocCell();
+ //e_WriteLog(Format(' 01: allocated cell for grid coords (%d,%d), body coords:(%d,%d): #%d', [gx, gy, dx, dy, cidx]), MSG_NOTIFY);
+ mCells[cidx].body := mUData;
+ mCells[cidx].next := pc^;
+ pc^ := cidx;
+ {$ENDIF}
+end;
+
+procedure TBodyGridBase.insert (body: TBodyProxyId);
var
px: PBodyProxyRec;
+ oudata: Integer;
begin
if (body < 0) or (body > High(mProxies)) then exit; // just in case
px := @mProxies[body];
+ oudata := mUData;
+ mUData := body;
forGridRect(px.mX, px.mY, px.mWidth, px.mHeight, inserter);
+ mUData := oudata;
end;
-// absolutely not tested
-procedure TBodyGrid.remove (body: TBodyProxy);
-
- function remover (grida: Integer): Boolean;
- var
- pidx, idx, tmp: Integer;
+function TBodyGridBase.remover (grida: Integer): Boolean;
+var
+ pidx, idx, tmp, f: Integer;
+ pc: PGridCell;
+begin
+ result := false; // never stop
+ // find and remove cell
+ pidx := -1;
+ idx := mGrid[grida];
+ while (idx >= 0) do
begin
- result := false; // never stop
- // find and remove cell
- pidx := -1;
- idx := mGrid[grida];
- while idx >= 0 do
+ tmp := mCells[idx].next;
+ {$IFDEF grid_use_buckets}
+ pc := @mCells[idx];
+ f := 0;
+ while (f < High(TGridCell.bodies)) do
begin
- tmp := mCells[idx].next;
- if (mCells[idx].body = body) then
- begin
- if (pidx = -1) then mGrid[grida] := tmp else mCells[pidx].next := tmp;
- freeCell(idx);
- break; // assume that we cannot have one object added to bucket twice
- end
- else
+ if (pc.bodies[f] = mUData) then
begin
- pidx := idx;
+ // i found her!
+ if (f = 0) and (pc.bodies[1] = -1) then
+ begin
+ // this cell contains no elements, remove it
+ tmp := mCells[idx].next;
+ if (pidx = -1) then mGrid[grida] := tmp else mCells[pidx].next := tmp;
+ freeCell(idx);
+ end
+ else
+ begin
+ // remove element from bucket
+ Inc(f);
+ while (f < High(TGridCell.bodies)) do
+ begin
+ pc.bodies[f-1] := pc.bodies[f];
+ if (pc.bodies[f] = -1) then break;
+ Inc(f);
+ end;
+ pc.bodies[High(TGridCell.bodies)] := -1; // just in case
+ end;
+ exit; // assume that we cannot have one object added to bucket twice
end;
- idx := tmp;
+ Inc(f);
+ end;
+ {$ELSE}
+ if (mCells[idx].body = mUData) then
+ begin
+ if (pidx = -1) then mGrid[grida] := tmp else mCells[pidx].next := tmp;
+ freeCell(idx);
+ exit; // assume that we cannot have one object added to bucket twice
end;
+ {$ENDIF}
+ pidx := idx;
+ idx := tmp;
end;
+end;
+
+// absolutely not tested
+procedure TBodyGridBase.remove (body: TBodyProxyId);
var
px: PBodyProxyRec;
+ oudata: Integer;
begin
if (body < 0) or (body > High(mProxies)) then exit; // just in case
px := @mProxies[body];
+ oudata := mUData;
+ mUData := body;
forGridRect(px.mX, px.mY, px.mWidth, px.mHeight, remover);
+ mUData := oudata;
end;
-function TBodyGrid.insertBody (aObj: TObject; aX, aY, aWidth, aHeight: Integer; aTag: Integer=0): TBodyProxy;
+function TBodyGridBase.insertBody (aObj: ITP; aX, aY, aWidth, aHeight: Integer; aTag: Integer=0): TBodyProxyId;
begin
result := allocProxy(aX, aY, aWidth, aHeight, aObj, aTag);
insert(result);
end;
-procedure TBodyGrid.removeBody (aObj: TBodyProxy);
+procedure TBodyGridBase.removeBody (aObj: TBodyProxyId);
begin
if (aObj < 0) or (aObj > High(mProxies)) then exit; // just in case
- removeBody(aObj);
+ remove(aObj);
freeProxy(aObj);
end;
-procedure TBodyGrid.moveResizeBody (body: TBodyProxy; dx, dy, sx, sy: Integer);
+procedure TBodyGridBase.moveResizeBody (body: TBodyProxyId; dx, dy, sx, sy: Integer);
var
px: PBodyProxyRec;
begin
insert(body);
end;
-procedure TBodyGrid.moveBody (body: TBodyProxy; dx, dy: Integer);
+procedure TBodyGridBase.moveBody (body: TBodyProxyId; dx, dy: Integer);
begin
moveResizeBody(body, dx, dy, 0, 0);
end;
-procedure TBodyGrid.resizeBody (body: TBodyProxy; sx, sy: Integer);
+procedure TBodyGridBase.resizeBody (body: TBodyProxyId; sx, sy: Integer);
begin
moveResizeBody(body, 0, 0, sx, sy);
end;
-function TBodyGrid.forEachInAABB (x, y, w, h: Integer; cb: GridQueryCB): Boolean;
- function iterator (grida: Integer): Boolean;
- var
- idx: Integer;
- px: PBodyProxyRec;
+function TBodyGridBase.iterator (grida: Integer): Boolean;
+var
+ idx: Integer;
+ px: PBodyProxyRec;
+ {$IFDEF grid_use_buckets}
+ pi: PGridCell;
+ f: Integer;
+ {$ENDIF}
+begin
+ result := false;
+ idx := mGrid[grida];
+ while (idx >= 0) do
begin
- result := false;
- idx := mGrid[grida];
- while (idx >= 0) do
+ {$IFDEF grid_use_buckets}
+ pi := @mCells[idx];
+ for f := 0 to High(TGridCell.bodies) do
begin
- if (mCells[idx].body <> -1) then
+ if (pi.bodies[f] = -1) then break;
+ px := @mProxies[pi.bodies[f]];
+ if (px.mQueryMark <> mLastQuery) and ((mTagMask = -1) or ((px.mTag and mTagMask) <> 0)) then
begin
- px := @mProxies[mCells[idx].body];
- if (px.mQueryMark <> mLastQuery) then
- begin
- //e_WriteLog(Format(' query #%d body hit: (%d,%d)-(%dx%d) tag:%d', [mLastQuery, mCells[idx].body.mX, mCells[idx].body.mY, mCells[idx].body.mWidth, mCells[idx].body.mHeight, mCells[idx].body.mTag]), MSG_NOTIFY);
- px.mQueryMark := mLastQuery;
- if (cb(px.mObj, px.mTag)) then begin result := true; exit; end;
- end;
+ //e_WriteLog(Format(' query #%d body hit: (%d,%d)-(%dx%d) tag:%d', [mLastQuery, mCells[idx].body.mX, mCells[idx].body.mY, mCells[idx].body.mWidth, mCells[idx].body.mHeight, mCells[idx].body.mTag]), MSG_NOTIFY);
+ px.mQueryMark := mLastQuery;
+ if (mItCB(px.mObj, px.mTag)) then begin result := true; exit; end;
+ end;
+ end;
+ idx := pi.next;
+ {$ELSE}
+ if (mCells[idx].body <> -1) then
+ begin
+ px := @mProxies[mCells[idx].body];
+ if (px.mQueryMark <> mLastQuery) and ((mTagMask = -1) or ((px.mTag and mTagMask) <> 0)) then
+ begin
+ //e_WriteLog(Format(' query #%d body hit: (%d,%d)-(%dx%d) tag:%d', [mLastQuery, mCells[idx].body.mX, mCells[idx].body.mY, mCells[idx].body.mWidth, mCells[idx].body.mHeight, mCells[idx].body.mTag]), MSG_NOTIFY);
+ px.mQueryMark := mLastQuery;
+ if (mItCB(px.mObj, px.mTag)) then begin result := true; exit; end;
end;
- idx := mCells[idx].next;
end;
+ idx := mCells[idx].next;
+ {$ENDIF}
end;
+end;
+function TBodyGridBase.forEachInAABB (x, y, w, h: Integer; cb: TGridQueryCB; tagmask: Integer=-1): Boolean;
var
idx: Integer;
+ otagmask: Integer;
+ ocb: TGridQueryCB;
begin
result := false;
if not assigned(cb) then exit;
end;
//e_WriteLog(Format('grid: query #%d: (%d,%d)-(%dx%d)', [mLastQuery, minx, miny, maxx, maxy]), MSG_NOTIFY);
+ otagmask := mTagMask;
+ mTagMask := tagmask;
+ ocb := mItCB;
+ mItCB := cb;
result := forGridRect(x, y, w, h, iterator);
-end;
-
-
-function TBodyGrid.getProxyForBody (aObj: TObject; x, y, w, h: Integer): TBodyProxy;
-var
- res: TBodyProxy = -1;
-
- function iterator (grida: Integer): Boolean;
- var
- idx: Integer;
- begin
- result := false;
- idx := mGrid[grida];
- while idx >= 0 do
- begin
- if (mCells[idx].body <> -1) and (mProxies[mCells[idx].body].mObj = aObj) then
- begin
- result := true;
- res := mCells[idx].body;
- exit;
- end;
- idx := mCells[idx].next;
- end;
- end;
-
-begin
- result := -1;
- if (aObj = nil) then exit;
- forGridRect(x, y, w, h, iterator);
- result := res;
-end;
-
-
-// ////////////////////////////////////////////////////////////////////////// //
-constructor TBinaryHeapObj.Create (alessfn: TBinaryHeapLessFn);
-begin
- if not assigned(alessfn) then raise Exception.Create('wutafuck?!');
- lessfn := alessfn;
- SetLength(elem, 8192); // 'cause why not?
- elemUsed := 0;
-end;
-
-
-destructor TBinaryHeapObj.Destroy ();
-begin
- inherited;
-end;
-
-
-procedure TBinaryHeapObj.clear ();
-begin
- elemUsed := 0;
-end;
-
-
-procedure TBinaryHeapObj.heapify (root: Integer);
-var
- smallest, right: Integer;
- tmp: TObject;
-begin
- while true do
- begin
- smallest := 2*root+1; // left child
- if (smallest >= elemUsed) then break; // anyway
- right := smallest+1; // right child
- if not lessfn(elem[smallest], elem[root]) then smallest := root;
- if (right < elemUsed) and (lessfn(elem[right], elem[smallest])) then smallest := right;
- if (smallest = root) then break;
- // swap
- tmp := elem[root];
- elem[root] := elem[smallest];
- elem[smallest] := tmp;
- root := smallest;
- end;
-end;
-
-
-procedure TBinaryHeapObj.insert (val: TObject);
-var
- i, par: Integer;
-begin
- if (val = nil) then exit;
- i := elemUsed;
- if (i = Length(elem)) then SetLength(elem, Length(elem)+16384); // arbitrary number
- Inc(elemUsed);
- while (i <> 0) do
- begin
- par := (i-1) div 2; // parent
- if not lessfn(val, elem[par]) then break;
- elem[i] := elem[par];
- i := par;
- end;
- elem[i] := val;
-end;
-
-function TBinaryHeapObj.front (): TObject;
-begin
- if elemUsed > 0 then result := elem[0] else result := nil;
-end;
-
-
-procedure TBinaryHeapObj.popFront ();
-begin
- if (elemUsed > 0) then
- begin
- Dec(elemUsed);
- elem[0] := elem[elemUsed];
- heapify(0);
- end;
+ mTagMask := otagmask;
+ mItCB := ocb;
end;