@mcmcjs/stan 0.1.0

package/dist/index.cjs

@@ -0,0 +1,1000 @@
+"use strict";
+var __defProp = Object.defineProperty;
+var __getOwnPropDesc = Object.getOwnPropertyDescriptor;
+var __getOwnPropNames = Object.getOwnPropertyNames;
+var __hasOwnProp = Object.prototype.hasOwnProperty;
+var __export = (target, all) => {
+  for (var name in all)
+    __defProp(target, name, { get: all[name], enumerable: true });
+};
+var __copyProps = (to, from, except, desc) => {
+  if (from && typeof from === "object" || typeof from === "function") {
+    for (let key of __getOwnPropNames(from))
+      if (!__hasOwnProp.call(to, key) && key !== except)
+        __defProp(to, key, { get: () => from[key], enumerable: !(desc = __getOwnPropDesc(from, key)) || desc.enumerable });
+  }
+  return to;
+};
+var __toCommonJS = (mod) => __copyProps(__defProp({}, "__esModule", { value: true }), mod);
+
+// src/index.ts
+var index_exports = {};
+__export(index_exports, {
+  INSTALLED_CMDSTAN_CHANNEL: () => INSTALLED_CMDSTAN_CHANNEL,
+  PINNED_CMDSTAN_VERSION: () => PINNED_CMDSTAN_VERSION,
+  addVersion: () => addVersion,
+  chainArgs: () => chainArgs,
+  columnToLeaf: () => columnToLeaf,
+  compileModel: () => compileModel,
+  createStanCsvTail: () => createStanCsvTail,
+  createStanSpawn: () => createStanSpawn,
+  fittedParamsCsv: () => fittedParamsCsv,
+  isCmdStanHome: () => isCmdStanHome,
+  listCmdStanInstalls: () => listCmdStanInstalls,
+  listVersions: () => listVersions,
+  managedCmdStanHome: () => managedCmdStanHome,
+  managedStanRoot: () => managedStanRoot,
+  matchesTarget: () => matchesTarget,
+  modelCacheDir: () => modelCacheDir,
+  parseIterationLine: () => parseIterationLine,
+  planSetup: () => planSetup,
+  predictData: () => predictData,
+  removeVersion: () => removeVersion,
+  resolveCmdStan: () => resolveCmdStan,
+  runDoctor: () => runDoctor,
+  runFit: () => runFit,
+  runMatrix: () => runMatrix,
+  runPredict: () => runPredict,
+  runSetup: () => runSetup,
+  stanEngine: () => stanEngine
+});
+module.exports = __toCommonJS(index_exports);
+
+// src/compile.ts
+var import_node_crypto = require("crypto");
+var import_node_fs2 = require("fs");
+var import_node_path2 = require("path");
+var import_engine3 = require("@mcmcjs/engine");
+
+// src/environment.ts
+var import_node_fs = require("fs");
+var import_node_os = require("os");
+var import_node_path = require("path");
+var import_engine = require("@mcmcjs/engine");
+var import_engine2 = require("@mcmcjs/engine");
+var import_core = require("@mcmcjs/core");
+var defaultRunner = (0, import_engine.createRunner)();
+var INSTALLED_CMDSTAN_CHANNEL = import_core.DEFAULT_CMDSTAN_CHANNEL;
+var PINNED_CMDSTAN_VERSION = "2.39.0";
+function managedStanRoot() {
+  const home = process.env.HOME || (0, import_node_os.homedir)();
+  const dataHome = process.env.XDG_DATA_HOME || (0, import_node_path.join)(home, ".local", "share");
+  return (0, import_node_path.join)(dataHome, "mcmcjs", "stan");
+}
+function isCmdStanHome(dir) {
+  return (0, import_node_fs.existsSync)((0, import_node_path.join)(dir, "makefile")) && (0, import_node_fs.existsSync)((0, import_node_path.join)(dir, "bin", "stanc"));
+}
+function versionFromDirName(dir) {
+  return (0, import_node_path.basename)(dir).match(/^cmdstan-(\d+\.\d+\.\d+(?:-rc\d+)?)$/)?.[1];
+}
+function versionFromMakefile(home) {
+  try {
+    const makefile = (0, import_node_fs.readFileSync)((0, import_node_path.join)(home, "makefile"), "utf8");
+    return makefile.match(/CMDSTAN_VERSION\s*:?=\s*(\S+)/)?.[1];
+  } catch {
+    return void 0;
+  }
+}
+function installAt(home) {
+  if (!isCmdStanHome(home)) return void 0;
+  const version = versionFromDirName(home) ?? versionFromMakefile(home);
+  return version ? { version, home } : void 0;
+}
+function scanRoot(root) {
+  let entries;
+  try {
+    entries = (0, import_node_fs.readdirSync)(root);
+  } catch {
+    return [];
+  }
+  const installs = [];
+  for (const name of entries) {
+    if (!name.startsWith("cmdstan-")) continue;
+    const install = installAt((0, import_node_path.join)(root, name));
+    if (install) installs.push(install);
+  }
+  return installs;
+}
+function compareVersions(a, b) {
+  const parse = (v) => {
+    const [core, rc] = v.split("-rc");
+    const nums = (core ?? "").split(".").map((p) => Number.parseInt(p, 10) || 0);
+    return { nums, rc: rc === void 0 ? Number.POSITIVE_INFINITY : Number.parseInt(rc, 10) || 0 };
+  };
+  const pa = parse(a);
+  const pb = parse(b);
+  for (let i = 0; i < Math.max(pa.nums.length, pb.nums.length); i++) {
+    const d = (pa.nums[i] ?? 0) - (pb.nums[i] ?? 0);
+    if (d !== 0) return d;
+  }
+  if (pa.rc === pb.rc) return 0;
+  return pa.rc < pb.rc ? -1 : 1;
+}
+function listCmdStanInstalls() {
+  const explicit = process.env.MCMCJS_CMDSTAN || process.env.CMDSTAN;
+  if (explicit) {
+    const install = installAt(explicit);
+    if (!install) {
+      throw new Error(
+        `MCMCJS_CMDSTAN/CMDSTAN points at ${explicit}, which is not a CmdStan directory (expected a makefile and bin/stanc); unset it or point it at a CmdStan home.`
+      );
+    }
+    return [install];
+  }
+  const home = process.env.HOME || (0, import_node_os.homedir)();
+  const installs = [...scanRoot(managedStanRoot()), ...scanRoot((0, import_node_path.join)(home, ".cmdstan"))];
+  return installs.sort((a, b) => compareVersions(b.version, a.version));
+}
+function resolveCmdStan(requested = INSTALLED_CMDSTAN_CHANNEL) {
+  const installs = listCmdStanInstalls();
+  if (requested === INSTALLED_CMDSTAN_CHANNEL) {
+    const newest = installs[0];
+    if (newest) return newest;
+    throw new Error(
+      "CmdStan not found. Run `mcmc setup --engine stan`, or point MCMCJS_CMDSTAN at a CmdStan directory."
+    );
+  }
+  const match = installs.find((i) => i.version === requested);
+  if (match) return match;
+  throw new Error(
+    `CmdStan ${requested} not found. Run \`mcmc setup --engine stan --stan-version ${requested}\`, or point MCMCJS_CMDSTAN at a CmdStan directory.`
+  );
+}
+async function detect(command, args, parseVersion, runner) {
+  try {
+    const version = parseVersion(await runner(command, args));
+    if (version) return { found: true, version, path: command };
+  } catch {
+  }
+  return { found: false };
+}
+var versionNumber = (stdout) => stdout.match(/(\d+\.\d+(?:\.\d+)?)/)?.[1];
+function detectMake(runner = defaultRunner) {
+  return detect("make", ["--version"], versionNumber, runner);
+}
+async function detectCxx(runner = defaultRunner) {
+  for (const command of ["g++", "clang++"]) {
+    const info = await detect(command, ["--version"], versionNumber, runner);
+    if (info.found) return info;
+  }
+  return { found: false };
+}
+async function detectStanc(install, runner = defaultRunner) {
+  if (!install) return { found: false };
+  const stanc = (0, import_node_path.join)(install.home, "bin", "stanc");
+  return detect(stanc, ["--version"], versionNumber, runner);
+}
+
+// src/compile.ts
+function cacheKeySource(source, modelDir, depth = 4) {
+  if (depth === 0) return source;
+  const parts = [source];
+  for (const match of source.matchAll(/^\s*#include\s+[<"']?([^>"'\s]+)[>"']?/gm)) {
+    const file = match[1];
+    if (!file) continue;
+    try {
+      const included = (0, import_node_fs2.readFileSync)((0, import_node_path2.isAbsolute)(file) ? file : (0, import_node_path2.join)(modelDir, file), "utf8");
+      parts.push(cacheKeySource(included, modelDir, depth - 1));
+    } catch {
+    }
+  }
+  return parts.join("\n");
+}
+function modelCacheDir(source, version, cacheRoot) {
+  const key = (0, import_node_crypto.createHash)("sha256").update(`${version}
+${source}`).digest("hex").slice(0, 16);
+  return (0, import_node_path2.join)(cacheRoot ?? (0, import_node_path2.join)(managedStanRoot(), "models"), key);
+}
+async function compileModel(install, modelPath, options = {}) {
+  const { runner = (0, import_engine3.createRunner)(10 * 6e4), cacheRoot } = options;
+  const source = (0, import_node_fs2.readFileSync)(modelPath, "utf8");
+  const modelDir = (0, import_node_path2.dirname)(modelPath);
+  const dir = modelCacheDir(cacheKeySource(source, modelDir), install.version, cacheRoot);
+  const binaryPath = (0, import_node_path2.join)(dir, "model");
+  const okPath = (0, import_node_path2.join)(dir, ".ok");
+  if ((0, import_node_fs2.existsSync)(okPath) && (0, import_node_fs2.existsSync)(binaryPath)) return { binaryPath, cached: true };
+  (0, import_node_fs2.mkdirSync)(dir, { recursive: true });
+  (0, import_node_fs2.writeFileSync)((0, import_node_path2.join)(dir, "model.stan"), source);
+  await runner("make", ["-C", install.home, `STANCFLAGS=--include-paths=${modelDir}`, binaryPath]);
+  if (!(0, import_node_fs2.existsSync)(binaryPath)) {
+    throw new Error("make reported success but produced no model executable");
+  }
+  (0, import_node_fs2.writeFileSync)(okPath, "");
+  return { binaryPath, cached: false };
+}
+
+// src/csv.ts
+var import_node_fs3 = require("fs");
+var import_core2 = require("@mcmcjs/core");
+var STAT_RENAME = {
+  lp__: "lp",
+  energy__: "energy",
+  divergent__: "diverging",
+  treedepth__: "tree_depth",
+  n_leapfrog__: "n_steps",
+  accept_stat__: "acceptance_rate",
+  stepsize__: "step_size"
+};
+function columnToLeaf(column) {
+  if (column.endsWith("__")) return STAT_RENAME[column] ?? null;
+  return (0, import_core2.fromStanName)(column);
+}
+function parseStanNumber(field) {
+  const value = Number(field);
+  if (!Number.isNaN(value) || field === "nan") return value;
+  if (field === "inf" || field === "+inf") return Number.POSITIVE_INFINITY;
+  if (field === "-inf") return Number.NEGATIVE_INFINITY;
+  return value;
+}
+function createStanCsvTail(filePath, opts) {
+  let position = 0;
+  let pending = "";
+  let leaves;
+  let seq = 0;
+  const rows = [];
+  const emit = (count) => {
+    if (count === 0 || !leaves) return;
+    const draws = {};
+    for (let c = 0; c < leaves.length; c++) {
+      const leaf = leaves[c];
+      if (!leaf) continue;
+      const values = new Array(count);
+      for (let r = 0; r < count; r++) values[r] = rows[r]?.[c] ?? Number.NaN;
+      draws[leaf] = values;
+    }
+    rows.splice(0, count);
+    opts.onBatch({ chain: opts.chain, seq, iteration: null, draws });
+    seq += 1;
+  };
+  const consume = (line) => {
+    const trimmed = line.trim();
+    if (trimmed.length === 0 || trimmed.startsWith("#")) return;
+    if (!leaves) {
+      leaves = trimmed.split(",").map(columnToLeaf);
+      return;
+    }
+    const fields = trimmed.split(",");
+    if (fields.length !== leaves.length) return;
+    rows.push(fields.map(parseStanNumber));
+    if (rows.length >= opts.batchSize) emit(opts.batchSize);
+  };
+  const poll = () => {
+    let fd;
+    try {
+      fd = (0, import_node_fs3.openSync)(filePath, "r");
+    } catch {
+      return;
+    }
+    try {
+      const chunk = Buffer.alloc(1 << 16);
+      for (; ; ) {
+        const read = (0, import_node_fs3.readSync)(fd, chunk, 0, chunk.length, position);
+        if (read <= 0) break;
+        position += read;
+        pending += chunk.toString("utf8", 0, read);
+        for (; ; ) {
+          const nl = pending.indexOf("\n");
+          if (nl < 0) break;
+          consume(pending.slice(0, nl));
+          pending = pending.slice(nl + 1);
+        }
+      }
+    } finally {
+      (0, import_node_fs3.closeSync)(fd);
+    }
+  };
+  return {
+    poll,
+    finish() {
+      poll();
+      emit(rows.length);
+    }
+  };
+}
+
+// src/doctor.ts
+async function runDoctor(runner) {
+  const install = listCmdStanInstalls()[0];
+  const cmdstan = install ? { found: true, version: install.version, path: install.home } : { found: false };
+  const [stanc, make, cxx] = await Promise.all([
+    detectStanc(install, runner),
+    detectMake(runner),
+    detectCxx(runner)
+  ]);
+  return {
+    cmdstan,
+    stanc,
+    make,
+    cxx,
+    install,
+    ready: cmdstan.found && make.found && cxx.found
+  };
+}
+
+// src/engine.ts
+var stanEngine = {
+  id: "stan",
+  displayName: "Stan (CmdStan)",
+  capabilities: { setup: true, versions: true, fit: true, predict: true },
+  async doctor(ctx) {
+    const report = await runDoctor(ctx.run);
+    const missingToolchain = !report.make.found || !report.cxx.found;
+    return {
+      engineId: "stan",
+      ready: report.ready,
+      tools: [
+        { name: "cmdstan", ...report.cmdstan },
+        { name: "stanc", ...report.stanc },
+        { name: "make", ...report.make },
+        { name: "c++", ...report.cxx }
+      ],
+      hint: report.ready ? void 0 : missingToolchain ? "Stan models compile with the system toolchain; install make and a C++ compiler (g++ or clang++), then run `mcmc setup --engine stan`." : "CmdStan not found. Run `mcmc setup --engine stan`."
+    };
+  }
+};
+
+// src/fit.ts
+var import_node_crypto2 = require("crypto");
+var import_node_fs4 = require("fs");
+var import_node_os2 = require("os");
+var import_node_path3 = require("path");
+var import_core3 = require("@mcmcjs/core");
+
+// src/runner.ts
+var import_node_child_process = require("child_process");
+var import_engine4 = require("@mcmcjs/engine");
+var DETACHED = process.platform !== "win32";
+var STDERR_LIMIT = 1 << 22;
+function createStanSpawn(defaultTimeoutMs = 30 * 6e4) {
+  return (command, args, opts = {}) => new Promise((resolve) => {
+    const child = (0, import_node_child_process.spawn)(command, args, {
+      stdio: ["ignore", "pipe", "pipe"],
+      detached: DETACHED
+    });
+    let stderr = "";
+    let stdoutPending = "";
+    let cancelled = false;
+    let settled = false;
+    const timeoutMs = opts.timeoutMs ?? defaultTimeoutMs;
+    const timer = setTimeout(() => {
+      stderr += "process timed out and was killed\n";
+      (0, import_engine4.killTree)(child);
+    }, timeoutMs);
+    timer.unref?.();
+    const onAbort = () => {
+      cancelled = true;
+      (0, import_engine4.killTree)(child);
+    };
+    if (opts.signal) {
+      if (opts.signal.aborted) onAbort();
+      else opts.signal.addEventListener("abort", onAbort, { once: true });
+    }
+    child.stdout?.on("data", (data) => {
+      stdoutPending += data.toString("utf8");
+      for (; ; ) {
+        const nl = stdoutPending.indexOf("\n");
+        if (nl < 0) break;
+        opts.onStdoutLine?.(stdoutPending.slice(0, nl));
+        stdoutPending = stdoutPending.slice(nl + 1);
+      }
+    });
+    child.stderr?.on("data", (data) => {
+      if (stderr.length < STDERR_LIMIT) stderr += data.toString("utf8");
+    });
+    const settle = (code) => {
+      if (settled) return;
+      settled = true;
+      clearTimeout(timer);
+      opts.signal?.removeEventListener("abort", onAbort);
+      if (stdoutPending.length > 0) opts.onStdoutLine?.(stdoutPending);
+      resolve({ code, stderr, cancelled: cancelled || void 0 });
+    };
+    child.on("error", (error) => {
+      stderr = stderr || error.message;
+      settle(1);
+    });
+    child.on("close", (code) => settle(code ?? 1));
+  });
+}
+function parseIterationLine(line) {
+  const match = line.match(/Iteration:\s*(\d+)\s*\/\s*(\d+)\s*\[\s*\d+%\]\s*\((Warmup|Sampling)\)/);
+  if (!match) return null;
+  return {
+    iteration: Number.parseInt(match[1] ?? "0", 10),
+    total: Number.parseInt(match[2] ?? "0", 10),
+    warmup: match[3] === "Warmup"
+  };
+}
+
+// src/fit.ts
+function sha256(text) {
+  return (0, import_node_crypto2.createHash)("sha256").update(text).digest("hex");
+}
+function tmpParent() {
+  const uid = typeof process.getuid === "function" ? `-${process.getuid()}` : "";
+  return (0, import_node_path3.join)((0, import_node_os2.tmpdir)(), `mcmcjs${uid}`);
+}
+function chainArgs(spec, chain1, dataPath, csvPath) {
+  const s = spec.sampler;
+  const total = s.draws + s.warmup;
+  const refresh = Math.max(1, Math.floor(total / 100));
+  return [
+    "sample",
+    `num_samples=${s.draws}`,
+    `num_warmup=${s.warmup}`,
+    "adapt",
+    `delta=${s.adapt_delta}`,
+    "data",
+    `file=${dataPath}`,
+    "random",
+    `seed=${spec.seed}`,
+    `id=${chain1}`,
+    "output",
+    `file=${csvPath}`,
+    `refresh=${refresh}`
+  ];
+}
+function errorTail(stderr) {
+  const lines = stderr.trim().split("\n").map((l) => l.trim()).filter(Boolean);
+  return lines.at(-1);
+}
+async function runFit(spec, install, io) {
+  const startedAt = (/* @__PURE__ */ new Date()).toISOString();
+  const start = performance.now();
+  const runtimeRequested = spec.backend.version;
+  const fail = (stage, error) => ({
+    status: "error",
+    runtimeRequested,
+    elapsedMs: Math.round(performance.now() - start),
+    stage,
+    error
+  });
+  if (io.signal?.aborted) {
+    return { status: "cancelled", runtimeRequested, elapsedMs: 0 };
+  }
+  let binaryPath;
+  try {
+    ({ binaryPath } = await compileModel(install, spec.modelPath, io.compile));
+  } catch (error) {
+    return fail("compile", error.message);
+  }
+  if (io.signal?.aborted) {
+    return {
+      status: "cancelled",
+      runtimeRequested,
+      elapsedMs: Math.round(performance.now() - start)
+    };
+  }
+  const ownTmp = io.tmpDir === void 0;
+  let tmp;
+  if (io.tmpDir === void 0) {
+    (0, import_node_fs4.mkdirSync)(tmpParent(), { recursive: true });
+    tmp = (0, import_node_fs4.mkdtempSync)((0, import_node_path3.join)(tmpParent(), "stan-fit-"));
+  } else {
+    tmp = io.tmpDir;
+  }
+  try {
+    const dataPath = (0, import_node_path3.join)(tmp, "data.json");
+    (0, import_node_fs4.writeFileSync)(dataPath, JSON.stringify(spec.data));
+    const chains = spec.sampler.chains;
+    const spawn2 = io.spawn ?? createStanSpawn();
+    const csvPaths = Array.from({ length: chains }, (_, i) => (0, import_node_path3.join)(tmp, `chain_${i + 1}.csv`));
+    const tails = io.onDraws ? csvPaths.map(
+      (path, i) => createStanCsvTail(path, {
+        chain: i,
+        batchSize: io.drawBatchSize ?? 25,
+        onBatch: io.onDraws
+      })
+    ) : [];
+    const pollTimer = tails.length > 0 ? setInterval(() => {
+      for (const tail of tails) tail.poll();
+    }, 150) : void 0;
+    pollTimer?.unref?.();
+    let results;
+    try {
+      results = await Promise.all(
+        csvPaths.map((csvPath, i) => {
+          const onProgress = io.onProgress;
+          return spawn2(binaryPath, chainArgs(spec, i + 1, dataPath, csvPath), {
+            signal: io.signal,
+            onStdoutLine: onProgress ? (line) => {
+              const p = parseIterationLine(line);
+              if (!p) return;
+              onProgress({
+                chain: i + 1,
+                of: chains,
+                fraction: p.total > 0 ? p.iteration / p.total : 0,
+                done: p.iteration === p.total
+              });
+            } : void 0
+          });
+        })
+      );
+    } finally {
+      if (pollTimer) clearInterval(pollTimer);
+    }
+    for (const tail of tails) tail.finish();
+    const elapsedMs = Math.round(performance.now() - start);
+    if (io.signal?.aborted || results.some((r) => r.cancelled)) {
+      return { status: "cancelled", runtimeRequested, elapsedMs };
+    }
+    const failed = results.findIndex((r) => r.code !== 0);
+    if (failed >= 0) {
+      const r = results[failed];
+      return fail(
+        "sample",
+        `chain ${failed + 1} exited with code ${r?.code}${r && errorTail(r.stderr) ? `: ${errorTail(r.stderr)}` : ""}`
+      );
+    }
+    let samplesJson;
+    try {
+      const texts = csvPaths.map((path) => (0, import_node_fs4.readFileSync)(path, "utf8"));
+      samplesJson = JSON.stringify((0, import_core3.toMCMCChainsJson)((0, import_core3.fromStanCSVFiles)(texts)));
+      (0, import_core3.parseSamples)(samplesJson);
+    } catch (error) {
+      return fail("load_samples", `could not read CmdStan output: ${error.message}`);
+    }
+    try {
+      const partial = `${io.outPath}.tmp`;
+      (0, import_node_fs4.writeFileSync)(partial, samplesJson);
+      (0, import_node_fs4.renameSync)(partial, io.outPath);
+    } catch (error) {
+      return fail("write", error.message);
+    }
+    const record = {
+      schema_version: import_core3.RUN_RECORD_SCHEMA_VERSION,
+      spec_hash: spec.specHash,
+      seed: spec.seed,
+      backend: { id: spec.backend.id, runtime: spec.backend.runtime },
+      runtime: { requested: runtimeRequested, actual: install.version, path: install.home },
+      model_sha256: (0, import_node_fs4.existsSync)(spec.modelPath) ? sha256((0, import_node_fs4.readFileSync)(spec.modelPath, "utf8")) : void 0,
+      data_sha256: io.dataSha256 ?? sha256((0, import_core3.canonicalJson)(spec.data)),
+      ...io.dataFile ? { data_file: io.dataFile } : {},
+      samples_file: io.outPath,
+      started_at: startedAt,
+      elapsed_ms: Math.round(performance.now() - start)
+    };
+    try {
+      (0, import_node_fs4.writeFileSync)(
+        io.recordPath ?? `${io.outPath}.run.json`,
+        `${JSON.stringify(record, null, 2)}
+`
+      );
+    } catch (error) {
+      return fail("write", `could not write the run record: ${error.message}`);
+    }
+    return {
+      status: "ok",
+      samplesFile: io.outPath,
+      runtimeRequested,
+      runtimeActual: install.version,
+      elapsedMs: Math.round(performance.now() - start)
+    };
+  } finally {
+    if (ownTmp) (0, import_node_fs4.rmSync)(tmp, { recursive: true, force: true });
+  }
+}
+
+// src/matrix.ts
+var import_node_path4 = require("path");
+async function runMatrix(spec, versions, io) {
+  const entries = [];
+  for (const version of versions) {
+    let entry;
+    try {
+      const install = resolveCmdStan(version);
+      const result = await runFit({ ...spec, backend: { ...spec.backend, version } }, install, {
+        ...io.fit,
+        outPath: (0, import_node_path4.join)(io.outDir, `${version}.samples.json`),
+        dataFile: io.dataFile,
+        dataSha256: io.dataSha256,
+        signal: io.signal
+      });
+      entry = {
+        version,
+        status: result.status,
+        samplesFile: result.samplesFile,
+        runtimeActual: result.runtimeActual,
+        elapsedMs: result.elapsedMs,
+        stage: result.stage,
+        error: result.error
+      };
+    } catch (error) {
+      entry = { version, status: "error", elapsedMs: 0, error: error.message };
+    }
+    entries.push(entry);
+    if (entry.status === "cancelled") break;
+    if (entry.status === "error" && !io.keepGoing) break;
+  }
+  return {
+    entries,
+    ok: entries.length === versions.length && entries.every((e) => e.status === "ok")
+  };
+}
+
+// src/predict.ts
+var import_node_crypto3 = require("crypto");
+var import_node_fs5 = require("fs");
+var import_node_os3 = require("os");
+var import_node_path5 = require("path");
+var import_core4 = require("@mcmcjs/core");
+function predictData(spec) {
+  if (!spec.predict) throw new Error("spec has no [predict] block");
+  return { ...spec.data, ...spec.predict.data ?? {} };
+}
+function fittedParamsCsv(samples, chain) {
+  const names = samples.variables;
+  const header = names.map(import_core4.toStanName).join(",");
+  const columns = names.map((name) => samples.draws.get(name));
+  const start = chain * samples.nDraws;
+  const rows = [header];
+  for (let i = 0; i < samples.nDraws; i++) {
+    rows.push(columns.map((col) => stanCell(col[start + i] ?? Number.NaN)).join(","));
+  }
+  return `${rows.join("\n")}
+`;
+}
+function stanCell(value) {
+  if (Number.isFinite(value)) return String(value);
+  if (Number.isNaN(value)) return "nan";
+  return value > 0 ? "inf" : "-inf";
+}
+function matchesTarget(name, targets) {
+  return targets.some((t) => name === t || name.startsWith(`${t}[`));
+}
+function filterToTargets(all, targets) {
+  const draws = /* @__PURE__ */ new Map();
+  for (const name of all.variables) {
+    if (matchesTarget(name, targets)) {
+      draws.set(name, all.draws.get(name));
+    }
+  }
+  return {
+    variables: [...draws.keys()],
+    nChains: all.nChains,
+    nDraws: all.nDraws,
+    draws,
+    sampleStats: /* @__PURE__ */ new Map()
+  };
+}
+function baseNames(variables) {
+  return [...new Set(variables.map((v) => v.replace(/\[.*$/, "")))];
+}
+function sha2562(text) {
+  return (0, import_node_crypto3.createHash)("sha256").update(text).digest("hex");
+}
+function tmpParent2() {
+  const uid = typeof process.getuid === "function" ? `-${process.getuid()}` : "";
+  return (0, import_node_path5.join)((0, import_node_os3.tmpdir)(), `mcmcjs${uid}`);
+}
+async function runPredict(spec, install, io) {
+  const startedAt = (/* @__PURE__ */ new Date()).toISOString();
+  const start = performance.now();
+  const runtimeRequested = spec.backend.version;
+  const elapsed = () => Math.round(performance.now() - start);
+  const fail = (stage, error) => ({
+    status: "error",
+    runtimeRequested,
+    elapsedMs: elapsed(),
+    stage,
+    error
+  });
+  const targets = spec.predict?.targets;
+  if (!targets || targets.length === 0) {
+    return fail("predict", "spec has no [predict] block");
+  }
+  if (io.signal?.aborted) {
+    return { status: "cancelled", runtimeRequested, elapsedMs: 0 };
+  }
+  let posterior;
+  try {
+    posterior = (0, import_core4.parseSamples)((0, import_node_fs5.readFileSync)(io.samplesPath, "utf8"));
+  } catch (error) {
+    return fail("load_samples", `posterior samples did not parse: ${error.message}`);
+  }
+  let binaryPath;
+  try {
+    ({ binaryPath } = await compileModel(install, spec.modelPath, io.compile));
+  } catch (error) {
+    return fail("compile", error.message);
+  }
+  const ownTmp = io.tmpDir === void 0;
+  let tmp;
+  if (io.tmpDir === void 0) {
+    (0, import_node_fs5.mkdirSync)(tmpParent2(), { recursive: true });
+    tmp = (0, import_node_fs5.mkdtempSync)((0, import_node_path5.join)(tmpParent2(), "stan-predict-"));
+  } else {
+    tmp = io.tmpDir;
+  }
+  try {
+    const data = predictData(spec);
+    const dataPath = (0, import_node_path5.join)(tmp, "data.json");
+    (0, import_node_fs5.writeFileSync)(dataPath, JSON.stringify(data));
+    const spawn2 = io.spawn ?? createStanSpawn();
+    const chains = posterior.nChains;
+    const results = await Promise.all(
+      Array.from({ length: chains }, (_, i) => {
+        const fittedPath = (0, import_node_path5.join)(tmp, `fitted_${i + 1}.csv`);
+        (0, import_node_fs5.writeFileSync)(fittedPath, fittedParamsCsv(posterior, i));
+        return spawn2(
+          binaryPath,
+          [
+            "generate_quantities",
+            `fitted_params=${fittedPath}`,
+            "data",
+            `file=${dataPath}`,
+            "random",
+            `seed=${spec.seed}`,
+            `id=${i + 1}`,
+            "output",
+            `file=${(0, import_node_path5.join)(tmp, `gq_${i + 1}.csv`)}`
+          ],
+          { signal: io.signal }
+        );
+      })
+    );
+    if (io.signal?.aborted || results.some((r) => r.cancelled)) {
+      return { status: "cancelled", runtimeRequested, elapsedMs: elapsed() };
+    }
+    const failed = results.findIndex((r) => r.code !== 0);
+    if (failed >= 0) {
+      const stderr = results[failed]?.stderr ?? "";
+      if (/doesn't generate any quantities/i.test(stderr)) {
+        return fail(
+          "predict",
+          `the Stan model has no generated quantities block; add one that computes ${targets.join(", ")}`
+        );
+      }
+      const tail = stderr.trim().split("\n").filter(Boolean).at(-1);
+      return fail(
+        "predict",
+        `generate_quantities failed for chain ${failed + 1}${tail ? `: ${tail}` : ""}`
+      );
+    }
+    let predictive;
+    try {
+      const texts = Array.from(
+        { length: chains },
+        (_, i) => (0, import_node_fs5.readFileSync)((0, import_node_path5.join)(tmp, `gq_${i + 1}.csv`), "utf8")
+      );
+      predictive = filterToTargets((0, import_core4.fromStanCSVFiles)(texts), targets);
+    } catch (error) {
+      return fail(
+        "load_samples",
+        `could not read generate_quantities output: ${error.message}`
+      );
+    }
+    if (predictive.variables.length === 0) {
+      const available = baseNames(fromStanCSVFilesNames(tmp, chains));
+      return fail(
+        "predict",
+        `no generated quantity matches targets ${targets.join(", ")}; the model generates: ${available.length > 0 ? available.join(", ") : "(nothing)"}`
+      );
+    }
+    try {
+      const json = JSON.stringify((0, import_core4.toMCMCChainsJson)(predictive));
+      (0, import_core4.parseSamples)(json);
+      const partial = `${io.outPath}.tmp`;
+      (0, import_node_fs5.writeFileSync)(partial, json);
+      (0, import_node_fs5.renameSync)(partial, io.outPath);
+    } catch (error) {
+      return fail("write", error.message);
+    }
+    const record = {
+      schema_version: import_core4.RUN_RECORD_SCHEMA_VERSION,
+      spec_hash: spec.specHash,
+      seed: spec.seed,
+      backend: { id: spec.backend.id, runtime: spec.backend.runtime },
+      runtime: { requested: runtimeRequested, actual: install.version, path: install.home },
+      model_sha256: (0, import_node_fs5.existsSync)(spec.modelPath) ? sha2562((0, import_node_fs5.readFileSync)(spec.modelPath, "utf8")) : void 0,
+      data_sha256: sha2562((0, import_core4.canonicalJson)(data)),
+      posterior_samples: io.samplesPath,
+      posterior_samples_sha256: sha2562((0, import_node_fs5.readFileSync)(io.samplesPath, "utf8")),
+      samples_file: io.outPath,
+      started_at: startedAt,
+      elapsed_ms: elapsed()
+    };
+    try {
+      (0, import_node_fs5.writeFileSync)(`${io.outPath}.run.json`, `${JSON.stringify(record, null, 2)}
+`);
+    } catch (error) {
+      return fail("write", `could not write the run record: ${error.message}`);
+    }
+    return {
+      status: "ok",
+      samplesFile: io.outPath,
+      runtimeRequested,
+      runtimeActual: install.version,
+      elapsedMs: elapsed()
+    };
+  } finally {
+    if (ownTmp) (0, import_node_fs5.rmSync)(tmp, { recursive: true, force: true });
+  }
+}
+function fromStanCSVFilesNames(tmp, chains) {
+  try {
+    const texts = Array.from(
+      { length: chains },
+      (_, i) => (0, import_node_fs5.readFileSync)((0, import_node_path5.join)(tmp, `gq_${i + 1}.csv`), "utf8")
+    );
+    return [...(0, import_core4.fromStanCSVFiles)(texts).variables];
+  } catch {
+    return [];
+  }
+}
+
+// src/setup.ts
+var import_node_fs6 = require("fs");
+var import_node_os4 = require("os");
+var import_node_path6 = require("path");
+var import_engine5 = require("@mcmcjs/engine");
+function releaseUrl(version) {
+  return `https://github.com/stan-dev/cmdstan/releases/download/v${version}/cmdstan-${version}.tar.gz`;
+}
+function shq(value) {
+  return `'${value.replaceAll("'", `'\\''`)}'`;
+}
+function managedCmdStanHome(version) {
+  return (0, import_node_path6.join)(managedStanRoot(), `cmdstan-${version}`);
+}
+function planSetup(report, platform, version = PINNED_CMDSTAN_VERSION) {
+  const steps = [];
+  if (!report.make.found || !report.cxx.found) {
+    const missing = [
+      ...report.make.found ? [] : ["make"],
+      ...report.cxx.found ? [] : ["a C++ compiler (g++ or clang++)"]
+    ].join(" and ");
+    steps.push({
+      tool: "toolchain",
+      label: `install ${missing} with your system package manager`,
+      command: null
+    });
+  }
+  if (report.cmdstan.found && report.cmdstan.version === version) return steps;
+  const home = managedCmdStanHome(version);
+  const posix = platform !== "win32";
+  if (!(0, import_node_fs6.existsSync)((0, import_node_path6.join)(home, "makefile"))) {
+    const root = shq(managedStanRoot());
+    const staging = shq((0, import_node_path6.join)(managedStanRoot(), `.download-${version}`));
+    const finalHome = shq(home);
+    const extracted = shq((0, import_node_path6.join)(managedStanRoot(), `.download-${version}`, `cmdstan-${version}`));
+    steps.push({
+      tool: "cmdstan",
+      label: `download CmdStan ${version}`,
+      command: posix ? {
+        command: "sh",
+        args: [
+          "-c",
+          `rm -rf ${staging} && mkdir -p ${staging} && curl -fsSL ${shq(releaseUrl(version))} | tar -xz -C ${staging} && rm -rf ${finalHome} && mv ${extracted} ${finalHome} && rm -rf ${staging} && mkdir -p ${root}`
+        ]
+      } : null
+    });
+  }
+  if (!(0, import_node_fs6.existsSync)((0, import_node_path6.join)(home, "bin", "stansummary"))) {
+    steps.push({
+      tool: "build",
+      label: `build CmdStan ${version} (one-time, a few minutes)`,
+      command: posix ? { command: "make", args: ["-C", home, "build", `-j${(0, import_node_os4.cpus)().length}`] } : null
+    });
+  }
+  return steps;
+}
+function validateVersion(version) {
+  if (!/^\d+\.\d+\.\d+(-rc\d+)?$/.test(version)) {
+    throw new Error(
+      `invalid CmdStan version: "${version}" (expected e.g. ${PINNED_CMDSTAN_VERSION})`
+    );
+  }
+}
+async function runSetup(options = {}) {
+  const {
+    runner,
+    installer = (0, import_engine5.createRunner)(30 * 6e4),
+    platform = process.platform,
+    dryRun = false,
+    version = PINNED_CMDSTAN_VERSION
+  } = options;
+  validateVersion(version);
+  const before = await runDoctor(runner);
+  const plan = planSetup(before, platform, version);
+  const strip = ({ install: _install, ...report }) => report;
+  if (plan.length === 0) return { ...strip(before), steps: [] };
+  if (dryRun) {
+    return {
+      ...strip(before),
+      steps: plan.map((step) => ({ ...step, status: step.command ? "skipped" : "unsupported" }))
+    };
+  }
+  const steps = [];
+  let failed = false;
+  for (const step of plan) {
+    if (!step.command) {
+      steps.push({ ...step, status: "unsupported" });
+      failed = failed || step.tool === "toolchain";
+      continue;
+    }
+    if (failed) {
+      steps.push({ ...step, status: "skipped" });
+      continue;
+    }
+    try {
+      await installer(step.command.command, step.command.args);
+      steps.push({ ...step, status: "ran" });
+    } catch (error) {
+      const detail = error instanceof Error ? error.message : String(error);
+      steps.push({ ...step, status: "failed", detail });
+      failed = true;
+    }
+  }
+  const after = await runDoctor(runner);
+  return { ...strip(after), steps };
+}
+
+// src/versions.ts
+var import_node_fs7 = require("fs");
+var import_node_path7 = require("path");
+function listVersions() {
+  return listCmdStanInstalls().map((install, index) => ({
+    id: install.version,
+    version: install.version,
+    path: install.home,
+    isDefault: index === 0
+  }));
+}
+function addVersion(version, options = {}) {
+  return runSetup({ ...options, version });
+}
+function removeVersion(version) {
+  const install = listCmdStanInstalls().find((i) => i.version === version);
+  if (!install) {
+    throw new Error(`CmdStan ${version} is not installed`);
+  }
+  const managedHome = (0, import_node_path7.join)(managedStanRoot(), `cmdstan-${version}`);
+  if (install.home !== managedHome) {
+    throw new Error(
+      `CmdStan ${version} at ${install.home} is not managed by mcmcjs; remove it yourself if intended`
+    );
+  }
+  (0, import_node_fs7.rmSync)(managedHome, { recursive: true, force: true });
+}
+// Annotate the CommonJS export names for ESM import in node:
+0 && (module.exports = {
+  INSTALLED_CMDSTAN_CHANNEL,
+  PINNED_CMDSTAN_VERSION,
+  addVersion,
+  chainArgs,
+  columnToLeaf,
+  compileModel,
+  createStanCsvTail,
+  createStanSpawn,
+  fittedParamsCsv,
+  isCmdStanHome,
+  listCmdStanInstalls,
+  listVersions,
+  managedCmdStanHome,
+  managedStanRoot,
+  matchesTarget,
+  modelCacheDir,
+  parseIterationLine,
+  planSetup,
+  predictData,
+  removeVersion,
+  resolveCmdStan,
+  runDoctor,
+  runFit,
+  runMatrix,
+  runPredict,
+  runSetup,
+  stanEngine
+});
+//# sourceMappingURL=index.cjs.map