import test from "node:test";
import assert from "node:assert/strict";
import * as presets from "../examples/render-graph-studio/render-graph/presets.js";
import { descriptors } from "@yawn/render-graph-fxnode/catalog";

test("the JSO example is a complete canonical AST with external pipelines", () => {
  assert.deepEqual(Object.keys(presets.renderGraphPresets), ["jso"]);
  for (const [name, graph] of Object.entries(presets.renderGraphPresets)) {
    assert.deepEqual([graph.kind, graph.version, graph.revision], ["yawn-render-graph", 1, 1], name);
    assert.equal(new Set(graph.nodes.map((node) => node.id)).size, graph.nodes.length, name);
    assert.equal(graph.nodes.filter((node) => node.executor.key === "frame_out").length, 1, name);
    for (const node of graph.nodes)
      assert.equal(node.executor.version, descriptors[node.executor.key].version, `${name}:${node.id}`);
    assert.deepEqual(graph.pipelines.render.map(({ name }) => name), [
      "ground_plane", "gltf_standard", "gltf_standard_double_sided", "frame_out",
    ]);
    assert.deepEqual(graph.pipelines.compute.map(({ name }) => name), ["initialize_scene"]);
  }
});

test("presets classify demo-owned enable and material bits through type.words[0] predicates", () => {
  for (const [name, graph] of Object.entries(presets.renderGraphPresets)) {
    const byId = Object.fromEntries(graph.nodes.map((node) => [node.id, node]));
    assert.deepEqual(byId.type_words.inputs.value, [{ node: "mesh", socket: "type" }], name);
    assert.deepEqual(byId.type_bits.inputs.value, [{ node: "type_words", socket: "word0" }], name);
    assert.deepEqual(byId.ground.inputs.predicate, [{ node: "ground_class", socket: "value" }], name);
    assert.deepEqual(byId.pbr.inputs.predicate, [{ node: "standard_class", socket: "value" }], name);
    assert.deepEqual(byId.pbr_double.inputs.predicate, [{ node: "double_class", socket: "value" }], name);
    for (const pipeline of [byId.ground, byId.pbr, byId.pbr_double]) {
      assert.deepEqual(pipeline.inputs.mesh, [{ node: "mesh", socket: "mesh" }]);
      assert.equal(pipeline.executor.version, 2);
    }
  }
});

test("the example adds a local-AABB expression to each material predicate", () => {
  const byId = Object.fromEntries(presets.culling.nodes.map((node) => [node.id, node]));
  assert.deepEqual(byId.cull.inputs, {
    mesh: [{ node: "mesh", socket: "mesh" }], localAabb: [{ node: "mesh", socket: "localAabb" }],
  });
  assert.deepEqual(byId.not_culled.inputs.operand, [{ node: "cull", socket: "isFrustumCulled" }]);
  for (const id of ["ground_class", "standard_class", "double_class"])
    assert.equal(byId[id].inputs.inputs.at(-1).node, "not_culled");
});

test("scene pipelines directly share matching transient color and depth targets", () => {
  for (const [name, graph] of Object.entries(presets.renderGraphPresets)) {
    const byId = Object.fromEntries(graph.nodes.map((node) => [node.id, node]));
    const color = byId.ground.inputs.color;
    const depth = byId.ground.inputs.depth;
    for (const id of ["ground", "pbr", "pbr_double"]) {
      assert.deepEqual(byId[id].inputs.color, color, `${name}:${id}`);
      assert.deepEqual(byId[id].inputs.depth, depth, `${name}:${id}`);
      assert.equal(["ground", "pbr", "pbr_double"].includes(color[0].node), false, name);
    }
    const colorDescriptor = byId[color[0].node].parameters.texture;
    const depthDescriptor = byId[depth[0].node].parameters.texture;
    assert.equal(depthDescriptor.format, "depth32_float", name);
    assert.deepEqual(depthDescriptor.extent, colorDescriptor.extent, name);
    assert.equal(depthDescriptor.sampleCount, colorDescriptor.sampleCount, name);
  }
});