233 lines
11 KiB
C#
233 lines
11 KiB
C#
using Age.Engine.Diagnostics;
|
||
using Age.Engine.Hosting;
|
||
using Age.Engine.Model;
|
||
namespace Age.Engine.Vm;
|
||
|
||
public sealed class VirtualMachine
|
||
{
|
||
private const long NoJump = 0xFFFFFFFF;
|
||
private const int HALT = int.MinValue;
|
||
private const int FRAME_RETURN = int.MinValue + 1;
|
||
private const int T_IMM = 0, T_STR = 2, T_GINT = 3, T_GFLOAT = 4, T_GSTR = 5, T_GPTR = 6,
|
||
T_LINT = 9, T_LFLOAT = 10, T_LSTR = 11, T_LPTR = 12;
|
||
|
||
private readonly Script _s;
|
||
private readonly OpcodeTable _t;
|
||
private readonly IHost _host;
|
||
private readonly VmOptions _o;
|
||
private readonly IScriptProvider? _provider;
|
||
private ExecFrame _cur = null!;
|
||
private int _depth;
|
||
private readonly ITraceSink _sink;
|
||
public long CallScriptDispatches { get; private set; }
|
||
|
||
public Dictionary<int, long> Globals { get; } = new();
|
||
public Dictionary<int, string> GlobalStrings { get; } = new();
|
||
public List<(int Offset, string Text, string Script)> Emitted { get; } = new();
|
||
public string? HaltReason { get; private set; }
|
||
public long Steps { get; private set; }
|
||
|
||
public VirtualMachine(Script s, OpcodeTable t, IHost host, VmOptions? o = null,
|
||
IScriptProvider? provider = null, ITraceSink? sink = null)
|
||
{ _s = s; _t = t; _host = host; _o = o ?? new VmOptions(); _provider = provider;
|
||
_sink = sink ?? NullTraceSink.Instance; }
|
||
|
||
private static long Gi(Dictionary<int, long> d, int k) => d.TryGetValue(k, out var v) ? v : 0;
|
||
private static string Gs(Dictionary<int, string> d, int k) => d.TryGetValue(k, out var v) ? v : "";
|
||
private static long PyDiv(long a, long b) { if (b == 0) return 0; long q = a / b, r = a % b; if (r != 0 && (r < 0) != (b < 0)) q--; return q; }
|
||
private static long PyMod(long a, long b) { if (b == 0) return 0; long r = a % b; if (r != 0 && (r < 0) != (b < 0)) r += b; return r; }
|
||
|
||
private static bool IsStr(Operand o) => o.Type == T_STR || o.Type == T_GSTR || o.Type == T_LSTR;
|
||
|
||
private long Read(Operand op) => op.Type switch
|
||
{
|
||
T_IMM => op.Value,
|
||
T_GINT or T_GFLOAT => Gi(Globals, (int)op.Value),
|
||
T_GPTR => Gi(Globals, (int)Gi(Globals, (int)op.Value)),
|
||
T_LINT => Gi(_cur.Locals.I, (int)op.Value),
|
||
T_LFLOAT => Gi(_cur.Locals.F, (int)op.Value),
|
||
T_LPTR => Gi(Globals, (int)Gi(_cur.Locals.P, (int)op.Value)),
|
||
_ => op.Value,
|
||
};
|
||
|
||
private void Write(Operand op, long val)
|
||
{
|
||
switch (op.Type)
|
||
{
|
||
case T_GINT: case T_GFLOAT: Globals[(int)op.Value] = val; break;
|
||
case T_GPTR: Globals[(int)Gi(Globals, (int)op.Value)] = val; break;
|
||
case T_LINT: _cur.Locals.I[(int)op.Value] = val; break;
|
||
case T_LFLOAT: _cur.Locals.F[(int)op.Value] = val; break;
|
||
case T_LPTR: Globals[(int)Gi(_cur.Locals.P, (int)op.Value)] = val; break;
|
||
}
|
||
}
|
||
|
||
private string ReadStr(Operand op) => op.Type switch
|
||
{
|
||
T_STR => _cur.Script.GetString((int)op.Value),
|
||
T_GSTR => Gs(GlobalStrings, (int)op.Value),
|
||
T_LSTR => Gs(_cur.Locals.S, (int)op.Value),
|
||
_ => "",
|
||
};
|
||
|
||
private void WriteStr(Operand op, string val)
|
||
{
|
||
switch (op.Type)
|
||
{
|
||
case T_GSTR: GlobalStrings[(int)op.Value] = val; break;
|
||
case T_LSTR: _cur.Locals.S[(int)op.Value] = val; break;
|
||
}
|
||
}
|
||
|
||
private long BaseAddr(Operand op) => op.Type switch
|
||
{
|
||
T_IMM or T_GINT or T_GFLOAT or T_GSTR or T_GPTR => op.Value,
|
||
T_LINT => Gi(_cur.Locals.I, (int)op.Value),
|
||
T_LPTR => Gi(_cur.Locals.P, (int)op.Value),
|
||
_ => op.Value,
|
||
};
|
||
|
||
private void LookupStore(Operand dst, long addr)
|
||
{
|
||
switch (dst.Type)
|
||
{
|
||
case T_LPTR: _cur.Locals.P[(int)dst.Value] = addr; break;
|
||
case T_GPTR: Globals[(int)dst.Value] = addr; break;
|
||
default: Write(dst, Gi(Globals, (int)addr)); break;
|
||
}
|
||
}
|
||
|
||
private enum FrameOutcome { Returned, Halted, RanOff }
|
||
|
||
public void Run(int entryOffset = 0)
|
||
{
|
||
var top = new ExecFrame(_s, _s.IndexByOffset.TryGetValue(entryOffset, out var idx) ? idx : 0);
|
||
var outcome = RunFrame(top, FrameCause.TopScene);
|
||
if (outcome == FrameOutcome.RanOff) HaltReason ??= "pc-out-of-range";
|
||
else if (outcome == FrameOutcome.Returned) HaltReason ??= "exit";
|
||
// Halted: HaltReason already set by the halting op.
|
||
_sink.Emit(TraceEvent.Halt(HaltReason ?? "unknown", Steps));
|
||
}
|
||
|
||
private FrameOutcome RunFrame(ExecFrame frame, FrameCause cause, long callId = 0)
|
||
{
|
||
var prev = _cur; _cur = frame; _depth++;
|
||
_sink.Emit(TraceEvent.FrameEnter(frame.Script.Name, _depth, cause, callId));
|
||
var outcome = FrameOutcome.RanOff;
|
||
int pc = frame.Pc;
|
||
while (pc >= 0 && pc < frame.Script.Instructions.Count)
|
||
{
|
||
if (Steps >= _o.MaxSteps) { HaltReason ??= "STEP-LIMIT"; outcome = FrameOutcome.Halted; break; }
|
||
Steps++;
|
||
if (_sink.TracingSteps) _sink.Emit(TraceEvent.Step(pc, frame.Script.Instructions[pc], _depth));
|
||
int next = Step(frame.Script.Instructions[pc], pc);
|
||
if (next == FRAME_RETURN) { outcome = FrameOutcome.Returned; break; }
|
||
if (next == HALT) { outcome = FrameOutcome.Halted; break; }
|
||
pc = next;
|
||
}
|
||
_sink.Emit(TraceEvent.FrameExit(frame.Script.Name, _depth, outcome.ToString()));
|
||
_cur = prev; _depth--;
|
||
return outcome;
|
||
}
|
||
|
||
private int Step(Instruction ins, int pc)
|
||
{
|
||
int op = ins.Opcode;
|
||
var a = ins.Args;
|
||
switch (_t.Label(op))
|
||
{
|
||
case "add": Write(a[0], Read(a[1]) + Read(a[2])); return pc + 1;
|
||
case "sub": Write(a[0], Read(a[1]) - Read(a[2])); return pc + 1;
|
||
case "mul": Write(a[0], Read(a[1]) * Read(a[2])); return pc + 1;
|
||
case "div": Write(a[0], PyDiv(Read(a[1]), Read(a[2]))); return pc + 1;
|
||
case "mod": Write(a[0], PyMod(Read(a[1]), Read(a[2]))); return pc + 1;
|
||
case "and": Write(a[0], Read(a[1]) & Read(a[2])); return pc + 1;
|
||
case "or": Write(a[0], Read(a[1]) | Read(a[2])); return pc + 1;
|
||
case "sar": Write(a[0], Read(a[1]) >> (int)(Read(a[2]) & 31)); return pc + 1;
|
||
case "shl": Write(a[0], Read(a[1]) << (int)(Read(a[2]) & 31)); return pc + 1;
|
||
case "eq": Write(a[0], Read(a[1]) == Read(a[2]) ? 1 : 0); return pc + 1;
|
||
case "ne": Write(a[0], Read(a[1]) != Read(a[2]) ? 1 : 0); return pc + 1;
|
||
case "lt": Write(a[0], Read(a[1]) < Read(a[2]) ? 1 : 0); return pc + 1;
|
||
case "lte": Write(a[0], Read(a[1]) <= Read(a[2]) ? 1 : 0); return pc + 1;
|
||
case "gr": Write(a[0], Read(a[1]) > Read(a[2]) ? 1 : 0); return pc + 1;
|
||
case "gre": Write(a[0], Read(a[1]) >= Read(a[2]) ? 1 : 0); return pc + 1;
|
||
case "mov":
|
||
case "set-string":
|
||
if (IsStr(a[0]) || IsStr(a[1])) WriteStr(a[0], ReadStr(a[1]));
|
||
else Write(a[0], Read(a[1]));
|
||
return pc + 1;
|
||
case "lookup-array":
|
||
LookupStore(a[0], BaseAddr(a[1]) + Read(a[2])); return pc + 1;
|
||
case "lookup-array-2d":
|
||
LookupStore(a[0], BaseAddr(a[1]) + Read(a[2]) * Read(a[3]) + Read(a[4])); return pc + 1;
|
||
case "bit-set": Write(a[0], Read(a[0]) | Read(a[1])); return pc + 1;
|
||
case "bit-reset": Write(a[0], Read(a[0]) & ~Read(a[1])); return pc + 1;
|
||
case "check-bit": Write(a[0], (Read(a[1]) >> (int)(Read(a[2]) & 31)) & 1); return pc + 1;
|
||
case "copy-to-global": Write(a[0], Read(a[1])); return pc + 1;
|
||
case "jmp": return _cur.Script.IndexByOffset.GetValueOrDefault((int)a[0].Value, pc + 1);
|
||
case "call": _cur.CallStack.Add(pc + 1); return _cur.Script.IndexByOffset.GetValueOrDefault((int)a[0].Value, pc + 1);
|
||
case "ret":
|
||
if (_cur.CallStack.Count > 0) { int r = _cur.CallStack[^1]; _cur.CallStack.RemoveAt(_cur.CallStack.Count - 1); return r; }
|
||
return FRAME_RETURN; // empty intra-call stack => return from the script frame
|
||
case "jcc":
|
||
{
|
||
long tgt = Read(a[0]) != 0 ? a[1].Value : a[2].Value;
|
||
return tgt == NoJump ? pc + 1 : _cur.Script.IndexByOffset.GetValueOrDefault((int)tgt, pc + 1);
|
||
}
|
||
case "exit":
|
||
case "exit-script": return FRAME_RETURN;
|
||
case "call-script":
|
||
{
|
||
long id = a.Count > 0 ? Read(a[0]) : 0;
|
||
CallScriptDispatches++;
|
||
if (_provider == null)
|
||
{
|
||
_sink.Emit(TraceEvent.CallScript(id, null)); // stub mode: notify only, no child pushed
|
||
return pc + 1;
|
||
}
|
||
if (_depth >= _o.CallDepthCap) { HaltReason ??= "call-depth-exceeded"; return HALT; }
|
||
var child = _provider.GetById(id);
|
||
_sink.Emit(TraceEvent.CallScript(id, child?.Name));
|
||
if (child == null) { HaltReason ??= $"callscript-unresolved:0x{id:x}"; return HALT; }
|
||
var entry = child.IndexByOffset.TryGetValue(0, out var ci) ? ci : 0;
|
||
var outcome = RunFrame(new ExecFrame(child, entry), FrameCause.CallScript, id);
|
||
if (outcome == FrameOutcome.Halted) return HALT; // propagate whole-VM halt up
|
||
return pc + 1; // Returned / RanOff: resume caller
|
||
}
|
||
case "show-text":
|
||
foreach (var o in a)
|
||
{
|
||
if (o.Type != T_STR) continue;
|
||
int off = (int)o.Value;
|
||
_cur.EmitSeen.TryGetValue(off, out var c); c++; _cur.EmitSeen[off] = c;
|
||
if (c > _o.EmitCap) { HaltReason = $"LOOP:line@0x{off:x}×{c}"; return HALT; }
|
||
string text = _cur.Script.GetString(off);
|
||
Emitted.Add((off, text, _cur.Script.Name));
|
||
_host.ShowText(off, text);
|
||
}
|
||
return pc + 1;
|
||
case "wait-for-input": _host.WaitForInput(); return pc + 1;
|
||
case "end-text-line": case "set-font":
|
||
case "comment": case "display-furigana": case "dev_ukn":
|
||
return pc + 1;
|
||
case "create-texture":
|
||
_host.CreateTexture((int)Read(a[0]), (int)Read(a[1]), (int)Read(a[2])); return pc + 1;
|
||
case "set-texture":
|
||
_host.SetTexture(Read(a[0]), (int)Read(a[1])); return pc + 1;
|
||
case "draw-texture": // (handle, slot, srcX, srcY, w, h, dstX, dstY)
|
||
_host.DrawTexture((int)Read(a[1]), (int)Read(a[2]), (int)Read(a[3]), (int)Read(a[4]),
|
||
(int)Read(a[5]), (int)Read(a[6]), (int)Read(a[7])); return pc + 1;
|
||
case "get-texture-size": // 0x208 (slot) (out_w) (out_h)
|
||
{
|
||
var (gw, gh) = _host.GetTextureSize((int)Read(a[0]));
|
||
Write(a[1], gw); Write(a[2], gh);
|
||
return pc + 1;
|
||
}
|
||
case "play-bgm": _host.PlayBgm(Read(a[0])); return pc + 1;
|
||
case "play-voice": _host.PlayVoice(Read(a[0])); return pc + 1;
|
||
default:
|
||
_sink.Emit(TraceEvent.Stub(op, pc)); return pc + 1;
|
||
}
|
||
}
|
||
}
|