mu/shell/environment.mu

528 lines
26 KiB
Forth

# The top-level data structure for the Mu shell.
#
# vim:textwidth&
# It would be nice for tests to use a narrower screen than the standard 0x80 of
# 1024 pixels with 8px-wide graphemes. But it complicates rendering logic to
# make width configurable, so we just use longer lines than usual.
type environment {
globals: global-table
sandbox: sandbox
partial-function-name: (handle gap-buffer)
cursor-in-globals?: boolean
cursor-in-function-modal?: boolean
}
# Here's a sample usage session and what it will look like on the screen.
fn test-environment {
var env-storage: environment
var env/esi: (addr environment) <- address env-storage
initialize-environment env
# setup: screen
var screen-on-stack: screen
var screen/edi: (addr screen) <- address screen-on-stack
initialize-screen screen, 0x80/width=72, 0x10/height, 0/no-pixel-graphics
#
type-into-repl env, screen, "(+ 3 4)" # we don't have any global definitions here, so no macros
# | global function definitions | sandbox
# top row blank for now
check-screen-row screen, 0/y, " ", "F - test-environment/0"
check-screen-row screen, 1/y, " screen: ", "F - test-environment/1"
# starting at the same screen row, render the fake screen that exists within the sandbox within env
check-background-color-in-screen-row screen, 0/bg, 1/y, " ........ ", "F - test-environment/1-2"
check-background-color-in-screen-row screen, 0/bg, 2/y, " ........ ", "F - test-environment/2"
check-background-color-in-screen-row screen, 0/bg, 3/y, " ........ ", "F - test-environment/3"
check-screen-row screen, 4/y, " ", "F - test-environment/4"
check-screen-row screen, 5/y, " keyboard: ", "F - test-environment/5"
check-background-color-in-screen-row screen, 0/bg, 5/y, " ................ ", "F - test-environment/5-2"
check-screen-row screen, 6/y, " ", "F - test-environment/6"
check-screen-row screen, 7/y, " (+ 3 4) ", "F - test-environment/7"
check-screen-row screen, 8/y, " ... trace depth: 4 ", "F - test-environment/8"
check-screen-row screen, 9/y, " => 7 ", "F - test-environment/9"
check-screen-row screen, 0xa/y, " ", "F - test-environment/0xa"
check-screen-row screen, 0xb/y, " ", "F - test-environment/0xb"
check-screen-row screen, 0xc/y, " ", "F - test-environment/0xc"
check-screen-row screen, 0xd/y, " ", "F - test-environment/0xd"
check-screen-row screen, 0xe/y, " ", "F - test-environment/0xe"
# bottom row is for a wordstar-style menu
check-screen-row screen, 0xf/y, " ^r run main ^s run sandbox ^g go to ^m to trace ^a << ^b <word ^f word> ^e >> ", "F - test-environment/0xf"
}
fn test-definition-in-environment {
var env-storage: environment
var env/esi: (addr environment) <- address env-storage
initialize-environment env
# setup: screen
var screen-on-stack: screen
var screen/edi: (addr screen) <- address screen-on-stack
initialize-screen screen, 0x80/width=72, 0x10/height, 0/no-pixel-graphics
# define a function on the right (sandbox) side
type-into-repl env, screen, "(define fn1 (fn() 42))" # we don't have any global definitions here, so no macros
# | global function definitions | sandbox
check-screen-row screen, 0/y, " ", "F - test-definition-in-environment/0"
# function definition is now on the left side
check-screen-row screen, 1/y, " (define fn1 (fn() 42)) screen: ", "F - test-definition-in-environment/1"
check-background-color-in-screen-row screen, 0/bg, 1/y, " ........ ", "F - test-definition-in-environment/1-2"
check-background-color-in-screen-row screen, 0/bg, 2/y, " ........ ", "F - test-definition-in-environment/2"
check-background-color-in-screen-row screen, 0/bg, 3/y, " ........ ", "F - test-definition-in-environment/3"
check-screen-row screen, 4/y, " ", "F - test-definition-in-environment/4"
check-screen-row screen, 5/y, " keyboard: ", "F - test-definition-in-environment/5"
check-background-color-in-screen-row screen, 0/bg, 5/y, " ................ ", "F - test-definition-in-environment/5-2"
check-screen-row screen, 6/y, " ", "F - test-definition-in-environment/6"
check-screen-row screen, 7/y, " ", "F - test-definition-in-environment/7"
# you can still see the trace on the right for what you just added to the left
check-screen-row screen, 8/y, " ... trace depth: 4 ", "F - test-definition-in-environment/8"
}
# helper for testing
fn type-into-repl self: (addr environment), screen: (addr screen), keys: (addr array byte) {
# clear the buffer
edit-environment self, 0x15/ctrl-u, 0/no-disk
render-environment screen, self
# type in all the keys
var input-stream-storage: (stream byte 0x40/capacity)
var input-stream/ecx: (addr stream byte) <- address input-stream-storage
write input-stream, keys
{
var done?/eax: boolean <- stream-empty? input-stream
compare done?, 0/false
break-if-!=
var key/eax: grapheme <- read-grapheme input-stream
edit-environment self, key, 0/no-disk
render-environment screen, self
loop
}
# submit
edit-environment self, 0x13/ctrl-s, 0/no-disk
render-environment screen, self
}
## interface
fn initialize-environment _self: (addr environment) {
var self/esi: (addr environment) <- copy _self
var globals/eax: (addr global-table) <- get self, globals
initialize-globals globals
var sandbox/eax: (addr sandbox) <- get self, sandbox
initialize-sandbox sandbox, 1/with-screen
var partial-function-name-ah/eax: (addr handle gap-buffer) <- get self, partial-function-name
allocate partial-function-name-ah
var partial-function-name/eax: (addr gap-buffer) <- lookup *partial-function-name-ah
initialize-gap-buffer partial-function-name, 0x40/function-name-capacity
}
fn render-environment screen: (addr screen), _self: (addr environment) {
# globals layout: 1 char padding, 41 code, 1 padding, 41 code, 1 padding = 85
# sandbox layout: 1 padding, 41 code, 1 padding = 43
# total = 128 chars
var self/esi: (addr environment) <- copy _self
var cursor-in-globals-a/eax: (addr boolean) <- get self, cursor-in-globals?
var cursor-in-globals?/eax: boolean <- copy *cursor-in-globals-a
var globals/ecx: (addr global-table) <- get self, globals
render-globals screen, globals, cursor-in-globals?
var sandbox/edx: (addr sandbox) <- get self, sandbox
var cursor-in-sandbox?/ebx: boolean <- copy 1/true
cursor-in-sandbox? <- subtract cursor-in-globals?
render-sandbox screen, sandbox, 0x55/sandbox-left-margin, 0/sandbox-top-margin, 0x80/screen-width, 0x2f/screen-height-without-menu, cursor-in-sandbox?
# modal if necessary
{
var cursor-in-function-modal-a/eax: (addr boolean) <- get self, cursor-in-function-modal?
compare *cursor-in-function-modal-a, 0/false
break-if-=
render-function-modal screen, self
render-function-modal-menu screen, self
return
}
# render menu
{
var cursor-in-globals?/eax: (addr boolean) <- get self, cursor-in-globals?
compare *cursor-in-globals?, 0/false
break-if-=
render-globals-menu screen, globals
return
}
render-sandbox-menu screen, sandbox
}
fn edit-environment _self: (addr environment), key: grapheme, data-disk: (addr disk) {
var self/esi: (addr environment) <- copy _self
var globals/edi: (addr global-table) <- get self, globals
var sandbox/ecx: (addr sandbox) <- get self, sandbox
# ctrl-r
# Assumption: 'real-screen' and 'real-keyboard' are 0
{
compare key, 0x12/ctrl-r
break-if-!=
var tmp/eax: (addr handle cell) <- copy 0
var nil: (handle cell)
tmp <- address nil
allocate-pair tmp
# (main real-screen real-keyboard)
var real-keyboard: (handle cell)
tmp <- address real-keyboard
allocate-keyboard tmp
# args = cons(real-keyboard, nil)
var args: (handle cell)
tmp <- address args
new-pair tmp, real-keyboard, nil
#
var real-screen: (handle cell)
tmp <- address real-screen
allocate-screen tmp
# args = cons(real-screen, args)
tmp <- address args
new-pair tmp, real-screen, *tmp
#
var main: (handle cell)
tmp <- address main
new-symbol tmp, "main"
# args = cons(main, args)
tmp <- address args
new-pair tmp, main, *tmp
# clear real screen
clear-screen 0/screen
set-cursor-position 0/screen, 0, 0
# run
var out: (handle cell)
var out-ah/ecx: (addr handle cell) <- address out
var trace-storage: trace
var trace/ebx: (addr trace) <- address trace-storage
initialize-trace trace, 1/only-errors, 0x10/capacity, 0/visible
evaluate tmp, out-ah, nil, globals, trace, 0/no-fake-screen, 0/no-fake-keyboard, 0/call-number
# wait for a keypress
{
var tmp/eax: byte <- read-key 0/keyboard
compare tmp, 0
loop-if-=
}
#
return
}
# ctrl-s: send multiple places
{
compare key, 0x13/ctrl-s
break-if-!=
{
# cursor in function modal? do nothing
var cursor-in-function-modal-a/eax: (addr boolean) <- get self, cursor-in-function-modal?
compare *cursor-in-function-modal-a, 0/false
break-if-!=
{
# cursor in globals? update current definition
var cursor-in-globals-a/edx: (addr boolean) <- get self, cursor-in-globals?
compare *cursor-in-globals-a, 0/false
break-if-=
edit-globals globals, key
}
# update sandbox whether the cursor is in globals or sandbox
edit-sandbox sandbox, key, globals, data-disk
}
return
}
# ctrl-g: go to a function (or the repl)
{
compare key, 7/ctrl-g
break-if-!=
var cursor-in-function-modal-a/eax: (addr boolean) <- get self, cursor-in-function-modal?
compare *cursor-in-function-modal-a, 0/false
break-if-!=
# look for a word to prepopulate the modal
var current-word-storage: (stream byte 0x40)
var current-word/edi: (addr stream byte) <- address current-word-storage
word-at-cursor self, current-word
var partial-function-name-ah/eax: (addr handle gap-buffer) <- get self, partial-function-name
var partial-function-name/eax: (addr gap-buffer) <- lookup *partial-function-name-ah
clear-gap-buffer partial-function-name
load-gap-buffer-from-stream partial-function-name, current-word
# enable the modal
var cursor-in-function-modal-a/eax: (addr boolean) <- get self, cursor-in-function-modal?
copy-to *cursor-in-function-modal-a, 1/true
return
}
# dispatch to function modal if necessary
{
var cursor-in-function-modal-a/eax: (addr boolean) <- get self, cursor-in-function-modal?
compare *cursor-in-function-modal-a, 0/false
break-if-=
# nested events for modal dialog
# ignore spaces
{
compare key, 0x20/space
break-if-!=
return
}
# esc = exit modal dialog
{
compare key, 0x1b/escape
break-if-!=
var cursor-in-function-modal-a/eax: (addr boolean) <- get self, cursor-in-function-modal?
copy-to *cursor-in-function-modal-a, 0/false
return
}
# enter = switch to function name and exit modal dialog
{
compare key, 0xa/newline
break-if-!=
# done with function modal
var cursor-in-function-modal-a/eax: (addr boolean) <- get self, cursor-in-function-modal?
copy-to *cursor-in-function-modal-a, 0/false
# if no function name typed in, switch to sandbox
var partial-function-name-ah/eax: (addr handle gap-buffer) <- get self, partial-function-name
var partial-function-name/eax: (addr gap-buffer) <- lookup *partial-function-name-ah
{
var empty?/eax: boolean <- gap-buffer-empty? partial-function-name
compare empty?, 0/false
break-if-=
var cursor-in-globals-a/eax: (addr boolean) <- get self, cursor-in-globals?
copy-to *cursor-in-globals-a, 0/false
return
}
# turn function name into a stream
var name-storage: (stream byte 0x40)
var name/ecx: (addr stream byte) <- address name-storage
emit-gap-buffer partial-function-name, name
clear-gap-buffer partial-function-name
# compute function index
var index/ecx: int <- find-symbol-in-globals globals, name
# if function not found, return
{
compare index, 0
break-if->=
return
}
# otherwise switch focus to function at index
var globals-cursor-index/eax: (addr int) <- get globals, cursor-index
copy-to *globals-cursor-index, index
var cursor-in-globals-a/ecx: (addr boolean) <- get self, cursor-in-globals?
copy-to *cursor-in-globals-a, 1/true
return
}
# otherwise process like a regular gap-buffer
var partial-function-name-ah/eax: (addr handle gap-buffer) <- get self, partial-function-name
var partial-function-name/eax: (addr gap-buffer) <- lookup *partial-function-name-ah
edit-gap-buffer partial-function-name, key
return
}
# dispatch the key to either sandbox or globals
{
var cursor-in-globals-a/eax: (addr boolean) <- get self, cursor-in-globals?
compare *cursor-in-globals-a, 0/false
break-if-=
edit-globals globals, key
return
}
edit-sandbox sandbox, key, globals, data-disk
}
## details
fn word-at-cursor _self: (addr environment), out: (addr stream byte) {
var self/esi: (addr environment) <- copy _self
var cursor-in-function-modal-a/eax: (addr boolean) <- get self, cursor-in-function-modal?
compare *cursor-in-function-modal-a, 0/false
{
break-if-=
# cursor in function modal
return
}
var cursor-in-globals-a/edx: (addr boolean) <- get self, cursor-in-globals?
compare *cursor-in-globals-a, 0/false
{
break-if-=
# cursor in some function editor
var globals/eax: (addr global-table) <- get self, globals
var cursor-index-addr/ecx: (addr int) <- get globals, cursor-index
var cursor-index/ecx: int <- copy *cursor-index-addr
var globals-data-ah/eax: (addr handle array global) <- get globals, data
var globals-data/eax: (addr array global) <- lookup *globals-data-ah
var cursor-offset/ecx: (offset global) <- compute-offset globals-data, cursor-index
var curr-global/eax: (addr global) <- index globals-data, cursor-offset
var curr-global-data-ah/eax: (addr handle gap-buffer) <- get curr-global, input
var curr-global-data/eax: (addr gap-buffer) <- lookup *curr-global-data-ah
word-at-gap curr-global-data, out
return
}
# cursor in sandbox
var sandbox/ecx: (addr sandbox) <- get self, sandbox
var sandbox-data-ah/eax: (addr handle gap-buffer) <- get sandbox, data
var sandbox-data/eax: (addr gap-buffer) <- lookup *sandbox-data-ah
word-at-gap sandbox-data, out
}
fn render-function-modal screen: (addr screen), _self: (addr environment) {
var self/esi: (addr environment) <- copy _self
var width/eax: int <- copy 0
var height/ecx: int <- copy 0
width, height <- screen-size screen
# xmin = max(0, width/2 - 0x20)
var xmin: int
var tmp/edx: int <- copy width
tmp <- shift-right 1
tmp <- subtract 0x20/half-function-name-capacity
{
compare tmp, 0
break-if->=
tmp <- copy 0
}
copy-to xmin, tmp
# xmax = min(width, width/2 + 0x20)
var xmax: int
tmp <- copy width
tmp <- shift-right 1
tmp <- add 0x20/half-function-name-capacity
{
compare tmp, width
break-if-<=
tmp <- copy width
}
copy-to xmax, tmp
# ymin = height/2 - 2
var ymin: int
tmp <- copy height
tmp <- shift-right 1
tmp <- subtract 2
copy-to ymin, tmp
# ymax = height/2 + 1
var ymax: int
tmp <- add 3
copy-to ymax, tmp
#
clear-rect screen, xmin, ymin, xmax, ymax, 0xf/bg=modal
add-to xmin, 4
set-cursor-position screen, xmin, ymin
draw-text-rightward-from-cursor screen, "go to function (or leave blank to go to REPL)", xmax, 8/fg=dark-grey, 0xf/bg=modal
var partial-function-name-ah/eax: (addr handle gap-buffer) <- get self, partial-function-name
var _partial-function-name/eax: (addr gap-buffer) <- lookup *partial-function-name-ah
var partial-function-name/edx: (addr gap-buffer) <- copy _partial-function-name
subtract-from xmin, 4
add-to ymin 2
var dummy/eax: int <- copy 0
var dummy2/ecx: int <- copy 0
dummy, dummy2 <- render-gap-buffer-wrapping-right-then-down screen, partial-function-name, xmin, ymin, xmax, ymax, 1/always-render-cursor, 0/fg=black, 0xf/bg=modal
}
fn render-function-modal-menu screen: (addr screen), _self: (addr environment) {
var self/esi: (addr environment) <- copy _self
var _width/eax: int <- copy 0
var height/ecx: int <- copy 0
_width, height <- screen-size screen
var width/edx: int <- copy _width
var y/ecx: int <- copy height
y <- decrement
var height/ebx: int <- copy y
height <- increment
clear-rect screen, 0/x, y, width, height, 0xc5/bg=blue-bg
set-cursor-position screen, 0/x, y
draw-text-rightward-from-cursor screen, " ^r ", width, 0/fg, 0x5c/bg=menu-highlight
draw-text-rightward-from-cursor screen, " run main ", width, 7/fg, 0xc5/bg=blue-bg
draw-text-rightward-from-cursor screen, " enter ", width, 0/fg, 0x5c/bg=menu-highlight
draw-text-rightward-from-cursor screen, " submit ", width, 7/fg, 0xc5/bg=blue-bg
draw-text-rightward-from-cursor screen, " esc ", width, 0/fg, 0x5c/bg=menu-highlight
draw-text-rightward-from-cursor screen, " cancel ", width, 7/fg, 0xc5/bg=blue-bg
draw-text-rightward-from-cursor screen, " ^a ", width, 0/fg, 0x5c/bg=menu-highlight
draw-text-rightward-from-cursor screen, " << ", width, 7/fg, 0xc5/bg=blue-bg
draw-text-rightward-from-cursor screen, " ^b ", width, 0/fg, 0x5c/bg=menu-highlight
draw-text-rightward-from-cursor screen, " <word ", width, 7/fg, 0xc5/bg=blue-bg
draw-text-rightward-from-cursor screen, " ^f ", width, 0/fg, 0x5c/bg=menu-highlight
draw-text-rightward-from-cursor screen, " word> ", width, 7/fg, 0xc5/bg=blue-bg
draw-text-rightward-from-cursor screen, " ^e ", width, 0/fg, 0x5c/bg=menu-highlight
draw-text-rightward-from-cursor screen, " >> ", width, 7/fg, 0xc5/bg=blue-bg
}
# Gotcha: some saved state may not load.
fn load-state _self: (addr environment), data-disk: (addr disk) {
var self/esi: (addr environment) <- copy _self
# data-disk -> stream
var s-storage: (stream byte 0x2000) # space for 16/sectors
var s/ebx: (addr stream byte) <- address s-storage
draw-text-wrapping-right-then-down-from-cursor-over-full-screen 0/screen, "loading sectors from data disk", 3/fg, 0/bg
move-cursor-to-left-margin-of-next-line 0/screen
load-sectors data-disk, 0/lba, 0x10/sectors, s
#? draw-stream-wrapping-right-then-down-from-cursor-over-full-screen 0/screen, s, 7/fg, 0xc5/bg=blue-bg
# stream -> gap-buffer (HACK: we temporarily cannibalize the sandbox's gap-buffer)
draw-text-wrapping-right-then-down-from-cursor-over-full-screen 0/screen, "parsing", 3/fg, 0/bg
move-cursor-to-left-margin-of-next-line 0/screen
var sandbox/eax: (addr sandbox) <- get self, sandbox
var data-ah/eax: (addr handle gap-buffer) <- get sandbox, data
var data/eax: (addr gap-buffer) <- lookup *data-ah
load-gap-buffer-from-stream data, s
draw-text-wrapping-right-then-down-from-cursor-over-full-screen 0/screen, " into gap buffer", 3/fg, 0/bg
move-cursor-to-left-margin-of-next-line 0/screen
clear-stream s
# read: gap-buffer -> cell
var initial-root-storage: (handle cell)
var initial-root/ecx: (addr handle cell) <- address initial-root-storage
var trace-storage: trace
var trace/edi: (addr trace) <- address trace-storage
initialize-trace trace, 1/only-errors, 0x10/capacity, 0/visible
read-cell data, initial-root, trace
draw-text-wrapping-right-then-down-from-cursor-over-full-screen 0/screen, " into s-expressions", 3/fg, 0/bg
move-cursor-to-left-margin-of-next-line 0/screen
clear-gap-buffer data
#
{
var initial-root-addr/eax: (addr cell) <- lookup *initial-root
compare initial-root-addr, 0
break-if-!=
return
}
# load globals from assoc(initial-root, 'globals)
var globals-literal-storage: (handle cell)
var globals-literal-ah/eax: (addr handle cell) <- address globals-literal-storage
new-symbol globals-literal-ah, "globals"
var globals-literal/eax: (addr cell) <- lookup *globals-literal-ah
var globals-cell-storage: (handle cell)
var globals-cell-ah/edx: (addr handle cell) <- address globals-cell-storage
clear-trace trace
lookup-symbol globals-literal, globals-cell-ah, *initial-root, 0/no-globals, trace, 0/no-screen, 0/no-keyboard
var globals-cell/eax: (addr cell) <- lookup *globals-cell-ah
{
compare globals-cell, 0
break-if-=
var globals/eax: (addr global-table) <- get self, globals
load-globals globals-cell-ah, globals
}
# sandbox = assoc(initial-root, 'sandbox)
draw-text-wrapping-right-then-down-from-cursor-over-full-screen 0/screen, "loading sandbox", 3/fg, 0/bg
var sandbox-literal-storage: (handle cell)
var sandbox-literal-ah/eax: (addr handle cell) <- address sandbox-literal-storage
new-symbol sandbox-literal-ah, "sandbox"
var sandbox-literal/eax: (addr cell) <- lookup *sandbox-literal-ah
var sandbox-cell-storage: (handle cell)
var sandbox-cell-ah/edx: (addr handle cell) <- address sandbox-cell-storage
clear-trace trace
lookup-symbol sandbox-literal, sandbox-cell-ah, *initial-root, 0/no-globals, trace, 0/no-screen, 0/no-keyboard
var sandbox-cell/eax: (addr cell) <- lookup *sandbox-cell-ah
{
compare sandbox-cell, 0
break-if-=
# print: cell -> stream
clear-trace trace
print-cell sandbox-cell-ah, s, trace
# stream -> gap-buffer
var sandbox/eax: (addr sandbox) <- get self, sandbox
var data-ah/eax: (addr handle gap-buffer) <- get sandbox, data
var data/eax: (addr gap-buffer) <- lookup *data-ah
load-gap-buffer-from-stream data, s
}
}
# Save state as an alist of alists:
# ((globals . ((a . (fn ...))
# ...))
# (sandbox . ...))
fn store-state data-disk: (addr disk), sandbox: (addr sandbox), globals: (addr global-table) {
compare data-disk, 0/no-disk
{
break-if-!=
return
}
var stream-storage: (stream byte 0x2000) # space enough for 16/sectors
var stream/edi: (addr stream byte) <- address stream-storage
write stream, "(\n"
write-globals stream, globals
write-sandbox stream, sandbox
write stream, ")\n"
store-sectors data-disk, 0/lba, 0x10/sectors, stream
}