
    +j0                         d dl mZmZmZ  e            Zd Zd Zd Zd Zd Z	d Z
d Zd	 Zd
 Z G d de          ZdefdZdS )    )ComfyExtensionioComfyAPIc                  J  K   t                       d{V  t                       d{V  t                       d{V  t                       d{V  t	                       d{V  t                       d{V  t                       d{V  t                       d{V  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     ?/home/wildlama/comfy/ComfyUI/comfy_extras/nodes_replacements.pyregister_replacementsr      s      
*
,
,,,,,,,,
+
-
--------
,
.
........
*
,
,,,,,,,,
&
(
((((((((
)
+
++++++++
*
,
,,,,,,,,
/
1
1111111111r   c                     K   t           j                            t          j        dddgdddddddd	d
gdddg                     d {V  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      


'
'22)?"h77)]CC+)DD )*a889)
 )
 )
           r   c            
         K   t           j                            t          j        ddddddddg                     d {V  d S )	NBatchImagesNode
ImageBatchzimages.image0image1r   zimages.image1image2r#   r$   r&   r(   r   r   r   r   r       s}      


'
')$*h??*h??)
 )
 )
           r   c                     K   t           j                            t          j        ddddgddddd	d
ddddddg                     d {V  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      


'
'-&,j9"g66(7LMM3zJJ)5EFF		
)
 
)
 
)
 
 
 
 
 
 
 
 
 
 
 
r   c            	         K   t           j                            t          j        dddddg                     d {V  d S )NControlNetLoaderT2IAdapterLoadercontrol_net_namet2i_adapter_namer   r2   r(   r   r   r   r
   r
   9   ss      


'
'**-9KLL)
 )
 )
           r   c                  |   K   t           j                            t          j        dd                     d {V  d S )NLoad3DLoad3DAnimationr#   r$   r(   r   r   r   r   r   C   sb      


'
' ))
 )
 )
           r   c                  |   K   t           j                            t          j        dd                     d {V  d S )N	Preview3DPreview3DAnimationrB   r(   r   r   r   r   r   J   sb      


'
'#,)
 )
 )
           r   c                  |   K   t           j                            t          j        dd                     d {V  d S )NSVD_img2vid_ConditioningSDV_img2vid_ConditioningrB   r(   r   r   r   r   r   Q   sb      


'
'22)
 )
 )
           r   c                  |   K   t           j                            t          j        dd                     d {V  d S )NConditioningAveragezConditioningAverage rB   r(   r   r   r   r   r   X   sb      


'
'-.)
 )
 )
           r   c                   D    e Zd ZddZdeeej                          fdZdS )NodeReplacementsExtensionreturnNc                 2   K   t                       d {V  d S N)r   selfs    r   on_loadz!NodeReplacementsExtension.on_load`   s*      #%%%%%%%%%%%r   c                 
   K   g S rO   r   rP   s    r   get_node_listz'NodeReplacementsExtension.get_node_listc   s      	r   )rM   N)	__name__
__module____qualname__rR   listtyper   	ComfyNoderT   r   r   r   rL   rL   _   sO        & & & &T$r|*<%=      r   rL   rM   c                  "   K   t                      S rO   )rL   r   r   r   comfy_entrypointr\   f   s      $&&&r   N)comfy_api.latestr   r   r   r)   r   r   r   r	   r
   r   r   r   r   rL   r\   r   r   r   <module>r^      s   9 9 9 9 9 9 9 9 9 9hjj	2 	2 	2  	 	 	                   ' 9 ' ' ' ' ' 'r   