46 lines
1.0 KiB
Python
46 lines
1.0 KiB
Python
|
|
|
|
class BadmanIntUtil:
|
|
def __init__(self):
|
|
pass
|
|
|
|
|
|
|
|
@classmethod
|
|
def INPUT_TYPES(s):
|
|
return {
|
|
"required": {
|
|
"Int1": ("INT", {
|
|
"default": 0,
|
|
"min": 0}),
|
|
"Int2": ("INT", {
|
|
"default": 0,
|
|
"min": 0}),
|
|
"math_function": (["add", "multiply", "sub", "divide"],),
|
|
},
|
|
}
|
|
|
|
RETURN_TYPES = ("INT",)
|
|
FUNCTION = "process_int"
|
|
|
|
CATEGORY = "Badman"
|
|
|
|
def process_int(self, Int1: int, Int2: int, math_function: str):
|
|
|
|
processed_int = self.math_mode(Int1, Int2, math_function)
|
|
|
|
return (processed_int,)
|
|
|
|
def math_mode(self, Int1, Int2, mode):
|
|
if mode == "add":
|
|
return Int1+Int2
|
|
elif mode == "multiply":
|
|
return Int1 * Int2
|
|
elif mode == "sub":
|
|
return Int1 - Int2
|
|
elif mode == "divide":
|
|
return Int1/Int1
|
|
else:
|
|
raise ValueError(f"Unsupported math mode: {mode}")
|
|
|