o i*@sddlmZddlZddlZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZmZmZmZmZddlmZGdd d ejZGd d d ejZGd d d ejZGdddejZGdddejZGdddejZGdddeZdddZdS)) annotationsN)Optional)override)Fraction)ComfyExtensioniouiInput InputImplTypes)argsc@s&eZdZeddZedddZdS) SaveWEBMcCsztjddgddtjdtjjdddtjjd d d gd tjjd dddddtjjdddddddgtjjtjj gddS)Nr z export webm image/videoTimagesfilename_prefixComfyUI)defaultcodecvp9av1)optionsfpsg8@g{Gz?g@@rminmaxstepcrfg@@rgO@zhHigher crf means lower quality with a smaller file size, lower crf means higher quality higher filesize.)rrrrtooltip)node_idsearch_aliasescategoryis_experimentalinputshiddenis_output_node) rSchemaImager StringComboFloatHiddenprompt extra_pnginfoclsr06/mnt/c/Users/fbmor/ComfyUI/comfy_extras/nodes_video.py define_schemas zSaveWEBM.define_schemareturn io.NodeOutputcCst|t|djd|djd\}}}} }|d|dd} tjtj|| dd} |j j dur>t |j j | j d<|j jdurW|j jD]} t |j j| | j | <qHd d d } | j| |tt|d d d }|jd|_|jd|_|dkr~dnd|_d|_dt|i|_|dkrd|jd<|D]0}tjjtj|dddfddddjtdtjddd}| |D]}| !|qq| !| | "t#j$t%&t%'| | t#j(j)gdS)Nrr_05z_.webmw)moder,z libvpx-vp9 libsvtav1)rri)rater yuv420p10leyuv420pr6preset.)rrcpu)devicedtypergb24)formatr)* folder_pathsget_save_image_pathget_output_directoryshapeavopenospathjoinr$r,jsondumpsmetadatar- add_streamrroundwidthheightpix_fmtbit_ratestrr VideoFrame from_ndarraytorchclamptorDuint8numpyencodemuxcloser NodeOutputr PreviewVideo SavedResult FolderTypeoutput)r/rrrrrfull_output_folderfilenamecounter subfolderfile containerx codec_mapstreamframepacketr0r0r1execute!s6         B "zSaveWEBM.executeNr3r4__name__ __module__ __qualname__ classmethodr2rvr0r0r0r1r   r c@s&eZdZeddZed d d Zd S) SaveVideocCsxtjddgddddtjjddd tjjd d d d tjjdtjdddtjjdtj dddgtj j tj j gdd S)Nr~z export videoz Save VideorBasicsz8Saves the input images to your ComfyUI output directory.videozThe video to save.rrz video/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.rrrGautoz The format to save the video as.)rrrrzThe codec to use for the video.T) rr display_namer!essentials_category descriptionr#r$r%) rr&Videor r(r)r VideoContaineras_input VideoCodecr+r,r-r.r0r0r1r2EszSaveVideo.define_schemar Input.VideorGr[r3r4cCs|\}}t|t||\}}} } }d} tjs=i} |jjdur)| |jj|jj dur5|jj | d<t | dkr=| } |d| ddt j |} |jtj|| t ||| dtjtt| | tjjgdS)Nr,rr5r6z_.)rGrrTrH)get_dimensionsrIrJrKr disable_metadatar$r-updater,lenr r get_extensionsave_torOrPrQrrfrrgrhrirj)r/rrrGrrWrXrkrlrmrnsaved_metadatarTror0r0r1rvXs0      "zSaveVideo.executeN)rrrGr[r3r4rxr0r0r0r1r~Ds  r~c@s(eZdZeddZeddd d ZdS) CreateVideocCsTtjddgdddtjjdddtjjd d d d d d tjjddddgtjgdS)Nrzimages to videoz Create VideorzCreate a video from images.rz"The images to create a video from.rrg>@g?g^@raudioTzThe audio to add to the video.)optionalrrr rr!rr#outputs)rr&r'r r*AudiorOutputr.r0r0r1r2vszCreateVideo.define_schemaNr Input.ImagerfloatrOptional[Input.Audio]r3r4c Cs tttj||t|dS)N)rr frame_rate)rrfr VideoFromComponentsr VideoComponentsr)r/rrrr0r0r1rvszCreateVideo.executeN)rrrrrrr3r4rxr0r0r0r1rus  rc@s&eZdZeddZed ddZd S) GetVideoComponentsc CsNtjdgddddtjjdddgtjjd d tjjd d tjjd d gd S)Nr)zextract framesz split videozvideo to imagesdemuxzGet Video ComponentsrzCExtracts all components from a video: frames, audio, and framerate.rz%The video to extract components from.rr)rrrr)rr&rr r'rrr*r.r0r0r1r2s   z GetVideoComponents.define_schemarrr3r4cCs |}t|j|jt|jSr)get_componentsrrfrrrr)r/r componentsr0r0r1rvszGetVideoComponents.executeN)rrr3r4rxr0r0r0r1rr}rc@s>eZdZeddZed ddZeddZed d Zd S) LoadVideoc sjtfddtD}t|dg}tjdgddddtjjd t |tj j d gtj gd S) Ncs&g|]}tjtj|r|qSr0)rOrPisfilerQ).0f input_dirr0r1 s&z+LoadVideo.define_schema..rr)z import videoz open videoz video filez Load Videorrro)rupload)rr rr!rr#r)rIget_input_directoryrOlistdirfilter_files_content_typesrr&r)r sorted UploadTyperrr)r/filesr0rr1r2szLoadVideo.define_schemar3r4cCst|}tt|Sr)rIget_annotated_filepathrrfr VideoFromFile)r/ro video_pathr0r0r1rvs zLoadVideo.executecCst|}tj|}|Sr)rIrrOrPgetmtime)srormod_timer0r0r1fingerprint_inputss  zLoadVideo.fingerprint_inputscCst|s d|SdS)NzInvalid video file: {}T)rIexists_annotated_filepathrG)rror0r0r1validate_inputss  zLoadVideo.validate_inputsNrw)ryrzr{r|r2rvrrr0r0r0r1rs   rc@s&eZdZeddZedd d ZdS) VideoSlicecCshtjddgdddtjdtjjdddd d d d tjjd ddd ddtjjddddgtjgdS)Nz Video Slice)ztrim video durationzskip first frameszframe load capz start timerz Video Toolsr start_timeggj@gjgMbP?zStart time in seconds)rrrrrdurationz0Duration in seconds, or 0 for unlimited duration)rrrrstrict_durationFzNIf True, when the specified duration is not possible, an error will be raised.r)rrr r!rr#r)rr&rr r*Booleanrr.r0r0r1r2s> zVideoSlice.define_schemar io.Video.Typerrrrboolr3r4cCs@|j|||d}|durt|Std|d|d|)N)rz(Failed to slice video: Source duration: z Start time: z Target duration: ) as_trimmedrrf ValueError get_duration)r/rrrrtrimmedr0r0r1rvs  zVideoSlice.executeN) rrrrrrrrr3r4rxr0r0r0r1rs  (rc@seZdZedddZdS)VideoExtensionr3list[type[io.ComfyNode]]csttttttgSr)r r~rrrr)selfr0r0r1 get_node_listszVideoExtension.get_node_listN)r3r)ryrzr{rrr0r0r0r1rsrr3cstSr)rr0r0r0r1comfy_entrypointsr)r3r) __future__rrOrMr^rIrRtypingrtyping_extensionsr fractionsrcomfy_api.latestrrrr r r comfy.cli_argsr ComfyNoder r~rrrrrrr0r0r0r1s&      61(4