^ jNB <ddlZddlZddlmZmZmZddlmZmZm Z m Z m Z ddl m Z ddlZddlmZddlmZddlmZmZmZmZmZddlmZdd lmZdd lmZd d d ddddZd d ddZiZ dZ!Gddej"Z#dZ$e$Z%ej&ddddddddddd Z'Gdd e Z(e()d!e()d"e()d#e(Z*Gd$d%eZ+d+d&Z,d'Z-d(Z.d+d)Z/d+d*Z0dS),N)ListDictOptional)WM_DEVICECHANGEDBT_DEVICEARRIVALDBT_DEVICEREMOVECOMPLETEDBT_DEVTYP_DEVICEINTERFACEDEVICE_NOTIFY_WINDOW_HANDLE)EventDispatcher)base)win32)dinput_user32DEV_BROADCAST_DEVICEINTERFACEcomDEV_BROADCAST_HDR) _kernel32) get_mapping)ControllerManagerxyzrxryrz)rwheel)rrrcP|j}|j}tj|}|tjzr2t |}tj|dd|}n|tj zr0t|}tj ||}nw|tj zr0t|}tj||}n8|tjzr'tjtjjdd|}ndS|j|_|S)Nril)tszNamedwTyperDIDFT_GETINSTANCE DIDFT_ABSAXIS_abs_instance_namesgetr AbsoluteAxis DIDFT_RELAXIS_rel_instance_names RelativeAxis DIDFT_BUTTON_btn_instance_namesButton DIDFT_POVHAT_type)object_instanceraw_name ctrl_typeinstancenamecontrols Y/home/agentuser/manim-venv/lib/python3.11/site-packages/pyglet/input/win32/directinput.py_create_controlr:)s &H&I' 22H6'' "&&x00#D!VX>> V) ) "&&x00#D(33 V( ("&&x00+dH-- V% %#D$5$91j(SS#*GM Nc\eZdZfdZdZdZdZdZdZdd Z d Z d Z d Z d Z xZS)DirectInputDevicecX|j}t|||jdz|_|jdz|_||_|||j |_ t|j j d|_||_dS)N08x)tszInstanceNamesuper__init__ dwDevTyper2_subtype_device_init_controls _set_formattszProductNameid_nameformat guidProductData1id_product_guiddevice_instance)selfdisplaydevicerPr7 __class__s r9rDzDirectInputDevice.__init__As. $'''$.5 '1F:    &5 %o&A&GOO.r;c8|jdSN)rGReleaserQs r9__del__zDirectInputDevice.__del__Os r;c|jdd|jddz}|jdd|jddz}d|d|dS) z2Generate an SDL2 style GUID from the product guid.rrr030000000000 000000000000)rO)rQfirstseconds r9get_guidzDirectInputDevice.get_guidRs_$QqS)D,@1,EE%ac*T-A!A#-FF9%99V9999r;cg|_|jtj|jdtj|jddS)Nc|jSrV)r2)cs r9z2DirectInputDevice._init_controls..[sr;)key)controlsrG EnumObjectsrLPDIENUMDEVICEOBJECTSCALLBACK _object_enum DIDFT_ALLsortrXs r9rHz DirectInputDevice._init_controlsXsY    !EdFW!X!XZ^`f`pqqq 0011111r;czt|j}|r|j|tjSrV)r:contentsrhappendrDIENUM_CONTINUE)rQr3argr8s r9rkzDirectInputDevice._object_enum]s9!/":;;  * M  ) ) )%%r;c|jsdStjt|jz}d}t ||jD]\}}||_|j|_|dz }tj}tj ||_ tj tj|_ d|_ ||_t||_tjtj|tj|_|j|tj}tj ||j_ tj |j|j_d|j_tj|j_dtj tjz|_|jtjtj|jdS)Nrr@) rhrDIOBJECTDATAFORMATlenzipdwOfsr2r$ DIDATAFORMATctypessizeofdwSize dwObjSizedwFlags dwDataSize dwNumObjscastpointerLPDIOBJECTDATAFORMATrgodfrG SetDataFormat DIPROPDWORDdiph dwHeaderSizedwObj DIPH_DEVICEdwHowdwData SetPropertyDIPROP_BUFFERSIZEbyref)rQobject_formatsoffset object_formatr8fmtprops r9rIzDirectInputDevice._set_formatcs}  F 3c$-6H6HHKK&).$-&H&H   "M7"(M #*=M aKFF!##]3''  f&?@@  N++ K~ > >@[\\  ""3'''!##!=.. !'ty!9!9   , 6=)<===    !96< ;R;RSSSSSr;NFc|jsdS|%tjj}tjjD]}t j}|r|t jz}n|t j z}tj dddd|_ |j |j tjj|j |j|j |j||j dSNF)rhpygletgl_shadow_windowappwindowsrDISCL_BACKGROUNDDISCL_EXCLUSIVEDISCL_NONEXCLUSIVEr CreateEventW _wait_objectrGSetEventNotificationplatform_event_loopadd_wait_object_dispatch_eventsSetCooperativeLevel_hwndAcquire)rQwindow exclusiveflagss r9openzDirectInputDevice.opens}  F >Y-F *,  '  / V+ +EE V. .E%24tLL ))$*;<<< &66t7H$J_``` ((u=== r;c|jsdStjj|j|j|jdtj |jdSrV) rhrrrremove_wait_objectrrG Unacquirerr CloseHandlerXs r9closezDirectInputDevice.closesq}  F &99$:KLLL     ))$///d/00000r;c|jSrV)rhrXs r9 get_controlszDirectInputDevice.get_controlss }r;c|jsdStjdz}tjt |} |jtj tjtj tj |tj tj |dn#t$rYdSwxYw|d|jD]#}|jdz}|j|j|_$dS)Nrtrr)rhrDIDEVICEOBJECTDATAr DWORDrvrG GetDeviceDatarzr{rrLPDIDEVICEOBJECTDATArOSErrorvaluerxr)rQeventsn_eventseventindexs r9rz"DirectInputDevice._dispatch_eventss }  F+b033;s6{{++  L & &v}V5N'O'O'-{6>&3I3I393N(P(P'-|H'='='(  * * * *     FF OX^O, 6 6EK1$E).DM% & & 6 6sA:B== C  C ct|jjjd}|j|krG|j|jjkr2|j|jjdzkr|j |jjdzkrdSdS)NrAr?r@TF) rLrorMrNrOrKrJr2rErF)rQguid_idrPs r9matcheszDirectInputDevice.matchessu1=CUKK  G + +  8 GGG o6@4GGG !9!Cf!LLL4ur;r)__name__ __module__ __qualname__rDrYrbrHrkrIrrrrr __classcell__)rTs@r9r=r=@s / / / / /::: 222 &&& TTT80 1 1 1666&       r;r=ctj}tjd}tj|tjtjtj|d|SrV) r IDirectInput8rGetModuleHandleWDirectInput8CreateDIRECTINPUT_VERSIONIID_IDirectInput8Wrzr) _i_dinput module_handles r9_init_directinputrs[$&&I.t44M m$8$7$l955" $$$ r;iUMioi0cHeZdZdZdZd dZdZdZdZd d Z d Z d Z dS)DIDeviceManagercd|_d|_d|_g|_|dr|dSdS)NFT) skip_warning) registeredr _devnotifydevicesregister_device_eventsset_current_devicesrXs r9rDzDIDeviceManager.__init__sY 02  & &D & 9 9 '  $ $ & & & & & ' 'r;cB|\}}||_dS)aSets all currently discovered devices in the devices list. Be careful when using this, as this creates new device objects. Should only be called on initialization of the manager and if for some reason the window connection event isn't registered. N) _get_devicesr)rQ new_devices_s r9rz#DIDeviceManager.set_current_devicess$ **,, Q" r;FNc|js|s'tjj}|stjjD]}||_|jt}tj ||_ t|_ t|_tj|jjtj|t&|_|j|jjt.<d|_|j|dS|st3jddS)zRegister the first OS Window to receive events of disconnect and connection of devices. Returns True if events were successfully registered. NTzQDirectInput Device Manager requires a window to receive device connection events.F)rrrrrrrrrzr{ dbcc_sizer dbcc_devicetypeGUID_DEVINTERFACE_HIDdbcc_classguidrRegisterDeviceNotificationWrrr r_event_devicechange_event_handlersr push_handlerswarningswarn)rQrrdbis r9rz&DIDeviceManager.register_device_eventss w 1"(*"4 DK{&355 & c 2 2 &@#%:"#*"EdkFWY_YefiYjYjmH#I#I?C?W +O<"& ))$///t#wM"uvvvur;ct|jjvr|jjt=tj|jd|_d|_dSr)rrrrUnregisterDeviceNotificationrrrXs r9_unregister_device_eventsz)DIDeviceManager._unregister_device_eventssD dk9 9 9 +O<,T_===r;c|jr|ddl}t|jjdkr4|jjD]'}||jkr|d|dS(d|_dS)NrT)rr)rr pyglet.apprvrrrr)rQrexisting_windows r9on_closezDIDeviceManager.on_closes ? -  * * , , , vz! " "a ' '#):#5  "dk11//T//ZZZFF2 r;c@|jr|dSdSrV)rrrXs r9rYzDIDeviceManager.__del__s. ? -  * * , , , , , - -r;c@t|jggtjds" ddlm}|n#t $rYnwxYwfd}ttj tj |dtj fS)zaEnumerate all the devices on the system. Returns two values: new devices, missing deviceswin32_disable_xinputrget_xinput_guidsct|jjjd}|vr tjSt D];}|||r#|tjcS._device_enum/s_5AGOOG/))--,-- 2 2;;w882$++C000!11112/11F  " "?#;#H&,W]J^J^`d e e e&w8PQQF    ' ' ') )r;N) rrroptionspyglet.input.win32.xinputr ImportErrorr EnumDevicesrDI8DEVCLASS_ALLLPDIENUMDEVICESCALLBACKDIEDFL_ATTACHEDONLY)rQrRrrrrrs ` @@@r9rzDIDeviceManager._get_devices!s  -- ~45  FFFFFF"2"2"4"4     * * * * * * * *& f4$<\JJ"$8 : : :---sA A Ac|\}}|r5|j||D]}|d||r5|D]4}|j||d|3dSdS)N on_connect on_disconnect)rrextenddispatch_eventr)rQrmissing_devicesrSs r9_recheck_devicesz DIDeviceManager._recheck_devicesHs'+'8'8':':$ _  : L   , , ,% : :##L&9999  =) = = ##F+++##OV<<<< = = = =r;c|dkrdS|tks |tkrjtj|tjt }|jjtkr'tj j |ddSdSdS)Nrr) rrrzrPOINTERrrodbch_devicetyper rrr post_event)rQmsgwParamlParamhdr_ptrs r9rz#DIDeviceManager._event_devicechangeTs Q;; F & & &&4L*L*Lk&&.9J*K*KLLG/3MMM .99$@RSSSSSNM+M*Lr;)FNrV) rrrrDrrrrrYrrrr;r9rrs'''###B   ---%.%.%.%.N = = =TTTTTr;rr r rcBeZdZddZdedeejfdZdZ dS)DIControllerManagerNc|_i_tjD]}|tjfd}tjfd}dS)Nc|jvrH|r5tjjdj|dSdSdS)Nr ) _controllers_add_controllerrrrr) di_devicerQs r9r z0DIControllerManager.__init__..on_connectosi 111'' 22pJ2==dLRVRcdmRnooooo21ppr;c|jvr=j|}j|=tjjd|dSdS)Nr )rrrrr)r! _controllerrQs r9r z3DIControllerManager.__init__..on_disconnectusUD---"/ : %i0 .99$Q\]]]]].-r;)_displayr _di_managerrr r)rQrRrSr r s` r9rDzDIControllerManager.__init__hs FH!) ) )F   ( ( ( (   p p p p   p   ^ ^ ^ ^   ^ ^ ^r;rSreturnc@t|}|r ||j|<|SdSrV)_create_controllerr)rQrS controllers r9r z#DIControllerManager._add_controller|s/'//  (2D f % tr;ctjs2ttt |jSrV)r%rrrrrvaluesrXs r9get_controllersz#DIControllerManager.get_controllerssN% .  . . 0 0 0  + + - - -D%,,..///r;rV) rrrrDr=rr Controllerr r,rr;r9rrfse^^^^(&7HT_._device_enums1=CUKK o % %) )+--7DflSYFZFZ\`aaa)'6?;STTUUU%%r;) rrrrrrrrrrr r )rRrrr0rs` @@r9 get_devicesr1sHO >0 1  B B B B B B..00OO    D  & & & & & & &&0 8FF 4666 Os: AAc(t|}|jtjtjtjfvrF|tj||Stj d|d|ddSdS)Nz Warning: z (GUID: zG) has no controller mappings. Update the mappings in the Controller DB.) rrbr2rDI8DEVTYPE_JOYSTICKDI8DEVTYPE_1STPERSONDI8DEVTYPE_GAMEPADr r-rr)rSmappings r9r(r(s&//++,,G |2F4OQWQjkkk  ?6733 3 Mcfccfoo6G6Gccc d d d d d lkr;c|jtjtjtjtjfvrt j|SdSrV)r2rr3r4r5DI8DEVTYPE_SUPPLEMENTALr Joystick)rSs r9_create_joystickr:sI |2316888}V$$$ 88r;ctjs2ttddtjDDS)Ncg|]}||SrVr).0joysticks r9 z!get_joysticks..s' % % %# ###r;c,g|]}t|Sr)r:r=rSs r9r?z!get_joysticks..s! H H H& f % % H H Hr;r%rrrrrRs r9 get_joysticksrDsc  !***,,,''))) % % H HK4G H H H % % %%r;ctjs2ttddtjDDS)Ncg|]}||SrVr)r=r)s r9r?z#get_controllers..s' ' ' ':% %%%r;c,g|]}t|Sr)r(rAs r9r?z#get_controllers..s! J J JF  ' ' J J Jr;rBrCs r9r,r,sc  !***,,,''))) ' ' J Jk6I J J J ' ' ''r;rV)1rzrtypingrrrpyglet.libs.win32.constantsrrrr r pyglet.eventr r pyglet.inputr pyglet.libsr pyglet.libs.win32rrrrrrpyglet.input.controllerrpyglet.input.baserr'r+r.r:Devicer=rrGUIDrrregister_event_typer%rr1r(r:rDr,rr;r9rSs ''''''''''<<<<<<<<<<<<<<)((((( dddddddddddddd''''''////////////         .AAAAA AAAH         VVT4tUY[_aegkllGTGTGTGTGToGTGTGTT##L111##O444##$6777o #0#0#0#0#0+#0#0#0L>ddd%%%%%%%''''''r;