support virtual node and master station

Author: langonginc

src/components/svg-canvas-graph.tsx

@@ -19,14 +19,14 @@ import {
     setSelected,
 } from '../redux/runtime/runtime-slice';
 import { getMousePosition, pointerPosToSVGCoord, roundToNearestN } from '../util/helpers';
+import { getNearestPolyline, getPolylineDistance, isNodeSupportPolyline } from '../util/graph';
 import { makeParallelIndex } from '../util/parallel';
 import { getLines, getNodes } from '../util/process-elements';
 import SvgLayer from './svg-layer';
 import { linePaths } from './svgs/lines/lines';
 import singleColor from './svgs/lines/styles/single-color';
 import miscNodes from './svgs/nodes/misc-nodes';
 import { default as stations } from './svgs/stations/stations';
-import { getNearestPolyline, getPolylineDistance } from '../util/graph';
 
 const connectableNodesType = [
     ...Object.values(StationType),
@@ -112,7 +112,7 @@ const SvgCanvas = () => {
                 let newX = toX;
                 let newY = toY;
 
-                if (node.startsWith('stn')) {
+                if (isNodeSupportPolyline(node, graph.current)) {
                     if (activePolylines.length !== 0) {
                         Object.values(activePolylines).forEach(p => {
                             const d = getPolylineDistance(p, toX, toY);

src/redux/runtime/runtime-slice.ts

@@ -7,9 +7,9 @@ import { MAX_MASTER_NODE_FREE, MAX_MASTER_NODE_PRO } from '../../constants/maste
 import { MiscNodeType } from '../../constants/nodes';
 import i18n from '../../i18n/config';
 import { countParallelLines, MAX_PARALLEL_LINES_FREE, MAX_PARALLEL_LINES_PRO } from '../../util/parallel';
+import { getPolylines } from '../../util/graph';
 import { setAutoParallel } from '../app/app-slice';
 import { redoAction, undoAction } from '../param/param-slice';
-import { getPolylines } from '../../util/graph';
 
 /**
  * RuntimeState contains all the data that do not require any persistence.

src/util/graph.ts

@@ -113,6 +113,16 @@ export const getMasterNodeTypes = (graph: MultiDirectedGraph<NodeAttributes, Edg
     return newList;
 };
 
+export const isNodeSupportPolyline = (
+    node: StnId | MiscNodeId,
+    graph: MultiDirectedGraph<NodeAttributes, EdgeAttributes, GraphAttributes>
+) =>
+    node.startsWith('stn') ||
+    (node.startsWith('misc_node') && graph.getNodeAttribute(node, 'type') === MiscNodeType.Virtual) ||
+    (node.startsWith('misc_node') &&
+        graph.getNodeAttribute(node, 'type') === MiscNodeType.Master &&
+        graph.getNodeAttributes(node)[MiscNodeType.Master]!.nodeType === 'Station');
+
 export const getPolylines = (graph: MultiDirectedGraph<NodeAttributes, EdgeAttributes, GraphAttributes>): Polylines => {
     const polylinesX: Polyline[] = [];
     const polylinesY: Polyline[] = [];
@@ -122,7 +132,7 @@ export const getPolylines = (graph: MultiDirectedGraph<NodeAttributes, EdgeAttri
     // const sortFunc = (x: Polyline, y: Polyline) => y.c - x.c;
 
     graph
-        .filterNodes(node => node.startsWith('stn'))
+        .filterNodes(node => isNodeSupportPolyline(node as StnId | MiscNodeId, graph))
         .forEach(node => {
             const x = graph.getNodeAttribute(node, 'x');
             const y = graph.getNodeAttribute(node, 'y');