Hovered connections on subway map!

1 files changed, 123 insertions, 6 deletions

@@ -2,6 +2,9 @@
 
 #include <wx/dcbuffer.h>
 
+#include <sstream>
+
+#include "ap_state.h"
 #include "game_data.h"
 #include "global.h"
 #include "tracker_state.h"
@@ -42,10 +45,61 @@ SubwayMap::SubwayMap(wxWindow *parent) : wxPanel(parent, wxID_ANY) {
   Bind(wxEVT_MOTION, &SubwayMap::OnMouseMove, this);
 }
 
+void SubwayMap::OnConnect() {
+  networks_.Clear();
+
+  std::map<std::string, std::vector<int>> tagged;
+  for (const SubwayItem &subway_item : GD_GetSubwayItems()) {
+    if (AP_IsPaintingShuffle() && !subway_item.paintings.empty()) {
+      continue;
+    }
+
+    for (const std::string &tag : subway_item.tags) {
+      tagged[tag].push_back(subway_item.id);
+    }
+
+    if (!AP_IsSunwarpShuffle() && subway_item.sunwarp && subway_item.sunwarp->type != SubwaySunwarpType::kFinal) {
+      std::ostringstream tag;
+      tag << "sunwarp" << subway_item.sunwarp->dots;
+
+      tagged[tag.str()].push_back(subway_item.id);
+    }
+  }
+
+  for (const auto &[tag, items] : tagged) {
+    // Pairwise connect all items with the same tag.
+    for (auto tag_it1 = items.begin(); std::next(tag_it1) != items.end();
+         tag_it1++) {
+      for (auto tag_it2 = std::next(tag_it1); tag_it2 != items.end();
+           tag_it2++) {
+        networks_.AddLink(*tag_it1, *tag_it2);
+      }
+    }
+  }
+
+  checked_paintings_.clear();
+}
+
 void SubwayMap::UpdateIndicators() {
   Redraw();
 }
 
+void SubwayMap::UpdatePainting(std::string from_painting_id,
+                               std::optional<std::string> to_painting_id) {
+  checked_paintings_.insert(from_painting_id);
+
+  if (to_painting_id) {
+    networks_.AddLink(GD_GetSubwayItemForPainting(from_painting_id),
+                      GD_GetSubwayItemForPainting(*to_painting_id));
+  }
+}
+
+void SubwayMap::UpdateSunwarp(SubwaySunwarp from_sunwarp,
+  SubwaySunwarp to_sunwarp) {
+  networks_.AddLink(GD_GetSubwayItemForSunwarp(from_sunwarp),
+                    GD_GetSubwayItemForSunwarp(to_sunwarp));
+}
+
 void SubwayMap::OnPaint(wxPaintEvent &event) {
   if (GetSize() != rendered_.GetSize()) {
     Redraw();
@@ -54,6 +108,73 @@ void SubwayMap::OnPaint(wxPaintEvent &event) {
   wxBufferedPaintDC dc(this);
   dc.DrawBitmap(rendered_, 0, 0);
 
+  if (hovered_item_ && networks_.IsItemInNetwork(*hovered_item_)) {
+    dc.SetBrush(*wxTRANSPARENT_BRUSH);
+
+    int network_id = networks_.GetNetworkWithItem(*hovered_item_);
+    for (const auto &[item_id1, item_id2] :
+         networks_.GetNetworkGraph(network_id)) {
+      const SubwayItem &item1 = GD_GetSubwayItem(item_id1);
+      const SubwayItem &item2 = GD_GetSubwayItem(item_id2);
+
+      int item1_x = (item1.x + AREA_ACTUAL_SIZE / 2) * render_width_ / map_image_.GetWidth() + render_x_;
+      int item1_y = (item1.y + AREA_ACTUAL_SIZE / 2) * render_width_ / map_image_.GetWidth() + render_y_;
+
+      int item2_x = (item2.x + AREA_ACTUAL_SIZE / 2) * render_width_ / map_image_.GetWidth() + render_x_;
+      int item2_y = (item2.y + AREA_ACTUAL_SIZE / 2) * render_width_ / map_image_.GetWidth() + render_y_;
+
+      int left = std::min(item1_x, item2_x);
+      int top = std::min(item1_y, item2_y);
+      int right = std::max(item1_x, item2_x);
+      int bottom = std::max(item1_y, item2_y);
+
+      int halfwidth = right - left;
+      int halfheight = bottom - top;
+
+      int ellipse_x;
+      int ellipse_y;
+      double start;
+      double end;
+
+      if (item1_x > item2_x) {
+        ellipse_y = top;
+
+        if (item1_y > item2_y) {
+          ellipse_x = left - halfwidth;
+
+          start = 0;
+          end = 90;
+        } else {
+          ellipse_x = left;
+
+          start = 90;
+          end = 180;
+        }
+      } else {
+        ellipse_y = top - halfheight;
+
+        if (item1_y > item2_y) {
+          ellipse_x = left - halfwidth;
+
+          start = 270;
+          end = 360;
+        } else {
+          ellipse_x = left;
+
+          start = 180;
+          end = 270;
+        }
+      }
+
+      dc.SetPen(*wxThePenList->FindOrCreatePen(*wxBLACK, 4));
+      dc.DrawEllipticArc(ellipse_x, ellipse_y, halfwidth * 2, halfheight * 2,
+                         start, end);
+      dc.SetPen(*wxThePenList->FindOrCreatePen(*wxCYAN, 2));
+      dc.DrawEllipticArc(ellipse_x, ellipse_y, halfwidth * 2, halfheight * 2,
+                         start, end);
+    }
+  }
+
   event.Skip();
 }
 
@@ -73,13 +194,9 @@ void SubwayMap::OnMouseMove(wxMouseEvent &event) {
   }
 
   if (new_hovered_item != hovered_item_) {
-    if (new_hovered_item) {
-      wxLogVerbose("Hovered: %d", *new_hovered_item);
-    } else {
-      wxLogVerbose("Un-hovered: %d", *hovered_item_);
-    }
-
     hovered_item_ = new_hovered_item;
+
+    Refresh();
   }
 
   event.Skip();