嵌套模板 C++ 的重载流运算符STL

Question

我有一个数据文件，它是由一系列嵌套的双精度数组成的单行，例如。

[[0.127279,0.763675,0.636396],[0.254558,0.890955,0.636396],
[0.127279,0.636396,0.763675],[0.254558,0.763675,0.763675],
[0.381838,0.890955,0.763675],[0.127279,0.509117,0.890955],
[0.254558,0.636396,0.890955],[0.509117,0.890955,0.890955]]

我希望能够将其读入 STL vector > 使用流运算符，该运算符是在 A 的内部类型中模板化的：

vector<vector<double> > A;
FIN >> A;

我已经找到了一种在向量未嵌套时执行此操作的方法，即。一个简单的 vector 如下：

template <class T>
istream& operator>>(istream& s, vector<T> &A){
  T x;
  string token; char blank;

  s >> blank; // Gobble the first '['
  while( getline(s, token, ',') ) {
   istringstream input(token);
   input >> x;
   A.push_back(x);
  }
  s >> blank; // Gobble the last ']'
  return s;
}

但是我在使用 istream& 时遇到了问题。 operator>>(istream& s, vector >&A) 部分，因为我似乎无法正确捕获内部 ] 。我确信 Boost 有办法做到这一点，但我希望看到一个用于教学目的的 STL 解决方案。

注意：我知道重载 vector 的流运算符可能会产生深远的不良后果，并且该实现应该包含在它自己的类中 - 我正在使用上面的示例就这样澄清了这个问题。

编辑： 我希望该方法足够强大，能够处理输入数组，其大小（和内部数组）大小事先不知道，而是从读取流中推断出来。

Answer 1

实际上，您的代码的问题是您希望对两者使用相同的函数，当 T 为：

• vector
• double

但是需要将数据读入vector和double的逻辑略有不同。所以你不能这样做，至少不能用那么简单的逻辑：

我宁愿编写两个函数来分别处理这两种情况。毕竟，即使在您的情况下，编译器也会为 T 的每个值生成两个不同的函数。

template <class T>
istream& operator>>(istream& s, vector<T> &A)
{
  T x;
  string token; char blank;

  s >> blank; // Gobble the first '['
  while( getline(s, token, ',') ) 
  {
   istringstream input(token);
   input >> x;
   A.push_back(x);
  }
// s >> blank; // Gobble the last ']'
  return s;
}

template <class T>
istream& operator>>(istream& s, vector<vector<T>> &A)
{
  vector<T> x;
  string token; 
  char blank;

  s >> blank; // Gobble the first '['
  while( getline(s, token, ']') ) 
  {
   istringstream input(token);
   input >> x;
   s >> blank; //read , after [...]
   A.push_back(x);
   x.clear();
  }
  s >> blank; // Gobble the last ']'
  return s;
}

测试代码：

int main() {
        vector<vector<double>> A;       
        cin >> A;
        for(size_t i = 0 ;i < A.size(); ++i)
        {
            for(size_t j = 0 ; j < A[i].size(); ++j)
                 cout << A[i][j] <<"   ";
            cout << endl;
        }
        return 0;
}

输入：

[[1,2,3],[4,5,6],
[7,8,9],[10,11,12],
[13,14,15],[16,17,18],
[19,20,21],[22,23,24]]

输出：

1   2   3   
4   5   6   
7   8   9   
10   11   12   
13   14   15   
16   17   18   
19   20   21   
22   23   24 

在线演示： http://ideone.com/iBbmw

Answer 2

在您的特定示例中，这非常简单。

• 将整行读入字符串。
• 将所有 [ , ] 和 , 替换为空白字符。
• 使用空格替换字符串创建一个简单的字符串流。

现在，在读取三个双精度数后，您可以有一个简单的循环

double x;
while( stringstreamp >> x )
{
}

和一些特殊的逻辑，将它们插入到一个新数组中。

Answer 3

几年后，我在这里为同样的问题而苦苦挣扎。

根据您的贡献，我开发了原始模板的修改版本。它能够解析多维数组，即使它们分布在几行中。

template <class T>
istream& operator>>(istream& s, vector<T> &A){
  while(true){
    T x;
    char c = s.peek();

    if( c == '[' || c == ','){ 
        s.ignore();
        if(s >> x) A.push_back(x);
        else throw invalid_argument("Bad, bad JS array!");
        continue;
    }

    if( c == ']') {
        s.ignore();
        return s;
    }

    /* Ignore line-break */
    s.ignore();     
  }
  return s;
}

希望这对某人有用。