| 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_name='MT19937'): | |
| """ | |
| Pickling helper function that returns a bit generator object | |
| Parameters | |
| ---------- | |
| bit_generator_name : str | |
| String containing the name of the BitGenerator | |
| Returns | |
| ------- | |
| bit_generator : BitGenerator | |
| BitGenerator instance | |
| """ | |
| if bit_generator_name in BitGenerators: | |
| bit_generator = BitGenerators[bit_generator_name] | |
| else: | |
| raise ValueError(str(bit_generator_name) + ' is not a known ' | |
| 'BitGenerator module.') | |
| return bit_generator() | |
| 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 | |
| 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 | |
| ------- | |
| rg : Generator | |
| Generator using the named core BitGenerator | |
| """ | |
| 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 | |
| """ | |
| return RandomState(bit_generator_ctor(bit_generator_name)) | |