jaclang 0.6.1__py3-none-any.whl → 0.7.0__py3-none-any.whl

jaclang/vendor/mypyc/transform/ir_transform.py

@@ -0,0 +1,368 @@
+"""Helpers for implementing generic IR to IR transforms."""
+
+from __future__ import annotations
+
+from typing import Final, Optional
+
+from mypyc.ir.ops import (
+    Assign,
+    AssignMulti,
+    BasicBlock,
+    Box,
+    Branch,
+    Call,
+    CallC,
+    Cast,
+    ComparisonOp,
+    DecRef,
+    Extend,
+    FloatComparisonOp,
+    FloatNeg,
+    FloatOp,
+    GetAttr,
+    GetElementPtr,
+    Goto,
+    IncRef,
+    InitStatic,
+    IntOp,
+    KeepAlive,
+    LoadAddress,
+    LoadErrorValue,
+    LoadGlobal,
+    LoadLiteral,
+    LoadMem,
+    LoadStatic,
+    MethodCall,
+    Op,
+    OpVisitor,
+    PrimitiveOp,
+    RaiseStandardError,
+    Return,
+    SetAttr,
+    SetMem,
+    Truncate,
+    TupleGet,
+    TupleSet,
+    Unborrow,
+    Unbox,
+    Unreachable,
+    Value,
+)
+from mypyc.irbuild.ll_builder import LowLevelIRBuilder
+
+
+class IRTransform(OpVisitor[Optional[Value]]):
+    """Identity transform.
+
+    Subclass and override to perform changes to IR.
+
+    Subclass IRTransform and override any OpVisitor visit_* methods
+    that perform any IR changes. The default implementations implement
+    an identity transform.
+
+    A visit method can return None to remove ops. In this case the
+    transform must ensure that no op uses the original removed op
+    as a source after the transform.
+
+    You can retain old BasicBlock and op references in ops. The transform
+    will automatically patch these for you as needed.
+    """
+
+    def __init__(self, builder: LowLevelIRBuilder) -> None:
+        self.builder = builder
+        # Subclasses add additional op mappings here. A None value indicates
+        # that the op/register is deleted.
+        self.op_map: dict[Value, Value | None] = {}
+
+    def transform_blocks(self, blocks: list[BasicBlock]) -> None:
+        """Transform basic blocks that represent a single function.
+
+        The result of the transform will be collected at self.builder.blocks.
+        """
+        block_map: dict[BasicBlock, BasicBlock] = {}
+        op_map = self.op_map
+        empties = set()
+        for block in blocks:
+            new_block = BasicBlock()
+            block_map[block] = new_block
+            self.builder.activate_block(new_block)
+            new_block.error_handler = block.error_handler
+            for op in block.ops:
+                new_op = op.accept(self)
+                if new_op is not op:
+                    op_map[op] = new_op
+            # A transform can produce empty blocks which can be removed.
+            if is_empty_block(new_block) and not is_empty_block(block):
+                empties.add(new_block)
+        self.builder.blocks = [block for block in self.builder.blocks if block not in empties]
+        # Update all op/block references to point to the transformed ones.
+        patcher = PatchVisitor(op_map, block_map)
+        for block in self.builder.blocks:
+            for op in block.ops:
+                op.accept(patcher)
+            if block.error_handler is not None:
+                block.error_handler = block_map.get(block.error_handler, block.error_handler)
+
+    def add(self, op: Op) -> Value:
+        return self.builder.add(op)
+
+    def visit_goto(self, op: Goto) -> None:
+        self.add(op)
+
+    def visit_branch(self, op: Branch) -> None:
+        self.add(op)
+
+    def visit_return(self, op: Return) -> None:
+        self.add(op)
+
+    def visit_unreachable(self, op: Unreachable) -> None:
+        self.add(op)
+
+    def visit_assign(self, op: Assign) -> Value | None:
+        return self.add(op)
+
+    def visit_assign_multi(self, op: AssignMulti) -> Value | None:
+        return self.add(op)
+
+    def visit_load_error_value(self, op: LoadErrorValue) -> Value | None:
+        return self.add(op)
+
+    def visit_load_literal(self, op: LoadLiteral) -> Value | None:
+        return self.add(op)
+
+    def visit_get_attr(self, op: GetAttr) -> Value | None:
+        return self.add(op)
+
+    def visit_set_attr(self, op: SetAttr) -> Value | None:
+        return self.add(op)
+
+    def visit_load_static(self, op: LoadStatic) -> Value | None:
+        return self.add(op)
+
+    def visit_init_static(self, op: InitStatic) -> Value | None:
+        return self.add(op)
+
+    def visit_tuple_get(self, op: TupleGet) -> Value | None:
+        return self.add(op)
+
+    def visit_tuple_set(self, op: TupleSet) -> Value | None:
+        return self.add(op)
+
+    def visit_inc_ref(self, op: IncRef) -> Value | None:
+        return self.add(op)
+
+    def visit_dec_ref(self, op: DecRef) -> Value | None:
+        return self.add(op)
+
+    def visit_call(self, op: Call) -> Value | None:
+        return self.add(op)
+
+    def visit_method_call(self, op: MethodCall) -> Value | None:
+        return self.add(op)
+
+    def visit_cast(self, op: Cast) -> Value | None:
+        return self.add(op)
+
+    def visit_box(self, op: Box) -> Value | None:
+        return self.add(op)
+
+    def visit_unbox(self, op: Unbox) -> Value | None:
+        return self.add(op)
+
+    def visit_raise_standard_error(self, op: RaiseStandardError) -> Value | None:
+        return self.add(op)
+
+    def visit_call_c(self, op: CallC) -> Value | None:
+        return self.add(op)
+
+    def visit_primitive_op(self, op: PrimitiveOp) -> Value | None:
+        return self.add(op)
+
+    def visit_truncate(self, op: Truncate) -> Value | None:
+        return self.add(op)
+
+    def visit_extend(self, op: Extend) -> Value | None:
+        return self.add(op)
+
+    def visit_load_global(self, op: LoadGlobal) -> Value | None:
+        return self.add(op)
+
+    def visit_int_op(self, op: IntOp) -> Value | None:
+        return self.add(op)
+
+    def visit_comparison_op(self, op: ComparisonOp) -> Value | None:
+        return self.add(op)
+
+    def visit_float_op(self, op: FloatOp) -> Value | None:
+        return self.add(op)
+
+    def visit_float_neg(self, op: FloatNeg) -> Value | None:
+        return self.add(op)
+
+    def visit_float_comparison_op(self, op: FloatComparisonOp) -> Value | None:
+        return self.add(op)
+
+    def visit_load_mem(self, op: LoadMem) -> Value | None:
+        return self.add(op)
+
+    def visit_set_mem(self, op: SetMem) -> Value | None:
+        return self.add(op)
+
+    def visit_get_element_ptr(self, op: GetElementPtr) -> Value | None:
+        return self.add(op)
+
+    def visit_load_address(self, op: LoadAddress) -> Value | None:
+        return self.add(op)
+
+    def visit_keep_alive(self, op: KeepAlive) -> Value | None:
+        return self.add(op)
+
+    def visit_unborrow(self, op: Unborrow) -> Value | None:
+        return self.add(op)
+
+
+class PatchVisitor(OpVisitor[None]):
+    def __init__(
+        self, op_map: dict[Value, Value | None], block_map: dict[BasicBlock, BasicBlock]
+    ) -> None:
+        self.op_map: Final = op_map
+        self.block_map: Final = block_map
+
+    def fix_op(self, op: Value) -> Value:
+        new = self.op_map.get(op, op)
+        assert new is not None, "use of removed op"
+        return new
+
+    def fix_block(self, block: BasicBlock) -> BasicBlock:
+        return self.block_map.get(block, block)
+
+    def visit_goto(self, op: Goto) -> None:
+        op.label = self.fix_block(op.label)
+
+    def visit_branch(self, op: Branch) -> None:
+        op.value = self.fix_op(op.value)
+        op.true = self.fix_block(op.true)
+        op.false = self.fix_block(op.false)
+
+    def visit_return(self, op: Return) -> None:
+        op.value = self.fix_op(op.value)
+
+    def visit_unreachable(self, op: Unreachable) -> None:
+        pass
+
+    def visit_assign(self, op: Assign) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_assign_multi(self, op: AssignMulti) -> None:
+        op.src = [self.fix_op(s) for s in op.src]
+
+    def visit_load_error_value(self, op: LoadErrorValue) -> None:
+        pass
+
+    def visit_load_literal(self, op: LoadLiteral) -> None:
+        pass
+
+    def visit_get_attr(self, op: GetAttr) -> None:
+        op.obj = self.fix_op(op.obj)
+
+    def visit_set_attr(self, op: SetAttr) -> None:
+        op.obj = self.fix_op(op.obj)
+        op.src = self.fix_op(op.src)
+
+    def visit_load_static(self, op: LoadStatic) -> None:
+        pass
+
+    def visit_init_static(self, op: InitStatic) -> None:
+        op.value = self.fix_op(op.value)
+
+    def visit_tuple_get(self, op: TupleGet) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_tuple_set(self, op: TupleSet) -> None:
+        op.items = [self.fix_op(item) for item in op.items]
+
+    def visit_inc_ref(self, op: IncRef) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_dec_ref(self, op: DecRef) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_call(self, op: Call) -> None:
+        op.args = [self.fix_op(arg) for arg in op.args]
+
+    def visit_method_call(self, op: MethodCall) -> None:
+        op.obj = self.fix_op(op.obj)
+        op.args = [self.fix_op(arg) for arg in op.args]
+
+    def visit_cast(self, op: Cast) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_box(self, op: Box) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_unbox(self, op: Unbox) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_raise_standard_error(self, op: RaiseStandardError) -> None:
+        if isinstance(op.value, Value):
+            op.value = self.fix_op(op.value)
+
+    def visit_call_c(self, op: CallC) -> None:
+        op.args = [self.fix_op(arg) for arg in op.args]
+
+    def visit_primitive_op(self, op: PrimitiveOp) -> None:
+        op.args = [self.fix_op(arg) for arg in op.args]
+
+    def visit_truncate(self, op: Truncate) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_extend(self, op: Extend) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_load_global(self, op: LoadGlobal) -> None:
+        pass
+
+    def visit_int_op(self, op: IntOp) -> None:
+        op.lhs = self.fix_op(op.lhs)
+        op.rhs = self.fix_op(op.rhs)
+
+    def visit_comparison_op(self, op: ComparisonOp) -> None:
+        op.lhs = self.fix_op(op.lhs)
+        op.rhs = self.fix_op(op.rhs)
+
+    def visit_float_op(self, op: FloatOp) -> None:
+        op.lhs = self.fix_op(op.lhs)
+        op.rhs = self.fix_op(op.rhs)
+
+    def visit_float_neg(self, op: FloatNeg) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_float_comparison_op(self, op: FloatComparisonOp) -> None:
+        op.lhs = self.fix_op(op.lhs)
+        op.rhs = self.fix_op(op.rhs)
+
+    def visit_load_mem(self, op: LoadMem) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_set_mem(self, op: SetMem) -> None:
+        op.dest = self.fix_op(op.dest)
+        op.src = self.fix_op(op.src)
+
+    def visit_get_element_ptr(self, op: GetElementPtr) -> None:
+        op.src = self.fix_op(op.src)
+
+    def visit_load_address(self, op: LoadAddress) -> None:
+        if isinstance(op.src, LoadStatic):
+            new = self.fix_op(op.src)
+            assert isinstance(new, LoadStatic)
+            op.src = new
+
+    def visit_keep_alive(self, op: KeepAlive) -> None:
+        op.src = [self.fix_op(s) for s in op.src]
+
+    def visit_unborrow(self, op: Unborrow) -> None:
+        op.src = self.fix_op(op.src)
+
+
+def is_empty_block(block: BasicBlock) -> bool:
+    return len(block.ops) == 1 and isinstance(block.ops[0], Unreachable)

