phyks/replot
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Implement axes swapping and rotation

Browse Source
This commit is contained in:
Lucas Verney 2016-03-11 17:04:14 +01:00
parent a736d41069
commit c80f3a4fac
2 changed files with 1650 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1589
Examples.ipynb
View File

File diff suppressed because one or more lines are too long

57
replot/__init__.py
View File

@ -302,6 +302,10 @@ class Figure():
- ``orthonormal`` (boolean) to force axis to be orthonormal.
- ``xlim`` and ``ylim`` which are tuples of intervals on the \
x and y axis.
- ``invert`` (boolean) invert X and Y axis on the plot. Invert \
the axes labels as well.
- ``rotate`` (angle in degrees) rotate the plot by the angle in \
degrees. Leave the labels untouched.
.. note:: Note that this API call considers list of tuples as \
list of (x, y) coordinates to plot, contrary to standard \
@ -347,6 +351,29 @@ class Figure():
# Keep track of the custom kwargs
plot_ += (custom_kwargs,)
# Handle inversion
if "invert" in custom_kwargs and custom_kwargs["invert"]:
# Invert X and Y data
plot_ = (
(plot_[0][1], plot_[0][0]) + plot_[0][2:],
plot_[1], plot_[2])
# Handle rotation
if "rotate" in custom_kwargs:
# Rotate X, Y data
# TODO: Not clean
new_X_list = []
new_Y_list = []
for x, y in zip(plot_[0][0], plot_[0][1]):
new_X_list.append(
np.cos(custom_kwargs["rotate"]) * x +
np.sin(custom_kwargs["rotate"]) * y)
new_Y_list.append(
-np.sin(custom_kwargs["rotate"]) * x +
np.cos(custom_kwargs["rotate"]) * y)
plot_ = (
(new_X_list, new_Y_list) + plot_[0][2:],
plot_[1], plot_[2])
# Add the plot to the correct group
if "group" in custom_kwargs:
self.plots[custom_kwargs["group"]].append(plot_)
@ -459,22 +486,43 @@ class Figure():
self.axes = axes
def _set_axes_properties(self, axis, group_):
is_inverted_axis = len(
[i
for i in self.plots[group_]
if "invert" in i[2] and i[2]["invert"]]
) > 0
# Set xlabel
if isinstance(self.xlabel, dict):
try:
if is_inverted_axis:
# Handle axis inversion
axis.set_ylabel(self.xlabel[group_])
else:
axis.set_xlabel(self.xlabel[group_])
except KeyError:
# No entry for this axis in the dict, pass it
pass
else:
if is_inverted_axis:
# Handle axis inversion
axis.set_ylabel(self.xlabel)
else:
axis.set_xlabel(self.xlabel)
# Set ylabel
if isinstance(self.ylabel, dict):
try:
if is_inverted_axis:
# Handle axis inversion
axis.set_xlabel(self.ylabel[group_])
else:
axis.set_ylabel(self.ylabel[group_])
except KeyError:
# No entry for this axis in the dict, pass it
pass
else:
if is_inverted_axis:
# Handle axis inversion
axis.set_xlabel(self.ylabel)
else:
axis.set_ylabel(self.ylabel)
# Set title
@ -772,6 +820,15 @@ def _handle_custom_plot_arguments(kwargs):
if "ylim" in kwargs:
custom_kwargs["ylim"] = kwargs["ylim"]
del kwargs["ylim"]
# Handle "invert" argument
if "invert" in kwargs:
custom_kwargs["invert"] = kwargs["invert"]
del kwargs["invert"]
# Handle "rotate" argument
if "rotate" in kwargs:
# Convert angle to radians
custom_kwargs["rotate"] = kwargs["rotate"] * np.pi / 180
del kwargs["rotate"]
return (kwargs, custom_kwargs)