211 lines
6.7 KiB
Python
211 lines
6.7 KiB
Python
|
from __future__ import print_function
|
||
|
|
||
|
from metakernel import MetaKernel
|
||
|
|
||
|
import sys
|
||
|
import signal
|
||
|
import yap
|
||
|
import yapex
|
||
|
|
||
|
import ipywidgets as widgets
|
||
|
|
||
|
|
||
|
def eprint(*args, **kwargs):
|
||
|
print(*args, file=sys.stderr, **kwargs)
|
||
|
|
||
|
|
||
|
class MetaKernelyap(MetaKernel):
|
||
|
implementation = 'MetaKernel YAP'
|
||
|
implementation_version = '1.0'
|
||
|
language = 'text'
|
||
|
language_version = '0.1'
|
||
|
banner = "MetaKernel YAP"
|
||
|
language_info = {
|
||
|
'mimetype': 'text/prolog',
|
||
|
'name': 'text',
|
||
|
# ------ If different from 'language':
|
||
|
'codemirror_mode': {
|
||
|
"version": 2,
|
||
|
"name": "prolog"
|
||
|
},
|
||
|
'pygments_lexer': 'prolog',
|
||
|
'version' : "0.0.1",
|
||
|
'file_extension': '.yap',
|
||
|
'help_links': MetaKernel.help_links,
|
||
|
}
|
||
|
|
||
|
def __init__(self, **kwargs):
|
||
|
MetaKernel.__init__(self, **kwargs)
|
||
|
self._start_yap(**kwargs)
|
||
|
self.olines = ""
|
||
|
self.ask = True
|
||
|
|
||
|
def _start_yap(self, **kwargs):
|
||
|
# Signal handlers are inherited by forked processes, and we can't easily
|
||
|
# reset it from the subprocess. Since kernelapp ignores SIGINT except in
|
||
|
# message handlers, we need to temporarily reset the SIGINT handler here
|
||
|
# so that yap and its children are interruptible.
|
||
|
sig = signal.signal(signal.SIGINT, signal.SIG_DFL)
|
||
|
try:
|
||
|
self.engine = yap.YAPEngine()
|
||
|
self.q = None
|
||
|
self.engine.query("load_files(library(python), [])").command()
|
||
|
banner = "YAP {0} Kernel".format(self.engine.version())
|
||
|
self.olines = banner
|
||
|
finally:
|
||
|
signal.signal(signal.SIGINT, sig)
|
||
|
|
||
|
# Register Yap function to write image data to temporary file
|
||
|
#self.yapwrapper.run_command(image_setup_cmd)
|
||
|
|
||
|
def get_usage(self):
|
||
|
return "This is the YAP kernel."
|
||
|
|
||
|
def query_prolog(self, s):
|
||
|
|
||
|
if not self.q:
|
||
|
self.q = self.engine.query(s)
|
||
|
if self.q.next():
|
||
|
vs = self.q.namedVarsCopy()
|
||
|
wrote = False
|
||
|
if vs:
|
||
|
i = 0
|
||
|
for eq in vs:
|
||
|
name = eq[0]
|
||
|
bind = eq[1]
|
||
|
if bind.isVar():
|
||
|
var = yap.YAPAtom('$VAR')
|
||
|
f = yap.YAPFunctor(var, 1)
|
||
|
bind.unify(yap.YAPApplTerm(f, (name)))
|
||
|
else:
|
||
|
i = bind.numberVars(i, True)
|
||
|
print(name.text() + " = " + bind.text())
|
||
|
wrote = True
|
||
|
print("yes")
|
||
|
if self.q.deterministic():
|
||
|
self.closeq()
|
||
|
return
|
||
|
print("No (more) answers")
|
||
|
self.closeq()
|
||
|
return
|
||
|
|
||
|
def closeq( self):
|
||
|
if self.q:
|
||
|
self.q.close()
|
||
|
self.q = None
|
||
|
|
||
|
def do_execute_direct(self, code):
|
||
|
if not code.strip():
|
||
|
return ""
|
||
|
lines = code.split("\n")
|
||
|
interrupted = False
|
||
|
self.doReset = True
|
||
|
nlines = ""
|
||
|
try:
|
||
|
for line in lines:
|
||
|
line = line.strip()
|
||
|
if line.startswith('#'):
|
||
|
# wait
|
||
|
print( "comment")
|
||
|
elif line.startswith('%'):
|
||
|
# wait
|
||
|
call_magic( line )
|
||
|
elif line.endswith(';'):
|
||
|
nlines += line.rstrip(';').rstrip()
|
||
|
self.doReset = False
|
||
|
break
|
||
|
elif line.endswith('!'):
|
||
|
nlines += line.rstrip('!').rstrip()
|
||
|
self.ask = False
|
||
|
self.doReset = False
|
||
|
break
|
||
|
else:
|
||
|
line = line.rstrip()
|
||
|
if line:
|
||
|
nlines += line + "\n"
|
||
|
if nlines != self.olines:
|
||
|
self.closeq( )
|
||
|
self.olines = nlines
|
||
|
elif self.doReset:
|
||
|
opt = widgets.ToggleButtons(
|
||
|
description='Query Solutions:',
|
||
|
options=['First', 'Next', 'All'],
|
||
|
)
|
||
|
print( opt )
|
||
|
if opt == 'First':
|
||
|
self.closeq( )
|
||
|
elif opt == 'Next':
|
||
|
self.doReset = False
|
||
|
else:
|
||
|
self.ask = False
|
||
|
self.doReset = False
|
||
|
self.query_prolog( nlines )
|
||
|
while not self.ask and self.q:
|
||
|
self.query_prolog( nlines )
|
||
|
|
||
|
except SyntaxError as err:
|
||
|
print("Syntax Error error: {0}".format(err))
|
||
|
except EOFError:
|
||
|
return
|
||
|
except RuntimeError as err:
|
||
|
print("YAP Execution Error: {0}".format(err))
|
||
|
except ValueError:
|
||
|
print("Could not convert data to an integer.")
|
||
|
except KeyboardInterrupt:
|
||
|
return 'stopped by user'
|
||
|
except:
|
||
|
print("Unexpected error:", sys.exc_info()[0])
|
||
|
raise
|
||
|
|
||
|
def do_complete(self, code, cursor_pos):
|
||
|
eprint( code, " -- ", str(cursor_pos) )
|
||
|
# code = code[:cursor_pos]
|
||
|
# default = {'matches': [], 'cursor_start': 0,
|
||
|
# 'cursor_end': cursor_pos, 'metadata': dict(),
|
||
|
# 'status': 'ok'}
|
||
|
|
||
|
# if not code or code[-1] == ' ':
|
||
|
# return default
|
||
|
|
||
|
# tokens = code.replace(';', ' ').split()
|
||
|
# if not tokens:
|
||
|
# return default
|
||
|
|
||
|
# matches = []
|
||
|
# token = tokens[-1]
|
||
|
# start = cursor_pos - len(token)
|
||
|
|
||
|
# if token[0] == '$':
|
||
|
# # complete variables
|
||
|
# cmd = 'compgen -A arrayvar -A export -A variable %s' % token[1:] # strip leading $
|
||
|
# output = self.bashwrapper.run_command(cmd).rstrip()
|
||
|
# completions = set(output.split())
|
||
|
# # append matches including leading $
|
||
|
# matches.extend(['$'+c for c in completions])
|
||
|
# else:
|
||
|
# # complete functions and builtins
|
||
|
# cmd = 'compgen -cdfa %s' % token
|
||
|
# output = self.bashwrapper.run_command(cmd).rstrip()
|
||
|
# matches.extend(output.split())
|
||
|
|
||
|
# if not matches:
|
||
|
# return default
|
||
|
# matches = [m for m in matches if m.startswith(token)]
|
||
|
|
||
|
# return {'matches': sorted(matches), 'cursor_start': start,
|
||
|
# 'cursor_end': cursor_pos, 'metadata': dict(),
|
||
|
# 'status': 'ok'}
|
||
|
|
||
|
|
||
|
|
||
|
|
||
|
def repr(self, data):
|
||
|
return repr(data)
|
||
|
|
||
|
if __name__ == '__main__':
|
||
|
try:
|
||
|
from ipykernel.kernelapp import IPKernelApp
|
||
|
except ImportError:
|
||
|
from jupyter_client.zmq.kernelapp import IPKernelApp
|
||
|
IPKernelApp.launch_instance(kernel_class=MetaKernelyap)
|