如何初始化并返回c ++头文件中的数组?

时间:2013-06-29 12:16:28

标签: c++ return-value header-files

我对头文件不是很好,但我想使用头文件从文件中读取数据并将数据作为主cpp文件中的向量返回。

这是我的readposcar.h文件:

#include <fstream>
#include <sstream>
#include <vector>

using namespace std;

int add(void) {

    double a1x, a1y, a1z, a2x, a2y, a2z, a3x, a3y, a3z; // I want all this stuff in vector form
    int i;
    double scale;
    string line; stringstream dum;

    ifstream poscar ("POSCAR");
    for (i=1; i<=5; i++) {
        getline(poscar,line);
        if (i==2) {stringstream dum(line); dum >> scale;}
        if (i==3) {stringstream dum(line); dum >> a1x >> a1y >> a1z;}
        if (i==4) {stringstream dum(line); dum >> a2x >> a2y >> a2z;}
        if (i==5) {stringstream dum(line); dum >> a3x >> a3y >> a3z;}
    }

    vector<double> myvec(3);
    myvec[0] = a1x;
    myvec[1] = a1y;
    myvec[2] = a1z;
    return myvec;
}

这是我的.cpp文件:

#include <iostream>
#include <fstream>

#include "readposcar.h"

using namespace std;

int main(void) {
    int nbasis = 2;
    int nkpts = 10;
    vector<double> myvec2(3);
    myvec2 = add();
    cout << "No. of k-points: " << nkpts << endl;
    return 0;
}

这显然不起作用。有人可以告诉我有什么问题以及我需要做些什么才能让它发挥作用?如果我在.h文件中返回说myvec [2]而不是整个数组,我只能让它工作。

如果向量不起作用,我不介意将它作为数组。是否可以将头文件中的数组初始化为一种全局数组,然后只需在.cpp文件中调用它?

以下是我得到的错误:

main.cpp中包含的文件:4:0:

readposcar.h: In function ‘int add()’:
readposcar.h:27:9: error: cannot convert ‘std::vector<double>’ to ‘int’ in return
main.cpp: In function ‘int main()’:
main.cpp:12:15: error: no match for ‘operator=’ in ‘myvec2 = add()’

3 个答案:

答案 0 :(得分:2)

您应该将add的返回类型从int更改为vector<double>

答案 1 :(得分:1)

您没有返回正确的类型。尝试:

vector<double> add() {
   ...
   return myvec;
}

但是,我个人会在调用者范围内传递对vector的引用并返回布尔成功(可选):

bool add(vector<double> &myvec) {
   ...
   return true;
}

因为这可以避免复制可能昂贵的vector,除非C ++编译器能够使用RVO来优化复制操作,在这种情况下你可以使用前一种方法语义。

(感谢@aryjczyk和@AlexB指出最后一点)。

答案 2 :(得分:0)

  1. 在调用getline()后解析该行。
  2. 将每个已解析的值转换为double。
  3. 在vector上调用push_back来添加double。
  4. 另外,请考虑传递对vector的引用。

    因此,您的函数的签名将更改为:

    int add( std::vector<double> & values )
    

    这样,从函数返回时,您将避免不必要的复制。