
    rh
                   h   S r SSKJr  SSKJr  SSKrSSKrSSKrSSK	J
r
  SSK	Jr  SSKJr  SSKJr  SS	KJr  SS
K	Jr  SSK	Jr  \
R$                  " S5      r " S S\R(                  5      r " S S\R,                  5      r " S S5      r " S S\R,                  5      r " S S5      r " S S\R,                  5      r " S S\R,                  5      r " S S\R,                  5      r " S S\R,                  5      r " S S \R>                  5      r \/r!\"S!:X  a  SSK	r	\	RF                  " \ 5        gg)"a  
**N.B. in Dec. 2014 MuseData access was removed from music21 because the rights conflicted with
access computationally from music21.  This module is retained for anyone who has such access,
however it is completely untested now and errors cannot and will not be fixed.**

Objects and resources for processing MuseData.

MuseData conversion from a file or URL to a :class:`~music21.stream.Stream` is available through
the music21 converter module's :func:`~music21.converter.parse` function.

>>> #_DOCS_SHOW from music21 import *
>>> #_DOCS_SHOW abcScore = converter.parse('d:/data/musedata/myScore.stage2')

Low level MuseData conversion is facilitated by the objects in this module and
:func:`music21.musedata.translate.museDataToStreamScore`.
    )annotations)IterableN)environment)exceptions21)	base12_26)base40)	translate)common)prebasemusedatac                      \ rS rSrSrg)MuseDataException4    N)__name__
__module____qualname____firstlineno____static_attributes__r       S/home/james-whalen/.local/lib/python3.13/site-packages/music21/musedata/__init__.pyr   r   4   s    r   r   c                      \ rS rSrSr  S SS jjrSS jrS rS rS r	S	 r
S
 rS rS rS rSS jrS rS rS rSS jrS rS rS rS rSrg)MuseDataRecord9   zd
Object for extracting data from a Note or other related record, or a
single line of musedata data.
Nc                    Xl         X l        U R                  b&  U R                  R                  R                  U l        OS U l        0 U l        g N)srcparentstage_cacheselfr   r   s      r   __init__MuseDataRecord.__init__>   s=     ;;"++11DJDJ+-r   c                N    U R                   (       a  U R                   S   S:X  a  gg)z
Return a boolean if this record is a rest.

>>> mdr = musedata.MuseDataRecord('D4     1        s     d  ]]')
>>> mdr.isRest()
False
>>> mdr = musedata.MuseDataRecord('measure 1       A')
>>> mdr.isRest()
False
r   rTFr   r"   s    r   isRestMuseDataRecord.isRestO   s      88s*r   c                    U R                   S:X  a.  [        U R                  5      S:  a  U R                  S   S:X  a  gg[        U R                  5      S:  a  U R                  S   S:X  a  gg)z
Return a boolean if this record is tied.

>>> mdr = musedata.MuseDataRecord('D4     8-       h     d        -')
>>> mdr.isTied()
True


>>> mdr = musedata.MuseDataRecord('C4     1        s     u  [[')
>>> mdr.isTied()
False
      -TF   r   lenr   r(   s    r   isTiedMuseDataRecord.isTied^   sY     ::?488}q TXXa[C%7488}q TXXa[C%7r   c                N    U R                   (       a  U R                   S   S;   a  gg)Nr   ABCDEFGTFr'   r(   s    r   isNoteMuseDataRecord.isNotet   s    88y0r   c                    [        U R                  5      S:  a'  U R                  S   S:X  a  U R                  S   S;   a  gg)zz
