l jL2dZddlmZddgZddlZddlmZmZddlZ ddl m Z ddl m Z dd lmZdd lmZdd lmZd d lmZd dlmZd dlmZd dlmZd dlmZmZd dlm Z d dl!m"Z"m#Z#d dl$m%Z%d dl&m'Z'd dl(m)Z)Gdde'Z*Gdde*Z+dS)zHA scene suitable for rendering three-dimensional objects and animations.) annotations ThreeDSceneSpecialThreeDSceneN)IterableSequence)Line) ThreeDAxes) OpenGLMobject)Sphere) ValueTracker)config) Animation) Transform) ThreeDCamera)DEGREES RendererType)Mobject)VectorizedPointVGroup) OpenGLCamera)Scene)merge_dicts_recursivelyceZdZdZeddffd Z d&d'd Zd(d)dZd*dZ d+d,dZ dZ ddddddgfd-dZ d.fd Z d/d!Z d/d"Zd/d#Zd/d$Zd%ZxZS)0rz} This is a Scene, with special configurations and properties that make it suitable for Three Dimensional Scenes. Nc ||_|dtzdtzd}||_tjdd|i|dS)NFiyphitheta camera_class)ambient_camera_rotationr(default_angled_camera_orientation_kwargssuper__init__)selfr r"r#kwargs __class__s T/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/scene/three_d_scene.pyr%zThreeDScene.__init__%se(?$ 3 ;G|88 4 5 5 ==l=f=====r float | Nonergammazoomfocal_distance frame_center Mobject | Sequence[float] | Nonec ||jj|||jj|||jj|||jj|||jj||&|jjj|dSdS)a This method sets the orientation of the camera in the scene. Parameters ---------- phi The polar angle i.e the angle between Z_AXIS and Camera through ORIGIN in radians. theta The azimuthal angle i.e the angle that spins the camera around the Z_AXIS. focal_distance The focal_distance of the Camera. gamma The rotation of the camera about the vector from the ORIGIN to the Camera. zoom The zoom factor of the scene. frame_center The new center of the camera frame in cartesian coordinates. N) renderercameraset_phi set_thetaset_focal_distance set_gammaset_zoom _frame_centermove_to)r&rrr,r-r.r/r's r)set_camera_orientationz"ThreeDScene.set_camera_orientation7sD ? M ( ( - - -   M * *5 1 1 1  % M 3 3N C C C   M * *5 1 1 1   M ) )$ / / /  # M . 6 6| D D D D D $ #r*{Gz?ratefloataboutstrc@ tjtjkr\|jj|jj|jjd}| fd| dStjtj krR|j}|j |j |jd| fd| |jdSdS#t$r}t!d|d}~wwxYw)a This method begins an ambient rotation of the camera about the Z_AXIS, in the anticlockwise direction Parameters ---------- rate The rate at which the camera should rotate about the Z_AXIS. Negative rate means clockwise rotation. about one of 3 options: ["theta", "phi", "gamma"]. defaults to theta. rrr,c4|zSN)increment_value)mdtr=xs r)z;ThreeDScene.begin_ambient_camera_rotation..~sA,=,=dRi,H,Hr*c,|zSrDr!)rFrGr?methodsr=s r)rIz;ThreeDScene.begin_ambient_camera_rotation..sngenTBY.G.Gr*Invalid ambient rotation angle.N)lowerrr2rCAIROr3 theta_tracker phi_tracker gamma_tracker add_updateraddOPENGLincrement_theta increment_phiincrement_gamma Exception ValueError)r&r=r?trackerscamerKrHs `` @@r)begin_ambient_camera_rotationz)ThreeDScene.begin_ambient_camera_rotationfs=[[]] G,"444![6;2![6 #+5/ HHHHHIII L$777$(K 0, 0  G G G G G GHHH%%%%%87 G G G>??Q F GsA4C>A*C>> DDDc|} tjtjkrW|jj|jj|jjd}||}| | |dStjtj kr|j dSdS#t$r}td|d}~wwxYw)z.This method stops all ambient camera rotation.rBrLN)rMrr2rrNr3rOrPrQclear_updatersremoverTrXrY)r&r?rZrHr\s r)stop_ambient_camera_rotationz(ThreeDScene.stop_ambient_camera_rotations[[]] G,"444![6;2![6 #+5/  """ AL$777 **,,,,,87 G G G>??Q F GsA/B>3B>> CCC origin_phi origin_thetac#|jjj#|jjjt dfd}|jjj|||jjjt dfd}|jjj|||jjjdS)a This method creates a 3D camera rotation illusion around the current camera orientation. Parameters ---------- rate The rate at which the camera rotation illusion should operate. origin_phi The polar angle the camera should move around. Defaults to the current phi angle. origin_theta The azimutal angle the camera should move around. Defaults to the current theta angle. Nrc|zdtjz}||zS)Ng?)rEnpsin get_value set_value)rFrGval_for_left_rightrdr=val_tracker_thetas r) update_thetazBThreeDScene.begin_3dillusion_camera_rotation..update_thetasU  - -b4i 8 8 8!$rv.?.I.I.K.K'L'L!L ;;|.@@AA Ar*c|zdtjzdz }||zS)Ng?)rErgcosrirj)rFrGval_for_up_downrcr=val_tracker_phis r) update_phiz@ThreeDScene.begin_3dillusion_camera_rotation..update_phisW  + +BI 6 6 6!BF?+D+D+F+F$G$GG#MO;;zO;<< FF|TTUUU _ 3 3 3 $ C88::D| $ G'lM::=#/#:#:#<#z2ThreeDScene.get_moving_mobjects..Fs(??r_$??????r*)r$get_moving_mobjectsr2r3get_value_trackersr9anymobjects)r&rcamera_mobjectsrr(s @r)rzThreeDScene.get_moving_mobjects8sz6%''5zB-.AACC M .G   ??????? ? ? != r*rrctjtjkr#|j||jjj|i|dStjtjkr.|D]-}|||,dSdS)a4 This method is used to prevent the rotation and tilting of mobjects as the camera moves around. The mobject can still move in the x,y,z directions, but will always be at the angle (relative to the camera) that it was at when it was passed through this method.) Parameters ---------- *mobjects The Mobject(s) whose orientation must be fixed. **kwargs Some valid kwargs are use_static_center_func : bool center_func : function N) rr2rrNrSr3add_fixed_orientation_mobjectsrTfix_orientation)r&rr'mobs r)rz*ThreeDScene.add_fixed_orientation_mobjectsJs$ ?l0 0 0 DHh   ?DM ? TV T T T T T _ 3 3 3  ##%%% 4 3  r*ctjtjkr|j|||jj|dStjtjkr.|D]-}|||,dSdS)a[ This method is used to prevent the rotation and movement of mobjects as the camera moves around. The mobject is essentially overlaid, and is not impacted by the camera's movement in any way. Parameters ---------- *mobjects The Mobjects whose orientation must be fixed. N) rr2rrNrSr3add_fixed_in_frame_mobjectsrT fix_in_framer&rrs r)rz'ThreeDScene.add_fixed_in_frame_mobjectses ?l0 0 0 DHh    % 3DK 3X > > > > _ 3 3 3    """ 4 3  r*ctjtjkr|jjj|dStjtjkr.|D]-}|||,dSdS)a This method "unfixes" the orientation of the mobjects passed, meaning they will no longer be at the same angle relative to the camera. This only makes sense if the mobject was passed through add_fixed_orientation_mobjects first. Parameters ---------- *mobjects The Mobjects whose orientation must be unfixed. N) rr2rrNr3!remove_fixed_orientation_mobjectsrTunfix_orientationr`rs r)rz-ThreeDScene.remove_fixed_orientation_mobjects{s ?l0 0 0 BDM BH M M M M _ 3 3 3 ! !%%''' C    4 3 ! !r*ctjtjkr|jjj|dStjtjkr.|D]-}|||,dSdS)a$ This method undoes what add_fixed_in_frame_mobjects does. It allows the mobject to be affected by the movement of the camera. Parameters ---------- *mobjects The Mobjects whose position and orientation must be unfixed. N) rr2rrNr3remove_fixed_in_frame_mobjectsrTunfix_from_framer`rs r)rz*ThreeDScene.remove_fixed_in_frame_mobjectss ?l0 0 0 ?DM ? J J J J _ 3 3 3 ! !$$&&& C    4 3 ! !r*c ht|j}|j||jdi|dS)aq This method sets the default_angled_camera_orientation to the keyword arguments passed, and sets the camera to that orientation. Parameters ---------- **kwargs Some recognised kwargs are phi, theta, focal_distance, gamma, which have the same meaning as the parameters in set_camera_orientation. Nr!)dict!default_camera_orientation_kwargsupdater;r&r'rs r)(set_to_default_angled_camera_orientationz4ThreeDScene.set_to_default_angled_camera_orientationsL  2    f##--f-----r*)NNNNNN) rr+rr+r,r+r-r+r.r+r/r0)r<r)r=r>r?r@)r)rbNN)r=r>rcr+rdr+)rr+rr+r,r+r-r+r.r+r/r0rwrx)rr)rr)__name__ __module__ __qualname____doc__rr%r;r]rarsrvrrrrrrr __classcell__r(s@r)rrs" $15 >>>>>>(!""!'+9=-E-E-E-E-E^$G$G$G$G$GLGGGG(#'%) ,3,3,3,3,3\666!""!'+9=+-c3c3c3c3c3J$6,!!!!(!!!!(.......r*c eZdZdZdddddddgdddddd d d ezd ezd ddiddiddidffd ZdZdZdZdZ xZ S)rakAn extension of :class:`ThreeDScene` with more settings. It has some extra configuration for axes, spheres, and an override for low quality rendering. Further key differences are: * The camera shades applicable 3DMobjects by default, except if rendering in low quality. * Some default params for Spheres and Axes have been added. T)should_apply_shadingexponential_projectionrbr )rrbr ) unit_sizetick_frequencynumbers_with_elongated_ticks stroke_width)num_axis_pieces axis_config)0)radius resolutionrirrFrr) r) camera_configthree_d_axes_config sphere_configc ||_||_||_||_||_||_|jjdtdkri}n|j}t||}tj di|dS)N pixel_widthr!) cut_axes_at_radiusrrrdefault_angled_camera_positionlow_quality_configr2rrr$r%) r&rrrrrrr'_configr(s r)r%zSpecialThreeDScene.__init__s6#5*#6 *.L+"4 = &} 5 9N N NGG-G)'6::##7#####r*c Ntdi|j}|D]}|jr|}|d}|d}|}t t||t||t||}|D] }d|_ | |j |j |j _ |j D]9} | td| z:|S)z}Return a set of 3D axes. Returns ------- :class:`.ThreeDAxes` A set of 3D axes. rbTg?r!)r rr get_startnumber_to_pointget_endrr shade_in_3d match_stylepieces submobjects tick_marksrSrr) r&axesaxisp0p1p2p3 new_piecespieceticks r)get_axeszSpecialThreeDScene.get_axess)55D455 C CD& A^^%%))"--))!,,\\^^#DRLL$r2,,R MM '--E(,E%%&&t{333*4*@ ' C Ct/@/@)@AABBBB C r*c Dt|j|}tdi|S)a1 Returns a sphere with the passed keyword arguments as properties. Parameters ---------- **kwargs Any valid parameter of :class:`~.Sphere` or :class:`~.Surface`. Returns ------- :class:`~.Sphere` The sphere object. r!)rrr rs r) get_spherezSpecialThreeDScene.get_spheres*));VDDr*c|jS)z Returns the default_angled_camera position. Returns ------- dict Dictionary of phi, theta, focal_distance, and gamma. )rrus r)get_default_camera_positionz.SpecialThreeDScene.get_default_camera_positions 22r*c*|jdi|jdS)z(Sets the camera to its default position.Nr!)r;rrus r)set_camera_to_default_positionz1SpecialThreeDScene.set_camera_to_default_positions$##JJd&IJJJJJr*) rrrrrr%rrrrrrs@r)rrs    /3tTT "#09 !   "#(;;<G^( ( 5e<$5q#9*H5  )&$&$&$&$&$&$P0   " 3 3 3KKKKKKKr*),r __future__r__all__rcollections.abcrrnumpyrgmanim.mobject.geometry.liner)manim.mobject.graphing.coordinate_systemsr #manim.mobject.opengl.opengl_mobjectr &manim.mobject.three_d.three_dimensionsr manim.mobject.value_trackerr ranimation.animationranimation.transformrcamera.three_d_camerar constantsrrmobject.mobjectr mobject.types.vectorized_mobjectrrrenderer.opengl_rendererr scene.scenerutils.config_opsrrrr!r*r)rsNN"""""" . /........,,,,,,@@@@@@======999999444444++++++++++++000000--------%%%%%%FFFFFFFF333333666666S.S.S.S.S.%S.S.S.l kKkKkKkKkKkKkKkKkKkKr*