From 749df22c85b83c86ec68ebc74829287e54498e26 Mon Sep 17 00:00:00 2001
From: Dan Goodliffe <dan@randomdan.homeip.net>
Date: Sun, 22 Dec 2024 13:53:27 +0000
Subject: Set terrain heights when creating new network links

---
 ui/builders/freeExtend.cpp |  8 ++++++--
 ui/builders/join.cpp       |  4 +++-
 ui/builders/straight.cpp   |  4 +++-
 ui/editNetwork.cpp         | 11 +++++++++++
 ui/editNetwork.h           |  3 +++
 5 files changed, 26 insertions(+), 4 deletions(-)

(limited to 'ui')

diff --git a/ui/builders/freeExtend.cpp b/ui/builders/freeExtend.cpp
index 904e144..fa08af6 100644
--- a/ui/builders/freeExtend.cpp
+++ b/ui/builders/freeExtend.cpp
@@ -61,11 +61,15 @@ BuilderFreeExtend::click(
 Link::CCollection
 BuilderFreeExtend::createJoin(Network * network, GlobalPosition3D p1, GlobalPosition3D p2) const
 {
-	return network->addJoins(p1, p2);
+	const auto links = network->addJoins(p1, p2);
+	setHeightsFor(network, links);
+	return links;
 }
 
 Link::CCollection
 BuilderFreeExtend::createExtend(Network * network, GlobalPosition3D p1, GlobalPosition3D p2) const
 {
-	return network->addExtend(p1, p2);
+	const auto links = network->addExtend(p1, p2);
+	setHeightsFor(network, links);
+	return links;
 }
diff --git a/ui/builders/join.cpp b/ui/builders/join.cpp
index 9fbbae9..ee14d63 100644
--- a/ui/builders/join.cpp
+++ b/ui/builders/join.cpp
@@ -49,5 +49,7 @@ BuilderJoin::click(
 Link::CCollection
 BuilderJoin::create(Network * network, const Node::Ptr & p1, const Node::Ptr & p2) const
 {
-	return network->addJoins(p1->pos, p2->pos);
+	const auto links = network->addJoins(p1->pos, p2->pos);
+	setHeightsFor(network, links);
+	return links;
 }
diff --git a/ui/builders/straight.cpp b/ui/builders/straight.cpp
index 014eea5..b9f1831 100644
--- a/ui/builders/straight.cpp
+++ b/ui/builders/straight.cpp
@@ -52,5 +52,7 @@ BuilderStraight::click(
 Link::CCollection
 BuilderStraight::create(Network * network, GlobalPosition3D p1, GlobalPosition3D p2) const
 {
-	return network->addStraight(p1, p2);
+	const auto links = network->addStraight(p1, p2);
+	setHeightsFor(network, links);
+	return links;
 }
diff --git a/ui/editNetwork.cpp b/ui/editNetwork.cpp
index ac2d93d..50d049d 100644
--- a/ui/editNetwork.cpp
+++ b/ui/editNetwork.cpp
@@ -63,6 +63,17 @@ EditNetwork::Builder::render(const SceneShader & shader) const
 	candidateLinks.apply<const Renderable>(&Renderable::render, shader);
 }
 
+void
+EditNetwork::Builder::setHeightsFor(Network * network, const Link::CCollection & links, GeoData::SetHeightsOpts opts)
+{
+	opts.surface = network->getBaseSurface();
+	const auto width = network->getBaseWidth();
+
+	for (const auto & link : links) {
+		gameState->geoData->setHeights(link->getBase(width), opts);
+	}
+}
+
 void
 EditNetwork::render(const UIShader & shader, const UIComponent::Position & parentPos) const
 {
diff --git a/ui/editNetwork.h b/ui/editNetwork.h
index ec06fa7..2ae467d 100644
--- a/ui/editNetwork.h
+++ b/ui/editNetwork.h
@@ -1,5 +1,6 @@
 #pragma once
 
+#include "game/geoData.h"
 #include "gameMainSelector.h"
 #include "modeHelper.h"
 #include "toolbar.h"
@@ -30,6 +31,8 @@ public:
 		virtual void click(Network *, const GeoData *, const SDL_MouseButtonEvent &, const Ray<GlobalPosition3D> &) = 0;
 		virtual void move(Network *, const GeoData *, const SDL_MouseMotionEvent &, const Ray<GlobalPosition3D> &) = 0;
 
+		static void setHeightsFor(Network *, const Link::CCollection &, GeoData::SetHeightsOpts = {});
+
 		using Ptr = std::unique_ptr<Builder>;
 
 	protected:
-- 
cgit v1.2.3