Dan The Monk
b7cb1796cb
- Created FloatToIntNode for converting float values to integers with rounding options. - Created IntToFloatNode for converting integer values to floats. - Introduced NODE_CLASS_MAPPINGS to aggregate conversion nodes. - Added .gitignore to exclude __pycache__ directories.
26 lines
688 B
Python
26 lines
688 B
Python
import math
|
|
|
|
class FloatToIntNode:
|
|
@classmethod
|
|
def INPUT_TYPES(cls):
|
|
return {
|
|
"required": {
|
|
"float_value": ("FLOAT", {"default": 0.0}),
|
|
"rounding_mode": (["down (floor)", "up (ceil)"], {"default": "down (floor)"}),
|
|
}
|
|
}
|
|
|
|
RETURN_TYPES = ("INT",)
|
|
RETURN_NAMES = ("int_value",)
|
|
FUNCTION = "cast_to_int"
|
|
CATEGORY = "Custom"
|
|
|
|
def cast_to_int(self, float_value, rounding_mode):
|
|
if rounding_mode == "up (ceil)":
|
|
return (int(math.ceil(float_value)),)
|
|
else:
|
|
return (int(math.floor(float_value)),)
|
|
|
|
NODE_CLASS_MAPPINGS = {
|
|
"FloatToInt": FloatToIntNode
|
|
} |