Class AstMap

lean_py/z3/core.py:3081–3109

Map from AST nodes to AST nodes.

class AstMap:
    """Map from AST nodes to AST nodes."""

    def __init__(self, ctx: Context | None = None) -> None:
        self._map: dict[Any, Any] = {}

    def __setitem__(self, k: Any, v: Any) -> None:
        self._map[k] = v

    def __getitem__(self, k: Any) -> Any:
        return self._map[k]

    def __contains__(self, k: Any) -> bool:
        return k in self._map

    def __len__(self) -> int:
        return len(self._map)

    def __repr__(self) -> str:
        return repr(self._map)

    def erase(self, k: Any) -> None:
        self._map.pop(k, None)

    def reset(self) -> None:
        self._map.clear()

    def keys(self):
        return list(self._map.keys())


# ---------------------------------------------------------------------------