l j#dZddlmZdgZddlmZddlmZddlm Z ddl m Z m Z m Z dd lmZdd lmZdd lmZmZmZmZdd lmZdd lmZddlmZmZGddeZdS)zDefines the MovingCamera class, a camera that can pan and zoom through a scene. .. SEEALSO:: :mod:`.moving_camera_scene` ) annotations MovingCamera)Iterable)Any)Context) PixelArrayPoint3D Point3DLike)config)Camera)DOWNLEFTRIGHTUP)ScreenRectangle)Mobject)WHITE ManimColorceZdZdZddedfd1fd Zed2dZejd3dZed2dZ e jd4dZ ed5dZ e jd6dZ d7fd Z d8d!Z d9d$Z d:d&Z d;draA camera that follows and matches the size and position of its 'frame', a Rectangle (or similar Mobject). The frame defines the region of space the camera displays and can move or resize dynamically. .. SEEALSO:: :class:`.MovingCameraScene` NrframeMobject | Nonefixed_dimensionintdefault_frame_stroke_colorrdefault_frame_stroke_widthkwargsrc ||_||_||_|;ttd}||j|j||_tjdi|dS)zFrame is a Mobject, (should almost certainly be a rectangle) determining which region of space the camera displays N frame_heightheight) rrrrr set_strokersuper__init__)selfrrrrr __class__s U/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/camera/moving_camera.pyr%zMovingCamera.__init__%s /*D'*D' =#6.+ABBBE   //    ""6"""""returnfloatc|jjS)z}Returns the height of the frame. Returns ------- float The height of the frame. )rr!r&s r(rzMovingCamera.frame_height=sz  r)rNonec:|j|dS)zSets the height of the frame in MUnits. Parameters ---------- frame_height The new frame_height. N)rstretch_to_fit_height)r&rs r(rzMovingCamera.frame_heightHs  ((66666r)c|jjS)zzReturns the width of the frame Returns ------- float The width of the frame. )rwidthr-s r( frame_widthzMovingCamera.frame_widthSszr)r3c:|j|dS)zSets the width of the frame in MUnits. Parameters ---------- frame_width The new frame_width. N)rstretch_to_fit_width)r&r3s r(r3zMovingCamera.frame_width^s  '' 44444r)r c4|jS)zReturns the centerpoint of the frame in cartesian coordinates. Returns ------- np.array The cartesian coordinates of the center of the frame. )r get_centerr-s r( frame_centerzMovingCamera.frame_centerisz$$&&&r)r8Point3DLike | Mobjectc:|j|dS)aSets the centerpoint of the frame. Parameters ---------- frame_center The point to which the frame must be moved. If is of type mobject, the frame will be moved to the center of that mobject. N)rmove_to)r&r8s r(r8zMovingCamera.frame_centerts  <(((((r)mobjectsIterable[Mobject]c <tj|fi|dS)N)r$capture_mobjects)r&r<rr's r(r?zMovingCamera.capture_mobjectss* ! 44V44444r) pixel_arrayrcdSzSince the frame can be moving around, the cairo context used for updating should be regenerated at each frame. So no caching. Nr")r&r@s r(get_cached_cairo_contextz%MovingCamera.get_cached_cairo_contexts tr)ctxrcdSrBr")r&r@rDs r(cache_cairo_contextz MovingCamera.cache_cairo_contexts r) list[Mobject]c|jgS)zReturns all mobjects whose movement implies that the camera should think of all other mobjects on the screen as moving Returns ------- list[Mobject] )rr-s r( get_mobjects_indicating_movementz-MovingCamera.get_mobjects_indicating_movements |r)FTmarginonly_mobjects_in_frameboolanimaterc|||\}}}}||zdz } ||zdz } t||z } t||z } |r |jjn|j} | |jjz | |jjz kr?| | | | |zS| | | | |zS)aZooms on to a given array of mobjects (or a singular mobject) and automatically resizes to frame all the mobjects. .. NOTE:: This method only works when 2D-objects in the XY-plane are considered, it will not work correctly when the camera has been rotated. Parameters ---------- mobjects The mobject or array of mobjects that the camera will focus on. margin The width of the margin that is added to the frame (optional, 0 by default). only_mobjects_in_frame If set to ``True``, only allows focusing on mobjects that are already in frame. animate If set to ``False``, applies the changes instead of returning the corresponding animation Returns ------- Union[_AnimationBuilder, ScreenRectangle] _AnimationBuilder that zooms the camera view to a given list of mobjects or ScreenRectangle with position and size updated to zoomed position. r )r2r ) _get_bounding_boxabsrrMr2r!set_xset_yset)r&r<rJrKrMscene_critical_x_leftscene_critical_x_rightscene_critical_y_upscene_critical_y_downxy new_width new_heightm_targets r( auto_zoomzMovingCamera.auto_zoomsR  " "8-C D D  ! "  !#%; ;q @ #8 8A =-0FFGG ,/DDEE )0@4:%%dj tz' '*tz7H*H H H>>!$$**1--11 F8J1KK K>>!$$**1--11f9L1MM Mr)!tuple[float, float, float, float]cd}d}d}d}d}|D]}||jks|r||s&|s|td}|td}|t d}|t d}d}|td|kr |td}|td|kr |td}|t d|kr |t d}|t d|kr |t d}|std||||fS)NFrTzFCould not determine bounding box of the mobjects given to 'auto_zoom'.)r is_in_frameget_critical_pointrrrr Exception) r&r<rKbounding_box_locatedrTrUrVrWms r(rOzMovingCamera._get_bounding_boxs %'(()%&'( J JATZ& /3/?/?/B/B ( J()(<(&>q&A#()(<(*>r*B*B1*E'''--a03HHH,-,@,@,F,Fq,I)# X  " "  !   r)) rrrrrrrrrr)r*r+)rr+r*r.)r3r+r*r.)r*r )r8r9r*r.)r<r=rrr*r.)r@rr*r.)r@rrDrr*r.)r*rG)rFT) r<r=rJr+rKrLrMrLr*r)r<r=rKrLr*r^)__name__ __module__ __qualname____doc__rr%propertyrsetterr3r8r?rCrFrIr]rO __classcell__)r's@r(rrs!% 16*+ #######0!!!X!7777   X 5555'''X' ) ) ) )555555     $', 8N8N8N8N8Nt/ / / / / / / / r)N)ri __future__r__all__collections.abcrtypingrcairor manim.typingrr r r camera.camerar constantsrrrr mobject.framermobject.mobjectr utils.colorrrrr"r)r(rysA#"""""  $$$$$$9999999999""""""------------++++++%%%%%%++++++++w w w w w 6w w w w w r)