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

# 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