Add a rather dumb builder for joining nodes

3 files changed, 61 insertions, 1 deletions

diff --git a/ui/builders/join.cpp b/ui/builders/join.cpp
new file mode 100644
index 0000000..90b1d61
--- /dev/null
+++ b/ui/builders/join.cpp
@@ -0,0 +1,43 @@
+#include "join.h"
+#include <game/geoData.h>
+
+void
+BuilderJoin::render(const Shader &) const
+{
+}
+
+std::string
+BuilderJoin::hint() const
+{
+	if (p1) {
+		return "Pick second node";
+	}
+	return "Pick first node";
+}
+
+void
+BuilderJoin::click(Network * network, const GeoData *, const SDL_MouseButtonEvent & e, const Ray & ray)
+{
+	switch (e.button) {
+		case SDL_BUTTON_LEFT:
+			if (const auto p = network->intersectRayNodes(ray)) {
+				if (p1) {
+					create(network, p1, p);
+					p1.reset();
+				}
+				else {
+					p1 = p;
+				}
+			}
+			return;
+		case SDL_BUTTON_MIDDLE:
+			p1.reset();
+			return;
+	}
+}
+
+void
+BuilderJoin::create(Network * network, const NodePtr & p1, const NodePtr & p2) const
+{
+	network->addJoins(p1->pos, p2->pos);
+}
diff --git a/ui/builders/join.h b/ui/builders/join.h
new file mode 100644
index 0000000..d4fb534
--- /dev/null
+++ b/ui/builders/join.h
@@ -0,0 +1,15 @@
+#pragma once
+#include "../editNetwork.h"
+
+class Network;
+class GeoData;
+
+class BuilderJoin : public EditNetwork::Builder {
+	void render(const Shader &) const override;
+	std::string hint() const override;
+	void click(Network * network, const GeoData * geoData, const SDL_MouseButtonEvent & e, const Ray & ray) override;
+
+	void create(Network * network, const NodePtr & p1, const NodePtr & p2) const;
+
+	NodePtr p1;
+};
diff --git a/ui/editNetwork.cpp b/ui/editNetwork.cpp
index a51d7fb..3ab7059 100644
--- a/ui/editNetwork.cpp
+++ b/ui/editNetwork.cpp
@@ -1,4 +1,5 @@
 #include "editNetwork.h"
+#include "builders/join.h"
 #include "builders/straight.h"
 #include "text.h"
 #include <game/gamestate.h>
@@ -7,6 +8,7 @@
 EditNetwork::EditNetwork(Network * n) :
 	network {n}, builderToolbar {
 						 {"ui/icon/network.png", mode.toggle<BuilderStraight>()},
+						 {"ui/icon/network.png", mode.toggle<BuilderJoin>()},
 				 }
 {
 }
@@ -16,7 +18,7 @@ EditNetwork::~EditNetwork() = default;
 bool
 EditNetwork::click(const SDL_MouseButtonEvent & e, const Ray & ray)
 {
-	if (builder) {
+	if (builder && (e.button == SDL_BUTTON_LEFT || e.button == SDL_BUTTON_MIDDLE)) {
 		builder->click(network, gameState->geoData.get(), e, ray);
 		return true;
 	}