Chords are specified as additional note records following a main chord tone.
The blank space defines this as chord tone.
r,   r    r5   TFr1   r   r(   s    r   isChordMuseDataRecord.isChordy   s8     txx=1!!3y8Pr   c                N    U R                   (       a  U R                   S   S;   a  gg)Nr   cgTFr'   r(   s    r   isCueOrGraceMuseDataRecord.isCueOrGrace   s    88t+r   c                ^    [        U R                  5      S:  a  U R                  SS S:X  a  gg)zE
>>> mdr = musedata.MuseDataRecord('back   4')
>>> mdr.isBack()
True
   r   backTFr:   r(   s    r   isBackMuseDataRecord.isBack   s+     txx=A$((1Q-6"9r   c                X   U R                  5       (       a!  U R                  SS R                  5       S   nOVU R                  5       (       d  U R	                  5       (       a!  U R                  SS R                  5       S   nO[        S5      eUS   /nSU;   a  UR                  S5        OGSU;   a  UR                  S5        O/SU;   a  UR                  S5        OS	U;   a  UR                  S
5        [        R                  " U5      u  p4UR                  U5        SR                  U5      $ )z

>>> mdr = musedata.MuseDataRecord('Ef4    1        s     d  ==')
>>> mdr.isNote()
True
>>> mdr._getPitchParameters()
'E-4'

>>> mdr = musedata.MuseDataRecord('F#4    1        s #   d  ==')
>>> mdr.isNote()
True
>>> mdr._getPitchParameters()
'F#4'
r   Nr,   z4cannot get pitch parameters from this kind of record#z##fr.   ffz-- )
r6   r   splitr?   r;   r   appendr
   getNumFromStrjoin)r"   datapStrnumbersjunks        r   _getPitchParameters"MuseDataRecord._getPitchParameters   s     ;;==88AB<%%'*D  DLLNN88AB<%%'*D#$Z[[Qy$;KKT\KKD[KKT\KK ,,T2Gwwt}r   c                   SSK Jn  [        U R                  5      S::  a  gU R                  S   nSnUS:X  a  UR	                  S5      nOUS:X  a  UR	                  S5      nOUS	:X  a  UR	                  S
5      nOwUS:X  a  UR	                  S5      nO_US:X  a  UR	                  S5      nOGUS:X  a  UR	                  S5      nO/US:X  a  UR	                  S5      nOUS:X  a  UR	                  S
5      nUb  SUl        SUl        U$ )a  
Return a music21 Accidental object for the representation.

>>> mdr = musedata.MuseDataRecord('Ef4    1        s     d  ==')
>>> mdr._getAccidentalObject()
>>> mdr = musedata.MuseDataRecord('F#4    1        s #   d  ==')
>>> mdr._getAccidentalObject()
<music21.pitch.Accidental sharp>
>>> mdr._getAccidentalObject().displayStatus == True
True
>>> mdr = musedata.MuseDataRecord('F#4    1        s ')
>>> mdr._getAccidentalObject() is None
True
r   pitch   NrG   sharpnnaturalrH   flatxzdouble-sharpX&zdouble-flatSFalwaysT)music21rW   r1   r   
AccidentaldisplayTypedisplayStatus)r"   rW   rO   accs       r   _getAccidentalObject#MuseDataRecord._getAccidentalObject   s     	"txx=Bxx|3;""7+CS[""9-CS[""6*CS["">2CS["">2CS[""=1CS[""7+CS[""6*C ? 'CO $C
r   c                j   SSK Jn  U R                  5       nUR                  U5      nU R                  S:X  a  U$ U R                  5       nUR                  b  SUR                  l        Ub  U R                  5       Ul        UR                  b&  U R                  5       (       a  SUR                  l        U$ )a  
Get the Pitch object defined by this record. This may be a note, chord, or grace pitch.

>>> mdr = musedata.MuseDataRecord('Ef4    1        s     d  ==')
>>> p = mdr.getPitchObject()
>>> p.nameWithOctave
'E-4'
>>> mdr = musedata.MuseDataRecord('F#4    1        s #   d  ==')
>>> p = mdr.getPitchObject()
>>> p.nameWithOctave
'F#4'
>>> p.accidental.displayStatus
True


Double sharps were giving octave problems.

>>> mdr = musedata.MuseDataRecord('F##5   2        e x   d')
>>> p = mdr.getPitchObject()
>>> p.nameWithOctave
'F##5'
r   rV   r,   FT)
rc   rW   rS   Pitchr   rh   
accidentalrf   _accidentalhasCautionaryAccidental)r"   rW   ppprg   s        r   getPitchObjectMuseDataRecord.getPitchObject   s    0 	"%%'KKO::?H ++-C ||' .3* $ 9 9 ;||'D,H,H,J,J-1* Hr   c                   U R                   S:X  a  [        U R                  SS 5      nO[        U R                  SS 5      nU R                  SS nUS:w  a/   [        U5      nUSU-  -  n[        SU R                  -   S	-   5        U R                  b(  U R                  R                  R                  5       nX%-  $ Ub  UnX%-  $ [        S5      e! [         a    [        SU R                  -   S
-   5      ef = f)a  
Gets the quarterLength of the note given the prevailing divisionsPerQuarterNote

Here there is one division:

>>> mdr = musedata.MuseDataRecord('Ef4    1        s     d  ==')
>>> mdr.getQuarterLength(4)
0.25
>>> mdr.getQuarterLength(8)
0.125

>>> mdr = musedata.MuseDataRecord('Ef4    6        s     d  ==')
>>> mdr.getQuarterLength(4)
1.5

>>> mdr = musedata.MuseDataRecord('back   4')
>>> mdr.getQuarterLength(4)
1.0
r,      r-   r/   rB   r9   d   zError in parsing: z>
   Column 5 must be blank. Parsing as a part of the divisionsz
   Column 5 must be blank.zMcannot access parent container of this record to obtain divisions per quarter)r   intr   print
ValueErrorr   r   getDivisionsPerQuarterNote)r"   divisionsPerQuarterNote	divisionsshouldBeBlankdivHundredsdpqs         r   getQuarterLengthMuseDataRecord.getQuarterLength  s   ( ::?DHHQqM*IDHHQqM*I1CV!-0S;..	*!YZ [ ;;"++$$??AC  %0)C  $ %H I I  V'(48836TTV VVs   .C &C5c                    U R                   S:X  a  g [        U R                  5      S:  a(  U R                  S   S:X  a  gU R                  S   S:X  a  gg)Nr,      .:   r   r0   r(   s    r   getDotsMuseDataRecord.getDotsK  sI    ::?488}r!88B<3&88B<3&r   c                    SnU R                   S:X  a  g[        U R                  5      S:  a  gU R                  SS nUR                  S5       Vs/ s H  o3R	                  5       PM     nnU$ s  snf )z
Return lyrics as a list.

>>> mdr = musedata.MuseDataRecord('D4     2        e     u                    con-')
>>> mdr.stage = 2
>>> mdr.getLyrics()
['con-']
Nr,   ,   +   |)r   r1   r   rK   strip)r"   rO   rawr]   s       r   	getLyricsMuseDataRecord.getLyricsa  se     ::? 488}r!((23-C'*yy~6~!GGI~D6 7s   A-c                p    U R                   S:X  a  gU R                  SS nUS:X  a  gUR                  5       $ )aA  
Return complete span of characters defining beams.

>>> mdr = musedata.MuseDataRecord('E2     1        s     u  =')
>>> mdr.getBeams()
'='
>>> mdr = musedata.MuseDataRecord('E2     1        s     u  ]')
>>> mdr.getBeams()
']'
>>> mdr = musedata.MuseDataRecord('E2     4        q     u')
>>> mdr.getBeams() is None
True

r,   N      rJ   )r   r   rstripr"   rO   s     r   getBeamsMuseDataRecord.getBeamsv  s7     ::?88Br?Drz;;= r   c                    U R                   S   $ ! [         a     Of = f[        U R                  5      S:  a  SnO/ nSnUS::  a]  U[        U R                  5      :  aD  UR	                  U R                  U   5        US-  nUS::  a  U[        U R                  5      :  a  MD  SR                  U5      R                  5       nXR                   S'   U$ )ar  
Return an articulation string or None

>>> mdr = musedata.MuseDataRecord('C4    12        e     u  [      .p')
>>> mdr._getAdditionalNotations()
'.p'

>>> mdr = musedata.MuseDataRecord('C4    12        e     u  =      .')
>>> mdr._getAdditionalNotations()
'.'

>>> mdr = musedata.MuseDataRecord('G4    24        q     u        (')
>>> mdr._getAdditionalNotations()
'('
_getAdditionalNotationsr   N*   r,   rJ   )r    KeyErrorr1   r   rL   rN   r   )r"   rO   	data_listis       r   r   &MuseDataRecord._getAdditionalNotations  s    "	;;899 		 txx=2D IAr'a#dhh-/  !-Q r'a#dhh-/ 779%++-D15-.s    
c                   SSK Jn  / nU R                  5       nUc  U$ U GH>  nUS:X  a!  UR                  UR	                  5       5        M+  US:X  a!  UR                  UR	                  5       5        MR  US:X  a!  UR                  UR                  5       5        My  US:X  a!  UR                  UR                  5       5        M  US:X  a!  UR                  UR                  5       5        M  US:X  a!  UR                  UR                  5       5        M  US	:X  a"  UR                  UR                  5       5        GM  US
:X  d  GM  UR                  UR                  5       5        GMA     U$ )au  
Return a list of 0 or more music21 Articulation objects

>>> mdr = musedata.MuseDataRecord('C4    12        e     u  [      .p')
>>> mdr.getArticulationObjects()
[<music21.articulations.Staccato>]

