Added painting room randomisation

1 files changed, 43 insertions, 0 deletions

@@ -118,6 +118,25 @@ private:
   std::vector<std::vector<std::string>> sets_;
 };
 
+class Paintings {
+public:
+  explicit Paintings(std::string filename) {
+    std::ifstream file(filename);
+    std::string line;
+    while (std::getline(file, line)) {
+      auto parts = hatkirby::split<std::vector<std::string>>(line, ",");
+      paintings_.emplace_back(parts[0], parts[1], std::atoi(parts[2].c_str()));
+    }
+  }
+
+  const std::tuple<std::string, std::string, int>& GetPainting(std::mt19937& rng) const {
+    return paintings_.at(std::uniform_int_distribution<int>(0, paintings_.size()-1)(rng));
+  }
+
+private:
+  std::vector<std::tuple<std::string, std::string, int>> paintings_;
+};
+
 class Generator {
 public:
 
@@ -125,6 +144,7 @@ public:
     database_ = std::make_unique<verbly::database>("/Users/hatkirby/Dropbox/Programming/verbly-datafiles/d1.3_lingo7");
     wanderlust_ = std::make_unique<Wanderlust>("../wanderlust_words.txt", "../wanderlust_puzzles.txt");
     cross_tower_ = std::make_unique<CrossTower>("../cross_tower.txt");
+    paintings_ = std::make_unique<Paintings>("../paintings.txt");
   }
 
   // Querying
@@ -136,6 +156,18 @@ public:
     return panels_.at(name);
   }
 
+  bool IsNodeRandomized(const std::string& name) const {
+    return replace_nodes_.count(name);
+  }
+
+  const std::tuple<std::string, int>& GetNode(const std::string& name) const {
+    return replace_nodes_.at(name);
+  }
+
+  const std::vector<std::tuple<std::string, std::string>>& GetResources() const {
+    return resources_;
+  }
+
   // Sets the panel's question and answer to static values.
   void GenerateStaticPanel(std::string name, std::string question, std::string answer = "");
 
@@ -196,6 +228,9 @@ public:
     std::string west_other_name2,
     std::string west_other_name3);
 
+  // Generate a painting/panel pair.
+  void GeneratePaintingPuzzle(std::string panel_name, std::string painting_name);
+
 private:
 
   verbly::filter MakeHintFilter(verbly::filter subfilter, Height height, Colour colour, FilterDirection filter_direction) const;
@@ -223,12 +258,20 @@ private:
   std::unique_ptr<verbly::database> database_;
   std::unique_ptr<Wanderlust> wanderlust_;
   std::unique_ptr<CrossTower> cross_tower_;
+  std::unique_ptr<Paintings> paintings_;
 
   // name, question, answer
   std::map<std::string, std::tuple<std::string, std::string>> panels_;
 
+  // name, resource_path, resource_id (0 if needs import)
+  std::map<std::string, std::tuple<std::string, int>> replace_nodes_;
+
+  // path, type
+  std::vector<std::tuple<std::string, std::string>> resources_;
+
   std::vector<std::string> reusable_;
   std::map<std::string, std::set<std::string>> pools_;
+  std::set<std::string> used_paintings_;
 };
 
 #endif /* end of include guard: GENERATOR_H_811386CE */