From 60b482651b68b578a6767f66a431911b618ccc4c Mon Sep 17 00:00:00 2001
From: ProGamerGov <ProGamerGov@users.noreply.github.com>
Date: Tue, 28 Jan 2025 15:09:35 -0700
Subject: [PATCH] General Improvements

---
 ReadMe.md      |  6 ++++++
 nodes.py       | 16 ++++++++--------
 pyproject.toml |  2 +-
 3 files changed, 15 insertions(+), 9 deletions(-)

diff --git a/ReadMe.md b/ReadMe.md
index 2083ed1..99f77ec 100644
--- a/ReadMe.md
+++ b/ReadMe.md
@@ -52,6 +52,12 @@ Or for Windows:
 path/to/python.exe -m pip install pytorch360convert
 ```
 
+#### ComfyUI registry install
+
+```
+comfy node registry-install pytorch360convert
+```
+
 ## 💡 Usage Examples
 
 See the provided workflow files in the `examples` directory for examples of how to use each new node.
diff --git a/nodes.py b/nodes.py
index 292edbd..f213f63 100644
--- a/nodes.py
+++ b/nodes.py
@@ -16,8 +16,8 @@ class C2ENode:
                 "e_img": ("IMAGE", {"default": None}),
                 "h": ("INT", {"default": -1}),
                 "w": ("INT", {"default": -1}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
-                "cube_format": ("STRING", {"default": "stack"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
+                "cube_format": (["stack", "dice", "horizon", "list", "dict"], {"default": "stack"}),
             },
         }
 
@@ -61,8 +61,8 @@ class E2CNode:
             "required": {
                 "e_img": ("IMAGE", {"default": None}),
                 "face_width": ("INT", {"default": -1}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
-                "cube_format": ("STRING", {"default": "stack"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
+                "cube_format": (["stack", "dice", "horizon", "list", "dict"], {"default": "stack"}),
             },
         }
 
@@ -112,7 +112,7 @@ class E2PNode:
                 "out_h": ("INT", {"default": 512}),
                 "out_w": ("INT", {"default": 512}),
                 "in_rot_deg": ("FLOAT", {"default": 0.0}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
             },
         }
 
@@ -163,7 +163,7 @@ class E2ENode:
                 "roll": ("FLOAT", {"default": 0.0}),
                 "h_deg": ("FLOAT", {"default": 0.0}),
                 "v_deg": ("FLOAT", {"default": 0.0}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
             },
         }
 
@@ -312,8 +312,8 @@ class C2EMaskedDiffNode:
                 "original_faces": ("IMAGE", {"default": None}),
                 "modified_faces": ("IMAGE", {"default": None}),
                 "original_equi": ("IMAGE", {"default": None}),
-                "padding_mode": ("STRING", {"default": "bilinear"}),
-                "cube_format": ("STRING", {"default": "stack"}),
+                "padding_mode": (["bilinear", "bicubic", "nearest"], {"default": "bilinear"}),
+                "cube_format": (["stack", "dice", "horizon", "list", "dict"], {"default": "stack"}),
             },
         }
 
diff --git a/pyproject.toml b/pyproject.toml
index 87b80e5..0e665ec 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -1,7 +1,7 @@
 [project]
 name = "pytorch360convert"
 description = "A collection of custom nodes for working with and converting between 360 degree equirectangular images, cubemap, and perspective images. Panoramic 360 images are also sometimes known as 'VR photography', 'image sphere', and 360 pano."
-version = "1.0.0"
+version = "1.0.1"
 license = {file = "LICENSE"}
 dependencies = ["pytorch360convert"]