k jJddlZddlmZmZmZddlZddlmZmZmZddl m Z ddl m Z ddl mZmZdZdZd Zd Zd Zd ZdZdZdZGd dZGddeZGddeZdS)N)AnyOptionalUnion)cosradianssin)BaseKeys) Projection3D)QWERTY KeyMapFactoryc eZdZdZ d!dedededefd Zed efd Zd ed eded dfdZ deded dfdZ ed efdZ e j ded dfdZ ed efdZ e j ded dfdZ ed ejfdZd"dZ d#deejdeeeeefd ejfdZdejdejdejd ejfd ZdS)$CameraaVSimple camera class containing projection. .. code:: python # create a camera camera = Camera(fov=60.0, aspect_ratio=1.0, near=1.0, far=100.0) # Get the current camera matrix as numpy array print(camera.matrix) # Get projection matrix as numpy array print(camera.projection.matrix) N@?Y@fov aspect_rationearfarc^tjddd|_tjddd|_tjddd|_tjddd|_d|_d|_tjddd|_t|||||_ dS)zInitialize camera using a specific projection Keyword Args: fov (float): Field of view aspect_ratio (float): Aspect ratio near (float): Near plane far (float): Far plane rggVN) glmvec3positionuprightdir_yaw_pitch_upr _projection)selfrrrrs W/home/agentuser/manim-venv/lib/python3.11/site-packages/moderngl_window/scene/camera.py__init__zCamera.__init__(sc3// (3S))Xc3,, 8Cd++  8Cc**( c4EEreturnc|jS)zN:py:class:`~moderngl_window.opengl.projection.Projection3D`: The 3D projection)r'r(s r) projectionzCamera.projectionBs r+xyzNc<tj||||_dS)zSet the 3D position of the camera. Args: x (float): x position y (float): y position z (float): z position N)rrr )r(r0r1r2s r) set_positionzCamera.set_positionGsAq)) r+yawpitchcJ||_||_|dS)zSet the rotation of the camera. Args: yaw (float): yaw rotation pitch (float): pitch rotation N)r%r$_update_yaw_and_pitch)r(r5r6s r) set_rotationzCamera.set_rotationQs*   ""$$$$$r+c|jS)zfloat: The current yaw angle.)r$r.s r)r5z Camera.yaw\s yr+valuec<||_|dSN)r$r8r(r;s r)r5z Camera.yawas!  ""$$$$$r+c|jS)zfloat: The current pitch angle.)r%r.s r)r6z Camera.pitchfs {r+c<||_|dSr=)r%r8r>s r)r6z Camera.pitchks!  ""$$$$$r+c|||j|j|jz|jS)0glm.mat4: The current view matrix for the camera)r8 _gl_look_atr r#r&r.s r)matrixz Camera.matrixps; ""$$$ t}tx/GRRRr+ctjddd}tt|jtt|jz|_tt|j|_tt|jtt|jz|_ tj ||_ tj tj |j |j |_tj tj |j|j |_dS)z=Updates the camera vectors based on the current yaw and pitchrN)rrrrr5r6r0rr1r2 normalizer#crossr&r"r!)r(fronts r)r8zCamera._update_yaw_and_pitchvsc3''gdh''((3wtz/B/B+C+CCgdj))**gdh''((3wtz/B/B+C+CC='']39TXtx#@#@AA - $*dh ? ?@@r+vecposc|tj|}|td||j||jS)aLook at a specific point Either ``vec`` or ``pos`` needs to be supplied. Keyword Args: vec (glm.vec3): position pos (tuple/list): list of tuple ``[x, y, x]`` / ``(x, y, x)`` Returns: glm.mat4x4: Camera matrix Nzvector or pos must be set)rr ValueErrorrCr r&)r(rIrJs r)look_atzCamera.look_atsF ?(3--C ;899 9 sDH===r+targetr!ctj||z }tjtjtj||}tj||}tj}|j |dd<|j |dd<|j |dd<tj}|d|dd<|d|dd<|d|dd<|d|dd<|d|dd<|d|dd<|d|dd<|d|dd<|d|dd<||zS)zThe standard lookAt method. Args: pos: current position target: target position to look at up: direction up Returns: glm.mat4: The matrix rrr r)rrFrGmat4r0r1r2) r(rJrNr!r2r0r1 translaterotates r)rCzCamera._gl_look_ats8 M#, ' ' M#)CM"$5$5q99 : : IaOOHJJ 5& ! Q5& ! Q5& ! Qtq ! tq ! tq ! tq ! tq ! tq ! tq ! tq ! tq !  !!r+)rrrr)r,N)NN)__name__ __module__ __qualname____doc__floatr*propertyr r/r4r9r5setterr6rrPrDr8rrtuplerMrCr+r)rrsf  ]bFFF/4FBGFTYFFFF4 L   X *e**%*D**** % %e % % % % %UX Z%%4%%%Z%uX \%5%T%%%\%SSSSXS A A A A[_>>CH%>3;E%PUBU >>>>*"sx""sx"CH""""""r+rc eZdZdZeddddfdedededed ed ef fd Ze d efd Z e j ded dfdZ e d efdZ e j ded dfdZ de de ded dfdZded dfdZded dfdZded dfdZded dfdZded dfdZded dfdZdeded dfdZd ed!ed dfd"Ze d ejfd#ZxZS)$KeyboardCameraaCamera controlled by mouse and keyboard. The class interacts with the key constants in the built in window types. Creating a keyboard camera: .. code:: python camera = KeyboardCamera( self.wnd.keys, fov=75.0, aspect_ratio=self.wnd.aspect_ratio, near=0.1, far=1000.0, ) We can also interact with the belonging :py:class:`~moderngl_window.opengl.projection.Projection3D` instance. .. code:: python # Update aspect ratio camera.projection.update(aspect_ratio=1.0) # Get projection matrix in bytes (f4) camera.projection.tobytes() rrrkeyskeymaprrrrc||_|||_t|_t|_t|_d|_d|_d|_d|_ t ||||dS)awInitialize the camera Args: keys (BaseKeys): The key constants for the current window type Keyword Args: keymap (KeyMapFactory) : The keymap to use. By default QWERTY. fov (float): Field of view aspect_ratio (float): Aspect ratio near (float): near plane far (float): far plane r$@g?)rrrrN) r^r_STILL_xdir_zdir_ydir _last_time_last_rot_time _velocity_mouse_sensitivitysuperr*)r(r^r_rrrr __class__s r)r*zKeyboardCamera.__init__sw* fTll    !"% S|$CPPPPPr+r,c|jS)zfloat: Mouse sensitivity (rotation speed). This property can also be set:: camera.mouse_sensitivity = 2.5 rir.s r)mouse_sensitivityz KeyboardCamera.mouse_sensitivitys &&r+r;Nc||_dSr=rmr>s r)rnz KeyboardCamera.mouse_sensitivity"'r+c|jS)zfloat: The speed this camera move based on key inputs The property can also be modified:: camera.velocity = 5.0 rhr.s r)velocityzKeyboardCamera.velocitys ~r+c||_dSr=rrr>s r)rszKeyboardCamera.velocitys r+keyaction modifierscv||jjkrP||jjkr|ddS||jjkr|ddSdS||jjkrP||jjkr|ddS||jjkr|ddSdS||jjkrN||jjkr| d||jjkr| ddSdS||jj krN||jjkr| d||jjkr| ddSdS||jj krN||jjkr| d||jjkr| ddSdS||jjkrL||jjkr|d||jjkr|ddSdSdS)zProcess key inputs and move camera Args: key: The key action: key action release/press modifiers: key modifier states such as ctrl or shit TFN)r_RIGHTr^ ACTION_PRESS move_rightACTION_RELEASELEFT move_leftFORWARD move_forwardBACKWARD move_backwardDOWN move_downUPmove_up)r(rurvrws r) key_inputzKeyboardCamera.key_inputsu $+# # #///%%%%%49333&&&&&43DK$ $ $///t$$$$$49333u%%%%%43DK' ' '///!!$'''111!!%(((((21DK( ( (///""4(((111""5)))))21DK$ $ $///t$$$111u%%%%%21DKN " "/// T"""111 U##### # "21r+activatec<|t|dS)zThe camera should be continiously moving to the left. Args: activate (bool): Activate or deactivate this state N) move_stater}r(rs r)r~zKeyboardCamera.move_leftB h'''''r+c<|t|dS)zThe camera should be continiously moving to the right. Args: activate (bool): Activate or deactivate this state N)rryrs r)r{zKeyboardCamera.move_rightJs x(((((r+c<|t|dS)zThe camera should be continiously moving forward. Args: activate (bool): Activate or deactivate this state N)rrrs r)rzKeyboardCamera.move_forwardRs *****r+c<|t|dS)zThe camera should be continiously moving backwards. Args: activate (bool): Activate or deactivate this state N)rrrs r)rzKeyboardCamera.move_backwardZs (+++++r+c<|t|dS)zThe camera should be continiously moving up. Args: activate (bool): Activate or deactivate this state N)rrrs r)rzKeyboardCamera.move_upbs H%%%%%r+c<|t|dS)zThe camera should be continiously moving down. Args: activate (bool): Activate or deactivate this state N)rrrs r)rzKeyboardCamera.move_downjrr+ directionc|tkr|rtnt|_dS|tkr|rt nt|_dS|t kr|rt nt|_dS|tkr|rtnt|_dS|tkr|rtnt|_ dS|tkr|rt nt|_ dSdS)zSet the camera position move state. Args: direction: What direction to update activate: Start or stop moving in the direction N) ryPOSITIVErbrcr}NEGATIVErrdrrrer)r(rrs r)rzKeyboardCamera.move_staters   %-85DJJJ $  %-85DJJJ ' ! !%-85DJJJ ( " "%-85DJJJ "__%-85DJJJ $  %-85DJJJ r+dxdyctj}||jz }||_|dkr0tt|t|dkrdS||jz}||jz}|xj|zc_|xj|z c_|jdkrd|_|jdkrd|_|dS)aUpdate the rotation of the camera. This is done by passing in the relative mouse movement change on x and y (delta x, delta y). In the past this method took the viewport position of the mouse. This does not work well when mouse exclusivity mode is enabled. Args: dx: Relative mouse position change on x dy: Relative mouse position change on y g?rNg@U@g@U) timergmaxabsrir$r%r6r8)r(rrnowdeltas r) rot_statezKeyboardCamera.rot_statesikkd))! 3;;3s2wwB00144 F d%% d%% R  r :  DJ :  DJ ""$$$$$r+ctj}t||jz d}||_|jtkr!|xj|j|jz|zz c_n0|jtkr |xj|j|jz|zzc_|j tkr!|xj|j |jz|zz c_n0|j tkr |xj|j |jz|zzc_|j tkr!|xj|j |jz|zz c_n0|j tkr |xj|j |jz|zzc_| |j|j|j z|jS)z2glm.mat4x4: The current view matrix for the camerar)rrrfrcrr r"rhrrdr#rer!rCr&)r(rts r)rDzKeyboardCamera.matrixsaikk do%q ) ) : ! ! MMTZ$.81< !'!!Q!Q!Q!Q !Q  !Q  !Q!Q!Q!Q!Q!Q!QF'5'''X'(u(((((%X_e_-$S-$#-$#-$$-$-$-$-$^($(4(((()4)D))))+T+d++++,d,t,,,,&&&&&&($(4((((9C949D9999(#%E#%u#%#%#%#%#%JSSSSXSSSSSr+r]c eZdZdZ ddeejeeeeffdedeeefde ffd Z e d ej fd Z e d efd Zejd ed dfdZe d efdZejd ed dfdZe d efdZejd ed dfdZe d efdZejd ed dfdZdeded dfdZded dfdZxZS) OrbitCameraaCamera controlled by the mouse to pan around the target. The functions :py:function:`~camera.OrbitCamera.rot_state` and :py:function:`~camera.OrbitCamera.rot_state` are used to update the rotation and zoom. Creating a orbit camera: .. code:: python camera = OrbitCamera( target=(0., 0., 0.), radius=2.0 fov=75.0, aspect_ratio=self.wnd.aspect_ratio, near=0.1, far=1000.0, ) We can also interact with the belonging :py:class:`~moderngl_window.opengl.projection.Projection3D` instance. .. code:: python # Update aspect ratio camera.projection.update(aspect_ratio=1.0) # Get projection matrix in bytes (f4) camera.projection.tobytes() rrr@gF@gFrNradiusangleskwargsc ||_|\|_|_tj||_tjddd|_d|_d|_tj di|dS)aoInitialize the camera Keyword Args: target (float, float, float): Target point radius (float): Radius angles (float, float): angle_x and angle_y in degrees fov (float): Field of view aspect_ratio (float): Aspect ratio near (float): near plane far (float): far plane rrNr[) rangle_xangle_yrrrNr!ri_zoom_sensitivityrjr*)r(rNrrrrks r)r*zOrbitCamera.__init__ss& %+" dlhv&& (3S))"%!$""6"""""r+r,c&tt|jtt|jz|jz|jdztt|j|jz|jdztt|jtt|jz|jz|jdzf}|j|tj ||j|j S)rBrr r) rrrrrrrNr4rlookAtr!)r(r s r)rDzOrbitCamera.matrixs  %% & &WT\-B-B)C)C Cdk QTXT_`aTb b  %% & & 4t{1~ E  %% & &WT\-B-B)C)C Cdk QTXT_`aTb b  8$$z  K G   r+c|jS)ztfloat: camera angle x in degrees. This property can also be set:: camera.angle_x = 45. _angle_xr.s r)rzOrbitCamera.angle_x }r+r;Nc||_dS)z!Set camera rotation_x in degrees.Nrr>s r)rzOrbitCamera.angle_x r+c|jS)ztfloat: camera angle y in degrees. This property can also be set:: camera.angle_y = 45. _angle_yr.s r)rzOrbitCamera.angle_y$rr+c||_dS)z!Set camera rotation_y in degrees.Nrr>s r)rzOrbitCamera.angle_y-rr+c|jS)zfloat: Mouse sensitivity (rotation speed). This property can also be set:: camera.mouse_sensitivity = 2.5 rmr.s r)rnzOrbitCamera.mouse_sensitivity2s &&r+c||_dSr=rmr>s r)rnzOrbitCamera.mouse_sensitivity;rpr+c|jS)zfloat: Mousewheel zooming sensitivity (zoom speed). This property can also be set:: camera.zoom_sensitivity = 2.5 rr.s r)zoom_sensitivityzOrbitCamera.zoom_sensitivity?s %%r+c||_dSr=rr>s r)rzOrbitCamera.zoom_sensitivityHs!&r+rrc|xj||jzdz z c_|xj||jzdz z c_tt |jdd|_dS)a"Update the rotation of the camera around the target point. This is done by passing relative mouse change in the x and y axis (delta x, delta y) Args: dx: Relative mouse position change on x axis dy: Relative mouse position change on y axis raggeN)rrnrrmin)r(rrs r)rzOrbitCamera.rot_stateLsa T33d::  T33d:: 3t|T22F;; r+y_offsetcj|xj||jzzc_td|j|_dS)Nr)rrr)r(rs r) zoom_statezOrbitCamera.zoom_state[s1 x$"888 #t{++ r+)rrr)rSrTrUrVrrrrZrWrr*rXrPrDrrYrrnrrrrrs@r)rrst@?N&3 ##cheUE&9 ::;##eUl# #  ######:      X  X ^Ut^X ^Ut^'5'''X'(u(((((&%&&&X&'e''''' rsj '''''''''' !!!!!!!!!!111111::::::????????      ["["["["["["["["|PSPSPSPSPSVPSPSPSfT,T,T,T,T,&T,T,T,T,T,r+