@@ -41,34 +41,34 @@ def convert_limits(lim, converter):
41
41
42
42
axis_map = axes ._axis_map
43
43
axis_limits = {
44
- axname : tuple (convert_limits (
45
- getattr (axes , f'get_{ axname } lim' )(), axis .converter
44
+ name : tuple (convert_limits (
45
+ getattr (axes , f'get_{ name } lim' )(), axis .converter
46
46
))
47
- for axname , axis in axis_map .items ()
47
+ for name , axis in axis_map .items ()
48
48
}
49
49
general = [
50
50
('Title' , axes .get_title ()),
51
51
sep ,
52
52
]
53
53
axes_info = [
54
54
(
55
- (None , f"<b>{ axis .upper ()} -Axis</b>" ),
56
- ('Min' , axis_limits [axis ][0 ]),
57
- ('Max' , axis_limits [axis ][1 ]),
58
- ('Label' , getattr ( axes , f"get_ { axis } label" ) ()),
59
- ('Scale' , [getattr ( axes , f"get_ { axis } scale" ) (),
55
+ (None , f"<b>{ name .upper ()} -Axis</b>" ),
56
+ ('Min' , axis_limits [name ][0 ]),
57
+ ('Max' , axis_limits [name ][1 ]),
58
+ ('Label' , axis . get_label (). get_text ()),
59
+ ('Scale' , [axis . get_scale (),
60
60
'linear' , 'log' , 'symlog' , 'logit' ]),
61
61
sep ,
62
62
)
63
- for axis in axis_map .keys ()
63
+ for name , axis in axis_map .items ()
64
64
]
65
65
general .extend (chain .from_iterable (axes_info ))
66
66
general .append (('(Re-)Generate automatic legend' , False ))
67
67
68
68
# Save the unit data
69
69
axis_units = {
70
- axis : getattr ( getattr ( axes , f" { axis } axis" ), " get_units" ) ()
71
- for axis in axis_map .keys ()
70
+ name : axis . get_units ()
71
+ for name , axis in axis_map .items ()
72
72
}
73
73
74
74
# Get / Curves
@@ -174,8 +174,8 @@ def prepare_data(d, init):
174
174
def apply_callback (data ):
175
175
"""A callback to apply changes."""
176
176
orig_limits = {
177
- axis : getattr (axes , f"get_{ axis } lim" )()
178
- for axis in axis_map .keys ()
177
+ name : getattr (axes , f"get_{ name } lim" )()
178
+ for name in axis_map .keys ()
179
179
}
180
180
181
181
general = data .pop (0 )
@@ -188,19 +188,19 @@ def apply_callback(data):
188
188
axes .set_title (title )
189
189
generate_legend = general .pop ()
190
190
191
- for i , (axname , ax ) in enumerate (axis_map .items ()):
192
- axmin = general [4 * i ]
193
- axmax = general [4 * i + 1 ]
194
- axlabel = general [4 * i + 2 ]
195
- axscale = general [4 * i + 3 ]
196
- if getattr (axes , f"get_{ axname } scale" )() != axscale :
197
- getattr (axes , f"set_{ axname } scale" )(axscale )
191
+ for i , (name , axis ) in enumerate (axis_map .items ()):
192
+ axis_min = general [4 * i ]
193
+ axis_max = general [4 * i + 1 ]
194
+ axis_label = general [4 * i + 2 ]
195
+ axis_scale = general [4 * i + 3 ]
196
+ if getattr (axes , f"get_{ name } scale" )() != axis_scale :
197
+ getattr (axes , f"set_{ name } scale" )(axis_scale )
198
198
199
- getattr (axes , f"set_{ axname } lim" )(axmin , axmax )
200
- getattr ( axes , f"set_ { axname } label" )( axlabel )
201
- setattr (ax , 'converter' , ax .converter )
202
- getattr ( ax , ' set_units' ) (axis_units [axname ])
203
- ax ._update_axisinfo ()
199
+ getattr (axes , f"set_{ name } lim" )(axis_min , axis_max )
200
+ axis . set_label_text ( axis_label )
201
+ setattr (axis , 'converter' , axis .converter )
202
+ axis . set_units (axis_units [name ])
203
+ axis ._update_axisinfo ()
204
204
205
205
# Set / Curves
206
206
for index , curve in enumerate (curves ):
@@ -247,8 +247,8 @@ def apply_callback(data):
247
247
# Redraw
248
248
figure = axes .get_figure ()
249
249
figure .canvas .draw ()
250
- for axis in axis_map .keys ():
251
- if getattr (axes , f"get_{ axis } lim" )() != orig_limits [axis ]:
250
+ for name in axis_map .keys ():
251
+ if getattr (axes , f"get_{ name } lim" )() != orig_limits [name ]:
252
252
figure .canvas .toolbar .push_current ()
253
253
break
254
254
0 commit comments