X-Git-Url: https://thelambdalab.xyz/gitweb/index.cgi?p=jars.git;a=blobdiff_plain;f=mars.scm;h=caccfa17c17225a774171547e878c45d1efa7976;hp=57e3e13e3f77809ca51d491c4682e1b5afcc0282;hb=80272dcf66d5e333302fa310c4c346d6af5d503c;hpb=c30656b58f082be264e382d0da5866c4c04a10f7

diff --git a/mars.scm b/mars.scm
index 57e3e13..caccfa1 100644
--- a/mars.scm
+++ b/mars.scm
@@ -2,62 +2,369 @@
 ;;; An implementation of the Memory Array Redcode Simulator (MARS)
 ;;;
 
-;;; Constants
+(import (chicken io)
+        (chicken string)
+        (chicken random)
+        (chicken condition)
+        (chicken process-context)
+        matchable wish)
+
+
+;;; Instructions
 ;;
 
-(define core-size 8000)
-(define max-steps 10000)
+(define (make-instr opcode modifier A-mode A-num B-mode B-num)
+  (lambda args
+    (match args
+      (('opcode) opcode)
+      (('modifier) modifier)
+      (('A-mode) A-mode)
+      (('A-num) A-num)
+      (('B-mode) B-mode)
+      (('B-num) B-num)
+      (('make-copy) (make-instr opcode modifier A-mode A-num B-mode B-num))
+      (('set! 'opcode x) (set! opcode x))
+      (('set! 'modifier x) (set! modifier x))
+      (('set! 'A-mode x) (set! A-mode x))
+      (('set! 'A-num x) (set! A-num x))
+      (('set! 'B-mode x) (set! B-mode x))
+      (('set! 'B-num x) (set! B-num x))
+      (('set-from! other)
+       (set! opcode (other 'opcode))
+       (set! modifier (other 'modifier))
+       (set! A-mode (other 'A-mode))
+       (set! A-num (other 'A-num))
+       (set! B-mode (other 'B-mode))
+       (set! B-num (other 'B-num)))
+      (('->string)
+       (conc opcode
+             "." modifier
+             " " (mode->string A-mode) A-num
+             " " (mode->string B-mode) B-num))
+      (else
+       (error "Invalid instr arguments" args)))))
 
+(define (mode->string mode)
+  (case mode
+    ((immediate) "#")
+    ((direct) "$")
+    ((indirect-A) "*")
+    ((indirect-B) "@")
+    ((pre-indirect-A) "{")
+    ((pre-indirect-B) "<")
+    ((post-indirect-A) "}")
+    ((post-indirect-B) ">")
+    (else
+     (error "Unknown mode."))))
 
-;;; Memory setup
+(define initial-instruction
+  (make-instr 'DAT 'F 'immediate 0 'immediate 0))
+
+
+;;; Memory setup and addressing
 ;;
 
-(define core (make-vector core-size '()))
+(define (make-core core-size initial-instr . set-functions)
+  (let ((core-vec (make-vector core-size '()))
+        (names-vec (make-vector core-size '())))
+    (define (norm-addr i)
+      (if (< i 0)
+          (norm-addr (+ i core-size))
+          (modulo i core-size)))
+    (define (norm-ref v i)
+      (vector-ref v (norm-addr i)))
+    (define (norm-set! v i x)
+      (vector-set! v (norm-addr i)
+                   (if (integer? x)
+                       (norm-addr x)
+                       x)))
+    (define (run-set-functions i n)
+      (let loop ((remaining-fns set-functions))
+        (unless (null? remaining-fns)
+          ((car remaining-fns) i n))))
+    (let loop ((i 0))
+      (unless (>= i core-size)
+        (vector-set! core-vec i (initial-instr 'make-copy))
+        (loop (+ i 1))))
+    (lambda args
+      (match args
+        ((i 'set-from! j n)
+         ((norm-ref core-vec i) 'set-from! (norm-ref core-vec j))
+         (norm-set! names-vec i n)
+         (run-set-functions i n))
+        ((i 'set-from-instr! instr n)
+         ((norm-ref core-vec i) 'set-from! instr)
+         (norm-set! names-vec i n)
+         (run-set-functions i n))
+        ((i 'set! v x n)
+         ((norm-ref core-vec i) 'set! v x)
+         (norm-set! names-vec i n)
+         (run-set-functions i n))
+        ((i 'name) (norm-ref names-vec i))
+        (((? integer? i) v) ((norm-ref core-vec i) v))
+        (('->addr (? integer? i)) (norm-addr i))
+        (('dump)
+         (let loop ((i 0))
+           (unless (>= i core-size)
+             (print* i ":\t" ((vector-ref core-vec i) '->string))
+             (let ((n (vector-ref names-vec i)))
+               (unless (null? n)
+                 (print* "\t;" n)))
+             (print)
+             (loop (+ i 1)))))
+        (('size) core-size)))))
 
 
-;;; Instructions
+;;; Programmes and task queues
 ;;
 
-(define (make-instr opcode modifier addrA modeA addrB modeB))
+(define (make-prog name instrs offset)
+  (list name instrs offset))
 
+(define (prog-name prog) (list-ref prog 0))
+(define (prog-instrs prog) (list-ref prog 1))
+(define (prog-offset prog) (list-ref prog 2))
 
-;;; Players
-;;
+(define (install-prog core prog addr)
+  (let loop ((ptr addr)
+             (instrs (prog-instrs prog)))
+    (unless (null? instrs)
+      (core ptr 'set-from-instr! (car instrs) (prog-name prog))
+      (loop (core '->addr (+ ptr 1)) (cdr instrs))))
+  (make-queue (prog-name prog)
+              (core '->addr (+ addr (prog-offset prog)))))
 
-(define (make-player name . ptrs)
-  (cons name ptrs))
+(define (can-install-prog? core prog-len addr)
+  (let loop ((ptr addr)
+             (remaining prog-len))
+    (if (= remaining 0)
+        #t
+        (if (null? (core ptr 'name))
+            (loop (core '->addr (+ ptr 1))
+                  (- remaining 1))
+            #f))))
 
-(define (player-ptrs player)
-  (cdr player))
+(define (install-progs core progs)
+  (let loop ((queues '())
+             (progs-left progs))
+    (if (null? progs-left)
+        queues
+        (let ((addr (pseudo-random-integer (core 'size)))
+              (prog (car progs-left)))
+          (if (can-install-prog? core (length (prog-instrs prog)) addr)
+              (loop (cons (install-prog core prog addr) queues)
+                    (cdr progs-left))
+              (loop queues progs-left))))))
 
-(define (player-name player)
-  (car player))
+(define (make-queue name ptr)
+  (list name ptr))
 
-(define (player-set-ptrs! player ptrs)
-  (set-cdr! player ptrs))
+(define (queue-owner queue) (car queue))
+(define (queue-ptrs queue) (cdr queue))
 
+(define (queue-set-ptrs! queue ptrs)
+  (set-cdr! queue ptrs))
 
-;;; Main loop
-;;
 
-(define (list-rot l)
-  (if (> (length l) 1)
-      (append (cdr l) (list (car l)))
-      l))
+;;; Executive function
+;;
 
-(define (run playerA playerB step)
+(define (run core queues steps-left)
   (cond
-   ((> step max-steps) 'tie)
-   ((null? (player-ptrs playerA) playerB))
-   ((null? (player-ptrs playerB) playerA))
+   ((<= steps-left 0) queues)      ;Tie between remaining players
+   ;; ((<= (length queues) 1) queues) ;There's only one player left who thus wins
    (else
-    (let ((ptrs (player-ptrs playerA)))
-      (player-set-ptrs! (append (cdr ptrs)
-                                (execute-instr (car ptrs)))))
-    (run playerB player A (+ step 1)))))
-
-(define (execute-instr ptr)
-  (let ((instr (vector-ref core ptr)))
-    (match instr
-      ())))
-               
+    (let* ((queue (car queues))
+           (remaining-queues (cdr queues))
+           (ptrs (queue-ptrs queue))
+           (new-ptrs (execute-instr core (car ptrs) (queue-owner queue))))
+      (if (null? new-ptrs)
+          (run remaining-queues (- steps-left 1))
+          (begin
+            (queue-set-ptrs! queue (append (cdr ptrs) new-ptrs))
+            (run core (append remaining-queues (list queue)) (- steps-left 1))))))))
+
+(define (execute-instr core ptr name)
+  (let* ((A-ptr (eval-operand core (core ptr 'A-mode) (core ptr 'A-num) ptr name))
+         (B-ptr (eval-operand core (core ptr 'B-mode) (core ptr 'B-num) ptr name))
+         (modifier (core ptr 'modifier)))
+    (case (core ptr 'opcode)
+      ((DAT)
+       '()) ;Game over, man, game over!
+      ((MOV)
+       (if (eq? modifier 'I)
+           (core B-ptr 'set-from! A-ptr name)
+           (combine-and-store core A-ptr B-ptr modifier name (lambda (x y) y)))
+       (list (core '->addr (+ ptr 1))))
+      ((ADD)
+       (combine-and-store core A-ptr B-ptr modifier name +)
+       (list (core '->addr (+ ptr 1))))
+      ((SUB)
+       (combine-and-store core A-ptr B-ptr modifier name -)
+       (list (core '->addr (+ ptr 1))))
+      ((MUL)
+       (combine-and-store core A-ptr B-ptr modifier name *)
+       (list (core '->addr (+ ptr 1))))
+      ((DIV)
+       (condition-case 
+           (begin
+             (combine-and-store core A-ptr B-ptr modifier name quotient)
+                                
+             (list (core '->addr (+ ptr 1))))
+         ((exn arithmetic) '())))
+      ((MOD)
+       (condition-case
+           (begin
+             (combine-and-store core A-ptr B-ptr modifier name modulo)
+             (list (core '->addr (+ ptr 1))))
+         ((exn arithmetic) '())))
+      ((JMP)
+       (list (core '->addr (+ ptr (core A-ptr 'A-num)))))
+      ((JMZ)
+       (list (core '->addr (+ ptr (if (instr-zero? B-ptr modifier #f name)
+                                      (core A-ptr 'A-num)
+                                      1)))))
+      ((JMN)
+       (list (core '->addr (+ ptr (if (not (instr-zero? B-ptr modifier #f name))
+                                      (core A-ptr 'A-num)
+                                      1)))))
+      ((DJN)
+       (list (core '->addr (+ ptr (if (not (instr-zero? B-ptr modifier #t name))
+                                      (core A-ptr 'A-num)
+                                      1)))))
+      ((SEQ CMP)
+       (list (core '->addr (+ ptr (if (compare-instrs core A-ptr B-ptr modifier =) 2 1)))))
+      ((SNE)
+       (list (core '->addr (+ ptr (if (compare-instrs core A-ptr B-ptr modifier =) 1 2)))))
+      ((SLT)
+       (list (core '->addr (+ ptr (if (compare-instrs core A-ptr B-ptr modifier <) 2 1)))))
+      ((SPL)
+       (list (core '->addr (+ ptr 1) (core '->addr (+ ptr (core A-ptr 'A-num))))))
+      ((NOP)
+       (list (core '->addr (+ ptr 1))))
+      (else
+       (error "Unrecognised opcode" (instr 'opcode))))))
+
+(define (compare-instrs core A-ptr B-ptr modifier test)
+  (case modifier
+    ((A) (test (core A-ptr 'A-num) (core B-ptr 'A-num)))
+    ((B) (test (core A-ptr 'B-num) (core B-ptr 'B-num)))
+    ((AB) (test (core A-ptr 'A-num) (core B-ptr 'B-num)))
+    ((BA) (test (core A-ptr 'B-num) (core B-ptr 'A-num)))
+    ((F) (and
+          (test (core A-ptr 'A-num) (core B-ptr 'A-num))
+          (test (core A-ptr 'B-num) (core B-ptr 'B-num))))
+    ((X) (and
+          (test (core A-ptr 'A-num) (core B-ptr 'B-num))
+          (test (core A-ptr 'B-num) (core B-ptr 'A-num))))
+    ((I) (and
+          (if (eq? test =)
+              (and
+               (eq? (core A-ptr 'opcode) (core B-ptr 'opcode))
+               (eq? (core A-ptr 'modifier) (core B-ptr 'modifier))
+               (eq? (core A-ptr 'A-mode) (core B-ptr 'B-mode))
+               (eq? (core A-ptr 'B-mode) (core B-ptr 'A-mode)))
+              #t)
+          (test (core A-ptr 'A-num) (core B-ptr 'B-num))
+          (test (core A-ptr 'B-num) (core B-ptr 'A-num))))))
+
+(define (instr-zero? core ptr modifier decrement name)
+  (case modifier
+    ((A AB)
+     (if decrement (core ptr 'set! 'A-num (- (core ptr 'A-num) 1) name))
+     (= 0 (core ptr 'A-num)))
+    ((A AB)
+     (if decrement (core ptr 'set! 'B-num (- (core ptr 'B-num) 1) name))
+     (= 0 (core ptr 'B-num)))
+    ((X I F)
+     (if decrement
+         (begin
+           (core ptr 'set! 'A-num (- (core ptr 'A-num) 1) name)
+           (core ptr 'set! 'B-num (- (core ptr 'B-num) 1) name)))
+     (and (= 0 (core ptr 'A-num))
+          (= 0 (core ptr 'B-num))))))
+
+(define (combine-and-store core A-ptr B-ptr modifier name f)
+  (case modifier
+    ((A) (core B-ptr 'set! 'A-num
+               (f (core B-ptr 'A-num) (core A-ptr 'A-num)) name))
+    ((B) (core B-ptr 'set! 'B-num
+               (f (core B-ptr 'B-num) (core A-ptr 'B-num)) name))
+    ((AB) (core B-ptr 'set! 'B-num
+                (f (core B-ptr 'B-num) (core A-ptr 'A-num)) name))
+    ((BA) (core B-ptr 'set! 'A-num
+                (f (core B-ptr 'A-num) (core A-ptr 'B-num)) name))
+    ((F I) (core B-ptr 'set! 'A-num
+                 (f (core B-ptr 'A-num) (core A-ptr 'A-num)) name)
+           (core B-ptr 'set! 'B-num
+                 (f (core B-ptr 'B-num) (core A-ptr 'B-num)) name))
+    ((X) (core B-ptr 'set! 'A-num
+               (f (core B-ptr 'A-num) (core A-ptr 'B-num)) name)
+         (core B-ptr 'set! 'B-num
+               (f (core B-ptr 'B-num) (core A-ptr 'A-num)) name))))
+
+(define (eval-operand core mode num ptr name)
+  (core '->addr (+ ptr
+                   (case mode
+                     ((immediate) 0)
+                     ((direct) num)
+                     ((indirect-A) (+ num (core (+ ptr num) 'A-num)))
+                     ((indirect-B) (+ num (core (+ ptr num) 'B-num)))
+                     ((pre-indirect-A)
+                      (let ((aux-ptr (+ ptr num)))
+                        (core aux-ptr 'set! 'A-num (- (core aux-ptr 'A-num) 1) name)
+                        (+ num (core aux-ptr 'A-num))))
+                     ((pre-indirect-B)
+                      (let ((aux-ptr (+ ptr num)))
+                        (core aux-ptr 'set! 'B-num (- (core aux-ptr 'B-num) 1) name)
+                        (+ num (core aux-ptr 'B-num))))
+                     ((post-indirect-A)
+                      (let* ((aux-ptr (+ ptr num))
+                             (old-A-num (core aux-ptr 'A-num)))
+                        (core aux-ptr 'set! 'A-num (+ (core aux-ptr 'A-num) 1) name)
+                        (+ num old-A-num)))
+                     ((post-indirect-B)
+                      (let* ((aux-ptr (+ ptr num))
+                             (old-B-num (core aux-ptr 'B-num)))
+                        (core aux-ptr 'set! 'B-num (+ (core aux-ptr 'B-num) 1) name)
+                        (+ num old-B-num)))
+                     (else
+                      (error "Unrecognized mode" mode))))))
+
+;;; Main procedure
+;; 
+
+;;; TEST CODE
+
+(define addressing-test
+  (make-prog 'at (list
+                   (make-instr 'DAT 'F 'immediate 42 'immediate 53)
+                   (make-instr 'DAT 'F 'immediate 123 'immediate 256)
+                   (make-instr 'MOV 'A 'indirect-B 4 'direct 7)
+                   (make-instr 'NOP 'I 'immediate 0 'immediate 0)
+                   (make-instr 'NOP 'I 'immediate 0 'immediate 0)
+                   (make-instr 'NOP 'I 'immediate 0 'immediate 0)
+                   (make-instr 'DAT 'F 'immediate -5 'immediate -6)) 2))
+
+(define imp
+  (make-prog 'imp (list (make-instr 'MOV 'I 'direct 0 'direct 1)) 0))
+
+(define dwarf
+  (make-prog 'dwarf (list
+                     (make-instr 'DAT 'F 'immediate 0 'immediate -1)
+                     (make-instr 'ADD 'AB 'immediate 5 'direct -1)
+                     (make-instr 'MOV 'I 'direct -2 'indirect-B -2)
+                     (make-instr 'JMP 'I 'immediate -2 'immediate 0)) 1))
+
+(define w (make-wish 640 480))
+(define colours '((imp . "red")
+                  (dwarf . "blue")))
+
+(define core (make-core (* 640 480) (make-instr 'DAT 'F 'immediate 0 'immediate 0)
+                        (lambda (i n)
+                          (set-wish-pixel w
+                                          (remainder i 640)
+                                          (quotient i 640)
+                                          (cdr (assoc n colours))))))
+(define queues (install-progs core (list dwarf imp)))
+