about summary refs log tree commit diff stats
path: root/gen.cpp
diff options
context:
space:
mode:
authorKelly Rauchenberger <fefferburbia@gmail.com>2016-05-26 21:46:41 -0400
committerKelly Rauchenberger <fefferburbia@gmail.com>2016-05-26 21:46:41 -0400
commit4d217ac6122120d9e86248432594864e114e3a46 (patch)
tree314d758bad971572d27582ef523c526e119d295a /gen.cpp
parent6fbc06504c448a2972a01438b920bc61523e4683 (diff)
downloadrawr-ebooks-4d217ac6122120d9e86248432594864e114e3a46.tar.gz
rawr-ebooks-4d217ac6122120d9e86248432594864e114e3a46.tar.bz2
rawr-ebooks-4d217ac6122120d9e86248432594864e114e3a46.zip
Fixed ebooks and gen ignoring the form distribution
Diffstat (limited to 'gen.cpp')
-rw-r--r--gen.cpp8
1 files changed, 4 insertions, 4 deletions
diff --git a/gen.cpp b/gen.cpp index eba0277..5e2d9db 100644 --- a/gen.cpp +++ b/gen.cpp
@@ -69,14 +69,14 @@ int main(int argc, char** args)
69 rawr kgramstats; 69 rawr kgramstats;
70 kgramstats.addCorpus(corpus); 70 kgramstats.addCorpus(corpus);
71 kgramstats.compile(4); 71 kgramstats.compile(4);
72 kgramstats.setTransformCallback([&] (std::string canonical, std::string) { 72 kgramstats.setTransformCallback([&] (std::string, std::string form) {
73 size_t pos = canonical.find("$name$"); 73 size_t pos = form.find("$name$");
74 if (pos != std::string::npos) 74 if (pos != std::string::npos)
75 { 75 {
76 canonical.replace(pos, 6, fv_names[rand() % fv_names.size()]); 76 form.replace(pos, 6, fv_names[rand() % fv_names.size()]);
77 } 77 }
78 78
79 return canonical; 79 return form;
80 }); 80 });
81 81
82 std::cout << "Generating..." << std::endl; 82 std::cout << "Generating..." << std::endl;