道岔和区段吸附

src/graphics/turnout/Turnout.ts

@@ -104,6 +104,10 @@ export class Turnout extends JlGraphic {
     return this.getStates<ITurnoutState>();
   }
 
+  getPortPoints() {
+    return [this.datas.pointA, this.datas.pointB, this.datas.pointC];
+  }
+
   doRepaint(): void {
     const { pointA, pointB, pointC } = this.datas;
 

src/graphics/turnout/TurnoutDrawAssistant.ts

@@ -1,4 +1,5 @@
 import {
+  AbsorbablePosition,
   DraggablePoint,
   GraphicApp,
   GraphicDrawAssistant,
@@ -25,6 +26,8 @@ import {
 } from 'pixi.js';
 import { GraphicEditPlugin } from 'src/jl-graphic/plugins/GraphicEditPlugin';
 import Vector2 from 'src/jl-graphic/math/Vector2';
+import { Section } from '../section/Section';
+import AbsorbablePoint from 'src/jl-graphic/graphic/AbsorbablePosition';
 
 export class TurnoutDraw extends GraphicDrawAssistant<
   TurnoutTemplate,
@@ -94,6 +97,41 @@ export class TurnoutHitArea implements IHitArea {
   }
 }
 
+function buildAbsorbablePositions(turnout: Turnout): AbsorbablePosition[] {
+  const aps: AbsorbablePosition[] = [];
+
+  const sections = turnout.queryStore.queryByType<Section>(Section.Type);
+  sections.forEach((section) => {
+    const ps = new AbsorbablePoint(
+      section.localToCanvasPoint(section.getStartPoint())
+    );
+    const pe = new AbsorbablePoint(
+      section.localToCanvasPoint(section.getEndPoint())
+    );
+    aps.push(ps, pe);
+  });
+
+  const turnouts = turnout.queryStore.queryByType<Turnout>(Turnout.Type);
+  turnouts.forEach((otherTurnout) => {
+    if (turnout.id === otherTurnout.id) return;
+    otherTurnout.getPortPoints().forEach((portPoint) => {
+      aps.push(new AbsorbablePoint(otherTurnout.localToCanvasPoint(portPoint)));
+    });
+  });
+
+  return aps;
+}
+
+function onEditPointCreate(turnout: Turnout, dp: DraggablePoint) {
+  dp.on('transformstart', (e: GraphicTransformEvent) => {
+    if (e.isShift()) {
+      turnout.getGraphicApp().setOptions({
+        absorbablePositions: buildAbsorbablePositions(turnout),
+      });
+    }
+  });
+}
+
 export class TurnoutPointsInteractionPlugin extends GraphicInteractionPlugin<Turnout> {
   static Name = 'TurnoutPointsDrag';
   static init(app: JlDrawApp) {
@@ -130,7 +168,7 @@ export class TurnoutPointsInteractionPlugin extends GraphicInteractionPlugin<Tur
       TurnoutEditPlugin.Name
     );
     if (!tep) {
-      tep = new TurnoutEditPlugin(turnout);
+      tep = new TurnoutEditPlugin(turnout, { onEditPointCreate });
       turnout.addAssistantAppend(tep);
     }
     tep.reset();
@@ -152,14 +190,22 @@ export class TurnoutPointsInteractionPlugin extends GraphicInteractionPlugin<Tur
   }
 }
 
+type onTurnoutEditPointCreate = (turnout: Turnout, dp: DraggablePoint) => void;
+
+export interface ITurnoutEditOptions {
+  onEditPointCreate?: onTurnoutEditPointCreate;
+}
+
 export class TurnoutEditPlugin extends GraphicEditPlugin<Turnout> {
   static Name = 'TurnoutEdit';
+  options: ITurnoutEditOptions;
   editPoints: DraggablePoint[] = [];
   labels: VectorText[] = [];
 
-  constructor(graphic: Turnout) {
+  constructor(graphic: Turnout, options?: ITurnoutEditOptions) {
     super(graphic);
     this.name = TurnoutEditPlugin.Name;
+    this.options = Object.assign({}, options);
     this.initEditPoints();
   }
   reset(): void {
@@ -208,6 +254,9 @@ export class TurnoutEditPlugin extends GraphicEditPlugin<Turnout> {
 
         this.graphic.repaint();
       });
+      if (this.options.onEditPointCreate) {
+        this.options.onEditPointCreate(this.graphic, dp);
+      }
       this.editPoints.push(dp);
     });
     this.addChild(...this.editPoints);