import { describe, expect, it } from 'vitest';

import { jumpLinkSegments, minimumRangeBetween, routeBetween } from './jump-links';
import { StarNeighbourhood, StarPoint } from './star-neighbourhood';

/** Stars a parsec apart along x, so a chain's length is the number of hops it takes. */
function chain(count: number): StarNeighbourhood {
  return new StarNeighbourhood(Array.from({ length: count }, (_, i) => ({ id: i, x: i, y: 0, z: 0 })));
}

function index(points: StarPoint[]): StarNeighbourhood {
  return new StarNeighbourhood(points);
}

describe('routeBetween', () => {
  it('walks the chain a hop at a time when that is all the range allows', () => {
    const { route } = routeBetween(chain(5), 0, 4, 1.5);

    expect(route?.stars).toEqual([0, 1, 2, 3, 4]);
    expect(route?.totalPc).toBeCloseTo(4);
    expect(route?.longestHopPc).toBeCloseTo(1);
  });

  it('goes straight there when the range reaches, however many stars lie between', () => {
    // The direct crossing is never longer than a chain through anything — Euclid says so — so a
    // range that covers it makes it the answer, and the stars in between are just scenery.
    const { route } = routeBetween(chain(5), 0, 4, 5);

    expect(route?.stars).toEqual([0, 4]);
    expect(route?.totalPc).toBeCloseTo(4);
  });

  it('picks the shorter of two ways round when neither is a straight line', () => {
    // 0 to 3 is 10 pc, out of a 6 pc range. Two ways round, both inside it: through 1, barely
    // off the line, or through 2, well off it. Shorter is what "the way there" means.
    const { route } = routeBetween(
      index([
        { id: 0, x: 0, y: 0, z: 0 },
        { id: 1, x: 5, y: 0.5, z: 0 },
        { id: 2, x: 5, y: 3, z: 0 },
        { id: 3, x: 10, y: 0, z: 0 }
      ]),
      0,
      3,
      6
    );

    expect(route?.stars).toEqual([0, 1, 3]);
    expect(route?.totalPc).toBeCloseTo(10.05, 1);
  });

  it('finds nothing across a gap wider than the range', () => {
    const split = index([
      { id: 0, x: 0, y: 0, z: 0 },
      { id: 1, x: 1, y: 0, z: 0 },
      { id: 2, x: 20, y: 0, z: 0 }
    ]);

    expect(routeBetween(split, 0, 2, 5)).toEqual({ route: null, gaveUp: false });
  });

  it('answers nothing for a star that is not there, or for going nowhere', () => {
    const line = chain(3);

    expect(routeBetween(line, 0, 0, 2).route).toBeNull();
    expect(routeBetween(line, 0, 99, 2).route).toBeNull();
    expect(routeBetween(line, 0, 2, 0).route).toBeNull();
  });

  it('reports the longest hop, which is what the range has to cover', () => {
    const { route } = routeBetween(
      index([
        { id: 0, x: 0, y: 0, z: 0 },
        { id: 1, x: 1, y: 0, z: 0 },
        { id: 2, x: 5, y: 0, z: 0 }
      ]),
      0,
      2,
      4
    );

    expect(route?.longestHopPc).toBeCloseTo(4);
  });

  it('says it gave up rather than that there is no chain, once it has spent its budget', { timeout: 30_000 }, () => {
    // Nothing reaches the island, but the crowd around the departure is larger than the budget, so
    // the search stops without having looked everywhere the range reaches. Read as "no chain", that
    // is a confident wrong answer — and the range search downstream would build on it.
    // Cells sized for the range asked of them, as the real catalogue's are: a search that settles
    // 40 000 stars scans every cell it touches 40 000 times.
    const search = routeBetween(knotAndChain(1.5), 0, ISLAND, 1.5);

    expect(search.route).toBeNull();
    expect(search.gaveUp).toBe(true);
  });

  it('reports a dead end proved with the last star of the budget as a dead end, not a give-up', () => {
    // Exactly the budget's worth of stars reach each other, and the destination is not among them.
    // The search does look everywhere the range reaches, so "no chain" is what it found — but the
    // set is full at the end of it, and a budget read off the settled count says it gave up.
    const search = routeBetween(budgetExactly(), 0, BUDGET_ISLAND, 1.5);

    expect(search).toEqual({ route: null, gaveUp: false });
  });

  it('heads for the destination rather than exhausting a dense knot around the departure', () => {
    // The Gaia catalogue in miniature: a crowd around the departure, larger than the search's
    // budget, with the only way on a thin chain leading out of it. A search widening evenly from
    // the departure spends the budget on the crowd and never reaches the chain's far end.
    const { route } = routeBetween(knotAndChain(), 0, CHAIN_END, 1.5);

    expect(route).not.toBeNull();
    expect(route!.stars[route!.stars.length - 1]).toBe(CHAIN_END);
    expect(route!.longestHopPc).toBeLessThanOrEqual(1.5);
  });
});

