From b49033154c7010f56214fc2e6c374f5466646779 Mon Sep 17 00:00:00 2001
From: Kawe Mazidjatari <48657826+Mauler125@users.noreply.github.com>
Date: Fri, 23 Aug 2024 14:30:22 +0200
Subject: [PATCH] Recast: add debug option to mark tile by ref

---
 src/naveditor/Editor_TileMesh.cpp | 77 ++++++++++++++++++++++++++-----
 1 file changed, 65 insertions(+), 12 deletions(-)

diff --git a/src/naveditor/Editor_TileMesh.cpp b/src/naveditor/Editor_TileMesh.cpp
index 944fe540..cf06a382 100644
--- a/src/naveditor/Editor_TileMesh.cpp
+++ b/src/naveditor/Editor_TileMesh.cpp
@@ -53,6 +53,7 @@ class NavMeshTileTool : public EditorTool
 	float m_nearestPos[3];
 	int m_selectedTraverseType;
 
+	dtTileRef m_markedTileRef;
 	dtPolyRef m_markedPolyRef;
 
 	enum TileToolCursorMode
@@ -81,6 +82,7 @@ class NavMeshTileTool : public EditorTool
 
 	int m_textOverlayDrawFlags;
 
+	char m_tileRefTextInput[MAX_POLYREF_CHARS];
 	char m_polyRefTextInput[MAX_POLYREF_CHARS];
 	bool m_hitPosSet;
 	
@@ -90,6 +92,7 @@ public:
 		m_editor(0),
 		m_navMesh(0),
 		m_selectedTraverseType(-2),
+		m_markedTileRef(0),
 		m_markedPolyRef(0),
 		m_cursorMode(TT_CURSOR_MODE_DEBUG),
 		m_textOverlayDrawMode(TO_DRAW_MODE_DISABLED),
@@ -98,6 +101,7 @@ public:
 	{
 		rdVset(m_hitPos, 0.0f,0.0f,0.0f);
 		rdVset(m_nearestPos, 0.0f,0.0f,0.0f);
+		memset(m_tileRefTextInput, '\0', sizeof(m_tileRefTextInput));
 		memset(m_polyRefTextInput, '\0', sizeof(m_polyRefTextInput));
 	}
 
@@ -153,9 +157,23 @@ public:
 		if (ImGui::RadioButton("Show Tile And Poly Indices", m_textOverlayDrawFlags & TO_DRAW_FLAGS_INDICES))
 			toggleTextOverlayDrawFlags(TO_DRAW_FLAGS_INDICES);
 
