@antv/x6/src/registry/edge-anchor/orth.ts

JavaScript diagramming library that uses SVG and HTML for rendering

import { FunctionExt } from '../../common'
import { Line, type Point } from '../../geometry'
import {
  getPointAtEdge,
  type ResolveOptions,
  resolve,
} from '../node-anchor/util'
import { getClosestPoint } from './closest'
import type {
  EdgeAnchorDefinition,
  EdgeAnchorResolvedDefinition,
} from './index'

export interface OrthEndpointOptions extends ResolveOptions {
  fallbackAt?: number | string
}

const orthogonal: EdgeAnchorResolvedDefinition<OrthEndpointOptions> = function (
  view,
  magnet,
  refPoint,
  options,
): Point {
  const OFFSET = 1e6
  const path = view.getConnection()!
  const segmentSubdivisions = view.getConnectionSubdivisions()
  const vLine = new Line(
    refPoint.clone().translate(0, OFFSET),
    refPoint.clone().translate(0, -OFFSET),
  )
  const hLine = new Line(
    refPoint.clone().translate(OFFSET, 0),
    refPoint.clone().translate(-OFFSET, 0),
  )

  const vIntersections = vLine.intersect(path, {
    segmentSubdivisions,
  })

  const hIntersections = hLine.intersect(path, {
    segmentSubdivisions,
  })

  const intersections = []
  if (vIntersections) {
    intersections.push(...vIntersections)
  }
  if (hIntersections) {
    intersections.push(...hIntersections)
  }

  if (intersections.length > 0) {
    return refPoint.closest(intersections)!
  }

  if (options.fallbackAt != null) {
    return getPointAtEdge(view, options.fallbackAt)!
  }

  return FunctionExt.call(
    getClosestPoint,
    this,
    view,
    magnet,
    refPoint,
    options,
  )
}

export const orth = resolve<
  EdgeAnchorResolvedDefinition<OrthEndpointOptions>,
  EdgeAnchorDefinition<OrthEndpointOptions>
>(orthogonal)