scipy.weave.inline与数学库无法正常工作

时间:2012-04-14 22:49:30

标签: python numpy scipy

我使用scipy.weave.inline时遇到问题。我想编写一个以lcenter为中心的unittep函数,并使用width_nm。我有两个版本:python版本,名为pm和一个名为pm_weave的优化函数,但它看起来像abs不能正常工作。请参阅下面的代码。如果你运行它,你会得到一个大小为1的窗口,无论输入是什么,所以它看起来像abs不起作用。例如,如果删除abs,它的工作方式与预期完全相同

我该如何解决这个问题?

def pm_weave(w,lcenter,width_nm):
  """ Return a unitstep function that is centered around lcenter with height 1.0 and width width_nm """
  lcenter = float(lcenter)
  w = float(w)
  width_nm = float(width_nm)
  code = """
  #include <math.h>
  float wl = 1.88495559215387594307758602997E3/w;

  if(abs(lcenter-wl) < width_nm) {
     return_val = 1.0;
     }
     else {
     return_val = 0.0;
     }
  """
  res = weave.inline(code,['w','lcenter','width_nm'],
                     type_converters = weave.converters.blitz,
                     compiler = "gcc", headers=["<math.h>"]
                    )
  return res



def pm(w,lcenter,width_nm):
  """ 
  Return a unitstep function centered around lcenter [nm] with width width_nm. w
  should be a radial frequency. 
  """
  return abs(600*np.pi/w - lcenter) < width_nm/2. and 1. or 0.



plot(wavelength_list,map(lambda w:pm(toRadialFrequency(w),778,1),wavelength_list),label="Desired behaviour")
plot(wavelength_list,map(lambda w:pm_weave(toRadialFrequency(w),778,1),wavelength_list),'^',label="weave.inline behaviour")
ylim(0,1.5)
show()

1 个答案:

答案 0 :(得分:2)

我认为您可能需要在C代码中使用fabs()而不是abs()abs()将截断结果,而fabs()将用于浮点运算。