コード例 #1
0
ファイル: TPS_wrapper.py プロジェクト: sapresearch/Tomo
class TPScpp(object):
    
    def __init__(self, pS, pD):
        self.TPScpp = WinDLL('CThinPlateSpline.dll')
        self.TPScpp.new_CThinPlateSpline.argtypes = [[Point],[Point]]
        self.TPScpp.new_CThinPlateSpline.restype = ctypes.c_void_p

        self.TPScpp.warpImage.argtypes = [ndpointer(np.float32, flags="C_CONTIGUOUS"), ndpointer(np.float32, flags="C_CONTIGUOUS")]
        self.TPScpp.warpImage.restype = ctypes.c_void_p

        self.obj = self.TPScpp.new_CThinPlateSpline(pS, pD)

    def warp(self, a, b):
        self.TPScpp.warpImage(self.obj, src, dst)

    def delete(): 
        self.TPScpp.del_CThinPlateSpline()