l j ddlmZddlZddlZddlZddlZddlmZddlm Z m Z m Z ddl Z ddl Zddl mZddlmZddlmZddlmZmZdd lmZmZdd lmZdd lmZmZdd lm Z dd l!m"Z"ddl#m$Z$ddl%m&Z&ddl'Tddl(m)Z)ddl*m+Z+ddl,m-Z-ddl.m/Z/m0Z0m1Z1m2Z2m3Z3ddl4m5Z5m6Z6ddl7m8Z8m9Z9e r@ddl:m;Z;ddlm Z ddlm?Z?m@Z@ddlAmBZBddlmCZCmDZDmEZEmFZFmGZGddlHmIZIdd lJmKZKdd!lLmMZMd"d#gZNGd$d"eZOGd%d#ZPdS)&) annotationsN)cached_property) TYPE_CHECKINGAnySelf) Framebuffer)Image)override)configlogger) OpenGLMobject OpenGLPoint)OpenGLVMobject)MatrixMNPoint3D)handle_caching_play) color_to_rgba)EndSceneEarlyException) straight_path)*)SceneFileWriter)opengl)clip)angle_of_vectorquaternion_from_angle_axisquaternion_multrotation_matrix_transpose)rotation_matrix_transpose_from_quaternion)MeshShader)%render_opengl_vectorized_mobject_fill'render_opengl_vectorized_mobject_stroke)Iterable)r) Animation)Mobject_AnimationBuilder)Scene) FloatRGBA PathFuncType Point3DLikeRGBAPixelArray Vector3DLike)ParsableManimColor)FlattenedMatrix4x4Window OpenGLCameraOpenGLRendererc TeZdZdZdddddde dz edz df dEfd ZdFdZdGdZedHdZ edId!Z dJd"Z dKd#Z dJd$Z eedfdLd)Z dMdNd.ZdOd/ZdPd0ZdQd1ZdRd3ZdSd5ZdTd7ZdUd9ZdFd:ZdVd;ZdVd<ZdVd=ZeefdWfdD ZxZS)Xr3a An OpenGL-based camera for 3D scene rendering. Attributes ---------- frame_shape : tuple[float, float] The width and height of the camera frame. center_point : np.ndarray The center point of the camera in 3D space. euler_angles : np.ndarray The Euler angles (theta, phi, gamma) representing the camera's orientation. focal_distance : float The focal distance of the camera. light_source_position : np.ndarray The position of the light source in 3D space. orthographic : bool Whether the camera uses orthographic projection instead of perspective. minimum_polar_angle : float The minimum polar angle for camera rotation. maximum_polar_angle : float The maximum polar angle for camera rotation. inverse_rotation_matrix : np.ndarray The inverse rotation matrix of the camera. Ng@Fr frame_shapetuple[float, float] | None center_pointPoint3DLike | None euler_anglesfocal_distancefloatlight_source_position orthographicboolminimum_polar_anglemaximum_polar_angle model_matrixMatrixMN | NonekwargsrreturnNonec  Hd|_d|_||_||_||_|jr3t j|_t jd|_n2t j |_t j d|_| tdtdf|_ n||_ | t|_ n tj|t |_ | t jd d d } ||_tj|pgd t |_t)|j|_| |_t/jd| dd | tj|pgd t }||_|dS)u` Initializes an OpenGLCamera instance. Parameters ---------- frame_shape : tuple[float, float], optional The width and height of the camera frame. If not provided, defaults to the global manim config values `frame_width` and `frame_height`. center_point : Point3DLike, optional The center point of the camera in 3D space. If not provided, defaults to the origin (0, 0, 0). euler_angles : Point3DLike, optional The Euler angles (theta, phi, gamma) representing the camera's orientation. If not provided, defaults to (0, 0, 0) (i.e., no rotation). focal_distance : float, optional The focal distance of the camera. Default is 2.0. light_source_position : Point3DLike, optional The position of the light source in 3D space. If not provided, defaults to (-10, 10, 10). orthographic : bool, optional Whether the camera uses orthographic projection instead of perspective. Default is False (perspective). minimum_polar_angle : float, optional The minimum polar angle in radian for camera rotation. Default is -π/2, i.e. no restriction. maximum_polar_angle : float, optional The maximum polar angle in radian for camera rotation. Default is π/2, i.e. no restriction. model_matrix : MatrixMN, optional The initial model matrix [1]_ for the camera. If not provided, defaults to a translation matrix that positions the camera at (0, 0, 11). **kwargs : Any Additional keyword arguments passed to the OpenGLMobject constructor. References ---------- .. [1] Wikipedia, "Camera matrix", https://en.wikipedia.org/wiki/Camera_matrix T<F)format_N frame_width frame_heightdtyper )i rO)rB should_render)rrr) use_z_index frame_rater>r@rArorthographic_projection_matrixprojection_matrixunformatted_projection_matrixperspective_projection_matrixr r6ORIGINr8npasarrayr<translation_matrixr;r=r light_sourcedefault_model_matrixsuper__init__r:refresh_rotation_matrix) selfr6r8r:r;r=r>r@rArBrD __class__s Y/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/renderer/opengl_renderer.pyr_zOpenGLCamera.__init___sj (#6 #6   %+%J%L%LD "171V222D . .&,%I%K%KD "171U222D .   &} 5vn7MND  *D    &D   " >L,%'Z ! 2]]]%& & & "((BCC$0!Rl%RR6RRRz,";)))5III %1 $$&&&&&rc4|jdddfddS)z+Retrieve the camera's position in 3D space.NrBras rc get_positionzOpenGLCamera.get_positions! A&rr**rdpositionrc2||jdddfdd<|S)z&Set the camera's position in 3D space.Nrfrg)rarjs rc set_positionzOpenGLCamera.set_positions%&.!!!Q$# rdr0c4tj|jS)z+The formatted view matrix for shader input.)rmatrix_to_shader_inputunformatted_view_matrixrhs rcformatted_view_matrixz"OpenGLCamera.formatted_view_matrixs,T-IJJJrdrcztjttj|jSN)typingcastrrYlinalginvrBrhs rcroz$OpenGLCamera.unformatted_view_matrixs%{8RY]]43D%E%EFFFrdc(|ttttt g||jdd||jdd| |j dS)zEInitialize the camera's points based on frame shape and center point.rT)stretchr N) set_pointsrXLEFTRIGHTDOWNUP set_widthr6 set_heightmove_tor8rhs rc init_pointszOpenGLCamera.init_pointssv udB7888 t'*D999 (+T::: T&'''''rdc||td|td|ddd|j|_|S)zlReset the camera to its default state (config frame size, centered at origin, no rotation). rKrJr)centerrr r~set_euler_anglesr]rBrhs rcto_default_statezOpenGLCamera.to_default_statesg  ~./// vm,--- aA&&& 5 rdc |j\}}}tt|tdt|tdt|td}t t j|t|_ dS)zGRefresh the camera's inverse rotation matrix based on its Euler angles.T)axis_normalizedrLN) r:rrOUTr{rrYrZr<inverse_rotation_matrix)rathetaphigammaquats rcr`z$OpenGLCamera.refresh_rotation_matrixs!-sE &uc4 H H H &sE4 H H H &uc4 H H H   (Q Jt5 ) ) )( ( $$$rdangleaxisr. about_pointc |j}t||}tj||}|d}tj|d} t |ddt dz z} tjt| tt| t} t tj| |j dddf} | | | | |S)a Rotate the camera by a given angle around a specified axis. Parameters ---------- angle : float The angle in radians to rotate the camera. axis : Vector3DLike, optional The axis around which to rotate the camera. Default is OUT (z-axis). about_point : Point3DLike, optional Ignored. For OpenGLCamera, rotation is always about the camera's center. **kwargs : Any Not used for OpenGLCamera. Passing additional keyword arguments has no effect. Returns ------- Self The rotated camera instance. Returned for chaining. rNr) rrrYdotarccosrPIr{rTr) rarrrrD curr_rot_T added_rot_T new_rot_TFzrr partial_rot_Trs rcrotatezOpenGLCamera.rotates:1 /t<< F:{33 q\i12A2''"q&0 %c5 1 1 %eS 1 1   }ik B B111a4 HII eS%000 rdr float | Nonerrcv| ||jd<| ||jd<| ||jd<||S)a Set the camera's Euler angles [1]_ (theta, phi, gamma). Parameters ---------- theta : float | None, optional The angle in radians for rotation around the OUT (z) axis. If None, the current theta value is retained. phi : float | None, optional The angle in radians for rotation around the RIGHT (x) axis. If None, the current phi value is retained. gamma : float | None, optional The angle in radians for rotation around the OUT (z) axis. If None, the current gamma value is retained. Returns ------- Self The camera instance with updated Euler angles. Returned for chaining. See Also -------- set_theta : Set the theta Euler angle. set_phi : Set the phi Euler angle. set_gamma : Set the gamma Euler angle. References ---------- .. [1] Wikipedia, "Euler angles", https://en.wikipedia.org/wiki/Euler_angles Nrr rr:r`)rarrrs rcrzOpenGLCamera.set_euler_anglessTJ  #(D a ?#&D a  #(D a  $$&&& rdc.||S)z Set the camera's theta Euler angle (in radians). See Also -------- set_euler_angles : Set all Euler angles at once. set_phi : Set the phi Euler angle. set_gamma : Set the gamma Euler angle. )rr)rars rc set_thetazOpenGLCamera.set_thetaK$$5$111rdc.||S)z Set the camera's phi Euler angle (in radians). See Also -------- set_euler_angles : Set all Euler angles at once. set_theta : Set the theta Euler angle. set_gamma : Set the gamma Euler angle. )rr)rars rcset_phizOpenGLCamera.set_phiWs$$$---rdc.||S)z Set the camera's gamma Euler angle (in radians). See Also -------- set_euler_angles : Set all Euler angles at once. set_theta : Set the theta Euler angle. set_phi : Set the phi Euler angle. )rr)rars rc set_gammazOpenGLCamera.set_gammacrrddthetacX|jdxx|z cc<||S)z Increment the camera's theta Euler angle by a given amount (in radians). See Also -------- set_euler_angles : Set all Euler angles at once. set_theta : Set the theta Euler angle. rr)rars rcincrement_thetazOpenGLCamera.increment_thetao: !& $$&&& rddphic|jd}t||zt dz tdz }||jd<||S)z Increment the camera's phi Euler angle by a given amount (in radians). See Also -------- set_euler_angles : Set all Euler angles at once. set_phi : Set the phi Euler angle. r r)r:rrr`)rarrnew_phis rc increment_phizOpenGLCamera.increment_phi|sT"sTzB37BF33&! $$&&& rddgammacX|jdxx|z cc<||S)z Increment the camera's gamma Euler angle by a given amount (in radians). See Also -------- set_euler_angles : Set all Euler angles at once. set_gamma : Set the gamma Euler angle. rr)rars rcincrement_gammazOpenGLCamera.increment_gammarrdtuple[float, float]cR||fS)z2Retrieve the width and height of the camera frame.) get_width get_heightrhs rc get_shapezOpenGLCamera.get_shapes!  $//"3"344rdcLtjt|jdS)z Retrieve the center point of the camera in 3D space. Notes ----- The center point is assumed to be the first point in the camera's points array. r)rsrtrpointsrhs rc get_centerzOpenGLCamera.get_centers{7DKN333rdcP|j}|d|dz }t|S)z'Retrieve the width of the camera frame.)rr)r rrr<rarouts rcrzOpenGLCamera.get_width'TlVD\)SzzrdcP|j}|d|dz }t|S)z(Retrieve the height of the camera frame.)r )rfr rrs rcrzOpenGLCamera.get_heightrrdc:|j|zS)z*Retrieve the focal distance of the camera.)r;rrhs rcget_focal_distancezOpenGLCamera.get_focal_distances"T__%6%666rdmobject1r mobject2alpha path_funcr+cxt||||||Srr)r^ interpolater`)rarrrrrbs rcrzOpenGLCamera.interpolates: HhyAAA $$&&& rd)r6r7r8r9r:r9r;r<r=r9r>r?r@r<rAr<rBrCrDrrErF)rEr)rjrrEr)rEr0)rErrErF)rEr) rr<rr.rr9rDrrEr)NNN)rrrrrrrEr)rr<rEr)rr<rEr)rr<rEr)rr<rEr)rr<rEr)rr<rEr)rEr)rEr<) rr rr rr<rr+rEr) __name__ __module__ __qualname____doc__rr_rirlrrprorrr`r rrrrrrrrrrrrrrrr __classcell__)rbs@rcr3r3Ds837+/+/ #48"&(S1W%'!V(,`'`'`'`'`'`'`'D++++ KKK_KGGG_G((((         !*. ((((X(X# " ,,,,,\ 2 2 2 2 . . . . 2 2 2 2            5555 4 4 4 4 7777 #0-//      X     rdceZdZdZedfdHd ZdId ZdJd ZdKdZdLdZ dMdZ dNdZ dOdZ dPdZ edQd!ZdPd"ZdRd'ZdId(ZdId)ZdJd*ZdSd,ZdTd/Z dUdVd5ZdWdXd9ZdYd;Z dZd[dAZed\dCZejd]dFZdGS)^r4a An OpenGL-based renderer. Attributes ---------- animation_elapsed_time : float The elapsed time of the current animation. animation_start_time : float The start time of the current animation. animations_hashes : list[str | None] List of animation hashes for caching. anti_alias_width : float The width used for anti-aliasing in pixel units. background_color : FloatRGBA The background color of the renderer. camera : OpenGLCamera The camera used for rendering. num_plays : float The number of animation plays executed. path_to_texture_id : dict[str, int] Mapping from texture file paths to OpenGL texture IDs. pressed_keys : set[int] Set of currently pressed key codes. skip_animations : bool Whether animations are currently being skipped. time : float The total elapsed time for the renderer. window : Window | None The window used for previewing, if any. Ffile_writer_classtype[SceneFileWriter]skip_animationsr?rErFcd|_||_||_||_d|_d|_d|_g|_d|_t|_ t|_ d|_ i|_td|_dS)aQInitializes the OpenGLRenderer. Parameters ---------- file_writer_class : type[SceneFileWriter], optional The class to use for writing scene files, by default SceneFileWriter. skip_animations : bool, optional Whether to skip animations during rendering, by default False. g?rNbackground_color)anti_alias_width_file_writer_class_original_skipping_statusranimation_start_timeanimation_elapsed_timetimeanimations_hashes num_playsr3cameraset pressed_keyswindowpath_to_texture_idr r)rarrs rcr_zOpenGLRenderer.__init__s!$"3)8&.$'!&)# 35"nn &)ee%) 24 &'9 :rdscener)c*g|_|||jj|_||_t d|_|rFddl m }|||_ |j j |_ |j |_n t!jd|_ n+#t$$rt!jdd|_ YnwxYw||j d|_|j|j t jt d |j _t jt jt jt jf|j _d S) aW Initializes the OpenGL rendering context and related resources for the given scene. Set up: - the file writer - the background color - the OpenGL context - the window (if needed) Parameters ---------- scene : Scene The scene to be rendered rr r1T) standaloneegl)rbackendrenable_wireframeN)partial_movie_filesrrbr file_writerrr rshould_create_windowopengl_renderer_windowr2rctxcontextdetect_framebufferframe_buffer_objectmodernglcreate_context Exceptionget_frame_buffer_objectuseenableBLEND wireframe SRC_ALPHAONE_MINUS_SRC_ALPHAONE blend_func)rarr2s rc init_scenezOpenGLRenderer.init_scenes{ 68 ,0,C,C  O $- -  &'9 :  $ $ & & + 6 6 6 6 6 6 &,,DK;?DL'+|'F'F'H'HD $ $ '6$GGG    '6#!      (,'C'CDLRS'T'TD $  $ ( ( * * * HN+++!'(:!;     ( L L #  s!B<<%C$#C$ctdrtjddStdo7td o)td otd o td S) z Determine whether a window should be created for rendering based on the current configuration. Notes ----- A windows is always created if the 'force_window' configuration is enabled. force_windowz}'--force_window' is enabled, this is intended for debugging purposes and may impact performance if used when outputting filesTpreviewsave_last_frameformatwrite_to_moviedry_run)r r warningrhs rcrz#OpenGLRenderer.should_create_window7s . !  NK   4 9  &,-- &8$$ &+,, &9%%  rdtuple[int, int] | NonecLt|dd}|dS|j\}}}}||fS)a  Retrieve the pixel dimensions of the current frame buffer object (2D). Returns ------- width : int The width of the frame buffer in pixels. height : int The height of the frame buffer in pixels. rN)getattrviewport)ra frame_buffer_ pixel_width pixel_heights rcget_pixel_shapezOpenGLRenderer.get_pixel_shapeNs>,349NPT+U+U  4*6*?'1k<L((rdrr3c8|}|d}t||\}}|\}}|j||z z }|j} |j} tj| | } ||t| ttj | j t| |d|_dS)a Update the perspective-related uniform variables used in the OpenGL renderer based on the current camera settings. Parameters ---------- camera : OpenGLCamera The camera object from which to extract perspective and lighting information. Raises ------ ValueError If the renderer's pixel shape is not available. Nz9Pixel shape is None, cannot refresh perspective uniforms.)r6r camera_centercamera_rotationr=r;)r ValueErrorrrrr\ get_locationrYrtuplerarrayrflattenrperspective_uniforms) rar pixel_shapemsgr r rJrKrrotation light_poss rcrefresh_perspective_uniformsz+OpenGLRenderer.refresh_perspective_uniforms_s **,,  MCS// !$/! \$*$4$4$6$6! \ 0L<4OP1#0==?? F8Y// "++-- 0"6#4#4#6#677$RXh%7%7%9%A%A%C%CDD%*9%5%5$7799 % % !!!rdmobjectOpenGLMobject | OpenGLVMobjectct|tr:tdrt||tdrt |||}|D]}}|j}t|j|t|nd}|j D]\}}| t|}|j |} t| tjtjfs!dt#| } t%| || _t)j|j |j D]L\}} t1jt45||| dddn #1swxYwYM t9jd|jj} |d| j |d| j!n#t4$rYnwxYw|j"r%|j#tj$n$|j%tj$|j&} tO||j(| tSj*| nd|j"|j+ }|,||-dS) a Render an OpenGL mobject (either OpenGLMobject or OpenGLVMobject) using the appropriate shaders and rendering pipeline. Parameters ---------- mobject : OpenGLMobject | OpenGLVMobject The mobject to render. Must be an instance of OpenGLMobject or OpenGLVMobject. Raises ------ TypeError If a shader texture is not a moderngl.Uniform or moderngl.UniformBlock. use_projection_fill_shadersuse_projection_stroke_shadersN)rnamez&Shader texture must be a uniform, got r3 u_view_matrixu_projection_matrix)indicesuse_depth_test primitive). isinstancerr r#r$get_shader_wrapper_list shader_folderr"rstr texture_pathsitemsget_texture_idshader_programrUniform UniformBlocktype TypeErrorvalueitchainuniformsr contextlibsuppressKeyError set_uniformrsrtrrrprU depth_testr DEPTH_TESTdisable vert_indicesr! vert_datarYrZrender_primitive set_uniformsrender)rarshader_wrapper_listshader_wrapperfoldershaderr"pathtidshader_texturerr4camr?meshs rcrender_mobjectzOpenGLRenderer.render_mobjects= g~ . . G34 E5dGDDD56 G7gFFF%==??13 3 N#1F &:L3v;;;RVF -:@@BB + + d))#d))44!'!6t!<!"X%5x7L$M)XnAUAUWW$C..('*$$ "x'--//)//11   4 4 e (2244&&tU333444444444444444 k.$*2CDD""?C4MNNN""#8#:OPPPP    ( : ##H$78888 $$X%8999*6L(4@4L <000RV-8!2 D   d # # # KKMMMMg3 3 s%F77F; >F; AH H'&H'rHr+intcX||jvr |j|n||S)a Retrieves the OpenGL texture ID associated with the given image file path. Automatically creates a new texture it it has not been loaded before. Parameters ---------- path : str The file path to the texture image. Returns ------- int The OpenGL texture ID corresponding to the given path. )r_create_texture)rarHs rcr.zOpenGLRenderer.get_texture_ids8$t...  #D ) )%%d++ rd image_pathctj|5}t|j}|jdkrd}d}n|d}d}d}|j|j|| }dddn #1swxYwYd|_ d|_ tj tj f|_||_|| ||j|<|S) a Create an OpenGL texture from the given image file path, get its texture ID, and store it in `self.path_to_texture_id[image_path]`. Parameters ---------- image_path : str The file path to the image to be loaded as a texture. Returns ------- int The texture ID assigned to the newly created texture. Lr RRR1RGBAr)size componentsdataNF)location)r openlenrmodeconvertrtexturerVtobytesrepeat_xrepeat_yrNEARESTfilterswizzler)rarQimgrIrWrdr^s rcrPzOpenGLRenderer._create_textures,Z # # sd-..Cx3  kk&))  l**X%[[]]+G               $! "*H,<=! S !!!.1 + sA2BBBc|jjdjrd|_tjdkr|jtjkrd|_tjdkr*|jtjkrd|_tdSdS)a^ Check and update the skipping status for the current animation (self.skip_animations flag) based on the configuration settings. Parameters ---------- None Raises ------ EndSceneEarlyException If the number of played animations exceeds the configured upper bound. TrN)rsectionsrr from_animation_numberrupto_animation_numberrrhs rcupdate_skipping_statusz%OpenGLRenderer.update_skipping_statuss   $R ( 8 (#'D  (1 , ,!===#'D  (A - -!===#'D (** * . -==rd animations'Animation | Mobject | _AnimationBuilderrDrc,tj|_|j|j |j|i|||r|||js;|j |ttj |j z|jb|jtj|jz |j kr% tj|jz |j k%|j |_n||j|j |xj|j z c_|xjdz c_dS)aj Plays the given animations or mobjects in the specified scene. "Playing" here refers to the process of compiling animation data, beginning the animations, updating frames, and finalizing the animation in the context of the renderer. Parameters ---------- scene Scene The scene in which to play the animations. *animations Animation | Mobject | _AnimationBuilder The animations, mobjects, or animation builders to play. **kwargs Any Additional keyword arguments to pass to the animation compilation. ) num_framesNr )rrrbegin_animationrcompile_animation_databegin_animations!is_current_animation_frozen_frame update_frame write_framerNr rSdurationr swap_buffersr play_internal end_animationr)rarrlrDs rcplayzOpenGLRenderer.play/s0%)IKK! ((T-A)ABBB$$j;F;;;     2 2 4 4 "   e $ $ $'  ,,S):U^)K%L%L-{& ((***ikkD$==NNikkD$==NN*/.D ' '    ! ! ! &&4+?'?@@@ U^#  !rdcr|jj|j|jdS|jdS)z Clears the current frame buffer and updates the display window accordingly. The screen is cleared using the background color specified in the renderer. N)rclearrrrwrhs rc clear_screenzOpenGLRenderer.clear_screenasA ' &(=>> ;  F   """""rd frame_offsetr<moving_mobjects list[Mobject]cF|||jrdS|j||j_|j|j|kr=|||j|j|k9dSdSdS)a Renders a single frame of the given scene using OpenGL. Parameters ---------- scene : Scene The scene to render. frame_offset : float The time offset for the current frame in seconds. If no window is present, this parameter is ignored, and a frame is a true snapshot of the scene at the current time. moving_mobjects : list[Mobject] List of mobjects that are currently moving and need to be updated. Not used at all, kept for compatibility with other renderers. Notes ----- - Updates the frame for the scene. - If animations are skipped, the method returns early. - Writes the current frame using the file writer. - If a window is present, swaps buffers and continues updating frames until the animation elapsed time reaches the frame offset. N)rtrrrurrwr)rarr~rs rcrCzOpenGLRenderer.renderns4 %      F $$T*** ; " K $ $ & & &- <<!!%((( ((***- <<<< # "<>k.%,77 ))#...~ & &G( ;?IID    % % % %<  C((  !!$'''  '+ikkD4M&M###rdc\|jdkr|jn#|jdkrtjr dt_|rOdt_|||j| dSdS)aJ Handle the finalization process after a scene has finished rendering. Performs the following actions: - If any plays (animations) have occurred, finalizes the file writing process. - If no plays have occurred but movie writing is enabled, disables movie writing to avoid creating an empty movie file. - If the configuration requires saving the last frame, updates and saves the final image of the scene. Parameters ---------- scene : Scene The scene that has finished rendering. rFTN) rrfinishr rshould_save_last_framerrt save_image get_image)rars rcscene_finishedzOpenGLRenderer.scene_finisheds$ >A     # # % % % % ^q V%: $)F !  & & ( ( :%)F "   e $ $ $   ' '(8(8 9 9 9 9 9 : :rdcRtdrdS|jjrdS|jdkS)a2 Determine whether the last frame of the scene should be saved, i.e. if one of the following conditions is met: - The configuration option 'save_last_frame' is enabled. - The scene is not in interactive mode. - This is the first play (i.e., num_plays == 0). rTFr)r rinteractive_moderrhs rcrz%OpenGLRenderer.should_save_last_frames6 # $ 4 : & 5~""rd Image.Imagec |}|}|d}t|tjd||dddd}|S)a Get the current OpenGL frame buffer as a PIL Image. Returns ------- Image.Image The image representation of the current frame buffer. Raises ------ ValueError If the pixel shape cannot be determined. Notes ----- The image is constructed from raw RGBA buffer data, with the origin at the bottom-left. Nz&Pixel shape is None, cannot get image.rUrawrrg) get_raw_frame_buffer_object_datarrr frombytes)raraw_buffer_datarrimages rcrzOpenGLRenderer.get_imagesj&??AA**,,  :CS// !          rdstatic_mobjectsIterable[Mobject]cdSrrrQ)rarrs rcsave_static_frame_dataz%OpenGLRenderer.save_static_frame_data s  rdrrmoderngl.Contextsamplesrctd}td}d}||||f|||||f|S)a Creates and returns a framebuffer object configured with color and depth attachments. Parameters ---------- context : moderngl.Context The ModernGL context used to create the framebuffer and its attachments. samples : int, optional The number of samples for multisample anti-aliasing (MSAA)[1]_. Default is 0 (no MSAA). Returns ------- Framebuffer A framebuffer object with a color texture attachment and a depth renderbuffer attachment, both sized according to the current configuration's pixel width and height. Notes ----- Framebuffer's color attachment is supposed RGBA. Pixel dimensions are taken from the global config of Manim. References ---------- .. [1] Wikipedia, "Multisample anti-aliasing", https://en.wikipedia.org/wiki/Multisample_anti-aliasing r r r)rWr)r)color_attachmentsdepth_attachment)r framebufferr^depth_renderbuffer)rarrr r  num_channelss rcrz&OpenGLRenderer.get_frame_buffer_objectsB]+ n-  ""%ool+'. %77l+8 #   rdf1rMbytescXd}|j|jj||}|S)a Get the raw data from the current frame buffer object as bytes. This method reads the pixel data from the frame buffer object using the specified data type. The data is read with 4 color channels (typically RGBA). Args: dtype (str, optional): The data type to use when reading the buffer. Defaults to "f1" (i.e., float with 1 byte). Returns: bytes: The raw pixel data from the frame buffer object. r)r rWrM)rreadr )rarMrrets rcrz/OpenGLRenderer.get_raw_frame_buffer_object_data@s<* -22-6#3    rdr-c|d}|}|d}t||d|ddf}tj|d|}tj|}|S) a} Get the current frame buffer as a Numpy array of RGBA pixel values. Returns ------- RGBAPixelArray A Numpy array of shape (height, width, 4) containing the RGBA pixel data of the current frame, with dtype uint8. Raises ------ ValueError If the pixel shape cannot be determined. rrLNz&Pixel shape is None, cannot get frame.r rruint8)rrrrY frombufferreshapeflipud)rarrrresult_dimensionsnp_bufs rc get_framezOpenGLRenderer.get_frame]s 33$3??**,,  :CS// !(^[^Q?s'222::;LMM6"" rdpxpyrelativetop_leftrc |}|.tjtt jgdS|\}}t d}|j} |r dt j||z ||z dgzS||z } |rdnd} tjt| | t j||dz z | ||dz z zdgzzS) a Converts pixel coordinates to space (scene) coordinates. top_left flag should be set to True when using a GUI framework where the (0,0) is at the top left: e.g. PySide6. Parameters ---------- px : float The x-coordinate in pixel space. py : float The y-coordinate in pixel space. relative : bool, optional If True, returns coordinates relative to the frame (normalized to [-1, 1]). If False, returns absolute space coordinates. Default is False. top_left : bool, optional If True, treats the origin (0, 0) as the top-left corner of the pixel space. If False, treats the origin as the bottom-left. Default is False. Returns ------- Point3D The corresponding coordinates in space as a NumPy array of shape (3,). Notes ----- If the pixel shape is not available, returns the origin [0, 0, 0]. N)rrrrKrrrgr r) rrsrtrrYrr rr) rarrrrrr r rK frame_centerscale y_directions rcpixel_coords_to_space_coordsz+OpenGLRenderer.pixel_coords_to_space_coords{s>**,,  ;w(A(ABB B$/! \n- {--//  Jrxk!12 3Da HIII I|+$+bb! {  h{Q&\A=M8M)NPST    rdr*c|jS)z3The background color of the renderer (RGBA format).)_background_colorrhs rcrzOpenGLRenderer.background_colors %%rdr4r/c0t|d|_dS)Ng?)rr)rar4s rcrzOpenGLRenderer.background_colors!.uc!:!:rdN)rrrr?rErF)rr)rErF)rEr?)rEr)rr3rErF)rrrErF)rHr+rErN)rQr+rErNr)rr)rlrmrDrrErF)rr)r~r<rrrErF)rEr)rr)rrrErF)r)rrrrNrEr)r)rMr+rEr)rEr-)FF) rr<rr<rr?rr?rEr)rEr*)r4r/rErF)rrrrrr_rrrrrMr.rPrkrrzr}rCrtrrrrrrrrpropertyrsetterrQrdrcr4r4sPB4C %;;;;;@0 0 0 0 d    .))))"% % % % NKKKKZ    ,''''R++++:////b # # # #%+%+%+%+N%N%N%N%NN::::8 # # # #####J    9:. . . . . `:>NS3 3 3 3 3 j&&&X&;;;;;;rd)Q __future__rr8 itertoolsr5rrs functoolsrrrrrnumpyrYrPILr typing_extensionsr manimr r #manim.mobject.opengl.opengl_mobjectr r.manim.mobject.opengl.opengl_vectorized_mobjectr manim.typingrrmanim.utils.cachingrmanim.utils.colorrmanim.utils.exceptionsrmanim.utils.pathsr constantsscene.scene_file_writerrutilsrutils.simple_functionsrutils.space_opsrrrrrrGr!r"vectorized_mobject_renderingr#r$collections.abcr%manim.animation.animationr&manim.mobject.mobjectr'r(manim.scene.scener)r*r+r,r-r.manim.utils.color.corer/manim.utils.openglr0rr2__all__r3r4rQrdrcrs"""""" %%%%%%++++++++++ &&&&&& JIIIII********333333++++++999999++++++555555))))))!  /((((((333333@@@@@@@@'''''':99999555555...... + ,~~~~~=~~~B r ;r ;r ;r ;r ;r ;r ;r ;r ;r ;rd