24 lines
542 B
Python
24 lines
542 B
Python
|
import ctypes
|
||
|
|
import os
|
||
|
|
|
||
|
|
so = os.path.join(os.path.dirname(__file__), 'maths.so')
|
||
|
|
lib = ctypes.CDLL(so)
|
||
|
|
|
||
|
|
lib.relax.argtypes = [
|
||
|
|
ctypes.c_void_p,
|
||
|
|
ctypes.c_void_p,
|
||
|
|
ctypes.c_void_p,
|
||
|
|
ctypes.c_void_p,
|
||
|
|
ctypes.c_void_p,
|
||
|
|
ctypes.c_void_p,
|
||
|
|
ctypes.c_void_p,
|
||
|
|
ctypes.c_int,
|
||
|
|
ctypes.c_int,
|
||
|
|
]
|
||
|
|
|
||
|
|
def relax(pos, links, mrel1, mrel2, lengths, push, pull, iters):
|
||
|
|
nlinks = links.shape[0]
|
||
|
|
lib.relax(pos.ctypes, links.ctypes, mrel1.ctypes, mrel2.ctypes, lengths.ctypes, push.ctypes, pull.ctypes, nlinks, iters)
|
||
|
|
|
||
|
|
|