From f3c045d51f9a0576f9d5573acddbe67f2f64b587 Mon Sep 17 00:00:00 2001
From: Lubos Lenco <lubos.lenco@gmail.com>
Date: Thu, 8 Jul 2021 10:20:55 +0200
Subject: [PATCH] Fix layer mask node preview

---
 Sources/arm/ui/UINodes.hx | 13 +++++++++++++
 1 file changed, 13 insertions(+)

diff --git a/Sources/arm/ui/UINodes.hx b/Sources/arm/ui/UINodes.hx
index 4ddd5a51..f189f5f9 100644
--- a/Sources/arm/ui/UINodes.hx
+++ b/Sources/arm/ui/UINodes.hx
@@ -627,6 +627,7 @@ class UINodes {
 			if (Config.raw.node_preview && nodes.nodesSelected.length > 0) {
 				var img: kha.Image = null;
 				var sel = nodes.nodesSelected[0];
+				var singleChannel = sel.type == "LAYER_MASK";
 				if (sel.type == "LAYER" || sel.type == "LAYER_MASK") {
 					var id = sel.buttons[0].default_value;
 					if (id < Project.layers.length) {
@@ -660,10 +661,22 @@ class UINodes {
 					var invertY = false;
 					#end
 
+					if (singleChannel) {
+						ui.g.pipeline = UIView2D.pipe;
+						#if kha_opengl
+						ui.currentWindow.texture.g4.setPipeline(UIView2D.pipe);
+						#end
+						ui.currentWindow.texture.g4.setInt(UIView2D.channelLocation, 1);
+					}
+
 					ui.g.color = 0xffffffff;
 					invertY ?
 						ui.g.drawScaledImage(img, tx, ty + th, tw, -th) :
 						ui.g.drawScaledImage(img, tx, ty, tw, th);
+
+					if  (singleChannel) {
+						ui.g.pipeline = null;
+					}
 				}
 			}