def make_nonnull_str(self, op, mode):
from rpython.jit.metainterp.optimizeopt import vstring
op = self.get_box_replacement(op)
if op.is_constant():
return
opinfo = op.get_forwarded()
if isinstance(opinfo, vstring.StrPtrInfo):
return
op.set_forwarded(vstring.StrPtrInfo(mode))
def ensure_ptr_info_arg0(self, op):
from rpython.jit.metainterp.optimizeopt import vstring
arg0 = self.get_box_replacement(op.getarg(0))
if arg0.is_constant():
return info.ConstPtrInfo(arg0)
opinfo = arg0.get_forwarded()
if isinstance(opinfo, info.AbstractVirtualPtrInfo):
return opinfo
elif opinfo is not None:
last_guard_pos = opinfo.get_last_guard_pos()
else:
last_guard_pos = -1
assert opinfo is None or opinfo.__class__ is info.NonNullPtrInfo
opnum = op.opnum
if (rop.is_getfield(opnum) or opnum == rop.SETFIELD_GC
or opnum == rop.QUASIIMMUT_FIELD):
descr = op.getdescr()
parent_descr = descr.get_parent_descr()
if parent_descr.is_object():
opinfo = info.InstancePtrInfo(parent_descr)
else:
opinfo = info.StructPtrInfo(parent_descr)
opinfo.init_fields(parent_descr, descr.get_index())
elif (rop.is_getarrayitem(opnum) or opnum == rop.SETARRAYITEM_GC
or opnum == rop.ARRAYLEN_GC):
opinfo = info.ArrayPtrInfo(op.getdescr())
elif opnum in (rop.GUARD_CLASS, rop.GUARD_NONNULL_CLASS):
opinfo = info.InstancePtrInfo()
elif opnum in (rop.STRLEN, ):
opinfo = vstring.StrPtrInfo(vstring.mode_string)
elif opnum in (rop.UNICODELEN, ):
opinfo = vstring.StrPtrInfo(vstring.mode_unicode)
else:
assert False, "operations %s unsupported" % op
assert isinstance(opinfo, info.NonNullPtrInfo)
opinfo.last_guard_pos = last_guard_pos
arg0.set_forwarded(opinfo)
return opinfo