| from .bit_generator import BitGenerator
|
| from .mtrand import RandomState
|
| from ._philox import Philox
|
| from ._pcg64 import PCG64, PCG64DXSM
|
| from ._sfc64 import SFC64
|
|
|
| from ._generator import Generator
|
| from ._mt19937 import MT19937
|
|
|
| BitGenerators = {'MT19937': MT19937,
|
| 'PCG64': PCG64,
|
| 'PCG64DXSM': PCG64DXSM,
|
| 'Philox': Philox,
|
| 'SFC64': SFC64,
|
| }
|
|
|
|
|
| def __bit_generator_ctor(bit_generator: str | type[BitGenerator] = 'MT19937'):
|
| """
|
| Pickling helper function that returns a bit generator object
|
|
|
| Parameters
|
| ----------
|
| bit_generator : type[BitGenerator] or str
|
| BitGenerator class or string containing the name of the BitGenerator
|
|
|
| Returns
|
| -------
|
| BitGenerator
|
| BitGenerator instance
|
| """
|
| if isinstance(bit_generator, type):
|
| bit_gen_class = bit_generator
|
| elif bit_generator in BitGenerators:
|
| bit_gen_class = BitGenerators[bit_generator]
|
| else:
|
| raise ValueError(
|
| str(bit_generator) + ' is not a known BitGenerator module.'
|
| )
|
|
|
| return bit_gen_class()
|
|
|
|
|
| def __generator_ctor(bit_generator_name="MT19937",
|
| bit_generator_ctor=__bit_generator_ctor):
|
| """
|
| Pickling helper function that returns a Generator object
|
|
|
| Parameters
|
| ----------
|
| bit_generator_name : str or BitGenerator
|
| String containing the core BitGenerator's name or a
|
| BitGenerator instance
|
| bit_generator_ctor : callable, optional
|
| Callable function that takes bit_generator_name as its only argument
|
| and returns an instantized bit generator.
|
|
|
| Returns
|
| -------
|
| rg : Generator
|
| Generator using the named core BitGenerator
|
| """
|
| if isinstance(bit_generator_name, BitGenerator):
|
| return Generator(bit_generator_name)
|
|
|
| return Generator(bit_generator_ctor(bit_generator_name))
|
|
|
|
|
| def __randomstate_ctor(bit_generator_name="MT19937",
|
| bit_generator_ctor=__bit_generator_ctor):
|
| """
|
| Pickling helper function that returns a legacy RandomState-like object
|
|
|
| Parameters
|
| ----------
|
| bit_generator_name : str
|
| String containing the core BitGenerator's name
|
| bit_generator_ctor : callable, optional
|
| Callable function that takes bit_generator_name as its only argument
|
| and returns an instantized bit generator.
|
|
|
| Returns
|
| -------
|
| rs : RandomState
|
| Legacy RandomState using the named core BitGenerator
|
| """
|
| if isinstance(bit_generator_name, BitGenerator):
|
| return RandomState(bit_generator_name)
|
| return RandomState(bit_generator_ctor(bit_generator_name))
|
|
|
