from rpython.jit.codewriter.effectinfo import EffectInfo
from rpython.jit.metainterp.optimizeopt.optimizer import Optimization
from rpython.jit.metainterp.resoperation import rop, OpHelpers


def is_raw_free(op, opnum):
    if not OpHelpers.is_real_call(opnum):
        return False
    einfo = op.getdescr().get_extra_info()
    return einfo.oopspecindex == EffectInfo.OS_RAW_FREE


class OptEarlyForce(Optimization):
    def propagate_forward(self, op):
        opnum = op.getopnum()

        if (opnum != rop.SETFIELD_GC and
            opnum != rop.SETARRAYITEM_GC and
            opnum != rop.SETARRAYITEM_RAW and
            opnum != rop.QUASIIMMUT_FIELD and
            opnum != rop.SAME_AS_I and
            opnum != rop.SAME_AS_R and
            opnum != rop.SAME_AS_F and
            not is_raw_free(op, opnum)):

            for arg in op.getarglist():
                self.optimizer.force_box(arg, self)
        return self.emit(op)

    def setup(self):
        self.optimizer.optearlyforce = self