^ jb|ddlZddlZddlmZmZddlmZmZmZm Z m Z m Z ddl m Z mZddlZddlmZddlmZddlmZddlmZdd lmZd d lmZed Zd ZdZGddZ Gddej!Z"Gddej!Z#GddZ$GddZ%GddZ&dS)N) defaultdict namedtuple)POINTERbyrefc_charc_floatcastpointer)DWORDFLOAT)com)get_audio_device_manager) DeviceFlow)MediaException) debug_print) lib_xaudio2 debug_mediactj}|j|_t |jt t|_|S)zCreates a XAUDIO2_BUFFER to be used with a source voice. Audio data cannot be purged until the source voice has played it; doing so will cause glitches. ) libXAUDIO2_BUFFERlength AudioBytesr r rr pAudioData) audio_databuffs a/home/agentuser/manim-venv/lib/python3.11/site-packages/pyglet/media/drivers/xaudio2/interface.pycreate_xa2_bufferrs;    D 'DO:-wv??DO Kc8|jdks |jdvrtd|tj}tj|_|j|_|j|_ |j|_ |j |jzdz|_ |j |j z|_ |S)N)zUnsupported audio format: r") channels sample_sizerr WAVEFORMATEXWAVE_FORMAT_PCM wFormatTag nChannels sample_ratenSamplesPerSecwBitsPerSample nBlockAlignnAvgBytesPerSec) audio_formatwfxs rcreate_xa2_waveformatr1sq  L$C Jrc2eZdZdZd dZdZdZdZdZdS) _VoiceResetterz'Manage a voice during its reset period.returnNc>||_||_||_||_dSN)drivervoice voice_keyremaining_data)selfr7r8r9r:s r__init__z_VoiceResetter.__init__/s%  ",rc|jjdkr1|j|jj_|jdSt j|j ddSNr) r8buffers_queuedflush_on_buffer_end _callback on_buffer_endflushpygletclock schedule_once_finishr;s rrunz_VoiceResetter.run5s\ : $ ) )151IDJ . J        L & &t|Q 7 7 7 7 7rc|jjdkr@|jtj|jddSdSr>)r8r?r:clearrDrErFrGr;_s rr@z"_VoiceResetter.flush_on_buffer_end<sO : $ ) )   % % ' ' ' L & &t|Q 7 7 7 7 7 * )rcd|jj_|jj|j_|j|j|jdSr6)r8rArBsamples_playedsamples_played_at_last_recycler7_return_reset_voicer9rLs rrGz_VoiceResetter._finishCs@-1 *48J4M 1 '' DNCCCCCrctj|jd|_d|_|jdSr6)rDrE unschedulerGr7r8r:rKrHs rdestroyz_VoiceResetter.destroyHsC  ---   !!#####r)r4N) __name__ __module__ __qualname____doc__r<rIr@rGrTrrr3r3,sm11---- 888888DDD $$$$$rr3c@eZdZejgZfdZdZdZdZ xZ S)XA2EngineCallbackcVt||_dSr6)superr<_lock)r;lock __class__s rr<zXA2EngineCallback.__init__Rs$  rc8|jdSr6)r^acquirerHs rOnProcessingPassStartz'XA2EngineCallback.OnProcessingPassStartV rc8|jdSr6)r^releaserHs rOnProcessingPassEndz%XA2EngineCallback.OnProcessingPassEndYrdrc|jr|jtd|)NzCritical Error:)r^lockedrf Exception)r;hresults rOnCriticalErrorz!XA2EngineCallback.OnCriticalError\s? :     ! J   )7333r) rUrVrWrIXAudio2EngineCallback _interfaces_r<rcrgrl __classcell__r`s@rr[r[Osp./L4444444rr[c>eZdZdZejgZfdZdZdZ xZ S)XAudio2VoiceCallbackaQCallback class used to trigger when buffers or streams end. WARNING: Whenever a callback is running, XAudio2 cannot generate audio. Make sure these functions run as fast as possible and do not block/delay more than a few milliseconds. MS Recommendation: At a minimum, callback functions must not do the following: - Access the hard disk or other permanent storage - Make expensive or blocking API calls - Synchronize with other parts of client code - Require significant CPU usage cVtd|_dSr6)r]r<rB)r;r`s rr<zXAudio2VoiceCallback.__init__rs' !rc0||dSr6)rB)r;pBufferContexts r OnBufferEndz XAudio2VoiceCallback.OnBufferEndvs >*****rc&td|)Nz&Error occurred during audio playback: )rj)r;rurks r OnVoiceErrorz!XAudio2VoiceCallback.OnVoiceErrorysJJJKKKr) rUrVrWrXrIXAudio2VoiceCallbackrnr<rvrxrorps@rrrrrest  -.L"""""+++LLLLLLLrrrceZdZdZejZejZdZ dZ dZ dZ de fdZddZed Zd Zd Zd Zd ZdZedZejdZdZdZdZdZdZdZdZdZ dZ!dZ"dS) XAudio2DriverTg@cXtdsJd|_d|_d|_t j|_t|j|_g|_ tt|_ i|_ i|_g|_||jrt%}|rrtdsJ|||t+dt,j|jddSdSdS)a.Creates an XAudio2 master voice and sets up 3D audio if specified. This attaches to the default audio device and will create a virtual audio endpoint that changes with the system. It will not recover if a critical error is encountered such as no more audio devices are present. zConstructing XAudio2DriverNFzAudio device instance found.z5No default audio device found, can not create driver.g?)_debug _listener_xaudio2_dead threadingLockr_r[_engine_callback_emitting_voicesrlist _voice_pool_in_use_resetting_voices_players _create_xa2restart_on_errorr push_handlersget_default_output ImportErrorrDrEschedule_interval_soft _check_state)r; audio_devicess rr<zXAudio2Driver.__init__s4 233333  N$$ 1$) < < "&t,, !#    L466M L<=====++D111 3355=%&]^^^ 33D4EsKKKKK L L L Lrc|jdur|jr|dSdS|jsN||jD]}|d|jdSdS)zeHack/workaround, you cannot shutdown/create XA2 within a COM callback, set a schedule to check state.Ton_driver_resetN)rr_shutdown_xaudio2rrdispatch_eventrK)r;dtplayers rrzXAudio2Driver._check_states :  } )&&((((( ) )= &  """"m==F))*;<<<< ##%%%%%  & &rflowc|tjkr> |tdsJd|_dS|jrtdsJd|_dSdSdS)Nz8Error: Default audio device was removed or went missing.Tz8Warning: Default audio device added after going missing.F)rOUTPUTr}r)r;devicers ron_default_changedz XAudio2Driver.on_default_changedst :$ $ $ t~XYYYYY! :'!"\]]]]]!&DJJJ % $ ''rNc tj|_ tjt |jd|jn#t $rtdwxYwtrqtj }d|_ tj tj z|_ tj |_|jt |d|j|jtj|_tj|_|jt |jtjtjd|d|j|jt |jd|_d|_|jr|dSdS)Nrz(XAudio2 driver could not be initialized.T)rIXAudio2r XAudio2Creater processorOSErrorrr}XAUDIO2_DEBUG_CONFIGURATION LogThreadIDXAUDIO2_LOG_ERRORSXAUDIO2_LOG_WARNINGS TraceMask BreakMaskSetDebugConfigurationRegisterForCallbacksrXAUDIO2_VOICE_DETAILS_mvoice_detailsIXAudio2MasteringVoice _master_voiceCreateMasteringVoiceXAUDIO2_DEFAULT_CHANNELSXAUDIO2_DEFAULT_SAMPLERATEcategoryGetVoiceDetails _x3d_handle _dsp_settingsallow_3d enable_3d)r; device_iddebugs rrzXAudio2Driver._create_xa2s   J  eDM22At~ F F F F J J JHII I J  D355E $E !4s7OOEO!6EO M / /e d C C C **4+@AAA"8:: 799 **51C+D+D+.+G+.+I+,it} N N N **51E+F+FGGG! =  NN       s -AA"c4|jSr6)rkeysrHs r active_voiceszXAudio2Driver.active_voicess|  """rc|jD]/}|D]}||0|jD]-\}}||.|j|jD]}||jdS)z"Destroy and clear all voice pools.N)rvaluesrTrKritemsrr)r;list_r8resetters r_destroy_voiceszXAudio2Driver._destroy_voicess%,,..  E    KKMMMM#5;;==  OE8 MMOOO        $$&&&'  E MMOOOO rc|||j|jD]}|d|jdS)zAAttach XA2 with a specific device rather than the virtual device.rN)rridrrrK)r;rrs r set_devicezXAudio2Driver.set_deviceso     ###m 5 5F  ! !"3 4 4 4 4 rc|jD]5}||j|j6|dS)zAStops and destroys all active voices, then destroys XA2 instance.N)rron_driver_destroyrappendr_delete_driver)r;rs rrzXAudio2Driver._shutdown_xaudio2sdl))++ 0 0F  $ $ & & & M  / / / / rch|jrtdsJ|jr$tj|j||j|j |j |j d|_dSdS)NzXAudio2Driver: Deleting) rr}rrDrErS_calculate_3d_sourcesrUnregisterForCallbacksr StopEngineReleaserHs rrzXAudio2Driver._delete_drivers = !344 4 44} D ''(BCCC  " " " M 0 01F G G G M $ $ & & & M ! ! # # # DMMM ! !rct}|jt|t j|_t j|jtj |jt|j j z}t j |_d|j_|j j |j_||j_t$j|jddS)z`Initializes the prerequisites for 3D positional audio and initializes with default DSP settings.rg?N)r rGetChannelMaskrrX3DAUDIO_HANDLErX3DAudioInitializevalueX3DAUDIO_SPEED_OF_SOUNDr r InputChannelsX3DAUDIO_DSP_SETTINGSrSrcChannelCountDstChannelCountpMatrixCoefficientsrDrErr)r; channel_maskmatrixs rrzXAudio2Driver.enable_3d)sww  ))% *=*=>>>.00 |133NPTP`aaa$.<<?? 688-.*-1-A-O*17. ++D,FQQQQQrczt}|jt||jSr6)rr GetVolumerrr;vols rvolumezXAudio2Driver.volume9s0ii $$U3ZZ000yrc<|j|ddS)z'Sets global volume of the master voice.rN)r SetVolumer;rs rrzXAudio2Driver.volume?s# $$UA.....rcz|jD]}||d|jddS)ziWe calculate the 3d emitters and sources every 15 fps, committing everything after deferring all changes.rN)r_apply3dr CommitChanges)r;r source_voices rrz#XAudio2Driver._calculate_3d_sourcesDsG 1 + +L MM, * * * * ##A&&&&&rcx|j2||d|jddSdS)zJApply and immediately commit positional audio effects for the given voice.Nr!)rrrr)r;rs rapply3dzXAudio2Driver.apply3dKsC   ' MM, * * * M ' ' * * * * * ( 'rc4tj|j|jj|jtj|j|j |j d|j j |jj ||j|jj|dS)aCalculates and sets output matrix and frequency ratio on the voice based on the listener and the voice's emitter. Commit determines the operation set, whether the settings are applied immediately (0) or to be committed together at a later time. rN)rX3DAudioCalculaterr~listener_emitterdefault_dsp_calculationr_voiceSetOutputMatrixrrrrSetFrequencyRatio DopplerFactor)r;rcommits rrzXAudio2Driver._apply3dQs    N #  !  '       ++D,>,-,0,@,N,0,>,R,2  4 4 4 --d.@.NPVWWWWWrcv|tj|jdSr6)rrDrErSrrHs rdeletezXAudio2Driver.deletees3   122222rcztj}|jt ||S)zTRetrieve some basic XAudio2 performance data such as memory usage and source counts.)rXAUDIO2_PERFORMANCE_DATArGetPerformanceDatar)r;pfs rget_performancezXAudio2Driver.get_performanceis1  ) + + ((r333 rcZ|j Jdt||_|jS)Nz!You can only create one listener.)r~XAudio2ListenerrHs rcreate_listenerzXAudio2Driver.create_listeneros0~%%'J%%%(..~rc"|jr|j||j|t d|dsJ|j|jf}t||||}||j |<| dS)abReset a voice and eventually return it to the pool. The voice must be stopped. `remaining_data` should contain the data this voice's remaining buffers point to. It will be `.clear()`ed shortly after as soon as the flush initiated by the driver completes in order to not have theoretical dangling pointers. zXA2AudioDriver: Resetting z...N) is_emitterrremoverpopr} channel_countr%r3rrI)r;r8r:r9rs r return_voicezXAudio2Driver.return_voicets   0  ! ( ( / / / =5===>>>>>(%*;< !$y.II(0u% rc|j||j||t d|dsJdS)NzXA2AudioDriver: z back in pool)rrrTrrr})r;r8r9s rrQz!XAudio2Driver._return_reset_voicesg ""5))11333 #**5111====>>>>>>>rc|j|jf}|j|sI||}|j|||n|j|}|jdksJ||j|j |j r|j |||j |<|S)zGet a source voice from the pool. Source voice creation can be slow to create/destroy. So pooling is recommended. We pool based on audio channels. A source voice handles all of the audio playing and state for a single source.r) r$r%r_create_new_voicerrr?acquiredrBr*rrr)r;r/rr9r8s rget_source_voicezXAudio2Driver.get_source_voices "*L,DE  * 6**<88E  Y ' . .t/E/El/S/S T T T T$Y/3355E#q(((( v+\-EFFF   0  ! ( ( / / /$ U rc tj}t|}t}|jt |t |d|j|ddt|||j |j S)zDHas the driver create a new source voice for the given audio format.rN) rIXAudio2SourceVoicer1rrrCreateSourceVoicermax_frequency_ratioXA2SourceVoicer$r%)r;r/r8 wfx_formatcallbacks rrzXAudio2Driver._create_new_voices'))*<88 ')) ''e (-j(9(9()(,(@(0(,(,  . . .eX|/DlF^___rr6)#rUrVrWrrXAUDIO2_DEFAULT_PROCESSORrAudioCategory_GameEffectsrrr r<rrrrpropertyrrrrrrrsetterrrrrrrrrQrrrYrrr{r{}sH-I,H)L)L)LV & & & 'z ' ' ' '    D##X#    ! ! !RRR X  ]//]/'''+++ XXX(333  $??? 0`````rr{c,eZdZdZdZdZedZedZedZ e j dZ edZ ed Z e j d Z ed Z e j d Z ed Zej dZedZej dZedddgZedZdZedZej dZedZej dZdZdZdZdZdS)r cxtj|_||_||_||_||_d|_|dkrqtj|_ ||j _ d|j _ tj }d|_ t||j _d|j _dSd|_ dS)Nrrg?)rXAUDIO2_VOICE_STATE _voice_staterrArr%rPX3DAUDIO_EMITTERr ChannelCountCurveDistanceScaler X3DAUDIO_CONE InnerVolumer pCone pVolumeCurve)r;r8rrr%cones rr<zXA2SourceVoice.__init__s355 !*&/0+ A  022DM)6DM &03DM -$&&D #D #*$--DM )-DM & & & DMMMrcpd|_|j |jd|_d|_dS)zCompletely destroy the voice.N)rr DestroyVoicerArHs rrTzXA2SourceVoice.destroys7 ; " K $ $ & & &DKrcR||j_|j|dS)z\A voice has been acquired. Set the callback as well as its new sample rate. N)rArBrSetSourceSampleRate)r;on_buffer_end_cbr*s rrzXA2SourceVoice.acquireds*(8$ '' 44444rc|jt|jtj|jjS)z_Get the amount of buffers in the current voice. Adding flag for no samples played is 3x faster.)rGetStaterrrXAUDIO2_VOICE_NOSAMPLESPLAYED BuffersQueuedrHs rr?zXA2SourceVoice.buffers_queueds6 U4#455s7XYYY ..rct|jt|jd|jjS)z.Get the amount of samples played by the voice.r)rr%rr SamplesPlayedrHs rrOzXA2SourceVoice.samples_playeds3 U4#455q999 ..rczt}|jt||jSr6)rrrrrrs rrzXA2SourceVoice.volumes/ii eCjj)))yrc<|j|ddSr>)rrrs rrzXA2SourceVoice.volumes  eQ'''''rc|jduSr6)rrHs rrzXA2SourceVoice.is_emitters}D((rcx|jr2|jjj|jjj|jjjfSdS)NrrrrrPositionxyzrHs rpositionzXA2SourceVoice.positions6 ? =)+T]-C-Et}G]G__ _7rc|jr;|\}}}||jj_||jj_||jj_dSdSr6r/)r;r4r1r2r3s rr4zXA2SourceVoice.positionsN ? )GAq!'(DM " $'(DM " $'(DM " $ $ $  ) )rc,|jr |jjSdS)zCurve distance scaler that is used to scale normalized distance curves to user-defined world units, and/or to exaggerate their effect.r)rrrrHs r min_distancezXA2SourceVoice.min_distances ? =4 41rc|jr6|jj|kr(t|tj|j_dSdSdSr6)rrrminrFLT_MAXrs rr7zXA2SourceVoice.min_distancesK ? L}0E9947s{4K4K 111 L L99rczt}|jt||jS)z[The actual frequency ratio. If voice is 3d enabled, will be overwritten next apply3d cycle.)rrGetFrequencyRatiorrrs r frequencyzXA2SourceVoice.frequencys1  %%eEll333{rcV|j|krdS|j|ddSr>)r=rrrs rr=zXA2SourceVoice.frequency%s2 >U " " F %%eQ/////rcx|jr2|jjj|jjj|jjjfSdS)z%The orientation of the sound emitter.r.rr OrientFrontr1r2r3rHs rcone_orientationzXA2SourceVoice.cone_orientation,s8 ? =,. 0I0KT]MfMhh h7rc|jr;|\}}}||jj_||jj_||jj_dSdSr6r@r;rr1r2r3s rrBzXA2SourceVoice.cone_orientation4sN ? ,GAq!*+DM % '*+DM % '*+DM % ' ' '  , ,r _ConeAnglesinsideoutsidec|jr>||jjjj|jjjjS|ddS);The inside and outside angles of the sound projection cone.r)rrErrcontents InnerAngle OuterAnglerHs r cone_angleszXA2SourceVoice.cone_angles>sP ? *##DM$7$@$KT]M`MiMtuu u##Aq)) )rcp|jr.||jjj_||jjj_dSdS)rIN)rrrrJrKrL)r;rFrGs rset_cone_angleszXA2SourceVoice.set_cone_anglesFs< ? >6 >rc@|jr|jjjjSdS)z5The volume scaler of the sound beyond the outer cone.rrrrrJ OuterVolumerHs rcone_outside_volumez"XA2SourceVoice.cone_outside_volumeL$ ? =&/; ;1rcD|jr||jjj_dSdSr6rQrs rrSz"XA2SourceVoice.cone_outside_volumeT, ? =7)rStartrHs rplayzXA2SourceVoice.playks  !Qrc<|jdddSr>)rr\rHs rstopzXA2SourceVoice.stopns  ArcV|jt|ddSr6)rSubmitSourceBufferr)r; x2_buffers r submit_bufferzXA2SourceVoice.submit_bufferqs( &&uY'7'7>>>>>rN)rUrVrWr<rTrrr?rOrrrr4r7r=rBrrErMrOrSrYrCr`rbrfrYrrr r s!!!@555//X/ //X/ X  ]((]())X)X _))_)XLLL X 000 X,,,*]Xy,ABBK **X*>>> X== =X===)))    ?????rr ceZdZdZdZedZejdZedZejdZdS)rc4tj||_tj|_d|jj_d|jj_d|jj_ d|jj _d|jj _d|jj _ dS)Nrr) weakrefproxy xa2_driverrX3DAUDIO_LISTENERrrAr1r2r3 OrientTop)r;r7s rr<zXAudio2Listener.__init__vsy!-//-// '( !#&' !#&' !#$% !$% !$% !!!rcd|_dSr6)rrHs rrzXAudio2Listener.deletes  rcf|jjj|jjj|jjjfSr6rr0r1r2r3rHs rr4zXAudio2Listener.positions'}%')?)A4=CYC[[[rcx|\}}}||jj_||jj_||jj_dSr6rprDs rr4zXAudio2Listener.positions:1a#$  #$  #$    rc|jjj|jjj|jjj|jjj|jjj|jjjfSr6rrAr1r2r3rmrHs r orientationzXAudio2Listener.orientationsL}(*DM,E,GIbId}&($-*A*CT]E\E^_ _rc|\}}}}}}||jj_||jj_||jj_||jj_||jj_||jj_dSr6rs)r;rtfront_xfront_yfront_ztop_xtop_ytop_zs rrtzXAudio2Listener.orientationsh9D6'5%&- !#&- !#&- !#$) !$) !$) !!!rN) rUrVrWr<rrr4rrtrYrrrrus & & &\\X\_%%_% __X_ * * * * *rr)'rri collectionsrrctypesrrrrr r ctypes.wintypesr r rDpyglet.libs.win32r pyglet.media.devicesrpyglet.media.devices.baserpyglet.media.exceptionsr pyglet.utilrrrr}rr1r3 COMObjectr[rrr{r rrYrrrsA////////AAAAAAAAAAAAAAAA(((((((( !!!!!!999999000000222222###### ] # #    $ $ $ $ $ $ $ $F44444 444,LLLLL3=LLL0t`t`t`t`t`t`t`t`n ~?~?~?~?~?~?~?~?B,*,*,*,*,*,*,*,*,*,*r