jaclang/vendor/mypyc/transform/lower.py

@@ -0,0 +1,33 @@
+"""Transform IR to lower-level ops.
+
+Higher-level ops are used in earlier compiler passes, as they make
+various analyses, optimizations and transforms easier to implement.
+Later passes use lower-level ops, as they are easier to generate code
+from, and they help with lower-level optimizations.
+
+Lowering of various primitive ops is implemented in the mypyc.lower
+package.
+"""
+
+from mypyc.ir.func_ir import FuncIR
+from mypyc.ir.ops import PrimitiveOp, Value
+from mypyc.irbuild.ll_builder import LowLevelIRBuilder
+from mypyc.lower.registry import lowering_registry
+from mypyc.options import CompilerOptions
+from mypyc.transform.ir_transform import IRTransform
+
+
+def lower_ir(ir: FuncIR, options: CompilerOptions) -> None:
+    builder = LowLevelIRBuilder(None, options)
+    visitor = LoweringVisitor(builder)
+    visitor.transform_blocks(ir.blocks)
+    ir.blocks = builder.blocks
+
+
+class LoweringVisitor(IRTransform):
+    def visit_primitive_op(self, op: PrimitiveOp) -> Value:
+        # The lowering implementation functions of various primitive ops are stored
+        # in a registry, which is populated using function decorators. The name
+        # of op (such as "int_eq") is used as the key.
+        lower_fn = lowering_registry[op.desc.name]
+        return lower_fn(self.builder, op.args, op.line)

