from .NoiseMathNode import NoiseMathNode
from .FloatMathNode import FloatMathNode
from .ConditioningMathNode import ConditioningMathNode
from .LatentMathNode import LatentMathNode
from .ImageMathNode import ImageMathNode
from .AudioMathNode import AudioMathNode
from .VideoMathNode import VideoMathNode

from .ModelMathNode import ModelMathNode, CLIPMathNode, VAEMathNode
from comfy_api.latest import ComfyExtension, io

class IntToFloatNode(io.ComfyNode):
    """
    Converts int to float.
    """
    @classmethod
    def define_schema(cls) -> io.Schema:
         """
         """
         return io.Schema(
             node_id="mrmth_IntToFloat",
             display_name="Int -> Float",
             category="More math",
             inputs=[
                 io.Int.Input(id="value", default=0),
             ],
             outputs=[
                 io.Float.Output(),
             ],
         )
    @classmethod
    def execute(cls, value):
        return (float(value),)

class FloatToIntNode(io.ComfyNode):
    """
    Converts float to int.
    """
    @classmethod
    def define_schema(cls) -> io.Schema:
        """
        """
        return io.Schema(
            node_id="mrmth_FloatToInt",
            category="More math",
            display_name="Float -> Int",
            inputs=[
                io.Float.Input(id="value", default=0.0),
            ],
            outputs=[
                io.Int.Output(),
            ],
        )
    @classmethod
    def execute(cls, value):
        return (int(value),)


class MoreMathExtension(ComfyExtension):
    def __init__(self):
        pass
    @classmethod
    async def get_node_list(self) -> list[type[io.ComfyNode]]:
            return [
                ConditioningMathNode,
                ModelMathNode,
                CLIPMathNode,
                VAEMathNode,
                LatentMathNode,
                ImageMathNode,
                FloatMathNode,
                NoiseMathNode,
                IntToFloatNode,
                FloatToIntNode,
                AudioMathNode,
                VideoMathNode
            ]
async def comfy_entrypoint() -> MoreMathExtension:
    return MoreMathExtension()