l jT=dZddlmZdgZddlmZmZddlmZddl Z ddl m Z ddl mZmZmZmZdd lmZdd lmZdd lmZmZmZmZmZd d lmZd dlmZd dl Td dl!m"Z"d dl#m$Z$d dl%m&Z&d dl'm(Z(m)Z)GddeZ*dS)zHA camera that can be positioned and oriented in three-dimensional space.) annotations ThreeDCamera)CallableIterable)AnyN)Mobject)get_3d_vmob_end_corner"get_3d_vmob_end_corner_unit_normalget_3d_vmob_start_corner$get_3d_vmob_start_corner_unit_normal)VMobject) ValueTracker)FloatRGBA_ArrayMatrixMNPoint3D Point3D_Array Point3DLike)config)Camera)*)Point)get_shaded_rgb)extract_mobject_family_members)rotation_about_zrotation_matrixc |eZdZddddezdezzdezzddd d ezd d f dQfd ZedRdZ e j dSd!Z dTfd$ Z dUd&Z dVd+Z dWdXd-ZdYd.ZdZfd1 Zd[d2Zd[d3Zd[d4Zd[d5Zd[d6Zd\d8Zd\d9Zd\d:Zd\d;Zd\d<Zd]d=Zd^d?Zd^d@Zd_dCZd`dDZdafdG Z ddHdIdbdMZ!dcdNZ"dcdOZ#dcdPZ$xZ%S)drg4@g?g@  TFrifocal_distancefloatshading_factordefault_distancelight_source_start_pointrshould_apply_shadingboolexponential_projectionphithetagammazoomkwargsrc  t| dtd|_t jdi| ||_||_||_| |_ | |_ ||_ ||_ ||_ t|j |_||_||_dt"dz|_t'|j|_t'|j|_t'|j|_t'|j |_t'|j |_i|_t5|_|dS)zInitializes the ThreeDCamera Parameters ---------- *kwargs Any keyword argument of Camera. frame_centerr) stroke_width frame_widthN)rgetORIGIN _frame_centersuper__init__r"r*r+r,r-r$r%r& light_sourcer'r)rmax_allowable_normr phi_tracker theta_trackerfocal_distance_tracker gamma_tracker zoom_trackerfixed_orientation_mobjectssetfixed_in_frame_mobjectsreset_rotation_matrix) selfr"r$r%r&r'r)r*r+r,r-r. __class__s V/home/agentuser/manim-venv/lib/python3.11/site-packages/manim/camera/three_d_camera.pyr9zThreeDCamera.__init__(s>*#6::nf#E#ETUVVV""6""",   , 0(@%!$"?@@$8!&<#"#f]&;";'11)$*55&243F&G&G#)$*55(33PR'58UU$ ""$$$$$returnrc&|jjdS)Nr)r7pointsrEs rGr0zThreeDCamera.frame_centerTs!(++rHpointNonec:|j|dSN)r7move_torErMs rGr0zThreeDCamera.frame_centerXs ""5)))))rHmobjectsIterable[Mobject]c d|tj|fi|dSrP)rDr8capture_mobjects)rErSr.rFs rGrVzThreeDCamera.capture_mobjects\s: ""$$$  44V44444rHlist[ValueTracker]cB|j|j|j|j|jgS)zA list of :class:`ValueTrackers <.ValueTracker>` of phi, theta, focal_distance, gamma and zoom. Returns ------- list list of ValueTracker objects )r<r=r>r?r@rLs rGget_value_trackerszThreeDCamera.get_value_trackers`s,      '       rHvmobjectr rgbasrc|js|S|jr|dkr|jjd}t |dkr|dd}ntj|dd}t|dddft|t|||dddf<t|dddft|t|||dddf<|S|S)Nrr)axisr2r!)r' shade_in_3dget_num_pointsr:rKlenrepeatnparrayrr r r r )rErZr[light_source_point shaded_rgbass rGmodified_rgbaszThreeDCamera.modified_rgbasqs( L   X%<%<%>%>%B%B!%!2!9!!< 5zzA~~$||AA|66 !xbqb 22 "0QU#(224X>>" ##LBQB  #1QU#&x0028<<" ##LBQB    rH backgroundcT||||SrP)rfget_stroke_rgbas)rErZrgs rGrizThreeDCamera.get_stroke_rgbass( ""8X-F-Fz-R-RSSSrHcR|||SrP)rfget_fill_rgbas)rErZs rGrkzThreeDCamera.get_fill_rgbass&""8X-D-D-F-FGGGrHargs list[Mobject]ctj|i|}|dfd }t||S)NmobrrIr#ct|dr|js tjStj|jd}|S)Nr^r)hasattrr^rbinfdotget_z_index_reference_pointT)rodistance rot_matrixs rGz_keyz3ThreeDCamera.get_mobjects_to_display..z_keysQC// CO v !fS%D%D%F%F UUVWXHOrH)key)rorrIr#)r8get_mobjects_to_displayget_rotation_matrixsorted)rErlr.rSrxrwrFs @rGrzz$ThreeDCamera.get_mobjects_to_displayse35772DCFCC--//       hE****rHc4|jS)zReturns the Polar angle (the angle off Z_AXIS) phi. Returns ------- float The Polar angle in radians. )r< get_valuerLs rGget_phizThreeDCamera.get_phis))+++rHc4|jS)zReturns the Azimuthal i.e the angle that spins the camera around the Z_AXIS. Returns ------- float The Azimuthal angle in radians. )r=r~rLs rG get_thetazThreeDCamera.get_thetas!++---rHc4|jS)zReturns focal_distance of the Camera. Returns ------- float The focal_distance of the Camera in MUnits. )r>r~rLs rGget_focal_distancezThreeDCamera.get_focal_distances*44666rHc4|jS)aReturns the rotation of the camera about the vector from the ORIGIN to the Camera. Returns ------- float The angle of rotation of the camera about the vector from the ORIGIN to the Camera in radians )r?r~rLs rG get_gammazThreeDCamera.get_gammas!++---rHc4|jS)zReturns the zoom amount of the camera. Returns ------- float The zoom amount of the camera. )r@r~rLs rGget_zoomzThreeDCamera.get_zooms **,,,rHvaluec:|j|dS)zSets the polar angle i.e the angle between Z_AXIS and Camera through ORIGIN in radians. Parameters ---------- value The new value of the polar angle in radians. N)r< set_valuerErs rGset_phizThreeDCamera.set_phis! ""5)))))rHc:|j|dS)zSets the azimuthal angle i.e the angle that spins the camera around Z_AXIS in radians. Parameters ---------- value The new value of the azimuthal angle in radians. N)r=rrs rG set_thetazThreeDCamera.set_theta! $$U+++++rHc:|j|dS)zSets the focal_distance of the Camera. Parameters ---------- value The focal_distance of the Camera. N)r>rrs rGset_focal_distancezThreeDCamera.set_focal_distances! #--e44444rHc:|j|dS)zSets the angle of rotation of the camera about the vector from the ORIGIN to the Camera. Parameters ---------- value The new angle of rotation of the camera. N)r?rrs rG set_gammazThreeDCamera.set_gammarrHc:|j|dS)zSets the zoom amount of the camera. Parameters ---------- value The zoom amount of the camera. N)r@rrs rGset_zoomzThreeDCamera.set_zooms! ##E*****rHc8||_dS)zySets the value of self.rotation_matrix to the matrix corresponding to the current position of the camera N)generate_rotation_matrixrrLs rGrDz"ThreeDCamera.reset_rotation_matrix s $<<>>rHrc|jS)zReturns the matrix corresponding to the current position of the camera. Returns ------- np.array The matrix corresponding to the current position of the camera. )rrLs rGr{z ThreeDCamera.get_rotation_matrixs ##rHcX|}|}|}t| dtzz t | t t|g}tjd}|D]}tj ||}|S)zGenerates a rotation matrix based off the current position of the camera. Returns ------- np.array The matrix corresponding to the current position of the camera. Zr2) rrrrDEGREESrRIGHTrbidentityrs)rEr*r+r,matricesresultmatrixs rGrz%ThreeDCamera.generate_rotation_matrixsllnn     eVb7l2 3 3 SD% ( ( U # #  Q , ,FVFF++FF rHrKrc|j}|}|}|}||z }t j||j}|dddf}dD]c}|jr/t j||z }|dk} |||| z z || <n|||z z }d|||z dk<|dd|fxx||zzcc<d|S)a&Applies the current rotation_matrix as a projection matrix to the passed array of points. Parameters ---------- points The list of points to project. Returns ------- np.array The points after projecting. Nr)rr!ri@B) r0rrr{rbrsrur)exp) rErKr0r"r-rwzsifactorlt0s rGproject_pointszThreeDCamera.project_points1s( 0022}}--// ,& -- AAAqD\ * *A* : ^ 3441f,C0HIs '>B+>?49+q01 111a4LLLFTM )LLLL rHcf||ddddfS)aApplies the current rotation_matrix as a projection matrix to the passed point. Parameters ---------- point The point to project. Returns ------- np.array The point after projection. )r!r2rN)rreshaperRs rG project_pointzThreeDCamera.project_pointVs1""5==#8#899!QQQ$??rHmobjectrc t||}||jv}||jv}|r|S|r4|j|}|}||}|||z zS||SrP)r8transform_points_pre_displayrArCrr) rErrKfixed_orientationfixed_in_frame center_funccenter new_centerrFs rGrz)ThreeDCamera.transform_points_pre_displayfs 55gvFF#t'FF D$@@  M  /9'BK []]F++F33JZ&01 1&&v.. .rHN)use_static_center_funcrrrCallable[[], Point3D] | Nonecdd}|D]=}|r|}n|r ||}n|j}|D] }||j|< >dS)aThis method allows the mobject to have a fixed orientation, even when the camera moves around. E.G If it was passed through this method, facing the camera, it will continue to face the camera even as the camera moves. Highly useful when adding labels to graphs and the like. Parameters ---------- *mobjects The mobject whose orientation must be fixed. use_static_center_func Whether or not to use the function that takes the mobject's center as centerpoint, by default False center_func The function which returns the centerpoint with respect to which the mobject will be oriented, by default None rrrICallable[[], Point3D]c6|fdS)NcSrPr4)rMsrGz]ThreeDCamera.add_fixed_orientation_mobjects..get_static_center_func..s5rH) get_center)rrMs @rGget_static_center_funczKThreeDCamera.add_fixed_orientation_mobjects..get_static_center_funcs &&((E === rHN)rrrIr)r get_familyrA)rErrrSrrfuncsubmobs rGadd_fixed_orientation_mobjectsz+ThreeDCamera.add_fixed_orientation_mobjectsys4 ! ! ! !  ? ?G *"' *--g66)!,,.. ? ?:>/77 ? ? ?rHc^t|D]}|j|dS)aThis method allows the mobject to have a fixed position, even when the camera moves around. E.G If it was passed through this method, at the top of the frame, it will continue to be displayed at the top of the frame. Highly useful when displaying Titles or formulae or the like. Parameters ---------- **mobjects The mobject to fix in frame. N)rrCaddrErSrs rGadd_fixed_in_frame_mobjectsz(ThreeDCamera.add_fixed_in_frame_mobjectss?6h?? 6 6G  ( , ,W 5 5 5 5 6 6rHcLt|D]}||jvr|j|=dS)aUIf a mobject was fixed in its orientation by passing it through :meth:`.add_fixed_orientation_mobjects`, then this undoes that fixing. The Mobject will no longer have a fixed orientation. Parameters ---------- mobjects The mobjects whose orientation need not be fixed any longer. N)rrArs rG!remove_fixed_orientation_mobjectsz.ThreeDCamera.remove_fixed_orientation_mobjectss?6h?? = =G$9993G< = =rHcpt|D]%}||jvr|j|&dS)a>If a mobject was fixed in frame by passing it through :meth:`.add_fixed_in_frame_mobjects`, then this undoes that fixing. The Mobject will no longer be fixed in frame. Parameters ---------- mobjects The mobjects which need not be fixed in frame any longer. N)rrCremovers rGremove_fixed_in_frame_mobjectsz+ThreeDCamera.remove_fixed_in_frame_mobjectssK6h?? = =G$666,33G<<< = =rH)r"r#r$r#r%r#r&rr'r(r)r(r*r#r+r#r,r#r-r#r.r)rIr)rMrrIrN)rSrTr.rrIrN)rIrW)rZr r[rrIr)F)rZr rgr(rIr)rZr rIr)rlrr.rrIrm)rIr#)rr#rIrN)rIrN)rIr)rKrrIr)rMrrIr)rrrKrrIr)rSrrr(rrrIrN)rSrrIrN)&__name__ __module__ __qualname__DOWNLEFTOUTrr9propertyr0setterrVrYrfrirkrzrrrrrrrrrrrDr{rrrrrrrr __classcell__)rFs@rGrr's!% #"%01D1t80Cb3h0N%)',W}*%*%*%*%*%*%*%X,,,X,****555555    ":!TTTTTHHHH ++++++ ,,,,....7777 . . . .----****,,,,5555,,,,++++???? $$$$*####J@@@@ //////,(-48 &?&?&?&?&?&?P6666 = = = = = = = = = = = =rH)+__doc__ __future__r__all__collections.abcrrtypingrnumpyrbmanim.mobject.mobjectr#manim.mobject.three_d.three_d_utilsr r r r &manim.mobject.types.vectorized_mobjectr manim.mobject.value_trackerr manim.typingrrrrrr camera.camerar constants!mobject.types.point_cloud_mobjectr utils.colorr utils.familyrutils.space_opsrrrr4rHrGrsNN""""""  /.......)))))) <;;;;;444444""""""555555((((((999999????????d=d=d=d=d=6d=d=d=d=d=rH