43 lines
1.4 KiB
Python
43 lines
1.4 KiB
Python
|
from nac3_types import *
|
||
|
|
|
||
|
|
|
||
|
|
TBool = TObj('bool', {}, [])
|
||
|
|
TInt = TObj('int', {}, [])
|
||
|
|
TFloat = TObj('float', {}, [])
|
||
|
|
|
||
|
|
TBool.fields['__eq__'] = TFunc([FuncArg('other', TBool, False)], TBool, [])
|
||
|
|
TBool.fields['__ne__'] = TFunc([FuncArg('other', TBool, False)], TBool, [])
|
||
|
|
|
||
|
|
|
||
|
|
def impl_cmp(ty):
|
||
|
|
ty.fields['__lt__'] = TFunc([FuncArg('other', ty, False)], TBool, [])
|
||
|
|
ty.fields['__le__'] = TFunc([FuncArg('other', ty, False)], TBool, [])
|
||
|
|
ty.fields['__eq__'] = TFunc([FuncArg('other', ty, False)], TBool, [])
|
||
|
|
ty.fields['__ne__'] = TFunc([FuncArg('other', ty, False)], TBool, [])
|
||
|
|
ty.fields['__gt__'] = TFunc([FuncArg('other', ty, False)], TBool, [])
|
||
|
|
ty.fields['__ge__'] = TFunc([FuncArg('other', ty, False)], TBool, [])
|
||
|
|
|
||
|
|
|
||
|
|
def impl_arithmetic(ty):
|
||
|
|
ty.fields['__add__'] = TFunc([FuncArg('other', ty, False)], ty, [])
|
||
|
|
ty.fields['__sub__'] = TFunc([FuncArg('other', ty, False)], ty, [])
|
||
|
|
ty.fields['__mul__'] = TFunc([FuncArg('other', ty, False)], ty, [])
|
||
|
|
|
||
|
|
|
||
|
|
impl_cmp(TInt)
|
||
|
|
impl_cmp(TFloat)
|
||
|
|
impl_arithmetic(TInt)
|
||
|
|
impl_arithmetic(TFloat)
|
||
|
|
|
||
|
|
TNum = TVar([TInt, TFloat])
|
||
|
|
|
||
|
|
TInt.fields['__truediv__'] = TFunc(
|
||
|
|
[FuncArg('other', TNum, False)], TFloat, [TNum])
|
||
|
|
TInt.fields['__floordiv__'] = TFunc(
|
||
|
|
[FuncArg('other', TNum, False)], TInt, [TNum])
|
||
|
|
TFloat.fields['__truediv__'] = TFunc(
|
||
|
|
[FuncArg('other', TNum, False)], TFloat, [TNum])
|
||
|
|
TFloat.fields['__floordiv__'] = TFunc(
|
||
|
|
[FuncArg('other', TNum, False)], TFloat, [TNum])
|
||
|
|
|