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

Clean code

Browse source
This commit is contained in:
Andika Wasisto 2020-09-07 21:11:39 +07:00
parent 5e9747eabe
commit 743ef277a8
2 changed files with 42 additions and 42 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: class QngWrapperLinux:
def __init__(self): def __init__(self):
self.qwqng_wrapper = cdll.LoadLibrary('./libqwqng-wrapper-x86-64.so' if platform.machine().endswith('64') else './libqwqng-wrapper.so') 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.GetQwqngInstance.restype = c_void_p
self.qwqng_wrapper.RandInt32.argtypes = [c_void_p] self._qwqng_wrapper.RandInt32.argtypes = [c_void_p]
self.qwqng_wrapper.RandInt32.restype = c_int self._qwqng_wrapper.RandInt32.restype = c_int
self.qwqng_wrapper.RandUniform.argtypes = [c_void_p] self._qwqng_wrapper.RandUniform.argtypes = [c_void_p]
self.qwqng_wrapper.RandUniform.restype = c_double self._qwqng_wrapper.RandUniform.restype = c_double
self.qwqng_wrapper.RandNormal.argtypes = [c_void_p] self._qwqng_wrapper.RandNormal.argtypes = [c_void_p]
self.qwqng_wrapper.RandNormal.restype = c_double self._qwqng_wrapper.RandNormal.restype = c_double
self.qwqng_wrapper.RandBytes.argtypes = [c_void_p, c_int] self._qwqng_wrapper.RandBytes.argtypes = [c_void_p, c_int]
self.qwqng_wrapper.RandBytes.restype = POINTER(c_char) self._qwqng_wrapper.RandBytes.restype = POINTER(c_char)
self.qwqng_wrapper.Clear.argtypes = [c_void_p] self._qwqng_wrapper.Clear.argtypes = [c_void_p]
self.qwqng_wrapper.Reset.argtypes = [c_void_p] self._qwqng_wrapper.Reset.argtypes = [c_void_p]
self.qng_pointer = self.qwqng_wrapper.GetQwqngInstance() self._qng_pointer = self._qwqng_wrapper.GetQwqngInstance()
def randint32(self): def randint32(self):
try: try:
return self.qwqng_wrapper.RandInt32(self.qng_pointer) return self._qwqng_wrapper.RandInt32(self._qng_pointer)
except: except:
self.qwqng_wrapper.Reset(self.qng_pointer) self._qwqng_wrapper.Reset(self._qng_pointer)
return self.qwqng_wrapper.RandInt32(self.qng_pointer) return self._qwqng_wrapper.RandInt32(self._qng_pointer)
def randuniform(self): def randuniform(self):
try: try:
return self.qwqng_wrapper.RandUniform(self.qng_pointer) return self._qwqng_wrapper.RandUniform(self._qng_pointer)
except: except:
self.qwqng_wrapper.Reset(self.qng_pointer) self._qwqng_wrapper.Reset(self._qng_pointer)
return self.qwqng_wrapper.RandUniform(self.qng_pointer) return self._qwqng_wrapper.RandUniform(self._qng_pointer)
def randnormal(self): def randnormal(self):
try: try:
return self.qwqng_wrapper.RandNormal(self.qng_pointer) return self._qwqng_wrapper.RandNormal(self._qng_pointer)
except: except:
self.qwqng_wrapper.Reset(self.qng_pointer) self._qwqng_wrapper.Reset(self._qng_pointer)
return self.qwqng_wrapper.RandNormal(self.qng_pointer) return self._qwqng_wrapper.RandNormal(self._qng_pointer)
def randbytes(self, length): def randbytes(self, length):
try: try:
return self._randbytes_arbitrary_length(length) return self._randbytes_arbitrary_length(length)
except: except:
self.qwqng_wrapper.Reset(self.qng_pointer) self._qwqng_wrapper.Reset(self._qng_pointer)
return self._randbytes_arbitrary_length(length) return self._randbytes_arbitrary_length(length)
def _randbytes_arbitrary_length(self, length): def _randbytes_arbitrary_length(self, length):
if length <= 8192: if length <= 8192:
return self.qwqng_wrapper.RandBytes(self.qng_pointer, length)[:length] return self._qwqng_wrapper.RandBytes(self._qng_pointer, length)[:length]
else: else:
data = bytearray() data = bytearray()
for x in range(length // 8192): for x in range(length // 8192):
data.extend(self.qwqng_wrapper.RandBytes(self.qng_pointer, 8192)[:8192]) data.extend(self._qwqng_wrapper.RandBytes(self._qng_pointer, 8192)[:8192])
bytes_needed = length % 8192 bytes_needed = length % 8192
if bytes_needed != 0: if bytes_needed != 0:
data.extend(self.qwqng_wrapper.RandBytes(self.qng_pointer, bytes_needed)[:bytes_needed]) data.extend(self._qwqng_wrapper.RandBytes(self._qng_pointer, bytes_needed)[:bytes_needed])
return bytes(data) return bytes(data)
def clear(self): def clear(self):
self.qwqng_wrapper.Clear(self.qng_pointer) self._qwqng_wrapper.Clear(self._qng_pointer)

30
quanttp/data/qng_wrapper_windows.py
View file

@ -23,47 +23,47 @@ class QngWrapperWindows:
def __init__(self): def __init__(self):
import win32com.client import win32com.client
self.qng = win32com.client.Dispatch("QWQNG.QNG") self._qng = win32com.client.Dispatch("QWQNG.QNG")
def randint32(self): def randint32(self):
try: try:
return self.qng.RandInt32 return self._qng.RandInt32
except: except:
self.qng.Reset() self._qng.Reset()
return self.qng.RandInt32 return self._qng.RandInt32
def randuniform(self): def randuniform(self):
try: try:
return self.qng.RandUniform return self._qng.RandUniform
except: except:
self.qng.Reset() self._qng.Reset()
return self.qng.RandUniform return self._qng.RandUniform
def randnormal(self): def randnormal(self):
try: try:
return self.qng.RandNormal return self._qng.RandNormal
except: except:
self.qng.Reset() self._qng.Reset()
return self.qng.RandNormal return self._qng.RandNormal
def randbytes(self, length): def randbytes(self, length):
try: try:
return self._randbytes_arbitrary_length(length) return self._randbytes_arbitrary_length(length)
except: except:
self.qng.Reset() self._qng.Reset()
return self._randbytes_arbitrary_length(length) return self._randbytes_arbitrary_length(length)
def _randbytes_arbitrary_length(self, length): def _randbytes_arbitrary_length(self, length):
if length <= 8192: if length <= 8192:
return bytes(self.qng.RandBytes(length)) return bytes(self._qng.RandBytes(length))
else: else:
data = bytearray() data = bytearray()
for x in range(length // 8192): for x in range(length // 8192):
data.extend(bytearray(self.qng.RandBytes(8192))) data.extend(bytearray(self._qng.RandBytes(8192)))
bytes_needed = length % 8192 bytes_needed = length % 8192
if bytes_needed != 0: if bytes_needed != 0:
data.extend(bytearray(self.qng.RandBytes(bytes_needed))) data.extend(bytearray(self._qng.RandBytes(bytes_needed)))
return bytes(data) return bytes(data)
def clear(self): def clear(self):
self.qng.Clear() self._qng.Clear()