l j%ddlmZddlmZddlmZmZddlZddl m Z ddl m Z m Z ddlmZdd lmZmZdd lmZdd lmZdd lmZerdd lmZddlmZddlmZdgZGddZdS)) annotations)Iterable) TYPE_CHECKINGAnyN)get_hash_from_play_call)configlogger)Camera)Mobject_AnimationBuilder)SceneFileWriter)EndSceneEarlyException) list_update) Animation)Scene) PixelArray CairoRendererceZdZdZeddfd/d Zd0dZd1dZ d2d3dZ d4d5dZ d6d Z d7d8d%Z d9d'Z d0d(Z d:d,Zd;d-Zd0d.ZdS)r1is_already_cachedadd_partial_movie_filer!appendstrbegin_animationbegin_animationssave_static_frame_datastatic_mobjects!is_current_animation_frozen_frame update_framemoving_mobjectsfreeze_current_frame play_internal end_animation)r%r*r4rr:s r'playzCairoRenderer.play@s $= ##%%%$$d5f555   0 L?t~?? @ @ @%) " II 'III'( 0 /000)HT^)H)H)H&&'333)@K$N **& #556LMM0KmT^mmm13IJ,0D(II/II //0FGGG %%&<=== H #d,RaR011 2   ((T-A)ABBB     ##E5+@AAA  2 2 4 4 "   ee.C  D D D  % %en 5 5 5 5    ! ! ! &&4+?'?@@@ !r)Tr>Iterable[Mobject] | Noneinclude_submobjectsignore_skippingc |jr|sdS|st|j|j}|j |j|jn|j||d<|jj|fi|dS)zUpdate the frame. Parameters ---------- scene mobjects list of mobjects include_submobjects ignore_skipping **kwargs NrV) rrr>foreground_mobjectsr$rset_frame_to_backgroundresetcapture_mobjects)r%r*r>rVrWrs r'rOzCairoRenderer.update_frame}s.     F ")H   ( K / /0A B B B B K     (;$%$ $X8888888r)r#floatrPc|||||dSr.)rO add_frame get_frame)r%r*r#rPs r'renderzCairoRenderer.renders< %111 t~~''(((((r)rc>tj|jjS)zGets the current frame as NumPy array. Returns ------- PixelArray NumPy array of pixel values of each pixel in screen. The shape of the array is height x width x 3. )nparrayr pixel_arrayr%s r'r`zCairoRenderer.get_framesx /000r)r?frame num_framesintcd|jjz }|jrdS|xj||zz c_|j||dS)zAdds a frame to the video_file_stream Parameters ---------- frame The frame to add, as a pixel array. num_frames The number of times to add frame. r?Nrh)r frame_raterr#r1 write_frame)r%rgrhdts r'r_zCairoRenderer.add_framesY' '    F Z"_$  $$Uz$BBBBBr)rCcd|jjz }||t ||z dS)zAdds a static frame to the movie for a given duration. The static frame is the current frame. Parameters ---------- duration [description] r?rkN)rrlr_r`ri)r%rCrns r'rQz"CairoRenderer.freeze_current_framesT' '  NN  8b=))      r)c||d|jdS)zTOpens the current frame in the Default Image Viewer of your system. T)rWN)rOr get_imageshowr2s r' show_framezCairoRenderer.show_framesC %666 $$&&&&&r)rMIterable[Mobject]PixelArray | Nonecd|_|sdS|||||_|jS)aCompute and save the static frame, that will be reused at each frame to avoid unnecessarily computing static mobjects. Parameters ---------- scene The scene played. static_mobjects Static mobjects of the scene. If None, self.static_image is set to None. Returns ------- PixelArray | None The static image computed. The return value is None if there are no static mobjects in the scene. Nr=)r$rOr`)r%r*rMs r'rLz$CairoRenderer.save_static_frame_datasM(! 4 %/::: NN,,  r)c:|jjdjrd|_tdrd|_tjdkr|jtjkrd|_tjdkr*|jtjkrd|_tdSdS)a?This method is used internally to check if the current animation needs to be skipped or not. It also checks if the number of animations that were played correspond to the number of animations that need to be played, and raises an EndSceneEarlyException if they don't correspond. Tsave_last_framerN)r1sectionsrr from_animation_numberr"upto_animation_numberrrfs r'r@z$CairoRenderer.update_skipping_statuss   $R ( 8 (#'D # $ (#'D  (1 , ,!===#'D  (A - -!===#'D (** * . -==r)c|jr|jnAtjrdt_dt_nd|_||tdrOd|_|||j|j dSdS)NTFry) r"r1finishr write_to_movieryr$rO save_imagerrqr2s r'scene_finishedzCairoRenderer.scene_finisheds > %   # # % % % %  " %%)F "$)F ! ! $D    e $ $ $ # $ A $D    e $ $ $   ' ' (=(=(?(? @ @ @ @ @ A Ar))rrrrrrrr)r*rr+r,)r*rr4r5rrr+r,)NTT) r*rr>rUrVrrWrrrr+r,r.)r*rr#r]rPrUr+r,)r+r)r?)rgrrhrir+r,)rCr]r+r,)r*rrMrtr+ru)r+r,)r0 __module__ __qualname____doc__rr(r3rTrOrar`r_rQrsrLr@rr)r'rrsK  4C,0 % 44444(    ;;;;@.2$( $ $9$9$9$9$9T59 ))))) 1 1 1 1CCCCC     ''''!!!!6++++0AAAAAAr)) __future__rcollections.abcrtypingrrnumpyrcmanim.utils.hashingrr r camera.camerar mobject.mobjectr r scene.scene_file_writerrutils.exceptionsrutils.iterablesrmanim.animation.animationrmanim.scene.scenerr__all__rrr)r'rsc""""""$$$$$$%%%%%%%%777777""""""88888888555555555555))))))$333333''''''######  CACACACACACACACACACAr)