mirror of
https://github.com/Ed94/pikuma_ps1.git
synced 2026-07-12 20:31:25 -07:00
191 lines
11 KiB
Lua
191 lines
11 KiB
Lua
--- word_count_eval.lua — Word-counting logic for the tape-atom metaprogram pipeline.
|
|
---
|
|
--- Three responsibilities:
|
|
--- 1. **Public utilities** (used by `passes/components.lua`, `passes/offsets.lua`, `passes/annotation.lua`):
|
|
--- - `M.count_token_words(token, wc)` — words emitted by one token
|
|
--- - `M.scan_dir(dir, suffix)` — glob walk for *.macs.h
|
|
--- 2. **Pass entry** `M.run(ctx)` — loads metadata.h + *.macs.h into `ctx.shared.word_counts` for downstream passes.
|
|
--- 3. **Internal helpers** for the body scanner.
|
|
---
|
|
--- **Conventions**: tabs (1/level), EmmyLua annotations, no regex,
|
|
--- Lua 5.3 compatible.
|
|
|
|
-- ════════════════════════════════════════════════════════════════════════════
|
|
-- Module-scope requires + package.path setup
|
|
-- ════════════════════════════════════════════════════════════════════════════
|
|
|
|
-- Resolve `arg[0]` to an absolute-ish script directory so that `require("duffle")` resolves against `scripts/` regardless of CWD.
|
|
-- Note: this boilerplate is duplicated in 6 other entry scripts; a Phase-6 extraction target (`duffle.setup_package_path()`).
|
|
-- Bootstrap: see `ps1_meta.lua` for the rationale.
|
|
-- Bootstrap: load `scripts/duffle_paths.lua` (sets package.path + package.cpath).
|
|
-- Uses `debug.getinfo` to find this file's own directory, so it works both standalone and when require'd from the orchestrator.
|
|
-- Bootstrap: load `duffle_paths.lua` via `debug.getinfo(1, "S").source` (works both standalone + when require'd).
|
|
-- duffle_paths.lua sets package.path then returns `require("duffle")` at the bottom, so the dofile value IS the duffle module.
|
|
local _bootstrap_dir = debug.getinfo(1, "S").source:match("^@?(.*[/\\])") or "./"
|
|
local duffle = dofile(_bootstrap_dir .. "../duffle_paths.lua")
|
|
|
|
-- ════════════════════════════════════════════════════════════════════════════
|
|
-- Constants
|
|
-- ════════════════════════════════════════════════════════════════════════════
|
|
|
|
-- Windows separator char — used by `fname:match` to recognize `.macs.h` files.
|
|
local PATH_SEP_BACKSLASH = "\\"
|
|
|
|
-- Required native extension: lfs (LuaFileSystem). Built by `update_deps.ps1` to
|
|
-- `toolchain/lfs/lfs.dll` and wired into package.cpath by `scripts/duffle_paths.lua`.
|
|
-- If lfs is missing, `require` throws — fail loud per the build-tool convention.
|
|
local lfs = require("lfs")
|
|
|
|
-- ════════════════════════════════════════════════════════════════════════════
|
|
-- Type declarations
|
|
-- ════════════════════════════════════════════════════════════════════════════
|
|
|
|
--- @class WordCounts
|
|
--- @field [string] integer -- macro name -> word count
|
|
|
|
--- @class SourceFile
|
|
--- @field path string -- absolute path to the source file
|
|
--- @field text string -- the full source text
|
|
--- @field dir string -- the directory containing the source
|
|
--- @field basename string -- filename without extension
|
|
|
|
--- @class PassCtx
|
|
--- @field sources SourceFile[] -- all source files in the build
|
|
--- @field metadata_path string -- path to word_count.metadata.h
|
|
--- @field shared table -- cross-pass shared state
|
|
--- @field shared.word_counts WordCounts -- populated by this pass
|
|
--- @field out_root string -- output root (e.g. "build/gen")
|
|
--- @field project_root string -- project root (e.g. "code/")
|
|
--- @field upstream table<string, table> -- per-pass upstream outputs
|
|
--- @field flags table -- CLI flags
|
|
--- @field dry_run boolean -- if true, compute but don't write
|
|
--- @field verbose boolean -- if true, log diagnostic info
|
|
|
|
--- @class PassResult
|
|
--- @field outputs table[] -- {kind=, path=} entries describing emit files
|
|
--- @field errors table[] -- {line=, msg=} entries; build-stops
|
|
--- @field warnings table[] -- {line=, msg=} entries; build-succeeds
|
|
|
|
-- ════════════════════════════════════════════════════════════════════════════
|
|
-- Module exports
|
|
-- ════════════════════════════════════════════════════════════════════════════
|
|
|
|
local M = {}
|
|
|
|
-- ┌────────────────────────────────────────────────────────────────────┐
|
|
-- │ Shared utility: count_token_words │
|
|
-- └────────────────────────────────────────────────────────────────────┘
|
|
|
|
--- Count words emitted by a single comma-separated token inside an atom body.
|
|
--- For most tokens (regular MIPS instructions) this returns 1.
|
|
--- For `mac_X(...)` calls, this returns the resolved word count from `wc` (recursively if needed). For `nop2` etc., returns wc[name].
|
|
--- For unknown macros, returns 1 and (optionally) warns.
|
|
---
|
|
--- @param token string -- a single token from split_top_level_commas
|
|
--- @param wc WordCounts -- the shared word-count table
|
|
--- @return integer
|
|
function M.count_token_words(token, wc)
|
|
local s = duffle.trim(token)
|
|
if s == "" then return 0 end
|
|
local name, after = duffle.read_ident(s, 1)
|
|
if not name then return 1 end
|
|
if wc[name] then return wc[name] end
|
|
local paren_pos = duffle.skip_ws_and_cmt(s, after)
|
|
if s:sub(paren_pos, paren_pos) == "(" then
|
|
io.stderr:write(" warning: unknown macro '" .. name .. "', assuming 1 word\n")
|
|
end
|
|
return 1
|
|
end
|
|
|
|
-- ┌────────────────────────────────────────────────────────────────────┐
|
|
-- │ Shared utility: scan_dir │
|
|
-- └────────────────────────────────────────────────────────────────────┘
|
|
|
|
--- Recursively scan a directory for files matching a glob suffix.
|
|
---
|
|
--- The `.macs.h` files produced by the components pass always live at `<project_root>/<module>/gen/`.
|
|
--- Native walk via lfs.attributes + lfs.dir: ~2ms vs ~56ms for the prior `dir /b /s` subprocess.
|
|
---
|
|
--- @param dir string -- directory to scan (absolute or relative)
|
|
--- @param suffix string -- file pattern, e.g. "*.macs.h"
|
|
--- @return string[]
|
|
-- Cache the scan_dir result per (dir, suffix) in package.loaded.
|
|
-- The cache persists for the lifetime of the Lua process (cleared when ps1_meta.lua exits).
|
|
-- If a build removes/creates .macs.h files mid-process, the caller can invalidate by calling `M._invalidate_scan_cache()`.
|
|
local SCAN_CACHE_KEY = "__word_count_eval_scan_cache__"
|
|
|
|
--- Scan `code/` for files matching `suffix` (e.g. `*.macs.h`).
|
|
--- Native directory enumeration via lfs (~2ms). Zero subprocess spawns.
|
|
---
|
|
--- @param dir string -- project root directory
|
|
--- @param suffix string -- file pattern, e.g. "*.macs.h"
|
|
--- @return string[]
|
|
function M.scan_dir(dir, suffix)
|
|
local key = dir .. "\0" .. suffix
|
|
|
|
local cache = package.loaded[SCAN_CACHE_KEY]
|
|
if cache and cache[key] then return cache[key] end
|
|
|
|
local results = {}
|
|
local code_dir = dir .. "/code"
|
|
if lfs.attributes(code_dir, "mode") == "directory" then
|
|
for mod_name in lfs.dir(code_dir) do
|
|
if mod_name ~= "." and mod_name ~= ".." then
|
|
local gen_path = code_dir .. "/" .. mod_name .. "/gen"
|
|
if lfs.attributes(gen_path, "mode") == "directory" then
|
|
for fname in lfs.dir(gen_path) do
|
|
if fname:match("%.macs%.h$") then
|
|
results[#results + 1] = gen_path .. "/" .. fname
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
end
|
|
|
|
-- Cache the result (including empty results).
|
|
cache = cache or {}
|
|
cache[key] = results
|
|
package.loaded[SCAN_CACHE_KEY] = cache
|
|
|
|
return results
|
|
end
|
|
|
|
--- Invalidate the scan cache (call after creating new .macs.h files in the same Lua process — usually not needed).
|
|
function M._invalidate_scan_cache() package.loaded[SCAN_CACHE_KEY] = nil end
|
|
|
|
-- ┌────────────────────────────────────────────────────────────────────┐
|
|
-- │ Pass entry: M.run(ctx) — "word-counts" pass │
|
|
-- └────────────────────────────────────────────────────────────────────┘
|
|
|
|
--- Load metadata.h + scan for existing *.macs.h files into ctx.shared.word_counts.
|
|
--- Loading the .macs.h files is idempotent: entries from later (current-build) .macs.h files override metadata.h entries of the same name.
|
|
---
|
|
--- @param ctx PassCtx
|
|
--- @return PassResult
|
|
function M.run(ctx)
|
|
local wc = {}
|
|
|
|
-- 1. Load metadata.h (the encoding-macro source of truth).
|
|
local meta_counts = duffle.load_word_counts(ctx.metadata_path)
|
|
for name, count in pairs(meta_counts) do wc[name] = count end
|
|
|
|
-- 2. Scan project_root recursively for *.macs.h files (component-macro source).
|
|
local macs_files = M.scan_dir(ctx.project_root, "*.macs.h")
|
|
for _, macs_path in ipairs(macs_files) do
|
|
local ok, mc = pcall(duffle.load_word_counts, macs_path)
|
|
if not ok then
|
|
io.stderr:write(string.format("[word_count_eval] parse error in '%s': %s\n", macs_path, tostring(mc)))
|
|
elseif type(mc) ~= "table" then
|
|
io.stderr:write(string.format("[word_count_eval] '%s' did not return a table (got %s)\n", macs_path, type(mc)))
|
|
else
|
|
for name, count in pairs(mc) do wc[name] = count end
|
|
end
|
|
end
|
|
|
|
ctx.shared.word_counts = wc
|
|
return { outputs = {}, errors = {}, warnings = {} }
|
|
end
|
|
|
|
return M
|