U
\ ¸aÿ$ ã @ sv d Z ddlZddlmZmZmZmZmZ ddl
m Z ddl m
Z
dddd d
d œZd d
d
dddœZejjZddd„ZdS )zHModule that provides a GUI-based editor for Matplotlib's figure options.é N)ÚcbookÚcmÚcolorsÚmarkersÚimage)ÚQtGui)Ú _formlayoutZSolidZDashedZDashDotZDottedÚNone)ú-z--z-.ú:r ZDefaultz Steps (Pre)z Steps (Mid)z Steps (Post))Údefaultz steps-preZstepsz steps-midz
steps-postc s¬ d}t tˆ ¡ ƒ\}}t tˆ ¡ ƒ\}}dˆ ¡ f|dd|fd|fdˆ ¡ fdˆ ¡ dd d
gf|d d |fd
|fdˆ ¡ fdˆ ¡ dd d
gf|dg}ˆ j j
‰ˆ j j
‰ ˆ j ¡ ‰ˆ j ¡ ‰
dd„ }i ‰ˆ
¡ D ] } | ¡ }
|
dkràqÊ| ˆ|
< qÊg } dd„ } tˆ|d‰ˆD ]à}
ˆ|
} tjt | ¡ | ¡ ¡dd}
tjt | ¡ | ¡ ¡dd}tjt | ¡ | ¡ ¡dd}d|
f|dd| t| ¡ ƒfd| t| ¡ ƒfd| ¡ fd|
f|d d | t | ¡ ƒfd | ¡ fd |fd |fg
}| ||
d!g¡ qt | ƒ‰i ‰ˆ j!ˆ j"•D ]2}| ¡ }
|
dks | #¡ d"kr(q |ˆ|
< q tˆ|d‰g }d#d$„ tt$j% &¡ ƒD ƒ}ˆD ]®}
ˆ|
}| '¡ }|t$j% (¡ kr”||j)ff|•}| *¡ \}}d|
fd%|j)g| fd&|fd'|fg}t+|d(ƒrüd)d$„ tt,j-ƒD ƒ}| d*| .¡ f|•f¡ | ||
d!g¡ q`t |ƒ‰|d+d!fg}| r:| | d,d!f¡ |rP| |d-d!f¡ ‡ ‡‡‡‡‡‡‡‡‡ ‡
f d.d/„}t/j0|d0|t1 2t3t4 5d1d2¡ƒ¡|d3}|d"k r¨||ƒ d"S )4z Edit matplotlib figure options)NNZTitle)Nz
X-AxisZLeftZRightZLabelZScaleZlinearÚlogZlogit)Nz
Y-AxisZBottomZTop)z (Re-)Generate automatic legendFc S s4 t d| ¡}|r(| d¡t| d¡ƒfS | dfS d S )Nz(_line|_image)(\d+)é é r )ÚreÚmatchÚgroupÚint)Úlabelr © r úO/tmp/pip-unpacked-wheel-wjyw_3jo/matplotlib/backends/qt_editor/figureoptions.pyÚcmp_key= s z figure_edit..cmp_keyZ
_nolegend_c S sb || kr| |t |ƒi–} dd„ | ¡ D ƒ}dd„ | ¡ D ƒ}|| | }|gt| ¡ dd„ d S )aâ
Prepare entry for FormLayout.
*d* is a mapping of shorthands to style names (a single style may
have multiple shorthands, in particular the shorthands `None`,
`"None"`, `"none"` and `""` are synonyms); *init* is one shorthand
of the initial style.
This function returns an list suitable for initializing a
FormLayout combobox, namely `[initial_name, (shorthand,
style_name), (shorthand, style_name), ...]`.
c S s i | ]\}}||“qS r r )Ú.0ÚshortÚnamer r r Ú
^ s z5figure_edit..prepare_data..c S s i | ]\}}||“qS r r )r r r r r r r ` s c S s | d S )Nr r )Zshort_and_namer r r Úf ó z3figure_edit..prepare_data..©Úkey)ÚstrÚitemsÚsorted)ÚdÚinitZ
name2shortZ
short2nameZcanonical_initr r r Ú prepare_dataM s
ÿÿz!figure_edit..prepare_datar T)Z
keep_alpha)Nz Linez
Line stylez
Draw styleZWidthz Color (RGBA))Nz
MarkerÚStyleÚSizezFace color (RGBA)zEdge color (RGBA)Ú Nc S s g | ]\}}||f‘qS r r )r r Úcmapr r r Ú
s z figure_edit..ZColormapz
Min. valuez
Max. valueÚget_interpolationc S s g | ] }||f‘qS r r )r r r r r r* s Ú
InterpolationZAxesZCurvesz Images, etc.c (
sÈ ˆ ¡ }ˆ ¡ }| d¡}ˆr(| d¡ng }ˆr:| d¡ng }| rJtdƒ‚|\
}}}} }
} } }
}}ˆ ¡ |
krxˆ |
¡ ˆ ¡ |krŽˆ |¡ ˆ |¡ ˆ ||¡ ˆ
| ¡ ˆ | | ¡ ˆ |
¡ ˆˆ j
_ˆ ˆ j_ˆ j
ˆ¡ ˆ j ˆ
¡ ˆ j
¡ ˆ j ¡ t|ƒD ]¤\}}ˆˆ| }|\ }}}}}}}}}| |¡ | |¡ | |¡ | |¡ t |¡} | d¡ | | ¡ |dkr| |¡ | |¡ | |¡ | |¡ qt|ƒD ]„\}} ˆˆ| } t | ƒdkrò| \}} } }!}"| |"¡ nt | ƒdkr | \}} } }!| |¡ | !t" #| ¡¡ | j$t%| |!gƒŽ q¶|rŠd}#d}$ˆ j&dk rnˆ '¡ }%|%j(dk }#|%j)}$ˆ j*|$d}&|&rŠ|& +|#¡ ˆ ,¡ }'|'j- .¡ ˆ ¡ |kr¸ˆ ¡ |ksÄ|'j-j/ 0¡ dS ) z A callback to apply changes.r zUnexpected fieldNÚnoneé é r )Úncol)1Úget_xlimÚget_ylimÚpopÚ
ValueErrorÚ
get_xscaleZ
set_xscaleÚ
get_yscaleZ
set_yscaleÚ set_titleZset_xlimZ
set_xlabelZset_ylimZ
set_ylabelÚxaxisÚ converterÚyaxisZ set_unitsZ_update_axisinfoÚ enumerateZ set_labelZ
set_linestyleZ
set_drawstyleZ
set_linewidthÚmcolorsÚto_rgbaZ set_alphaZ set_colorZ
set_markerZset_markersizeZset_markerfacecolorZset_markeredgecolorÚlenZset_interpolationZset_cmapr Úget_cmapZset_climr" Zlegend_Z
get_legendZ
_draggableZ_ncolZlegendZ
set_draggableZ
get_figureZcanvasZdrawZtoolbarZ push_current)(ÚdataZ orig_xlimZ orig_ylimÚgeneralÚcurvesÚ mappablesÚtitleÚxminÚxmaxZxlabelZxscaleÚyminÚymaxZylabelZyscaleZgenerate_legendÚindexZcurveÚliner Z linestyleZ drawstyleZ linewidthÚcolorÚmarkerZ
markersizeZmarkerfacecolorZmarkeredgecolorZrgbaZmappable_settingsÚmappabler) ÚlowÚhighÚ
interpolationZ draggabler0 Z
old_legendZ
new_legendÚfigure© ÚaxesZ curvelabelsZ has_curveZhas_smZlinedictZ mappabledictZmappablelabelsZ
xconverterZxunitsZ
yconverterZyunitsr r Úapply_callback¬ s†
ÿ
ÿ
z#figure_edit..apply_callbackzFigure optionsÚimageszqt4_editor_options.svg)rD ÚparentZiconÚapply)6ÚmapÚfloatr1 r2 Z get_titleZ
get_xlabelr5 Z
get_ylabelr6 r8 r9 r: Z get_unitsÚ get_linesZ get_labelr" r< Zto_hexr= Z get_colorZ get_alphaZget_markeredgecolorZget_markerfacecolorÚ
LINESTYLESZ
get_linestyleÚ
DRAWSTYLESZ
get_drawstyleZ
get_linewidthÚMARKERSZ
get_markerZget_markersizeÚappendÚboolrU Ú collectionsZ get_arrayr Z_cmap_registryr! r? Úvaluesr Zget_climÚhasattrÚmimageZinterpolations_namesr+ r Zfeditr ZQIconr r Z_get_data_path) rS rV ÚseprE rF rG rH rA r rJ r rB r% rK ZecZfcZ curvedatarM rC Zcmapsr) rN rO Z mappabledataZinterpolationsZdatalistrT r@ r rR r Ú figure_edit sÜ
ô
þþþ
ó ü ÿ þ R ÿü
re )N)Ú__doc__r Z
matplotlibr r r r< r r rc Z matplotlib.backends.qt_compatr Z matplotlib.backends.qt_editorr r[ r\ Z MarkerStyler] re r r r r Ú s$ ü ü