新增flag 帮助处理位掩码

2020-06-12 08:47:13 +08:00
parent ad68f0e1a0
commit 6fa56dd572
12 changed files with 537 additions and 8 deletions
--- a/source/src/Physics/Verlet/SpatialHash.ts
+++ b/source/src/Physics/Verlet/SpatialHash.ts
@@ -1,7 +1,111 @@
 class SpatialHash {
-    public overlapCircle(circleCenter: Vector2, radius: number, results: any[], layerMask){
+    private _raycastParser: RaycastResultParser;
+    private _cellSize: number;
+    private _inverseCellSize: number;
+    private _overlapTestCircle: Circle;
+    private _tempHashSet: Collider[] = [];
+    private _cellDict: NumberDictionary = new NumberDictionary();
+
+    constructor(cellSize: number = 100){
+        this._cellSize = cellSize;
+        this._inverseCellSize = 1 / this._cellSize;
+        this._raycastParser = new RaycastResultParser();
+    }
+
+    public overlapCircle(circleCenter: Vector2, radius: number, results: Collider[], layerMask){
+        let bounds = new Rectangle(circleCenter.x - radius, circleCenter.y - radius, radius * 2, radius * 2);
+
+        this._overlapTestCircle.radius = radius;
+        this._overlapTestCircle.position = circleCenter;
+
        let resultCounter = 0;
+        let potentials = this.aabbBroadphase(bounds, null, layerMask);
+        potentials.forEach(collider => {
+            if (resultCounter == results.length)
+                return resultCounter;
+        });

        return resultCounter;
    }
+
+    public aabbBroadphase(bounds: Rectangle, excludeCollider: Collider, layerMask: number){
+        this._tempHashSet.length = 0;
+
+        let p1 = this.cellCoords(bounds.x, bounds.y);
+        let p2 = this.cellCoords(bounds.right, bounds.bottom);
+
+        for (let x = p1.x; x <= p2.x; x ++){
+            for (let y = p1.y; y <= p2.y; y ++){
+                let cell = this.cellAtPosition(x, y);
+                if (!cell)
+                    continue;
+
+                for (let i = 0; i < cell.length; i ++){
+                    let collider = cell[i];
+
+                    if (collider == excludeCollider || !Flags.isFlagSet(layerMask, collider.physicsLayer))
+                        continue;
+
+                    if (bounds.intersects(collider.bounds))
+                        this._tempHashSet.push(collider);
+                }
+            }
+        }
+
+        return this._tempHashSet;
+    }
+
+    private cellAtPosition(x: number, y: number, createCellIfEmpty: boolean = false){
+        let cell: Collider[] = this._cellDict.tryGetValue(x, y);
+        if (!cell){
+            if (createCellIfEmpty){
+                cell = [];
+                this._cellDict.add(x, y, cell);
+            }
+        }
+        return cell;
+    }
+
+    private cellCoords(x: number, y: number): Point {
+        return new Point(Math.floor(x * this._inverseCellSize), Math.floor(y * this._inverseCellSize));
+    }
+}
+
+class RaycastResultParser {
+
+}
+
+class NumberDictionary {
+    private _store: Map<number, Collider[]> = new Map<number, Collider[]>();
+
+    private getKey(x: number, y: number): number{
+        return x << 32 | y;
+    }
+
+    public add(x: number, y: number, list: Collider[]){
+        this._store.set(this.getKey(x, y), list);
+    }
+
+    public remove(obj: Collider){
+        this._store.forEach(list => {
+            if (list.contains(obj))
+                list.remove(obj);
+        })
+    }
+
+    public tryGetValue(x: number, y: number): Collider[]{
+        return this._store.get(this.getKey(x, y));
+    }
+
+    public getAllObjects(): Collider[]{
+        let set: Collider[] = [];
+
+        this._store.forEach(list => set.concat(list));
+
+        return set;
+    }
+
+    public clear(){
+        this._store.clear();
+    }
 }