source: branches/Release2.0/python/asaplotgui.py@ 1274

Last change on this file since 1274 was 707, checked in by mar637, 19 years ago

split asaplot into gui and non-gui versions inheriting from asaplotbase. This is the 'old' gui version

  • Property svn:eol-style set to native
  • Property svn:keywords set to Author Date Id Revision
File size: 5.4 KB
Line 
1"""
2ASAP plotting class based on matplotlib.
3"""
4
5from asap.asaplotbase import *
6import Tkinter as Tk
7from matplotlib.backends.backend_tkagg import FigureCanvasTkAgg, \
8 FigureManagerTkAgg
9# Force use of the newfangled toolbar.
10import matplotlib
11matplotlib.use("TkAgg")
12matplotlib.rcParams['toolbar'] = 'toolbar2'
13
14class asaplotgui(asaplotbase):
15 """
16 ASAP plotting class based on matplotlib.
17 """
18
19 def __init__(self, rows=1, cols=0, title='', size=(8,6), buffering=False):
20 """
21 Create a new instance of the ASAPlot plotting class.
22
23 If rows < 1 then a separate call to set_panels() is required to define
24 the panel layout; refer to the doctext for set_panels().
25 """
26 v = vars()
27 del v['self']
28
29 asaplotbase.__init__(self, **v)
30 self.window = Tk.Tk()
31 def dest_callback():
32 self.is_dead = True
33 self.window.destroy()
34
35 self.window.protocol("WM_DELETE_WINDOW", dest_callback)
36 self.canvas = FigureCanvasTkAgg(self.figure, master=self.window)
37 self.canvas.get_tk_widget().pack(side=Tk.TOP, fill=Tk.BOTH, expand=1)
38 # Simply instantiating this is enough to get a working toolbar.
39 self.figmgr = FigureManagerTkAgg(self.canvas, 1, self.window)
40 self.window.wm_title('ASAPlot graphics window')
41
42 self.events = {'button_press':None,
43 'button_release':None,
44 'motion_notify':None}
45
46 matplotlib.interactive = True
47 self.buffering = buffering
48
49 self.canvas.show()
50
51 def map(self):
52 """
53 Reveal the ASAPlot graphics window and bring it to the top of the
54 window stack.
55 """
56 self.window.wm_deiconify()
57 self.window.lift()
58
59 def position(self):
60 """
61 Use the mouse to get a position from a graph.
62 """
63
64 def position_disable(event):
65 self.register('button_press', None)
66 print '%.4f, %.4f' % (event.xdata, event.ydata)
67
68 print 'Press any mouse button...'
69 self.register('button_press', position_disable)
70
71
72 def quit(self):
73 """
74 Destroy the ASAPlot graphics window.
75 """
76 self.window.destroy()
77
78
79 def region(self):
80 """
81 Use the mouse to get a rectangular region from a plot.
82
83 The return value is [x0, y0, x1, y1] in world coordinates.
84 """
85
86 def region_start(event):
87 height = self.canvas.figure.bbox.height()
88 self.rect = {'fig': None, 'height': height,
89 'x': event.x, 'y': height - event.y,
90 'world': [event.xdata, event.ydata,
91 event.xdata, event.ydata]}
92 self.register('button_press', None)
93 self.register('motion_notify', region_draw)
94 self.register('button_release', region_disable)
95
96 def region_draw(event):
97 self.canvas._tkcanvas.delete(self.rect['fig'])
98 self.rect['fig'] = self.canvas._tkcanvas.create_rectangle(
99 self.rect['x'], self.rect['y'],
100 event.x, self.rect['height'] - event.y)
101
102 def region_disable(event):
103 self.register('motion_notify', None)
104 self.register('button_release', None)
105
106 self.canvas._tkcanvas.delete(self.rect['fig'])
107
108 self.rect['world'][2:4] = [event.xdata, event.ydata]
109 print '(%.2f, %.2f) (%.2f, %.2f)' % (self.rect['world'][0],
110 self.rect['world'][1], self.rect['world'][2],
111 self.rect['world'][3])
112
113 self.register('button_press', region_start)
114
115 # This has to be modified to block and return the result (currently
116 # printed by region_disable) when that becomes possible in matplotlib.
117
118 return [0.0, 0.0, 0.0, 0.0]
119
120
121 def register(self, type=None, func=None):
122 """
123 Register, reregister, or deregister events of type 'button_press',
124 'button_release', or 'motion_notify'.
125
126 The specified callback function should have the following signature:
127
128 def func(event)
129
130 where event is an MplEvent instance containing the following data:
131
132 name # Event name.
133 canvas # FigureCanvas instance generating the event.
134 x = None # x position - pixels from left of canvas.
135 y = None # y position - pixels from bottom of canvas.
136 button = None # Button pressed: None, 1, 2, 3.
137 key = None # Key pressed: None, chr(range(255)), shift,
138 win, or control
139 inaxes = None # Axes instance if cursor within axes.
140 xdata = None # x world coordinate.
141 ydata = None # y world coordinate.
142
143 For example:
144
145 def mouse_move(event):
146 print event.xdata, event.ydata
147
148 a = asaplot()
149 a.register('motion_notify', mouse_move)
150
151 If func is None, the event is deregistered.
152
153 Note that in TkAgg keyboard button presses don't generate an event.
154 """
155
156 if not self.events.has_key(type): return
157
158 if func is None:
159 if self.events[type] is not None:
160 # It's not clear that this does anything.
161 self.canvas.mpl_disconnect(self.events[type])
162 self.events[type] = None
163
164 # It seems to be necessary to return events to the toolbar.
165 if type == 'motion_notify':
166 self.canvas.mpl_connect(type + '_event',
167 self.figmgr.toolbar.mouse_move)
168 elif type == 'button_press':
169 self.canvas.mpl_connect(type + '_event',
170 self.figmgr.toolbar.press)
171 elif type == 'button_release':
172 self.canvas.mpl_connect(type + '_event',
173 self.figmgr.toolbar.release)
174
175 else:
176 self.events[type] = self.canvas.mpl_connect(type + '_event', func)
177
178
179 def show(self):
180 """
181 Show graphics dependent on the current buffering state.
182 """
183 if not self.buffering:
184 asaplotbase.show(self)
185 self.window.wm_deiconify()
186 self.canvas.show()
187
188 def terminate(self):
189 """
190 Clear the figure.
191 """
192 self.window.destroy()
193
194 def unmap(self):
195 """
196 Hide the ASAPlot graphics window.
197 """
198 self.window.wm_withdraw()
Note: See TracBrowser for help on using the repository browser.