main.cpp: added command-option "-is" to specify usage of std::istream interface in reading initial file

Author: firewave

main.cpp

@@ -25,6 +25,7 @@
 int main(int argc, char **argv)
 {
     const char *filename = nullptr;
+    bool use_istream = false;
 
     // Settings..
     simplecpp::DUI dui;
@@ -49,6 +50,8 @@ int main(int argc, char **argv)
             case 'i':
                 if (std::strncmp(arg, "-include=",9)==0)
                     dui.includes.push_back(arg+9);
+                else if (std::strncmp(arg, "-is",3)==0)
+                    use_istream = true;
                 break;
             case 's':
                 if (std::strncmp(arg, "-std=",5)==0)
@@ -72,20 +75,29 @@ int main(int argc, char **argv)
         std::cout << "  -UNAME          Undefine NAME." << std::endl;
         std::cout << "  -std=STD        Specify standard." << std::endl;
         std::cout << "  -q              Quiet mode (no output)." << std::endl;
+        std::cout << "  -is             Use std::istream interface." << std::endl;
         std::exit(0);
     }
 
     // Perform preprocessing
     simplecpp::OutputList outputList;
     std::vector<std::string> files;
-    //std::ifstream f(filename);
-    simplecpp::TokenList rawtokens(files,filename,&outputList);
-    rawtokens.removeComments();
-    std::map<std::string, simplecpp::TokenList*> included = simplecpp::load(rawtokens, files, dui, &outputList);
+    simplecpp::TokenList *rawtokens;
+    if (use_istream) {
+        std::ifstream f(filename);
+        rawtokens = new simplecpp::TokenList(f, files,filename,&outputList);
+    }
+    else {
+        rawtokens = new simplecpp::TokenList(files,filename,&outputList);
+    }
+    rawtokens->removeComments();
+    std::map<std::string, simplecpp::TokenList*> included = simplecpp::load(*rawtokens, files, dui, &outputList);
     for (std::pair<std::string, simplecpp::TokenList *> i : included)
         i.second->removeComments();
     simplecpp::TokenList outputTokens(files);
-    simplecpp::preprocess(outputTokens, rawtokens, files, included, dui, &outputList);
+    simplecpp::preprocess(outputTokens, *rawtokens, files, included, dui, &outputList);
+    delete rawtokens;
+    rawtokens = nullptr;
 
     // Output
     if (!quiet) {