import Delaunator from "delaunator";
import FlatQueue from "flatqueue";

import {TIME} from "config/logging";
import {ELEVATION, MIN_LAND_HEIGHT, ROUTES} from "config/generation";
import {dist2} from "utils/functionUtils";

type TCellsData = Pick<IPack["cells"], "c" | "p" | "g" | "h" | "t" | "haven" | "biome" | "state" | "burg">;

export function generateRoutes(burgs: TBurgs, temp: Int8Array, cells: TCellsData) {
  const cellRoutes = new Uint8Array(cells.h.length);

  const {capitalsByFeature, burgsByFeature, portsByFeature} = sortBurgsByFeature(burgs);
  const connections: Map<string, boolean> = new Map();

  const mainRoads = generateMainRoads();
  const trails = generateTrails();
  const seaRoutes = generateSeaRoutes();

  const routes = combineRoutes();
  return {cellRoutes, routes};

  function sortBurgsByFeature(burgs: TBurgs) {
    const burgsByFeature: Dict<IBurg[]> = {};
    const capitalsByFeature: Dict<IBurg[]> = {};
    const portsByFeature: Dict<IBurg[]> = {};

    const isBurg = (burg: IBurg | TNoBurg): burg is IBurg => burg.i !== 0;
    const addBurg = (object: Dict<IBurg[]>, feature: number, burg: IBurg) => {
      if (!object[feature]) object[feature] = [];
      object[feature].push(burg);
    };

    for (const burg of burgs) {
      if (isBurg(burg)) {
        const {feature, capital, port} = burg;
        addBurg(burgsByFeature, feature, burg);
        if (capital) addBurg(capitalsByFeature, feature, burg);
        if (port) addBurg(portsByFeature, port, burg);
      }
    }

    return {burgsByFeature, capitalsByFeature, portsByFeature};
  }

  function generateMainRoads() {
    TIME && console.time("generateMainRoads");
    const mainRoads: {feature: number; cells: number[]}[] = [];

    for (const [key, featureCapitals] of Object.entries(capitalsByFeature)) {
      const points: TPoints = featureCapitals.map(burg => [burg.x, burg.y]);
      const urquhartEdges = calculateUrquhartEdges(points);
      urquhartEdges.forEach(([fromId, toId]) => {
        const start = featureCapitals[fromId].cell;
        const exit = featureCapitals[toId].cell;

        const segments = findPathSegments({isWater: false, cellRoutes, connections, start, exit});
        for (const segment of segments) {
          addConnections(segment, ROUTES.MAIN_ROAD);
          mainRoads.push({feature: Number(key), cells: segment});
        }
      });
    }

    TIME && console.timeEnd("generateMainRoads");
    return mainRoads;
  }

  function generateTrails() {
    TIME && console.time("generateTrails");

    const trails: {feature: number; cells: number[]}[] = [];

    for (const [key, featureBurgs] of Object.entries(burgsByFeature)) {
      const points: TPoints = featureBurgs.map(burg => [burg.x, burg.y]);
      const urquhartEdges = calculateUrquhartEdges(points);
      urquhartEdges.forEach(([fromId, toId]) => {
        const start = featureBurgs[fromId].cell;
        const exit = featureBurgs[toId].cell;

        const segments = findPathSegments({isWater: false, cellRoutes, connections, start, exit});
        for (const segment of segments) {
          addConnections(segment, ROUTES.TRAIL);
          trails.push({feature: Number(key), cells: segment});
        }
      });
    }

    TIME && console.timeEnd("generateTrails");
    return trails;
  }

  function generateSeaRoutes() {
    TIME && console.time("generateSearoutes");
    const mainRoads: {feature: number; cells: number[]}[] = [];

    for (const [key, featurePorts] of Object.entries(portsByFeature)) {
      const points: TPoints = featurePorts.map(burg => [burg.x, burg.y]);
      const urquhartEdges = calculateUrquhartEdges(points);
      urquhartEdges.forEach(([fromId, toId]) => {
        const start = featurePorts[fromId].cell;
        const exit = featurePorts[toId].cell;

        const segments = findPathSegments({isWater: true, cellRoutes, connections, start, exit});
        for (const segment of segments) {
          addConnections(segment, ROUTES.MAIN_ROAD);
          mainRoads.push({feature: Number(key), cells: segment});
        }
      });
    }

    TIME && console.timeEnd("generateSearoutes");
    return mainRoads;
  }

  function addConnections(segment: number[], roadTypeId: number) {
    for (let i = 0; i < segment.length; i++) {
      const cellId = segment[i];
      const nextCellId = segment[i + 1];
      if (nextCellId) connections.set(`${cellId}-${nextCellId}`, true);
      cellRoutes[cellId] = roadTypeId;
    }
  }

  function findPathSegments({
    isWater,
    cellRoutes,
    connections,
    start,
    exit
  }: {
    isWater: boolean;
    cellRoutes: Uint8Array;
    connections: Map<string, boolean>;
    start: number;
    exit: number;
  }): number[][] {
    const from = findPath(isWater, cellRoutes, temp, cells, start, exit, connections);
    if (!from) return [];

    const pathCells = restorePath(start, exit, from);
    const segments = getRouteSegments(pathCells, connections);
    return segments;
  }

  function combineRoutes() {
    const routes: TRoutes = [];

    for (const {feature, cells} of mainRoads) {
      routes.push({i: routes.length, type: "road", feature, cells});
    }

    for (const {feature, cells} of trails) {
      routes.push({i: routes.length, type: "trail", feature, cells});
    }

    for (const {feature, cells} of seaRoutes) {
      routes.push({i: routes.length, type: "sea", feature, cells});
    }

    return routes;
  }
}

