嵌套模板 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]]

我希望能够使用跨 A 的内部类型模板化的流运算符将其读入 STL vector<vector<double> >：

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

当向量没有嵌套时，我已经找到了一种方法，即。一个简单的vector<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;
}

但我对istream& operator>>(istream& s, vector<vector<T> >&A) 部分有疑问，因为我似乎无法正确捕捉到内部]。我确信 Boost 有办法做到这一点，但我希望看到 STL 的解决方案用于教学目的。

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

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

Answer 1

实际上，当T 时，您希望对两者使用相同功能的代码的问题是：

• vector<double>
• 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;
}

希望这对某人有用。