forked from IUCompilerCourse/python-student-support-code
-
Notifications
You must be signed in to change notification settings - Fork 0
/
type_check_Cany.py
49 lines (46 loc) · 1.71 KB
/
type_check_Cany.py
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
from type_check_Clambda import TypeCheckClambda
from utils import *
class TypeCheckCany(TypeCheckClambda):
def type_check_exp(self, e, env):
match e:
case ValueOf(value, typ):
self.type_check_exp(value, env)
return typ
case TagOf(value):
t = self.type_check_exp(value, env)
self.check_type_equal(t, AnyType(), e)
return IntType()
case Call(Name('any_load'), [tup, index]):
t = self.type_check_exp(tup, env)
self.check_type_equal(t, AnyType(), e)
return AnyType()
# case Call(Name('any_tuple_store'), [tup, index, value]):
# t = self.type_check_exp(tup, env)
# self.check_type_equal(t, AnyType(), e)
# v = self.type_check_exp(value, env)
# self.check_type_equal(v, AnyType(), e)
# return type(None) # ??
case Call(Name('any_len'), [tup]):
t = self.type_check_exp(tup, env)
self.check_type_equal(t, AnyType(), e)
return IntType()
case Call(Name('make_any'), [value, tag]):
v = self.type_check_exp(value, env)
t = self.type_check_exp(tag, env)
self.check_type_equal(t, IntType(), e)
return AnyType()
case Call(Name('exit'), []):
return Bottom()
case Call(Name('arity'), [fun]):
ty = self.type_check_exp(fun, env)
match ty:
case FunctionType(ps, rt):
return IntType()
case TupleType([FunctionType(ps,rs)]): # after closure conversion
return IntType()
case Bottom():
return IntType()
case _:
raise Exception('type_check_exp arity unexpected ' + repr(ty))
case _:
return super().type_check_exp(e, env)