@hoox-sh/pynets 0.1.0 → 0.2.0

package/package.json

@@ -1,6 +1,6 @@
 {
   "name": "@hoox-sh/pynets",
-  "version": "0.1.0",
+  "version": "0.2.0",
   "type": "module",
   "license": "AGPL-3.0-or-later",
   "author": "jango_blockchained",

package/src/cli/rich.ts

@@ -12,7 +12,7 @@ const MUTED = [203, 204, 189] as const;
 const FAIL = [231, 0, 11] as const;
 const WARN = [235, 169, 65] as const;
 
-export const VERSION = "0.1.0";
+export const VERSION = "0.2.0";
 
 type Rgb = readonly [number, number, number];
 
@@ -122,7 +122,7 @@ export class Rich {
       this.write(this.paint("accent", "COMMANDS"));
     } else {
       this.write(
-        "usage: pynets <check|format|run|dump|info> [file] [--bars N] [--commission N] [--slippage N] [--pyramiding N] [--json] [--plain] [--rich] [--full]",
+        "usage: pynets <check|format|run|dump|info> [file] [--bars N] [--mode interpret|compile|auto] [--commission N] [--slippage N] [--pyramiding N] [--json] [--plain] [--rich] [--full]",
       );
       this.write("");
       this.write("commands:");
@@ -130,7 +130,7 @@ export class Rich {
     const cmds: Array<[string, string]> = [
       ["check", "Parse-only validation (CI-friendly exit codes)"],
       ["format", "Parse → unparse (pretty-print on TTY)"],
-      ["run", "Interpret on synthetic OHLCV"],
+      ["run", "Run on synthetic OHLCV (interpret or compile JS emit)"],
       ["dump", "Dump the ASDL-shaped AST"],
       ["info", "Version and runtime extras"],
     ];
@@ -146,6 +146,7 @@ export class Rich {
     else this.write("options:");
     const opts: Array<[string, string]> = [
       ["--bars N", "Synthetic bar count for run (default 20, max 100000)"],
+      ["--mode MODE", "run engine: interpret | compile | auto (JS emit; default interpret)"],
       ["--commission N", "Broker commission fraction for run (e.g. 0.001)"],
       ["--slippage N", "Broker slippage in price units for run"],
       ["--pyramiding N", "Broker max same-direction adds for run"],

package/src/cli.ts

@@ -20,6 +20,8 @@ import {
   useRich,
 } from "./cli/rich.ts";
 
+export type RunMode = "interpret" | "compile" | "auto";
+
 export const MAX_BARS = 100_000;
 
 export class UsageError extends Error {
@@ -155,6 +157,11 @@ function takeValue(args: string[], i: number, flag: string): { raw: string | und
   return { raw: a.slice(flag.length + 1), next: i };
 }
 
+function requireMode(raw: string | undefined): RunMode {
+  if (raw === "interpret" || raw === "compile" || raw === "auto") return raw;
+  throw new UsageError("error: --mode must be interpret|compile|auto");
+}
+
 export function parseArgs(argv: string[]): {
   cmd: string;
   file: string | undefined;
@@ -162,6 +169,7 @@ export function parseArgs(argv: string[]): {
   json: boolean;
   indent: number;
   full: boolean;
+  mode: RunMode;
   commission: number | undefined;
   slippage: number | undefined;
   pyramiding: number | undefined;
@@ -174,13 +182,14 @@ export function parseArgs(argv: string[]): {
   let json = false;
   let indent = 2;
   let full = false;
+  let mode: RunMode = "interpret";
   let commission: number | undefined;
   let slippage: number | undefined;
   let pyramiding: number | undefined;
   for (let i = 1; i < args.length; i++) {
     const a = args[i]!;
     if (a === "--help" || a === "-h") {
-      return { cmd: "help", file: undefined, bars, json, indent, full, commission, slippage, pyramiding };
+      return { cmd: "help", file: undefined, bars, json, indent, full, mode, commission, slippage, pyramiding };
     }
     if (a === "--json") {
       json = true;
@@ -190,6 +199,12 @@ export function parseArgs(argv: string[]): {
       full = true;
       continue;
     }
+    if (a === "--mode" || a.startsWith("--mode=")) {
+      const got = takeValue(args, i, "--mode");
+      mode = requireMode(got.raw);
+      i = got.next;
+      continue;
+    }
     if (a === "--bars" || a.startsWith("--bars=")) {
       const got = takeValue(args, i, "--bars");
       bars = clampBars(requireUint(got.raw, "--bars"));
@@ -230,7 +245,7 @@ export function parseArgs(argv: string[]): {
     if (file != null) throw new UsageError(`error: unexpected argument: ${a}`);
     file = a;
   }
-  return { cmd, file, bars, json, indent, full, commission, slippage, pyramiding };
+  return { cmd, file, bars, json, indent, full, mode, commission, slippage, pyramiding };
 }
 
 function fmtNum(v: number | null | undefined): string {
@@ -346,21 +361,27 @@ function runJson(out: RuntimeOut): void {
     series: typeof out.series;
     count: number;
     script_name: string | null;
+    mode: RuntimeOut["mode"];
     events?: RuntimeEvent[];
     drawings?: DrawingLike[];
     fills?: RuntimeFill[];
     alerts?: AlertLike[];
+    compile_fallback_reason?: string;
+    auto_backend?: RuntimeOut["auto_backend"];
     error?: string;
   } = {
     plots: out.plots,
     series: out.series,
     count: out.count,
     script_name: out.script_name,
+    mode: out.mode,
   };
   if (out.events != null) payload.events = out.events;
   if (out.drawings != null) payload.drawings = out.drawings;
   if (out.fills != null) payload.fills = out.fills;
   if (out.alerts != null) payload.alerts = out.alerts;
+  if (out.compile_fallback_reason != null) payload.compile_fallback_reason = out.compile_fallback_reason;
+  if (out.auto_backend != null) payload.auto_backend = out.auto_backend;
   if (out.error != null) payload.error = out.error;
   writeJson(payload);
   if (out.error != null) process.exit(1);
@@ -506,11 +527,12 @@ function run(
   bars: number,
   asJson: boolean,
   broker: BrokerSettings,
+  mode: RunMode,
 ): void {
   const t0 = performance.now();
   let out: RuntimeOut;
   try {
-    out = new Runtime("AAPL", { broker }).run(source, syntheticBars(bars));
+    out = new Runtime("AAPL", { broker, mode }).run(source, syntheticBars(bars));
   } catch (err) {
     fail(ui, err instanceof Error ? err.message : String(err), 1);
   }
@@ -538,7 +560,7 @@ function infoPayload(ui: Rich): {
     version: VERSION,
     runtime: "bun",
     bun: typeof Bun !== "undefined" ? Bun.version : null,
-    mode: "interpret",
+    mode: "interpret+compile",
     rich: ui.enabled,
     docs: "https://hoox.sh/pyne",
   };
@@ -558,7 +580,7 @@ function info(ui: Rich, asJson: boolean): void {
       ["package", "@hoox-sh/pynets"],
       ["version", VERSION],
       ["runtime", `bun ${payload.bun ?? "?"}`],
-      ["engine", "interpret"],
+      ["engine", "interpret + compile (JS emit)"],
       ["rich", "yes (TTY)"],
       ["docs", payload.docs],
     ],
@@ -577,7 +599,7 @@ function main(): void {
     failUsage(ui, msg);
   }
 
-  const { cmd, file, bars, json, indent, full } = parsed;
+  const { cmd, file, bars, json, indent, full, mode } = parsed;
   const broker = brokerFromFlags(parsed);
   if (cmd === "help" || cmd === "-h" || cmd === "--help") {
     ui.help();
@@ -610,7 +632,7 @@ function main(): void {
     return;
   }
   if (cmd === "run") {
-    run(ui, readSource(ui, file), file, bars, json, broker);
+    run(ui, readSource(ui, file), file, bars, json, broker, mode);
     return;
   }
   failUsage(ui, `error: unknown command: ${cmd}`);

package/src/index.ts

@@ -20,6 +20,27 @@ export type {
   StrategyEvent,
   StreamEvent,
 } from "./runtime/interpret.ts";
+export {
+  compileScript,
+  transpile,
+  compileEligible,
+  clearCompileCache,
+  compileCacheStats,
+  runScript,
+  compileToResult,
+  CompileError,
+  CompileEmitError,
+  CompileIneligibleError,
+  CompileLoadError,
+} from "./runtime/compile/index.ts";
+export type {
+  CompileBackend,
+  CompileCacheStats,
+  CompileEligibility,
+  CompilePlotMeta,
+  CompiledScript,
+  RuntimeMode,
+} from "./runtime/compile/index.ts";
 export { NA, PineSeries } from "./runtime/series.ts";
 export { TaEngine } from "./runtime/ta.ts";
 export type { BrokerSettings, StrategySummary } from "./runtime/strategy.ts";

package/src/runtime/compile/collections.ts

@@ -0,0 +1,339 @@
+/**
+ * Copyright (C) 2024-2026 jango_blockchained
+ * SPDX-License-Identifier: AGPL-3.0-or-later
+ *
+ * Compile-path array / map / matrix facades. Wraps PineArray / PineMap / PineMatrix.
+ */
+import { PineArray } from "../array.ts";
+import { PineMap } from "../map.ts";
+import { PineMatrix } from "../matrix.ts";
+import { naNum } from "./helpers.ts";
+
+function n(v: unknown, fallback = 0): number {
+  const x = naNum(v);
+  return x == null ? fallback : x;
+}
+
+function cell(v: unknown): number | null {
+  if (v != null && typeof v === "object") return null;
+  return naNum(v);
+}
+
+function asArray(id: unknown): PineArray | null {
+  return id instanceof PineArray ? id : null;
+}
+
+function asMap(id: unknown): PineMap | null {
+  return id instanceof PineMap ? id : null;
+}
+
+function asMatrix(id: unknown): PineMatrix | null {
+  return id instanceof PineMatrix ? id : null;
+}
+
+function sortOrder(order: unknown): "asc" | "desc" {
+  return String(order ?? "asc").toLowerCase().includes("desc") ? "desc" : "asc";
+}
+
+function wrapCells(cells: ReadonlyArray<number | null> | null | undefined): PineArray | null {
+  if (cells == null) return null;
+  const out = new PineArray();
+  for (const v of cells) out.push(v);
+  return out;
+}
+
+export const compileArray = {
+  new(size?: unknown, initial?: unknown): PineArray {
+    return new PineArray(n(size, 0), cell(initial));
+  },
+  get(id: unknown, index?: unknown): number | null {
+    return asArray(id)?.get(n(index, 0)) ?? null;
+  },
+  set(id: unknown, index?: unknown, value?: unknown): unknown {
+    asArray(id)?.set(n(index, 0), cell(value));
+    return id ?? null;
+  },
+  push(id: unknown, value?: unknown): unknown {
+    asArray(id)?.push(cell(value));
+    return id ?? null;
+  },
+  pop(id: unknown): number | null {
+    return asArray(id)?.pop() ?? null;
+  },
+  unshift(id: unknown, value?: unknown): unknown {
+    asArray(id)?.unshift(cell(value));
+    return id ?? null;
+  },
+  shift(id: unknown): number | null {
+    return asArray(id)?.shift() ?? null;
+  },
+  size(id: unknown): number {
+    return asArray(id)?.size() ?? 0;
+  },
+  includes(id: unknown, value?: unknown): boolean {
+    return asArray(id)?.includes(cell(value)) ?? false;
+  },
+  first(id: unknown): number | null {
+    return asArray(id)?.first() ?? null;
+  },
+  last(id: unknown): number | null {
+    return asArray(id)?.last() ?? null;
+  },
+  insert(id: unknown, index?: unknown, value?: unknown): unknown {
+    asArray(id)?.insert(n(index, 0), cell(value));
+    return id ?? null;
+  },
+  remove(id: unknown, index?: unknown): number | null {
+    return asArray(id)?.remove(n(index, 0)) ?? null;
+  },
+  clear(id: unknown): unknown {
+    asArray(id)?.clear();
+    return id ?? null;
+  },
+  fill(id: unknown, value?: unknown): unknown {
+    asArray(id)?.fill(cell(value));
+    return id ?? null;
+  },
+  copy(id: unknown): PineArray | null {
+    return asArray(id)?.copy() ?? null;
+  },
+  reverse(id: unknown): unknown {
+    asArray(id)?.reverse();
+    return id ?? null;
+  },
+  sort(id: unknown, order?: unknown): unknown {
+    const o = String(order ?? "asc").toLowerCase();
+    asArray(id)?.sort(o.startsWith("desc") ? "desc" : "asc");
+    return id ?? null;
+  },
+  concat(id: unknown, other?: unknown): PineArray | null {
+    const a = asArray(id);
+    const b = asArray(other);
+    if (a == null) return b?.copy() ?? null;
+    if (b == null) return a;
+    return a.concat(b);
+  },
+  slice(id: unknown, from?: unknown, to?: unknown): PineArray | null {
+    return asArray(id)?.slice(n(from, 0), to == null ? undefined : n(to, 0)) ?? null;
+  },
+  indexof(id: unknown, value?: unknown): number | null {
+    return asArray(id)?.indexof(cell(value)) ?? null;
+  },
+  lastindexof(id: unknown, value?: unknown): number | null {
+    return asArray(id)?.lastIndexOf(cell(value)) ?? null;
+  },
+  avg(id: unknown): number | null {
+    return asArray(id)?.avg() ?? null;
+  },
+  min(id: unknown): number | null {
+    return asArray(id)?.min() ?? null;
+  },
+  max(id: unknown): number | null {
+    return asArray(id)?.max() ?? null;
+  },
+  sum(id: unknown): number | null {
+    return asArray(id)?.sum() ?? null;
+  },
+  stdev(id: unknown, biased?: unknown): number | null {
+    return asArray(id)?.stdev(biased !== false && biased !== 0) ?? null;
+  },
+  variance(id: unknown, biased?: unknown): number | null {
+    return asArray(id)?.variance(biased !== false && biased !== 0) ?? null;
+  },
+  join(id: unknown, sep?: unknown): string {
+    return asArray(id)?.join(sep == null ? "," : String(sep)) ?? "";
+  },
+  every(id: unknown): boolean {
+    return asArray(id)?.every() ?? false;
+  },
+  some(id: unknown): boolean {
+    return asArray(id)?.some() ?? false;
+  },
+  abs(id: unknown): PineArray | null {
+    return asArray(id)?.abs() ?? null;
+  },
+  range(id: unknown): number | null {
+    return asArray(id)?.range() ?? null;
+  },
+  percentile_nearest(id: unknown, percentage?: unknown): number | null {
+    const a = asArray(id);
+    const p = naNum(percentage);
+    return a != null && p != null ? a.percentileNearestRank(p) : null;
+  },
+  percentileNearestRank(id: unknown, percentage?: unknown): number | null {
+    return compileArray.percentile_nearest(id, percentage);
+  },
+  percentile_linear(id: unknown, percentage?: unknown): number | null {
+    const a = asArray(id);
+    const p = naNum(percentage);
+    return a != null && p != null ? a.percentileLinearInterpolation(p) : null;
+  },
+  percentileLinearInterpolation(id: unknown, percentage?: unknown): number | null {
+    return compileArray.percentile_linear(id, percentage);
+  },
+  percentrank(id: unknown, value?: unknown): number | null {
+    return asArray(id)?.percentrank(cell(value)) ?? null;
+  },
+  binary_search(id: unknown, value?: unknown): number | null {
+    return asArray(id)?.binarySearch(cell(value)) ?? null;
+  },
+  binary_search_leftmost(id: unknown, value?: unknown): number | null {
+    return asArray(id)?.binarySearchLeftmost(cell(value)) ?? null;
+  },
+  binary_search_rightmost(id: unknown, value?: unknown): number | null {
+    return asArray(id)?.binarySearchRightmost(cell(value)) ?? null;
+  },
+  sort_indices(id: unknown, order?: unknown): PineArray | null {
+    const idx = asArray(id)?.sortIndices(sortOrder(order));
+    return wrapCells(idx);
+  },
+};
+
+export const compileMap = {
+  new(): PineMap {
+    return new PineMap();
+  },
+  put(id: unknown, key?: unknown, value?: unknown): unknown {
+    asMap(id)?.put(key, cell(value));
+    return id ?? null;
+  },
+  get(id: unknown, key?: unknown): number | null {
+    return asMap(id)?.get(key) ?? null;
+  },
+  contains(id: unknown, key?: unknown): boolean {
+    return asMap(id)?.contains(key) ?? false;
+  },
+  remove(id: unknown, key?: unknown): number | null {
+    return asMap(id)?.remove(key) ?? null;
+  },
+  clear(id: unknown): unknown {
+    asMap(id)?.clear();
+    return id ?? null;
+  },
+  size(id: unknown): number {
+    return asMap(id)?.size() ?? 0;
+  },
+  keys(id: unknown): unknown[] {
+    return asMap(id)?.keys() ?? [];
+  },
+  values(id: unknown): Array<number | null> {
+    return asMap(id)?.values() ?? [];
+  },
+  copy(id: unknown): PineMap | null {
+    return asMap(id)?.copy() ?? null;
+  },
+  put_all(id: unknown, other?: unknown): unknown {
+    const m = asMap(id);
+    const o = asMap(other);
+    if (m != null && o != null) m.putAll(o);
+    return id ?? null;
+  },
+};
+
+export const compileMatrix = {
+  new(rows?: unknown, cols?: unknown, initial?: unknown): PineMatrix {
+    return new PineMatrix(n(rows, 0), n(cols, 0), cell(initial));
+  },
+  get(id: unknown, row?: unknown, col?: unknown): number | null {
+    return asMatrix(id)?.get(n(row, 0), n(col, 0)) ?? null;
+  },
+  set(id: unknown, row?: unknown, col?: unknown, value?: unknown): unknown {
+    asMatrix(id)?.set(n(row, 0), n(col, 0), cell(value));
+    return id ?? null;
+  },
+  rows(id: unknown): number {
+    return asMatrix(id)?.rows() ?? 0;
+  },
+  columns(id: unknown): number {
+    return asMatrix(id)?.columns() ?? 0;
+  },
+  fill(id: unknown, value?: unknown): unknown {
+    asMatrix(id)?.fill(cell(value));
+    return id ?? null;
+  },
+  transpose(id: unknown): PineMatrix | null {
+    return asMatrix(id)?.transpose() ?? null;
+  },
+  copy(id: unknown): PineMatrix | null {
+    return asMatrix(id)?.copy() ?? null;
+  },
+  row(id: unknown, index?: unknown): PineArray | null {
+    const m = asMatrix(id);
+    return m == null ? null : wrapCells(m.row(n(index, 0)));
+  },
+  col(id: unknown, index?: unknown): PineArray | null {
+    const m = asMatrix(id);
+    return m == null ? null : wrapCells(m.col(n(index, 0)));
+  },
+  sum(id: unknown): number | null {
+    return asMatrix(id)?.sum() ?? null;
+  },
+  avg(id: unknown): number | null {
+    return asMatrix(id)?.avg() ?? null;
+  },
+  min(id: unknown): number | null {
+    return asMatrix(id)?.min() ?? null;
+  },
+  max(id: unknown): number | null {
+    return asMatrix(id)?.max() ?? null;
+  },
+  det(id: unknown): number | null {
+    return asMatrix(id)?.det() ?? null;
+  },
+  mult(id: unknown, other?: unknown): PineMatrix | null {
+    const m = asMatrix(id);
+    if (m == null) return null;
+    const o = asMatrix(other);
+    if (o != null) return m.mult(o);
+    const s = naNum(other);
+    return s == null ? null : m.mult(s);
+  },
+  inv(id: unknown): PineMatrix | null {
+    return asMatrix(id)?.inv() ?? null;
+  },
+  pinv(id: unknown): PineMatrix | null {
+    return asMatrix(id)?.pinv() ?? null;
+  },
+  kron(id: unknown, other?: unknown): PineMatrix | null {
+    const m = asMatrix(id);
+    const o = asMatrix(other);
+    return m != null && o != null ? m.kron(o) : null;
+  },
+  rank(id: unknown): number | null {
+    return asMatrix(id)?.rank() ?? null;
+  },
+  add(id: unknown, other?: unknown): PineMatrix | null {
+    const m = asMatrix(id);
+    if (m == null) return null;
+    const o = asMatrix(other);
+    if (o != null) {
+      const neg = o.mult(-1);
+      return neg == null ? null : m.diff(neg);
+    }
+    const s = naNum(other);
+    if (s == null) return null;
+    return m.diff(new PineMatrix(m.rows(), m.columns(), -s));
+  },
+  diff(id: unknown, other?: unknown): PineMatrix | null {
+    const m = asMatrix(id);
+    const o = asMatrix(other);
+    return m != null && o != null ? m.diff(o) : null;
+  },
+  reshape(id: unknown, rows?: unknown, cols?: unknown): PineMatrix | null {
+    return asMatrix(id)?.reshape(n(rows, 0), n(cols, 0)) ?? null;
+  },
+  reverse(id: unknown): unknown {
+    asMatrix(id)?.reverse();
+    return id ?? null;
+  },
+  sort(id: unknown, column?: unknown, order?: unknown): unknown {
+    asMatrix(id)?.sort(n(column, 0), sortOrder(order));
+    return id ?? null;
+  },
+  eigenvalues(id: unknown): PineArray | null {
+    return wrapCells(asMatrix(id)?.eigenvalues());
+  },
+  eigenvectors(id: unknown): PineMatrix | null {
+    return asMatrix(id)?.eigenvectors() ?? null;
+  },
+};