c/p.py at master · effbiae/c · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
m=["av;","['/\\",":+-*%!&|<>=~,^#_$?@.",";)]\n "]
c=dict([(chr(x),' ')for x in range(128)])
def f(x,s):
 for a in s:c[a]=x
for i in range(3):f(m[0][i],m[i+1])
def cs(i):
 if(i<len(s)):return c[s[i][0]]
 return ';'
def n():
 global i
 if len(s)!=i:i+=1;return s[i-1]
 return ' ';
def q():return ';'==cs(i)
def E(x):
 while n() in ";[(":
  if type(x)!=tuple:x=(x,)
  x=x+(e(t()),)
 return x
def t():
 if q():return()
 x=n()if'('!=s[i]else x[1]if 3>len(x:=E(s[i]))else x
 while'a'==cs(i):
  x=E(x)if'['==s[i]else(n(),x)
 return x
def verb(x):#primitive or derived verb?
 if type(x)==str and len(x)>0:return c[x[0]]=='v'
 return len(x)>0 and type(x[0])==str and c[x[0]]=='a'
def train(x):#is train? if a verb or projection or composition
 if verb(x):return 1
 if type(x)==tuple:
  if len(x)==3 and x[0]=="'":return 1
  if len(x)==3 and x[2]==():return 1
 return 0
def o3(x):
 if train(x[2]):return("'",(x[0],x[1],()),x[2])
 return x
def monad(x):
 if type(x)==str and c[x[0]]=='v':
  return x+':'
 return x
def o2(x):
 if train(x[1]):return("'",monad(x[0]),x[1])
 return x
def v():return';'<cs(i-1)
def e(x):
 if q():return x
 v_=v();f=t();return o3((f,x,e(t())))if v()>v_ else o2((x,e(f)))
def lex(x):
 r=[];i=0
 while i<len(x):
  if i<len(x)-1 and c[x[i]]in'av'and x[i+1]==':':r.append(x[i:2+i]);i+=1
  else:r.append(x[i])
  i+=1
 return r
def parse(x):
 global s,i;s=lex('('+x);i=0;r=t()
 if type(r)==tuple and len(r)and x[0]!='('and r[0]=='(':r=('[',)+r[1:]
 return r
def test():
    import sys
    if 0: #generate test cases
        print('[')
        for x in ["x;y","(x;y)","f[x;y]","x+y","x+*y","1+3*x","(+x)%y","(+/x)%#x","x+m[*i]/y","1+2-","3*:/2","+-","3+-","5(+\\|:)\\x","f::x,0"]:
            print("        [%12s,%s ],"%(x.__repr__(),parse(x)))
        print(']')
    for x,y in [
        [          '',() ],
        [       'x;y',('[', 'x', 'y') ],
        [     '(x;y)',('(', 'x', 'y') ],
        [    'f[x;y]',('f', 'x', 'y') ],
        [       'x+y',('+', 'x', 'y') ],
        [      'x+*y',('+', 'x', ('*', 'y')) ],
        [     '1+3*x',('+', '1', ('*', '3', 'x')) ],
        [    '(+x)%y',('%', ('+', 'x'), 'y') ],
        [  '(+/x)%#x',('%', (('/', '+'), 'x'), ('#', 'x')) ],
        [ 'x+m[*i]/y',('+', 'x', (('/', ('m', ('*', 'i'))), 'y')) ],
        [      '1+2-',("'",('+', '1', ()),('-', '2', ())) ],
        [     '3*:/2',(('/', '*:'), '3', '2') ],
        [        '+-',("'", '+:', '-') ],
        [       '3+-',("'", ('+', '3', ()),'-') ],
        ['5(+\\|:)\\x',(('\\', ("'", ('\\', '+'), '|:')), '5', 'x') ],
        [    'f::x,0',('::', 'f', (',', 'x', '0')) ],
        ]:
        if (r:=parse(x))!=y:print('!',repr(x),repr(r),'!=',y);sys.exit(1)
if __name__=='__main__':
 test()