/**
 * 45 000 stars scattered through the 30 pc cube around the origin, twenty times the density around
 * the real Sun and more than a search's budget, with a chain a parsec a hop running along x from
 * the origin out through the crowd and on to 75 pc — and one star at 500 pc that nothing reaches.
 */
const CHAIN_END = 75;
const ISLAND = 999;
function knotAndChain(cellSizePc?: number): StarNeighbourhood {
  let seed = 7;
  const random = () => ((seed = (seed * 1103515245 + 12345) % 2147483648) / 2147483648) * 30 - 15;
  const knot: StarPoint[] = Array.from({ length: 45000 }, (_, i) => ({ id: 1000 + i, x: random(), y: random(), z: random() }));
  const chainOut: StarPoint[] = Array.from({ length: CHAIN_END }, (_, i) => ({ id: i + 1, x: i + 1, y: 0, z: 0 }));
  return new StarNeighbourhood([{ id: 0, x: 0, y: 0, z: 0 }, ...knot, ...chainOut, { id: ISLAND, x: 500, y: 0, z: 0 }], cellSizePc);
}

/**
 * Exactly a search's budget of stars that reach one another — 40 000 a parsec apart along x, which
 * a 1.5 pc range walks end to end — and one 500 pc off that line, which nothing reaches. The dead
 * end is real and the search proves it, with the last star it is allowed.
 *
 * A line rather than a crowd because the count has to be exact: a random cloud dense enough to
 * connect leaves clumps the departure never reaches, and 39 662 of 40 000 settled is a budget that
 * was never spent.
 */
const BUDGET_ISLAND = 99_999;
function budgetExactly(): StarNeighbourhood {
  const line: StarPoint[] = Array.from({ length: 40_000 }, (_, i) => ({ id: i, x: i, y: 0, z: 0 }));
  return new StarNeighbourhood([...line, { id: BUDGET_ISLAND, x: 0, y: 500, z: 0 }], 1.5);
}

/**
 * 45 000 stars in a 10 pc cube — dense enough to stay one connected piece at half a parsec, where
 * walking it costs more than a search's budget — with a chain a parsec a hop leaving its edge for
 * 30 pc. Its cells are sized for the ranges asked of it, as the real catalogue's are for its own.
 */
const CROWD_CHAIN_END = 25;
const CROWD_ISLAND = 999999;
function crowdedKnot(): StarNeighbourhood {
  let seed = 11;
  const random = () => ((seed = (seed * 1103515245 + 12345) % 2147483648) / 2147483648) * 10 - 5;
  const knot: StarPoint[] = Array.from({ length: 45000 }, (_, i) => ({ id: 1000 + i, x: random(), y: random(), z: random() }));
  const chainOut: StarPoint[] = Array.from({ length: CROWD_CHAIN_END }, (_, i) => ({ id: i + 1, x: 5 + i + 1, y: 0, z: 0 }));
  // One star nothing reaches, for the questions that have no answer.
  return new StarNeighbourhood([{ id: 0, x: 0, y: 0, z: 0 }, ...knot, ...chainOut, { id: CROWD_ISLAND, x: 500, y: 0, z: 0 }], 0.25);
}