>>> mdr = musedata.MuseDataRecord('C4    12        e     u  [      .p>')
>>> mdr.getArticulationObjects()
[<music21.articulations.Staccato>, <music21.articulations.Accent>]

r   )articulationsAV>r   _=r   ,)rc   r   r   rL   StrongAccentAccentStaccatoTenutoDetachedLegatoSpiccato
BreathMark)r"   r   postrO   chars        r   getArticulationObjects%MuseDataRecord.getArticulationObjects  s    	*++-<KDs{M6689M6689M0023M2245M0023M88:;M2245M4467' ( r   c                   SSK Jn  / nU R                  5       nUc  U$ U H  nUS:X  a!  UR                  UR	                  5       5        M*  US:X  a!  UR                  UR	                  5       5        MQ  US:X  a!  UR                  UR                  5       5        Mx  US:X  a!  UR                  UR                  5       5        M  US:X  d  M  UR                  UR                  5       5        M     U$ )aI  
Return a list of 0 or more music21 Articulation objects

>>> mdr = musedata.MuseDataRecord('C4    12        e     u  [      t')
>>> mdr.getExpressionObjects()
[<music21.expressions.Trill>]

>>> mdr = musedata.MuseDataRecord('C4    12        e     u  [      .p>F')
>>> mdr.getExpressionObjects()
[<music21.expressions.Fermata>]

r   )expressionsra   Etr&   M)rc   r   r   rL   FermataTrillTurnMordent)r"   r   r   rO   r   s        r   getExpressionObjects#MuseDataRecord.getExpressionObjects  s     	(++-<KDs{K//12K//12K--/0K,,./K//12  r   c                D   SSK Jn  / nU R                  5       nUc  U$ SnU H  nUR                  U5      nUS:  a  M  USU X6[	        U5      -   S -   nUS;   a"  UR                  UR                  U5      5        MZ  US:X  a"  UR                  UR                  S5      5        M  US:X  a"  UR                  UR                  S	5      5        M  US
:X  a"  UR                  UR                  S	5      5        M  US:X  d  M  UR                  UR                  S	5      5        M     U$ )a>  
Return a list of 0 or more music21 Dynamic objects

>>> mdr = musedata.MuseDataRecord('C5    12        e     u         ff')
>>> mdr.getDynamicObjects()
[<music21.dynamics.Dynamic ff>]

>>> mdr = musedata.MuseDataRecord('E4    48        h     u        (pp')
>>> mdr.getDynamicObjects()
[<music21.dynamics.Dynamic pp>]
r   )dynamicsN)pppfffro   rI   fpmpmfrp   rH   mZZpR)	r   r   ro   rI   rp   rH   r   r   r   r   r   r   sfr   r   )rc   r   r   findr1   rL   Dynamic)r"   r   r   rO   targetstargetposs          r   getDynamicObjects MuseDataRecord.getDynamicObjects  s    	%++-<K3 F))F#CQw:3v;%6%7 88DOOH,,V453H,,T233H,,T234H,,T233H,,T23! $ r   c                :    U R                  5       nUc  gSU;   a  gg)a  
Return a boolean if this note has a cautionary accidental.

>>> mdr = musedata.MuseDataRecord('F5     3        t n   d  ==[   (+')
>>> mdr.hasCautionaryAccidental()
True

>>> mdr = musedata.MuseDataRecord('C4    12        e     u  [')
>>> mdr.hasCautionaryAccidental()
False

F+T)r   r   s     r   rn   &MuseDataRecord.hasCautionaryAccidental(  s'     ++-<$;r   )r    r   r   r   )rJ   N)r   str)returnboolr   )r   z
str | None)r   r   r   r   __doc__r#   r)   r2   r6   r;   r?   rD   rS   rh   rq   r   r   r   r   r   r   r   r   rn   r   r   r   r   r   r   9   s~     ..",

$L0d0d-^	,*!6"H&P D'Rr   r   c                  *    \ rS rSrSrS rS rS rSrg)MuseDataRecordIteratori@  z3
Create MuseDataRecord objects on demand, in order
c                *    Xl         SU l        X l        g Nr   )r   indexr   r!   s      r   r#   MuseDataRecordIterator.__init__E  s    
r   c                    U $ r   r   r(   s    r   __iter__MuseDataRecordIterator.__iter__J      r   c                    U R                   [        U R                  5      :  a  [        e[	        U R                  U R                      U R
                  5      nU =R                   S-  sl         U$ Nr,   )r   r1   r   StopIterationr   r   r"   mdrs     r   __next__MuseDataRecordIterator.__next__M  sK    ::TXX&TXXdjj14;;?

a

r   )r   r   r   N	r   r   r   r   r   r#   r   r   r   r   r   r   r   r   @  s    
r   r   c                  L    \ rS rSrSrSS jrS rS rS rS r	S	 r
S
 rS rSrg)MuseDataMeasureiX  a  
A MuseDataMeasure is an abstraction of the data contained within a measure definitions.

This needs to be an object to gracefully handle the following cases.
Some Measures do not have any notes, for example, and the end of encoding where a
final bar line is defined. Some measures do not have numbers or barlin definitions,
such as pickup notes. Some measures define barline characteristics.
Backup and forward presumably only is contained within a measure.
Nc                    Uc  / nXl         X l        U R                  b  U R                  R                  U l        g S U l        g r   )r   r   r   r!   s      r   r#   MuseDataMeasure.__init__c  s8    ;C;;"**DJDJr   c                    g)Nzsize={len(self.src)}r   r(   s    r   _reprInternalMuseDataMeasure._reprInternalq  s    %r   c                   SSK Jn  U R                  S   R                  5       nUS   S:w  a  SnUSS nUS:X  a  SnODUS	:X  a  S	nO;US
:X  a  SnO2US;   a  SnO)US:X  a  SnO US:X  a  SnOUS;   a  SnO[	        SU 35      eUR                  U5      n[        U5      S:  af  USS R                  5       S:w  aO  USS R                  5       nSU;   a  SnUR                  SS9nXEl        U$ SU;   a  SnUR                  SS9nXEl        U$ )zu
Return a configured music21 bar object. This can be used with the current
Measure or applied to a previous Measure.
r   )barr   measurer,   r-   easureregulardotteddoublezlight-light)heavy1heavyr   heavy2zlight-heavyheavy3zheavy-light)heavy4heave4zheavy-heavyz$cannot process bar data definition:    NrJ   z:|end)	directionz|:start)	rc   r   r   r   r   Barliner1   Repeattype)r"   r   rO   dataBarbarlineTypebldataFlagunused_repeatForms           r   getBarObjectMuseDataMeasure.getBarObjectt  s=    	 xx{  "7c>Dq)h#K "K 'K++!K 'K 'K,,'K#&J7)$TUU[[% t9r>d23ioo/25BCy(Hx$(!ZZ%Z0%
 		 !$(!ZZ'Z2%	r   c                X   SSK Jn  U R                  S   R                  5       nUS   S:w  a  SnSn[	        U5      S:  a/  USS R                  5       S	:w  a  [
        R                  " U5      u  p4UR                  5       nU R                  5       Ul	        US	:w  a  [        U5      Ul        U$ )
z?
Return a configured music21 :class:`~music21.stream.Measure`.
r   )streamr   r   1	   r/   NrJ   )rc   r  r   r   r1   r
   rM   Measurer  leftBarlinerv   number)r"   r  rO   mNumberrR   r   s         r   getMeasureObject MuseDataMeasure.getMeasureObject  s     	#xx{  "7c>D t9>d12hnn."4"006MGNN ))+ b=7|AHr   c                R    U R                    H  nU(       d  M  US   S;   d  M    g   g)z*
