mirror of
https://github.com/Ed94/Odin.git
synced 2026-06-20 12:44:59 -07:00
Jeroen van Rijn
79 lines
1.8 KiB
Python
79 lines
1.8 KiB
Python
from math import *
|
|
from ctypes import *
|
|
import os
|
|
|
|
#
|
|
# Where is the DLL? If missing, build using: `odin build . -build-mode:dll`
|
|
#
|
|
LIB_PATH = os.getcwd() + os.sep + "big.dll"
|
|
|
|
#
|
|
# Result values will be passed in a struct { res: cstring, err: Error }
|
|
#
|
|
class Res(Structure):
|
|
_fields_ = [("res", c_char_p), ("err", c_byte)]
|
|
|
|
#
|
|
# Error enum values
|
|
#
|
|
E_None = 0
|
|
E_Out_Of_Memory = 1
|
|
E_Invalid_Pointer = 2
|
|
E_Invalid_Argument = 3
|
|
E_Unknown_Error = 4
|
|
E_Max_Iterations_Reached = 5
|
|
E_Buffer_Overflow = 6
|
|
E_Integer_Overflow = 7
|
|
E_Division_by_Zero = 8
|
|
E_Math_Domain_Error = 9
|
|
E_Unimplemented = 127
|
|
|
|
#
|
|
# Set up exported procedures
|
|
#
|
|
|
|
try:
|
|
l = cdll.LoadLibrary(LIB_PATH)
|
|
except:
|
|
print("Couldn't find or load " + LIB_PATH + ".")
|
|
exit(1)
|
|
|
|
try:
|
|
l.test_add_two.argtypes = [c_char_p, c_char_p, c_longlong]
|
|
l.test_add_two.restype = Res
|
|
except:
|
|
print("Couldn't find exported function 'test_add_two'")
|
|
exit(2)
|
|
|
|
add_two = l.test_add_two
|
|
|
|
try:
|
|
l.test_error_string.argtypes = [c_byte]
|
|
l.test_error_string.restype = c_char_p
|
|
except:
|
|
print("Couldn't find exported function 'test_error_string'")
|
|
exit(2)
|
|
|
|
def error(res: Res, param=[]):
|
|
if res.err != E_None:
|
|
error_type = l.test_error_string(res.err).decode('utf-8')
|
|
error_loc = res.res.decode('utf-8')
|
|
|
|
error_string = "'{}' error in '{}'".format(error_type, error_loc)
|
|
if len(param):
|
|
error_string += " with params {}".format(param)
|
|
|
|
print(error_string, flush=True)
|
|
os._exit(res.err)
|
|
|
|
|
|
def test_add_two(a = 0, b = 0, radix = 10):
|
|
res = add_two(str(a).encode('utf-8'), str(b).encode('utf-8'), radix)
|
|
error(res, [str(a), str(b), radix])
|
|
|
|
if __name__ == '__main__':
|
|
print("---- core:math/big tests ----")
|
|
print()
|
|
|
|
test_add_two(1234, 5432, 10)
|
|
test_add_two(1234, 5432, 110) |