describe('minimumRangeBetween', () => {
  it('works out the range past a dense knot around the departure', () => {
    // Past the crowd the chain's hops of a parsec are the only way on, so a parsec is the
    // answer, to the half-step the panel rounds up to.
    expect(minimumRangeBetween(knotAndChain(), 0, CHAIN_END, 8).rangePc).toBeCloseTo(1, 1);
  });

  it('stops bisecting where a search gave up, and hands back a range that does work', { timeout: 30_000 }, () => {
    // Below the chain's own hop of a parsec, the crowd is still one connected piece and larger than
    // the budget, so those probes give up. Reading a give-up as "no chain at this range" is what
    // used to report ranges up to 29% wider than needed, and went on paying for probes whose
    // answers it could not use; the answer now is the narrowest range a chain was found at.
    const knot = crowdedKnot();

    const needed = minimumRangeBetween(knot, 0, CROWD_CHAIN_END, 1.2);

    expect(needed.least).toBe(false);
    expect(needed.rangePc).not.toBeNull();
    expect(routeBetween(knot, 0, CROWD_CHAIN_END, needed.rangePc!).route).not.toBeNull();
    // Narrower than the ceiling's own route, too: stopping before the bisection has found a range
    // of its own hands back the ceiling, which is the control's maximum — the question, not an answer.
    expect(needed.rangePc!).toBeLessThan(routeBetween(knot, 0, CROWD_CHAIN_END, 1.2).route!.longestHopPc);
  });

  it('names the shortest range that opens a way through', () => {
    // Hops of 1 and 4: no range under 4 connects them, and 4 exactly does.
    const stepped = index([
      { id: 0, x: 0, y: 0, z: 0 },
      { id: 1, x: 1, y: 0, z: 0 },
      { id: 2, x: 5, y: 0, z: 0 }
    ]);

    expect(minimumRangeBetween(stepped, 0, 2, 50)).toEqual({ rangePc: expect.closeTo(4) as number, least: true });
    expect(routeBetween(stepped, 0, 2, 4).route).not.toBeNull();
    expect(routeBetween(stepped, 0, 2, 3.99).route).toBeNull();
  });

  it('prefers a longer way whose worst hop is shorter, since that is what the range pays for', () => {
    // Direct: one hop of 10. Round: three hops of at most 4. The range only has to cover 4.
    const both = index([
      { id: 0, x: 0, y: 0, z: 0 },
      { id: 1, x: 0, y: 4, z: 0 },
      { id: 2, x: 6, y: 7, z: 0 },
      { id: 3, x: 10, y: 0, z: 0 }
    ]);

    const needed = minimumRangeBetween(both, 0, 3, 50);

    expect(needed.rangePc).toBeLessThan(10);
    expect(routeBetween(both, 0, 3, needed.rangePc!).route).not.toBeNull();
  });

  it('claims nothing about a ceiling its own search gave up on', () => {
    // Nothing reaches the island at any range here, but the crowd spends the budget first, so the
    // widest search proves nothing — and neither does the null it hands back.
    const needed = minimumRangeBetween(crowdedKnot(), 0, CROWD_ISLAND, 0.5);

    expect(needed).toEqual({ rangePc: null, least: false });
  });

  it('finds nothing when even the ceiling does not reach', () => {
    const split = index([
      { id: 0, x: 0, y: 0, z: 0 },
      { id: 1, x: 100, y: 0, z: 0 }
    ]);

    expect(minimumRangeBetween(split, 0, 1, 50)).toEqual({ rangePc: null, least: true });
  });
});

/**
 * The links a segment buffer draws, as unordered pairs of star ids, read back from where each end
 * sits. Positions are compared as the float32 the buffer holds.
 */
function linksDrawn(segments: Float32Array, points: readonly StarPoint[]): string[] {
  const idAt = new Map(points.map((point) => [[point.x, point.y, point.z].map(Math.fround).join(), point.id]));
  const links: string[] = [];
  for (let at = 0; at < segments.length; at += 6) {
    const a = idAt.get(Array.from(segments.subarray(at, at + 3)).join())!;
    const b = idAt.get(Array.from(segments.subarray(at + 3, at + 6)).join())!;
    links.push(a < b ? `${a}-${b}` : `${b}-${a}`);
  }
  return links;
}

