From 32cfe9dbfcd318dc8b45d6483d5679f672530f88 Mon Sep 17 00:00:00 2001
From: luboslenco <lubos.lenco@gmail.com>
Date: Sun, 7 Feb 2021 12:10:53 +0100
Subject: [PATCH] Remove unused node groups when material is deleted

---
 Sources/arm/ui/TabMaterials.hx | 1 +
 Sources/arm/ui/UINodes.hx      | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/Sources/arm/ui/TabMaterials.hx b/Sources/arm/ui/TabMaterials.hx
index 0efb4b32..4f19ea02 100644
--- a/Sources/arm/ui/TabMaterials.hx
+++ b/Sources/arm/ui/TabMaterials.hx
@@ -145,6 +145,7 @@ class TabMaterials {
 								materials.splice(i, 1);
 								UISidebar.inst.hwnd1.redraws = 2;
 								for (m in Project.materials) updateMaterialPointers(m.canvas.nodes, i);
+								for (n in m.canvas.nodes) UINodes.onNodeRemove(n);
 							}
 
 							var baseHandle = Id.handle().nest(m.id, {selected: m.paintBase});
diff --git a/Sources/arm/ui/UINodes.hx b/Sources/arm/ui/UINodes.hx
index 9a8cb673..7697f3a7 100644
--- a/Sources/arm/ui/UINodes.hx
+++ b/Sources/arm/ui/UINodes.hx
@@ -190,7 +190,7 @@ class UINodes {
 		}
 	}
 
-	function onNodeRemove(node: TNode) {
+	public static function onNodeRemove(node: TNode) {
 		if (node.type == "GROUP") { // Remove unused groups
 			var found = false;
 			var canvases: Array<TNodeCanvas> = [];