[main.cpp] added flag '-C' to keep comments

Author: Oleg Sesov

main.cpp

@@ -18,6 +18,7 @@ int main(int argc, char **argv)
     bool error = false;
     const char *filename = nullptr;
     bool use_istream = false;
+    bool keep_comments = false;
 
     // Settings..
     simplecpp::DUI dui;
@@ -70,6 +71,10 @@ int main(int argc, char **argv)
                 error_only = true;
                 found = true;
                 break;
+            case 'C':
+                keep_comments = true;
+                found = true;
+                break;
             }
             if (!found) {
                 std::cout << "error: option '" << arg << "' is unknown." << std::endl;
@@ -102,10 +107,11 @@ int main(int argc, char **argv)
         std::cout << "  -q              Quiet mode (no output)." << std::endl;
         std::cout << "  -is             Use std::istream interface." << std::endl;
         std::cout << "  -e              Output errors only." << std::endl;
+        std::cout << "  -C              Keep comments." << std::endl;
         std::exit(0);
     }
 
-    dui.removeComments = true;
+    dui.removeComments = !keep_comments;
 
     // Perform preprocessing
     simplecpp::OutputList outputList;
@@ -121,7 +127,8 @@ int main(int argc, char **argv)
     } else {
         rawtokens = new simplecpp::TokenList(filename,files,&outputList);
     }
-    rawtokens->removeComments();
+    if (!keep_comments)
+        rawtokens->removeComments();
     simplecpp::TokenList outputTokens(files);
     std::map<std::string, simplecpp::TokenList*> filedata;
     simplecpp::preprocess(outputTokens, *rawtokens, files, filedata, dui, &outputList);