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 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,
                LatentMathNode,
                ImageMathNode,
                FloatMathNode,
                NoiseMathNode,
                IntToFloatNode,
                FloatToIntNode,
                AudioMathNode,
                VideoMathNode
            ]
async def comfy_entrypoint() -> MoreMathExtension:
    return MoreMathExtension()