import Benchmark from "benchmark";
import {
  expression,
  LayerSpecification,
  StyleExpression,
  EvaluationContext,
  Feature,
} from "@maplibre/maplibre-gl-style-spec";

import { build } from "../src/layer/index.js";
import { VectorTile } from "@mapbox/vector-tile";
import Pbf from "pbf";

const layers = build(["en"])
  .filter(
    (layer: LayerSpecification) =>
      layer["source-layer"] === "transportation" && "filter" in layer
  )
  //TODO: this map call will fail if there is a BackgroundLayerSpecification in the layers array, because it doesn't have a filter property.
  .map(
    (layer: LayerSpecification) =>
      (
        expression.createExpression((layer as any).filter)
          .value as StyleExpression
      ).expression
  );

const suite = new Benchmark.Suite();

async function addTest(
  name: string,
  z: number,
  x: number,
  y: number
): Promise<void> {
  const tile = await (
    await fetch(
      `https://d1zqyi8v6vm8p9.cloudfront.net/planet/${z}/${x}/${y}.mvt`
    )
  ).arrayBuffer();

  const transportation = new VectorTile(new Pbf(tile)).layers["transportation"];
  const features: Feature[] = [];

  for (let i = 0; i < transportation.length; i++) {
    const feature = transportation.feature(i);
    features.push({
      type: feature.type,
      properties: feature.properties,
      geometry: [],
    });
  }
  suite.add(`evaluate expressions ${name}`, () => {
    let num = 0;
    const context: EvaluationContext = {
      globals: {
        zoom: 12,
      },
      properties: () => context.feature!.properties,
      geometryType: () => context.feature!.type.toString(),
      feature: null as any,
      featureState: {},
      formattedSection: null as any,
      availableImages: [],
      canonical: null as any,
      _parseColorCache: {},
      id: () => null,
      geometry: () => [],
      canonicalID: () => context.canonical,
      parseColor: () => null as any,
    };
    for (const layer of layers) {
      for (const feature of features) {
        context.feature = feature;
        if (layer.evaluate(context)) {
          num++;
        }
      }
    }
  });
}

await addTest("nyc z12", 12, 1207, 1539);
await addTest("boston z12", 12, 1239, 1514);
await addTest("kansas z14", 14, 3707, 6302);

suite
  .on("error", (event: any) => console.log(event.target.error))
  .on("cycle", (event: any) => {
    const time = 1_000 / event.target.hz;
    console.log(`${time.toPrecision(4)}ms ${event.target}`);
  })
  .run();