Return True of if this Measure has Notes
r   
ABCDEFGrgcTFr'   r"   lines     r   hasNotesMuseDataMeasure.hasNotes  s*     HHDtQ</  r   c                l    U R                    H$  nU(       d  M  UR                  S5      (       d  M$    g   g)z
Return True of if this Measure defines one or more 'back' indication.

Note: this does not instantiate MuseDataRecord instances.
rC   TFr   
startswithr  s     r   	hasVoicesMuseDataMeasure.hasVoices  s-     HHDt//  r   c                .    [        U R                  U 5      $ z:
Iterating over this part returns MuseDataMeasure objects
)r   r   r(   s    r   r   MuseDataMeasure.__iter__  s     &dhh55r   c                    [        U 5      $ )zH
Return a list of all records stored in this measure as MuseDataRecord.
listr(   s    r   
getRecordsMuseDataMeasure.getRecords       Dzr   )r   r   r   NN)r   r   r   r   r   r#   r   r  r  r  r  r   r%  r   r   r   r   r   r   X  s0    &.`4	6r   r   c                  *    \ rS rSrSrS rS rS rSrg)MuseDataMeasureIteratori  z4
Create MuseDataMeasure objects on demand, in order
c                6    Xl         X l        SU l        X0l        g r   )r   
boundariesr   r   )r"   r   r,  r   s       r   r#    MuseDataMeasureIterator.__init__  s    $
r   c                    U $ r   r   r(   s    r   r    MuseDataMeasureIterator.__iter__  r   r   c                   U R                   [        U R                  5      :  a  [        eU R                  U R                      u  p[	        U R
                  XS-    U R                  5      nU =R                   S-  sl         U$ r   )r   r1   r,  r   r   r   r   )r"   r   r   mdms       r   r    MuseDataMeasureIterator.__next__  s`    ::T__--__TZZ0
dhhu1W5t{{C

a

r   )r,  r   r   r   Nr   r   r   r   r*  r*    s    r   r*  c                      \ rS rSrSrS$S jrS rS rS rS r	  S%S	 jr
S
 rS rS rS rS rS rS rS rS&S jrS&S jrS rS rS rS rS rS rS rS'S jrS rS rS rS(S jr S  r!S! r"S" r#S#r$g))MuseDataParti  z3
A MuseData part is defined by collection of lines
Nc                   Uc  / nXl         / U l        S U l        S U l        X l        U R                  c&  U R                   (       a  U R                  5       U l        U R                  S:X  a!  U R                  U R                   5      U l         g g r   )r   _measureBoundaries_divisionsPerQuarter_divisionsPerQuarterNoter   _determineStage_scrubStage1)r"   r   r   s      r   r#   MuseDataPart.__init__  sx    ;C #%$(! )-%
::$((--/DJ::?((2DH r   c                    gNrJ   r   r(   s    r   r   MuseDataPart._reprInternal      r   c                Z    U R                    H  nUR                  S5      (       d  M    g   g)z
Determine the stage of this file. This is done by looking for an
attributes record starting with a $; if not found, it is stage 1. if it is found
it is stage 2.
$r   r,   r  )r"   	attributes     r   r9  MuseDataPart._determineStage  s+     I##C(( " r   c                    [        U5      S::  a  [        S5      eSn/ nU H3  nU(       a  UR                  5       S:X  a  M   SnUR                  U5        M5     U$ )z
Some stage1 files start with a leading line of space.
This needs to be removed, as each line matters. Provide a list of character lines.
r,   zcannot scrub empty sourceTrJ   F)r1   r   r   rL   )r"   r   checkr   r  s        r   r:  MuseDataPart._scrubStage1!  s[    
 s8q=#$?@@D::<2%!EKK  r   c                v   / nX!;   a  UR                  U5      [        U5      -   nU[        U5      :  av  X   R                  5       (       a  UR                  X   5        O5X   R	                  5       (       a  OX   S;   a  UR                  X   5        OOUS-  nU[        U5      :  a  Mv  SR                  U5      $ )ar  

>>> mdp = musedata.MuseDataPart()
>>> mdp._getDigitsFollowingTag('junk WK#:2345', 'WK#:')
'2345'
>>> mdp._getDigitsFollowingTag('junk WK#: 2345 junk', 'WK#:')
'2345'
>>> mdp._getDigitsFollowingTag('$ K:-3   Q:4   T:3/4   C:22', 'Q:')
'4'
>>> mdp._getDigitsFollowingTag('$ K:-3   Q:4   T:3/4   C:22', 'T:')
'3/4'
>>> mdp._getDigitsFollowingTag('$ K:-3   Q:4', 'T:')
''
z-/r,   rJ   )r   r1   isdigitrL   isspacerN   )r"   r  tagr   r   s        r   _getDigitsFollowingTag#MuseDataPart._getDigitsFollowingTag5  s     ;		#S)Ac$i-7??$$KK(W__&&W_KK(Q c$i- wwt}r   c                6   U(       d   UR                  5       nUR                  5       n/ nX!;   a  UR                  U5      [        U5      -   nU[        U5      :  a  X   R                  5       (       a  UR	                  X   5        OnX   R                  5       (       a  U(       d  OOX   R                  5       (       a  U(       a  UR	                  X   5        OX   S;   a  UR	                  X   5        OOUS-  nU[        U5      :  a  M  SR                  U5      $ )z

>>> mdp = musedata.MuseDataPart()
>>> mdp._getAlphasFollowingTag('Group memberships: sound, score', 'Group memberships:')
'sound,score'
r   r,   rJ   )lowerr   r1   isalpharL   rI  rN   )r"   r  rJ  	keepSpacekeepCaser   r   s          r   _getAlphasFollowingTag#MuseDataPart._getAlphasFollowingTagS  s     ::<D))+C;		#S)Ac$i-7??$$KK(W__&&yW__&&9KK(W^KK(Q c$i- wwt}r   c                    U R                   S:X  a<  U R                  S   SS R                  5       nSU;   a  UR                  S5      S   nU$ U R	                  U R                  S   S5      $ )a  
Returns a String not an int, representing an opus number

>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getWorkNumber()
'581'
r,   r      r   rB   zWK#:)r   r   r   rK   rK  r   s     r   getWorkNumberMuseDataPart.getWorkNumbero  sg     ::? 88A;q#))+Dd{zz#q)K..txx{FCCr   c                    U R                   S:X  a   U R                  S   SS R                  5       $ U R                  U R                  S   S5      $ )z
Returns a string, not an int.

>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getMovementNumber()
'3'
r,   rU  NrB   zMV#:)r   r   r   rK  r(   s    r   getMovementNumberMuseDataPart.getMovementNumber  sG     ::?88A;qr?((**..txx{FCCr   c                    U R                   S:X  a  gU R                  5       nU R                  USSSS9nUS:X  a  gUR                  5       $ )aU  
The directive field is generally used to store tempo indications.
This indication, however, is frequently not provided.

>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getDirective() is None
True
r,   NzD:T)rP  rQ  rJ   )r   _getAttributesRecordrR  r   )r"   r  alphass      r   getDirectiveMuseDataPart.getDirective  sU     ::?,,.D00tt:> 1 @F|||~%r   c                
   U R                   S:X  ae  / nU R                  S   U R                  S   U R                  S   4 H"  nUR                  UR                  5       5        M$     SR	                  U5      $ U R                  S   $ )u  
