Class TupleGet

mypyc/ir/ops.py:1082–1110

Get item of a fixed-length tuple (src[index]).

@final
class TupleGet(RegisterOp):
    """Get item of a fixed-length tuple (src[index])."""

    error_kind = ERR_NEVER

    def __init__(self, src: Value, index: int, line: int = -1, *, borrow: bool = False) -> None:
        super().__init__(line)
        assert isinstance(
            src.type, RTuple
        ), f"TupleGet only operates on tuples, not {type(src.type).__name__}"
        src_len = len(src.type.types)
        self.src = src
        self.index = index
        if index < 0:
            self.index += src_len
        assert (
            self.index <= src_len - 1
        ), f"Index out of range.\nsource type: {src.type}\nindex: {index}"
        self.type = src.type.types[index]
        self.is_borrowed = borrow

    def sources(self) -> list[Value]:
        return [self.src]

    def set_sources(self, new: list[Value]) -> None:
        (self.src,) = new

    def accept(self, visitor: OpVisitor[T]) -> T:
        return visitor.visit_tuple_get(self)


@final