function findPath(
  isWater: boolean,
  cellRoutes: Uint8Array,
  temp: Int8Array,
  cells: TCellsData,
  start: number,
  exit: number,
  connections: Map<string, boolean>
) {
  const from: number[] = [];
  const cost: number[] = [];
  const queue = new FlatQueue<number>();
  queue.push(start, 0);

  return isWater ? findWaterPath() : findLandPath();

  function findLandPath() {
    while (queue.length) {
      const priority = queue.peekValue()!;
      const next = queue.pop()!;

      for (const neibCellId of cells.c[next]) {
        if (cells.h[neibCellId] < MIN_LAND_HEIGHT) continue; // ignore water cells

        const habitability = biomesData.habitability[cells.biome[neibCellId]];
        if (!habitability) continue; // inhabitable cells are not passable (eg. lava, glacier)

        const distanceCost = dist2(cells.p[next], cells.p[neibCellId]);

        const habitabilityModifier = 1 + Math.max(100 - habitability, 0) / 1000; // [1, 1.1];
        const heightModifier = 1 + Math.max(cells.h[neibCellId] - ELEVATION.HILLS, 0) / 50; // [1, 2];
        const roadModifier = cellRoutes[neibCellId] ? 1 : 2;
        const burgModifier = cells.burg[neibCellId] ? 1 : 2;

        const cellsCost = distanceCost * habitabilityModifier * heightModifier * roadModifier * burgModifier;
        const totalCost = priority + cellsCost;

        if (from[neibCellId] || totalCost >= cost[neibCellId]) continue;
        from[neibCellId] = next;

        if (neibCellId === exit) return from;

        cost[neibCellId] = totalCost;
        queue.push(neibCellId, totalCost);
      }
    }

    return null; // path is not found
  }

  function findWaterPath() {
    const MIN_PASSABLE_TEMP = -4;

    while (queue.length) {
      const priority = queue.peekValue()!;
      const next = queue.pop()!;

      for (const neibCellId of cells.c[next]) {
        if (neibCellId === exit) {
          from[neibCellId] = next;
          return from;
        }

        if (cells.h[neibCellId] >= MIN_LAND_HEIGHT) continue; // ignore land cells
        if (temp[cells.g[neibCellId]] < MIN_PASSABLE_TEMP) continue; // ignore to cold cells

        const distanceCost = dist2(cells.p[next], cells.p[neibCellId]);
        const typeModifier = Math.abs(cells.t[neibCellId]); // 1 for coastline, 2 for deep ocean, 3 for deeper ocean
        const routeModifier = cellRoutes[neibCellId] ? 1 : 2;
        const connectionModifier =
          connections.has(`${next}-${neibCellId}`) || connections.has(`${neibCellId}-${next}`) ? 1 : 3;

        const cellsCost = distanceCost * typeModifier * routeModifier * connectionModifier;
        const totalCost = priority + cellsCost;

        if (from[neibCellId] || totalCost >= cost[neibCellId]) continue;
        from[neibCellId] = next;

        cost[neibCellId] = totalCost;
        queue.push(neibCellId, totalCost);
      }
    }

    return null; // path is not found
  }
}

function restorePath(start: number, end: number, from: number[]) {
  const cells: number[] = [];

  let current = end;
  let prev = end;

  while (current !== start) {
    cells.push(current);
    prev = from[current];
    current = prev;
  }

  cells.push(current);

  return cells;
}

function getRouteSegments(pathCells: number[], connections: Map<string, boolean>) {
  const segments: number[][] = [];
  let segment: number[] = [];

  for (let i = 0; i < pathCells.length; i++) {
    const cellId = pathCells[i];
    const nextCellId = pathCells[i + 1];
    const isConnected = connections.has(`${cellId}-${nextCellId}`) || connections.has(`${nextCellId}-${cellId}`);

    if (isConnected) {
      if (segment.length) {
        // segment stepped into existing segment
        segment.push(pathCells[i]);
        segments.push(segment);
        segment = [];
      }
      continue;
    }

    segment.push(pathCells[i]);
  }

  if (segment.length > 1) segments.push(segment);

  return segments;
}

// Urquhart graph is obtained by removing the longest edge from each triangle in the Delaunay triangulation
// this gives us an aproximation of a desired road network, i.e. connections between burgs
// code from https://observablehq.com/@mbostock/urquhart-graph
function calculateUrquhartEdges(points: TPoints) {
  const score = (p0: number, p1: number) => dist2(points[p0], points[p1]);

  const {halfedges, triangles} = Delaunator.from(points);
  const n = triangles.length;

  const removed = new Uint8Array(n);
  const edges = [];

  for (let e = 0; e < n; e += 3) {
    const p0 = triangles[e],
      p1 = triangles[e + 1],
      p2 = triangles[e + 2];

    const p01 = score(p0, p1),
      p12 = score(p1, p2),
      p20 = score(p2, p0);

    removed[
      p20 > p01 && p20 > p12
        ? Math.max(e + 2, halfedges[e + 2])
        : p12 > p01 && p12 > p20
        ? Math.max(e + 1, halfedges[e + 1])
        : Math.max(e, halfedges[e])
    ] = 1;
  }

  for (let e = 0; e < n; ++e) {
    if (e > halfedges[e] && !removed[e]) {
      const t0 = triangles[e];
      const t1 = triangles[e % 3 === 2 ? e - 2 : e + 1];
      edges.push([t0, t1]);
    }
  }

  return edges;
}