jaclang 0.7.34__py3-none-any.whl → 0.8.1__py3-none-any.whl

jaclang/vendor/mypyc/ir/func_ir.py

@@ -1,370 +0,0 @@
-"""Intermediate representation of functions."""
-
-from __future__ import annotations
-
-from typing import Final, Sequence
-
-from mypy.nodes import ARG_POS, ArgKind, Block, FuncDef
-from mypyc.common import BITMAP_BITS, JsonDict, bitmap_name, get_id_from_name, short_id_from_name
-from mypyc.ir.ops import (
-    Assign,
-    AssignMulti,
-    BasicBlock,
-    ControlOp,
-    DeserMaps,
-    LoadAddress,
-    Register,
-    Value,
-)
-from mypyc.ir.rtypes import RType, bitmap_rprimitive, deserialize_type
-from mypyc.namegen import NameGenerator
-
-
-class RuntimeArg:
-    """Description of a function argument in IR.
-
-    Argument kind is one of ARG_* constants defined in mypy.nodes.
-    """
-
-    def __init__(
-        self, name: str, typ: RType, kind: ArgKind = ARG_POS, pos_only: bool = False
-    ) -> None:
-        self.name = name
-        self.type = typ
-        self.kind = kind
-        self.pos_only = pos_only
-
-    @property
-    def optional(self) -> bool:
-        return self.kind.is_optional()
-
-    def __repr__(self) -> str:
-        return "RuntimeArg(name={}, type={}, optional={!r}, pos_only={!r})".format(
-            self.name, self.type, self.optional, self.pos_only
-        )
-
-    def serialize(self) -> JsonDict:
-        return {
-            "name": self.name,
-            "type": self.type.serialize(),
-            "kind": int(self.kind.value),
-            "pos_only": self.pos_only,
-        }
-
-    @classmethod
-    def deserialize(cls, data: JsonDict, ctx: DeserMaps) -> RuntimeArg:
-        return RuntimeArg(
-            data["name"],
-            deserialize_type(data["type"], ctx),
-            ArgKind(data["kind"]),
-            data["pos_only"],
-        )
-
-
-class FuncSignature:
-    """Signature of a function in IR."""
-
-    # TODO: Track if method?
-
-    def __init__(self, args: Sequence[RuntimeArg], ret_type: RType) -> None:
-        self.args = tuple(args)
-        self.ret_type = ret_type
-        # Bitmap arguments are use to mark default values for arguments that
-        # have types with overlapping error values.
-        self.num_bitmap_args = num_bitmap_args(self.args)
-        if self.num_bitmap_args:
-            extra = [
-                RuntimeArg(bitmap_name(i), bitmap_rprimitive, pos_only=True)
-                for i in range(self.num_bitmap_args)
-            ]
-            self.args = self.args + tuple(reversed(extra))
-
-    def real_args(self) -> tuple[RuntimeArg, ...]:
-        """Return arguments without any synthetic bitmap arguments."""
-        if self.num_bitmap_args:
-            return self.args[: -self.num_bitmap_args]
-        return self.args
-
-    def bound_sig(self) -> FuncSignature:
-        if self.num_bitmap_args:
-            return FuncSignature(self.args[1 : -self.num_bitmap_args], self.ret_type)
-        else:
-            return FuncSignature(self.args[1:], self.ret_type)
-
-    def __repr__(self) -> str:
-        return f"FuncSignature(args={self.args!r}, ret={self.ret_type!r})"
-
-    def serialize(self) -> JsonDict:
-        if self.num_bitmap_args:
-            args = self.args[: -self.num_bitmap_args]
-        else:
-            args = self.args
-        return {"args": [t.serialize() for t in args], "ret_type": self.ret_type.serialize()}
-
-    @classmethod
-    def deserialize(cls, data: JsonDict, ctx: DeserMaps) -> FuncSignature:
-        return FuncSignature(
-            [RuntimeArg.deserialize(arg, ctx) for arg in data["args"]],
-            deserialize_type(data["ret_type"], ctx),
-        )
-
-
-def num_bitmap_args(args: tuple[RuntimeArg, ...]) -> int:
-    n = 0
-    for arg in args:
-        if arg.type.error_overlap and arg.kind.is_optional():
-            n += 1
-    return (n + (BITMAP_BITS - 1)) // BITMAP_BITS
-
-
-FUNC_NORMAL: Final = 0
-FUNC_STATICMETHOD: Final = 1
-FUNC_CLASSMETHOD: Final = 2
-
-
-class FuncDecl:
-    """Declaration of a function in IR (without body or implementation).
-
-    A function can be a regular module-level function, a method, a
-    static method, a class method, or a property getter/setter.
-    """
-
-    def __init__(
-        self,
-        name: str,
-        class_name: str | None,
-        module_name: str,
-        sig: FuncSignature,
-        kind: int = FUNC_NORMAL,
-        is_prop_setter: bool = False,
-        is_prop_getter: bool = False,
-        implicit: bool = False,
-    ) -> None:
-        self.name = name
-        self.class_name = class_name
-        self.module_name = module_name
-        self.sig = sig
-        self.kind = kind
-        self.is_prop_setter = is_prop_setter
-        self.is_prop_getter = is_prop_getter
-        if class_name is None:
-            self.bound_sig: FuncSignature | None = None
-        else:
-            if kind == FUNC_STATICMETHOD:
-                self.bound_sig = sig
-            else:
-                self.bound_sig = sig.bound_sig()
-
-        # If True, not present in the mypy AST and must be synthesized during irbuild
-        # Currently only supported for property getters/setters
-        self.implicit = implicit
-
-        # This is optional because this will be set to the line number when the corresponding
-        # FuncIR is created
-        self._line: int | None = None
-
-    @property
-    def line(self) -> int:
-        assert self._line is not None
-        return self._line
-
-    @line.setter
-    def line(self, line: int) -> None:
-        self._line = line
-
-    @property
-    def id(self) -> str:
-        assert self.line is not None
-        return get_id_from_name(self.name, self.fullname, self.line)
-
-    @staticmethod
-    def compute_shortname(class_name: str | None, name: str) -> str:
-        return class_name + "." + name if class_name else name
-
-    @property
-    def shortname(self) -> str:
-        return FuncDecl.compute_shortname(self.class_name, self.name)
-
-    @property
-    def fullname(self) -> str:
-        return self.module_name + "." + self.shortname
-
-    def cname(self, names: NameGenerator) -> str:
-        partial_name = short_id_from_name(self.name, self.shortname, self._line)
-        return names.private_name(self.module_name, partial_name)
-
-    def serialize(self) -> JsonDict:
-        return {
-            "name": self.name,
-            "class_name": self.class_name,
-            "module_name": self.module_name,
-            "sig": self.sig.serialize(),
-            "kind": self.kind,
-            "is_prop_setter": self.is_prop_setter,
-            "is_prop_getter": self.is_prop_getter,
-            "implicit": self.implicit,
-        }
-
-    # TODO: move this to FuncIR?
-    @staticmethod
-    def get_id_from_json(func_ir: JsonDict) -> str:
-        """Get the id from the serialized FuncIR associated with this FuncDecl"""
-        decl = func_ir["decl"]
-        shortname = FuncDecl.compute_shortname(decl["class_name"], decl["name"])
-        fullname = decl["module_name"] + "." + shortname
-        return get_id_from_name(decl["name"], fullname, func_ir["line"])
-
-    @classmethod
-    def deserialize(cls, data: JsonDict, ctx: DeserMaps) -> FuncDecl:
-        return FuncDecl(
-            data["name"],
-            data["class_name"],
-            data["module_name"],
-            FuncSignature.deserialize(data["sig"], ctx),
-            data["kind"],
-            data["is_prop_setter"],
-            data["is_prop_getter"],
-            data["implicit"],
-        )
-
-
-class FuncIR:
-    """Intermediate representation of a function with contextual information.
-
-    Unlike FuncDecl, this includes the IR of the body (basic blocks).
-    """
-
-    def __init__(
-        self,
-        decl: FuncDecl,
-        arg_regs: list[Register],
-        blocks: list[BasicBlock],
-        line: int = -1,
-        traceback_name: str | None = None,
-    ) -> None:
-        # Declaration of the function, including the signature
-        self.decl = decl
-        # Registers for all the arguments to the function
-        self.arg_regs = arg_regs
-        # Body of the function
-        self.blocks = blocks
-        self.decl.line = line
-        # The name that should be displayed for tracebacks that
-        # include this function. Function will be omitted from
-        # tracebacks if None.
-        self.traceback_name = traceback_name
-
-    @property
-    def line(self) -> int:
-        return self.decl.line
-
-    @property
-    def args(self) -> Sequence[RuntimeArg]:
-        return self.decl.sig.args
-
-    @property
-    def ret_type(self) -> RType:
-        return self.decl.sig.ret_type
-
-    @property
-    def class_name(self) -> str | None:
-        return self.decl.class_name
-
-    @property
-    def sig(self) -> FuncSignature:
-        return self.decl.sig
-
-    @property
-    def name(self) -> str:
-        return self.decl.name
-
-    @property
-    def fullname(self) -> str:
-        return self.decl.fullname
-
-    @property
-    def id(self) -> str:
-        return self.decl.id
-
-    def cname(self, names: NameGenerator) -> str:
-        return self.decl.cname(names)
-
-    def __repr__(self) -> str:
-        if self.class_name:
-            return f"<FuncIR {self.class_name}.{self.name}>"
-        else:
-            return f"<FuncIR {self.name}>"
-
-    def serialize(self) -> JsonDict:
-        # We don't include blocks in the serialized version
-        return {
-            "decl": self.decl.serialize(),
-            "line": self.line,
-            "traceback_name": self.traceback_name,
-        }
-
-    @classmethod
-    def deserialize(cls, data: JsonDict, ctx: DeserMaps) -> FuncIR:
-        return FuncIR(
-            FuncDecl.deserialize(data["decl"], ctx), [], [], data["line"], data["traceback_name"]
-        )
-
-
-INVALID_FUNC_DEF: Final = FuncDef("<INVALID_FUNC_DEF>", [], Block([]))
-
-
-def all_values(args: list[Register], blocks: list[BasicBlock]) -> list[Value]:
-    """Return the set of all values that may be initialized in the blocks.
-
-    This omits registers that are only read.
-    """
-    values: list[Value] = list(args)
-    seen_registers = set(args)
-
-    for block in blocks:
-        for op in block.ops:
-            if not isinstance(op, ControlOp):
-                if isinstance(op, (Assign, AssignMulti)):
-                    if op.dest not in seen_registers:
-                        values.append(op.dest)
-                        seen_registers.add(op.dest)
-                elif op.is_void:
-                    continue
-                else:
-                    # If we take the address of a register, it might get initialized.
-                    if (
-                        isinstance(op, LoadAddress)
-                        and isinstance(op.src, Register)
-                        and op.src not in seen_registers
-                    ):
-                        values.append(op.src)
-                        seen_registers.add(op.src)
-                    values.append(op)
-
-    return values
-
-
-def all_values_full(args: list[Register], blocks: list[BasicBlock]) -> list[Value]:
-    """Return set of all values that are initialized or accessed."""
-    values: list[Value] = list(args)
-    seen_registers = set(args)
-
-    for block in blocks:
-        for op in block.ops:
-            for source in op.sources():
-                # Look for uninitialized registers that are accessed. Ignore
-                # non-registers since we don't allow ops outside basic blocks.
-                if isinstance(source, Register) and source not in seen_registers:
-                    values.append(source)
-                    seen_registers.add(source)
-            if not isinstance(op, ControlOp):
-                if isinstance(op, (Assign, AssignMulti)):
-                    if op.dest not in seen_registers:
-                        values.append(op.dest)
-                        seen_registers.add(op.dest)
-                elif op.is_void:
-                    continue
-                else:
-                    values.append(op)
-
-    return values

