# -*- coding: utf-8 -*-

#SELECT. Interpreter v0.6 by Quintopia

#SELECT. Language by Por Gammer

#TODO: make debug output work with optimized code

try:

from Tkinter import *

except ImportError:

from tkinter import *

import mpmath

import sys,os

import re,random

from collections import deque,defaultdict

from itertools import islice,repeat

import optparse

myname = "SELECT. interpreter v0.6"

parser = optparse.OptionParser(description=myname+" by Quintopia. SELECT. Language by Por Gammer.", usage="%prog [options] <filename>")

parser.add_option('-v', '--verbose', help="Turn on verbose output.", dest="verbose", action="store_true", default=False)

parser.add_option('-d', '--debug', help="Print out all of memory after every cycle.", dest="debug", action="store_true", default=False)

parser.add_option('-m', '--minify', help="Write back a minified version of the program instead of executing it.", dest="minify", action="store_true", default=False)

parser.add_option('-0', '--no_opt', help="Turn off optimization.", dest="optimize", action="store_false", default=True)

parser.add_option('-k', help="Set the tape initialization value.", metavar="<value>", action="store", dest="val", type=float, default=1)

parser.add_option('-p', help="Set the precision of arithmetic.", metavar="<value>", action="store", dest="precision", type=int, default=100)

(options, args) = parser.parse_args()

#open and load file

if len(args)<1:

parser.error('Please specify the filename of the SELECT. program to be run.')

filename = args[0]

sys.stdout.write("Reading file...");sys.stdout.flush()

try:

openfile = open(filename,"r")

except IOError:

print("File '"+filename+"' not found.")

sys.exit()

else:

theFile = openfile.read()

origFile=theFile

openfile.close()

#get Canvas dimensions from file, dike them out, create the canvas

pattern = re.compile(r"""\(\s*(?P<x>[0-9]+?)\s*

,\s*(?P<y>[0-9]+?)\s*

,\s*(?P<z>[0-9]+?)\s*\)""",re.VERBOSE)

match=pattern.match(theFile)

if match is None:

print("Program must begin with (xres,yres,pixsize)")

sys.exit()

x = int(match.group("x"))

y = int(match.group("y"))

z = int(match.group("z"))

if x%2==0 or y%2==0:

print('Width and height must both be odd numbers.')

sys.exit()

w = x*z

h = y*z

centerx = (x+1)/2

centery = (y+1)/2

theFile = theFile.replace(match.group(),"")

sys.stdout.write("done.\n")

#replace all commands with single character for ease of parsing

sys.stdout.write("Compressing...");sys.stdout.flush()

theFile = theFile.replace("\n","")

theFile = re.sub('<<<\$.*%>>>','',theFile)

theFile = theFile.replace('SELECT.','<<<$s%>>>')

theFile = theFile.replace('EXP.','<<<$e%>>>')

theFile = theFile.replace('LOG.','<<<$l%>>>')

theFile = theFile.replace('LEFT.','<<<$<%>>>')

theFile = theFile.replace('RIGHT.','<<<$>%>>>')

theFile = theFile.replace('CONJ.','<<<$*%>>>')

theFile = theFile.replace('LOOP.','<<<$[%>>>')

theFile = theFile.replace('END.','<<<$]%>>>')

theFile = theFile.replace('PRINT.','<<<$.%>>>')

theFile = theFile.replace('GET.','<<<$,%>>>')

theFile = theFile.replace('CLEAR.','<<<$X%>>>')

theFile = theFile.replace('COLOR.','<<<$c%>>>')

theFile = theFile.replace('HALT.','<<<$h%>>>')

#dike out non-commands

theFile = re.sub('%>>>.*?<<<\$','',theFile)

theFile = re.sub('%>>>.*?$','',theFile)

theFile = re.sub('^.*?<<<\$','',theFile)

sys.stdout.write("done.\n")

if options.minify:

with open(filename,'w') as openfile:

sys.stdout.write("Writing golfed file...");sys.stdout.flush()

openfile.write("(%d,%d,%d)"%(x,y,z))

i=0

for c in theFile:

if c=='s': openfile.write("SELECT.")

elif c=='e': openfile.write("EXP.")

elif c=='l': openfile.write("LOG.")

elif c=='<': openfile.write("LEFT.")

elif c=='>': openfile.write("RIGHT.")

elif c=='*': openfile.write("CONJ.")

elif c=='[': openfile.write("LOOP.")

elif c==']': openfile.write("END.")

elif c=='.': openfile.write("PRINT.")

elif c==',': openfile.write("GET.")

elif c=='X': openfile.write("CLEAR.")

elif c=='c': openfile.write("COLOR.")

elif c=='h': openfile.write("HALT.")

i+=1

if i%16==0:openfile.write("\n")

sys.stdout.write("done.\n")

sys.exit(0)

#optimizations

if options.optimize:

newFile = "";i=0

sys.stdout.write("Optimizing");sys.stdout.flush()

re1 = re.compile(r"e(>+)s")

re2 = re.compile(r"e(<+)s")

re3 = re.compile(r"l(>+)s")

re4 = re.compile(r"l(<+)s")