/**
 * What a budget should keep, worked out the slow way: every link sorted by how near its nearer end
 * is to the centre, then taken until one does not fit. Lengths and distances as the float32 buffer
 * holds them.
 */
function nearestFirst(points: readonly StarPoint[], rangePc: number, centre: { x: number; y: number; z: number }, lengthPc: number): string[] {
  const all = jumpLinkSegments(index([...points]), rangePc);
  const links = Array.from({ length: all.length / 6 }, (_, link) => {
    const v = Array.from(all.subarray(link * 6, link * 6 + 6));
    const nearer = Math.fround(Math.sqrt(Math.min((v[0] - centre.x) ** 2 + (v[1] - centre.y) ** 2 + (v[2] - centre.z) ** 2, (v[3] - centre.x) ** 2 + (v[4] - centre.y) ** 2 + (v[5] - centre.z) ** 2)));
    return { link, nearer, length: Math.fround(Math.hypot(v[3] - v[0], v[4] - v[1], v[5] - v[2])), key: linksDrawn(all.subarray(link * 6, link * 6 + 6), points)[0] };
  }).sort((a, b) => a.nearer - b.nearer || a.link - b.link);
  const kept: string[] = [];
  let total = 0;
  for (const { length, key } of links) {
    if (total + length > lengthPc) {
      break;
    }
    total += length;
    kept.push(key);
  }
  return kept;
}

/** Stars a parsec apart along x, as points, for reading a segment buffer back. */
function chainPoints(count: number): StarPoint[] {
  return Array.from({ length: count }, (_, i) => ({ id: i, x: i, y: 0, z: 0 }));
}

