Class VariableType

src/Def.py:99–124

Allows type checking in a structured manner.

class VariableType:
    """
    Allows type checking in a structured manner.
    """

    def __init__(self, ckind: VariableCompKind, elem_ckind: VariableCompKind = VariableCompKind(VariableKind.INT64, VariableMetaKind.PRIM), name: str = '') -> None:
        self.name = name
        self.ckind = ckind
        self.elem_ckind = elem_ckind

    def __eq__(self, other):
        if isinstance(other, VariableType):
            return (self.ckind, self.elem_ckind, self.name) == (other.ckind, other.elem_ckind, other.name)
        return False

    def __str__(self) -> str:
        attrs = []
        for key, value in vars(self).items():
            attrs.append(f"{key}={value}")
        return f"{self.__class__.__name__}({', '.join(attrs)})"

    def kind(self):
        return self.ckind.kind

    def meta_kind(self):
        return self.ckind.meta_kind


@dataclass