The Meson Build System
http://mesonbuild.com/
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
83 lines
2.6 KiB
83 lines
2.6 KiB
3 years ago
|
# Copyright 2021 The Meson development team
|
||
|
# SPDX-license-identifier: Apache-2.0
|
||
|
|
||
|
from ...interpreterbase import (
|
||
|
ObjectHolder,
|
||
|
MesonOperator,
|
||
|
typed_operator,
|
||
|
noKwargs,
|
||
|
noPosargs,
|
||
|
|
||
|
TYPE_var,
|
||
|
TYPE_kwargs,
|
||
|
|
||
|
InvalidArguments
|
||
|
)
|
||
|
|
||
|
import typing as T
|
||
|
|
||
|
if T.TYPE_CHECKING:
|
||
|
# Object holders need the actual interpreter
|
||
|
from ...interpreter import Interpreter
|
||
|
|
||
|
class IntegerHolder(ObjectHolder[int]):
|
||
|
def __init__(self, obj: int, interpreter: 'Interpreter') -> None:
|
||
|
super().__init__(obj, interpreter)
|
||
|
self.methods.update({
|
||
|
'is_even': self.is_even_method,
|
||
|
'is_odd': self.is_odd_method,
|
||
|
'to_string': self.to_string_method,
|
||
|
})
|
||
|
|
||
|
self.trivial_operators.update({
|
||
|
# Arithmetic
|
||
|
MesonOperator.UMINUS: (None, lambda x: -self.held_object),
|
||
|
MesonOperator.PLUS: (int, lambda x: self.held_object + x),
|
||
|
MesonOperator.MINUS: (int, lambda x: self.held_object - x),
|
||
|
MesonOperator.TIMES: (int, lambda x: self.held_object * x),
|
||
|
|
||
|
# Comparison
|
||
|
MesonOperator.EQUALS: (int, lambda x: self.held_object == x),
|
||
|
MesonOperator.NOT_EQUALS: (int, lambda x: self.held_object != x),
|
||
|
MesonOperator.GREATER: (int, lambda x: self.held_object > x),
|
||
|
MesonOperator.LESS: (int, lambda x: self.held_object < x),
|
||
|
MesonOperator.GREATER_EQUALS: (int, lambda x: self.held_object >= x),
|
||
|
MesonOperator.LESS_EQUALS: (int, lambda x: self.held_object <= x),
|
||
|
})
|
||
|
|
||
|
# Use actual methods for functions that require additional checks
|
||
|
self.operators.update({
|
||
|
MesonOperator.DIV: self.op_div,
|
||
|
MesonOperator.MOD: self.op_mod,
|
||
|
})
|
||
|
|
||
|
def display_name(self) -> str:
|
||
|
return 'int'
|
||
|
|
||
|
@noKwargs
|
||
|
@noPosargs
|
||
|
def is_even_method(self, args: T.List[TYPE_var], kwargs: TYPE_kwargs) -> bool:
|
||
|
return self.held_object % 2 == 0
|
||
|
|
||
|
@noKwargs
|
||
|
@noPosargs
|
||
|
def is_odd_method(self, args: T.List[TYPE_var], kwargs: TYPE_kwargs) -> bool:
|
||
|
return self.held_object % 2 != 0
|
||
|
|
||
|
@noKwargs
|
||
|
@noPosargs
|
||
|
def to_string_method(self, args: T.List[TYPE_var], kwargs: TYPE_kwargs) -> str:
|
||
|
return str(self.held_object)
|
||
|
|
||
|
@typed_operator(MesonOperator.DIV, int)
|
||
|
def op_div(self, other: int) -> int:
|
||
|
if other == 0:
|
||
|
raise InvalidArguments('Tried to divide by 0')
|
||
|
return self.held_object // other
|
||
|
|
||
|
@typed_operator(MesonOperator.MOD, int)
|
||
|
def op_mod(self, other: int) -> int:
|
||
|
if other == 0:
|
||
|
raise InvalidArguments('Tried to divide by 0')
|
||
|
return self.held_object % other
|