From 0b91c7874c5259c5a05ce6e90ebfc706cfd392b7 Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Thu, 23 Jan 2025 21:04:02 +0200
Subject: [PATCH] Update nodes.py

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

diff --git a/nodes.py b/nodes.py
index 7b2e22b..07145b5 100644
--- a/nodes.py
+++ b/nodes.py
@@ -453,9 +453,10 @@ class Hy3DSampleMultiView:
             selected_camera_azims = camera_config["selected_camera_azims"]
             selected_camera_elevs = camera_config["selected_camera_elevs"]
         
-        camera_info = [(((azim // 30) + 9) % 12) // {-20: 1, 0: 1, 20: 1, -90: 3, 90: 3}[
-            elev] + {-20: 0, 0: 12, 20: 24, -90: 36, 90: 40}[elev] for azim, elev in
-                       zip(selected_camera_azims, selected_camera_elevs)]
+        camera_info = [(((azim // 30) + 9) % 12) // {-90: 3, -45: 2, -20: 1, 0: 1, 20: 1, 45: 2, 90: 3}[
+            elev] + {-90: 36, -45: 30, -20: 0, 0: 12, 20: 24, 45: 30, 90: 40}[elev] for azim, elev in
+                    zip(selected_camera_azims, selected_camera_elevs)]
+        print(camera_info)
         
         normal_maps_np = (normal_maps * 255).to(torch.uint8).cpu().numpy()
         normal_maps_pil = [Image.fromarray(normal_map) for normal_map in normal_maps_np]
@@ -504,6 +505,9 @@ class Hy3DBakeFromMultiview:
                 "images": ("IMAGE", ),
                 "renderer": ("MESHRENDER",),
             },
+            "optional": {
+                "camera_config": ("HY3DCAMERA",),
+            }
         }
 
     RETURN_TYPES = ("IMAGE", "MASK", "MESHRENDER")