feat: optimize snap method

F-star · Jul 3, 2024 · 2862a3b · 2862a3b
1 parent 2e92805
commit 2862a3b
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 3 deletions.
diff --git a/packages/geo/src/geo/geo_line.ts b/packages/geo/src/geo/geo_line.ts
@@ -35,7 +35,7 @@ export const closestPtOnLine = (
 /**
  * get closest point of polar coords. 0, 45, 90, 135, 150
  */
-export const closestPolarPt = (center: IPoint, p: IPoint, count = 4) => {
+export const getPolarTrackSnapPt = (center: IPoint, p: IPoint, count = 4) => {
   let closestPt: IPoint = { x: 0, y: 0 };
   let closestDist = Infinity;
   for (let i = 1; i <= count; i++) {
@@ -46,6 +46,9 @@ export const closestPolarPt = (center: IPoint, p: IPoint, count = 4) => {
     };
     const { point } = closestPtOnLine(center, pt, p);
     const dist = distance(point, p);
+    if (dist === 0) {
+      return point;
+    }
     if (dist < closestDist) {
       closestDist = dist;
       closestPt = point;

diff --git a/packages/geo/src/geo/geo_resize_line.ts b/packages/geo/src/geo/geo_resize_line.ts
@@ -1,6 +1,6 @@
 import { type IPoint, type ITransformRect } from '../type';
 import { getSweepAngle } from './geo_angle';
-import { closestPolarPt } from './geo_line';
+import { getPolarTrackSnapPt } from './geo_line';
 import { Matrix } from './geo_matrix_class';
 import { distance } from './geo_point';
 
@@ -49,7 +49,7 @@ export const resizeLine = (
   }
 
   if (options.keepPolarSnap) {
-    newPos = closestPolarPt(globalOrigin, newPos);
+    newPos = getPolarTrackSnapPt(globalOrigin, newPos);
   }
 
   let width = distance(newPos, globalOrigin);