l j) dZddlmZdgZddlZddlZddlZddlmZddl m Z ddl m Z ddl mZmZdd lmZdd lmZmZddlZddlZddlZdd lmZej5ejd d eddlmZdddn #1swxYwYddl m!Z!ddl"m#Z#m$Z$ddl%m&Z&ddl'm(Z(ddl)m*Z*m+Z+m,Z,m-Z-m.Z.m/Z/m0Z0ddl1m2Z2ddl3m4Z4m5Z5er ddl6m7Z7ddl8m9Z9ddl:m;Z;ddlm?Z?m@Z@d,d"ZAd-d*ZBGd+dZCdS).z(The interface between scenes and ffmpeg.) annotationsSceneFileWriterN)Fraction)Path)Queue)NamedTemporaryFile_TemporaryFileWrapper)Thread) TYPE_CHECKINGAny)Imageignorez.*ffmpeg or avconv.*)messagecategory) AudioSegment) __version__)configlogger)set_file_logger) RendererType)add_extension_if_not_presentadd_version_before_extensionguarantee_existence is_gif_format is_png_format modify_atimewrite_to_movie)get_full_sound_file_path)DefaultSectionTypeSection)OutputContainer)Stream) CairoRenderer)OpenGLRenderer) PixelArrayStrPathfpsfloatreturnrcLd}d}t|tr|d}}nwt|t|z |krt|d}}nBd}t||zdz dz}t|||z z |krt dt ||S)Ng-C6?g{Gz?r izinvalid frame rate) isinstanceintabsround ValueErrorr)r)epsilon1epsilon2numdenoms X/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/scene/scene_file_writer.pyto_av_frame_rater8;sHH#s3Qe S5::    ) )c AeC%K$&''$. sS5[ ! !X - -122 2 C   input_pathr output_path#Path | _TemporaryFileWrapper[bytes] codec_namestrNonectj|5}tj|d5}|jjd}||}||D]/}||D]}||0|D]}|| dddn #1swxYwYddddS#1swxYwYdS)Nwr)avopenstreamsaudio add_streamdecodeencodemux) r:r;r= input_audio output_audioinput_audio_streamoutput_audio_streamframepackets r7 convert_audiorPLs   %*  S!! %%1(06q9*55jAA ''(:;; ) )E-44U;; ) )  (((( )*0022 % %F   V $ $ $ $ % % % % % % % % % % % % % % % % % % % % % % % % % % % % % % % % % %s5C)BC C)C C)C C))C-0C-ceZdZdZdZdLd ZdMd ZdNd ZdOdZdPdZ dQdZ dNdZ dNdZ dRdSdZ dRdTd#Z dUdVd'ZdWdXd(ZdNd)ZdYd.Z dZd[d2Zd\d8Zd]d9ZdNd:Zd^d_d;ZdNd<Zd`d>Z dadbdEZdNdFZdNdGZdNdHZdNdIZdNdJZdcdKZ dS)draSceneFileWriter is the object that actually writes the animations played, into video files, using FFMPEG. This is mostly for Manim's internal use. You will rarely, if ever, have to use the methods for this class, unless tinkering with the very fabric of Manim's reality. Attributes ---------- sections : list of :class:`.Section` used to segment scene sections_output_dir : :class:`pathlib.Path` where are section videos stored output_name : str name of movie without extension and basis for section video names Some useful attributes are: "write_to_movie" (bool=False) Whether or not to write the animations into a video file. "movie_file_extension" (str=".mp4") The file-type extension of the outputted video. "partial_movie_files" List of all the partial-movie files. FrendererCairoRenderer | OpenGLRenderer scene_namer>kwargsr r+r?c ||_|||d|_g|_g|_g|_|dtj ddS)Nr autocreatedF)nametype_skip_animations) rRinit_output_directories init_audio frame_countpartial_movie_files subcaptionssections next_sectionr!NORMAL)selfrRrTrUs r7__init__zSceneFileWriter.__init__{s !  $$Z000 57 /1')  &8&?QV      r9ctdrdStdrtjdjnd}tjrt ||_nHtdr'tdstjd|_nt ||_tdrAttjd|| }|t|jd z |_ tr]ttjd || }|t|jtd z |_ t d|_ tj r)ttjd || |_ trOt|jd|_tdst!|j|_||jz |_ttjd|||_tdr7ttjd}t%|||dSdSdS)zInitialise output directories. Notes ----- The directories are read from ``config``, for example ``config['media_dir']``. If the target directories don't already exist, they will be created. dry_runN input_file output_file write_all media_dir images_dir) module_namerTz.png video_dirmovie_file_extension sections_dirz.gifpartial_movie_dir)rTrm log_to_filelog_dir)rTrmrs)rget_dirstemrforce_output_as_scene_namer output_namerrimage_file_pathrmovie_file_pathsections_output_dir save_sectionsr gif_file_pathrpartial_movie_directoryr)rcrTrm image_dir movie_dirrss r7r[z'SceneFileWriter.init_output_directoriess )   F;A,;OWfn\2277UW  5 0#J//D   M " 06++> 0%~m<N&KJ,,(  D%A$f&&"m,)E***D&&/1C%C"+>') +,,D (m$ -fnY.G.GHH){GS+ + N  r9ct|jr:|jdr|jdSdSdS)z&Delete current section if it is empty.N)lenr`is_emptypoprcs r7finish_last_sectionz#SceneFileWriter.finish_last_sections\ t}   $-"3"<"<">"> M           r9rXrYrZboolc2|d}tjsJtrr}r)rcrnew_partial_movie_files r7add_partial_movie_filez&SceneFileWriter.add_partial_movie_filest677 ~?O?O  F  !  $ + +D 1 1 1 M"  1 8 8 > > > > >%(,#EV,B%CEEF&& "  $ + +,B C C C M"  1 8 89O P P P P Pr9cDtd}td}|d|S)aGet the name of the resolution directory directly containing the video file. This method gets the name of the directory that immediately contains the video file. This name is ``p``. For example, if you are rendering an 854x480 px animation at 15fps, the name of the directory that immediately contains the video, file will be ``480p15``. The file structure should look something like:: MEDIA_DIR |--Tex |--texts |--videos |-- |--p |--partial_movie_files |--.mp4 |--.srt Returns ------- :class:`str` The name of the directory. pixel_height frame_ratep)r)rcrrs r7get_resolution_directoryz(SceneFileWriter.get_resolution_directorys,6n- L) -----r9cd|_dS)z/Preps the writer for adding audio to the movie.FN)includes_soundrs r7r\zSceneFileWriter.init_audio5s#r9c6tj|_dS)z(Creates an empty, silent, Audio Segment.N)rsilent audio_segmentrs r7create_audio_segmentz$SceneFileWriter.create_audio_segment9s)022r9N new_segmentrtime float | Nonegain_to_backgroundc |jsd|_||j}|j}||}|dkrt d||jz}||z }|dkrK|t jttj |dzd}| |td|z||_dS)amThis method adds an audio segment from an AudioSegment type object and suitable parameters. Parameters ---------- new_segment The audio segment to add time the timestamp at which the sound should be added. gain_to_background The gain of the segment from the background. TNrzAdding sound at timestamp < 0r-) crossfade)positiongain_during_overlay) rrrduration_secondsr2rrrr/npceiloverlay)rcrrrsegmentcurr_endnew_enddiffs r7add_audio_segmentz!SceneFileWriter.add_audio_segment=s(" ("&D   % % ' ' '$+ <D !88<== =55! !88nn#Ct (<(<$=$=>>%G%__ %% 2-  r9 sound_filer(gainc t|}|jdvrtdd5}t||dt j|j}tjd|ddddn #1swxYwYt|j nt j|}|r| |}|j ||fi|dS) aThis method adds an audio segment from a sound file. Parameters ---------- sound_file The path to the sound file. time The timestamp at which the audio should be added. gain The gain of the given audio segment. **kwargs This method uses add_audio_segment, so any keyword arguments used there can be referenced here. ).wavz.rawrF)suffixdelete pcm_s16lezAutomatically converted z to .wavN) rrrrPr from_filerXrinforunlink apply_gainr)rcrrrrU file_path wav_file_pathrs r7 add_soundzSceneFileWriter.add_soundhsJ2-Z88   #3 3 3$6%@@@ LMi DDD*4]5GHH  JyJJJKKK L L L L L L L L L L L L L L L # $ $ + + - - - -&0;;K  7%0066K{D;;F;;;;;sAA99A=A= allow_writerStrPath | NonecZtr|r||dSdSdS)zUsed internally by manim to stream the animation to FFMPEG for displaying or writing to a file. Parameters ---------- allow_write Whether or not to write to a video file. )rN)ropen_partial_movie_stream)rcrrs r7begin_animationzSceneFileWriter.begin_animationsP    @  @  * *Y * ? ? ? ? ? @ @ @ @r9cVtr|r|dSdSdS)zInternally used by Manim to stop streaming to FFMPEG gracefully. Parameters ---------- allow_write Whether or not to write to a video file. N)rclose_partial_movie_stream)rcrs r7 end_animationzSceneFileWriter.end_animationsE    .  .  + + - - - - - . . . .r9cr |j\}}|dS|||7)zHFor internal use only: blocks until new frame is available on the queue.TN)queuegetencode_and_write_frame)rc num_frames frame_datas r7listen_and_writez SceneFileWriter.listen_and_writesE @%)Z^^%5%5 "J !  ' ' J ? ? ?  @r9rNr'rr/ct|D]Z}tj|d}|j|D]}|j|[dS)zpFor internal use only: takes a given frame in ``np.ndarray`` format and writes it to the stream rgba)formatN)rangerB VideoFrame from_ndarray video_streamrHvideo_containerrI)rcrNrrav_framerOs r7rz&SceneFileWriter.encode_and_write_framesz"" 1 1A}11%1GGH+228<< 1 1$((0000 1 1 1r9r frame_or_rendererPixelArray | OpenGLRendererctrkt|tjr|}n0tjt jkr|n|}||f}|j |trtdst|tjrtj |}nBtjt jkr|ntj |}|jj|jjz }|jj}||||tddSdSdS)zUsed internally by Manim to write a frame to the FFMPEG input buffer. Parameters ---------- frame_or_renderer Pixel array of the frame. num_frames The number of times to write frame. rfzero_padN)rr.rndarrayrrRrOPENGL get_framerputrr fromarray get_imagerxparentrur output_image)rcrrrNmsgimage target_dir extensions r7 write_framezSceneFileWriter.write_frames^    +RZ88 ),*===&//111* u%C JNN3    ?? 6)#4 +RZ88 (9::,*===&//111):;; -4t7K7PPJ,3I   z"          r9r Image.Imagerextrc|rA||t|j||n |||j||xjdz c_dS)Nr )saver>r]zfill)rcrrrrs r7rzSceneFileWriter.output_images  ? JJ*Rc$*:&;&;&A&A(&K&KRSRR S S S S JJ*=d&6=== > > > Ar9ctdrdStdst|j|_||j||jdS)zThis method saves the image passed to it in the default image directory. Parameters ---------- image The pixel array of the image to save. rfNri)rrrxrprint_file_ready_message)rcrs r7 save_imagezSceneFileWriter.save_imagesi )   Fm$ V#?@T#U#UD  4'((( %%d&:;;;;;r9ctrk|tjr|tdr|nq|n\trNtdsA|jj |jj z }tj d|j t||jr|dSdS)zFinishes writing to the FFMPEG buffer or writing images to output directory. Combines the partial movie files into the whole scene. If save_last_frame is True, saves the last frame in the default image directory. flush_cacherfz %i images ready at %s N)rcombine_to_movierr{combine_to_section_videosflush_cache_directory clean_cacherrxrrurrr]r>r_write_subcaption_file)rcrs r7finishzSceneFileWriter.finish s    X  ! ! # # ## 1..000m$ #**,,,,  """" __ XVI%6 X-4t7K7PPJ K3T5Es: W W W   )  & & ( ( ( ( ( ) )r9cN||j|jj}||_t t j}d}d}ddd}t jdkrd}d|d <t jrd }nt jrd }d }tj |d }| |||}||_ t j |_t j|_||_||_t'|_t+|jd|_|jdS)zOpen a container holding a video stream. This is used internally by Manim initialize the container holding the video stream of a partial movie file. Nlibx264yuv420p123)ancrf.webmz libvpx-vp9z -auto-alt-refyuva420pqtrleargbrAmode)rateoptions)targetargs)r^rR num_playspartial_movie_file_pathr8rrro transparentrBrCrFpix_fmt pixel_widthwidthrheightrrrrr r writer_threadstart)rcrr)partial_movie_file_codecpartial_movie_file_pix_fmt av_optionsrstreams r7rz)SceneFileWriter.open_partial_movie_streams=  01HII'0$v011#, %."   &' 1 1'3 $*-J '! 8-7*   0'. $)/ &')#666 ++ $,   4) + 0?$*;@77 #4+@rJJJ   """""r9cf|jd|j|jD]}|j||jtj d|j j ddd|j didS)zClose the currently opened video container. Used internally by Manim to first flush the remaining packages in the video stream holding a partial file, and then close the corresponding container. )rNz Animation z) : Partial movie file written in %(path)spath'N)rrrjoinrrHrrIcloserrrRrr )rcrOs r7rz*SceneFileWriter.close_partial_movie_streamJs z""" !!!'..00 - -F  $ $V , , , , ""$$$ [0 [ [ [ 85888 9     r9hash_invocationct|drtsdS|j|tdz }|S)a9Will check if a file named with `hash_invocation` exists. Parameters ---------- hash_invocation The hash corresponding to an invocation to either `scene.play` or `scene.wait`. Returns ------- :class:`bool` Whether the file exists. r}Fro)rrr}rexists)rcrrs r7is_already_cachedz!SceneFileWriter.is_already_cached^s^t677 ~?O?O 5  ( B&)?"@BB C {{}}r9 input_files list[str]rir create_gifrc|jdz }tjdt|dd|ddi|dd5}|d |D]<}t |}|d |d = dddn #1swxYwYd d i}|sd|d<tjt||d} | j j d} tjt|d} dt| j d<|rs | d} d| _t jrd| _t j| _t j| _t-t j| _tj} | | }| d}| dd}| dd}| d }||||dd||d!d||dd!||| | d"D]}| || dd} | !}| j"j#| j"j#|_#||_$|d!z }| %| &|n#tj'j($rYnwxYw| &D]}| %|n|| )| } t jrt j*d$krd%| _| +| D]-}|j, d|_,| |_-| %|.| .| .dS)&Npartial_movie_file_list.txtz Partial movie files to combine (z files): %(p)srrAutf-8encodingz*# This file is used internally by FFMPEG. z file 'file:z' safe0rrconcat)rrrrzRendered with Manim Community vcommentgif)r=rgb8pal8templatesplit palettegenzstats_mode=diff paletteusez.dither=bayer:bayer_scale=5:diff_mode=rectangle buffersinkr )videoTrr)/r}rdebugrrCwriteras_posixrBr>rDr4rmetadatarFr rr r r rrr8rrfilterGraph add_bufferaddlink_to configurerGpushpull codec_context time_baseptsrIrHerrorEOFErroradd_stream_from_templaterodemuxdtsrr)rcrrir r file_listfppf_pathrpartial_movies_inputpartial_movies_streamoutput_container output_streamgraph input_bufferr0r1r2 output_sinkrNframes_writtenrOs r7 combine_fileszSceneFileWriter.combine_filesss03PP  Os;/?/? O O O +bqb/ "   ^^C'^ 2 2 5b HHB C C C& 5 5w--00223w3334444 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 5 C  #"Jt !w  NNJx   !5 < B1 E73{#3#3#>>> ;k ; ; !), D -  -77 8M%+M !! /(. %"("4M #)#6M !1&2C!D!DM IOO%%E ++5J+KKLIIg&&E<1BCCJNJ ))L11K   ' ' ' MM*a + + + MM*a + + +   z1a 0 0 0   { + + + OO   -4414== " " 5!!!! JJt   N !JJLLE$2<H*7*E*O .EI"a'N$(()=)=e)D)DEEEEx(E (..00 - - $$V,,,, --EE.FM! 3f&AW&L&L(2 %.445JKK - -:%! !.  $$V,,,,""$$$     s&AB44B8;B8=A%M##M:9M:cd|jD}|j}tr|j}t |dkrt jddSt jd|||t|j|jrtj dkr| d}| tjd|j|dd tjd kr)| d }t%||d |}n8tjdkr(| d}t%||d|}||jd|j}ddt,d}t/j|5}t/j|5} |jjd} | jjd} t/jt9|d|} | | } | | }|| D]&}|j | |_ | !|'| | D]&}|j ||_ | !|'| "dddn #1swxYwYdddn #1swxYwYtGj$t9|t9||%|&t9|tOr|D]}tQ|dSdS)zUsed internally by Manim to combine the separate partial movie files that make up a Scene into a single video file for that Scene. cg|]}||SNr).0els r7 z4SceneFileWriter.combine_to_movie..sWWWbrr9rz*No animations are contained in this scene.NzCombining to Movie file.r+rwav312k)rbitraterz.ogg libvorbisz.mp4z.aacaac_temprz'comment=Rendered with Manim Community v)shortestr8rA)rrr.))r^ryrr|rrrrTrrr with_suffixrrrrexportrorP with_namerurrrBrCrDr4rEr>rFrGrHrrIrshutilmoverrrr)rcr^rysound_file_pathogg_sound_file_pathaac_sound_file_pathtemp_file_pathr video_input audio_inputr audio_streamrNoutput_video_streamrMrOrs r7rz SceneFileWriter.combine_to_moviesa XWD,DWWW. ?? 1"0O " # #q ( ( KD E E E F .///    OO        G %6=E#9#9-99&AAO  " "<#6q#9#9 : : :   % % &   *g55&5&A&A&&I&I#o/BKPPP"5,66'6&A&A&&I&I#o/BEJJJ"5,66"'FFo.DFFN SkSSJ ((" ),7((" ),7*28; *28; #%7''c:$$$ '7&O&O)'P''#'7&O&O)'P''#*// ==11Fz) %8FM$((0000)// ==11Fz) %8FM$((0000 &&(((E" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )" )H KN++S-A-A B B B  " " $ $ $ %%c/&:&:;;;    (0 ( ( Y'''' ( ( ( (s7<K5DK K5K" "K5%K" &K55K9<K9c|g}|jD]}|jtjd|jd|||j|jz | | |j|j|j dz d5}tj||dddddS#1swxYwYdS)z1Concatenate partial movie files for each section.Nz%Combining partial files for section 'rz.jsonrA)indent)rr`r4rrrXrTget_clean_partial_movie_filesrzrget_dictrwrCjsondump)rcsections_indexsectionfiles r7rz)SceneFileWriter.combine_to_section_videosIsO   """/1} R RG}( SGLSSSTTT""99;;,w}<%%g&6&6t7O&P&PQQQ&D,<)C)C)CC I I# N N 6RV Ind1 5 5 5 5 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6 6s C11C58C5c~fdjD}t|tdkrxt|tdz }t |dd|}|D]}|t jdtdd|ddSdS) z@Will clean the cache by removing the oldest partial_movie_files.c0g|]}|dkj|z Sr"r}rX file_namercs r7rZz/SceneFileWriter.clean_cache..[s7! ! ! 999 )I 5999r9max_files_cachedc4|jSrW)statst_atime)rs r7z-SceneFileWriter.clean_cache..fs!5r9)keyNz'The partial movie directory is full (> z* files). Therefore, manim has removed the zV oldest file(s). You can change this behaviour by changing max_files_cached in config.)r}iterdirrrsortedrrr)rccached_partial_moviesnumber_files_to_deleteoldest_files_to_deletefile_to_deletes` r7rzSceneFileWriter.clean_cacheYs#! ! ! ! !9AACC! ! !  $ % %/A(B B B)**V4F-GG #&,%55&&&&%%&' "#9 ( (%%'''' KY&AS:TYYAWYYY      C Br9cfdjD}|D]}|tjdt |ddjidS)z)Delete all the cached partial movie filesc0g|]}|dkj|z Sr{r|r}s r7rZz9SceneFileWriter.flush_cache_directory..qs7! ! ! 999  (9 4999r9zCache flushed. z file(s) deleted in %(par_dir)s.par_dirN)r}rrrrr)rcrfs` r7rz%SceneFileWriter.flush_cache_directoryos! ! ! ! !9AACC! ! !  '  A HHJJJJ Zc"788 Z Z Z 4 5     r9ctjdSttjd}|t j|jdtj d|dS)zWrites the subcaption file.Nz.srtr$r%z$Subcaption file has been written as ) rrirrb write_textsrtcomposer_rr)rcsubcaption_files r7rz%SceneFileWriter.write_subcaption_file}ss   % Fv122>>vFF""3;t/?#@#@7"SSS L?LLMMMMMr9cP|td<tjddd|didS)z*Prints the "File Ready" message to STDOUT.riz File ready at %(file_path)s rrN)rrr)rcrs r7rz(SceneFileWriter.print_file_ready_messages6 )} 5 EUEUEUEU7VWWWWWr9)rRrSrTr>rUr r+r?)rTr>r+r?)r+r?)rXr>rYr>rZrr+r?)rrr+r?)r+r>)NN)rrrrrrr+r?) rr(rrrrrUr r+r?)FN)rrrrr+r?)F)rrr+r?)rNr'rr/r+r?)r )rrrr/r+r?) rrrr(rr>rr/r+r?)rrr+r?rW)rrr+r?)rr>r+r)FF) rrrirr rrrr+r?)rr(r+r?)!__name__ __module__ __qualname____doc__rvrdr[rrarrr\rrrrrrrrrrrrrrrTrrrrrrrr9r7rr]s6"'    &KKKKZ        2QQQQ8....@$$$$3333"+/ ) ) ) ) ) \"! )<)<)<)<)r+r?)Dr __future__r__all__rtrewarnings fractionsrpathlibrrrtempfilerr threadingr typingr r rBnumpyrrPILr catch_warningsfilterwarningsRuntimeWarningpydubrmanimrrhrr_config.logger_utilsr constantsrutils.file_opsrrrrrrr utils.soundsrrwr!r"av.container.outputr# av.streamr$manim.renderer.cairo_rendererr%manim.renderer.opengl_rendererr& manim.typingr'r(r8rPrrr9r7rs5..""""""   >>>>>>>>%%%%%%%%   X##H' #""""" ###############222222$$$$$$433333000000001333333      ;;;;;;======00000000    "%%%%"k Xk Xk Xk Xk Xk Xk Xk Xk Xk Xs!BB B