diff options
-rw-r--r-- | ebooks.cpp | 20 |
1 files changed, 10 insertions, 10 deletions
diff --git a/ebooks.cpp b/ebooks.cpp index 27065d9..6bbe25e 100644 --- a/ebooks.cpp +++ b/ebooks.cpp | |||
@@ -19,34 +19,34 @@ int main(int argc, char** args) | |||
19 | YAML::Node config = YAML::LoadFile("config.yml"); | 19 | YAML::Node config = YAML::LoadFile("config.yml"); |
20 | int delay = config["delay"].as<int>(); | 20 | int delay = config["delay"].as<int>(); |
21 | 21 | ||
22 | ifstream infile(config["corpus"].as<std::string>().c_str()); | 22 | std::ifstream infile(config["corpus"].as<std::string>().c_str()); |
23 | string corpus; | 23 | std::string corpus; |
24 | string line; | 24 | std::string line; |
25 | while (getline(infile, line)) | 25 | while (getline(infile, line)) |
26 | { | 26 | { |
27 | corpus += " " + line; | 27 | corpus += " " + line; |
28 | } | 28 | } |
29 | 29 | ||
30 | cout << "Preprocessing corpus..." << endl; | 30 | std::cout << "Preprocessing corpus..." << std::endl; |
31 | kgramstats* stats = new kgramstats(corpus, 3); | 31 | kgramstats* stats = new kgramstats(corpus, 3); |
32 | 32 | ||
33 | cout << "Preprocessing freevars..." << endl; | 33 | std::cout << "Preprocessing freevars..." << std::endl; |
34 | freevars* vars = new freevars(); | 34 | freevars* vars = new freevars(); |
35 | vars->addVar("name", "names.txt"); | 35 | vars->addVar("name", "names.txt"); |
36 | vars->addVar("noun", "nouns.txt"); | 36 | vars->addVar("noun", "nouns.txt"); |
37 | 37 | ||
38 | cout << "Generating..." << endl; | 38 | std::cout << "Generating..." << std::endl; |
39 | for (;;) | 39 | for (;;) |
40 | { | 40 | { |
41 | vector<string> doc = stats->randomSentence(rand() % 25 + 5); | 41 | std::vector<std::string> doc = stats->randomSentence(rand() % 25 + 5); |
42 | string hi; | 42 | std::string hi; |
43 | for (vector<string>::iterator it = doc.begin(); it != doc.end(); ++it) | 43 | for (std::vector<std::string>::iterator it = doc.begin(); it != doc.end(); ++it) |
44 | { | 44 | { |
45 | hi += vars->parse(*it) + " "; | 45 | hi += vars->parse(*it) + " "; |
46 | } | 46 | } |
47 | 47 | ||
48 | size_t lastperiod = hi.find_last_of("."); | 48 | size_t lastperiod = hi.find_last_of("."); |
49 | if ((lastperiod != string::npos) && (rand() % 3 > 0)) | 49 | if ((lastperiod != std::string::npos) && (rand() % 3 > 0)) |
50 | { | 50 | { |
51 | hi = hi.substr(0, lastperiod+1); | 51 | hi = hi.substr(0, lastperiod+1); |
52 | } | 52 | } |