mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-05-30 19:07:25 +08:00
174208df6b
Some checks are pending
Python Linting / Run Ruff (push) Waiting to run
Python Linting / Run Pylint (push) Waiting to run
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.10, [self-hosted Linux], stable) (push) Waiting to run
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.11, [self-hosted Linux], stable) (push) Waiting to run
Full Comfy CI Workflow Runs / test-stable (12.1, , linux, 3.12, [self-hosted Linux], stable) (push) Waiting to run
Full Comfy CI Workflow Runs / test-unix-nightly (12.1, , linux, 3.11, [self-hosted Linux], nightly) (push) Waiting to run
Execution Tests / test (macos-latest) (push) Waiting to run
Execution Tests / test (ubuntu-latest) (push) Waiting to run
Execution Tests / test (windows-latest) (push) Waiting to run
Test server launches without errors / test (push) Waiting to run
Unit Tests / test (macos-latest) (push) Waiting to run
Unit Tests / test (ubuntu-latest) (push) Waiting to run
Unit Tests / test (windows-2022) (push) Waiting to run
* Move dataset/text nodes to text category * Rename category utils into utilities * Rename category api node into partner * Move categories conditioning, latent, sampling, model_patches, training, etc. under model category * Dispatch partner nodes in to 3d, audio, image, text, video categories * Move PreviewAny node to utilities category
92 lines
2.8 KiB
Python
92 lines
2.8 KiB
Python
"""Number Convert node for unified numeric type conversion.
|
|
|
|
Provides a single node that converts INT, FLOAT, STRING, and BOOL
|
|
inputs into FLOAT and INT outputs.
|
|
"""
|
|
|
|
|
|
import math
|
|
|
|
from typing_extensions import override
|
|
|
|
from comfy_api.latest import ComfyExtension, io
|
|
|
|
|
|
class NumberConvertNode(io.ComfyNode):
|
|
"""Converts various types to numeric FLOAT and INT outputs."""
|
|
|
|
@classmethod
|
|
def define_schema(cls) -> io.Schema:
|
|
return io.Schema(
|
|
node_id="ComfyNumberConvert",
|
|
display_name="Convert Number",
|
|
category="utilities",
|
|
search_aliases=[
|
|
"int to float", "float to int", "number convert",
|
|
"int2float", "float2int", "cast", "parse number",
|
|
"string to number", "bool to int",
|
|
],
|
|
inputs=[
|
|
io.MultiType.Input(
|
|
"value",
|
|
[io.Int, io.Float, io.String, io.Boolean],
|
|
display_name="value",
|
|
),
|
|
],
|
|
outputs=[
|
|
io.Float.Output(display_name="FLOAT"),
|
|
io.Int.Output(display_name="INT"),
|
|
],
|
|
)
|
|
|
|
@classmethod
|
|
def execute(cls, value) -> io.NodeOutput:
|
|
if isinstance(value, bool):
|
|
float_val = 1.0 if value else 0.0
|
|
int_val = 1 if value else 0
|
|
elif isinstance(value, int):
|
|
float_val = float(value)
|
|
int_val = value
|
|
elif isinstance(value, float):
|
|
float_val = value
|
|
int_val = int(value)
|
|
elif isinstance(value, str):
|
|
text = value.strip()
|
|
if not text:
|
|
raise ValueError("Cannot convert empty string to number.")
|
|
try:
|
|
float_val = float(text)
|
|
except ValueError:
|
|
raise ValueError(
|
|
f"Cannot convert string to number: {value!r}"
|
|
) from None
|
|
if not math.isfinite(float_val):
|
|
raise ValueError(
|
|
f"Cannot convert non-finite value to number: {float_val}"
|
|
)
|
|
try:
|
|
int_val = int(text)
|
|
except ValueError:
|
|
int_val = int(float_val)
|
|
else:
|
|
raise TypeError(
|
|
f"Unsupported input type: {type(value).__name__}"
|
|
)
|
|
|
|
if not math.isfinite(float_val):
|
|
raise ValueError(
|
|
f"Cannot convert non-finite value to number: {float_val}"
|
|
)
|
|
|
|
return io.NodeOutput(float_val, int_val)
|
|
|
|
|
|
class NumberConvertExtension(ComfyExtension):
|
|
@override
|
|
async def get_node_list(self) -> list[type[io.ComfyNode]]:
|
|
return [NumberConvertNode]
|
|
|
|
|
|
async def comfy_entrypoint() -> NumberConvertExtension:
|
|
return NumberConvertExtension()
|