From 6ed62d190222e9debe937efe07770641bf03ffe7 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Sun, 12 May 2024 11:14:19 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 10 +++++++---
 1 file changed, 7 insertions(+), 3 deletions(-)

diff --git a/nodes.py b/nodes.py
index 4bef689..861944f 100644
--- a/nodes.py
+++ b/nodes.py
@@ -210,8 +210,8 @@ class CalculateNormalsFromImages:
             }
         }
     
-    RETURN_TYPES = ("IMAGE",)
-    RETURN_NAMES = ("normal", )
+    RETURN_TYPES = ("IMAGE", "IMAGE",)
+    RETURN_NAMES = ("normal", "divided",)
     FUNCTION = "execute"
     CATEGORY = "IC-Light"
     DESCRIPTION = """
@@ -271,8 +271,12 @@ left, right, bottom, top
         
         normal = F.normalize(normal * 2 - 1, dim=3) / 2 + 0.5
         normal = torch.clamp(normal, 0, 1)
+        divided = np.stack([left, right, bottom, top])
+        divided = torch.from_numpy(divided)
+        print(divided.shape)
+        divided = torch.clamp(divided, 0, 1)
    
-        return (normal,)
+        return (normal, divided, )
 
 class LoadHDRImage:
     @classmethod