3j*dZddlmZddlmZddlmZddlZddlm Z ddl m Z ddlm Z dd l mZGd d e ZGd d ZGddej$ZGddej$Zedk(rddlZej,eyy)a Objects for realtime playback of Music21 Streams as MIDI. From an idea of Joe "Codeswell": https://joecodeswell.wordpress.com/2012/06/13/how-to-produce-python-controlled-audio-output-from-music-made-with-music21 https://stackoverflow.com/questions/10983462/how-can-i-produce-real-time-audio-output-from-music-made-with-music21 Requires pygame 2 (or 1.9) install with: pip3 install pygame ) annotations) find_spec)BytesION)defaults)Music21Exception)stream) translatec eZdZy)StreamPlayerExceptionN__name__ __module__ __qualname__B/DATA/.local/lib/python3.12/site-packages/music21/midi/realtime.pyr r "rr ceZdZdZdZ d d dZ deddddZd Z dedddd Z d Z y) StreamPlayera) Create a player for a stream that plays its midi version in realtime using pygame. Set up a detuned piano (where each key has a random but consistent detuning from 30 cents flat to sharp) and play a Bach Chorale on it in real time. >>> import random >>> keyDetune = [] >>> for i in range(127): ... keyDetune.append(random.randint(-30, 30)) >>> #_DOCS_SHOW b = corpus.parse('bwv66.6') >>> #_DOCS_SHOW for n in b.flatten().notes: >>> class PitchMock: midi = 20 #_DOCS_HIDE >>> class Mock: pitch = PitchMock() #_DOCS_HIDE >>> #_DOCS_HIDE -- should not play back in doctests, see TestExternal >>> n = Mock() #_DOCS_HIDE >>> for i in [1]: #_DOCS_HIDE ... n.pitch.microtone = keyDetune[n.pitch.midi] >>> #_DOCS_SHOW sp = midi.realtime.StreamPlayer(b) >>> #_DOCS_SHOW sp.play() The stream is stored (unaltered) in `StreamPlayer.streamIn`, and can be changed any time the midi file is not playing. A number of mixer controls can be passed in with keywords: * mixerFreq (default 44100 -- CD quality) * mixerBitSize (default -16 (=unsigned 16bit) -- really, are you going to do 24bit audio with Python?? :-) ) * mixerChannels (default 2 = stereo) * mixerBuffer (default 1024 = number of samples) Fc ddl}||_|jdus|r|jj ||||||_y#t$r tdwxYw)Nrz,StreamPlayer requires pygame. Install firstF)pygame ImportErrorr mixerInitializedmixerinitstreamIn)selfr reinitMixer mixerFreq mixerBitSize mixerChannels mixerBufferrs r__init__zStreamPlayer.__init__Ksb X  DK  E )[ LL  i}k R   X'(VW W Xs AANinfT)playForMillisecondsblockedc V|j}|j||||||||y)a busyFunction is a function that is called with busyArgs when the music is busy every busyWaitMilliseconds. endFunction is a function that is called with endArgs when the music finishes playing. playForMilliseconds is the amount of time in milliseconds after which the playback will be automatically stopped. If blocked is False, the method will finish before ending the stream, allowing you to completely control whether to stop it. Ignore every other arguments ) busyFunctionbusyArgs endFunctionendArgsbusyWaitMillisecondsr%r&N)getStringOrBytesIOFileplayStringIOFile) rr(r)r*r+r,r%r&streamStringIOFiles rplayzStreamPlayer.play_s>*"88: 0+7'/*5&-3G2E&-  /rcvtj|j}|j}t |SN) midiTranslatestreamToMidiFilerwritestrr)rstreamMidiFilestreamMidiWrittens rr-z#StreamPlayer.getStringOrBytesIOFile~s1&77 F*335())rc|jjj} |jjjj ||jjjj|sytd|z } |jjj} |jjjjr||||jjj| z |kDr/|jjjjn@| j| |jjjjr| ||yy#|jj $r} td|d| | d} ~ wwxYw)a' busyFunction is a function that is called with busyArgs when the music is busy every busyWaitMilliseconds. endFunction is a function that is called with endArgs when the music finishes playing. playForMilliseconds is the amount of time in milliseconds after which the playback will be automatically stopped. If blocked is False, the method will finish before ending the stream, allowing you to completely control whether to stop it. Ignore every other arguments but for stringIOFile zCould not play music file z because: N)rtimeClockrmusicloaderrorr r0int get_ticksget_busystoptick) r stringIOFiler(r)r*r+r,r%r& pygameClockpge framerate start_times rr.zStreamPlayer.playStringIOFilesr kk&&,,.   KK   # # ( ( 6 $$& 334 [[%%//1 kk%%..0'X&{{))+j8;NN !!'',,.   Y ' kk%%..0  "   ##{{   ',\N*SEJ  s/FG.GGc`|jjjjyr2)rrr<rB)rs rrBzStreamPlayer.stops $$&r)FiDii) rz stream.Streamrboolrr?r r?r!r?r"r?)NNNN2) r rr__doc__rr#floatr0r-r.rBrrrrr&s!D "!!! !  !  !!*"$ /"'u/>* JNNP%!.35\4%!N'rrc eZdZy)TestNr rrrrPrPrrrPcneZdZedZedZndZejeddZdZ dZ d Z y) TestExternalrNTFzpygame is not installedcdddlm}ddl}|jd}g}t dD]#}|j |j dd%|jjD]*}||jj|j_ ,t|}|jy)Nrcorpusbwv66.6)music21rUrandomparserangeappendrandintrecursenotespitchmidi microtonerr0)rrUr[b keyDetuneinsps rtestBachDetunezTestExternal.testBachDetunes" LL # sA   V^^C4 5""A )!'',, 7AGG # !_  rcddlm}ddl}d}Gdd}|}d|_|j d}g}t d D]#}|j |jd d %|jjD]*} || jj| j_ ,t|} | j||gd y) zN tests to see if the busyCallback function is called properly rrTNc~|d}|xj|jz c_td|jdy)Nrz hi! waited z milliseconds)times updateTimeprint)timeListtimeCounter_inners r busyCounterz4TestExternal.x_testBusyCallback..busyCounters= (   # #'8'C'C C # K 1 7 78 F GrceZdZdZy)-TestExternal.x_testBusyCallback..MockrN)r rrrmrrrMockrtsErruz bach/bwv66.6rWrXrYr(r)r,)rZrUr[rnr\r]r^r_r`rarbrcrdrr0) rrUr[rrru timeCounterrerfrgrhris rx_testBusyCallbackzTestExternal.x_testBusyCallbacks # H   f !$  LL ( sA   V^^C4 5""A )!'',, 7AGG # !_ [K=WZ[rchddlm}dt_|j d}g}t |j djtj}t|D]"}|j|j|$t|}|D]}||_|jy)NrrTrV)rZrUr ticksAtStartr\lenpartsgetElementsByClassrMeasurer]r^measurerrr0)rrUremeasures maxMeasurergrirs rx_testPlayOneMeasureAtATimez(TestExternal.x_testPlayOneMeasureAtATimes" ! LL #66v~~FG z"A OOAIIaL )# !_G!BK GGI rc4ddlmddlfdfd}Gdd}|}}t|}|j |_|j dkDr9d|_|j|j |||gd |j dkDr8yy) zf doesn't work -- no matter what there's always at least a small lag, even with queues r)noteNctj}tdD]:}j}j dd|_|j |<j}|j ||S)N0H)rStreamr]Noter_psr^)srgrhlastNrr[s rgetRandomStreamz8TestExternal.x_testPlayRealTime..getRandomStreamsa A1XIIK~~b"- IIKE HHUOHrc|d}|d}|jjjj}|dcxkr|jkrnn|xj dzc_|j dkDrb|_|j|_|jjjj|j||_yy||_y)Nrrv) rrr<get_poslastPosrmrr- storedIOFilequeue)rprx streamPlayer currentPosrs r restoreListz4TestExternal.x_testPlayRealTime..restoreList s"1+K#A;L%,,2288@@BJC6;#6#66!!Q&!$$q(,;,=L)/;/R/R/TK, ''--3399+:R:RS*4K' )'1 #rceZdZdZdZdZy)3TestExternal.x_testPlayRealTime..TimePlayerFr9N)r rrreadyrmrrrr TimePlayerrsEEGrrFrYrw) rZrr[rr-rrmrr.) rrrrxrerirrr[s @@@rx_testPlayRealTimezTestExternal.x_testPlayRealTimes !  1   !l   !_#%#<#<#>  !# %K     8 8-8*5r):57  9!#r) r rrrloaderpygame_installedunittest skipUnlessrjryrrrrrrRrRsR x F  X)+DE F ,\6 29rrR__main__)rM __future__rimportlib.utilriorrrZrmusic21.exceptions21rr music21.midir r3r rTestCaserPrRr mainTestrrrrs #$13 , E'E'P 8   y98$$y9x zG\"r