codekingpro/portable-devtools

1from typing import TypedDict, type_check_only
2
3from numpy._typing import _ArrayLikeInt_co
4from numpy.random.bit_generator import BitGenerator, SeedSequence
5
6__all__ = ["PCG64"]
7
8@type_check_only
9class _PCG64Internal(TypedDict):
10    state: int
11    inc: int
12
13@type_check_only
14class _PCG64State(TypedDict):
15    bit_generator: str
16    state: _PCG64Internal
17    has_uint32: int
18    uinteger: int
19
20class PCG64(BitGenerator):
21    def __init__(self, seed: _ArrayLikeInt_co | SeedSequence | None = ...) -> None: ...
22    def jumped(self, jumps: int = 1) -> PCG64: ...
23    @property  # type: ignore[override]
24    def state(
25        self,
26    ) -> _PCG64State: ...
27    @state.setter
28    def state(
29        self,
30        value: _PCG64State,
31    ) -> None: ...
32    def advance(self, delta: int) -> PCG64: ...
33
34class PCG64DXSM(BitGenerator):
35    def __init__(self, seed: _ArrayLikeInt_co | SeedSequence | None = ...) -> None: ...
36    def jumped(self, jumps: int = 1) -> PCG64DXSM: ...
37    @property  # type: ignore[override]
38    def state(self) -> _PCG64State: ...
39    @state.setter
40    def state(self, value: _PCG64State) -> None: ...
41    def advance(self, delta: int) -> PCG64DXSM: ...
42