Implemented freevars

Arbitrary variable tokens can now be defined (though at this point only in the code itself) as a pair of a variable name and a filename pointing to a plain text file containing a newline-delimited list of elements. When a token of the form $name$ (where name is the name of a variable) is encountered, the output will include a random element from the appropriate list. The variables $name$ and $noun$ are hard-coded at this point, but the program will not crash if names.txt and nouns.txt do not exist and will instead just silently ignore the variables.

1 files changed, 7 insertions, 1 deletions

@@ -10,6 +10,7 @@
 #include <twitcurl.h>
 #include <unistd.h>
 #include <yaml-cpp/yaml.h>
+#include "freevars.h"
 
 using namespace::std;
 
@@ -28,6 +29,11 @@ int main(int argc, char** args)
 
     cout << "Preprocessing corpus..." << endl;
         kgramstats* stats = new kgramstats(corpus, 5);
+    
+    cout << "Preprocessing freevars..." << endl;
+    freevars* vars = new freevars();
+    vars->addVar("name", "names.txt");
+    vars->addVar("noun", "nouns.txt");
 
     cout << "Generating..." << endl;
         for (;;)
@@ -36,7 +42,7 @@ int main(int argc, char** args)
                 string hi;
                 for (vector<string>::iterator it = doc.begin(); it != doc.end(); ++it)
                 {
-                        hi += *it + " ";
+                        hi += vars->parse(*it) + " ";
                 }
         
                 hi = hi.substr(0,140);