j ddlmZddlZddlmZddlZddlmZddlmZddlm Z ej dZ Gdd Z Gd d e Z Gd d ZGddejZedkrddlZejedSdS)) annotationsN)inf) environment)pitch)musedatazanalysis.enharmonicsceZdZdZdS)EnharmonicScoreRulesc>d|_d|_d|_d|_dS)NF) sameStaffLinealterationPenalty augDimPenaltymixSharpsFlatsPenaltyselfs /Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/site-packages/music21/analysis/enharmonics.py__init__zEnharmonicScoreRules.__init__s("!"%*"""N)__name__ __module__ __qualname__rrrr r s#+++++rr ceZdZfdZxZS)ChordEnharmonicScoreRulescVtd|_dS)Nr )superrr)r __class__s rrz"ChordEnharmonicScoreRules.__init__ s' %&"""r)rrrr __classcell__)rs@rrrs8'''''''''rrc@eZdZdZefdZdZdZdZdZ dZ dZ d S) EnharmonicSimplifiera Takes any pitch list input and returns the best enharmonic respelling according to the input criteria and rule weightings. Those criteria and rule weightings are currently fixed, but in future the user should be able to select their own combination and weighting of rules according to preferences, with predefined defaults for melodic and harmonic norms. Note: EnharmonicSimplifier itself returns nothing. ct|dtr d|D}||_||_d|_g|_|dS)Nrc6g|]}tj|Sr)rPitch.0ps r z1EnharmonicSimplifier.__init__../s ;;;AQ;;;r) isinstancestr pitchList ruleObjectallPossibleSpellings allSpellingsgetRepresentations)rr+ ruleClasss rrzEnharmonicSimplifier.__init__-sg ilC ( ( <;;;;;I"#)++$(! !!!!!rcg}|jD]0}|g|dz}||1||_dS)z Takes a list of pitches or pitch names and retrieves all enharmonic spellings. Note: getRepresentations itself returns nothing. N)r+getAllCommonEnharmonicsappendr.)rr.r' spellingss rr/z'EnharmonicSimplifier.getRepresentations7sY   + +Aa77:::I    * * * *(rcZttj|j|_|jS)N)list itertoolsproductr.r-rs r getProductzEnharmonicSimplifier.getProductBs&$():D>> pList1 = [pitch.Pitch('C'), pitch.Pitch('D'), pitch.Pitch('E')] >>> es = analysis.enharmonics.EnharmonicSimplifier(pList1) >>> es.bestPitches() (, , ) >>> pList2 = ['D--', 'E', 'F##'] >>> es = analysis.enharmonics.EnharmonicSimplifier(pList2) >>> es.bestPitches() (, , ) )r:rr-getAugDimScoregetAlterationScoregetMixSharpFlatsScore)r bestPitchesminScore possibilitythisAugDimScorethisAlterationScorethisMixSharpsFlatScore thisScores rr?z EnharmonicSimplifier.bestPitchesFs  4 * *K"11+>>O"&"9"9+"F"F %)%?%? %L%L "'*==@VVI8##$) rc|jjdurdSdd|D}|d}|d}||zdz|jjz}|S)z Returns a score according to the number of sharps and flats in a possible spelling. The score is the sum of the flats and sharps + 1, multiplied by the alterationPenalty. Fr2cg|] }|j Srnamer%s rr(z;EnharmonicSimplifier.getAlterationScore..i$A$A$AQV$A$A$Ar-#)r,rjoincountrrAjoinedPossibility flatCount sharpCountscores rr=z'EnharmonicSimplifier.getAlterationScoreas~ ? , 5 51GG$A$A[$A$A$ABB%++C00 &,,S11 Z'!+t/PP rc|jjdurdSdd|D}|d}|d}t ||g|jjz}|S)z Returns a score based on the mixture of sharps and flats in a possible spelling: the score is given by the number of the lesser used accidental (sharps or flats) multiplied by the mixSharpsFlatsPenalty. Fr2rGcg|] }|j SrrIr%s rr(z>EnharmonicSimplifier.getMixSharpFlatsScore..xrKrrLrM)r,rrNrOminrPs rr>z*EnharmonicSimplifier.getMixSharpFlatsScoreos ? 0E 9 91GG$A$A[$A$A$ABB%++C00 &,,S11 Y +,,t/TT rc|jjdurdSd}tt|dz D]y}tjj||j}tjj||dzj}||z dz}|tjj |dz }z| d}| d}||zdz|jjz} | S)z Returns a score based on the number of augmented and diminished intervals between successive pitches in the given spelling. Fr2rG(dddAd) r,rrangelenrbase40base40RepresentationrJbase40IntervalTablegetrO) rrA intervalStrip0p1 base40diffdimCountaugCountrTs rr<z#EnharmonicSimplifier.getAugDimScore~s ? (E 1 11 s;''!+,, V VA5k!n6IJB5k!a%6H6MNBr'RJ 8?>BB:uUU UKK$$S))$$S))H$q(DO,II rN) rrr__doc__r rr/r:r?r=r>r<rrrr!r!$s-A"""" ) ) ))))6      rr!c&eZdZdZdZdZdZdS)Testctjdtjdtjdg}t|}|}|t |d|t |d||dtjdS)NCDEr)rr$r!r? assertEqualr^assertIsInstance)rpListes bestPitchLists rtestBestPitcheszTest.testBestPitchessS!!5;s#3#3U[5E5EF !% ( (((  UQ''' ]++Q/// mA. <<<<rrr^rsrz)rrtrur{ testMixScores rtestGetMixSharpFlatsScorezTest.testGetMixSharpFlatsScoresS!!5;s#3#3U[5E5EF !% ( ( C  %+c"2"2EK4D4DE//55  UQ''' lC00000rctjdtjdtjdg}t|}tjdtjdtjdg}||}|t |d||tdSry)rr$r!r<rrr^rsrz)rrtrur{testAugDimScores rtestGetAugDimScorezTest.testGetAugDimScoresS!!5;s#3#3U[5E5EF !% ( ( C  %+c"2"2EK4D4DE++D11 UQ''' os33333rN)rrrrwr}rrrrrrlrlsP===11111144444rrl__main__) __future__rr8mathrunittestmusic21rrr Environment environLocalr rr!TestCaserlrmainTestrrrrsr#"""""&{&'=>> ++++++++''''' 4''' kkkkkkkk\$4$4$4$4$48 $4$4$4P zNNNGTr