3j LdZddlmZddlZddlZddlZddlZddlZejrddl Z ddl Z ddl mZddlmZddl mZddlmZmZmZmZddl mZej0d ZGd d ej4ZGd d eZGddeZGddeZGddeZGddeZ GddeZ!GddeZ"GddeZ#GddeZ$GddejJZ&Gd d!ejJZ'e(d"k(rddl Z e jRe&yy)#a Object definitions for graphing and plotting :class:`~music21.stream.Stream` objects. The :class:`~music21.graph.primitives.Graph` object subclasses primitive, abstract fundamental graphing archetypes using the matplotlib library. From the highest level to the lowest level of usage, ways of graphing are as follows: 1. streamObj.plot('graphName') 2. graph.plot.Class(streamObj).run() 3. plotter = graph.primitives.Class(); plotter.data = ...; plotter.process() 4. Use matplotlib directly to create your graph. ) annotationsN)common) SubConverter) environment)getExtendedModulesGraphExceptiongetColoraccidentalLabelToUnicode)prebasezgraph.primitivesc`eZdZUdZdZdZded<dZded<d Zded <d d Z d d Z d!d Z d"dZ e d#dZejd$dZd%dZ d&dZ d' d(dZ d) d*dZd+d,dZed+d-dZd.dZd.dZ d/dZd dZd.dZd0d1dZd dZ d0 d2dZy)3Grapha A music21.graph.primitives.Graph is an object that represents a visual graph or plot, automating the creation and configuration of this graph in matplotlib. It is a low-level object that most music21 users do not need to call directly; yet, as most graphs will take keyword arguments that specify the look of graphs, they are important to know about. The keyword arguments can be provided for configuration are: * doneAction (see below) * alpha (which describes how transparent elements of the graph are) * dpi * colorBackgroundData * colorBackgroundFigure * colorGrid, * title (a string) * figureSize (a tuple of two ints) * colors (a list of colors to cycle through) * tickFontSize * tickColors (a dict of 'x': '#color', 'y': '#color') * titleFontSize * labelFontSize * fontFamily * hideXGrid * hideYGrid * xTickLabelRotation * marker * markersize Graph objects do not manipulate Streams or other music21 data; they only manipulate raw data formatted for each Graph subclass, hence it is unlikely that users will call this class directly. The `doneAction` argument determines what happens after the graph has been processed. Currently, there are three options, 'write' creates a file on disk (this is the default), while 'show' opens an interactive GUI browser. The third option, None, does the processing but does not write any output. figureSize: A two-element iterable. Scales all graph components but because of matplotlib limitations (esp. on 3d graphs) no all labels scale properly. defaults to .figureSizeDefault >>> a = graph.primitives.Graph(title='a graph of some data to be given soon', tickFontSize=9) >>> a.data = [[0, 2], [1, 3]] >>> a.graphType 'genericGraph' genericGraphxytuple[str, ...]axisKeys)rztuple[int | float, ...]figureSizeDefault)alphacolorBackgroundDatacolorBackgroundFigure colorGridcolors doneActiondpi figureSize fontFamily hideXGrid hideYGrid labelFontSizemarker markersize tickColors tickFontSizetitle titleFontSizexTickLabelHorizontalAlignmentxTickLabelRotationxTickLabelVerticalAlignmentkeywordConfigurablesc t}|j|_d|_d|_d|_i|_|j D]#}i|j |<d|j |d<%d|_i|_|j D]}d|j|<d|_ d|_ d|_ d|_ d|_ d|_|j|_d|_d |_gd |_d |_d d d |_d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_d|_ |jBD]}||vstE||||y)NrangeTF皙?z#ffffffz#ddddddz Music21 Graphor)z#605c7fz#5c7f60z#988969z#628297z#ad776dz#80a364#000000r serifrcenterwrite)#rpltdatafiguresubplotaxisrgridaxisRangeHasBeenSetrrrrrr&rrr"r#rr%r$r'r!rrr r)r(r*hideLeftBottomSpines _doneAction_dataColorIndexr+setattr)selfkeywordsextmaxaxisKeykws E/DATA/.local/lib/python3.12/site-packages/music21/graph/primitives.py__init__zGraph.__init__ss!#88!% 7; 48 &( --BDIIbM%)DIIbM' "  46 }}G05D $ $W -%  #(1 *3"') 373I3I  "  "#09 *J"$"$&$$'(2:*08(*/!'.$%++BX~b(2,/,ct|dr3|j&|jj|jyyy)z] Matplotlib Figure objects need to be explicitly closed when no longer used. r9N)hasattrr9r7closerBs rH__del__z Graph.__del__s3 4 "t{{'> HHNN4;; '(? "rJc@|jj}|d=|S)zc The wrapper to matplotlib.pyplot stored as self.plt cannot be pickled/deepcopied. r7)__dict__copy)rBstates rH __getstate__zGraph.__getstate__s" ""$ %L rJcp|jj|t}|j|_yN)rQupdaterr7)rBrSrDs rH __setstate__zGraph.__setstate__s' U#!#88rJc|jS)z{ returns or sets what should happen when the graph is created (see docs above) default is 'write'. )r?rNs rHrzGraph.doneActions rJc6|dvr||_ytd|)N)showr6Nzno such done action: )r?r)rBactions rHrzGraph.doneActions& , ,%D  #8!AB BrJct|j|jt|jz}|xjdz c_|S)ah Utility function that cycles through the colors of self.colors. >>> g = graph.primitives.Graph() >>> g.colors = ['#605c7f', '#5c7f60', '#715c7f'] >>> g.nextColor() '#605c7f' >>> g.nextColor() '#5c7f60' >>> g.nextColor() '#715c7f' >>> g.nextColor() '#605c7f' )r rr@len)rBcs rH nextColorzGraph.nextColorsA& T[[!5!5DKK8H!HI J !rJc|y||jvrtd|dg}g}|D]'\}}|j||j|)||f|j|d<y)a| Set the tick-labels for a given graph or plot's axisKey (generally 'x', and 'y') with a set of pairs Pairs are iterables of positions and labels. N.B. -- both 'x' and 'y' ticks have to be set in order to get matplotlib to display either (and presumably 'z' for 3D graphs). >>> g = graph.primitives.GraphHorizontalBar() >>> g.axis['x']['ticks'] Traceback (most recent call last): KeyError: 'ticks' >>> g.axis['x'] {'range': None} >>> g.setTicks('x', [(0, 'a'), (1, 'b')]) >>> g.axis['x']['ticks'] ([0, 1], ['a', 'b']) >>> g.setTicks('m', [('a', 'b')]) Traceback (most recent call last): music21.graph.utilities.GraphException: Cannot find key 'm' in self.axis >>> g.setTicks('x', []) >>> g.axis['x']['ticks'] ([], []) NzCannot find key 'z' in self.axisticks)r;rappend)rBrFpairs positionslabelsvaluelabels rHsetTickszGraph.setTickssy> =  $)) # #4WI^!LM M !LE5   U # MM% "'0&7 '7#rJctt|}t||z }g}|dkrd}td|dz|D]}|j ||f|j ||y)a Set the ticks for an axis (usually 'y') given unsorted data. Data steps shows how many ticks to make from the data. >>> g = graph.primitives.GraphHorizontalBar() >>> g.setIntegerTicksFromData([10, 5, 3, 8, 20, 11], dataSteps=4) >>> g.axis['y']['ticks'] ([0, 5, 10, 15, 20], ['0', '5', '10', '15', '20']) TODO: should this not also use min? instead of always starting from zero? r^rN)intmaxroundr-rdrj)rB unsortedDatarF dataStepsmaxDatatickSteptickListrs rHsetIntegerTicksFromDatazGraph.setIntegerTicksFromData)sj c,'(9,-*, q=Hq'A+x0A OOQ1#K (1 gx(rJc||jvrtd||dk7r|d|dz }||z}nd}|d|z |d|zf|j|d<d|j|<y)aM Set the range for the axis for a given axis key (generally, 'x', or 'y') ValueRange is a two-element tuple of the lowest number and the highest. By default, there is a padding of 10% of the range in either direction. Set paddingFraction = 0 to eliminate this shift No such axis exists: rr^r-TN)rrr;r=)rBrF valueRangepaddingFraction totalRangeshifts rH setAxisRangezGraph.setAxisRangeCs $-- ' #8 !BC C a #AA6J0EE'1!}u'<'1!}u'<'> '7#-1  )rJc||jvrtd||r$d|j|vr|j|dry||j|d<y)Nrwri)rrr;)rBrFri conditionals rH setAxisLabelzGraph.setAxisLabel`s[ $-- ' #8 !BC C 7dii&88TYYw=OPW=X &+ '7#rJc|jjD]A\}}|dvr|s |jd|dvr|jd5td|t |j |j zD]3\}}|r|jd|dzdk(s#|jd5y) a Remove the right and top spines from the diagram. If leftBottom is True, remove the left and bottom spines as well. Spines are removed by setting their colors to 'none' and every other tick line set_visible to False. leftbottomnone)righttopzunknown spine location: Frlr^N)spinesitems set_color ValueError enumerateget_xticklinesget_yticklines set_visible)r: leftBottomlocspineilines rHhideAxisSpineszGraph.hideAxisSpineshs"....0JC((OOF+((' #;C5!ABB1!!7!7!9Gtjdt|z|j}|j t |j |jD]}|j|||jr2|j|j|j|j|jD]V}||jvr|jtj tj"d||j|X|j%||j& t)d|j&J|j&j+|j,d|j&j/|j,dy) z Apply formatting to the Subplot (Axes) container and Figure instance. ax should be an AxesSubplot object or an Axes3D object or something similar. zcalling applyFormatting on fontsizefamily)bothrr)r;rNz>Cannot apply formatting without a figure; call process() firstrr^) environLocal printDebugreprpatch set_facecolorr rrapplyFormattingToOneAxisr& set_titler'rr$ tick_paramstcastLiteral applyGridr9r set_figwidthr set_figheight)rBr:rectr; thisAxisNames rHapplyFormattingzGraph.applyFormattingsL  =W MN}} 8D$<$<=>MMD  ) )'4 8" ::   djj43E3Edoo  ^ !MML4??2   VVAII&67F|4  6* w ;; ]^ ^{{&&&   !34 !!$//!"45rJcB|jrD|j8|jd|jddt|j|jr|j jd|j r|jjdyy)zQ Apply the Grid to the subplot such that it goes below the data. NTmajor)whichcolorF)r<r set_axisbelowr r yaxisrxaxisrBr:s rHrzGraph.applyGridst 993  ! !$ ' LLWHT^^4LL M >> MM  u % >> MM  u % rJc |j|}|dvryd|vr>|d9d|zdz}t|jdk(r|dz }t||}||dd|vr9|d4t|d|zdz}||d|j|j d |vr"|d t|d|zd z}||d  t|d |zd z}t|d|zd z}t|d|zd z} d |vr2| 0|} || | | |j|j |S|d \} } ||| |dk(rL|j| |j|j |j|j|jd|S|dk(r,|j| |j|j dd|St| r| | |j|j |S#t $r d}d}d} YwxYw)al Given a matplotlib.Axes object (a subplot) and a string of 'x', 'y', or 'z', set the Axes object's xlim (or ylim or zlim or xlim3d, etc.) from self.axis[axis]['range'], Set the label from self.axis[axis]['label'], the scale, the ticks, and the ticklabels. Returns the matplotlib Axis object that has been modified rrzNr-set_lim3drirscaleget_rc ticklabelsrg{Gz)rrhorizontalalignmentverticalalignmentrotationrrrr5rrrr)r;r_rgetattrr!rAttributeErrorr%set_xticklabelsr(r*r)set_yticklabelscallable) rBr:r;thisAxis rangeFuncName thisRangeFuncsetLabelFunctiongetTickFunctionsetTickFunctionsetTickLabelFunctionrcvaluesrgs rHrzGraph.applyFormattingToOneAxissm99T?  & h 8G#4#@"TME1M4==!Q&% #G];M 8G, - h 8G#4#@&w 0GH  Xg.&*&8&8 R h 8G#4#@&w 0GH  Xg. / (%gv}w/FGO%gv}w/FGO#*7FTML4P#Q  ( "';'G#%E E " *.*;*;(, 981&g.NFF*'s{''151B1B/3<@<^<^:>:Z:Z151H1H*/ (1(''151B1B/3rcallDoneActionrBrDr7s rHprocessz Graph.processs"#hh jj4+E+EjF {{..q!Q7   4<<( DLLT5N5NO T\\* rJcy)zI Calls the subclass specific information to get the data Nrs rHrzGraph.renderSubplot*s rJNc|jdk(r|jy|jdk(r|j|y|jyy)zI Implement the desired doneAction, after data processing r[r6N)rr[r6)rBfps rHrzGraph.callDoneAction1sB ??f $ IIK __ ' JJrN __ $ %rJc|j td|jJ|jjy)z Calls the show() method of the matplotlib plot. For most matplotlib back ends, this will open a GUI window with the desired graph. Nz8Cannot show graph without a figure; call process() first)r9rr[rNs rHr[z Graph.show<s: ;; WX X{{&&& rJcr|j td|jJ|tjd}|j}|d}|jj ||t js/tjtj|dy|jS)za Writes the graph to a file. If no file path is given, a temporary file is used. Nz9Cannot write graph without a figure; call process() first.png,)rpng)fmt) r9rr getTempFilersavefigrrunningInNotebookrlaunchpathlibPath)rBrrs rHr6z Graph.writeGs ;; XY Y{{&&& :))&1Bhh ;C B!$  % '') N ! !',,r"2 ! >;; rJreturnNone)rdict)rSrrr)r str | None)r\rrr)rstr)rFrrez%t.Iterable[t.Sequence[object]] | Nonerr)r)rpzt.Sequence[int | float]rFrrqrmrr)皙?)rFrrxztuple[int | float, int | float]ryfloatrr)F)rFrrirr~boolrr)r:matplotlib.axes.Axesrrrrr:rrr)r:rr;rrz dict | NonerV)rrrr)rzstr | pathlib.Path | Nonerzmatplotlib.figure.Figure | None)__name__ __module__ __qualname____doc__ graphTyper__annotations__rr+rIrOrTrXpropertyrsetterrarjrur|r staticmethodrrrrrrrr[r6rrJrHr r 2so4jI *Ho*/5,5-/0;0z(   CC .,8,8=,8BF,8`0312).E)),),/)8<):/21!1!@1',17;1:,((6(6\ & K*>K'*K/:KZ6   /3+2rJr cpeZdZUdZddiZded<dZdZejdzZ d ed <d fd Z dd Z xZ S)GraphColorGrida Grid of discrete colored "blocks" to visualize results of a windowed analysis routine. Data is provided as a list of lists of colors, where colors are specified as a hex triplet, or the common HTML color codes, and based on analysis-specific mapping of colors to results. >>> #_DOCS_SHOW g = graph.primitives.GraphColorGrid() >>> g = graph.primitives.GraphColorGrid(doneAction=None) #_DOCS_HIDE >>> data = [['#55FF00', '#9b0000', '#009b00'], ... ['#FFD600', '#FF5600'], ... ['#201a2b', '#8f73bf', '#a080d5', '#403355', '#999999']] >>> g.data = data >>> g.process() .. image:: images/GraphColorGrid.* :width: 600 r>:bool to hide the left and bottom axis spines; default Truedict[str, str] _DOC_ATTRr) rr>rr+c 2d|_t|di|y)NTr)r>superrIrBrC __class__s rHrIzGraphColorGrid.__init__}s$(! $8$rJc |j td|jJ|j td|jJ|jjdt |j}t |D]}|j|}g}g}g}t |D];\}} |j|dz|j| |jd=|jj|d||z } | j||d|ddd | jjD]8\} } | jd| jd| jd:t | j| j!zD]\}} | j#d | j%g| j'g| j)d gt | j+z| j-d gt | j/z| j1d t |j|f| j3j#d | j5j#d |d kDr|jjd n|jjdd}|j7d|d d |_y)NxA))A,KIGI )+ 6 9  U+  +q! !7((1hlCB FF9" &$  &&(YY__%6! E##C( *" &7%R%6%6%82;L;L;N%NO4  'P MM"  MM"    tc"*<*<*>&?? @   tc"*<*<*>&?? @ KKC ! -. / LLN & &u - LLN & &u -]!d b= KK ' 'q ' 1 KK ' 's ' 3! #/3 rJrr) rrrrrrrrr r+rIr __classcell__rs@rHrrcsM$  \!I~I,1,F,FIb,b/b%PrJrceZdZUdZddiZded<dZdZejdzZ d ed <dfd Z dd Z dd Z xZ S)GraphColorGridLegenda Grid of discrete colored "blocks" where each block can be labeled Data is provided as a list of lists of colors, where colors are specified as a hex triplet, or the common HTML color codes, and based on analysis-specific mapping of colors to results. >>> #_DOCS_SHOW g = graph.primitives.GraphColorGridLegend() >>> g = graph.primitives.GraphColorGridLegend(doneAction=None) #_DOCS_HIDE >>> data = [ ... ('Major', [('C#', '#00aa55'), ('D-', '#5600ff'), ('G#', '#2b00ff')]), ... ('Minor', [('C#', '#004600'), ('D-', '#00009b'), ('G#', '#00009b')]) ... ] >>> g.data = data >>> g.process() .. image:: images/GraphColorGridLegend.* :width: 600 r>rrrcolorGridLegend)?rrr+c Jd|_t|di|d|vrd|_yy)NTr&Legendr)r>rrIr&rs rHrIzGraphColorGridLegend.__init__s/$(! $8$ ( "!DJ #rJc|j td|jJ|j td|jJt|jD]-\}}|d}|d}|j |j|||/|j ddd|j |jz}t|D]\}}|jd|jjdd d |jd d |jdd |jd g|jdgy)Nrrrr^rrFr+g?r.)rrrrr) r9rr8rmakeOneRowOfGraphr|rrrrrrj) rBr:rrowLabelAndDatarowLabelrowData allTickLinesr!rs rHrz"GraphColorGridLegend.renderSubplotsC ;; [\ \{{&&& 99 FG Gyy$$$"+DII"6 A&q)H%a(G  " "4;;8W E#7 #vq)--/'2H2H2JJ  .GAt   U #/ ##3D#E #r" #r" c2 c2rJcL|j td|jJg}g}g}t|D]>\}} |jd|z|j| d|jd@t |jd|dzf} |j | } d} | j ||| |dd| jjD]$} | jd| jd&| j| jz}t|D]\}}|jd| jdg| j|g|j |j"d d | j%t't |Dcgc]}|dz c}|Dcgc]\}}t)|}}}| j+||j |j"d d d | j-dt |dzf| Scc}wcc}}w)a Makes a subplot for one row of data (such as for the Major label) and returns a matplotlib.axes.AxesSubplot instance representing the subplot. Here we create an axis with a part of Scriabin's mapping of colors to keys in Prometheus: The Poem of Fire. >>> import matplotlib.pyplot >>> colorLegend = graph.primitives.GraphColorGridLegend() >>> rowData = [('C', '#ff0000'), ('G', '#ff8800'), ('D', '#ffff00'), ... ('A', '#00ff00'), ('E', '#4444ff')] >>> colorLegend.data = [['Scriabin Mapping', rowData]] >>> fig = matplotlib.pyplot.figure() >>> subplot = colorLegend.makeOneRowOfGraph(fig, 0, 'Scriabin Mapping', rowData) >>> subplot zOCannot create row of graph without data; set self.data before calling process()?r^r r1)rr r Fr rr5rgٿ)rrrrr)r8rrrdr_rrrrrrrrrrrr%rrr-r rr)rBr9rowIndexr1r2rfrr r!oneColorMapping posTriplerEwidth spineArtistr3rrunused_ysubstitutedAccidentalLabelss rHr/z&GraphColorGridLegend.makeOneRowOfGraphs<2 99 FG Gyy$$$  "+G"4 A   S1W %   _Q/ 0 NN1 #5 ^Q1 5 V   + y'5 ST]^99++-K  % %c *  ! !) ,. ((*R->->-@@  .GAt   U #/ se H:$($5$5"&///6-5  7 eCL&9:&9q1u&9:;;B'C:A;1h(@'B:A $'C  '&&?? (&   S#g,,-. ;'Cs  H7H rr) r9zmatplotlib.figure.Figurer6rmr1rr2zlist[tuple[str, str]]rr)rrrrrrrrr r+rIrr/r%r&s@rHr(r(s(  \!I~"I ,1,F,FIb,b/b"R UrJr(ceZdZUdZdddZded<dZdZejdzZ d ed <dfd Z e d Z dd Z xZS)GraphHorizontalBara Numerous horizontal bars in discrete channels, where bars can be incomplete and/or overlap. Data provided is a list of pairs, where the first value becomes the key, the second value is a list of x-start, x-length values. Note how the second element in each data point is the length, so subtracting death year from birth year gives the appropriate length. Example: Plot the life-span of four composers whose lives were intertwined: Chopin, Robert and Clara Schumann, and Brahms. >>> a = graph.primitives.GraphHorizontalBar() >>> a.doneAction = None #_DOCS_HIDE >>> data = [('Chopin', [(1810, 1849-1810)]), ... ('Schumanns', [(1810, 1856-1810), (1819, 1896-1819)]), ... ('Brahms', [(1833, 1897-1833)])] >>> a.data = data >>> a.process() .. image:: images/GraphHorizontalBar.* :width: 600 Data is a list of tuples in the form, where each entry represents a space on the Y axis: * Label * List of tuples of numeric data where each tuple has two or three elements: * Start x-position * Length of bar * Optional: dictionary of format information about this point. * Optional: dictionary of format information for all points at this level. (this will be overridden by any information for the particular point) To make an equally spaced plot, like in a Pitch Space plot, leave empty data in the form: `('', [], {})` 2Amount of vertical space each bar takes; default 8zj Vertical space above and below the bars, default 2 (= total4 space between bars) barSpacemarginrr horizontalBarr3rr+c Xd|_d|_t| di|d|vrd|_yy)Nrrlr333333?rrArBrrIrrs rHrIzGraphHorizontalBar.__init__s5   $8$ ( "DJ #rJc:|j|jdzz SNrlr@rNs rH barHeightzGraphHorizontalBar.barHeight}} a00rJcP|j td|jJ|j td|jJ|jjdd}t }g}g}g}d}|jD]k}t |dk(r|\} } i} n|\} } } |j | | jd|j} | rt | ddk(} | r| }| }n4| Dcgc]}|djd| }}| Dcgc]}|dd }}||jz|jf}|j||||j| D]W}t |dk\r |dd\}}ntd t |d |||z}||fD]}||vs|j|Y|j ||jd zz| g||jz }|d z }nd }d }|rt|}t!|}||z }|j#ddt ||jzf|j#d||f|j%d|d|j&dvr|j&ddst)|t+|dz z}|dk(rd }t-t)t/j0|t+t/j2||D]}|j ||g|j%d|yyycc}wcc}w)Nrrrr rrlr) facecolorsrz"Points must be length 2 or 3, not z: r r^rrrcr3)r9rr8rsetr_rdgetrarBrK broken_barhraddrAminrnr|rjr;rmror-mathfloorceil)rBr:yPosxPointsyTicksxTickskeysrinfokeypointsfullRowFormatDict faceColoruniformFormatPerRow rowFaceColorspositionPointspyRangexStartxLenxEndrxMinxMaxxRange rangeSteps rHrz GraphHorizontalBar.renderSubplotsO ;; [\ \{{&&& 99 FG Gyy$$$ ###."%% IID4yA~" V$&!15.V. KK *--gt~~7GHI'*6!9~':#&$-M%+NKQ$R6aQqTXXgy%A6M$R5;% #d|, c6" diin $TYYs^G-DD5"#556IA~ 3tzz$/0 41$& qQCk*& MM#v &.E $O%S%r>nsl'RI!I~ I,1,F,FJ-/ 11S'rJr>ceZdZUdZdddZded<dZdZejdzZ d ed <dfd Z e d Z dd Z xZS)GraphHorizontalBarWeightedz Numerous horizontal bars in discrete channels, where bars can be incomplete and/or overlap, and can have different heights and colors within their respective channel. r?z Space around the bars, default 2r@rrhorizontalBarWeightedrDrr+c Xd|_d|_t| di|d|vrd|_yy)Nr?rr^rrHrs rHrIz#GraphHorizontalBarWeighted.__init__s7 !  $8$ ( "DJ #rJc:|j|jdzz SrJr@rNs rHrKz$GraphHorizontalBarWeighted.barHeight.rLrJc F|j td|jJ|j td|jJ|jjdd}g}g}g}d}t |j|_|jj |jD]\}}|j |g} g} g} g} t|D]h\}} d}d}d}|j}|j}d}t| dk(r| \}}}nDt| dk(r| \}}}}n.t| d k(r | \}}}}}nt| d k(r | \}}}}}}t|}|td | |J| j ||f| j || j |||vr|j |||z|vr|j ||z|j|z}|j|z d z}|jd|z zd z}||jz|z||zz}| j ||fkt| D](\}}|j|g| || || || | *|j ||jd zz|g||jz }|dz }t!|}t#|}|j%ddt||jzfd|j%d||fd|j'd|y)Nrrrr rr^rrEr*rzLData point must have at least 3 elements (x, span, heightScalar, ...); got: r )rNrr r皙?)ryrg{Gz?)r9rr8rlistreverserdrrarr_r rKrBrRrArTrnr|rj)rBr:rXrYrZr\rr^r_xRangesyRangesalphasrr8rspan heightScalarrryShifthyAdjust yShiftUnit adjustedYrlrjrks rHrz(GraphHorizontalBarWeighted.renderSubplot2s` ;; [\ \{{&&& 99 FG Gyy$$$ ###. O  99KC KK GGFF$V,4 ( t9>,0)At\Y!^370At\5Y!^:>7At\5%Y!^BF?At\5% <$(KKO&&RSS'''4y) e$ e$G#NN1%HW,NN1t8, NN\1>>A-4!^^q>> g = graph.primitives.GraphScatterWeighted() >>> g.doneAction = None #_DOCS_HIDE >>> data = [(23, 15, 234), (10, 23, 12), (4, 23, 5), (15, 18, 120)] >>> g.data = data >>> g.process() .. image:: images/GraphScatterWeighted.* :width: 600 z1the maximum diameter of any ellipse, default 1.25z1the minimum diameter of any ellipse, default 0.25 maxDiameter minDiameterrrscatterWeighted)r*r*rr+c Xd|_d|_t| di|d|vrd|_yy)Ng?rsrrGr)rrrrIrrs rHrIzGraphScatterWeighted.__init__s7"&"& $8$ ( "DJ #rJc4|j|jz SrVrrNs rH rangeDiameterz"GraphScatterWeighted.rangeDiameters$"2"222rJc X|j td|jJ|j td|jJt}|j}|jj dd|jDcgc]}|d }}|jDcgc]}|d }}|jDcgc]}|d }}g}|jD])}t |dkDr|d} ni} |j| +t|} t|} t| | z } t|} t|}t| |z }t|}t|}t||z }d }d }| |kDr|| z }n || kDr| |z }g}|D]Z}|dk(r|jddg|dk7r ||z |z }nd }|j|j|zz}|j||g\tt |jD]}||}||}||\}}||} ||z}||z}|jd||f||d | }|j||j!|j"|j%|j&|j)|j+||dkDs||d zd |zzz} |d z}!|j-| |!t/||dddd|j1d|| f|j1d| | fycc}wcc}wcc}w)Nrrrrrr^rlrr5r )xyr9heightrvrrbaseliner)sizevahamultialignmentrrr)r9rr8rpatchesrr_rdrnrTrrrr-Ellipse add_artist set_clip_boxbboxrrrratextrr|)"rBr:rDrdxListyListzListformatDictList formatDictrkrjrlyMaxyMinrfzMaxzMinzRangexDistortyDistortzNormrscalarscaledrrrunused_zScalarr9re adjustedXrs" rHrz"GraphScatterWeighted.renderSubplotsX ;; [\ \{{&&& 99 FG Gyy$$$!#,, ##d#; $yy)y!1y)#yy)y!1y)#yy)y!1y)A1vzqT   ! !* - 5z5ztd{#5z5ztd{#5z5ztd{#  F?H f_HAAv aV$Q;$h&0F F))T-?-?&-HI ff-.s499~&AaAaA %a A~'*JLE\FTAq6vTTA   q ! NN7<< ( KK # OODNN, - Qx!| %#+$/!BC H  Y& q]"# * &,2 49'H #d|, #d|,k*))s L( L" L'rr)rrrrrrrrr r+rIrrrr%r&s@rHrrsi KJ!I~ "I,1,F,FJ&-&/&33d-rJrceZdZdZdZddZy) GraphScatteraP Graph two parameters in a scatter plot. Data representation is a list of points of values. >>> g = graph.primitives.GraphScatter() >>> g.doneAction = None #_DOCS_HIDE >>> data = [(x, x * x) for x in range(50)] >>> g.data = data >>> g.process() .. image:: images/GraphScatter.* :width: 600 scatterc l|j td|jJ|j td|jJ|jjdg}g}d}|jD]}t |dkr t d|d}|d}|j ||j ||j}|j} |j} |j} t |d k\r)|d} d | vr| d } d | vr| d }d | vr| d } d | vr| d } |j|||| | | |dz }|jds&|jdt|t|f|jds'|jdt|t|fyy)Nrrrr rrlz1Need at least two points for a graph data object!r^rrr"rr#)r"rrr#rr)r9rr8rr_rrdr"rarr#plotr=r|rTrn) rBr:xValuesyValuesrrowrrr"rrr# displayDatas rHrzGraphScatter.renderSubplot:s ;; [\ \{{&&& 99 FG Gyy$$$ ###. 99C3x!|$%XYYAAAA NN1  NN1 [[FNN$EJJEJ3x1}!!f k)'0E{*(2Fk)'0E;.!,\!:J LLAfES]L ^ FA/2'',   cCL#g,#? @'',   cCL#g,#? @-rJNr)rrrrrrrrJrHrr*s I*ArJrcteZdZUdZddiZded<dZejdzZded <d d d dfd Z dd Z xZ S)GraphHistograma Graph the count of a single element. Data set is simply a list of x and y pairs, where there is only one of each x value, and y value is the count or magnitude of that value >>> import random >>> g = graph.primitives.GraphHistogram() >>> g.doneAction = None #_DOCS_HIDE >>> g.graphType 'histogram' >>> data = [(x, random.choice(range(30))) for x in range(50)] >>> g.data = data >>> g.process() .. image:: images/GraphHistogram.* :width: 600 binWidthz Size of each bin; if the bins are equally spaced at intervals of 1, then 0.8 is a good default to allow a little space. 1.0 will give no space. rr histogram)rrr+皙?)rrc 6||_t|dd|i|y)Nrr)rrrI)rBrrrCrs rHrIzGraphHistogram.__init__s   1u11rJc|j td|jJ|j td|jJ|jjdg}g}|j}t |j d}|j}|jD]>}t|dkDr|\}} } n|\}} |j||j| @|j|||||y)Nrrrr rrlr9rr) r9rr8rrr rrr_rdr) rBr:rrrrrpointabunused_formatDicts rHrzGraphHistogram.renderSubplots ;; [\ \{{&&& 99 FG Gyy$$$ ###.  ==Q( YYE5zA~*/'1'1 HHQK HHQK   AqU CrJ)rrrrr) rrrrrrrr r+rIrr%r&s@rHrrgsM* !I~I,1,F,F,V/V,/2DrJrcbeZdZUdZdZej dzZded<d fd Zd dZ d dZ xZ S) GraphGroupedVerticalBara Graph the count of one or more elements in vertical bars. The `.data` is simply a list of x and y pairs, where there is only one of each x value, and y value is a list of values. >>> from collections import OrderedDict >>> g = graph.primitives.GraphGroupedVerticalBar() >>> g.doneAction = None #_DOCS_HIDE >>> lengths = OrderedDict( [('a', 3), ('b', 2), ('c', 1)] ) >>> data = [('bar' + str(x), lengths) for x in range(3)] >>> data [('bar0', OrderedDict([('a', 3), ('b', 2), ('c', 1)])), ('bar1', OrderedDict([('a', 3), ('b', 2), ('c', 1)])), ('bar2', OrderedDict([('a', 3), ('b', 2), ('c', 1)]))] >>> g.data = data >>> g.process() groupedVerticalBar)r roundDigitsgroupLabelHeightrr+c Nd|_d|_d|_t|di|y)Nr^rOr)rrrrrIrs rHrIz GraphGroupedVerticalBar.__init__s*%& !'* $8$rJc  |D]}|j|jdz z}|j}|j||t t ||j dd|j|jy)Nrlr5r)rrrr) get_x get_width get_heightrrrorr%r)rBr:rectsrrrs rH labelBarsz!GraphGroupedVerticalBar.labelBarssuD (81(<=I__&F LLU64+;+;<=$$"&"3"3 $  1rJc  |j td|jJt}|j}d}g}|jD]*\}}t |}t |j }n|j}||z } g} g} t|jD]R\} \}}| j| | jt |j D cgc]} ||  c} Tg}t|D]} g}t| D]\}}|j| || g}| D]}|j|| | zz|j||| d|j}|j|g}|D](}|j|||j|d*|jj|j |j"}|j%|||ycc} w)Nrr^rrr)rr)prop)r8rr matplotlibr_sortedr\rrrdr-rrar font_managerFontPropertiesr%rlegend)rBr:rDr barsPerGroup subLabelsunused_arr widthShiftxValsyBundlesrr^ryValsr!_x xValsShiftedrrr fontPropss rHrz%GraphGroupedVerticalBar.renderSubplots 99 FG Gyy$$$!#__    99KHaq6Lqvvx(I  % == ,  )$)) 4 A}! LLO OOvaffh/?@/?QsV/?@ A!5 |$AE"5)2 Xa[^,*L##Aa$89;;|$%/%(%)^^%5 7D LL %"D NN7D ) MM$q' "++::@Q@QBF//;S vyy97As# H rr) rrrrrr r+rrIrrr%r&s@rHrrs<$%I,1,F,FJ8-8/8% 15:rJrcPeZdZdZdZdZdddd d fdZd dZd d ZxZ S) Graph3DBarsa  Graph multiple parallel bar graphs in 3D. Data definition: A list of lists where the inner list of (x, y, z) coordinates. For instance, a graph where the x values increase (left to right), the y values increase in a step pattern (front to back), and the z values decrease (top to bottom): >>> g = graph.primitives.Graph3DBars() >>> g.doneAction = None #_DOCS_HIDE >>> data = [] >>> for i in range(1, 10 + 1): ... q = [i, i//2, 10 - i] ... data.append(q) >>> g.data = data >>> g.process() When creating the graph these keyword arguments can be used (in addition to any such as `colors` that all Graph classes can use). * ``alpha`` -- opacity of each bar, from 0.0 (transparent) to 1.0 (opaque). Defaults to 0.8. * ``barWidth`` -- width of each bar along the x-axis. When ``None`` (default) the width is computed automatically. Individual data points may also override this via a ``'barWidth'`` key in their per-point format dict. * ``barDepth`` -- depth of each bar along the y-axis. When ``None`` (default) the depth is computed automatically. Individual data points may also override this via a ``'barDepth'`` key in their per-point format dict. 3DBarsrrN)rbarWidthbarDepthc `t|dd|i|||_||_d|vr gd|_yy)Nrr)z#ff0000z#00ff00z#6666ffr)rrIrrr)rBrrrrCrs rHrIzGraph3DBars.__init__0s< 1u11(0 (0 8 #;DK $rJc.t}|j}|j|_|jjdddd|_|j |j|j |j|jy)Nr^r) projection)rr7r9rr:rrrrs rHrzGraph3DBars.process<so!#hhjjl {{..q!Q4.H  4<<( T\\* rJc |j td|jJg}g}g}|jD]D}|d|d|d}}}|j||j||j|Ft|t |} } t|t |} } t|t |}} |j dd| | f|j dd<|j dd| |f|j dd<|j dd| | f|j dd<|j |j n| | z d z }|j |jn| | z d z }|jD]}t|d k(r |\}}}i}n!t|d k(r|\}}}}n td d |vr|d }n|j}d|vr|d}d|vr|d}|j||dz z ||dz z d|||||j|jddd|jddd|jdddy)Nrrr^rlrr-rrrrEz/Cannot plot a point unless it has 3 or 4 valuesrrr)rrT)r~)r8rrdrTrnr;rrr_rrabar3drr)rBr:rrzValsrrrrrjrkrrrrrr dataPointrrs rHrzGraph3DBars.renderSubplotHs 99 FG Gyy$$$YYEAha%(!qA LLO LLO LLO  ZUdZUdZUd 99S>' " *&*DjDIIcN7 # 99S>' " *&*DjDIIcN7 # 99S>' " *&*DjDIIcN7 #$(MM$=4==D4KSUCU$(MM$=4==D4KSUCUI9~"#1a Y1$&/#1a$%VWW*$"7+(Z'%j1Z'%j1 MM!x!|,a8a<.@!"Ha % $   ,'#0 #s5 #s5 #s5rJ)rrrfloat | int | Nonerrrr) rrrrrrrIrrr%r&s@rHrr sK"FIH!$,0,0 < <* <* < 76rJrceZdZdZy)Testc2ddlm}||ty)Nr) testCopyAll)music21.test.commonTestrglobals)rBrs rHtestCopyAndDeepcopyzTest.testCopyAndDeepcopys7D')$rJN)rrrrrrJrHrrs%rJrcBeZdZdZdZdZdZdZdZdZ d d Z d Z y) TestExternalTc vtddd}tdDcgc] }|||zf }}||_|jt dd}tdDcgc]"}|t j tdf$}}||_|jtddd d d g }g}tdd D]T}tdDcgc]-}|t j td|zd|dzz|f/}}|j|V||_|j~ycc}wcc}wcc}w)Nzx to x*xr^)rr&r2z50 x with random(30) y counts)rr&z,50 x with random values increase by 10 per xrrg)rr&rrrEr3) rr-r8rrrandomchoicerextend)rBrrr8rqs rH testBasiczTestExternal.testBasics$ D ! D$)"I.IqAE I. d2Q R7?CYAW KKN )/ BXsD,'D1 2D6cJg}dD]y}g}tdD]T}tjtd}|tjtdz}|j||fV|j||g{t d}||_|j y)N) rrr`rrfrrrr!r3rr)r-rrrdr>r8r)rBr8rir_rstartendrs rHtestBrokenHorizontalz!TestExternal.testBrokenHorizontalsGEF2Y eCj1fmmE"I66 ucl+ KK ( H $ / rJcng}tdD]q}tjtd}tjtd}tjtdd}|j|||gs|jrd}nd}t |}||_|jy)Nrrr^r6r)r-rrrdr[rr8r)rBr8rrrrrrs rHtestScatterWeightedz TestExternal.testScatterWeightedsrA eBi(A eBi(A eArl+A KKAq "  99 JJ J 7 rJc Fg}tddD]P}tdDcgc])}|tjtd|dzz|f+}}|j|Rtddgfddd gfd d gfgft gd ft td Dcgc] }|||zf c}fttd Dcgc]"}|tjtdf$c}ft|ftdgdfdgdfgftgdddggdgfg}|D]\}}|d}||_ |j|jjdz} tt j#| z } t j%d| g|j'| ycc}wcc}wcc}w)z Write a graphic file for all graphs, naming them after the appropriate class. This is used to generate documentation samples. r^rErr3Chopin)' Schumanns)r .)iMBrahms)i)@)))r3rr2)rErr*)rxrrMajor))Cz#00AA55)Dz#5600FF)Gz#2B00FFMinor))rz#004600)rz#00009b)rz#00009B)z#8968CDz#96CDCDz#CD4F39z#FFD600z#FF5600)z#201a2bz#8f73bfz#a080d5z#6495EDz#FF83FANrr writing fp:)r-rrrr>rrrrr(rr8rrrrrgetRootTempDirrr6) rBdata3DPolygonBarsrrr graphClassesgraphClassNamer8objfnrs rHx_test_writeAllGraphsz"TestExternal.x_test_writeAllGraphssq!AEJ2YOY!V]]5q1u#67;YAO  $ $Q ' -./02EFG-./1  " E G "') ,)Qq!a%j) , . 5:2Y ?Yq&--b *+Y ? A + , !NONOS ?()4U  # 2%1 ND D1CCH KKM''&0B\002R78B  # #]B$7 8 IIbM%1=P- ?s.FF 4'F ctd}gdgdgdgdgdgdg}||_|j|jjdz}t t j|z }|j|y) Nr)#525252#5f5f5f#797979#858585#727272#6c6c6c#8c8c8cr*r)#999999r+r&r)r%r$z#464646#3f3f3fr,#4c4c4cr-r&r&r-r-r$r%r&r'r(r))r+r+r+r+r+r+r+r+r+r+r+r&r)r%r%r'r&r&r&r&r&r&r'#929292r+)r+r+r+r+r+r+r+r+r+r+r+r+r*r*r*r'r&r'r.r+)r+r+r+r+r+r+r+r+r+r+r+r+r*r.r+) r+r+r+r+r+r+r+r+r+r+)r+r+r+r+r+r) rr8rrrrrrr6)rBrr8r!rs rHx_test_writeGraphColorGridz'TestExternal.x_test_writeGraphColorGrids{ d +S ^'2=G%I& [[ ! !F * ,,.3 4  rJc ,g}td}tdDcgc] }|||zf }}||_|j|dgtddd}tdDcgc] }|||zf }}||_|j|dgt d}tdDcgc]"}|t j td f$}}||_|j|d gtd}g}tdd D]P}td Dcgc])}|t j td |dzz|f+}}|j|R||_|j|dgtddddgd}||_|j|dg|D]b\}} |j| dz} ttj| z } tjd| g|j| dycc}wcc}wcc}wcc}w)z6 Write graphing examples for the docs Nrrz graphing-01zExponential Graphr^)r&rrz graphing-02rz graphing-03rErr3z graphing-04z Random Datarr.)rrr)r&rrrrz graphing-05rr)rr-r8rdrrrrrrrrrrr6) rBpostrrr8rrrr namer!rs rHx_test_writeGraphingDocsz%TestExternal.x_test_writeGraphingDocss D )$)"I.IqAE I. Q &' 2! M$)"I.IqAE I. Q &' d +7rprrrrrTestCaserrrmainTestrrJrHrLs; #  ??8'' &{&&'9: nG " "nbnUnbW5WtW'W'tE#E#hK-5K-\:A5:Az<DU<D~`:e`:Fv6%v6r%8  %A8$$AH zGTrJ