re5 = re.compile(r"<{2,}")

re6 = re.compile(r">{2,}")

numdots = 1

while True:

type=0

m = re1.search(theFile,pos=i)

if m is not None: type = 1

m2 = re2.search(theFile,pos=i)

if m is None or (m2 is not None and m2.start()<m.start()): m=m2;type = 2

m2 = re3.search(theFile,pos=i)

if m is None or (m2 is not None and m2.start()<m.start()): m=m2;type = 3

m2 = re4.search(theFile,pos=i)

if m is None or (m2 is not None and m2.start()<m.start()): m=m2;type = 4

m2 = re5.search(theFile,pos=i)

if m is None or (m2 is not None and m2.start()<m.start()): m=m2;type = 5

m2 = re6.search(theFile,pos=i)

if m is None or (m2 is not None and m2.start()<m.start()): m=m2;type = 6

if m is not None:

newFile += theFile[i:m.start()]

if type == 1:

newFile += "f"+str(len(m.group(1)))

elif type == 2:

newFile += "g"+str(len(m.group(1)))

elif type == 3:

newFile += "m"+str(len(m.group(1)))

elif type == 4:

newFile += "n"+str(len(m.group(1)))

elif type == 5:

newFile += "{"+str(len(m.group(0)))

elif type == 6:

newFile += "}"+str(len(m.group(0)))

else:

assert False

i=m.end()

if i > numdots*len(theFile)/6:

sys.stdout.write(".");sys.stdout.flush()

numdots+=1

else:

theFile = newFile + theFile[i:]

sys.stdout.write("done.\n")

break

#create a Tk instance

sys.stdout.write("Preparing interface...");sys.stdout.flush()

master = Tk()

master.resizable(width = False, height = False)

master.title(myname)

#set precision and default value

#pick a random fill value (very close to 1 to protect against rounding errors when doing things like k^(k^n) )

if options.val==1:

val = mpmath.mpf(random.uniform(1,1.0000001))

else:

val = mpmath.mpmathify(options.val)

mpmath.mp.dps=options.precision

verbose=options.verbose

if verbose:

print("Using tape value: "+mpmath.nstr(val))

if verbose:

print("Width: "+str(w)+" Height: "+str(h))

canv = Canvas(master, width=w, height=h, background="white")

canv.pack()

#waiting for input

waiting = False

#counter for screen update

cycles=0

#lastop: 0 = none, 1 = exp, 2 = log

lastop=0

#index of the location to be modified by exp or log

argindex=0

#the index into the list

listindex=0

#the list data structure

a=defaultdict(repeat(val).next)

#instruction pointer

pointer=0

#the dictionary of pixels on the canvas

pixdict={}

#tolerance for zeroing

tol = int(0.8*mpmath.mp.dps)

tol = mpmath.mpf('1e-'+str(tol))

#colors

red = 0

green = 0

blue = 0

#clickevent queue

eventqueue = deque([])

def click(event):

master.after_idle(main)

#bind click

canv.bind("<Button-1>",click)

canv.bind("<B1-Motion>",click)

#create loop point dictionary

brackstack = []

loopdict = {}

for i in range(len(theFile)):

if theFile[i]=='[':

brackstack.append(i)

elif theFile[i]==']':

if len(brackstack)==0:

print('Found END. unpreceded by corresponding LOOP. at instruction '+str(i))

sys.exit()

b=brackstack.pop()

loopdict[b]=i

if len(brackstack)>0:

print('Found LOOP. unterminated by END. at instruction '+str(brackstack[0]))

sys.exit()

def rgb(red, green, blue):

return '#%02x%02x%02x' % (red, green, blue)

def greporig(num):

return origFile[left+1:thematch.start()]+'***'+thematch.group()+'***'+origFile[thematch.end():right]

#define all commands

def select():

oparg=0

def repexpright():

global a,pointer,listindex

pass

def repexpleft():

global a,pointer,listindex

pass

def expright():

select()

def expleft():

select()

def logright():

select()

def logleft():

select()

def repright():

listindex+=int(num)

def repleft():

listindex-=int(num)

def exp():

return

def log():

global lastop,argindex,listindex

lastop = 2

argindex=listindex

def left():

global listindex,a,val,lastop,argindex

listindex-=1

def right():

global listindex,a,val

listindex+=1

def conj():

global a,listindex

a[listindex]=mpmath.conj(a[listindex])

def color():

blue = int(mpmath.fmod(mpmath.im(a[listindex]),256))

def drawbox():

pass

def deletepix():

eventqueue = deque([]) #if the screen is cleared, we probably don't want the old events hanging around either

def getmouse():

pointer-=1

def loop():

pointer=loopdict[pointer]

def end():

pointer=next(key for key,value in loopdict.items() if value==pointer)

def halt():

global pointer

pointer=len(theFile)

#create command map (switch-case)

commands = {'s': select,

}

def main():

raise

sys.stdout.write("done.\n");sys.stdout.flush()

master.after_idle(main)

try:

mainloop()

except KeyboardInterrupt,SystemExit:

print('Program terminated by user while executing instruction '+str(pointer))

print(greporig(pointer))

quit()

except:

raise