diff options
Diffstat (limited to 'gen.cpp')
-rw-r--r-- | gen.cpp | 8 |
1 files changed, 7 insertions, 1 deletions
diff --git a/gen.cpp b/gen.cpp index e4a58e5..1d381c8 100644 --- a/gen.cpp +++ b/gen.cpp | |||
@@ -7,6 +7,7 @@ | |||
7 | #include <cstdlib> | 7 | #include <cstdlib> |
8 | #include <fstream> | 8 | #include <fstream> |
9 | #include <iostream> | 9 | #include <iostream> |
10 | #include "freevars.h" | ||
10 | 11 | ||
11 | using namespace::std; | 12 | using namespace::std; |
12 | 13 | ||
@@ -45,6 +46,11 @@ int main(int argc, char** args) | |||
45 | cout << "Preprocessing corpus..." << endl; | 46 | cout << "Preprocessing corpus..." << endl; |
46 | kgramstats* stats = new kgramstats(corpus, 5); | 47 | kgramstats* stats = new kgramstats(corpus, 5); |
47 | 48 | ||
49 | cout << "Preprocessing freevars..." << endl; | ||
50 | freevars* vars = new freevars(); | ||
51 | vars->addVar("name", "names.txt"); | ||
52 | vars->addVar("noun", "nouns.txt"); | ||
53 | |||
48 | cout << "Generating..." << endl; | 54 | cout << "Generating..." << endl; |
49 | for (;;) | 55 | for (;;) |
50 | { | 56 | { |
@@ -52,7 +58,7 @@ int main(int argc, char** args) | |||
52 | string hi; | 58 | string hi; |
53 | for (vector<string>::iterator it = doc.begin(); it != doc.end(); ++it) | 59 | for (vector<string>::iterator it = doc.begin(); it != doc.end(); ++it) |
54 | { | 60 | { |
55 | hi += *it + " "; | 61 | hi += vars->parse(*it) + " "; |
56 | } | 62 | } |
57 | 63 | ||
58 | cout << hi << endl; | 64 | cout << hi << endl; |