Method gen_condition

mypyc/irbuild/for_helpers.py:752–776

(self)

        self.target_type = target_type

    def gen_condition(self) -> None:
        builder = self.builder
        line = self.line
        self.return_value = Register(object_rprimitive)
        err = builder.add(LoadErrorValue(object_rprimitive, undefines=True))
        builder.assign(self.return_value, err, line)

        # Call generated generator helper method, passing a PyObject ** as the final
        # argument that will be used to store the return value in the return value
        # register. We ignore the return value but the presence of a return value
        # indicates that the generator has finished. This is faster than raising
        # and catching StopIteration, which is the non-native way of doing this.
        ptr = builder.add(LoadAddress(object_pointer_rprimitive, self.return_value))
        nn = builder.none_object()
        helper_call = MethodCall(
            builder.read(self.iter_target, line),
            GENERATOR_HELPER_NAME,
            [nn, nn, nn, nn, ptr],
            line,
        )
        # We provide custom handling for error values.
        helper_call.error_kind = ERR_NEVER

        self.next_reg = builder.add(helper_call)
        builder.add(Branch(self.next_reg, self.loop_exit, self.body_block, Branch.IS_ERROR))

    def begin_body(self) -> None:
        # Assign the value obtained from the generator helper method to the