@logicflow/core/src/algorithm/outline.ts

LogicFlow, help you quickly create flowcharts

import { ModelType } from '../constant'
import {
  Model,
  BaseNodeModel,
  BaseEdgeModel,
  LineEdgeModel,
  BezierEdgeModel,
  PolylineEdgeModel,
} from '../model'
import { points2PointsList, getBBoxOfPoints, getBezierPoints } from '../util'

import OutlineInfo = Model.OutlineInfo

// 获取节点的out
export const getNodeOutline = (node: BaseNodeModel): OutlineInfo => {
  const { x, y, width, height } = node
  return {
    x: x - width / 2,
    y: y - height / 2,
    x1: x + width / 2,
    y1: y + height / 2,
  }
}

export const getLineOutline = (edge: LineEdgeModel): OutlineInfo => {
  const { startPoint, endPoint } = edge
  const x = (startPoint.x + endPoint.x) / 2
  const y = (startPoint.y + endPoint.y) / 2
  const width = Math.abs(startPoint.x - endPoint.x) + 10
  const height = Math.abs(startPoint.y - endPoint.y) + 10
  return {
    x: x - width / 2,
    y: y - height / 2,
    x1: x + width / 2,
    y1: y + height / 2,
  }
}

export const getPolylineOutline = (edge: PolylineEdgeModel): OutlineInfo => {
  const { points } = edge
  const pointsList = points2PointsList(points)
  const bbox = getBBoxOfPoints(pointsList, 8)
  const { x, y, width, height } = bbox
  return {
    x: x - width / 2,
    y: y - height / 2,
    x1: x + width / 2,
    y1: y + height / 2,
  }
}

export const getBezierOutline = (edge: BezierEdgeModel): OutlineInfo => {
  const { path } = edge
  const pointsList = getBezierPoints(path)
  const bbox = getBBoxOfPoints(pointsList, 8)
  const { x, y, width, height } = bbox
  return {
    x: x - width / 2,
    y: y - height / 2,
    x1: x + width / 2,
    y1: y + height / 2,
  }
}

export const getEdgeOutline = (
  edge: BaseEdgeModel,
): OutlineInfo | undefined => {
  if (edge.modelType === ModelType.LINE_EDGE) {
    return getLineOutline(edge as LineEdgeModel)
  }
  if (edge.modelType === ModelType.POLYLINE_EDGE) {
    return getPolylineOutline(edge as PolylineEdgeModel)
  }
  if (edge.modelType === ModelType.BEZIER_EDGE) {
    return getBezierOutline(edge as BezierEdgeModel)
  }
}