Return the source (original print encoding) for the score.

>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '03.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getSource()
'Breitkopf & Härtel, Vol. 13'

Note that the publisher is Härtel. MuseData predates the standardization
of encoding as UTF-8, so many
MuseData files are in Windows (ISO-8859-1) encodings.
Music21 tries first
in utf-8 and if there are errors will try again in ISO-8859-1, and if it
fails it will try to read ignoring any errors (leaving "Hrtel").
In this case, both UTF-8 and ISO-8859-1 support this character (Latin-1 space
in Unicode).

If you know the encoding you can pass that to MuseDataWork or MuseDataFile.
It is not yet a property you can specify in converter.

>>> mdw = musedata.MuseDataWork()
>>> mdw.encoding = 'ISO-8859-1'
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getSource()
'Breitkopf & Härtel, Vol. 13'
r,   r      rB   rJ   rt   )r   r   rL   r   rN   )r"   rO   r  s      r   	getSourceMuseDataPart.getSource  sn    : ::?D!dhhqk488A;?DJJL) @ 774= 88A;r   c                z    U R                   S:X  a  U R                  S   R                  5       $ U R                  S   $ )a  
For stage 1 just gets the catalogue number

>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getWorkTitle()
'Clarinet Quintet'
r,   r   rU  r   r   r   r(   s    r   getWorkTitleMuseDataPart.getWorkTitle  s4     ::?88A;$$&&88A;r   c                B    U R                   S:X  a  gU R                  S   $ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getMovementTitle()
'Trio II'
r,   Nr-   )r   r   r(   s    r   getMovementTitleMuseDataPart.getMovementTitle  s      ::?88A;r   c                ^    U R                   S:X  a  gU R                  S   R                  5       $ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getPartName()
'Clarinet in A'
r,   Nr/   re  r(   s    r   getPartNameMuseDataPart.getPartName  s)     ::?88A;$$&&r   c                   U R                   S:X  a  / $ U R                  U R                  S   S5      n/ nUR                  S5       H8  nUR	                  5       S:w  d  M  UR                  UR	                  5       5        M:     U$ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getGroupMemberships()
['sound', 'score']
r,   
   zgroup memberships:r   rJ   )r   rR  r   rK   r   rL   )r"   r   r   entrys       r   getGroupMemberships MuseDataPart.getGroupMemberships  sp     ::?I--dhhrl.BDCD3;;=B&KK. ( Kr   c                   U R                   S:X  a*  [        U R                  S   R                  S5      S   5      $ SnSnU R                  U   R	                  S5      (       db  U R                  U   nUR	                  U5      (       a  U R                  US5      nO*US-  nU R                  U   R	                  S5      (       d  Mb  Uc  g[        U5      $ )	z