jaclang/vendor/mypyc/transform/refcount.py

@@ -67,9 +67,7 @@ def insert_ref_count_opcodes(ir: FuncIR) -> None:
     args: set[Value] = set(ir.arg_regs)
     live = analyze_live_regs(ir.blocks, cfg)
     borrow = analyze_borrowed_arguments(ir.blocks, cfg, borrowed)
-    defined = analyze_must_defined_regs(
-        ir.blocks, cfg, args, values, strict_errors=True
-    )
+    defined = analyze_must_defined_regs(ir.blocks, cfg, args, values, strict_errors=True)
     ordering = make_value_ordering(ir)
     cache: BlockCache = {}
     for block in ir.blocks.copy():
@@ -94,10 +92,7 @@ def is_maybe_undefined(post_must_defined: set[Value], src: Value) -> bool:
 
 
 def maybe_append_dec_ref(
-    ops: list[Op],
-    dest: Value,
-    defined: AnalysisDict[Value],
-    key: tuple[BasicBlock, int],
+    ops: list[Op], dest: Value, defined: AnalysisDict[Value], key: tuple[BasicBlock, int]
 ) -> None:
     if dest.type.is_refcounted and not isinstance(dest, Integer):
         ops.append(DecRef(dest, is_xdec=is_maybe_undefined(defined[key], dest)))
