Mercurial > vba-clojure

     1 (ns com.aurellem.gb.gb-driver

     2     (:import com.aurellem.gb.Gb)

     3     (:import java.io.File)

     4     (:import org.apache.commons.io.FileUtils)

     5     (:import (java.nio IntBuffer ByteOrder)))

     6 

     7 ;; Savestates

     8 

     9 (defrecord SaveState [data])

    10 

    11 (def user-home (File. (System/getProperty "user.home")))

    12 

    13 (def ^:dynamic *save-state-cache*

    14   (File. user-home "proj/vba-clojure/save-states/"))

    15 

    16 (def current-state (atom nil))

    17 

    18 (defn state-cache-file [name]

    19   (File. *save-state-cache* (str name ".sav")))

    20 

    21 (defn write-state!

    22   ([^SaveState name]

    23      (write-state! @current-state name))

    24   ([^SaveState save ^String name]

    25      (let [buffer (:data save)

    26            bytes (byte-array (.limit buffer))

    27            dest (state-cache-file name)]

    28        (.get buffer bytes)

    29        (FileUtils/writeByteArrayToFile dest bytes)

    30        (.rewind buffer)

    31        dest)))

    32 

    33 (defn read-state [name]

    34   (let [save (state-cache-file name)]

    35     (assert (.exists save))

    36     (let [buffer (Gb/saveBuffer)

    37           bytes (FileUtils/readFileToByteArray save)]

    38       (.put buffer bytes)

    39       (.flip buffer)

    40       (SaveState. buffer))))

    41 ;;;;;;;;;;;;;;;;

    42 

    43 ;; Gameboy management

    44 (Gb/loadVBA)

    45 

    46 (def yellow-rom-image

    47   (File. user-home "proj/pokemon-escape/roms/yellow.gbc"))

    48 

    49 (def yellow-save-file

    50   (File. user-home "proj/pokemon-escape/roms/yellow.sav"))

    51 

    52 (def on? (atom nil))

    53 

    54 (defn shutdown! [] (Gb/shutdown) (reset! on? false))

    55 

    56 (defn restart! []

    57   (shutdown!)

    58   (.delete yellow-save-file)

    59   (Gb/startEmulator (.getCanonicalPath yellow-rom-image))

    60   (reset! on? true))

    61 

    62 ;;; The first state!

    63 (defn gen-root! []

    64   (restart!)

    65   (let [state (SaveState. (Gb/saveState))]

    66     (write-state! state "root" ) state))

    67 

    68 (defn root []

    69   (if (.exists (state-cache-file "root"))

    70     (read-state "root")

    71     (gen-root!)))

    72 

    73 ;;;; Press Buttons

    74 

    75 (def button-code

    76   {;; main buttons

    77    :a         0x0001

    78    :b         0x0002

    79 

    80    ;; directional pad

    81    :r         0x0010

    82    :l         0x0020

    83    :u         0x0040

    84    :d         0x0080

    85 

    86    ;; meta buttons

    87    :select    0x0004

    88    :start     0x0008

    89 

    90    ;; pseudo-buttons

    91    :restart   0x0800 ; hard reset 

    92    :listen -1        ; listen for user input

    93    })

    94 

    95 (defn button-mask [buttons]

    96   (reduce bit-or 0x0000 (map button-code buttons)))

    97 

    98 (defn set-state! [^SaveState state]

    99   (assert (:data state) "Not a valid state!")

   100   (if (not @on?) (restart!))

   101   (if (not= state @current-state)

   102     (do 

   103       (Gb/loadState (:data state))

   104       (reset! current-state state))))

   105 

   106 (defn update-state []

   107   (reset! current-state

   108           (SaveState. (Gb/saveState))))

   109 

   110 (defn step

   111   ([^SaveState state buttons]

   112      (set-state! state)

   113      (Gb/step (button-mask buttons))

   114      (reset! current-state 

   115              (SaveState. (Gb/saveState))))

   116   ([^SaveState state]

   117      (step state [:listen]))

   118   ([] (step (if @current-state @current-state (root)))))

   119 

   120 (defn tick

   121   ([] (tick @current-state))

   122   ([^SaveState state]

   123      (set-state! state)

   124      (Gb/tick)

   125      (update-state)))

   126 

   127 (defn play

   128   ([^SaveState state n]

   129      (try

   130        (set-state! state)

   131        (dorun (dotimes [_ n]

   132                 (Thread/sleep 1)

   133                 (Gb/step)))

   134        (finally

   135              (update-state))))

   136   ([n]

   137      (play @current-state n)))

   138 

   139 (defn continue!

   140   ([state]

   141      (play state Integer/MAX_VALUE))

   142   ([]

   143     (continue! @current-state)))

   144 

   145 (defn run-moves [state moves]

   146   (set-state! state)

   147   (dorun (map #(Gb/step (button-mask %))

   148               moves))

   149   (update-state))

   150 

   151 (defn play-moves

   152   ([moves [prev state]]

   153      (set-state! state)

   154      (dorun (map (fn [move] (step @current-state move)) moves))

   155      [(concat prev moves) @current-state]))

   156   

   157 ;;;;;;;;;;;

   158 

   159 

   160 ;;;;;;;;;;;;;;; CPU data

   161 

   162 (defn cpu-data [size arr-fn]

   163   (let [store (int-array size)]

   164     (fn get-data

   165       ([] (get-data @current-state))

   166       ([state]

   167          (set-state! state) (arr-fn store) store))))

   168 

   169 (defn write-cpu-data [size store-fn]

   170   (fn store-data

   171     ([state new-data]

   172        (set-state! state)

   173        (let [store (int-array new-data)]

   174          (assert (= size (count new-data)))

   175          (store-fn store)

   176          (update-state)))

   177     ([new-data]

   178        (store-data @current-state new-data))))

   179     

   180 

   181 (def memory

   182   (cpu-data Gb/GB_MEMORY #(Gb/getMemory %)))

   183 

   184 (def ram

   185   (cpu-data Gb/RAM_SIZE #(Gb/getRAM %)))

   186 

   187 (def rom 

   188   (cpu-data Gb/ROM_SIZE #(Gb/getROM %)))

   189 

   190 (def working-ram 

   191   (cpu-data Gb/WRAM_SIZE #(Gb/getWRAM %)))

   192 

   193 (def video-ram 

   194   (cpu-data Gb/VRAM_SIZE #(Gb/getVRAM %)))

   195 

   196 (def registers

   197   (cpu-data Gb/NUM_REGISTERS #(Gb/getRegisters %)))

   198 

   199 (def write-memory!

   200   (write-cpu-data Gb/GB_MEMORY #(Gb/writeMemory %)))

   201 

   202 (def write-registers!

   203   (write-cpu-data Gb/NUM_REGISTERS #(Gb/writeRegisters %)))

   204 

   205 (def write-rom!

   206   (write-cpu-data Gb/ROM_SIZE #(Gb/writeROM %)))

   207 

   208 ;;;;;  Registers  ;;;;;;;;;;;;;;;;;;;;;;;;;;;

   209 

   210 (defmacro gen-get-set-register [name index]

   211   (let [name-bang (symbol (str name "!"))]

   212     `(do

   213        (defn ~name

   214          ~(str "Retrieve the " name " register from state, or "

   215                "from @current-state if state is absent.")

   216          ([state#]

   217             (nth (registers state#) ~index))

   218          ([]

   219             (~name @current-state)))

   220        (defn ~name-bang

   221          ~(str "Set the " name " register for state, or "

   222                "for @current-state if state is absent.")

   223          ([state# new-register#]

   224             (set-state! state#)

   225             (let [registers# (registers state#)]

   226               (aset registers# ~index new-register#)

   227               (Gb/writeRegisters registers#)

   228               (update-state)))

   229          ([new-register#]

   230             (~name-bang @current-state new-register#))))))

   231 

   232 ;; 16 bit registers

   233 (gen-get-set-register PC 0)

   234 (gen-get-set-register SP 1)

   235 (gen-get-set-register AF 2)

   236 (gen-get-set-register BC 3)

   237 (gen-get-set-register DE 4)

   238 (gen-get-set-register HL 5)

   239 (gen-get-set-register IFF 6)

   240 

   241 ;; 8 bit registers

   242 (gen-get-set-register DIV 7)

   243 (gen-get-set-register TIMA 8)

   244 (gen-get-set-register TMA 9)

   245 (gen-get-set-register IF 11)

   246 (gen-get-set-register LCDC 12)

   247 (gen-get-set-register STAT 13)

   248 (gen-get-set-register SCY 14)

   249 (gen-get-set-register SCX 15)

   250 (gen-get-set-register LY 16)

   251 (gen-get-set-register DMA 18)

   252 (gen-get-set-register WY 19)

   253 (gen-get-set-register WX 20)

   254 (gen-get-set-register VBK 21)

   255 (gen-get-set-register HDMA1 22)

   256 (gen-get-set-register HDMA2 23)

   257 (gen-get-set-register HDMA3 24)

   258 (gen-get-set-register HDMA4 25)

   259 (gen-get-set-register HDMA5 26)

   260 (gen-get-set-register SVBK 27)

   261 (gen-get-set-register IE 28)

   262 

   263 ;;;;;;;;;;;;;;;

   264 

   265 (defmacro defn-memo

   266   [& forms]

   267   (let [fun-name (first forms)]

   268     `(do

   269        (defn ~@forms)

   270        (alter-var-root (var ~fun-name) memoize))))

   271 

   272 

   273 

   274 (def original-rom (rom (root)))
author	Robert McIntyre <rlm@mit.edu>
date	Sat, 24 Mar 2012 14:42:34 -0500
parents	d5dddf33543c
children	d263df762c59