>>> fp1 = str(common.getSourceFilePath() /'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getGroupMembershipsTotal()
5
r,   rt   r9   r      NrA  of)r   rv   r   rK   r  rK  r"   
membershipr   r   r  s        r   getGroupMembershipsTotal%MuseDataPart.getGroupMembershipsTotal  s     ::?txx{((-a011AChhqk,,S11xx{??:..55dDACQ hhqk,,S11 {3xr   c                   U R                   S:X  a!  U R                  S   R                  S5      S   $ SnSnU R                  U   R                  S5      (       db  U R                  U   nUR                  U5      (       a  U R	                  US5      nO*US-  nU R                  U   R                  S5      (       d  Mb  Uc  g[        U5      $ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getGroupMembershipNumber()
1
r,   rt   r9   rt  NrA  part)r   r   rK   r  rK  rv   rv  s        r   getGroupMembershipNumber%MuseDataPart.getGroupMembershipNumber0  s     ::?88A;$$S)!,,AChhqk,,S11xx{??:..55dFCCQ hhqk,,S11 {3xr   c                ^   U R                   S:X  a@  U R                  S   R                  5       S-   U R                  S   R                  5       -   nU$ SnU R                  U   R                  S5      (       d*  US-  nU R                  U   R                  S5      (       d  M*  U R                  U   $ )a\  
The attributes record is not in a fixed position,
but is the first line that starts with a $.


>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0]._getAttributesRecord()
'$  K:0   Q:6   T:3/4   X:-11   C:4'
r,   rU  r9   r-   rt  rA  )r   r   r   r  )r"   recordr   s      r   r\  !MuseDataPart._getAttributesRecordJ  s     ::?XXa[&&(3.!1B1B1DDFMAhhqk,,S11Q hhqk,,S1188A;r   c                    U R                  5       nU R                  S:X  a  [        UR                  S5      S   5      $ [        U R	                  US5      5      $ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                   / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getKeyParameters()
0
r,   r9   zK:r\  r   rv   rK   rK  r  s     r   getKeyParametersMuseDataPart.getKeyParametersb  sO     ((*::?tzz#q)** t224>??r   c                L    SSK Jn  UR                  U R                  5       5      $ )a  
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getKeySignature()
<music21.key.KeySignature of no sharps or flats>
r   )key)rc   r  KeySignaturer  )r"   r  s     r   getKeySignatureMuseDataPart.getKeySignaturer  s!     	  5 5 788r   c                \   U R                  5       nU R                  S:X  a;  [        UR                  S5      S   5      n[        UR                  S5      S   5      nO8U R	                  US5      R                  S5      u  p#[        U5      [        U5      p2US:X  a  US:X  d  US:X  a  gU SU 3$ )	z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                   / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getTimeSignatureParameters()
'3/4'
r,   r9   rB   rt   zT:/r   z4/4r  )r"   r  rZ   ds       r   getTimeSignatureParameters'MuseDataPart.getTimeSignatureParameters~  s     ((*::?DJJsOA&'ADJJsOA&'A ..tT:@@EDAq63q6q FqAv!q&S!:r   c                L    SSK Jn  UR                  U R                  5       5      $ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getTimeSignatureObject()
<music21.meter.TimeSignature 3/4>
r   )meter)rc   r  TimeSignaturer  )r"   r  s     r   getTimeSignatureObject#MuseDataPart.getTimeSignatureObject  s!     	"""4#B#B#DEEr   c                    U R                   S:X  a  gU R                  5       nU R                  US5      nUS:X  a  g[        U5      $ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0]._getNumberOfStaves()
1
r,   zS:rJ   )r   r\  rK  rv   r"   r  r   s      r   _getNumberOfStavesMuseDataPart._getNumberOfStaves  sE     ::?,,.D--dD9Cby3xr   c                |   U R                  5       nU R                  S:X  a  UR                  S5      S   nU/$ / nU R                  US5      nUS:w  a  UR	                  U5        US:X  aS  [        SU R                  5       S-   5       H2  nU R                  USU S35      nUS:w  d  M!  UR	                  U5        M4     U$ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                   / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0]._getClefParameters()
['4']
r,   r9   r/   zC:rJ   Cr   )r\  r   rK   rK  rL   ranger  )r"   r  rO   r   r   r   s         r   _getClefParametersMuseDataPart._getClefParameters  s     ((*::? ::c?1%D6M D--dD9CbyC byq$"9"9";a"?@A55das!HECbyC( A Kr   c                   U R                   S:X  a  gU R                  5       US-
     n[        [        U5      5      nSSKJn  US:X  a  UR                  5       $ US:X  a  UR                  5       $ US:X  a  UR                  5       $ US:X  a  UR                  5       $ US	:X  a  UR                  5       $ US
:X  a  UR                  5       $ US:X  a  UR                  5       $ US:X  a  UR                  5       $ US:X  a  UR                  5       $ US:X  a  UR                  5       $ US:X  a  UR!                  5       $ US:X  a  UR#                  5       $ US:X  a  UR%                  5       $ US:X  a  UR'                  5       $ [)        SU5      e)a!  
Return a music21 clef object based on a two character clef definition.

>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getClefObject().sign
'G'
r,   Nr   )clef5434643111213141522235282zcannot determine clef from:)r   r  r   rv   rc   r  FrenchViolinClef
TrebleClefTreble8vbClefTreble8vaClefGSopranoClefCBaritoneClef	TenorClefAltoClefMezzoSopranoClefSopranoClefBassClefFBaritoneClefBass8vbClefBass8vaClefr   )r"   voicecharPairr  s       r   getClefObjectMuseDataPart.getClefObject  s    ::? ..0;H3x=)H$3,,..S((T!))++T!))++S((**T!))++T!~~''T!}}&T!,,..T!''))T!}}&T!))++T!''))T!''))'(ExPPr   c                    U R                  5       nU R                  S:X  a  gU R                  US5      nUS:X  a  g[        U5      $ )a  
Get the transposition, if defined, from the Metadata header.


>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> fp2 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '02.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.addFile(fp2)
>>> mdw.getParts()[0]._getTranspositionParameters()
-11
>>> mdw.getParts()[1]._getTranspositionParameters() is None
True
r,   NzX:rJ   )r\  r   rK  rv   r  s      r   _getTranspositionParameters(MuseDataPart._getTranspositionParameters  sE    " ((*::?--dD9Cby3xr   c                V    U R                  5       nUc  g[        R                  " U5      $ )a  
If this part defines a transposition, return a corresponding Interval object.

