19 lines
894 B
Python
19 lines
894 B
Python
from __future__ import annotations
|
|
|
|
from mypyc.ir.ops import ComparisonOp, GetElementPtr, Integer, LoadMem, Value
|
|
from mypyc.ir.rtypes import PyVarObject, c_pyssize_t_rprimitive, object_rprimitive
|
|
from mypyc.irbuild.ll_builder import LowLevelIRBuilder
|
|
from mypyc.lower.registry import lower_primitive_op
|
|
|
|
|
|
@lower_primitive_op("var_object_size")
|
|
def var_object_size(builder: LowLevelIRBuilder, args: list[Value], line: int) -> Value:
|
|
elem_address = builder.add(GetElementPtr(args[0], PyVarObject, "ob_size"))
|
|
return builder.add(LoadMem(c_pyssize_t_rprimitive, elem_address))
|
|
|
|
|
|
@lower_primitive_op("propagate_if_error")
|
|
def propagate_if_error_op(builder: LowLevelIRBuilder, args: list[Value], line: int) -> Value:
|
|
# Return False on NULL. The primitive uses ERR_FALSE, so this is an error.
|
|
return builder.add(ComparisonOp(args[0], Integer(0, object_rprimitive), ComparisonOp.NEQ))
|