From 2ab1bfebb88744022ca833ee384a5557f65a6bde Mon Sep 17 00:00:00 2001
From: adriflex <rouquie.adrien@gmail.com>
Date: Tue, 23 Jan 2024 15:25:54 +0100
Subject: [PATCH] Add Sampler_Setting class to nodes.py

---
 nodes.py | 41 +++++++++++++++++++++++++++++++++++++++--
 1 file changed, 39 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index f3ca000..95efafc 100644
--- a/nodes.py
+++ b/nodes.py
@@ -1,6 +1,7 @@
 import hashlib
 import os
 
+import comfy
 import numpy as np
 import torch
 from PIL import Image
@@ -80,12 +81,48 @@ class ViewportDepth:
         return m.digest().hex()
 
 
+class Sampler_Setting:
+    @classmethod
+    def INPUT_TYPES(cls):
+        # input = float, string and int
+        return {
+            "required": {
+                "seed": ("INT", {"min": 0}),
+                "sampler_steps": ("INT", {"min": 0}),
+                "cgf_scale": ("FLOAT", {"min": 0.0}),
+                # "sampler_name": ("STRING", {"default": ""}),
+                # "scheduler": ("STRING", {"default": ""}),
+                "sampler_name": (comfy.samplers.KSampler.SAMPLERS,),
+                "scheduler": (comfy.samplers.KSampler.SCHEDULERS,),
+                "denoise": ("FLOAT", {"min": 0.0, "max": 1.0}),
+            }
+        }
+
+    RETURN_TYPES = (
+        "INT",
+        "INT",
+        "FLOAT",
+        comfy.samplers.KSampler.SAMPLERS,
+        comfy.samplers.KSampler.SCHEDULERS,
+        "FLOAT",
+    )
+    FUNCTION = "get_sampler_setting"
+    CATEGORY = "Blender TexDiff"
+
+    def get_sampler_setting(
+        self, sampler_steps, cgf_scale, sampler_name, scheduler, denoise
+    ):
+        return (sampler_steps, cgf_scale, sampler_name, scheduler, denoise)
+
+
 NODE_CLASS_MAPPINGS = {
     "ViewportColor": ViewportColor,
     "ViewportDepth": ViewportDepth,
+    "Sampler_Setting": Sampler_Setting,
 }
 
 NODE_DISPLAY_NAME_MAPPINGS = {
-    "ViewportColor": "Blender viewport color",
-    "ViewportDepth": "Blender Viewport depth",
+    "ViewportColor": "From blender viewport color",
+    "ViewportDepth": "From blender Viewport depth",
+    "Sampler_Setting": "From blender sampler setting",
 }