>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> fp2 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '02.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.addFile(fp2)
>>> mdw.getParts()[0].getTranspositionIntervalObject()
<music21.interval.Interval m-3>
N)r  r   base40DeltaToInterval)r"   argss     r   getTranspositionIntervalObject+MuseDataPart.getTranspositionIntervalObject+  s+     //1<//55r   c                `   U R                   c  U R                  5       nU R                  S:X  aV  [        UR	                  S5      S   5      nU R                  5       nX#R                  R                  -  U l         U R                   $ [        U R                  US5      5      U l         U R                   $ )z
>>> fp1 = (common.getSourceFilePath() / 'musedata' / 'testPrimitive'
...                    / 'test01' / '01.md')
>>> mdw = musedata.MuseDataWork()
>>> mdw.addFile(fp1)
>>> mdw.getParts()[0].getDivisionsPerQuarterNote()
6
r,   r9   r   zQ:)	r8  r\  r   rv   rK   r  barDurationquarterLengthrK  )r"   r  rO   tss       r   ry   'MuseDataPart.getDivisionsPerQuarterNote@  s     ((0,,.DzzQ4::c?1-.002 15~~7S7S0S- ,,, 14//d;1=- ,,,r   c                b   Uc  U R                   n/ n/ nSn[        U5      S-
  nU R                  S:X  aC  [        U5       H.  u  pgUR	                  S5      (       d  M  UR                  U5        M0     US   nOgSn[        U5       HV  u  pgUR	                  S5      (       a  US-  nM"  US:  a  Uc  UnUR	                  S5      (       d  ME  UR                  U5        MX     Sn	US   U:X  aG  [        U5      S:X  a  UR                  US   U45        Sn	O9UR                  US   US   S-
  45        Sn	OUR                  XCS   S-
  45        Sn	U	bb  [        U	[        U5      5       HI  nU[        U5      S-
  :X  a  UR                  X6   U45        M,  UR                  X6   X6S-      S-
  45        MK     U$ )ad  

>>> mdp = musedata.MuseDataPart()
>>> mdp.stage is None
True
>>> mdp._getMeasureBoundaryIndices(['$', 'A', 'B', 'm', 'C', 'm', 'D'])
[(1, 2), (3, 4), (5, 6)]
>>> mdp._getMeasureBoundaryIndices(['$', 'm', 'B', 'C', 'm', 'D', 'E', 'm', 'F', 'D'])
[(1, 3), (4, 6), (7, 9)]
>>> mdp._getMeasureBoundaryIndices(['$', 'B', 'C', 'm', 'D', 'E'])
[(1, 2), (3, 5)]
Nr,   r   r   rA  )r   r1   r   	enumerater  rL   r  )
r"   r   r,  mIndicesfirstPostAttributesIndex	lastIndexr   r  mAttributeRecordCountstartIterIndexs
             r   _getMeasureBoundaryIndices'MuseDataPart._getMeasureBoundaryIndices`  s    ;((C
#' HqL	::?$S>??3''OOA& * (0{$ %&!$S>??3'')Q.)(1,1I1Q/0,??3''OOA& * A;228}!!!8A;	":;!%!!8A;a"@A!" 7!qIJN%>3x=9H))%%x{I&>?%%x{HUOa4G&HI : r   c                X    U R                  5       U l        U R                  5       U l        g)z
After setting the source string, this method must be called to configure
the _measureNumberToLine method and set additional attributes.
N)ry   r7  r  r6  r(   s    r   updateMuseDataPart.update  s&    
 %)$C$C$E!"&"A"A"Cr   c                D    [        U R                  U R                  U 5      $ r   )r*  r   r6  r(   s    r   r   MuseDataPart.__iter__  s     'txx1H1H$OOr   c                    [        U 5      $ )zO
Return a list of all measures stored in this part as MuseDataMeasure objects.
r#  r(   s    r   getMeasuresMuseDataPart.getMeasures  r'  r   )r7  r8  r6  r   r   r(  )FF)score)r,   r   )%r   r   r   r   r   r#   r   r9  r:  rK  rR  rV  rY  r^  rb  rf  ri  rl  rq  rx  r|  r\  r  r  r  r  r  r  r  r  r  ry   r  r  r   r  r   r   r   r   r4  r4    s    3(	(< ;@(-8D,D"&0%N"'( 4 40@ 
90
F * D5Qn 66*-@@DDPr   r4  c                  @    \ rS rSrSrSS jrS rS rS rS r	S r
S	rg
)MuseDataFilei  z
A MuseDataFile file may describe one or more MuseDataPart;
a Score might need multiple files for complete definition.
A MuseDataFile object can be created from a string.

When read, one or more MuseDataPart objects are created and stored on self.parts.
c                <    / U l         S U l        S U l        SU l        g Nzutf-8)partsfilenamefileencodingr(   s    r   r#   MuseDataFile.__init__  s    )+
"&%)	$r   c                    gr=  r   r(   s    r   r   MuseDataFile._reprInternal  r?  r   c                2    [        US5      U l        Xl        g )Nrb)openr  r  )r"   r   s     r   r  MuseDataFile.open  s     TN	r   c                    U R                   R                  5       n UR                  U R                  5      nU R                  U5      $ ! [         a    UR                  SS5      n N/f = f)Nz
ISO-8859-1ignore)r  readdecoder  UnicodeDecodeErrorreadstr)r"   fileContentss     r   r  MuseDataFile.read  sb    yy~~'	G'..t}}=L ||L)) " 	G'..|XFL	Gs   A A'&A'c                8    U R                   R                  5         g r   )r  closer(   s    r   r  MuseDataFile.close  s    		r   c                   Sn/ nUR                  S5      n[        U5       H  u  pVUR                  S5      (       a  U(       a  SnM&  SnM*  U(       a  M3  U(       a  US   S:X  a  ME  UR                  S5      (       d  UR                  S5      (       aM  [        U5      S	::  a  / nM  [	        U5      nUR                  5         U R                  R                  U5        / nM  UR                  S
5      (       a  M  UR                  U5        M     g)z3
Read a string, dividing it into individual parts.
F
r_   Tr   @z/ENDENDr,   z/eofN)rK   r  r  r1   r4  r  r  rL   )r"   	input_strcommentTogglelinessrcLinesr   r  mdps           r   r  MuseDataFile.readstr  s     ??4( *GA s## $)M$(M$q'S.((DOOE,B,B u:?E"5)



!!#&((T"C +r   )r  r  r  r  Nr   None)r   r   r   r   r   r#   r   r  r  r  r  r   r   r   r   r  r    s%    %*-#r   r  c                  8    \ rS rSrSrS	S jrS
S jrS rS rSr	g)MuseDataWorki  z,
A work might consist of one or more files.
c                     / U l         SU l        g r  )filesr  r(   s    r   r#   MuseDataWork.__init__  s    )+
$r   c                \   [         R                  " U5      (       d  [        R                  " [        U5      nU/nOUnU Hj  n[        5       nU R                  Ul        UR                  U5        UR                  5         UR                  5         U R                  R                  U5        Ml     g)zZ
Open and read this file path or list of paths as MuseDataFile objects
and set self.files
N)r
   
isIterabler   castr   r  r  r  r  r  r  rL   )r"   r   	fp_singlefpListfpInnermdfs         r   addFileMuseDataWork.addFile  s|       $$sBI[FFG.C==CLHHWHHJIIKJJc" r   c                    [         R                  " U5      (       d  U/nOUnU H9  n[        5       nUR                  U5        U R                  R                  U5        M;     g)a  
Add a string representation acting like a part file

>>> mdw = musedata.MuseDataWork()
>>> mdw.addString('WK#:581       MV#:3c\nBreitkopf & Hartel, Vol. 13\n' +
...               'Clarinet Quintet\n' +
...               'Trio II\n' +
...               '$  K:0   Q:6   T:3/4   X:-11   C:4\n' +
...               'C5     3        e     d  [     (&0p\n' +
...               'E5     3        e     d  ]')

# TODO: Okay, so what? did we test this or demo anything?
N)r
   r
  r  r  r  rL   )r"   r  strList
thisStringr  s        r   	addStringMuseDataWork.addString+  sP    "   ++ kGG!J.CKK
#JJc" "r   c                x    / nU R                    H'  nUR                   H  nUR                  U5        M     M)     U$ )z
Get all parts contained in all files associated with this work.
A list of MuseDataPart objects that were created in a MuseDataFile.
)r  r  rL   )r"   r   r  r   s       r   getPartsMuseDataWork.getPartsH  s8     ::CyyC  !  r   )r  r  Nr  )r   zstr | Iterable[str]r   r  )
r   r   r   r   r   r#   r  r  r  r   r   r   r   r  r    s    %#*#:r   r  c                  4    \ rS rSrSrS rS rS rS	S jrSr	g)
MuseDataDirectoryiW  aD  
This class manages finding musedata files stored in a directory,
comparing file names and examining subdirectories to determine which files are parts.

