python mount回调函数_为python回调函数设置argtype

我对Python很在行,所以希望我能正确地表达这个问题。在

整个问题涉及从Python调用C例程。我可以通过把一些相关的问题/答案凑在一起来接近,但我似乎不能把事情安排得很好。有两个方面:第一个是用指针调用C例程,第二个是使用回调函数。在

背景

Rubner提供了一个用C[EMD C code location]编写的EMD例程,他还提供了两个调用EMD例程的示例C程序。我正在尝试开发一个Python例程,作为example2.c的替代,它将调用EMD例程。(是的,我熟悉EMD的OpenCV实现。)

为了方便起见,下面是我想从python调用的emd.c代码的头文件:/* DEFINITIONS */

#define MAX_SIG_SIZE 100

#define MAX_ITERATIONS 500

#define INFINITY 1e20

#define EPSILON 1e-6

/*****************************************************************************/

/* feature_t SHOULD BE MODIFIED BY THE USER TO REFLECT THE FEATURE TYPE */

typedef int feature_t;

/* typedef struct { int X,Y,Z; } feature_t;*/

/*typedef struct { int X; } feature_t; */

/*****************************************************************************/

typedef struct

{

int n; /* Number of features in the signature */

feature_t *Features; /* Pointer to the features vector */

float *Weights; /* Pointer to the weights of the features */

} signature_t;

typedef struct

{

int from; /* Feature number in signature 1 */

int to; /* Feature number in signature 2 */

float amount; /* Amount of flow from "from" to "to" */

} flow_t;

float emd(signature_t *Signature1, signature_t *Signature2,

float (*func)(feature_t *, feature_t *),

flow_t *Flow, int *FlowSize);

#endif

最后,这里是我迄今为止拼凑起来的python代码。我认为(但不确定)我已经正确地设置了结构。(请注意,这是Rubner emd.c代码中可能的功能结构的简化版本。我最终想让整个工作正常,但现在我开始变得简单了。)我遇到的第一个问题是调用函数的argtypes中的某个地方。我尝试过一些变体,但是在网上可以找到的例子非常少,我遇到了困难。在

^{pr2}$

如果有任何关于我哪里出错的建议,我们将不胜感激。在

我肯定会遇到的第二个问题是返回指针的argtypes;这里的任何建议都将受到欢迎。在

提前谢谢。在

------------更新(工作)代码import ctypes

import math

import itertools

MAX_FEATURE_SIZE = 25

FEATURE_t = ctypes.c_int

FEATURE_ptr = ctypes.POINTER(FEATURE_t)

WEIGHT_t = ctypes.c_float

WEIGHT_ptr = ctypes.POINTER(WEIGHT_t)

COUNT_t = ctypes.c_int

COUNT_ptr = ctypes.POINTER(COUNT_t)

class FLOW_t(ctypes.Structure):

_fields_ = [("frm", ctypes.c_int),

("to", ctypes.c_int),

("amount", ctypes.c_float)]

# Note that ctypes.POINTER is compatible with a ctypes array declared

# as TYPE * array_len. This is equivalent to the way we can say 'char

# *foo = "ABCDEF"' in C.

class SIGNATURE_t(ctypes.Structure):

_fields_ = [("N", COUNT_t ),

("feature", FEATURE_ptr),

("weight", WEIGHT_ptr)]

FLOW_ARRAY_t = FLOW_t * (2*MAX_FEATURE_SIZE - 1)

CMPFUNC_t = ctypes.CFUNCTYPE(ctypes.c_float, FEATURE_ptr, FEATURE_ptr)

SIGNATURE_ptr = ctypes.POINTER(SIGNATURE_t)

FLOW_ptr = ctypes.POINTER(FLOW_t)

# Convenience function - keeps us from having to remember all the types and parameters later on

def make_signature(features, weights):

sig = SIGNATURE_t()

sig.N = len(features)

sig.feature = (len(features) * FEATURE_t)(*features)

sig.weight = (len(weights) * WEIGHT_t)(*weights)

return sig

# We want to pass into C a custom distance function from Python

def py_dist_func(f1,f2):

# print "f1, f2: %d, %d" % ( f1[0], f2[0] )

d= distance(f1[0],f2[0])

return d

# set this up as a holder for distance function between any two n-D points

def distance(p0,p1):

return(math.fabs(p0-p1))

dist_callback = CMPFUNC_t(py_dist_func)

#print "Importing emdlib"

emdlib = ctypes.CDLL('emdlib.dylib')

#print "Setting argtypes"

emdlib.emd.argtypes = [ SIGNATURE_ptr,

SIGNATURE_ptr,

CMPFUNC_t,

FLOW_ptr,

COUNT_ptr ]

#print "Setting restype"

emdlib.emd.restype = ctypes.c_float

feature1 = [0, 1,2,3,4,5,6,7,8]

feature2 = [0, 1,2,3,4,5,6,7,8]

weight1 = [0.275,0.296,0.002,0.131,0.208,0.048,0.058,0.098,0.455]

weight2 = [0.285,0.421,0.028,0.021,0.240,0.166,0.023,0.054,0.469]

#print "Creating signatures"

signature1 = make_signature(feature1, weight1)

signature2 = make_signature(feature2, weight2)

flow_array = FLOW_ARRAY_t()

flow_size = COUNT_t()

#print "Calling EMD"

e = emdlib.emd(ctypes.byref(signature1),

ctypes.byref(signature2),

dist_callback,

flow_array,

ctypes.byref(flow_size))

print "EMD= ", e

print "Number of FlowS", flow_size.value

print "Flow"

print "from to amount"

totalFlow=0.0

for i in range(0,flow_size.value):

# print "Flow from %d to %d amount :%f" %(flow_array[i].frm, flow_array[i].to, flow_array[i].amount)

print " %d %d %f" %(flow_array[i].frm, flow_array[i].to, flow_array[i].amount)

totalFlow=totalFlow+flow_array[i].amount

#

# now adjust EMD to account for different signature masses and make it a metric

alpha=1.0

mass1=sum(weight1)

mass2=sum(weight2)

fList=[feature1,feature2]

max_distance= 0.0

for p0, p1 in list(itertools.product(*fList)):

# print p0,p1, distance(p0,p1), max_distance

max_distance = max(max_distance, distance(p0, p1))

print "\nMax distance= %f" % max_distance

print "Total Source = %f" % mass1

print "Total Demand = %f" % mass2

print "Total Flow= %f\n " % totalFlow

print "Alpha = %f\n" %alpha

# emdHat = e*totalFlow+math.sqrt((mass1-mass2)*(mass1-mass2))*alpha*max_distance

emdHat = e*totalFlow+math.fabs((mass1-mass2))*alpha*max_distance

print "Corrected Earth Movers Distance \n"

print "emdHat = %f\n" % emdHat;

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值