o
    i                     @   s   d dl mZmZmZ e Zdd Zdd Zdd Zdd	 Zd
d Z	dd Z
dd Zdd Zdd ZG dd deZdefddZdS )    )ComfyExtensionioComfyAPIc                      sf   t  I dH  t I dH  t I dH  t I dH  t I dH  t I dH  t I dH  t I dH  dS )z(Register all built-in node replacements.N) register_replacements_longeredge!register_replacements_batchimages"register_replacements_upscaleimage register_replacements_controlnetregister_replacements_load3dregister_replacements_preview3d register_replacements_svdimg2vid%register_replacements_conditioningavg r   r   =/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_replacements.pyregister_replacements   s   r   c                      sJ   t jtjdddgdddddddd	d
gdddgdI d H  d S )NImageScaleToMaxDimensionResizeImagesByLongerEdgelonger_edgeimageimagesnew_idold_idlargest_sizeupscale_methodlanczosr   	set_valuer   )new_idxold_idx)new_node_idold_node_idold_widget_idsinput_mappingoutput_mappingapinode_replacementregisterr   NodeReplacer   r   r   r   r      s   

r   c                	      s4   t jtjddddddddgdI d H  d S )	NBatchImagesNode
ImageBatchzimages.image0image1r   zimages.image1image2r   r    r"   r$   r   r   r   r   r       s   
r   c                      sJ   t jtjddddgddddd	d
ddddddgdI d H  d S )NResizeImageMaskNodeImageScaleByr   scale_byinputr   r   resize_typezscale by multiplierr   zresize_type.multiplierscale_method)r   r    r!   r"   r$   r   r   r   r   r   +   s   
r   c                      s,   t jtjdddddgdI d H  d S )NControlNetLoaderT2IAdapterLoadercontrol_net_namet2i_adapter_namer   r-   r$   r   r   r   r   r   9   s   
r   c                      "   t jtjdddI d H  d S )NLoad3DLoad3DAnimationr   r    r$   r   r   r   r   r	   C   
   
r	   c                      r8   )N	Preview3DPreview3DAnimationr;   r$   r   r   r   r   r
   J   r<   r
   c                      r8   )NSVD_img2vid_ConditioningSDV_img2vid_Conditioningr;   r$   r   r   r   r   r   Q   r<   r   c                      r8   )NConditioningAveragezConditioningAverage r;   r$   r   r   r   r   r   X   r<   r   c                   @   s.   e Zd ZdddZdeeej  fddZdS )NodeReplacementsExtensionreturnNc                    s   t  I d H  d S N)r   selfr   r   r   on_load`   s   z!NodeReplacementsExtension.on_loadc                    s   g S rD   r   rE   r   r   r   get_node_listc   s   z'NodeReplacementsExtension.get_node_list)rC   N)	__name__
__module____qualname__rG   listtyper   	ComfyNoderH   r   r   r   r   rB   _   s    
rB   rC   c                      s   t  S rD   )rB   r   r   r   r   comfy_entrypointf   s   rO   N)comfy_api.latestr   r   r   r%   r   r   r   r   r   r	   r
   r   r   rB   rO   r   r   r   r   <module>   s    
