forked from ipython/ipython
-
Notifications
You must be signed in to change notification settings - Fork 2
/
GnuplotInteractive.py
148 lines (108 loc) · 5.14 KB
/
GnuplotInteractive.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
# -*- coding: utf-8 -*-
"""Interactive functions and magic functions for Gnuplot usage.
This requires the Gnuplot.py module for interfacing python with Gnuplot, which
can be downloaded from:
http://gnuplot-py.sourceforge.net/
See gphelp() below for details on the services offered by this module.
Inspired by a suggestion/request from Arnd Baecker.
"""
__all__ = ['Gnuplot','gp','gp_new','plot','plot2','splot','replot',
'hardcopy','gpdata','gpfile','gpstring','gpfunc','gpgrid',
'gphelp']
import IPython.GnuplotRuntime as GRun
from IPython.utils.genutils import warn
from IPython.core import page
# Set global names for interactive use
Gnuplot = GRun.Gnuplot
gp_new = GRun.gp_new
gp = GRun.gp
plot = gp.plot
plot2 = gp.plot2
splot = gp.splot
replot = gp.replot
hardcopy = gp.hardcopy
# Accessors for the main plot object constructors:
gpdata = Gnuplot.Data
gpfile = Gnuplot.File
gpstring = Gnuplot.String
gpfunc = Gnuplot.Func
gpgrid = Gnuplot.GridData
def gphelp():
"""Print information about the Gnuplot facilities in IPython."""
page("""
IPython provides an interface to access the Gnuplot scientific plotting
system, in an environment similar to that of Mathematica or Matlab.
New top-level global objects
----------------------------
Please see their respective docstrings for further details.
- gp: a running Gnuplot instance. You can access its methods as
gp.<method>. gp(`a string`) will execute the given string as if it had been
typed in an interactive gnuplot window.
- plot, splot, replot and hardcopy: aliases to the methods of the same name in
the global running Gnuplot instance gp. These allow you to simply type:
In [1]: plot(x,sin(x),title='Sin(x)') # assuming x is a Numeric array
and obtain a plot of sin(x) vs x with the title 'Sin(x)'.
- gp_new: a function which returns a new Gnuplot instance. This can be used to
have multiple Gnuplot instances running in your session to compare different
plots, each in a separate window.
- Gnuplot: alias to the Gnuplot2 module, an improved drop-in replacement for
the original Gnuplot.py. Gnuplot2 needs Gnuplot but redefines several of its
functions with improved versions (Gnuplot2 comes with IPython).
- gpdata, gpfile, gpstring, gpfunc, gpgrid: aliases to Gnuplot.Data,
Gnuplot.File, Gnuplot.String, Gnuplot.Func and Gnuplot.GridData
respectively. These functions create objects which can then be passed to the
plotting commands. See the Gnuplot.py documentation for details.
Keep in mind that all commands passed to a Gnuplot instance are executed in
the Gnuplot namespace, where no Python variables exist. For example, for
plotting sin(x) vs x as above, typing
In [2]: gp('plot x,sin(x)')
would not work. Instead, you would get the plot of BOTH the functions 'x' and
'sin(x)', since Gnuplot doesn't know about the 'x' Python array. The plot()
method lives in python and does know about these variables.
New magic functions
-------------------
%gpc: pass one command to Gnuplot and execute it or open a Gnuplot shell where
each line of input is executed.
%gp_set_default: reset the value of IPython's global Gnuplot instance.""")
# Code below is all for IPython use
# Define the magic functions for communicating with the above gnuplot instance.
def magic_gpc(self,parameter_s=''):
"""Execute a gnuplot command or open a gnuplot shell.
Usage (omit the % if automagic is on). There are two ways to use it:
1) %gpc 'command' -> passes 'command' directly to the gnuplot instance.
2) %gpc -> will open up a prompt (gnuplot>>>) which takes input like the
standard gnuplot interactive prompt. If you need to type a multi-line
command, use \\ at the end of each intermediate line.
Upon exiting of the gnuplot sub-shell, you return to your IPython
session (the gnuplot sub-shell can be invoked as many times as needed).
"""
if parameter_s.strip():
self.shell.gnuplot(parameter_s)
else:
self.shell.gnuplot.interact()
def magic_gp_set_default(self,parameter_s=''):
"""Set the default gnuplot instance accessed by the %gp magic function.
%gp_set_default name
Call with the name of the new instance at the command line. If you want to
set this instance in your own code (using an embedded IPython, for
example), simply set the variable __IPYTHON__.gnuplot to your own gnuplot
instance object."""
gname = parameter_s.strip()
G = eval(gname,self.shell.user_ns)
self.shell.gnuplot = G
self.shell.user_ns.update({'plot':G.plot,'splot':G.splot,'plot2':G.plot2,
'replot':G.replot,'hardcopy':G.hardcopy})
try:
__IPYTHON__
except NameError:
pass
else:
# make the global Gnuplot instance known to IPython
__IPYTHON__.gnuplot = GRun.gp
__IPYTHON__.gnuplot.shell_first_time = 1
print """*** Type `gphelp` for help on the Gnuplot integration features."""
# Add the new magic functions to the class dict
from IPython.core.iplib import InteractiveShell
InteractiveShell.magic_gpc = magic_gpc
InteractiveShell.magic_gp_set_default = magic_gp_set_default
#********************** End of file <GnuplotInteractive.py> *******************