feat(distance): add INeighborhood.includesPosition()

- update all impls: Nearest/KNearest/Radial
zdccool · Jun 9, 2023 · 2018e41 · 2018e41
1 parent 065e40c
commit 2018e41
Show file tree

Hide file tree

Showing 4 changed files with 20 additions and 0 deletions.
diff --git a/packages/distance/src/api.ts b/packages/distance/src/api.ts
@@ -59,6 +59,14 @@ export interface INeighborhood<P, T> extends IReset {
 	 */
 	includesDistance(d: number, eucledian?: boolean): boolean;
 
+	/**
+	 * Computes distance metric between `pos` and this neighborhood's target
+	 * pos. Returns true if result is <= current radius.
+	 *
+	 * @param pos
+	 */
+	includesPosition(pos: P): boolean;
+
 	/**
 	 * Computes distance metric between `pos` and this neighborhood's target
 	 * pos. If result distance is <= current radius, adds `val` to neighborhood

diff --git a/packages/distance/src/knearest.ts b/packages/distance/src/knearest.ts
@@ -102,6 +102,10 @@ export class KNearest<D, T>
 		return (eucledian ? this.dist.to(d) : d) <= this._currR;
 	}
 
+	includesPosition(pos: D) {
+		return this.dist.metric(this.target, pos) < this._currR;
+	}
+
 	consider(pos: D, val: T) {
 		const d = this.dist.metric(this.target, pos);
 		if (d <= this._currR) {

diff --git a/packages/distance/src/nearest.ts b/packages/distance/src/nearest.ts
@@ -48,6 +48,10 @@ export class Nearest<D, T>
 		return (eucledian ? this.dist.to(d) : d) <= this._currR;
 	}
 
+	includesPosition(pos: D) {
+		return this.dist.metric(this.target, pos) < this._currR;
+	}
+
 	consider(pos: D, val: T) {
 		const d = this.dist.metric(this.target, pos);
 		if (d <= this._currR) {

diff --git a/packages/distance/src/radial.ts b/packages/distance/src/radial.ts
@@ -83,6 +83,10 @@ export class Radial<D, T>
 		return (eucledian ? this.dist.to(d) : d) <= this._r;
 	}
 
+	includesPosition(pos: D) {
+		return this.dist.metric(this.target, pos) < this._r;
+	}
+
 	consider(pos: D, val: T) {
 		const d = this.dist.metric(this.target, pos);
 		if (d <= this._r) {