import { ensureDirSync } from "jsr:@std/fs@^1.0.6";
import { Creature } from "../mod.ts";
import { createBackPropagationConfig } from "../src/propagate/BackPropagation.ts";
import { compactUnused } from "../src/compact/CompactUnused.ts";
import { SparseConfig } from "../src/propagate/sparse/SparseConfig.ts";

Deno.test("Traced", () => {
  const traceDir = ".trace";
  ensureDirSync(traceDir);
  const json = JSON.parse(Deno.readTextFileSync("./test/data/traced.json"));

  const creature = Creature.fromJSON(json);

  Deno.writeTextFileSync(
    `${traceDir}/A.json`,
    JSON.stringify(creature.exportJSON(), null, 2),
  );
  const config = createBackPropagationConfig();

  const compact = compactUnused(json, config.plankConstant);
  if (compact) {
    Deno.writeTextFileSync(
      `${traceDir}/C.json`,
      JSON.stringify(compact.exportJSON(), null, 2),
    );
  }
  const sparseConfig = new SparseConfig(creature.exportJSON(), config);

  creature.applyLearnings(config, sparseConfig);

  Deno.writeTextFileSync(
    `${traceDir}/B.json`,
    JSON.stringify(creature.exportJSON(), null, 2),
  );
});