Thanks to visit codestin.com
Credit goes to github.com

Skip to content

dynamically generate pyplot functions #9173

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Closed
wants to merge 7 commits into from
Closed

dynamically generate pyplot functions #9173

Changes from 1 commit
Commits
Show all changes
7 commits
Select commit Hold shift + click to select a range
458546b
Dynamically generate pyplot functions.
anntzer Sep 9, 2017
9b8f94e
Also dynamically generate scalarmappable functions.
anntzer Sep 9, 2017
894e4ec
Get rid of boilerplate.py.
anntzer Sep 9, 2017
7874128
Define pyplot.__all__.
anntzer Sep 9, 2017
bbbc93f
Fix docstrings affected by move to dynamic pyplot.
anntzer Sep 9, 2017
3ff6304
Also autogenerate isinteractive and rcdefaults.
anntzer Sep 10, 2017
9ea3ce3
pep8-ify pyplot.
anntzer Sep 10, 2017
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Also dynamically generate scalarmappable functions.
  • Loading branch information
@anntzer
anntzer committed Jan 16, 2018
commit 9b8f94e025f9f449d40ae7e73c10e85d2d46683a
67 changes: 27 additions & 40 deletions lib/matplotlib/pyplot.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -316,26 +316,8 @@ def rcdefaults():
draw_all()


# The current "image" (ScalarMappable) is retrieved or set
# only via the pyplot interface using the following two
# functions:
def gci():
"""
Get the current colorable artist. Specifically, returns the
current :class:`~matplotlib.cm.ScalarMappable` instance (image or
patch collection), or *None* if no images or patch collections
have been defined. The commands :func:`~matplotlib.pyplot.imshow`
and :func:`~matplotlib.pyplot.figimage` create
:class:`~matplotlib.image.Image` instances, and the commands
:func:`~matplotlib.pyplot.pcolor` and
:func:`~matplotlib.pyplot.scatter` create
:class:`~matplotlib.collections.Collection` instances. The
current image is an attribute of the current axes, or the nearest
earlier axes in the current figure that contains an image.
"""
return gcf()._gci()


# This function is not autogenerated because it is needed in when
# autogenerating other functions, see `_post_processors`.
def sci(im):
"""
Set the current image. This image will be the target of colormap
Expand Down Expand Up @@ -1991,23 +1973,6 @@ def getname_val(identifier):
fig.autofmt_xdate()


def _autogen_docstring(base):
"""Autogenerated wrappers will get their docstring from a base function
with an addendum."""
msg = ''
addendum = docstring.Appender(msg, '\n\n')
return lambda func: addendum(docstring.copy_dedent(base)(func))


# This function cannot be generated by boilerplate.py because it may
# return an image or a line.
@_autogen_docstring(Axes.spy)
def spy(Z, precision=0, marker=None, markersize=None, aspect='equal', **kwargs):
ret = ax.spy(Z, precision, marker, markersize, aspect, **kwargs)
if isinstance(ret, cm.ScalarMappable):
sci(ret)
return ret

# just to be safe. Interactive mode can be turned on without
# calling `plt.ion()` so register it again here.
# This is safe because multiple calls to `install_repl_displayhook`
Expand All @@ -2022,11 +1987,14 @@ def _make_wrapper(obj_getter, cls, command):
else:
method_name = func_name = command
method = getattr(cls, method_name)
post_processor = _post_processors.get(func_name, lambda obj: None)

def func(*args, **kwargs):
# We actually need to refetch the method here in case it has been
# monkey-patched.
return getattr(obj_getter(), method_name)(*args, **kwargs)
ret = getattr(obj_getter(), method_name)(*args, **kwargs)
post_processor(ret)
return ret

if six.PY2:
func.__name__ = func_name.encode("ascii")
Expand All @@ -2047,7 +2015,7 @@ def func(*args, **kwargs):
"clf", "figimage", "gca", "ginput", "savefig", "subplots_adjust",
"suptitle", "tight_layout", "waitforbuttonpress",
# Renamed commands.
"legend:figlegend", "text:figtext"]
"_gci:gci", "legend:figlegend", "text:figtext"]
_axes_commands = [
"acorr", "angle_spectrum", "annotate", "arrow", "autoscale", "axhline",
"axhspan", "axis", "axvline", "axvspan", "bar", "barbs", "barh", "boxplot",
Expand All @@ -2063,7 +2031,26 @@ def func(*args, **kwargs):
# Renamed commands.
"set_title:title", "set_xlabel:xlabel", "set_xscale:xscale",
"set_ylabel:ylabel", "set_yscale:yscale"]

_post_processors = {
# For the following functions, an additional callable will be called with
# the return value as argument.
"hexbin": sci,
"imshow": sci,
"pcolor": sci,
"pcolormesh": sci,
"quiver": sci,
"scatter": sci,
"tripcolor": sci,
"contour": lambda ret: sci(ret) if ret._A is not None else None,
"contourf": lambda ret: sci(ret) if ret._A is not None else None,
"tricontour": lambda ret: sci(ret) if ret._A is not None else None,
"tricontourf": lambda ret: sci(ret) if ret._A is not None else None,
"hist2d": lambda ret: sci(ret[-1]),
"specgram": lambda ret: sci(ret[-1]),
"streamplot": lambda ret: sci(ret.lines),
"spy": (
lambda ret: sci(ret) if isinstance(ret, cm.ScalarMappable) else None)
}

for _command in _canvas_commands:
_make_wrapper(
Expand Down