OpenFrontIO/tests/MiniMapTransformerPlanSegments.test.ts

import { describe, expect, it } from "vitest";
import { GameMapImpl } from "../src/core/game/GameMap";
import { densePathToKeypointSegments } from "../src/core/game/MotionPlans";

function makeMap(width: number, height: number): GameMapImpl {
  return new GameMapImpl(width, height, new Uint8Array(width * height), 0);
}

function expandPlanDda(
  map: GameMapImpl,
  points: Uint32Array,
  segmentSteps: Uint32Array,
): number[] {
  const out: number[] = [];
  if (points.length === 0) return out;
  out.push(points[0] >>> 0);
  for (let i = 0; i < segmentSteps.length; i++) {
    const steps = segmentSteps[i] >>> 0;
    const a = points[i] >>> 0;
    const b = points[i + 1] >>> 0;
    const ax = map.x(a);
    const ay = map.y(a);
    const bx = map.x(b);
    const by = map.y(b);
    const dx = bx - ax;
    const dy = by - ay;
    for (let t = 1; t <= steps; t++) {
      out.push(
        map.ref(
          Math.round(ax + (dx * t) / steps),
          Math.round(ay + (dy * t) / steps),
        ) >>> 0,
      );
    }
  }
  return out;
}

describe("densePathToKeypointSegments", () => {
  it("expands back to the dense path for axis segments", () => {
    const map = makeMap(10, 10);

    const dense = [
      map.ref(1, 1),
      map.ref(2, 1),
      map.ref(3, 1),
      map.ref(4, 1),
      map.ref(4, 2),
      map.ref(4, 3),
      map.ref(4, 4),
    ];

    const plan = densePathToKeypointSegments(dense);
    expect(plan).not.toBeNull();
    if (!plan) return;

    const expanded = expandPlanDda(map, plan.points, plan.segmentSteps);
    expect(expanded).toEqual(dense.map((t) => t >>> 0));
  });
});