From 4681411ca1f9fcbe5eba5c4009175033e76c865c Mon Sep 17 00:00:00 2001
From: NebelNidas
Date: Mon, 9 Oct 2023 12:29:35 +0200
Subject: Use `VisitableMappingTree` where possible

---
 enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'enigma-swing/src/main/java')

diff --git a/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java b/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java
index 0eecc7b..2c15305 100644
--- a/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java
+++ b/enigma-swing/src/main/java/cuchaz/enigma/gui/GuiController.java
@@ -33,6 +33,7 @@ import net.fabricmc.mappingio.MappingReader;
 import net.fabricmc.mappingio.MappingWriter;
 import net.fabricmc.mappingio.tree.MemoryMappingTree;
 import net.fabricmc.mappingio.tree.VisitOrder;
+import net.fabricmc.mappingio.tree.VisitableMappingTree;
 
 import cuchaz.enigma.Enigma;
 import cuchaz.enigma.EnigmaProfile;
@@ -167,7 +168,7 @@ public class GuiController implements ClientPacketHandler {
 					}
 
 					progress.init(1, loadingMessage);
-					MemoryMappingTree mappingTree = new MemoryMappingTree();
+					VisitableMappingTree mappingTree = new MemoryMappingTree();
 					MappingReader.read(path, format.getMappingIoCounterpart(), mappingTree);
 					mappings = MappingIoConverter.fromMappingIo(mappingTree, progress);
 				} else {
@@ -229,7 +230,7 @@ public class GuiController implements ClientPacketHandler {
 			loadedMappingPath = path;
 
 			if (useMappingIo) {
-				MemoryMappingTree mappingTree = MappingIoConverter.toMappingIo(mapper.getObfToDeobf(), progress);
+				VisitableMappingTree mappingTree = MappingIoConverter.toMappingIo(mapper.getObfToDeobf(), progress);
 
 				progress.init(1, I18n.translate("progress.mappings.writing"));
 				MappingWriter writer = MappingWriter.create(path, format.getMappingIoCounterpart());
-- 
cgit v1.2.3