U a+d" @sddlZddlZddlZddlZddlmZddlm Z ddl m Z m Z ddl mZzddlZdZWnek r|dZYnXe e dd d Ze e dddZdddZdS)N)DEVNULL) get_setting)requires_durationuse_clip_fps_by_default)subprocess_callTF ImageMagickOptimizeTransparencybarc Cst| } tj|\} } td|jd|} g}| d|d| dd| jt t | dD]2\}}d| |df}| ||j ||d d q^t d |}|d kr| d dtddd|dd|rdnddd|d| ddd|ddd|g | dk r dd| gng|g}n4|dkrNtddd d!d"t|d#| d$d"t||g }zt|| d%| d&|dWnRttfk r}z.d'|t|f}|d kr|d(}t|W5d}~XYnX|D]}t|qdS))a* Write the VideoClip to a GIF file. Converts a VideoClip into an animated GIF using ImageMagick or ffmpeg. Does the same as write_gif (see this one for more docstring), but writes every frame to a file instead of passing them in the RAM. Useful on computers with little RAM. r?zMoviePy - Building file %s messagez!MoviePy - - Generating GIF frames)tz%s_GIFTEMP%04d.pngr T)withmaskY@rz.MoviePy - - Optimizing GIF with ImageMagick...IMAGEMAGICK_BINARY-delay%d-dispose-loopz%s_GIFTEMP*.png -coalesce-fuzzz%02d%-layersz%sN-colorsffmpeg FFMPEG_BINARY-y-fZimage2-r-iz_GIFTEMP%04d.png)loggerzMoviePy - GIF ready: %s.zLMoviePy Error: creation of %s failed because of the following error: %s. .zThis error can be due to the fact that ImageMagick is not installed on your computer, or (for Windows users) that you didn't specify the path to the ImageMagick binary in file config_defaults.py.)proglogdefault_bar_loggerospathsplitextnpZarangedurationZiter_barlist enumerateappendZ save_frameintrstrrIOErrorOSErrorremove)clipfilenamefpsprogramoptfuzzverboseloopdisposecolorsr"fileNameextttZ tempfilesirnamedelaycmderrerrorfrF@/tmp/pip-unpacked-wheel-0yp4gafk/moviepy/video/io/gif_writers.pywrite_gif_with_tempfilessp            rHc Csd|} t| } |jdkr d}tdddddd d d d d |d d|j|jfd|rVdndddg} tttd}tjdkr~d|d<|dkrt j |d<t|d<t j | d|rdndd d ||gf|}n0t j |d<t j |d<t j | ddd ddgf|}|dkrtddd | d d!| rd"nd#d$d!|dd%g }|d&kr^|j |d<t|d<t j ||gf|}n(|j |d<t j |d<t j |d'gf|}|rtddd(d!|d)d*|g| dk rd+d!| gng|g}|j |d<t|d<t j |f|}| d,|d-| d.d-z\|j || d/d0d1D]D\}}|rFd2|j|}t||gd0}|j|qWnNtk r}z.d3|t|f}|dkr|d4}t|W5d}~XYnX|dkr| d5d-|j||dkr||r|| d6|d-dS)7aC Write the VideoClip to a GIF file, without temporary files. Converts a VideoClip into an animated GIF using ImageMagick or ffmpeg. Parameters ----------- filename Name of the resulting gif file. fps Number of frames per second (see note below). If it isn't provided, then the function will look for the clip's ``fps`` attribute (VideoFileClip, for instance, have one). program Software to use for the conversion, either 'ImageMagick' or 'ffmpeg'. opt (ImageMagick only) optimalization to apply, either 'optimizeplus' or 'OptimizeTransparency'. fuzz (ImageMagick only) Compresses the GIF by considering that the colors that are less than fuzz% different are in fact the same. Notes ----- The gif will be playing the clip in real time (you can only change the frame rate). If you want the gif to be played slower than the clip you will use :: >>> # slow down clip 50% and make it a gif >>> myClip.speedx(0.5).write_gif('myClip.gif') rNFrrz -loglevelrDrZrawvideoz-vcodecr z%.02fz-sz%dx%dz-pix_fmtZrgbaZrgb24r!-)stdoutstderrstdinnti creationflagsrrLrJZ image2pipeZbmprrrrrrr rr)FNzgif:-rrrrzMoviePy - Building file %sr z"MoviePy - - Generating GIF frames.Tuint8)r4r"Z with_timesdtypezN[MoviePy] Error: creation of %s failed because of the following error: %s. .zThis can be due to the fact that ImageMagick is not installed on your computer, or (for Windows users) that you didn't specify the path to the ImageMagick binary in file config_defaults.py.z,MoviePy - - Optimizing GIF with ImageMagick.zMoviePy - - File ready: %s.)r#r$maskrwhrr%r@spPIPEPopenrJ iter_framesZ get_framer(ZdstackZastyperLwritetostringr/r.closewait)r2r3r4r5r6r7r8rr9r:r;r"rAZcmd1Z popen_paramsZproc1Zcmd2Zproc2Zcmd3Zproc3rframerRrCrDrFrFrG write_gif^s>                      r^c Cs|dkr d}t|}ts"td|dkr0|j}|dkrsR    I