akkartik
/
mu
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

shell primitive: initialize array of some size

This commit is contained in:
Kartik K. Agaram 2021-07-26 00:56:30 -07:00
parent 4c224c5375
commit 3e76e0540b
1 changed files with 78 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

78
shell/primitives.mu
View File

@ -57,6 +57,7 @@ fn initialize-primitives _self: (addr global-table) {
append-primitive self, "rewind"
# for arrays
append-primitive self, "array"
append-primitive self, "populate"
append-primitive self, "index"
append-primitive self, "iset"
# misc
@ -108,6 +109,10 @@ fn render-primitives screen: (addr screen), xmin: int, xmax: int, ymax: int {
y <- increment
set-cursor-position screen, right-min, y
draw-text-wrapping-right-then-down-from-cursor screen, " array index iset len", right-min, y, xmax, ymax, 0x2a/fg=orange, 0xdc/bg=green-bg
y <- increment
var tmpx/eax: int <- copy right-min
tmpx <- draw-text-rightward screen, " populate", tmpx, xmax, y, 0x2a/fg=orange, 0xdc/bg=green-bg
tmpx <- draw-text-rightward screen, ": int _ -> array", tmpx, xmax, y, 7/fg=grey, 0xdc/bg=green-bg
#? {
#? compare screen, 0
#? break-if-!=
@ -560,6 +565,13 @@ fn apply-primitive _f: (addr cell), args-ah: (addr handle cell), out: (addr hand
apply-array args-ah, out, trace
return
}
{
var populate?/eax: boolean <- string-equal? f-name, "populate"
compare populate?, 0/false
break-if-=
apply-populate args-ah, out, trace
return
}
{
var index?/eax: boolean <- string-equal? f-name, "index"
compare index?, 0/false
@ -3505,6 +3517,72 @@ fn apply-array _args-ah: (addr handle cell), _out-ah: (addr handle cell), trace:
}
}
fn apply-populate _args-ah: (addr handle cell), _out-ah: (addr handle cell), trace: (addr trace) {
trace-text trace, "eval", "apply 'populate'"
var args-ah/eax: (addr handle cell) <- copy _args-ah
var _args/eax: (addr cell) <- lookup *args-ah
var args/esi: (addr cell) <- copy _args
{
var args-type/eax: (addr int) <- get args, type
compare *args-type, 0/pair
break-if-=
error trace, "args to 'populate' are not a list"
return
}
var empty-args?/eax: boolean <- nil? args
compare empty-args?, 0/false
{
break-if-=
error trace, "'populate' needs 2 args but got 0"
return
}
# args->left
var first-ah/ecx: (addr handle cell) <- get args, left
# args->right->left
var right-ah/eax: (addr handle cell) <- get args, right
var right/eax: (addr cell) <- lookup *right-ah
{
var right-type/eax: (addr int) <- get right, type
compare *right-type, 0/pair
break-if-=
error trace, "'populate' encountered non-pair"
return
}
{
var nil?/eax: boolean <- nil? right
compare nil?, 0/false
break-if-=
error trace, "'populate' needs 2 args but got 1"
return
}
var second-ah/edx: (addr handle cell) <- get right, left
#
var first/eax: (addr cell) <- lookup *first-ah
{
var first-type/eax: (addr int) <- get first, type
compare *first-type, 1/number
break-if-=
error trace, "first arg for 'populate' is not a number"
return
}
var first-value/eax: (addr float) <- get first, number-data
var capacity/ecx: int <- convert *first-value
var out-ah/edi: (addr handle cell) <- copy _out-ah
new-array out-ah, capacity
var out/eax: (addr cell) <- lookup *out-ah
var data-ah/eax: (addr handle array handle cell) <- get out, array-data
var data/eax: (addr array handle cell) <- lookup *data-ah
var i/ebx: int <- copy 0
{
compare i, capacity
break-if->=
var curr-ah/ecx: (addr handle cell) <- index data, i
copy-object second-ah, curr-ah
i <- increment
loop
}
}
fn apply-index _args-ah: (addr handle cell), out: (addr handle cell), trace: (addr trace) {
trace-text trace, "eval", "apply 'index'"
var args-ah/eax: (addr handle cell) <- copy _args-ah