+
+		const bool hasMarker = m_markedTileRef || m_markedPolyRef;
+
+		if (m_navMesh || hasMarker)
+		{
+			ImGui::Separator();
+			ImGui::Text("Markers");
+		}
+
 		if (m_navMesh)
 		{
 			ImGui::PushItemWidth(83);
+			if (m_navMesh && ImGui::InputText("Mark Tile By Ref", m_tileRefTextInput, sizeof(m_tileRefTextInput), ImGuiInputTextFlags_EnterReturnsTrue))
+			{
+				char* pEnd = nullptr;
+				m_markedTileRef = (dtPolyRef)STR_TO_ID(m_tileRefTextInput, &pEnd, 10);
+			}
 			if (m_navMesh && ImGui::InputText("Mark Poly By Ref", m_polyRefTextInput, sizeof(m_polyRefTextInput), ImGuiInputTextFlags_EnterReturnsTrue))
 			{
 				char* pEnd = nullptr;
@@ -164,19 +182,20 @@ public:
 			ImGui::PopItemWidth();
 		}
 
-		if (m_markedPolyRef)
+		if (hasMarker && ImGui::Button("Clear Markers"))
 		{
-			if (ImGui::Button("Clear Marker"))
-			{
-				m_markedPolyRef = 0;
-				rdVset(m_nearestPos, 0.0f,0.0f,0.0f);
-			}
+			m_markedTileRef = 0;
+			m_markedPolyRef = 0;
+			rdVset(m_nearestPos, 0.0f, 0.0f, 0.0f);
 		}
 
 		dtNavMeshQuery* query = m_editor->getNavMeshQuery();
 
 		if (query && m_navMesh)
 		{
+			ImGui::Separator();
+			ImGui::Text("Dumpers");
+
 			ImGui::PushItemWidth(83);
 			ImGui::SliderInt("Selected Traverse Type", &m_selectedTraverseType, -2, 31);
 			ImGui::PopItemWidth();
@@ -229,10 +248,19 @@ public:
 				const float halfExtents[3] = { 2, 2, 4 };
 				dtQueryFilter filter;
 
-				if (dtStatusFailed(m_editor->getNavMeshQuery()->findNearestPoly(m_hitPos, halfExtents, &filter, &m_markedPolyRef, m_nearestPos)))
+				if (shift)
 				{
-					m_markedPolyRef = 0;
-					rdVset(m_nearestPos, 0.0f, 0.0f, 0.0f);
+					if (dtStatusFailed(m_editor->getNavMeshQuery()->findNearestPoly(m_hitPos, halfExtents, &filter, &m_markedPolyRef, m_nearestPos)))
+					{
+						m_markedPolyRef = 0;
+						rdVset(m_nearestPos, 0.0f, 0.0f, 0.0f);
+					}
+				}
+				else
+				{
+					int tx, ty;
+					m_editor->getTilePos(m_hitPos, tx, ty);
+					m_markedTileRef = m_navMesh->getTileRefAt(tx, ty, 0);
 				}
 			}
 		}
@@ -262,14 +290,39 @@ public:
 			glLineWidth(1.0f);
 		}
 
+		const float* debugDrawOffset = m_editor->getDetourDrawOffset();
+
+		if (m_markedTileRef && m_editor && m_navMesh)
+		{
+			const dtMeshTile* tile = m_navMesh->getTileByRef(m_markedTileRef);
+
+			if (tile)
+			{
+				duDrawTraverseLinkParams params;
+				duDebugDrawMeshTile(&m_editor->getDebugDraw(), *m_navMesh, 0, tile, debugDrawOffset, m_editor->getNavMeshDrawFlags(), params);
+
+				const unsigned char side = rdClassifyPointInsideBounds(m_hitPos, tile->header->bmin, tile->header->bmax);
+
+				const int MAX_NEIS = 32; // Max neighbors
+				dtMeshTile* neis[MAX_NEIS];
+
+				const int nneis = m_navMesh->getNeighbourTilesAt(tile->header->x, tile->header->y, side, neis, MAX_NEIS);
+
+				for (int i = 0; i < nneis; i++)
+				{
+					duDebugDrawMeshTile(&m_editor->getDebugDraw(), *m_navMesh, 0, neis[i], debugDrawOffset, m_editor->getNavMeshDrawFlags(), params);
+				}
+			}
+		}
+
 		if (m_markedPolyRef && m_editor && m_navMesh)
 		{
-			const float* debugDrawOffset = m_editor->getDetourDrawOffset();
 			duDebugDrawNavMeshPoly(&m_editor->getDebugDraw(), *m_navMesh, m_markedPolyRef,
 				debugDrawOffset, m_editor->getNavMeshDrawFlags(), duRGBA(255, 0, 170, 190), false);
-
-			duDebugDrawCross(&m_editor->getDebugDraw(), m_nearestPos[0], m_nearestPos[1], m_nearestPos[2], 20.f, duRGBA(0,0,255,255), 2, debugDrawOffset);
 		}
+
+		if (m_markedTileRef || m_markedPolyRef)
+			duDebugDrawCross(&m_editor->getDebugDraw(), m_nearestPos[0], m_nearestPos[1], m_nearestPos[2], 20.f, duRGBA(0, 0, 255, 255), 2, debugDrawOffset);
 	}
 	
 	virtual void handleRenderOverlay(double* proj, double* model, int* view)