Once found, a MuseDataWork, or a list of paths, can be returned

A directory, or a list of file path stubs, such as that obtained within a zip file,
can both be provided.
c                B    / U l         0 U l        U R                  U5        g r   )pathsgroups_prepareGroups)r"   	dirOrLists     r   r#   MuseDataDirectory.__init__b  s    
I&r   c                .   / n[         R                  " U5      (       a  UnO[        R                  R	                  U5      (       aU  [        [        R                  " U5      5       H1  nUR                  [        R                  R                  X5      5        M3     O[        SU5      eU Hw  n[        R                  R                  U5      u  pSU R                  U5      (       d  M<  [         R                  " U5      u  pgUS:X  a  M\  U R                  R                  U5        My     / n[        U R                  5      S:  a  [        U R                  5       Ho  u  p[        R                  R                  U5      u  pS[         R                  " U5      u  pgUS:w  d  MF  UR!                  S5      (       d  M^  UR                  U	5        Mq     UR#                  5         U H  n	U R                  R%                  U	5        M      O U R                  R'                  5         g )Nz*cannot get files from the following entityrJ   r,   s)r
   r
  ospathisdirsortedlistdirrL   rN   r   rK   isMusedataFilerM   r  r1   r  r  reversepopsort)
r"   r   allPathsfnr   unused_directorynumStr	nonNumStrpopListr   s
             r   r   MuseDataDirectory._prepareGroupsh  s     Y'' !H WW]]9%% RZZ	23Y ;< 4 $$PR[\\B#%77==#4 &&r** & 4 4R 8F|

!!"%  tzz?Q"4::.')ww}}R'8$ $*$8$8$<!R<I$8$8$=$=NN1% / OO

q!   	

r   c                   [         R                  R                  U5      u  p#UR                  S5      (       a  gUR	                  S5      (       a  gUR                  S5      (       df  UR                  S5      (       dP  UR                  [         R
                  5      (       d,  UR	                  S5      (       d  UR                  S5      (       a  gg)	Nz.mdTmchanFz.pyr  r   z	.svn-base)r$  r%  rK   endswithr  sep)r"   r   
unused_dirr.  s       r   r)   MuseDataDirectory.isMusedataFile  s    
 r*
;;u]]7##kk%  S!!RVV$$s##[))r   Nc                    U R                   $ )z*
Return sorted paths for a group, or None
)r  )r"   groups     r   getPathsMuseDataDirectory.getPaths  s    
 zzr   )r  r  r   )
r   r   r   r   r   r#   r  r)  r<  r   r   r   r   r  r  W  s    '<@&r   r  c                  &    \ rS rSrS rS rS rSrg)Testi  c                
   [        [        R                  " 5       S-  S-  5      n[        5       n[        R
                  R                  US5      nS H3  n[        R
                  R                  X45      nUR                  U5        M5     UR                  5       nU R                  [        U5      S5        U R                  US   R                  S   S5        U R                  US   R                  S	   S
5        U R                  US   R                  S   S5        U R                  US   R                  S	   S5        U R                  US   R                  S   S5        U R                  US   R                  S	   S5        U R                  US   R                  S   S5        U R                  US   R                  S	   S5        U R                  US   R                  S   S5        U R                  US   R                  S	   S5        [        S5       GH&  nU R                  XV   R                  5       S5        U R                  XV   R                  5       S5        U R                  XV   R!                  5       R#                  S5      5        U R                  XV   R%                  5       S5        U R                  XV   R'                  5       S5        U R                  XV   R)                  5       SS/5        U R                  XV   R+                  S5      S5        U R                  XV   R+                  S5      S5        GM)     U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R-                  S5      S5        U R                  US   R/                  5       S5        U R                  US   R1                  5       S5        U R                  US   R3                  5       S5        g )Nr   testPrimitivetest01)z01.mdz02.mdz03.mdz04.mdz05.mdrt   r   rB   zWK#:581       MV#:3c   zscore: part 1 of 5r,   zscore: part 2 of 5r   zscore: part 3 of 5ra  zscore: part 4 of 5zscore: part 5 of 5581r  	BreitkopfzClarinet QuintetzTrio IIsoundr  z3/4rU  )r   r
   getSourceFilePathr  r$  r%  rN   r  r  assertEqualr1   r   r  rV  rY  
assertTruerb  r  rf  ri  rq  rx  r|  r  r  ry   )r"   r   mdwdirLibr.  mdpObjsr   s          r   testLoadFromFileTest.testLoadFromFile  s9   ))+j8?JKnb(+?Bf)B KKO	 @ ,,.Wq)*,BC+-AB*,BC+-AB*,BC+-AB*,BC+-AB*,BC+-AB qAWZ557?WZ99;SAOOGJ002==kJKWZ4468JKWZ88:IFWZ;;=?QRWZ@@I1MWZ@@I1M  	<<WEqI<<WEqI<<WEqI<<WEqI<<WEqI<<WEqI<<WEqI<<WEqI<<WEqI<<WEqI446:>>@%H>>@!Dr   c                h    [        [        R                  " 5       S-  S-  S-  5      n[        U5      ng )Nr   rA  rB  )r   r
   rG  r  )r"   fpDir
unused_mdds      r   testMuseDataDirectoryTest.testMuseDataDirectory<  s1     F,,.;oMPXXY&u-
r   c                8   [        S5      nSUl        U R                  UR                  5       S/5        [        S5      nSUl        U R                  UR                  5       S/5        [        S5      nSUl        U R                  UR                  5       SS/5        g )Nz/D4     2        e     u                    con-r   zcon-z,F#4    2        e     u                    aaz0F#4    2        e     u                    a | bb)r   r   rH  r   r   s     r   testGetLyricsTest.testGetLyrics]  s    NO	6(3KL	3%0OP	3*5r   r   N)r   r   r   r   rM  rR  rW  r   r   r   r   r?  r?    s    @9En.B6r   r?  __main__)$r   
__future__r   collections.abcr   r$  typingr   unittestrc   r   r   music21.musedatar   r   r	   r
   r   EnvironmentenvironLocalMusic21Exceptionr   ProtoM21Objectr   r   r   r*  r4  r  r  r  TestCaser?  
_DOC_ORDERr   mainTestr   r   r   <module>rf     s"    # $ 	      & # &  &&z2	55 	
BW++ BN 0Dg,, DP 4z
7)) z
|Q#7)) Q#jF7)) FTi.. i\c68 c6h ^
 zT r   