打开文件的I/O和使用命令行重定向

Question

我正在写一个程序，它从文本文件（C++代码）中获取输入并将其修改为在文本文件（html）中输出。该程序需要从标准输入读取并写入标准输出。它使用命令行参数-i文件名输入和-o文件名输出以及shell输入/输出重定向。我对Linux相当陌生，不确定如何以有效的方式做到这一点。这是我到目前为止：

#include <fstream> 
#include <iostream> 
#include <cstdlib> 
#include <vector> 
#include <string.h> 
#include <string> 

using namespace std; 

int main(int argc, char * argv[]) { 
    vector<string> mod; 
    int modNum = 0; 
    int i = 1; 
    string input = ""; 
    string inFilename = ""; 
    string outFilename = ""; 
    ifstream inFile; 
    ofstream outFile; 

    while(i < argc){ 
     if (strcmp(argv[i],"-i") == 0) { 
       i++; 
       if (i<argc){ 
        inFilename = argv[i]; 
       } else { 
        cout << "\t-i require a filename for input" << endl; 
       } 
       i++;     
     } else if (strcmp(argv[i],"-o") == 0) { 
       i++; 
       if (i<argc){ 
        outFilename = argv[i]; 
       } else { 
        cout << "\t-i require a filename for output" << endl; 
       } 
       i++; 
     } 
    } 
    if (inFilename != "") { 
     inFile.open(inFilename.c_str()); 
    }  
    if (outFilename != "") { 
     outFile.open(outFilename.c_str()); 
    }  
    if (inFile.is_open() and outFile.is_open()) { 
     outFile << "<script src=\"https://cdn.rawgit.com/google/code-prettify/master/loader/run_prettify.js\"></script>" << endl; 
     outFile << "<pre class=\"prettyprint\">" << endl; 
     while (!inFile.eof()) { 
      getline(inFile, input); 
      outFile << input << endl;    
     } 
     outFile << "</pre>"; 
    } else { 
     cout << "<script src=\"https://cdn.rawgit.com/google/code-prettify/master/loader/run_prettify.js\"></script>" << endl; 
     cout << "<pre class=\"prettyprint\">" << endl; 
     while (cin) { 
      if (getline(cin,input)) { 
       cout << input << endl; 
      } 
     } 
     cout << "</pre>"; 
    } 
    inFile.close(); 
    outFile.close(); 
} 

我遇到的问题是没有这种方法的多功能性。如果我得到

./c < code.cpp -o page.html 

或

./c -i code.cpp > page.html 

程序将无法正常执行。我道歉，如果有任何小错误，我复制并粘贴代码块只显示必需品。该程序将做更多，但现在我只是想让输入/输出正常工作。

Answer 1

好，到目前为止，你已经处理只有两个四例运行的程序：你的代码假定如果都-i和-o指定的，那么它从输入文件和打印输出文件读取，否则读取从输入和写入到输出。所以，当然，你的程序不能读取文件并写入标准输出（反之亦然） - 代码只是不在这里。

我建议你在一个单独的函数中提取处理文件的代码，它需要istream& input和ostream& output作为参数以避免代码重复。你会用不同的参数来调用它，这取决于你所遇到的四种情况，是这样的：

process_file(inFile.is_open() ? inFile : cin, outFile.is_open() ? outFile : cout); 

这将让你进一步纠正代码，但它仍然不理想。我建议去codereview网站获得更多的反馈。