Miroirs/planckforth
1
0
Fork 0
mirror of https://github.com/nineties/planckforth synced 2024-12-25 21:58:22 +01:00
Code Issues Projects Releases Packages Wiki Activity Actions

speed up python version using struct module

Browse source
This commit is contained in:
Koichi Nakamura 2021-12-29 21:17:00 +09:00
parent 4764a103ab
commit 7701f3d825
1 changed files with 13 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

23
others/planck.py
View file

@ -6,9 +6,9 @@
import os import os
import sys import sys
import operator import operator
import array
import ctypes import ctypes
import platform import platform
from struct import pack, unpack
RUNTIME_NAME = "Python {}".format(platform.python_version()) RUNTIME_NAME = "Python {}".format(platform.python_version())
COPYRIGHT = "Copyright (c) 2021 Koichi Nakamura <koichi@idein.jp>" COPYRIGHT = "Copyright (c) 2021 Koichi Nakamura <koichi@idein.jp>"
@ -38,11 +38,14 @@ def aligned(n):
def align(): def align():
write(HERE_CELL, aligned(read(HERE_CELL))) write(HERE_CELL, aligned(read(HERE_CELL)))
def read(addr, signed=False): def readi(addr):
return int.from_bytes(memory[addr:addr+CELL], 'little', signed=signed) return unpack('<i', memory[addr:addr+CELL])[0]
def read(addr):
return unpack('<I', memory[addr:addr+CELL])[0]
def write(addr, v): def write(addr, v):
memory[addr:addr+CELL] = ctypes.c_uint32(v).value.to_bytes(CELL, 'little') memory[addr:addr+CELL] = pack('<I', ctypes.c_uint32(v).value)
def comma(v, signed=False): def comma(v, signed=False):
here = read(HERE_CELL) here = read(HERE_CELL)
@ -103,7 +106,7 @@ def push(v):
def pop(): def pop():
global sp global sp
v = read(sp, signed=True) v = readi(sp)
sp += CELL sp += CELL
return v return v
@ -114,7 +117,7 @@ def rpush(v):
def rpop(): def rpop():
global rp global rp
v = read(rp, signed=True) v = readi(rp)
rp += CELL rp += CELL
return v return v
@ -190,11 +193,11 @@ def key():
exit(0) exit(0)
add_simple_operator('k', key) add_simple_operator('k', key)
add_simple_operator('t', lambda: sys.stdout.write(chr(pop()))) add_simple_operator('t', lambda: sys.stdout.write(chr(pop())))
add_operator('j', lambda ip,np: next(np + read(np, signed=True))) add_operator('j', lambda ip,np: next(np + readi(np)))
add_operator('J', lambda ip,np: next(np + (CELL if pop() else read(np, signed=True)))) add_operator('J', lambda ip,np: next(np + (CELL if pop() else readi(np))))
add_simple_operator('f', lambda: push(find(chr(pop())))) add_simple_operator('f', lambda: push(find(chr(pop()))))
add_operator('x', lambda ip,np: (pop(), np)) add_operator('x', lambda ip,np: (pop(), np))
add_simple_operator('@', lambda: push(read(pop(), signed=True))) add_simple_operator('@', lambda: push(readi(pop())))
# NB: Python evaluates expressions from left to right # NB: Python evaluates expressions from left to right
# https://docs.python.org/3/reference/expressions.html#evaluation-order # https://docs.python.org/3/reference/expressions.html#evaluation-order
@ -214,7 +217,7 @@ add_simple_operator('R', set_rp)
add_simple_operator('i', lambda: push(DOCOL_ID)) add_simple_operator('i', lambda: push(DOCOL_ID))
add_operator('e', lambda ip,np: next(rpop())) add_operator('e', lambda ip,np: next(rpop()))
def lit(ip, np): def lit(ip, np):
push(read(np, signed=True)) push(readi(np))
return next(np + CELL) return next(np + CELL)
add_operator('L', lit) add_operator('L', lit)
def litstring(ip, np): def litstring(ip, np):