nodes/frontend/src/lib/components/edges/Edge.svelte

<script context="module">
  const color = new Color(0x202020);
  color.convertLinearToSRGB();

  const color2 = color.clone();
  color2.convertSRGBToLinear();

  const circleMaterial = new MeshBasicMaterial({
    color,
    toneMapped: false,
  });
</script>

<script lang="ts">
  import { T } from "@threlte/core";
  import { MeshLineGeometry, MeshLineMaterial } from "@threlte/extras";
  import { MeshBasicMaterial, type Mesh, LineBasicMaterial } from "three";
  import { Color } from "three/src/math/Color.js";
  import { CubicBezierCurve } from "three/src/extras/curves/CubicBezierCurve.js";
  import { Vector3 } from "three/src/math/Vector3.js";
  import { Vector2 } from "three/src/math/Vector2.js";

  export let from: { x: number; y: number };
  export let to: { x: number; y: number };

  let samples = 5;

  const curve = new CubicBezierCurve(
    new Vector2(from.x, from.y),
    new Vector2(from.x + 2, from.y),
    new Vector2(to.x - 2, to.y),
    new Vector2(to.x, to.y),
  );

  let points: Vector3[] = [];

  let last_from_x = 0;
  let last_from_y = 0;

  let mesh: Mesh;

  export const update = function (force = false) {
    if (!force) {
      const new_x = from.x + to.x;
      const new_y = from.y + to.y;
      if (last_from_x === new_x && last_from_y === new_y) {
        return;
      }
      last_from_x = new_x;
      last_from_y = new_y;
    }

    const mid = new Vector2((from.x + to.x) / 2, (from.y + to.y) / 2);

    const length = Math.floor(
      Math.sqrt(Math.pow(to.x - from.x, 2) + Math.pow(to.y - from.y, 2)) / 4,
    );
    samples = Math.min(Math.max(10, length), 100);

    curve.v0.set(from.x, from.y);
    curve.v1.set(mid.x, from.y);
    curve.v2.set(mid.x, to.y);
    curve.v3.set(to.x, to.y);

    points = curve.getPoints(samples).map((p) => new Vector3(p.x, 0, p.y));
  };

  update();
  $: if (from || to) {
    update();
  }
</script>

<T.Mesh
  position.x={from.x}
  position.z={from.y}
  position.y={0.8}
  rotation.x={-Math.PI / 2}
  material={circleMaterial}
>
  <T.CircleGeometry args={[0.3, 16]} />
</T.Mesh>

<T.Mesh
  position.x={to.x}
  position.z={to.y}
  position.y={0.8}
  rotation.x={-Math.PI / 2}
  material={circleMaterial}
>
  <T.CircleGeometry args={[0.3, 16]} />
</T.Mesh>

<T.Mesh position.y={0.5} bind:ref={mesh}>
  {#key samples}
    <MeshLineGeometry {points} />
  {/key}
  <MeshLineMaterial
    width={4}
    attenuate={false}
    color={color2}
    toneMapped={false}
  />
</T.Mesh>