3jddlmZddlZddlmZddlZddlmZddlmZddlm Z ejdZ Gdd Z Gd d e Z Gd d ZGddejZedk(rddlZej$eyy)) annotationsN)inf) environment)pitch)musedatazanalysis.enharmonicsceZdZdZy)EnharmonicScoreRulesc<d|_d|_d|_d|_y)NF) sameStaffLinealterationPenalty augDimPenaltymixSharpsFlatsPenaltyselfs I/DATA/.local/lib/python3.12/site-packages/music21/analysis/enharmonics.py__init__zEnharmonicScoreRules.__init__s""!"%*"N)__name__ __module__ __qualname__rrrr r s+rr ceZdZfdZxZS)ChordEnharmonicScoreRulesc0t|d|_y)Nr )superrr)r __class__s rrz"ChordEnharmonicScoreRules.__init__ s %&"r)rrrr __classcell__)rs@rrrs ''rrc>eZdZdZefdZdZdZdZdZ dZ dZ y ) 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"|Dcgc]}tj|}}||_||_d|_g|_|jycc}w)Nr) isinstancestrrPitch pitchList ruleObjectallPossibleSpellings allSpellingsgetRepresentations)rr& ruleClassps rrzEnharmonicSimplifier.__init__-s] ilC (1:;AQI;"#+$(! ! >> 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() (, , ) )r6rr(getAugDimScoregetAlterationScoregetMixSharpFlatsScore)r bestPitchesminScore possibilitythisAugDimScorethisAlterationScorethisMixSharpsFlatScore thisScores rr;z EnharmonicSimplifier.bestPitchesFs  44K"11+>O"&"9"9+"F %)%?%? %L "'*==@VVI8#$) 5rc|jjdurydj|Dcgc]}|jc}}|j d}|j d}||zdz|jjz}|Scc}w)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. Fr.-#)r'rjoinnamecountrr=r,joinedPossibility flatCount sharpCountscores rr9z'EnharmonicSimplifier.getAlterationScoreas ?? , , 5GG[$A[QVV[$AB%++C0 &,,S1 Z'!+t/P/PP %BsBc |jjdurydj|Dcgc]}|jc}}|j d}|j d}t ||g|jjz}|Scc}w)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. Fr.rCrDrE)r'rrFrGrHminrIs rr:z*EnharmonicSimplifier.getMixSharpFlatsScoreos ?? 0 0E 9GG[$A[QVV[$AB%++C0 &,,S1 Y +,t/T/TT %BsB c|jjduryd}tt|dz D]}tj j ||j}tj j ||dzj}||z dz}|tj jj|dz }|jd}|jd}||zdz|jjz} | S)z Returns a score based on the number of augmented and diminished intervals between successive pitches in the given spelling. Fr.rC(dddAd) r'rrangelenrbase40base40RepresentationrGbase40IntervalTablegetrH) rr= intervalStrip0p1 base40diffdimCountaugCountrMs rr8z#EnharmonicSimplifier.getAugDimScore~s ?? ( (E 1 s;'!+,A55k!n6I6IJB55k!a%6H6M6MNBr'RJ 8??>>BB:uU UK - $$S)$$S)H$q(DOO,I,II rN) rrr__doc__r rr*r6r;r9r:r8rrrr!r!$s0-A" ))6  rr!c$eZdZdZdZdZdZy)Testchtjdtjdtjdg}t|}|j}|j t |d|j t |d|j |dtjy)NCDEr)rr%r!r; assertEqualrVassertIsInstance)rpListes bestPitchLists rtestBestPitcheszTest.testBestPitchessS!5;;s#3U[[5EF !% ((  UQ' ]+Q/ mA. rs#&{&&'=> ++' 4' kk\$48  $4P zGTr