如何将 std::valarray<double> 与 gsl 集成?

How to integrate std::valarray<double> with gsl?

我对 C++ 比较陌生,但我有一些(稀缺的)编码和数值经验。

我知道这个问题时不时地被发布,你如何整合一个数组。在 MATLAB 中,你可以强制你的数组成为一个函数(我忘了怎么做,但我知道我以前做过)并将它发送给内置的积分器,所以我的问题是你如何在 C++ 中做到这一点。

我有这个积分:

I = integral(A(z)*sin(qz)*dz)

q 只是 double const,z 是积分变量,但 A(z) 是一个数组(从现在起我将称之为 actualfunction),在我的代码中具有与 z 轴相同数量的点。积分边界是 z[0] 和 z[nz-1].

我使用梯形规则计算了这个积分,对于 5000 点的 z 轴,这需要 0.06 秒。我的问题是这个计算大约发生了 300 * 30 * 20 次(我有 3 个 for 循环),这个 0.06 秒很快增长到模拟的 3 小时。我的代码的整个瓶颈就是这个集成(我显然可以通过减少 z 来加快速度,但这不是重点。)

我知道库函数通常比用户编写的要好得多。我也知道我不能使用像辛普森规则这样更简单的东西,因为被积函数是高度振荡的,我想避免自己实现一些复杂的数值算法。

GSL 需要以下形式的函数:

F = f(double x, void *params)

我或许可以使用 gsl 中的 QAWO 自适应集成,但是如何将我的函数制作成可以将我的数组转换为函数的形式?

我在想:

F(double z, void *params)
{
  std::valarray<double> actualfunction = *(std::valarray<double> *) params;
  double dz = *(double *) params; // Pretty sure this is wrong
  unsigned int actual_index = z / dz; // crazy assumption (my z[0] was 0)
  return actualfunction[actual_index];
 }

这样的事情可能吗?我怀疑数值算法是否会使用与 actualfunction 相同的空间差异,我是否应该以某种方式对 actualfunction 进行插值或其他操作?

还有比gsl更好的吗?

template<class F>
struct func_ptr_helper {
  F f;
  void* pvoid(){ return std::addressof(f); }
  template<class R, class...Args>
  using before_ptr=R(*)(void*,Args...);
  template<class R, class...Args>
  using after_ptr=R(*)(Args...,void*);
  template<class R, class...Args>
  static before_ptr<R,Args...> before_func() {
    return [](void* p, Args...args)->R{
      return (*static_cast<F*>(p))(std::forward<Args>(args)...);
    };
  }
  template<class R, class...Args>
  static after_ptr<R,Args...> after_func() {
    return [](Args...args, void* p)->R{
      return (*static_cast<F*>(p))(std::forward<Args>(args)...);
    };
  }
};
template<class F>
func_ptr_helper<F> lambda_to_pfunc( F f ){ return {std::move(f)}; }

使用:

auto f = lambda_to_pfunc([&actualfunction, &dz](double z){
  unsigned int actual_index = z / dz; // crazy assumption (my z[0] was 0)
  return actualfunction[actual_index];
});

然后

void* pvoid - f.pvoid();
void(*pfun)(double, void*) = f.after_func();

你可以通过pfunpvoid

对于任何打字错误,我们深表歉意。

我们的想法是编写一个执行我们想要的操作的 lambda。然后 lambda_to_pfunc 将其包装起来,以便我们可以将其作为 void* 和指向 C 风格 API 的函数指针传递。

当然,您必须妥善管理所有内容的生命周期。