U a+d*@sbdZddlmZddlZddlmZedddZGdd d ZGd d d e Z edddZ dS)zN This module contains everything that can help automatize the cuts in MoviePy ) defaultdictN)use_clip_fps_by_default333333?csRfddt|jd|dd}dfdd|D}|t|S) z9 Finds the period of a video based on frames correlation cs|SN)Z get_frameflatten)t)clipz#find_video_period..?Nrcs g|]}t|dqS))rr)npZcorrcoef).0r)framerefr r sz%find_video_period..)rZarangedurationZargmax)rfpsZtminttZcorrsr )rrrr find_video_period s  rc@s0eZdZdZddZddZddZdd Zd S) FramesMatchz Parameters ----------- t1 Starting time t2 End time d_min Lower bound on the distance between the first and last frames d_max Upper bound on the distance between the first and last frames cCs&||_||_||_||_|||_dSr)t1t2d_mind_maxZ time_span)selfrrrrr r r __init__*s zFramesMatch.__init__cCsd|j|j|j|jfSNz(%.04f, %.04f, %.04f, %.04f)rrrrrr r r __str__1s zFramesMatch.__str__cCsd|j|j|j|jfSrr r!r r r __repr__6s zFramesMatch.__repr__cCst|j|j|j|jfSr)iterrrrrr!r r r __iter__:szFramesMatch.__iter__N)__name__ __module__ __qualname____doc__rr"r#r%r r r r rs rc@sZeZdZddZdddZddZd d Zed d Zedd dZ dddZ ddZ dS) FramesMatchescCst|t|ddddS)NcSs|jSr)r)er r r r Br z(FramesMatches.__init__..key)listrsorted)rlstr r r r@szFramesMatches.__init__rNcCs8|dk rt||d}|dkr(|dSt|d|S)Ndrr)lenr*)rnpercentr r r bestDszFramesMatches.bestcCstt||S)a Returns a FramesMatches object obtained by filtering out the FramesMatch which do not satistify the condition ``cond``. ``cond`` is a function (FrameMatch -> bool). Examples --------- >>> # Only keep the matches corresponding to (> 1 second) sequences. >>> new_matches = matches.filter( lambda match: match.time_span > 1) )r*filter)rZcondr r r r6Is zFramesMatches.filtercCs&tj|tdd|DddddS)NcSsg|]}tt|qSr )rarrayr.rr+r r r rWsz&FramesMatches.save..z%.03f )fmt delimiter)rZsavetxtr7)rfilenamer r r saveVszFramesMatches.savecCs t|}dd|D}t|S)zp Loads a FramesMatches object from a file. >>> matching_frames = FramesMatches.load("somefile") cSsg|] }t|qSr rr8r r r r`sz&FramesMatches.load..)rZloadtxtr*)r<ZarrZmfsr r r loadZs zFramesMatches.loadcs|j|jdfddifdd}g}|jdddD]\}d |}||}t|} tD]j} | |kr| qvt | d | | d | d | <| d |k| d <qvt } ||| d<t | D]\} } | d r$q|| } | | d <| d<| |k| d <| | ddD]t}|| |}}t |d| |d|d<t |d | |d|d | |d <|d |krrd|d <qrq|fddD7}q@tdd|DS)a Finds all the frames tht look alike in a clip, for instance to make a looping gif. This teturns a FramesMatches object of the all pairs of frames with (t2-t1 < max_d) and whose distance is under dist_thr. This is well optimized routine and quite fast. Examples --------- We find all matching frames in a given video and turn the best match with a duration of 1.5s or more into a GIF: >>> from moviepy.editor import VideoFileClip >>> from moviepy.video.tools.cuts import find_matching_frames >>> clip = VideoFileClip("foo.mp4").resize(width=200) >>> matches = find_matching_frames(clip, 10, 3) # will take time >>> best = matches.filter(lambda m: m.time_span > 1.5).best() >>> clip.subclip(best.t1, best.t2).write_gif("foo.gif") Parameters ----------- clip A MoviePy video clip, possibly transformed/resized dist_thr Distance above which a match is rejected max_d Maximal duration (in seconds) between two matching frames fps Frames per second (default will be clip.fps) cs||Srsum)ZF1ZF2)N_pixelsr r r r z)FramesMatches.from_clip..csJ|d|d}|d|d}}t||d|S)Nr|F|sq)rsqrt)rrZuvuv)F dot_productr r distancesz)FramesMatches.from_clip..distanceTbar)Z with_timesloggerr |F|)minmaxrOrejected)rrDrNrPrNcsHg|]@}|kr|ds||d|dfqS)rQrOrPr )rr)rIrr r rsz+FramesMatches.from_clip..cSsg|] }t|qSr r>r8r r r rs)wh iter_framesrrrFr.keyspopabsr/ enumeraterOrPr*)rZdist_thrZmax_drrKZmatching_framesrZ flat_frameZ F_norm_sqZF_normrZt_Fidistt3Zt3tZt2t3r )rIrCrJrr from_clipesD(      "    zFramesMatches.from_cliprcs.dkr tdd}|D] \}}|||gqt|ddd}g} d} |D]\} | krpq^dd| D} fd d| D} fd d| D}|sq^fd d | D}fd d | D}||sq^tdd|Dtfdd|D\}}| t|||} q^t| S)a! match_thr The smaller, the better-looping the gifs are. min_time_span Only GIFs with a duration longer than min_time_span (in seconds) will be extracted. nomatch_thr If None, then it is chosen equal to match_thr NcSsgSrr r r r r r r z-FramesMatches.select_scenes..cSs|dS)Nrr )kr r r r r r,rcSsg|]\}}}|qSr r rendrrr r r rsz/FramesMatches.select_scenes..cs$g|]\}}}|kr|||fqSr r r^) match_thrr r rscs(g|] \}}}|kr|||fqSr r r^) min_time_spanstartr r rs csh|]\}}}|kr|qSr r r^) nomatch_thrr r sz.FramesMatches.select_scenes..csh|]}|dkr|qS)g333333?r )rr_)rbr r rds css|]\}}}|VqdSrr r^r r r sz.FramesMatches.select_scenes..c3s|]}|dkr|VqdS)rNr r8)r_r r res ) rappendr/items intersectionrPnextrr*)rr`rarcZ time_distanceZ dict_startsrrZ starts_endsresultZ min_startZends_distancesZendsZ great_matchesZgreat_long_matchesZ poor_matchesZ short_matchesr )r_r`rarcrbr select_sceness:      zFramesMatches.select_scenescCsB|D]8\}}}}d|d|d|f}|||j|ddqdS)Nz%s/%08d_%08d.gifr1F)verbose)ZsubclipZ write_gif)rrZgif_dirrbr__namer r r write_gifsszFramesMatches.write_gifs)rN)N)Nr) r&r'r(rr5r6r= staticmethodr?r\rkror r r r r*>s    d .Zuint32)rdtyperM)rsr rrcSsg|]\}}||fqSr r )rrrr r r rGs) rTrr7floatrr2rWZdiffZmeanZnonzeror.zip) rZ luminositiesZthrrMrr_Z lum_diffsavgZluminosity_jumpsrZcutsr r r detect_sceness +   rw)Nr)NNrqrLN) r) collectionsrZnumpyrZmoviepy.decoratorsrrrr.r*rwr r r r s   (Q