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.

5 files changed, 82 insertions, 4 deletions

@@ -2,7 +2,7 @@ AUTOMAKE_OPTIONS = subdir-objects
 ACLOCAL_AMFLAGS = ${ACLOCAL_FLAGS}
 
 bin_PROGRAMS = rawr-ebooks rawr-gen
-rawr_ebooks_SOURCES = ebooks.cpp kgramstats.cpp
-rawr_gen_SOURCES = gen.cpp kgramstats.cpp
+rawr_ebooks_SOURCES = ebooks.cpp kgramstats.cpp freevars.cpp
+rawr_gen_SOURCES = gen.cpp kgramstats.cpp freevars.cpp
 rawr_ebooks_CPPFLAGS = $(LIBTWITCURL_CFLAGS) $(YAML_CFLAGS)
 rawr_ebooks_LDADD = $(LIBTWITCURL_LIBS) $(YAML_LIBS)
\ No newline at end of file
@@ -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);
@@ -0,0 +1,45 @@
+#include "freevars.h"
+#include <fstream>
+
+freevars::freevars()
+{
+    vars = new map<string, vector<string>* >();
+}
+
+void freevars::addVar(string name, string filename)
+{
+    vector<string>* eltlist = new vector<string>();
+    
+    ifstream infile(filename.c_str());
+    if (infile)
+    {
+        string line;
+        
+        while (getline(infile, line))
+        {
+            eltlist->push_back(line);
+        }
+    } else {
+        eltlist->push_back("");
+    }
+    
+    (*vars)[name] = eltlist;
+}
+
+string freevars::parse(string in)
+{
+    string res(in);
+    
+    for (map<string, vector<string>* >::iterator it = vars->begin(); it != vars->end(); it++)
+    {
+        string tofind = "$" + it->first + "$";
+        size_t fpos = res.find(tofind);
+        if (fpos != string::npos)
+        {
+            int r = rand() % it->second->size();
+            res.replace(fpos, tofind.length(), (*it->second)[r], 0, string::npos);
+        }
+    }
+    
+    return res;
+}
\ No newline at end of file
@@ -0,0 +1,21 @@
+#include <map>
+#include <string>
+#include <vector>
+
+using namespace::std;
+
+#ifndef FREEVARS_H
+#define FREEVARS_H
+
+class freevars
+{
+public:
+    freevars();
+    void addVar(string name, string filename);
+    string parse(string in);
+    
+private:
+    map<string, vector<string>* >* vars;
+};
+
+#endif FREEVARS_H
\ No newline at end of file
@@ -7,6 +7,7 @@
 #include <cstdlib>
 #include <fstream>
 #include <iostream>
+#include "freevars.h"
 
 using namespace::std;
 
@@ -45,6 +46,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 (;;)
         {
@@ -52,7 +58,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) + " ";
                 }
 
                 cout << hi << endl;