g(CdZddlmZddlZddlmZmZeedgd dZGddZ Gd d e Z e d d Z y)z?Contains everything that can help automate the cuts in MoviePy.) defaultdictN)convert_parameter_to_secondsuse_clip_fps_by_default start_timec fd}tj|jd|z dd}|d}|Dcgc]!}tj|||d#}}|tj|Scc}w)aFind the period of a video based on frames correlation. Parameters ---------- clip : moviepy.Clip.Clip Clip for which the video period will be computed. fps : int, optional Number of frames per second used computing the period. Higher values will produce more accurate periods, but the execution time will be longer. start_time : float, optional First timeframe used to calculate the period of the clip. Examples -------- .. code:: python from moviepy import * from moviepy.video.tools.cuts import find_video_period clip = VideoFileClip("media/chaplin.mp4").subclipped(0, 1).loop(2) round(videotools.find_video_period(clip, fps=80), 6) 1 cBj|jSN) get_frameflatten)tclips a/var/www/it7/html/youtubeDownloader/venv/lib/python3.12/site-packages/moviepy/video/tools/cuts.pyframez find_video_period..frame)s~~a ((**Nr)rr)nparangedurationcorrcoefargmax)r fpsrrtimingsrefr corrss` rfind_video_periodr su>+ii DMM1s7;AB?G (C7> ?!R[[eAh ' - ?E ? 299U# $$ @s&A:c.eZdZdZdZdZdZdZdZy) FramesMatchaWFrames match inside a set of frames. Parameters ---------- start_time : float Starting time. end_time : float End time. min_distance : float Lower bound on the distance between the first and last frames max_distance : float Upper bound on the distance between the first and last frames cP||_||_||_||_||z |_yr )rend_time min_distance max_distance time_span)selfrrr r!s r__init__zFramesMatch.__init__Es,$  ((!J.rcdd|j|j|j|jfzS)Nz(%.04f, %.04f, %.04f, %.04f)rrr r!r#s r__str__zFramesMatch.__str__Ls4- OO MM       1   rc"|jSr )r(r's r__repr__zFramesMatch.__repr__Ts||~rcpt|j|j|j|jfSr )iterrrr r!r's r__iter__zFramesMatch.__iter__Ws/ __dmmT->->@Q@Q R  rc|j|jk(xrO|j|jk(xr4|j|jk(xr|j|jk(Sr r&)r#others r__eq__zFramesMatch.__eq__\sb    / 8$--/ 8""d&7&77 8""d&7&77  rN) __name__ __module__ __qualname____doc__r$r(r*r-r0rrrr2s $/   rrc\eZdZdZdZd dZdZdZedZ ed dZ d d Z d Z y) FramesMatcheszFrames matches inside a set of frames. You can instantiate it passing a list of FramesMatch objects or using the class methods ``load`` and ``from_clip``. Parameters ---------- lst : list Iterable of FramesMatch objects. cHtj|t|dy)Nc|jSr )r!)es rz(FramesMatches.__init__..ss annrkey)listr$sorted)r#lsts rr$zFramesMatches.__init__rs dF3,DEFrNcj|t||zdz }|dk(r|dSt|dt|S)aReturns a new instance of FramesMatches object or a FramesMatch from the current class instance given different conditions. By default returns the first FramesMatch that the current instance stores. Parameters ---------- n : int, optional Number of matches to retrieve from the current FramesMatches object. Only has effect when ``percent=None``. percent : float, optional Percent of the current match to retrieve. Returns ------- FramesMatch or FramesMatches : If the number of matches to retrieve is greater than 1 returns a FramesMatches object, otherwise a FramesMatch. Ndrr)lenr7int)r#npercents rbestzFramesMatches.bestusA2  D G#c)Aq&tAwCmD3q6N&CCrc,tt||S)aReturn a FramesMatches object obtained by filtering out the FramesMatch which do not satistify a condition. Parameters ---------- condition : func Function which takes a FrameMatch object as parameter and returns a bool. Examples -------- .. code:: python # Only keep the matches corresponding to (> 1 second) sequences. new_matches = matches.filter(lambda match: match.time_span > 1) )r7filter)r# conditions rrIzFramesMatches.filters$VIt455rc tj|tj|Dcgc] }tjt|"c}ddycc}w)zSave a FramesMatches object to a file. Parameters ---------- filename : str Path to the file in which will be dumped the FramesMatches object data. z%.03f )fmt delimiterN)rsavetxtarrayr>)r#filenamer:s rsavezFramesMatches.saves@  HH6AbhhtAw'6 7  6s%Acvtj|}|Dcgc] }t| }}t|Scc}w)aLoad a FramesMatches object from a file. Parameters ---------- filename : str Path to the file to use loading a FramesMatches object. Examples -------- >>> matching_frames = FramesMatches.load("somefile") )rloadtxtrr7)rQarrr:mfss rloadzFramesMatches.loads9jj"(+,1{A,,S!!-s6c H|j|jzdzfdifd}g}|jd|D]\}}d|jz} | | } t j | } t jD]Y} || z |kDrj| t| d| z | d| zd| |<| |d |kD| |d <[tj} | | | d |<t| D]\}} | |d r||| }|x| |d <| |d <||k\| |d <| |d zdD]W}||| |}}t|d ||d z|d <t|d ||d z |d |z |d <|d |kDsSd|d <Y|Dcgc]*}||k7r#||d s||||d ||d f,c}z }t|Dcgc] }t| c}Scc}wcc}w)aFinds all the frames that look alike in a clip, for instance to make a looping GIF. Parameters ---------- clip : moviepy.video.VideoClip.VideoClip A MoviePy video clip. distance_threshold : float Distance above which a match is rejected. max_duration : float Maximal duration (in seconds) between two matching frames. fps : int, optional Frames per second (default will be ``clip.fps``). logger : str, optional Either ``"bar"`` for progress bar or ``None`` or any Proglog logger. Returns ------- FramesMatches All pairs of frames with ``end_time - start_time < max_duration`` and whose distance is under ``distance_threshold``. Examples -------- We find all matching frames in a given video and turn the best match with a duration of 1.5 seconds or more into a GIF: .. code:: python from moviepy import VideoFileClip from moviepy.video.tools.cuts import FramesMatches clip = VideoFileClip("foo.mp4").resize(width=200) matches = FramesMatches.from_clip( clip, distance_threshold=10, max_duration=3, # will take time ) best = matches.filter(lambda m: m.time_span > 1.5).best() clip.subclipped(best.start_time, best.end_time).write_gif("foo.gif") c0||zjz Sr )sum)F1F2N_pixelss r dot_productz,FramesMatches.from_clip..dot_productsG==?X- -rc|d|d}|d|d}}tj||zd|zz S)Nr|F|sq)rsqrt)t1t2uvuvr_ frame_dicts rdistancez)FramesMatches.from_clip..distancesWZ^G4jnW6MNBb>'*JrN7,CqA771q51r6>* *rT) with_timeslogger?|F|)minmaxrorejected)rrarnrprN)wh iter_framesr rrcr>keyspopabsr? enumeraterorpr7r)r distance_threshold max_durationrrlrjmatching_framesr r flat_frame F_norm_sqF_normret_Fidistt3t3tt2t3rdr:r^r_ris @@@r from_clipzFramesMatches.from_clips`66DFF?Q& .  + ((D(H1 HAuu}}.J#J ;IWWY'F:??,- Fl*NN2& #:b>%#86#AB)"~e4v=)JrN1% #2q)%03EErN1%j1 *+C&09VTJqM"3 /2b>!$Z02FJJ 2q!%(:b>!+"3EC!$SZU 1C!DCJ!$SZU 1CT%[SWEW!XCJ5z$66*.J / /, $ !GZ^A%6z%BQ 2q)%0*R.2CE2JK  O[1 fG!k1oGHH   Hs /HHc  ||}td}|D]\} }}||j ||g t|jd} g} d} | D]\}} || kr | D cgc]\} }}|  }}} }| D cgc]\} }}||kr| ||f}}} }|D cgc]\} }}| |z |kDr| ||f}}} }|sg| D chc] \} }}||kDr| }}} }|D chc] } | |z dks | }} |j |st d|D t fd|D\ }}| jt| ||||z} t| Scc}}} wcc}}} wcc}}} wcc}}} wcc} w)avSelect the scenes at which a video clip can be reproduced as the smoothest possible way, mainly oriented for the creation of GIF images. Parameters ---------- match_threshold : float Maximum distance possible between frames. The smaller, the better-looping the GIFs are. min_time_span : float Minimum duration for a scene. Only matches with a duration longer than the value passed to this parameters will be extracted. nomatch_threshold : float, optional Minimum distance possible between frames. If is ``None``, then it is chosen equal to ``match_threshold``. time_distance : float, optional Minimum time offset possible between matches. Returns ------- FramesMatches : New instance of the class with the selected scenes. Examples -------- .. code:: python from pprint import pprint from moviepy import * from moviepy.video.tools.cuts import FramesMatches ch_clip = VideoFileClip("media/chaplin.mp4").subclipped(1, 4) mirror_and_clip = [ch_clip.with_effects([vfx.TimeMirror()]), ch_clip] clip = concatenate_videoclips(mirror_and_clip) result = FramesMatches.from_clip(clip, 10, 3).select_scenes( 1, 2, nomatch_threshold=0, ) print(result) # [(1.0000, 4.0000, 0.0000, 0.0000), # (1.1600, 3.8400, 0.0000, 0.0000), # (1.2800, 3.7200, 0.0000, 0.0000), # (1.4000, 3.6000, 0.0000, 0.0000)] cgSr r5r5rrr;z-FramesMatches.select_scenes..qs"rc |dSNrr5)ks rr;z-FramesMatches.select_scenes..us !rr<rg333333?c3(K|] \}}}| ywr r5).0endr r!s r z.FramesMatches.select_scenes..sV?sL,cVsc34K|]}|dk(s |yw)rNr5)rr:rs rrz.FramesMatches.select_scenes..s 313s ) rappendr?items intersectionrpnextrr7)r#match_threshold min_time_spannomatch_threshold time_distance dict_startsstartr r! starts_endsresult min_startends_distancesrends great_matchesgreat_long_matches poor_matches short_matchess ` r select_sceneszFramesMatches.select_scenes;sf  $ / !*- 6: I 2E3 l   % %sL,&G H I[..0nE  %0$ . !E>y AOPP=S, CPDP:H5S, /1lL1M:G""5S, %K=0lL1" " &:H5S, "33L -1ISS5[S4HSIMI,,];VCUVVC.23-3/ +C| MM+eS, M N -II$ .LV$$EQ " Js$) E E$ E+ E2  E9,E9c ~|D]8\}}}}d|d|zd|zfz}|j||j|fi|:y)aNExtract the matching frames represented by the instance from a clip and write them as GIFs in a directory, one GIF for each matching frame. Parameters ---------- clip : video.VideoClip.VideoClip A video clip whose frames scenes you want to obtain as GIF images. gif_dir : str Directory in which the GIF images will be written. kwargs Passed as ``clip.write_gif`` optional arguments. Examples -------- .. code:: python import os from pprint import pprint from moviepy import * from moviepy.video.tools.cuts import FramesMatches ch_clip = VideoFileClip("media/chaplin.mp4").subclipped(1, 4) clip = concatenate_videoclips([ch_clip.time_mirror(), ch_clip]) result = FramesMatches.from_clip(clip, 10, 3).select_scenes( 1, 2, nomatch_threshold=0, ) os.mkdir("foo") result.write_gifs(clip, "foo") # MoviePy - Building file foo/00000100_00000400.gif with imageio. # MoviePy - Building file foo/00000115_00000384.gif with imageio. # MoviePy - Building file foo/00000128_00000372.gif with imageio. # MoviePy - Building file foo/00000140_00000360.gif with imageio. z%s/%08d_%08d.gifrBN) subclipped write_gif)r#r gifs_dirkwargsrr_names r write_gifszFramesMatches.write_gifssXP!% B E31%3;c (JJD 1DOOE3 ' 1 1$ A& A Br)rN)Nbarr) r1r2r3r4r$rGrIrR staticmethodrWrrrr5rrr7r7es^ GD:6( """pIpIfUVd%L*Brr7c4|/|j|d|Dcgc]}|j}}tj|t}| |j }nt |d|z z}ttj|}|j}dtjtj|||zkDdz} dgtd|z | zz|gz} t| | ddD cgc] \} } | | f } } } | |fScc}wcc} } w)a+Detects scenes of a clip based on luminosity changes. Note that for large clip this may take some time. Returns ------- tuple : cuts, luminosities cuts is a series of cuts [(0,t1), (t1,t2),...(...,tf)] luminosities are the luminosities computed for each frame of the clip. Parameters ---------- clip : video.VideoClip.VideoClip, optional A video clip. Can be None if a list of luminosities is provided instead. If provided, the luminosity of each frame of the clip will be computed. If the clip has no 'fps' attribute, you must provide it. luminosities : list, optional A list of luminosities, e.g. returned by detect_scenes in a previous run. luminosity_threshold : float, optional Determines a threshold above which the 'luminosity jumps' will be considered as scene changes. A scene change is defined as a change between 2 consecutive frames that is larger than (avg * thr) where avg is the average of the absolute changes between consecutive frames. logger : str, optional Either ``"bar"`` for progress bar or ``None`` or any Proglog logger. fps : int, optional Frames per second value. Must be provided if you provide no clip or a clip without fps attribute. Nuint32)rdtyperl)rrmrr) rtr[rrPfloatrrCrwdiffmeannonzeror>zip)r luminositiesluminosity_thresholdrlrfrluminosity_diffsavgluminosity_jumpsrrdrecutss r detect_scenesrs%V!--#Xf-U AEEG  88L6L mm,39-277<01    !C BHHRZZ 03G#3M MN OPQ RRcD#)'7788C5@G#&w #< =RRH =D =  !  >s D:D)Ng333333?)NN rN) r4 collectionsrnumpyrmoviepy.decoratorsrrrrr>r7rr5rrrshE#T|n-#%.#%L0 0 ffBDfBR MQ;;r