From fd7630b8f46c75b0e38aece193a3d6a2e2a7ccc1 Mon Sep 17 00:00:00 2001
From: Lubos Lenco <lubos.lenco@gmail.com>
Date: Thu, 8 Jul 2021 11:50:28 +0200
Subject: [PATCH] Improve node context menu

---
 Sources/arm/ui/UINodes.hx | 9 +++++++--
 1 file changed, 7 insertions(+), 2 deletions(-)

diff --git a/Sources/arm/ui/UINodes.hx b/Sources/arm/ui/UINodes.hx
index f189f5f9..716c848f 100644
--- a/Sources/arm/ui/UINodes.hx
+++ b/Sources/arm/ui/UINodes.hx
@@ -230,7 +230,12 @@ class UINodes {
 			if (!Nodes.socketReleased) {
 				UIMenu.draw(function(uiMenu: Zui) {
 					uiMenu._y += 1;
-					uiMenu.enabled = nodes.nodesSelected.length > 0;
+					var protected = selected == null ||
+									selected.type == "OUTPUT_MATERIAL_PBR" ||
+									selected.type == "GROUP_INPUT" ||
+									selected.type == "GROUP_OUTPUT" ||
+									selected.type == "BrushOutputNode";
+					uiMenu.enabled = !protected;
 					if (menuButton(uiMenu, tr("Cut"), "ctrl+x")) {
 						Zui.isCopy = true;
 						Zui.isCut = true;
@@ -242,7 +247,7 @@ class UINodes {
 					if (menuButton(uiMenu, tr("Paste"), "ctrl+v")) {
 						Zui.isPaste = true;
 					}
-					uiMenu.enabled = nodes.nodesSelected.length > 0;
+					uiMenu.enabled = !protected;
 					if (menuButton(uiMenu, tr("Delete"), "delete")) {
 						ui.isDeleteDown = true;
 						App.notifyOnNextFrame(function() { ui.isDeleteDown = false; });