From 260c070f34ea43b3df77d955ffa729cb3a1cf9f2 Mon Sep 17 00:00:00 2001
From: jtuu <joona.tuukkanen@gmail.com>
Date: Wed, 30 Oct 2019 21:02:35 +0200
Subject: [PATCH] [VM] Implemented get_random opcode and added a unit test for
 it.

---
 src/quest_editor/scripting/vm/index.test.ts | 47 +++++++++++++++++++++
 src/quest_editor/scripting/vm/index.ts      | 22 +++++++++-
 src/quest_editor/scripting/vm/windows.ts    | 23 ++++++++++
 3 files changed, 91 insertions(+), 1 deletion(-)
 create mode 100644 src/quest_editor/scripting/vm/windows.ts

diff --git a/src/quest_editor/scripting/vm/index.test.ts b/src/quest_editor/scripting/vm/index.test.ts
index c7d16098..c85ee4e1 100644
--- a/src/quest_editor/scripting/vm/index.test.ts
+++ b/src/quest_editor/scripting/vm/index.test.ts
@@ -1,7 +1,12 @@
+/**
+ * @jest-environment jsdom
+ */
+
 import { VirtualMachine, ExecutionResult } from ".";
 import { VMIOStub } from "./VMIOStub";
 import { to_instructions } from "../../../../test/src/utils";
 import { Segment } from "../instructions";
+import { srand } from "./windows";
 
 test("integer arithmetic opcodes", () => {
     class TestIO extends VMIOStub {
@@ -158,3 +163,45 @@ test("basic window_msg output", () => {
     expect(io.winend).toBeCalledTimes(1);
     expect(io.error).toBeCalledTimes(0);
 });
+
+test("opcode get_random", () => {
+    const result_reg = 102;
+    const obj_code = to_instructions(`
+    .code
+    0:
+        leti r100, 0
+        leti r101, 65535
+        get_random r100, r${result_reg}
+        get_random r100, r${result_reg}
+        get_random r100, r${result_reg}
+        get_random r100, r${result_reg}
+        get_random r100, r${result_reg}
+        get_random r100, r${result_reg}
+        get_random r100, r${result_reg}
+    `);
+
+    const vm = new VirtualMachine();
+    srand(123);
+    vm.load_object_code(obj_code);
+    vm.start_thread(0);
+
+    // run `let`s
+    expect(vm.execute()).toBe(ExecutionResult.Ok);
+    expect(vm.execute()).toBe(ExecutionResult.Ok);
+
+    // test correct get_random sequence
+    expect(vm.execute()).toBe(ExecutionResult.Ok);
+    expect(vm.get_register_unsigned(result_reg)).toBe(879);
+    expect(vm.execute()).toBe(ExecutionResult.Ok);
+    expect(vm.get_register_unsigned(result_reg)).toBe(38105);
+    expect(vm.execute()).toBe(ExecutionResult.Ok);
+    expect(vm.get_register_unsigned(result_reg)).toBe(46149);
+    expect(vm.execute()).toBe(ExecutionResult.Ok);
+    expect(vm.get_register_unsigned(result_reg)).toBe(26207);
+    expect(vm.execute()).toBe(ExecutionResult.Ok);
+    expect(vm.get_register_unsigned(result_reg)).toBe(64725);
+    expect(vm.execute()).toBe(ExecutionResult.Ok);
+    expect(vm.get_register_unsigned(result_reg)).toBe(6529);
+    expect(vm.execute()).toBe(ExecutionResult.Halted);
+    expect(vm.get_register_unsigned(result_reg)).toBe(61497);
+});
diff --git a/src/quest_editor/scripting/vm/index.ts b/src/quest_editor/scripting/vm/index.ts
index c14a8ed2..b7347c7a 100644
--- a/src/quest_editor/scripting/vm/index.ts
+++ b/src/quest_editor/scripting/vm/index.ts
@@ -79,6 +79,7 @@ import {
     OP_LETA,
     OP_FLET,
     OP_FLETI,
+    OP_GET_RANDOM,
 } from "../opcodes";
 import { VirtualMachineMemoryBuffer, VirtualMachineMemory } from "./memory";
 import {
@@ -92,6 +93,7 @@ import {
 } from "./utils";
 import { VirtualMachineIO } from "./io";
 import { VMIOStub } from "./VMIOStub";
+import { rand, srand, GetTickCount } from "./windows";
 
 const REGISTERS_BASE_ADDRESS = 0x00a954b0;
 const REGISTER_COUNT = 256;
@@ -125,7 +127,9 @@ export class VirtualMachine {
     private thread_idx = 0;
     private window_msg_open = false;
 
-    constructor(private io: VirtualMachineIO = new VMIOStub()) {}
+    constructor(private io: VirtualMachineIO = new VMIOStub()) {
+        srand(GetTickCount());
+    }
 
     /**
      * Halts and resets the VM, then loads new object code.
@@ -547,6 +551,22 @@ export class VirtualMachine {
                     this.io.winend();
                 }
                 break;
+            case OP_GET_RANDOM.code:
+                {
+                    const low = this.get_register_signed(arg0);
+                    const hi = this.get_register_signed(arg0 + 1);
+
+                    const r = rand();
+                    let result = Math.floor(Math.fround(r / 32768.0) * hi);
+
+                    // intentional. this is how the game does it.
+                    if (low >= result) {
+                        result = low;
+                    }
+
+                    this.set_register_signed(arg1, result);
+                }
+                break;
             default:
                 throw new Error(`Unsupported instruction: ${inst.opcode.mnemonic}.`);
         }
diff --git a/src/quest_editor/scripting/vm/windows.ts b/src/quest_editor/scripting/vm/windows.ts
new file mode 100644
index 00000000..c09ab273
--- /dev/null
+++ b/src/quest_editor/scripting/vm/windows.ts
@@ -0,0 +1,23 @@
+/**
+ * @file Implementations of some parts of the Win32 API and the MSVCRT C standard library.
+ */
+
+
+var holdrand = 1;
+
+export function srand(seed: number): void {
+    holdrand = seed;
+}
+
+export function rand(): number {
+    const r = (holdrand * 0x343fd + 0x269ec3) >>> 0;
+    holdrand = r;
+    return r >>> 0x10 & 0x7fff;
+}
+
+export function GetTickCount(): number {
+    // GetTickCount returns time elapsed since system start
+    // performance.now returns time elapsed since document load
+    // but both are monotonic so it's probably close enough?
+    return Math.floor(performance.now());
+}