@@ -146,11 +141,7 @@ def transform_block(
 
         for src in op.unique_sources():
             # Decrement source that won't be live afterwards.
-            if (
-                src not in post_live[key]
-                and src not in pre_borrow[key]
-                and src not in stolen
-            ):
+            if src not in post_live[key] and src not in pre_borrow[key] and src not in stolen:
                 maybe_append_dec_ref(ops, src, post_must_defined, key)
         # Decrement the destination if it is dead after the op and
         # wasn't a borrowed RegisterOp
@@ -207,17 +198,9 @@ def insert_branch_inc_and_decrefs(
             omitted = ()
 
         decs = after_branch_decrefs(
-            target,
-            pre_live,
-            source_defined,
-            source_borrowed,
-            source_live_regs,
-            ordering,
-            omitted,
-        )
-        incs = after_branch_increfs(
-            target, pre_live, pre_borrow, source_borrowed, ordering
+            target, pre_live, source_defined, source_borrowed, source_live_regs, ordering, omitted
         )
+        incs = after_branch_increfs(target, pre_live, pre_borrow, source_borrowed, ordering)
         term.set_target(i, add_block(decs, incs, cache, blocks, target))
 
 
@@ -253,19 +236,13 @@ def after_branch_increfs(
     incref = (source_borrowed - target_borrowed) & target_pre_live
     if incref:
         return tuple(
-            reg
-            for reg in sorted(incref, key=lambda r: ordering[r])
-            if reg.type.is_refcounted
+            reg for reg in sorted(incref, key=lambda r: ordering[r]) if reg.type.is_refcounted
         )
     return ()
 
 
 def add_block(
-    decs: Decs,
-    incs: Incs,
-    cache: BlockCache,
-    blocks: list[BasicBlock],
-    label: BasicBlock,
+    decs: Decs, incs: Incs, cache: BlockCache, blocks: list[BasicBlock], label: BasicBlock
 ) -> BasicBlock:
     if not decs and not incs:
         return label

jaclang/vendor/mypyc/transform/uninit.py

@@ -2,12 +2,7 @@
 
 from __future__ import annotations
 
-from mypyc.analysis.dataflow import (
-    AnalysisDict,
-    analyze_must_defined_regs,
-    cleanup_cfg,
-    get_cfg,
-)
+from mypyc.analysis.dataflow import AnalysisDict, analyze_must_defined_regs, cleanup_cfg, get_cfg
 from mypyc.common import BITMAP_BITS
 from mypyc.ir.func_ir import FuncIR, all_values
 from mypyc.ir.ops import (
@@ -75,9 +70,7 @@ def split_blocks_at_uninits(
                     and not isinstance(op, LoadAddress)
                 ):
                     new_block, error_block = BasicBlock(), BasicBlock()
-                    new_block.error_handler = error_block.error_handler = (
-                        cur_block.error_handler
-                    )
+                    new_block.error_handler = error_block.error_handler = cur_block.error_handler
                     new_blocks += [error_block, new_block]
 
                     if src not in init_registers_set:
@@ -117,9 +110,7 @@ def split_blocks_at_uninits(
             cur_block.ops.append(op)
 
     if bitmap_backed:
-        update_register_assignments_to_set_bitmap(
-            new_blocks, bitmap_registers, bitmap_backed
-        )
+        update_register_assignments_to_set_bitmap(new_blocks, bitmap_registers, bitmap_backed)
 
     if init_registers:
         new_ops: list[Op] = []
@@ -169,9 +160,7 @@ def check_for_uninit_using_bitmap(
 
 
 def update_register_assignments_to_set_bitmap(
-    blocks: list[BasicBlock],
-    bitmap_registers: list[Register],
-    bitmap_backed: list[Register],
+    blocks: list[BasicBlock], bitmap_registers: list[Register], bitmap_backed: list[Register]
 ) -> None:
     """Update some assignments to registers to also set a bit in a bitmap.
 

jaclang/vendor/pluggy/__init__.py

@@ -18,16 +18,20 @@ __all__ = [
     "HookspecMarker",
     "HookimplMarker",
     "Result",
+    "PluggyWarning",
+    "PluggyTeardownRaisedWarning",
 ]
 
-from ._manager import PluginManager, PluginValidationError
-from ._result import HookCallError, Result
-from ._hooks import (
-    HookspecMarker,
-    HookimplMarker,
-    HookCaller,
-    HookRelay,
-    HookspecOpts,
-    HookimplOpts,
-    HookImpl,
-)
+from ._hooks import HookCaller
+from ._hooks import HookImpl
+from ._hooks import HookimplMarker
+from ._hooks import HookimplOpts
+from ._hooks import HookRelay
+from ._hooks import HookspecMarker
+from ._hooks import HookspecOpts
+from ._manager import PluginManager
+from ._manager import PluginValidationError
+from ._result import HookCallError
+from ._result import Result
+from ._warnings import PluggyTeardownRaisedWarning
+from ._warnings import PluggyWarning

jaclang/vendor/pluggy/_callers.py

@@ -7,24 +7,49 @@ from __future__ import annotations
 from typing import cast
 from typing import Generator
 from typing import Mapping
+from typing import NoReturn
 from typing import Sequence
 from typing import Tuple
 from typing import Union
+import warnings
 
 from ._hooks import HookImpl
-from ._result import _raise_wrapfail
 from ._result import HookCallError
 from ._result import Result
+from ._warnings import PluggyTeardownRaisedWarning
 
 
 # Need to distinguish between old- and new-style hook wrappers.
-# Wrapping one a singleton tuple is the fastest type-safe way I found to do it.
+# Wrapping with a tuple is the fastest type-safe way I found to do it.
 Teardown = Union[
-    Tuple[Generator[None, Result[object], None]],
+    Tuple[Generator[None, Result[object], None], HookImpl],
     Generator[None, object, object],
 ]
 
 
+def _raise_wrapfail(
+    wrap_controller: (
+        Generator[None, Result[object], None] | Generator[None, object, object]
+    ),
+    msg: str,
+) -> NoReturn:
+    co = wrap_controller.gi_code
+    raise RuntimeError(
+        "wrap_controller at %r %s:%d %s"
+        % (co.co_name, co.co_filename, co.co_firstlineno, msg)
+    )
+
+
+def _warn_teardown_exception(
+    hook_name: str, hook_impl: HookImpl, e: BaseException
+) -> None:
+    msg = "A plugin raised an exception during an old-style hookwrapper teardown.\n"
+    msg += f"Plugin: {hook_impl.plugin_name}, Hook: {hook_name}\n"
+    msg += f"{type(e).__name__}: {e}\n"
+    msg += "For more information see https://pluggy.readthedocs.io/en/stable/api_reference.html#pluggy.PluggyTeardownRaisedWarning"  # noqa: E501
+    warnings.warn(PluggyTeardownRaisedWarning(msg), stacklevel=5)
+
+
 def _multicall(
     hook_name: str,
     hook_impls: Sequence[HookImpl],
@@ -61,7 +86,7 @@ def _multicall(
                         res = hook_impl.function(*args)
                         wrapper_gen = cast(Generator[None, Result[object], None], res)
                         next(wrapper_gen)  # first yield
-                        teardowns.append((wrapper_gen,))
+                        teardowns.append((wrapper_gen, hook_impl))
                     except StopIteration:
                         _raise_wrapfail(wrapper_gen, "did not yield")
                 elif hook_impl.wrapper:
@@ -129,9 +154,13 @@ def _multicall(
                 if isinstance(teardown, tuple):
                     try:
                         teardown[0].send(outcome)
-                        _raise_wrapfail(teardown[0], "has second yield")
                     except StopIteration:
                         pass
+                    except BaseException as e:
+                        _warn_teardown_exception(hook_name, teardown[1], e)
+                        raise
+                    else:
+                        _raise_wrapfail(teardown[0], "has second yield")
                 else:
                     try:
                         if outcome._exception is not None: