oakscriptjs 0.9.5 → 0.10.0

package/dist/cjs/script/index.js

@@ -69,9 +69,12 @@ __export(script_exports, {
   open: () => open,
   plot: () => plot,
   plotarrow: () => plotarrow,
+  plotbar: () => plotbar,
+  plotcandle: () => plotcandle,
   plotchar: () => plotchar,
   plotshape: () => plotshape,
   polyline: () => polyline,
+  position: () => import_types.position,
   request: () => request,
   second: () => second,
   seriesOf: () => seriesOf,
@@ -80,6 +83,7 @@ __export(script_exports, {
   strategy: () => strategy,
   syminfo: () => syminfo,
   ta: () => ta,
+  table: () => table,
   text: () => text,
   ticker: () => ticker,
   time: () => time,
@@ -102,10 +106,12 @@ var labelCore = __toESM(require("../label/index.js"), 1);
 var boxCore = __toESM(require("../box/index.js"), 1);
 var linefillCore = __toESM(require("../linefill/index.js"), 1);
 var polylineCore = __toESM(require("../polyline/index.js"), 1);
+var tableCore = __toESM(require("../table/index.js"), 1);
 var chartPointCore = __toESM(require("../chartpoint/index.js"), 1);
 var textCore = __toESM(require("../text/index.js"), 1);
 var mathCore = __toESM(require("../math/index.js"), 1);
 var import_registry = require("../drawing/registry.js");
+var import_output = require("../drawing/output.js");
 var timeframeCore = __toESM(require("../timeframe/index.js"), 1);
 var timeCore = __toESM(require("../time/index.js"), 1);
 var strCore = __toESM(require("../str/index.js"), 1);
@@ -113,7 +119,9 @@ var import_messageformat = require("../str/messageformat.js");
 var import_calendar = require("../session/calendar.js");
 var import_bars = require("../session/bars.js");
 var import_resample = require("../security/resample.js");
+var import_types = require("../types/index.js");
 var import_strategy = require("../strategy/index.js");
+var import_session = require("../time/session.js");
 var import_strategy2 = require("../strategy/index.js");
 var compare = __toESM(require("../compare/index.js"), 1);
 const ctxBars = new import_series.BarData([]);
@@ -144,8 +152,11 @@ function freshCollector(inputValues) {
     arrowConfig: [],
     arrows: [],
     barColorConfig: [],
-    bgcolors: [],
-    barcolors: [],
+    bgColors: [],
+    barColors: [],
+    candleConfig: [],
+    plotCandles: {},
+    plotBars: {},
     alertConfig: [],
     alerts: []
   };
@@ -164,6 +175,7 @@ function executeScript(body, barsInput, inputs = {}, chart2 = {}, options = {})
   scriptOptions = options;
   ctx = freshCollector(inputs);
   (0, import_registry.resetDrawings)();
+  mathCore.resetRandom();
   try {
     body();
     const c = ctx;
@@ -182,6 +194,7 @@ function executeScript(body, barsInput, inputs = {}, chart2 = {}, options = {})
       shapeConfig: c.shapeConfig,
       arrowConfig: c.arrowConfig,
       barColorConfig: c.barColorConfig,
+      candleConfig: c.candleConfig,
       defaultInputs: c.defaultInputs,
       alertConfig: c.alertConfig,
       strategyConfig: c.strategy?.properties,
@@ -191,8 +204,11 @@ function executeScript(body, barsInput, inputs = {}, chart2 = {}, options = {})
         fills: c.fills.length ? c.fills : void 0,
         markers: c.markers.length ? c.markers : void 0,
         arrows: c.arrows.length ? c.arrows : void 0,
-        bgcolors: c.bgcolors.length ? c.bgcolors : void 0,
-        barcolors: c.barcolors.length ? c.barcolors : void 0,
+        bgColors: c.bgColors.length ? c.bgColors : void 0,
+        barColors: c.barColors.length ? c.barColors : void 0,
+        plotCandles: Object.keys(c.plotCandles).length ? c.plotCandles : void 0,
+        plotBars: Object.keys(c.plotBars).length ? c.plotBars : void 0,
+        ...(0, import_output.drawingOutputs)(ctxBars.bars, chartCtx.timeUnit),
         alerts: c.alerts.length ? c.alerts : void 0
       }
     };
@@ -299,6 +315,7 @@ const hl2 = new import_series.Series(ctxBars, (b) => (b.high + b.low) / 2);
 const hlc3 = new import_series.Series(ctxBars, (b) => (b.high + b.low + b.close) / 3);
 const ohlc4 = new import_series.Series(ctxBars, (b) => (b.open + b.high + b.low + b.close) / 4);
 const hlcc4 = new import_series.Series(ctxBars, (b) => (b.high + b.low + b.close + b.close) / 4);
+for (const s of [hl2, hlc3, ohlc4, hlcc4]) s.fullHistory = true;
 const SOURCES = { open, high, low, close, volume, hl2, hlc3, ohlc4, hlcc4 };
 function sourceSeries(name) {
   return SOURCES[name] ?? close;
@@ -372,11 +389,11 @@ function plotImpl(series, title, options = {}) {
   const c = collector();
   const id = `plot${c.plotConfig.length}`;
   const perBar2 = Array.isArray(options.color) ? options.color : void 0;
-  const staticColor = typeof options.color === "string" ? options.color : perBar2?.[perBar2.length - 1] ?? "#2962FF";
+  const staticColor2 = typeof options.color === "string" ? options.color : perBar2?.[perBar2.length - 1] ?? "#2962FF";
   c.plotConfig.push({
     id,
     title: title ?? id,
-    color: staticColor,
+    color: staticColor2,
     lineWidth: options.linewidth,
     style: options.style,
     linestyle: options.linestyle,
@@ -440,7 +457,7 @@ function fill(a, b, ...args) {
   const c = collector();
   const third = args[0];
   let perBar2;
-  let staticColor;
+  let staticColor2;
   let title;
   let gradient;
   if (typeof third === "number" || third instanceof import_series.Series) {
@@ -455,12 +472,12 @@ function fill(a, b, ...args) {
   } else if (third !== null && typeof third === "object" && !Array.isArray(third)) {
     const options = third;
     perBar2 = Array.isArray(options.color) ? options.color : void 0;
-    staticColor = typeof options.color === "string" ? options.color : void 0;
+    staticColor2 = typeof options.color === "string" ? options.color : void 0;
     title = options.title;
   } else {
     const color2 = third;
     perBar2 = Array.isArray(color2) ? color2.map((x) => x ?? "") : void 0;
-    staticColor = typeof color2 === "string" ? color2 : void 0;
+    staticColor2 = typeof color2 === "string" ? color2 : void 0;
     title = args[1];
   }
   if (a.kind === "hline" && b.kind === "hline") {
@@ -468,7 +485,7 @@ function fill(a, b, ...args) {
       id: `fill${c.fillConfig.length}`,
       plot1: a.id,
       plot2: b.id,
-      color: staticColor,
+      color: staticColor2,
       colors: perBar2,
       gradient,
       title
@@ -478,11 +495,40 @@ function fill(a, b, ...args) {
   c.fills.push({
     plot1: a.id,
     plot2: b.id,
-    options: { color: staticColor, title },
+    options: { color: staticColor2, title },
     colors: perBar2,
     gradient
   });
 }
+const MARKER_SHAPES = {
+  xcross: "xcross",
+  cross: "cross",
+  triangleup: "triangleUp",
+  triangledown: "triangleDown",
+  flag: "flag",
+  circle: "circle",
+  arrowup: "arrowUp",
+  arrowdown: "arrowDown",
+  labelup: "labelUp",
+  labeldown: "labelDown",
+  square: "square",
+  diamond: "diamond"
+};
+function markerPosition(location, shape) {
+  switch (location) {
+    case "abovebar":
+      return "aboveBar";
+    case "belowbar":
+      return "belowBar";
+    case "top":
+      return "top";
+    case "bottom":
+      return "bottom";
+    case "absolute":
+      return shape === "labeldown" ? "atPriceTop" : shape === "labelup" ? "atPriceBottom" : "atPriceMiddle";
+  }
+}
+const MARKER_COLOR = "#2962FF";
 function emitMarkers(kind, condition, title, options) {
   const c = collector();
   const id = `${kind}${c.shapeConfig.filter((s) => s.kind === kind).length}`;
@@ -503,6 +549,19 @@ function emitMarkers(kind, condition, title, options) {
     offset: options.offset,
     forceOverlay: options.force_overlay
   });
+  const shapeStyle = kind === "shape" ? style : void 0;
+  const template = {
+    position: markerPosition(location, shapeStyle),
+    shape: shapeStyle ? MARKER_SHAPES[shapeStyle] : "circle",
+    color: options.color ?? MARKER_COLOR,
+    id
+  };
+  if (char !== void 0) template.char = char;
+  if (options.text !== void 0) template.text = options.text;
+  if (options.textcolor !== void 0) template.textColor = options.textcolor;
+  if (options.size !== void 0) template.size = options.size;
+  if (options.tooltip !== void 0) template.tooltip = options.tooltip;
+  if (options.force_overlay) template.forceOverlay = true;
   const values = condition.toArray();
   const b = bars();
   const off = options.offset ?? 0;
@@ -511,19 +570,8 @@ function emitMarkers(kind, condition, title, options) {
     if (v === void 0 || Number.isNaN(v) || v === 0) continue;
     const j = i + off;
     if (j < 0 || j >= b.length) continue;
-    c.markers.push({
-      time: b[j].time,
-      id,
-      location,
-      style,
-      char,
-      color: options.color,
-      text: options.text,
-      textcolor: options.textcolor,
-      size: options.size,
-      tooltip: options.tooltip,
-      price: location === "absolute" ? v : void 0
-    });
+    const time2 = b[j].time;
+    c.markers.push(location === "absolute" ? { time: time2, ...template, price: v } : { time: time2, ...template });
   }
 }
 function plotshape(condition, title, options = {}) {
@@ -568,13 +616,14 @@ function emitBarColors(kind, colors, options) {
   c.barColorConfig.push({ id, kind, title: options.title, offset: options.offset, forceOverlay: options.force_overlay });
   const b = bars();
   const off = options.offset ?? 0;
-  const target = kind === "bgcolor" ? c.bgcolors : c.barcolors;
+  const target = kind === "bgcolor" ? c.bgColors : c.barColors;
+  const forceOverlay = kind === "bgcolor" && options.force_overlay === true;
   for (let i = 0; i < b.length; i++) {
     const col = typeof colors === "string" ? colors : colors[i];
     if (!col) continue;
     const j = i + off;
     if (j < 0 || j >= b.length) continue;
-    target.push({ time: b[j].time, color: col });
+    target.push(forceOverlay ? { time: b[j].time, color: col, forceOverlay } : { time: b[j].time, color: col });
   }
 }
 function bgcolor(colors, options = {}) {
@@ -583,11 +632,78 @@ function bgcolor(colors, options = {}) {
 function barcolor(colors, options = {}) {
   emitBarColors("barcolor", colors, options);
 }
+function candleColor(col, i) {
+  if (col === void 0) return void 0;
+  if (typeof col === "string") return col;
+  return col[i] ?? "transparent";
+}
+function ohlcRows(o, h, l, cl) {
+  const [ov, hv, lv, cv] = [o.toArray(), h.toArray(), l.toArray(), cl.toArray()];
+  const rows = [];
+  for (let i = 0; i < bars().length; i++) {
+    const [open2, high2, low2, close2] = [ov[i], hv[i], lv[i], cv[i]];
+    if (Number.isFinite(open2) && Number.isFinite(high2) && Number.isFinite(low2) && Number.isFinite(close2)) {
+      rows.push({ i, open: open2, high: high2, low: low2, close: close2 });
+    }
+  }
+  return rows;
+}
+const staticColor = (col) => typeof col === "string" ? col : void 0;
+function plotcandle(open2, high2, low2, close2, title, options = {}) {
+  const c = collector();
+  const id = `plotcandle${c.candleConfig.filter((x) => x.kind === "candle").length}`;
+  c.candleConfig.push({
+    id,
+    kind: "candle",
+    title,
+    color: staticColor(options.color),
+    wickColor: staticColor(options.wickcolor),
+    borderColor: staticColor(options.bordercolor),
+    display: options.display,
+    forceOverlay: options.force_overlay
+  });
+  const b = bars();
+  c.plotCandles[id] = ohlcRows(open2, high2, low2, close2).map(({ i, ...ohlc }) => {
+    const candle = { time: b[i].time, ...ohlc };
+    const color2 = candleColor(options.color, i);
+    const wickColor = candleColor(options.wickcolor, i);
+    const borderColor = candleColor(options.bordercolor, i);
+    if (color2 !== void 0) candle.color = color2;
+    if (wickColor !== void 0) candle.wickColor = wickColor;
+    if (borderColor !== void 0) candle.borderColor = borderColor;
+    if (options.force_overlay) candle.forceOverlay = true;
+    return candle;
+  });
+}
+function plotbar(open2, high2, low2, close2, title, options = {}) {
+  const c = collector();
+  const id = `plotbar${c.candleConfig.filter((x) => x.kind === "bar").length}`;
+  c.candleConfig.push({
+    id,
+    kind: "bar",
+    title,
+    color: staticColor(options.color),
+    display: options.display,
+    forceOverlay: options.force_overlay
+  });
+  const b = bars();
+  c.plotBars[id] = ohlcRows(open2, high2, low2, close2).map(({ i, ...ohlc }) => {
+    const bar = { time: b[i].time, ...ohlc };
+    const color2 = candleColor(options.color, i);
+    if (color2 !== void 0) bar.color = color2;
+    if (options.force_overlay) bar.forceOverlay = true;
+    return bar;
+  });
+}
+const HISTORY_START = 242;
+const HISTORY_GROWTH = 1.618034;
 function eachBar(fn) {
   collector();
   const b = bars();
   const out = new Array(b.length).fill(NaN);
   const cache = /* @__PURE__ */ new Map();
+  const buffers = /* @__PURE__ */ new Map();
+  let pendingError = null;
   let idx = 0;
   const ctx2 = {
     get i() {
@@ -614,6 +730,13 @@ function eachBar(fn) {
     },
     get(src, offset = 0) {
       if (offset < 0) throw new RangeError(`c.get(): the offset cannot be negative (${offset} would read a future bar).`);
+      if (!src.fullHistory && offset > 0) {
+        const k = Math.floor(offset);
+        if (k > import_series.MAX_BARS_BACK) throw new Error((0, import_series.historyError)(idx, k));
+        let buffer = buffers.get(src);
+        if (!buffer) buffers.set(src, buffer = { size: HISTORY_START, previous: 0, barMax: 0 });
+        if (k > buffer.barMax) buffer.barMax = k;
+      }
       let vals = cache.get(src);
       if (!vals) {
         vals = src.toArray();
@@ -629,9 +752,20 @@ function eachBar(fn) {
   };
   try {
     for (idx = 0; idx < b.length; idx++) {
+      if (pendingError && pendingError.bar === idx) throw new Error((0, import_series.historyError)(idx, pendingError.barsBack));
       currentBar = idx;
       const r = fn(ctx2);
       out[idx] = typeof r === "number" ? r : typeof r === "boolean" ? r ? 1 : 0 : NaN;
+      for (const buffer of buffers.values()) {
+        const k = buffer.barMax;
+        if (k > buffer.size) {
+          const next = k - buffer.previous > 1 ? k : Math.round(buffer.size * HISTORY_GROWTH);
+          if (next > import_series.MAX_BARS_BACK && !pendingError) pendingError = { bar: idx + 1, barsBack: next };
+          buffer.size = next;
+        }
+        buffer.previous = k;
+        buffer.barMax = 0;
+      }
     }
   } finally {
     currentBar = -1;
@@ -900,6 +1034,8 @@ function inSession(time2, session2, timezone) {
 }
 const str = {
   ...strCore,
+  // format.mintick rounds to the chart symbol's tick (syminfo.mintick)
+  tostring: ((value, format, mintick) => strCore.tostring(value, format, format === "mintick" ? mintick ?? syminfo.mintick : mintick)),
   format_time(time2, format = "yyyy-MM-dd'T'HH:mm:ssZ", timezone) {
     return strCore.format_time(time2, format, timezone ?? exchangeTimezone());
   },
@@ -933,6 +1069,11 @@ function calendarFor(session2, timezone) {
   const { bars: bars2 } = sessionBars();
   if (!session2 && !timezone) return { bars: bars2, test: false };
   const tz = timezone ?? exchangeTimezone();
+  if (session2 !== void 0 && (0, import_session.parsePineSession)(session2) === null) {
+    const name = session2.trim();
+    const symbol = name === "regular" && chartCtx.regularSession ? chartCtx.regularSession : chartCtx.session;
+    return { bars: new import_bars.SessionBars(new import_calendar.TradingCalendar(tz, { session: toSpec(symbol).session }, "symbol"), bars2.chart.period), test: true };
+  }
   const spec = session2 ? { session: session2 } : { session: toSpec(chartCtx.session).session };
   return { bars: new import_bars.SessionBars(new import_calendar.TradingCalendar(tz, spec, session2 ? "pine" : "symbol"), bars2.chart.period), test: true };
 }
@@ -1139,6 +1280,12 @@ const polyline = {
     return polylineCore.all();
   }
 };
+const table = {
+  ...tableCore,
+  get all() {
+    return tableCore.all();
+  }
+};
 const text = { ...textCore };
 const chart = {
   point: {