o
iy @ s d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlZd dl Z
d dl m Z d dl
mZmZmZ d dlmZ ejjZejZG dd dejZG dd d ejZG d
d d ejZG d d
d
ejZG dd dejZG dd dejZ G dd dejZ G dd dejZ G dd dejZ G dd dejZ G dd dejZ!G d d d ejZ"G d d d ejZ#G d d! d!ejZ$G d"d# d#ejZ%G d$d% d%ejZ&G d&d' d'ejZ'G d(d) d)eZ(d-d+d,Z)dS ). ) annotationsN) PromptServer)ComfyExtensionIOUI)overridec @ * e Zd Zedd ZedddZeZdS ) ImageCropc C s t jddgddddt jdt jjdd d
tjd
d t jjd d d
tjd
d t jjd
ddtjd
d t jjdddtjd
d gt j gdS )Nr trimzImage Crop (Deprecated)image/transformT Image Toolsimagewidth defaultminmaxstepheightxr y)node_idsearch_aliases display_namecategory
is_deprecatedessentials_categoryinputsoutputs)r SchemaImageInputIntnodesMAX_RESOLUTIONOutputcls r* 7/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_images.py
define_schema s
zImageCrop.define_schemareturn
IO.NodeOutputc C sb t ||jd d }t ||jd d }|| }|| }|d d ||||d d f }t|S )N r )r shaper
NodeOutput) r) r
r r r r to_xto_yimgr* r* r+ execute) s
zImageCrop.executeNr- r. )__name__
__module__ __qualname__ classmethodr, r5 cropr* r* r* r+ r s
r c @ & e Zd Zedd ZedddZdS ) ImageCropV2c C s< t jddgddddt jdt jjdd d
gt j gd S ) Nr= r
z
Image Cropr r Tr
crop_regionr ) component)r r r r r has_intermediate_outputr r )r r! r" r# BoundingBoxr' r( r* r* r+ r, 6 s
zImageCropV2.define_schemar- r. c
C s | dd}| dd}| dd}| dd}t||jd d }t||jd d }|| }|| }|d d ||||d d f } tj| t| d S )
Nr r r r r r r/ r ui)getr r0 r r1 r PreviewImage)
r) r
r> r r r r r2 r3 r4 r* r* r+ r5 F s zImageCropV2.executeNr6 r7 r8 r9 r: r, r5 r* r* r* r+ r= 5 s
r= c @ r< ) rA c
C sd t jdddt jjdddtdt jjdddtdt jjdd d
tdt jjd d d
tdgt j gd S )
NPrimitiveBoundingBoxz Bounding Boxzutils/primitiver r r r r r r r r r )r r r r r )r r! r$ r# r& rA r' r( r* r* r+ r, V s
zBoundingBox.define_schemar- r. c C s t ||||dS )N)r r r r )r r1 )r) r r r r r* r* r+ r5 e s zBoundingBox.executeNr6 rF r* r* r* r+ rA U s
rA c @ r ) RepeatImageBatchc C s< t jdddgdt jdt jjddddd gt j gd
S ) NrI zduplicate imagez clone image image/batchr
amountr rH r r r r r r r! r" r# r$ r' r( r* r* r+ r, k
z RepeatImageBatch.define_schemar- r. c C s | |dddf}t|S )Nr )repeatr r1 )r) r
rK sr* r* r+ r5 x s
zRepeatImageBatch.executeNr6 r7 r8 r9 r: r, r5 rP r* r* r* r+ rI j s
rI c @ r ) ImageFromBatchc C sN t jdg ddt jdt jjdddddt jjd d
d
d dgt j gd S )
NrS )z select imagezpick from batchz
extract imagerJ r
batch_indexr i rH lengthr rL rM rN r( r* r* r+ r, s
z ImageFromBatch.define_schemar- r. c C sJ |}t |jd d |}t |jd | |}|||| }t|S )Nr r )r r0 cloner r1 )r) r
rT rU s_inrQ r* r* r+ r5 s
zImageFromBatch.executeNr6 )r7 r8 r9 r: r, r5 frombatchr* r* r* r+ rS s
rS c @ r )
ImageAddNoisec
C sR t jddgdt jdt jjddddddd t jjd
d d d
ddgt j gdS )NrY z
film grainr
seedr l Tz,The random seed used for creating the noise.)r r r control_after_generatetooltipstrengthg ? ?{Gz?r rM )r r! r" r# r$ Floatr' r( r* r* r+ r, s"
zImageAddNoise.define_schemar- r. c C sB t |}t j||t j| |dd| ddd}t|S )Ncpu) generatordevicer^ r_ )r r )torch manual_seedcliprandnsizetor r1 )r) r
rZ r] rc rQ r* r* r+ r5 s
.
zImageAddNoise.executeNr6 rR r* r* r* r+ rY s
rY c @ s8 e Zd ZddddZedd Zed d
d d
ZeZd S )SaveAnimatedWEBP r )r fastestslowestc
C s t jddt jdt jjdddt jjddd d
d d t jjd d
dt jjdddddt jjdt | j
dgt j j
t j jgd
dS )Nrk image/animationimagesfilename_prefixComfyUIr fps @r` @@r losslessTqualityP r d rH methodoptionsr r r hiddenis_output_node)r r! r" r# Stringra Booleanr$ CombolistCOMPRESS_METHODSkeysHiddenprompt
extra_pnginfor( r* r* r+ r, s
z SaveAnimatedWEBP.define_schemar- r. c C s* t jtjj||| |||| j|ddS )N)rq rr r) ru rx ry r| rB )r r1 r ImageSaveHelperget_save_animated_webp_uir rD )r) rq ru rr rx ry r|
num_framesr* r* r+ r5 s
zSaveAnimatedWEBP.executeN)r r6 )r7 r8 r9 r r: r, r5 save_imagesr* r* r* r+ rk s
rk c @ s, e Zd Zedd Zed d
ddZeZdS ) SaveAnimatedPNGc
C s` t jddt jdt jjdddt jjddd d
d d t jjd d
ddddgt jjt jj gddS )Nr rp rq rr rs rt ru rv r` rw r compress_levelrl r T)r r r advancedr )
r r! r" r# r ra r$ r r r r( r* r* r+ r, s
z SaveAnimatedPNG.define_schemars r- r. c C s t jtjj||| ||ddS )N)rq rr r) ru r rB )r r1 r r get_save_animated_png_ui)r) rq ru r rr r* r* r+ r5 s zSaveAnimatedPNG.executeN)rs r6 )r7 r8 r9 r: r, r5 r r* r* r* r+ r s
r c @ s2 e Zd ZdZedd Ze d d
ddZeZdS ) ImageStitchz8Upstreamed from https://github.com/kijai/ComfyUI-KJNodesc C s t jdg ddddt jdt jjdg dd d
t jjd d d
t jjdddddd dt jjdg ddd dt jjdd dgt j gdS )Nr )zcombine imagesz join imageszconcatenate imagesz side by sidez Image StitchzStitches image2 to image1 in the specified direction.
If image2 is not provided, returns image1 unchanged.
Optional spacing can be added between images.r image1 direction)rightdownleftupr )r~ r match_image_sizeTrt
spacing_widthr r/ )r r r r r
spacing_colorwhiteblackredgreenbluer )r~ r r image2)optional)r r r descriptionr r r )r r! r" r# r r r$ r' r( r* r* r+ r, s
zImageStitch.define_schemaNr- r. c C s |d u r t |S |jd |jd krXt|jd |jd }|jd |k r;t||dd ||jd dddg}|jd |k rXt||dd ||jd dddg}|r|jdd \}} |jdd \}
} | |
} |dv r~|t|| }
}n | t| | }}
tj
| dd||
dd dd}dd d
d d d
}|| }|sl|jdd \}} |jdd \}
} d }t |t
s|}|dv r ||
krt||
}
||
k r|
| }|d ||d }}tjjj|dddd||fd|d}|
|
k r|
|
}|d ||d }}tjjj|dddd||fd|d}nP| | krlt| | }| |k rI|| }|d ||d }}tjjj|dd||fd|d}| |k rl|| }|d ||d }}tjjj|dd||fd|d}|jd |jd krt|jd |jd }|jd |k rtj|tjg |jd d ||jd R d|jigdd}|jd |k rtj|tjg |jd d ||jd R d|jigdd}|dkrY||d }|dv r|jd t|jd |jd ||jd f}n|jd |t|jd |jd |jd f}tj|d |jd}t |t
r?t|D ]\}} ||jd k r0| |d|f< q |jd dkr>d|d< n||dd td|jd f< |jd dkrYd|d< |dv rb||gn||g} |dkrq| d| |dv rxdnd} t tj| | dS )Nr r )r r lanczosdisabledr_ r^ )r_ r^ r^ )r^ r_ r^ )r^ r^ r_ r r/ constant)modevaluerd )dim)rd .rl ).r )r r )r r1 r0 r re catrP intcomfyutilscommon_upscalemovedim
isinstancetuplenn
functionalpadonesrd full enumerater insert) r) r r r r r r max_batchh1w1h2w2 aspect_ratiotarget_htarget_w color_map color_val pad_valuepad_hpad_top
pad_bottompad_wpad_left pad_right max_channels
spacing_shapespacingicrq
concat_dimr* r* r+ r5 s
$$
"
"
zImageStitch.executeNr6 )r7 r8 r9 __doc__r: r, r5 stitchr* r* r* r+ r s
r c @ r ) ResizeAndPadImagec
C s| t jddgdt jdt jjdddtjddt jjd ddtjddt jjd
d d gd
dt jjdg dd
dgt j gdS )Nr z fit to sizer r
target_widthr r r
target_height
padding_colorr r T)r~ r
interpolation)areabicubic
nearest-exactbilinearr rM ) r r! r" r# r$ r% r& r r' r( r* r* r+ r, s
z ResizeAndPadImage.define_schemar- r. c C s |j \}}}} || }
|| } t|
| } t|| }
t|| }|dddd}tj||
||d}|dkr8dnd}tj|| ||f||j |j
d }|| d }||
d }||d d d d ||| |||
f< |dddd}t |S )
Nr r r r/ r r r^ r_ )dtyperd )
r0 r r permuter r r re r r rd r r1 )r) r
r r r r
batch_size orig_height
orig_widthchannelsscale_wscale_hscale new_width
new_heightimage_permutedresizedr paddedy_offsetx_offsetoutputr* r* r+ r5 s(
(
zResizeAndPadImage.executeNr6 )r7 r8 r9 r: r, r5 resize_and_padr* r* r* r+ r s
r c @ s, e Zd Zedd Zed d dd ZeZd
S )
SaveSVGNodec C sB t jdddgddt jdt jjddd d
gt jjt jjgd d S )
Nr z
export vectorzsave vector graphicszSave SVG files on disk.z
image/savesvgrr svg/ComfyUIzThe prefix for the file to save. This may include formatting information such as %date:yyyy-MM-dd% or %Empty Latent Image.width% to include values from nodes.)r r\ T)r r r r r r r )r r! SVGr# r r r r r( r* r* r+ r, s
zSaveSVGNode.define_schemar r IO.SVG.Typer- r. c s^ t |t \}}}}}g }i }| jjd ur | jj|d< | jjd ur*|| jj |r3tj|ddnd } t |j
D ]l\}
} | dt |
} | d|dd}
|
d | d }| rsd
| d fd d
}tjd||tjd}ttj||
d}||d W d n1 sw Y |tj|
|t j j d |d7 }q:t j d|idS )Nr r/ )indentz %batch_num%_05z_.svgr zutf-8z3
c s | dd S )Nr
)group)matchmetadata_elementr* r+ replacement s z(SaveSVGNode.execute..replacementz (]*>))flagswb)filename subfoldertyper rq rB ) folder_pathsget_save_image_pathget_output_directoryr r r updatejsondumpsr datareplacestrseekreaddecoderesubUNICODEopenospathjoinwriteencodeappendr SavedResultr
FolderTyper r1 )r) r rr full_output_folderr counterr results
metadata_dict
metadata_json batch_number svg_bytesfilename_with_batch_numfile svg_contentr svg_filer* r r+ r5 s2
zSaveSVGNode.executeN)r )r r r- r. )r7 r8 r9 r: r, r5 save_svgr* r* r* r+ r s
0r c @ r ) GetImageSizec C sR t jdg ddddt jdgt jjddt jjddt jjd dgt jjgd
S ) Nr+ )
dimensions
resolutionz
image infozGet Image SizezGReturns width and height of the image, and passes it through unchanged.r
r )r r r )r r r r r r r r )r r! r" r# r$ r' r unique_idr( r* r* r+ r, + s
zGetImageSize.define_schemar- r. c C sX |j d }|j d }|j d }| jjr%tjd| d| d| | jj t|||S )Nr r/ r zwidth: z
, height: z
batch size: )r0 r r. r instancesend_progress_textr r1 )r) r
r r r r* r* r+ r5 > s
$zGetImageSize.executeNr6 )r7 r8 r9 r: r, r5 get_sizer* r* r* r+ r+ * s
r+ c @ r ) ImageRotatec C s@ t jddddgddt jdt jjdg d d
gt j gd S ) Nr2 z Image Rotateturnzflip orientationr r r
rotation)nonez
90 degreesz 180 degreesz 270 degreesr} )r r r r r r r r r! r" r# r r' r( r* r* r+ r, N s
zImageRotate.define_schemar- r. c C sP d}| dr
d}n| drd}n| drd}tj||ddgd}t|S ) Nr 90r 180r/ 270r )kdims)
startswithre rot90r r1 )r) r
r4 rotate_byr* r* r+ r5 ] s
zImageRotate.executeNr6 )r7 r8 r9 r: r, r5 rotater* r* r* r+ r2 M s
r2 c @ r ) ImageFlipc C s< t jdddgdt jdt jjdddgd gt j gd
S ) Nr@ mirrorreflectr r
flip_methodzx-axis: verticallyzy-axis: horizontallyr} rM r6 r( r* r* r+ r, n rO zImageFlip.define_schemar- r. c C s@ | drtj|dgd}n
| drtj|dgd}t|S )Nr r )r; r r/ )r< re flipr r1 )r) r
rC r* r* r+ r5 { s
zImageFlip.executeNr6 )r7 r8 r9 r: r, r5 rD r* r* r* r+ r@ m s
r@ c @ r ) ImageScaleToMaxDimensionc C sJ t jddt jdt jjdg ddt jjddd td
d gt j gd S )
NrE zimage/upscalingr
upscale_method)r r r r r r r} largest_sizer r r r )r r r r )r r! r" r# r r$ r& r' r( r* r* r+ r, s
z&ImageScaleToMaxDimension.define_schemar- r. c C s |j d }|j d }||krt|| | }|}n||kr(t|| | }|}n|}|}|dd}tj||||d}|dd}t|S )Nr r/ r r )r0 roundr r r r r r1 )r) r
rF rG r r samplesrQ r* r* r+ r5 s
z ImageScaleToMaxDimension.executeNr6 )r7 r8 r9 r: r, r5 upscaler* r* r* r+ rE s
rE c @ s0 e Zd Zedd Zedd Zedd ZdS )SplitImageToTileListc C sh t jddg dddt jdt jjddd td
t jjd dd td
t jjd d
ddd
gt jjddgdS )NrK rJ z split imagez
tile imagez slice imagez Split Image into List of TileszFSplits an image into a batched list of tiles with a specified overlap.r
tile_widthr @ rH tile_heightoverlap r rL Tis_output_list)r r r r r r r )r r! r" r# r$ r& r' r( r* r* r+ r, s
z"SplitImageToTileList.define_schemac C s g }t t|d || }t t|d || }d}||k red} t|| |}
td|
| } | | k rVt| | | } td| | }
||
| | |
f | | krNn| |7 } | | k s2|
|kr] |S ||7 }||k s |S )Ng ?r )rH r r r )r r rM rO rP coordsstride_xstride_yr r y_endy_startx_endx_startr* r* r+ get_grid_coords s,
z$SplitImageToTileList.get_grid_coordsc C sj |j \}}}}| |||||} g }
| D ]\} } }
}|d d | || |
d d f }|
| qt|
S r )r0 r[ r r r1 )r) r
rM rO rP bhwr rT output_listrZ rX rY rW tiler* r* r+ r5 s
z SplitImageToTileList.executeN)r7 r8 r9 r: r, staticmethodr[ r5 r* r* r* r+ rK s
rK c @ s$ e Zd Zedd Zedd ZdS )ImageMergeTileListc C sh t jdddg ddt jdt jjddd d
d t jjd dd d
d t jjd
dddd gt jjddgdS )Nrb z Merge List of Tiles to ImagerJ rL T
image_list final_widthr rN i rH final_heightrP rQ r rL FrR )r r r r
is_input_listr r rN r( r* r* r+ r, s
z ImageMergeTileList.define_schemac '
C sZ |d }|d }|d }d}|d } | j \}
} } }
| j}| j}t||| | |}tj|
|||
f||d}tj|
||df||d}|dkrttj tj
dd| ||d }ttj tj
dd| ||d }tj |dd}tj |dd}| d| d d d}t
|}t|||}n tjd| | df||d}t|D ]~\}\}}}} |t|kr no|| } | | } || } t| | j d } t| | j d }!|| }"||! }#| d d d | d |!d d f }$|d d d | d |!d d f }%|d d ||"||#d d f |$|% 7 < |d d ||"||#d d f |%7 < qd||dk< || }&t|&S ) Nr r_ )rd r r gh㈵>)r r r/ )r0 rd r rK r[ re zerossinmathpilinspaceclamp unsqueeze ones_likelerpr r lenr r r1 )'r) rc rd re rP r^ r] ovlp feather_str
first_tiler\ t_ht_wr rd r rT canvasweightsy_wx_w sine_mask flat_mask weight_maskr rZ rX rY rW r` region_hregion_wreal_hreal_w y_end_actual x_end_actual tile_crop mask_crop merged_imager* r* r+ r5 sJ
,*
zImageMergeTileList.executeNrF r* r* r* r+ rb s
rb c @ s e Zd ZedddZdS )ImagesExtensionr- list[type[IO.ComfyNode]]c s( t ttttttttt t
t t t
tttgS r )r r= rA rI rS rY rk r r r r r+ r2 r@ rE rK rb )selfr* r* r+
get_node_list; s& z ImagesExtension.get_node_listN)r- r )r7 r8 r9 r r r* r* r* r+ r : s r r- c s t S r )r r* r* r* r+ comfy_entrypointR s r )r- r )*
__future__r r% r r r r ri re comfy.utilsr serverr comfy_api.latestr r r typing_extensionsr r Typer& ComfyNoder r= rA rI rS rY rk r r r r r+ r2 r@ rE rK rb r r r* r* r* r+ sD & 13H# )=M