o
    i                     @   s   d dl Z d dlmZ d dlmZmZ G dd dejZG dd dejZG dd	 d	ejZ	G d
d dejZ
G dd dejZG dd deZdefddZdS )    N)override)ComfyExtensionioc                   @   0   e Zd Zedd ZededejfddZdS )Stringc                 C   &   t jdddt jdgt j gdS )NPrimitiveStringr   utils/primitivevaluenode_iddisplay_namecategoryinputsoutputsr   Schemar   InputOutputcls r   :/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_primitive.pydefine_schema      

zString.define_schemar
   returnc                 C   
   t |S Nr   
NodeOutputr   r
   r   r   r   execute      
zString.executeN	__name__
__module____qualname__classmethodr   strr   r   r!   r   r   r   r   r      
    
r   c                   @   r   )StringMultilinec              	   C   s,   t jddddt jjdddgt j gdS )	NPrimitiveStringMultilinezString (Multiline)r	   Basicsr
   T)	multiline)r   r   r   essentials_categoryr   r   r   r   r   r   r   r      s   
zStringMultiline.define_schemar
   r   c                 C   r   r   r   r    r   r   r   r!   '   r"   zStringMultiline.executeNr#   r   r   r   r   r*      s
    
r*   c                   @   r   )Intc              
   C   4   t jdddt jjdtj tjddgt j gdS )NPrimitiveIntr/   r	   r
   T)minmaxcontrol_after_generater   )r   r   r/   r   sysmaxsizer   r   r   r   r   r   -      
zInt.define_schemar
   r   c                 C   r   r   r   r    r   r   r   r!   9   r"   zInt.executeN)	r$   r%   r&   r'   r   intr   r   r!   r   r   r   r   r/   ,   r)   r/   c                   @   r   )Floatc              
   C   r0   )NPrimitiveFloatr9   r	   r
   g?)r2   r3   stepr   )r   r   r9   r   r5   r6   r   r   r   r   r   r   ?   r7   zFloat.define_schemar
   r   c                 C   r   r   r   r    r   r   r   r!   K   r"   zFloat.executeN)	r$   r%   r&   r'   r   floatr   r   r!   r   r   r   r   r9   >   r)   r9   c                   @   r   )Booleanc                 C   r   )NPrimitiveBooleanr=   r	   r
   r   )r   r   r=   r   r   r   r   r   r   r   Q   r   zBoolean.define_schemar
   r   c                 C   r   r   r   r    r   r   r   r!   ]   r"   zBoolean.executeN)	r$   r%   r&   r'   r   boolr   r   r!   r   r   r   r   r=   P   r)   r=   c                   @   s(   e Zd Zedeeej  fddZdS )PrimitivesExtensionr   c                    s   t ttttgS r   )r   r*   r/   r9   r=   )selfr   r   r   get_node_listc   s   z!PrimitivesExtension.get_node_listN)	r$   r%   r&   r   listtyper   	ComfyNoderB   r   r   r   r   r@   b   s    r@   r   c                      s   t  S r   )r@   r   r   r   r   comfy_entrypointm   s   rF   )r5   typing_extensionsr   comfy_api.latestr   r   rE   r   r*   r/   r9   r=   r@   rF   r   r   r   r   <module>   s    