o
g$ @ sv d Z ddlZddlmZmZmZmZmZ ddl
m Z ddl m
Z
dddd d
d Zd d
d
dddZejjZdddZdS )zHModule that provides a GUI-based editor for Matplotlib's figure options. N)cbookcmcolorsmarkersimage)QtGui) _formlayoutSolidDashedDashDotDottedNone)-z--z-.:r
Defaultz Steps (Pre)z Steps (Mid)z Steps (Post))defaultz steps-prestepsz 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 ]} | }
|
dkrpqe| |
< qeg } dd } t|dD ]o}
|
} 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 g j! j"D ]}| }
|
dks|# d"u rq||
< qt|dg }d#d$ tt$j%& D }D ]W}
|
}|' }|t$j%( vrK||j)fg|}|* \}}d|
fd%|j)g| fd&|fd'|fg}t+|d(rd)d$ tt,j-D }| d*|. g|f | ||
d!g q1t ||d+d!fg}| r| | d,d!f |r| |d-d!f
f d.d/}t/j0|d0|t12t3t45d1d2|d3}|d"ur|| d"S d"S )4z Edit matplotlib figure options)NNTitle)Nz
X-AxisLeftRightLabelScalelinearloglogit)Nz
Y-AxisBottomTop)z (Re-)Generate automatic legendFc S s0 t d| }|r|dt|dfS | dfS )Nz(_line|_image)(\d+) r )rematchgroupint)labelr r$ i/home/ubuntu/cloudmapper/venv/lib/python3.10/site-packages/matplotlib/backends/qt_editor/figureoptions.pycmp_key= s z figure_edit..cmp_key
_nolegend_c S sf || vri | |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 i | ]\}}||qS r$ r$ ).0shortnamer$ r$ r%
^ z5figure_edit..prepare_data..c S r( r$ r$ )r) r+ r* r$ r$ r% r, ` r- c S s | d S )Nr r$ )short_and_namer$ r$ r% f s z3figure_edit..prepare_data..key)stritemssorted)dinit
name2short
short2namecanonical_initr$ r$ r% prepare_dataM s
z!figure_edit..prepare_datar0 T)
keep_alpha)Nz Linez
Line stylez
Draw styleWidthz Color (RGBA))Nz
MarkerStyleSizezFace color (RGBA)zEdge color (RGBA) Nc S s g | ]\}}||fqS r$ r$ )r) r+ cmapr$ r$ r%
s z figure_edit..Colormapz
Min. valuez
Max. valueget_interpolationc S s g | ]}||fqS r$ r$ )r) r+ r$ r$ r% rA s
InterpolationAxesCurvesz Images, etc.c (
s } }| d}r| dng }r | dng }| r%td|\
}}}} }
} } }
}} |
kr< |
|krG | | ||
| | | |
j
_ j_ j
j
j
j t|D ]P\}}| }|\ }}}}}}}}}|| || || || t|} |d || |dkr|| | | | | | | qt|D ]@\}} | } t | dkr| \}} } }!}"| |" n
t | dkr| \}} } }!| | | !t"#| | j$t%| |!g q|rAd}#d}$ j&dur3 ' }%|%j(du}#|%j)}$ j*|$d}&|&rA|&+|# , }'|'j-. |krX |ks`|'j-j/0 dS dS ) z A callback to apply changes.r zUnexpected fieldNnone r )ncol)1get_xlimget_ylimpop
ValueError
get_xscale
set_xscale
get_yscale
set_yscale set_titleset_xlim
set_xlabelset_ylim
set_ylabelxaxis converteryaxis set_units_update_axisinfo enumerate set_label
set_linestyle
set_drawstyle
set_linewidthmcolorsto_rgba set_alpha set_color
set_markerset_markersizeset_markerfacecolorset_markeredgecolorlenset_interpolationset_cmapr get_cmapset_climr4 legend_
get_legend
_draggable_ncollegend
set_draggable
get_figurecanvasdrawtoolbar push_current)(data orig_xlim orig_ylimgeneralcurves mappablestitlexminxmaxxlabelxscaleyminymaxylabelyscalegenerate_legendindexcurveliner# linestyle drawstyle linewidthcolormarker
markersizemarkerfacecolormarkeredgecolorrgbamappable_settingsmappabler@ lowhigh
interpolation draggablerJ
old_legend
new_legendfigure axes curvelabels has_curvehas_smlinedict mappabledictmappablelabels
xconverterxunits
yconverteryunitsr$ r% apply_callback s
z#figure_edit..apply_callbackzFigure optionsimageszqt4_editor_options.svg)r parenticonapply)6mapfloatrK rL get_title
get_xlabelrO
get_ylabelrQ rX rY rZ get_units get_lines get_labelr4 rb to_hexrc get_color get_alphaget_markeredgecolorget_markerfacecolor
LINESTYLES
get_linestyle
DRAWSTYLES
get_drawstyle
get_linewidthMARKERS
get_markerget_markersizeappendboolr collections get_arrayr _cmap_registryr3 rm valuesr+ get_climhasattrmimageinterpolations_namesrC r feditr QIconr2 r _get_data_path) r r sepr r r r r} r& r r# r~ r: r ecfc curvedatar r cmapsr@ r r mappabledatainterpolationsdatalistr rz r$ r r% figure_edit s
R
r )N)__doc__r
matplotlibr r r rb r r r matplotlib.backends.qt_compatr matplotlib.backends.qt_editorr r r MarkerStyler r r$ r$ r$ r% s$