Skip to content
Draft
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions .jules/bolt.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,3 @@
## 2024-06-19 - O(N) filtering in Backtest runner hot loop
**Learning:** The `backtestRunner.ts` logic loops through historical timeframe points and was using `Array.prototype.filter` sequentially to find the last price closing index up to the current as-of-time (i.e. `bars.filter(b => b.time <= asOf)`). Since financial time-series data is chronologically sorted, filtering the entire historical dataset on every backtest iteration and for every ticker scales at O(N) when it can be done in O(log N).
**Action:** Use the `findLastBarIndex` utility (now exported from `src/data/sources/dnsePublic.ts`) for fast binary search price lookups, instead of traversing arrays linearly with `.filter()`. Avoid doing array allocations and traversals inside tight loops in the backtest runner.
2 changes: 1 addition & 1 deletion package.json
Original file line number Diff line number Diff line change
Expand Up @@ -61,7 +61,7 @@
"ink-text-input": "^6.0.0",
"react": "^18.3.1",
"technicalindicators": "^3.1.0",
"undici": "^6.20.0",
"undici": "^6.27.0",
"yaml": "^2.6.0",
"zod": "^3.23.8"
},
Expand Down
26 changes: 13 additions & 13 deletions pnpm-lock.yaml

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

14 changes: 9 additions & 5 deletions src/agent/backtestRunner.ts
Original file line number Diff line number Diff line change
Expand Up @@ -2,7 +2,7 @@ import { randomUUID } from "node:crypto";
import { loadConfig } from "../config/loader.js";
import { getDb } from "../storage/db.js";
import { getBacktestBroker } from "../broker/index.js";
import { getStockOhlcv, getIndexOhlcv, type Bar } from "../data/sources/dnsePublic.js";
import { getStockOhlcv, getIndexOhlcv, findLastBarIndex, type Bar } from "../data/sources/dnsePublic.js";
import { DISCOVERY_UNIVERSE, discoverTickers } from "../tools/discover.js";
import { setActiveAsOf } from "./clock.js";
import { runTeamAnalysis } from "./team/index.js";
Expand Down Expand Up @@ -298,8 +298,9 @@ export async function runBacktestSession(
);

const vnindexAt = (asOf: number): number | null => {
const series = vnindex.filter((b) => b.time <= asOf);
return series.length ? series[series.length - 1]!.close : null;
// Optimization: O(log n) binary search instead of O(n) array filter
const idx = findLastBarIndex(vnindex, asOf);
return idx !== -1 ? vnindex[idx]!.close : null;
};
const vnindexBaseline = vnindexAt(intervalTurns[0]!);
if (vnindexBaseline == null) throw new Error(`no VNINDEX data at first ${interval.label} turn`);
Expand All @@ -312,8 +313,11 @@ export async function runBacktestSession(
throwIfAborted(cb.signal);
const dateIso = ictLabel(asOf);
const priceOverride = (sym: string): number | null => {
const series = bars[sym]?.filter((b) => b.time <= asOf) ?? [];
return series.length ? series[series.length - 1]!.close : null;
const symBars = bars[sym];
if (!symBars) return null;
// Optimization: O(log n) binary search instead of O(n) array filter
const idx = findLastBarIndex(symBars, asOf);
return idx !== -1 ? symBars[idx]!.close : null;
};
broker.setPriceOverride(priceOverride);
cb.onTurnStart?.({ asOf, dateIso });
Expand Down
27 changes: 26 additions & 1 deletion src/data/sources/dnsePublic.ts
Original file line number Diff line number Diff line change
Expand Up @@ -45,6 +45,27 @@ async function fetchOhlcs(
return (await body.json()) as OhlcvSeries;
}

/**
* O(log n) binary search to find the index of the last bar <= asOf time.
* Expects chronologically sorted bars.
* Returns -1 if no such bar exists.
*/
export function findLastBarIndex(bars: Bar[], asOf: number): number {
let low = 0;
let high = bars.length - 1;
let ans = -1;
while (low <= high) {
const mid = (low + high) >>> 1;
if (bars[mid]!.time <= asOf) {
ans = mid;
low = mid + 1;
} else {
high = mid - 1;
}
}
return ans;
}

export function seriesToBars(s: OhlcvSeries): Bar[] {
const out: Bar[] = [];
for (let i = 0; i < s.t.length; i++) {
Expand All @@ -68,7 +89,11 @@ function clipBars(bars: Bar[]): Bar[] {
asOfClock.getStore()?.asOfSec != null || isAsOfOverridden();
if (!hasOverride) return bars;
const asOf = nowSec();
return bars.filter((b) => b.time <= asOf);
const lastIdx = findLastBarIndex(bars, asOf);
if (lastIdx === -1) return [];
// Optimization: use slice (O(1) shallow copy of array view or O(N) allocation)
// instead of filter (O(N) iteration + allocation) since bars are chronologically sorted.
return bars.slice(0, lastIdx + 1);
}

export async function getStockOhlcv(
Expand Down
Loading