Generate wanderlust and paintings

2 files changed, 96 insertions, 1 deletions

@@ -857,6 +857,25 @@ void generator::run() {
     }
   }
 
+  // Orange addition
+  std::vector<std::string> orange_addition;
+  {
+    hatkirby::progress ppgs("Generating orange addition puzzles...",
+                            wanderlust_.size());
+
+    for (const auto& [cipher, form_id] : wanderlust_) {
+      for (const auto& [cipher2, form_id2] : wanderlust_) {
+        if (cipher2 >= cipher) {
+          break;
+        }
+        if (wanderlust_.count(cipher - cipher2)) {
+          orange_addition.push_back(fmt::format(
+              "[{},{},{}]", form_id2, wanderlust_[cipher - cipher2], form_id));
+        }
+      }
+    }
+  }
+
   // Count up all of the generated puzzles.
   int total_puzzles = 0;
   int reusable_words = 0;
@@ -916,6 +935,27 @@ void generator::run() {
   std::ofstream output_file(outputPath_);
   output_file << "extends Node\n\nvar forms = ["
               << hatkirby::implode(form_entry, ",") << "]" << std::endl;
+
+  std::vector<std::string> painting_entries;
+  {
+    std::list<std::string> paintings(readFile(datadirPath_ / "paintings.txt"));
+    for (const std::string& line : paintings) {
+      auto parts = hatkirby::split<std::vector<std::string>>(line, ",");
+      painting_entries.push_back(
+          fmt::format("[\"{}\",\"{}\"]", parts[0], parts[1]));
+    }
+  }
+  output_file << "var paintings = [" << hatkirby::implode(painting_entries, ",")
+              << "]" << std::endl;
+
+  std::vector<std::string> cipher_lines;
+  for (const auto& [cipher, form_id] : wanderlust_) {
+    cipher_lines.push_back(std::to_string(form_id));
+  }
+  output_file << "var wanderlust = [" << hatkirby::implode(cipher_lines, ",")
+              << "]" << std::endl;
+  output_file << "var addition = [" << hatkirby::implode(orange_addition, ",")
+              << "]" << std::endl;
 }
 
 size_t generator::LookupOrCreatePronunciation(const std::string& phonemes) {
@@ -996,7 +1036,59 @@ size_t generator::LookupOrCreateForm(const std::string& word) {
   } else {
     size_t form_id = forms_.size();
     form_by_text_[word] = form_id;
-    forms_.push_back({.id = form_id, .text = word});
+    std::optional<int> ciphered;
+    if (std::all_of(word.begin(), word.end(), [](char ch) {
+          return ch == 'w' || ch == 'a' || ch == 'n' || ch == 'd' ||
+                 ch == 'e' || ch == 'r' || ch == 'l' || ch == 'u' ||
+                 ch == 's' || ch == 't';
+        })) {
+      ciphered = 0;
+      std::string to_cipher = word;
+      while (!to_cipher.empty()) {
+        *ciphered *= 10;
+        switch (to_cipher.back()) {
+          case 'w': {
+            *ciphered += 1;
+            break;
+          }
+          case 'a': {
+            *ciphered += 2;
+            break;
+          }
+          case 'n': {
+            *ciphered += 3;
+            break;
+          }
+          case 'd': {
+            *ciphered += 4;
+            break;
+          }
+          case 'e': {
+            *ciphered += 5;
+            break;
+          }
+          case 'r': {
+            *ciphered += 6;
+            break;
+          }
+          case 'l': {
+            *ciphered += 7;
+            break;
+          }
+          case 'u': {
+            *ciphered += 8;
+            break;
+          }
+          case 's': {
+            *ciphered += 9;
+            break;
+          }
+        }
+        to_cipher.pop_back();
+      }
+      wanderlust_[*ciphered] = form_id;
+    }
+    forms_.push_back({.id = form_id, .text = word, .ciphered = ciphered});
 
     std::string sortedText = word;
     std::sort(sortedText.begin(), sortedText.end());
@@ -92,6 +92,7 @@ class generator {
     std::vector<size_t> pronunciation_ids;
     std::optional<size_t> anagram_set_id;
     std::optional<size_t> reverse_form_id;
+    std::optional<int> ciphered;
 
     std::unordered_map<PuzzleType, std::set<size_t>> puzzles;
   };
@@ -124,6 +125,8 @@ class generator {
 
   std::vector<std::vector<size_t>> synsets_;
   std::unordered_map<int, size_t> synset_by_wnid_;
+
+  std::map<int, size_t> wanderlust_;
 };
 
 #endif /* end of include guard: GENERATOR_H_D5C6A724 */
\ No newline at end of file