j<xUdZddlmZddlZddlZddlZddlZddlm Z ddlm Z ddlm Z e j e j e j e je jdZded <gd Zd ed <d eDZd%dZd&dZd'dZdZd(dZd)dZd*dZ d+d,d!ZGd"d#ejZed$krddlZej edSdS)-z7 Functions that find appropriate plots for graph.plot. ) annotationsN)axis)plot) primitives)ambitusdolan instrumentskey pianorollz%dict[str, type[plot.PlotStreamMixin]]PLOTCLASS_SHORTCUTS)) horizontalbarbar horizontalr piano) histogramhistocount)scatterpoint)scatterweightedweightedscatterweighted)3dbars3d) colorgridgridwindowwindowed)horizontalbarweighted barweighted weightedbarzlist[tuple[str, ...]]FORMAT_SYNONYMScg|] }|d S)r).0syns /Users/tedpeters/Documents/Projects/Claude Projects/Misc Tools/notename-sheets/.venv/lib/python3.11/site-packages/music21/graph/findPlot.py r(/s - - -c3q6 - - -return list[type[plot.PlotStreamMixin]]cg}ttjD]}tt|}t |rt |t jsst|drctj |j vrPtj |j vr=| tjt tj ||S)a{ return a list of all PlotStreamMixin subclasses. Returns a list sorted by name >>> graph.findPlot.getPlotClasses() [, , , , , ...] __mro__)sortedr__dict__getattrcallable isinstancetypes FunctionTypehasattrPlotStreamMixinr-rGraphappendtcasttype)allPlotinames r'getPlotClassesr?2s13G DM " "EEtQ TNN E"4);<< ED),, E(DL88$ 44 NN16$t';">> graph.findPlot.getAxisClasses() [, , , , ...] ) r.rr/r0r1r2r3r4 issubclassAxisr8)allAxisr=r>s r'getAxisClassesrEKs&(G DM " "!!tQ TNN !"4);<< !tTY// ! NN4 Nr)Fc|t}g}|D]?}|r||j||jd@|S)a >>> graph.findPlot.getAxisQuantities() ['generic', 'count', 'dynamic', 'offset', 'offsetEnd', 'pitchGeneric', 'pitchClass', 'pitchSpace', 'octave', 'position', 'quarterLength'] >>> graph.findPlot.getAxisQuantities(synonyms=True) ['generic', 'one', 'nothing', 'blank', 'count', 'quantity', 'frequency', ...] >>> theseAxes = [graph.axis.CountingAxis, graph.axis.OffsetAxis] >>> graph.findPlot.getAxisQuantities(axesToCheck=theseAxes) ['count', 'offset'] >>> graph.findPlot.getAxisQuantities(True, axesToCheck=theseAxes) ['count', 'quantity', 'frequency', 'counting', 'offset', 'measure', 'offsets', 'measures', 'time'] Nr)rEextend quantitiesr8)synonyms axesToCheck allQuantitiesaxClasss r'getAxisQuantitiesrM`sp$$&& M88  8  !3 4 4 4 4  !3A!6 7 7 7 7 r)c|}|dd}tD]}||vr |dcS|S)a Replace possible user format strings with defined format names as used herein. Returns string unaltered if no match. >>> graph.findPlot.userFormatsToFormat('horizontal') 'horizontalbar' >>> graph.findPlot.userFormatsToFormat('Weighted Scatter') 'scatterweighted' >>> graph.findPlot.userFormatsToFormat('3D') '3dbars' Unknown formats pass through unaltered. >>> graph.findPlot.userFormatsToFormat('4D super chart') '4dsuperchart'  r)lowerreplacer") userFormatopts r'userFormatsToFormatrU}s^$!!##J##C,,J   q6MMM  r)ct|}|t}g}|D]4}|j|kr||5|S)a| Given a graphFormat, find a list of plots that match: >>> graph.findPlot.getPlotClassesFromFormat('scatterweighted') [, , , ] Or give a list of plot classes to check: >>> pcs = [graph.plot.ScatterWeighted, graph.plot.Dolan] >>> graph.findPlot.getPlotClassesFromFormat('scatterweighted', pcs) [] )rUrQr? graphTyper8) graphFormatcheckPlotClasses filteredPlotsps r'getPlotClassesFromFormatr\sw"&k2288::K)++M $$ ;    + - -   # # # r) axisValuestrtype[axis.Axis] | NonecPtD]}t||r|cSdS)a_ given an axis value return the single best axis for the value, or None uses Axis.quantities >>> getAxis = graph.findPlot.getAxisClassFromValue >>> getAxis('counting') >>> getAxis('pc') >>> print(getAxis('boogie')) None N)rEaxisMatchesValue)r]thisAxiss r'getAxisClassFromValuercs>"#$$ Hi 0 0 OOO  4r) axisClasstype[axis.Axis] | axis.Axisboolcx|}|jD]}||krdSdS)a Returns Bool about whether axisValue.lower() is anywhere in axisClass.quantities >>> ax = graph.axis.CountingAxis >>> graph.findPlot.axisMatchesValue(ax, 'counting') True >>> graph.findPlot.axisMatchesValue(ax, 'count') True >>> graph.findPlot.axisMatchesValue(ax, 'offset') False Works on an instantiated object as well: >>> ax = graph.axis.CountingAxis() >>> graph.findPlot.axisMatchesValue(ax, 'counting') True >>> graph.findPlot.axisMatchesValue(ax, 'flute') False * Changed in v8: Must send a subclass of axis.Axis or an instance. `None` is no longer supported. TF)rQrH)rdr]vs r'rarasI0!!I  ! 7799 ! !44 " 5r)rX str | Nonec(fd}|gdgdzkrd}|tvrt|g}nt|}|s|r|cd}t}gdfdfdffD]^\}}|D]V}|||jvr|$|j|} t | |s|W_fd|D} | r"|st | d kr| S|| SgdfdfdffD]Y\} }|D]Q}|d } |jD]\} } t | |rd } n| s|RZfd |D} | r"|st | d kr| S|| Stj }dfdfdffD]\}}|t|}||||< t |d kr |d |fgS||}|r |d |fgS|d |fgS)a Returns either a list of plot classes to make if there is a predetermined class or a list of tuples where the first element of each tuple is the plot class and the second is a dict of {'x': axisXClass, 'y': axisYClass} etc. Default is pianoroll >>> graph.findPlot.getPlotsToMake() [] >>> graph.findPlot.getPlotsToMake('scatter') [, , , , ] >>> graph.findPlot.getPlotsToMake('scatter', 'offset', 'pitchClass') [] Try in wrong order: >>> graph.findPlot.getPlotsToMake('scatter', 'pitchClass', 'offset') [] Try giving just one value: >>> graph.findPlot.getPlotsToMake('scatter', 'offset') [] >>> graph.findPlot.getPlotsToMake('scatter', 'ql') # abbreviation [, ] Just one value, but it is in the wrong axis: >>> graph.findPlot.getPlotsToMake('scatter', 'pitchClass') [, ] Create a graph that does not exist: >>> graph.findPlot.getPlotsToMake('scatter', 'offset', 'dynamics') [(, OrderedDict([('x', ), ('y', )]))] Just a couple of values: >>> graph.findPlot.getPlotsToMake('offset', 'dynamics') [(, OrderedDict([('x', ), ('y', )]))] Just one value: >>> graph.findPlot.getPlotsToMake('octave') [(, OrderedDict([('x', )]))] Three values: >>> graph.findPlot.getPlotsToMake('offset', 'dynamics', 'count') [(, OrderedDict([('x', ), ('y', ), ('z', )]))] ctdfD}d}|dkrd}n|dkrd}n|dkrd}t||}|r|S|S) Ncg|]}|dS)Nr$)r%vals r'r(z9getPlotsToMake.._bestPlotType..9sPPPSqr)rPrrrmr)lenr\)graphClassesToChooseFromnumAxes bestGraphTypeinnerFilteredClassesxValueyValuezValues r' _bestPlotTypez%getPlotsToMake.._bestPlotType7s~PPVVV$<PPPQQ a<<-MM \\%MM \\'M7 G_``  ,' '+ +r)Nr xyzcg|]}|v| Sr$r$r%r{ graphRemoves r'r(z"getPlotsToMake..d#LLL!q 7K7KA7K7K7Kr)rmFTcg|]}|v| Sr$r$rs r'r(z"getPlotsToMake..zrr)r) r r\r? axesClassesr8rarqitems collections OrderedDictrc)rXrvrwrxry graphClasses axisLetterr]gc axisObjClassgraphClassesFiltered _axisLetterfoundunused_axisLetteraxisDictrdfilteredClassesrs ``` @r'getPlotsToMakersmT,,,,,,," VVV, ::! )))+K89 / <<  (K(!,ff %'' K#&-#vf !N ' ' I ' 'B //""2&&&>*5L#L)<< '""2&&& 'MLLL|LLL7  7#233q88' ' =!566 6K$'=3-#v"O ' ' Y ' 'B E35>3G3G3I3I  /!<#L)<< EE '""2&&& 'MLLL|LLL7  7#233q88' ' =!566 6&((H#&-#vf !N)) I   ))44   ( <Aa(+,,'- 55  1$Q'23 3!!_h/0 0r)ceZdZdZdZdS)Testc,td}||tjgtd}||tjgt}||tjgtd}||tjgtd}||tjgtd}||tjgtd}||tjgtd}||tjgtd}||tjgtd dd }||tj gtd d d }||tj gdS) Nrr duration quarterLengthpspitch pitchspace pitchClassrqlpcoffset) r assertEqualrWindowedAmbitus WindowedKeyHorizontalBarPitchSpaceOffsetHistogramQuarterLengthHistogramPitchSpaceHistogramPitchClassScatterPitchSpaceQuarterLengthScatterPitchClassOffsetselfposts r'testGetPlotsToMakeAzTest.testGetPlotsToMakeAsi((  45666e$$  01222  BCDDDj))  ;<===o..  ;<===d##  89:::g&&  89:::l++  89:::l++  89:::i$77  CDEEEix88  <=>>>>>r)ctd}||tjgtd}||tjgdS)Nrr )rrrDolanrs r'testGetPlotsToMakeBzTest.testGetPlotsToMakeBsVg&&  |,,,m,,  |,,,,,r)N)__name__ __module__ __qualname__rrr$r)r'rrs2???<-----r)r__main__)r*r+)r*r@)FN)N)r]r^r*r_)rdrer]r^r*rf)NNNN)rXri)!__doc__ __future__rrr3unittesttypingr9 music21.graphrrrrrrrr __annotations__r"FORMATSr?rErMrUr\rcrarTestCaserrmusic21mainTestr$r)r'rs#""""" $$$$$$# Z:  3 >>*** . -_ - - -2*:<8.>,0g1g1g1g1g1T#-#-#-#-#-8 #-#-#-L zNNNGTr)