randogoth/quanttp2
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update libqwqng-wrapper

Browse source
This commit is contained in:
Andika Wasisto 2020-09-04 21:02:23 +07:00
parent 0936dc353e
commit 5e1407ec4b
3 changed files with 27 additions and 27 deletions
Download patch file Download diff file Expand all files Collapse all files

54
quanttp/data/qng_wrapper_linux.py
View file

@ -25,59 +25,59 @@ from ctypes import cdll, c_void_p, c_int, c_double, POINTER, c_char
class QngWrapperLinux:
def __init__(self):
self.libqwqng_wrapper = cdll.LoadLibrary('./libqwqng-wrapper-x86-64.so' if platform.machine().endswith('64') else './libqwqng-wrapper.so')
self.libqwqng_wrapper.GetQwqngInstance.restype = c_void_p
self.libqwqng_wrapper.RandInt32.argtypes = [c_void_p]
self.libqwqng_wrapper.RandInt32.restype = c_int
self.libqwqng_wrapper.RandUniform.argtypes = [c_void_p]
self.libqwqng_wrapper.RandUniform.restype = c_double
self.libqwqng_wrapper.RandNormal.argtypes = [c_void_p]
self.libqwqng_wrapper.RandNormal.restype = c_double
self.libqwqng_wrapper.RandBytes.argtypes = [c_void_p, c_int]
self.libqwqng_wrapper.RandBytes.restype = POINTER(c_char)
self.libqwqng_wrapper.Clear.argtypes = [c_void_p]
self.libqwqng_wrapper.Reset.argtypes = [c_void_p]
self.qng = self.libqwqng_wrapper.GetQwqngInstance()
self.qwqng_wrapper = cdll.LoadLibrary('./libqwqng-wrapper-x86-64.so' if platform.machine().endswith('64') else './libqwqng-wrapper.so')
self.qwqng_wrapper.GetQwqngInstance.restype = c_void_p
self.qwqng_wrapper.RandInt32.argtypes = [c_void_p]
self.qwqng_wrapper.RandInt32.restype = c_int
self.qwqng_wrapper.RandUniform.argtypes = [c_void_p]
self.qwqng_wrapper.RandUniform.restype = c_double
self.qwqng_wrapper.RandNormal.argtypes = [c_void_p]
self.qwqng_wrapper.RandNormal.restype = c_double
self.qwqng_wrapper.RandBytes.argtypes = [c_void_p, c_int]
self.qwqng_wrapper.RandBytes.restype = POINTER(c_char)
self.qwqng_wrapper.Clear.argtypes = [c_void_p]
self.qwqng_wrapper.Reset.argtypes = [c_void_p]
self.qng_pointer = self.qwqng_wrapper.GetQwqngInstance()
def randint32(self):
try:
return self.libqwqng_wrapper.RandInt32(self.qng)
return self.qwqng_wrapper.RandInt32(self.qng_pointer)
except:
self.libqwqng_wrapper.Reset(self.qng)
return self.libqwqng_wrapper.RandInt32(self.qng)
self.qwqng_wrapper.Reset(self.qng_pointer)
return self.qwqng_wrapper.RandInt32(self.qng_pointer)
def randuniform(self):
try:
return self.libqwqng_wrapper.RandUniform(self.qng)
return self.qwqng_wrapper.RandUniform(self.qng_pointer)
except:
self.libqwqng_wrapper.Reset(self.qng)
return self.libqwqng_wrapper.RandUniform(self.qng)
self.qwqng_wrapper.Reset(self.qng_pointer)
return self.qwqng_wrapper.RandUniform(self.qng_pointer)
def randnormal(self):
try:
return self.libqwqng_wrapper.RandNormal(self.qng)
return self.qwqng_wrapper.RandNormal(self.qng_pointer)
except:
self.libqwqng_wrapper.Reset(self.qng)
return self.libqwqng_wrapper.RandNormal(self.qng)
self.qwqng_wrapper.Reset(self.qng_pointer)
return self.qwqng_wrapper.RandNormal(self.qng_pointer)
def randbytes(self, length):
try:
return self._randbytes_arbitrary_length(length)
except:
self.libqwqng_wrapper.Reset(self.qng)
self.qwqng_wrapper.Reset(self.qng_pointer)
return self._randbytes_arbitrary_length(length)
def _randbytes_arbitrary_length(self, length):
if length <= 8192:
return self.libqwqng_wrapper.RandBytes(self.qng, length)[:length]
return self.qwqng_wrapper.RandBytes(self.qng_pointer, length)[:length]
else:
data = bytearray()
for x in range(length // 8192):
data.extend(self.libqwqng_wrapper.RandBytes(self.qng, 8192)[:8192])
data.extend(self.qwqng_wrapper.RandBytes(self.qng_pointer, 8192)[:8192])
bytes_needed = length % 8192
if bytes_needed != 0:
data.extend(self.libqwqng_wrapper.RandBytes(self.qng, bytes_needed)[:bytes_needed])
data.extend(self.qwqng_wrapper.RandBytes(self.qng_pointer, bytes_needed)[:bytes_needed])
return bytes(data)
def clear(self):
self.libqwqng_wrapper.Clear(self.qng)
self.qwqng_wrapper.Clear(self.qng_pointer)