describe('jumpLinkSegments', () => {
  it('draws each pair once, not once from either end', () => {
    const segments = jumpLinkSegments(chain(4), 1.5);

    expect(linksDrawn(segments, chainPoints(4)).sort()).toEqual(['0-1', '1-2', '2-3']);
  });

  it('puts both ends of every link where its stars are', () => {
    const segments = jumpLinkSegments(chain(3), 2.5);

    expect(segments).toHaveLength(3 * 6);
    expect(linksDrawn(segments, chainPoints(3)).sort()).toEqual(['0-1', '0-2', '1-2']);
  });

  it('draws nothing at no range', () => {
    expect(jumpLinkSegments(chain(4), 0)).toHaveLength(0);
  });

  it('keeps the links nearest the centre first, for as much length as the budget holds', () => {
    // A parsec apart from 0 to 20, the centre at 10.3. By nearer end: 9-10 and 10-11 (0.3 away),
    // then 11-12 (0.7), then 8-9 (1.3). Three parsecs of them fit in 3.5; a fourth would not.
    const budget = { centre: { x: 10.3, y: 0, z: 0 }, lengthPc: 3.5 };

    const segments = jumpLinkSegments(chain(21), 1.5, budget);

    expect(linksDrawn(segments, chainPoints(21)).sort()).toEqual(['10-11', '11-12', '9-10']);
    expect(segments.buffer.byteLength).toBe(segments.byteLength);
  });

  it('keeps exactly the links a full nearest-first sort would, without sorting them all', () => {
    let seed = 7;
    const random = () => ((seed = (seed * 1103515245 + 12345) % 2147483648) / 2147483648) * 40 - 20;
    const points: StarPoint[] = Array.from({ length: 600 }, (_, id) => ({ id, x: random(), y: random(), z: random() }));
    const centre = { x: 3, y: -2, z: 1 };

    for (const lengthPc of [0, 5, 60, 900, 4000, 1e9]) {
      expect(linksDrawn(jumpLinkSegments(index(points), 4, { centre, lengthPc }), points).sort()).toEqual(nearestFirst(points, 4, centre, lengthPc).sort());
    }
  });

  it('sorts the distance band the budget runs out in, and stops at the first link there that does not fit', () => {
    // One pair 4 kpc out makes each band about a parsec deep, so dozens of short links near the
    // centre share the band the budget ends in, in whatever order the grid walks them.
    let seed = 3;
    const random = () => (seed = (seed * 1103515245 + 12345) % 2147483648) / 2147483648;
    const points: StarPoint[] = [{ id: 0, x: 4000, y: 0, z: 0 }, { id: 1, x: 4000.03, y: 0, z: 0 }];
    for (let pair = 0; pair < 40; pair++) {
      const r = 0.05 + random() * 0.9;
      const theta = random() * Math.PI * 2;
      const x = r * Math.cos(theta);
      const y = r * Math.sin(theta);
      points.push({ id: 2 + pair * 2, x, y, z: 0 }, { id: 3 + pair * 2, x, y, z: 0.005 + random() * 0.04 });
    }
    const centre = { x: 0, y: 0, z: 0 };

    for (const lengthPc of [0.1, 0.3, 0.5]) {
      expect(linksDrawn(jumpLinkSegments(index(points), 0.05, { centre, lengthPc }), points).sort()).toEqual(nearestFirst(points, 0.05, centre, lengthPc).sort());
    }
  });

  it('counts the budget in parsecs of link, not in links', () => {
    // Stars at 0, 1 and 3: a 2 pc link nearest the centre, then a 1 pc one. Two and a half parsecs
    // hold the first and not both, though two links would fit a count of two and a half.
    const points: StarPoint[] = [{ id: 0, x: 0, y: 0, z: 0 }, { id: 1, x: 1, y: 0, z: 0 }, { id: 2, x: 3, y: 0, z: 0 }];

    const segments = jumpLinkSegments(index(points), 2.5, { centre: { x: 3, y: 0, z: 0 }, lengthPc: 2.5 });

    expect(linksDrawn(segments, points)).toEqual(['1-2']);
  });

  it('grows past its first buffer without losing a link', () => {
    // 5 000 stars a tenth of a parsec apart, ten neighbours each way in range: some 50 000 links, far past
    // the 4 096 the buffer starts with, so it has to grow several times.
    const count = 5000;
    const line = new StarNeighbourhood(Array.from({ length: count }, (_, i) => ({ id: i, x: i / 10, y: 0, z: 0 })));
    // 1.05 rather than 1: the tenth neighbour sits at 1.0, which float steps of a tenth put either side of it.
    const segments = jumpLinkSegments(line, 1.05);

    let expected = 0;
    for (let i = 0; i < count; i++) {
      expected += Math.min(10, count - 1 - i);
    }
    expect(segments.length / 6).toBe(expected);
    expect(segments.buffer.byteLength).toBe(segments.byteLength);
  });

  it('agrees with every route it makes possible', () => {
    // The graph drawn and the graph walked have to be the same graph, or the map shows a way
    // the route cannot take.
    let seed = 11;
    const random = () => ((seed = (seed * 1103515245 + 12345) % 2147483648) / 2147483648) * 30 - 15;
    const points: StarPoint[] = Array.from({ length: 120 }, (_, id) => ({ id, x: random(), y: random(), z: random() }));
    const cloud = index(points);
    // 9 rather than 6: at 6 this cloud falls into pieces and 0 never reaches 119, which an
    // earlier version of this test hid by only checking the route it happened to find.
    const range = 9;

    const drawn = new Set(linksDrawn(jumpLinkSegments(cloud, range), points));

    const { route } = routeBetween(cloud, 0, 119, range);
    // Asserted, not guarded: a skipped body would let the two disagree unnoticed.
    expect(route).not.toBeNull();
    expect(route!.stars.length).toBeGreaterThan(2);
    for (let i = 1; i < route!.stars.length; i++) {
      const [a, b] = [route!.stars[i - 1], route!.stars[i]].sort((x, y) => x - y);
      expect(drawn.has(`${a}-${b}`)).toBe(true);
    }
    expect(drawn.size).toBeGreaterThan(0);
  });
});