Question

我试图通过一个函数使用boost来创建我的数组。我写了下面的代码，但它不起作用。我怎么能这样做，我的数组是由我的函数读取的？在此代码的先前版本中，我刚刚将我的数组定义为双U [N + 1] [4]。它起作用了。使用boost时我做错了什么？

#include "boost/multi_array.hpp"
#include <cassert>
#include <iostream>
#include <cmath>
#include <time.h>

int const N=400;
int const x1=0;
int const x2=1;
int const gammaValue=1.4;

void Output(double U[N][3])
{
    double x,dx;
    dx=(double(x2-x1))/double(N);
    FILE *fp;
    double rho,u,p;
    fp=fopen("result.dat","w+");
    fprintf(fp,"%2.30s\n %20.60s\n %20.18s\t %2.3d\t %2.18s\t ","TITLE = \"1D-EULER.dat \"","variables = \"x\", \"rho\", \"u\", \"p\"","zone i=",N+1,"f=point\n");


    for(int n=0;n<N+1;n++)
    {
        x=x1+n*dx;
        rho=U[n][0];
        u=U[n][1]/U[n][0];
        p=(gammaValue-1)*(U[n][2]-0.5*U[n][0]*u*u);
        fprintf(fp,"%20.5f\t%20.5f\t%20.5f\t%20.5f\n",x,rho,u,p);
    }
    fclose(fp);
}

int main () {
    // 3 x 4 x 2
    typedef boost::multi_array<double, 2> array_type;
    typedef array_type::index index;
    array_type A(boost::extents[N][3]);

    int values = 0;
    for(index i = 0; i != N; ++i) {
        for(index j = 0; j != 3; ++j){
            A[i][j] = i+j;
        }
    }

    Output(A);
    return 0;
}

Answer 1

使用multi_array的成员函数data()，它返回一个指向包含数组数据的连续块开头的指针。一旦获得第一个元素的地址，就可以获得其他元素，因为您已经知道了数组的维度。

    double * p = A.data();
    double (*array)[3] = (double (*)[3])p;
    Output(array);

Answer 2

为什么不改变输出功能的签名？将array_type typedef移动到文件顶部，然后将Output函数更改为：

void Output( array_type& U )

或者，更好的是，制作参数const。这是一些代码：

#include <boost/multi_array.hpp>                                                                                                                                                                                                              
#include <iostream>                                                                                                                                                                                                                           

typedef boost::multi_array< double, 2 > array_type;                                                                                                                                                                                           

void Output( const array_type& arr )                                                                                                                                                                                                          
{                                                                                                                                                                                                                                             
    std::cout << "here" << std::endl;                                                                                                                                                                                                         
}                                                                                                                                                                                                                                             

int main( int argc, char ** argv )                                                                                                                                                                                                            
{                                                                                                                                                                                                                                             
    array_type arr;                                                                                                                                                                                                                           
    Output( arr );                                                                                                                                                                                                                            
    return 0;                                                                                                                                                                                                                                 
}

除了演示适当更改函数签名的原则之外，它什么也没做。

如何将Boost multi_array传递给函数？

2 个答案: