@aethervtt/shard-core 0.1.0 → 0.2.1

package/dist/test-env.js

@@ -1,8 +1,73 @@
+import { ShardError } from "./error.js";
 const env = globalThis.process?.env ?? {};
 const timingMode = env.SHARD_BENCH ? "bench" : env.SHARD_CI ? "ci" : "test";
 const slack = timingMode === "bench" ? 1 : Number.POSITIVE_INFINITY;
-function budget(ms) {
-  return ms * slack;
+function unknownBudget(key, detail = "") {
+  return new ShardError("perf/unknown-budget", `No budget "${key}"${detail} in budgets.json`, {
+    path: key,
+    hint: "Add it to bench/perf/budgets.json with a number for every machine and a note."
+  });
+}
+function resolveLimit(budgets2, key, opts = {}, enforce = true) {
+  if (!budgets2) {
+    if (enforce) {
+      throw new ShardError("perf/no-budgets", `No budgets to resolve "${key}" against`, {
+        hint: "pnpm bench passes them in SHARD_BUDGETS; run tests through it or through testFiles()."
+      });
+    }
+    return { limit: Number.POSITIVE_INFINITY, reference: null, kind: "target" };
+  }
+  const on = enforce && budgets2.enforce;
+  const span = budgets2.spans[key];
+  if (span) {
+    let reference2 = span.value;
+    if (reference2 !== null && span.kind === "unit") reference2 *= opts.count ?? 1;
+    if (reference2 !== null && span.kind === "guard") reference2 *= 1 + (span.margin ?? 0);
+    const off = span.kind === "rate" ? 0 : Number.POSITIVE_INFINITY;
+    return { limit: on && reference2 !== null ? reference2 : off, reference: reference2, kind: span.kind };
+  }
+  const scenario = budgets2.scenarios[key];
+  if (!scenario) throw unknownBudget(key);
+  const track = opts.track ?? (opts.slice?.startsWith("gpu:") ? "gpu" : "cpu");
+  const slices = scenario.slices[track];
+  if (opts.slice !== void 0 && !(opts.slice in slices)) {
+    throw unknownBudget(key, ` slice "${opts.slice}" (${track})`);
+  }
+  const reference = opts.slice !== void 0 ? slices[opts.slice] : scenario.frame[track];
+  return {
+    limit: on && reference !== null ? reference : Number.POSITIVE_INFINITY,
+    reference,
+    kind: "slice"
+  };
+}
+let resolved;
+let parsed = false;
+function budgets() {
+  if (!parsed) {
+    parsed = true;
+    if (env.SHARD_BUDGETS) resolved = JSON.parse(env.SHARD_BUDGETS);
+  }
+  return resolved;
+}
+function scenarioSlices(name) {
+  const scenario = budgets()?.scenarios[name];
+  if (!scenario) throw unknownBudget(name);
+  const keys = (track) => Object.keys(scenario.slices[track]).filter((k) => k !== "headroom");
+  return { gpu: keys("gpu"), cpu: keys("cpu") };
+}
+function passTiming() {
+  return timingMode === "bench" && budgets()?.passTiming === "ablation" ? "ablation" : "timestamps";
+}
+function budget(key, opts = {}) {
+  if (typeof key === "number") return key * slack;
+  const bench = timingMode === "bench";
+  const data = budgets();
+  const result = resolveLimit(data, key, opts, bench);
+  if (bench && env.SHARD_PERF_REPORT) {
+    const call = { key, opts, used: false, ...result };
+    globalThis.__shardBudget = call;
+  }
+  return result.limit;
 }
 function timeout(ms) {
   return timingMode === "ci" ? ms * 5 : ms;
@@ -31,6 +96,9 @@ export {
   allocationChecks,
   budget,
   gcWindow,
+  passTiming,
+  resolveLimit,
+  scenarioSlices,
   slack,
   timeout,
   timingMode

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@aethervtt/shard-core",
-  "version": "0.1.0",
+  "version": "0.2.1",
   "license": "MIT",
   "type": "module",
   "sideEffects": false,
@@ -9,6 +9,10 @@
       "types": "./dist/index.d.ts",
       "default": "./dist/index.js"
     },
+    "./capture": {
+      "types": "./dist/scheduler/capture/index.d.ts",
+      "default": "./dist/scheduler/capture/index.js"
+    },
     "./math": {
       "types": "./dist/math/index.d.ts",
       "default": "./dist/math/index.js"
@@ -33,6 +37,6 @@
     "provenance": true
   },
   "shard": {
-    "buildSha": "8e14c9896f82a3554b156f1ffdd68f2cc3e7d78a"
+    "buildSha": "772eb0d2b8a5fc499105708f4ae92c58788cc9ea"
   }
 }