feat: ✨ distance can be a function

antvis · Sep 4, 2020 · b0b251c · b0b251c
1 parent c487154
commit b0b251c
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/packages/x6/src/model/model.ts b/packages/x6/src/model/model.ts
@@ -727,11 +727,18 @@ export class Model extends Basecoat<Model.EventArgs> {
     return result
   }
 
-  protected matchDistance(distance: number, preset?: number | number[]) {
+  protected matchDistance(
+    distance: number,
+    preset?: number | number[] | ((d: number) => boolean),
+  ) {
     if (preset == null) {
       return true
     }
 
+    if (typeof preset === 'function') {
+      return preset(distance)
+    }
+
     if (Array.isArray(preset) && preset.includes(distance)) {
       return true
     }
@@ -1178,7 +1185,7 @@ export namespace Model {
   }
 
   export interface GetPredecessorsOptions extends Cell.GetDescendantsOptions {
-    distance?: number | number[]
+    distance?: number | number[] | ((distance: number) => boolean)
   }
 }