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, line))

@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, line)

)