149 lines
3.8 KiB
Python
149 lines
3.8 KiB
Python
# -*- coding: utf-8 -*-
|
|
#
|
|
# Copyright © 2014, 2015 Simon Forman
|
|
#
|
|
# This file is part of joy.py
|
|
#
|
|
# joy.py is free software: you can redistribute it and/or modify
|
|
# it under the terms of the GNU General Public License as published by
|
|
# the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# joy.py is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU General Public License
|
|
# along with joy.py. If not see <http://www.gnu.org/licenses/>.
|
|
#
|
|
import os, pickle, sys
|
|
from inspect import getdoc
|
|
|
|
from joy.joy import run
|
|
from joy.parser import Symbol
|
|
from joy.utils.stack import stack_to_string
|
|
from joy.utils.polytypes import type_check
|
|
|
|
|
|
def is_numerical(s):
|
|
try:
|
|
float(s)
|
|
except ValueError:
|
|
return False
|
|
return True
|
|
|
|
|
|
class World(object):
|
|
|
|
def __init__(self, stack=(), dictionary=None, text_widget=None):
|
|
self.stack = stack
|
|
self.dictionary = dictionary or {}
|
|
self.text_widget = text_widget
|
|
|
|
def check(self, name):
|
|
return type_check(name, self.stack)
|
|
|
|
def do_lookup(self, name):
|
|
if name in self.dictionary:
|
|
self.stack = (Symbol(name), ()), self.stack
|
|
self.print_stack()
|
|
else:
|
|
assert is_numerical(name)
|
|
self.interpret(name)
|
|
|
|
def do_opendoc(self, name):
|
|
if is_numerical(name):
|
|
print 'The number', name
|
|
else:
|
|
try:
|
|
word = self.dictionary[name]
|
|
except KeyError:
|
|
print repr(name), '???'
|
|
else:
|
|
print getdoc(word)
|
|
self.print_stack()
|
|
|
|
def pop(self):
|
|
if self.stack:
|
|
self.stack = self.stack[1]
|
|
self.print_stack()
|
|
|
|
def push(self, it):
|
|
it = it.encode('utf8')
|
|
self.stack = it, self.stack
|
|
self.print_stack()
|
|
|
|
def peek(self):
|
|
if self.stack:
|
|
return self.stack[0]
|
|
|
|
def interpret(self, command):
|
|
if len(command.split()) == 1 and not is_numerical(command):
|
|
assert self.has(command), repr(command)
|
|
if self.check(command) == False: # not in {True, None}:
|
|
return
|
|
try:
|
|
self.stack, _, self.dictionary = run(
|
|
command,
|
|
self.stack,
|
|
self.dictionary,
|
|
)
|
|
finally:
|
|
self.print_stack()
|
|
|
|
def has(self, name):
|
|
return self.dictionary.has_key(name)
|
|
|
|
def save(self):
|
|
pass
|
|
|
|
def print_stack(self):
|
|
stack_out_index = self.text_widget.search('<' 'STACK', 1.0)
|
|
if stack_out_index:
|
|
self.text_widget.see(stack_out_index)
|
|
s = stack_to_string(self.stack) + '\n'
|
|
self.text_widget.insert(stack_out_index, s)
|
|
|
|
|
|
class StackDisplayWorld(World):
|
|
|
|
|
|
def __init__(self, repo, filename, rel_filename, dictionary=None, text_widget=None):
|
|
self.filename = filename
|
|
stack = self.load_stack() or ()
|
|
World.__init__(self, stack, dictionary, text_widget)
|
|
self.repo = repo
|
|
self.relative_STACK_FN = rel_filename
|
|
|
|
def interpret(self, command):
|
|
if (
|
|
is_numerical(command)
|
|
or len(command.split()) > 1
|
|
or self.has(command)
|
|
and self.check(command) in {True, None}
|
|
):
|
|
print '\njoy?', command
|
|
super(StackDisplayWorld, self).interpret(command)
|
|
|
|
def print_stack(self):
|
|
print '\n%s <-' % stack_to_string(self.stack)
|
|
|
|
def save(self):
|
|
with open(self.filename, 'wb') as f:
|
|
os.chmod(self.filename, 0600)
|
|
pickle.dump(self.stack, f)
|
|
f.flush()
|
|
os.fsync(f.fileno())
|
|
self.repo.stage([self.relative_STACK_FN])
|
|
commit_id = self.repo.do_commit(
|
|
'message',
|
|
committer='Simon Forman <forman.simon@gmail.com>',
|
|
)
|
|
print >> sys.stderr, commit_id
|
|
|
|
def load_stack(self):
|
|
if os.path.exists(self.filename):
|
|
with open(self.filename) as f:
|
|
return pickle.load(f)
|