jaclang/vendor/mypyc/ir/module_ir.py

@@ -1,88 +0,0 @@
-"""Intermediate representation of modules."""
-
-from __future__ import annotations
-
-from typing import Dict
-
-from mypyc.common import JsonDict
-from mypyc.ir.class_ir import ClassIR
-from mypyc.ir.func_ir import FuncDecl, FuncIR
-from mypyc.ir.ops import DeserMaps
-from mypyc.ir.rtypes import RType, deserialize_type
-
-
-class ModuleIR:
-    """Intermediate representation of a module."""
-
-    def __init__(
-        self,
-        fullname: str,
-        imports: list[str],
-        functions: list[FuncIR],
-        classes: list[ClassIR],
-        final_names: list[tuple[str, RType]],
-    ) -> None:
-        self.fullname = fullname
-        self.imports = imports.copy()
-        self.functions = functions
-        self.classes = classes
-        self.final_names = final_names
-
-    def serialize(self) -> JsonDict:
-        return {
-            "fullname": self.fullname,
-            "imports": self.imports,
-            "functions": [f.serialize() for f in self.functions],
-            "classes": [c.serialize() for c in self.classes],
-            "final_names": [(k, t.serialize()) for k, t in self.final_names],
-        }
-
-    @classmethod
-    def deserialize(cls, data: JsonDict, ctx: DeserMaps) -> ModuleIR:
-        return ModuleIR(
-            data["fullname"],
-            data["imports"],
-            [ctx.functions[FuncDecl.get_id_from_json(f)] for f in data["functions"]],
-            [ClassIR.deserialize(c, ctx) for c in data["classes"]],
-            [(k, deserialize_type(t, ctx)) for k, t in data["final_names"]],
-        )
-
-
-def deserialize_modules(data: dict[str, JsonDict], ctx: DeserMaps) -> dict[str, ModuleIR]:
-    """Deserialize a collection of modules.
-
-    The modules can contain dependencies on each other.
-
-    Arguments:
-        data: A dict containing the modules to deserialize.
-        ctx: The deserialization maps to use and to populate.
-             They are populated with information from the deserialized
-             modules and as a precondition must have been populated by
-             deserializing any dependencies of the modules being deserialized
-             (outside of dependencies between the modules themselves).
-
-    Returns a map containing the deserialized modules.
-    """
-    for mod in data.values():
-        # First create ClassIRs for every class so that we can construct types and whatnot
-        for cls in mod["classes"]:
-            ir = ClassIR(cls["name"], cls["module_name"])
-            assert ir.fullname not in ctx.classes, "Class %s already in map" % ir.fullname
-            ctx.classes[ir.fullname] = ir
-
-    for mod in data.values():
-        # Then deserialize all of the functions so that methods are available
-        # to the class deserialization.
-        for method in mod["functions"]:
-            func = FuncIR.deserialize(method, ctx)
-            assert func.decl.id not in ctx.functions, (
-                "Method %s already in map" % func.decl.fullname
-            )
-            ctx.functions[func.decl.id] = func
-
-    return {k: ModuleIR.deserialize(v, ctx) for k, v in data.items()}
-
-
-# ModulesIRs should also always be an *OrderedDict*, but if we
-# declared it that way we would need to put it in quotes everywhere...
-ModuleIRs = Dict[str, ModuleIR]