Removed yamlcpp dependency from rawr-gen

rawr-gen now takes the input corpus as a command-line argument, so as to increase the ease-of-use. It also now shows a usage message if provided with a non-existent file or no argument.

1 files changed, 22 insertions, 6 deletions

@@ -7,18 +7,34 @@
 #include <cstdlib>
 #include <fstream>
 #include <iostream>
-#include <unistd.h>
-#include <yaml-cpp/yaml.h>
 
 using namespace::std;
 
 int main(int argc, char** args)
 {
         srand(time(NULL));
-        
-        YAML::Node config = YAML::LoadFile("config.yml");
-        
-        ifstream infile(config["corpus"].as<std::string>().c_str());
+    
+    if (argc == 1)
+    {
+        cout << "rawr-gen, version 1.0" << endl;
+        cout << "Usage: rawr-gen corpus-file" << endl;
+        cout << "  where 'corpus-file' is the path to your input" << endl;
+        
+        return 0;
+    }
+    
+        ifstream infile(args[1]);
+    if (!infile)
+    {
+        cout << "rawr-gen, version 1.0" << endl;
+        cout << "Usage: rawr-gen corpus-file" << endl;
+        cout << "  where 'corpus-file' is the path to your input" << endl;
+        cout << endl;
+        cout << "The file you specified does not exist." << endl;
+        
+        return 0;
+    }
+    
         string corpus;
         string line;
         while (getline(infile, line))