1 files changed, 27 insertions, 6 deletions

@@ -1,11 +1,11 @@
 #include "paintings_pane.h"
 
-#include <map>
-#include <set>
-
 #include <fmt/core.h>
 #include <wx/dataview.h>
 
+#include <map>
+#include <set>
+
 #include "ap_state.h"
 #include "game_data.h"
 #include "tracker_state.h"
@@ -21,25 +21,34 @@ std::string GetPaintingDisplayName(const std::string& id) {
 
 }  // namespace
 
-PaintingsPane::PaintingsPane(wxWindow* parent)
-    : wxPanel(parent, wxID_ANY) {
-  wxStaticText* label = new wxStaticText(this, wxID_ANY, "Shuffled paintings grouped by destination:");
+PaintingsPane::PaintingsPane(wxWindow* parent) : wxPanel(parent, wxID_ANY) {
+  wxStaticText* label = new wxStaticText(
+      this, wxID_ANY, "Shuffled paintings grouped by destination:");
   tree_ctrl_ = new wxDataViewTreeCtrl(this, wxID_ANY);
 
+  reveal_btn_ = new wxButton(this, wxID_ANY, "Reveal shuffled paintings");
+  reveal_btn_->Disable();
+
   wxBoxSizer* top_sizer = new wxBoxSizer(wxVERTICAL);
   top_sizer->Add(label, wxSizerFlags().Border());
   top_sizer->Add(tree_ctrl_, wxSizerFlags().Expand().Proportion(1));
+  top_sizer->Add(reveal_btn_, wxSizerFlags().Border().Expand());
 
   SetSizerAndFit(top_sizer);
+
+  reveal_btn_->Bind(wxEVT_BUTTON, &PaintingsPane::OnClickRevealPaintings, this);
 }
 
 void PaintingsPane::UpdateIndicators() {
   tree_ctrl_->DeleteAllItems();
 
   if (!AP_IsPaintingShuffle()) {
+    reveal_btn_->Disable();
     return;
   }
 
+  reveal_btn_->Enable();
+
   std::map<std::string, std::set<std::string>> grouped_paintings;
 
   for (const auto& [from, to] : AP_GetPaintingMapping()) {
@@ -59,3 +68,15 @@ void PaintingsPane::UpdateIndicators() {
     }
   }
 }
+
+void PaintingsPane::OnClickRevealPaintings(wxCommandEvent& event) {
+  if (wxMessageBox("Clicking yes will reveal the mapping between all shuffled "
+                   "paintings. This is usually considered a spoiler, and is "
+                   "likely not allowed during competitions. This action is not "
+                   "reversible. Are you sure you want to proceed?",
+                   "Warning", wxYES_NO | wxICON_WARNING) == wxNO) {
+    return;
+  }
+
+  AP_RevealPaintings();
+}