drag_rectangle2.py 6.43 KB
 1 2 3 4 5 6 7 ``````# draggable rectangle with the animation blit techniques; see # http://www.scipy.org/Cookbook/Matplotlib/Animations import numpy as np import matplotlib.pyplot as plt from scipy.interpolate import interp1d from scipy import interpolate from pylab import * `````` Matheus Müller committed Oct 24, 2011 8 ``````from concentracao import * `````` 9 10 `````` listay = list() `````` Matheus Müller committed Oct 21, 2011 11 ``````listax = list() `````` 12 13 14 ``````drs = [] fig = plt.figure() ax = fig.add_subplot(111) `````` Matheus Müller committed Oct 24, 2011 15 16 17 18 19 ``````filename = "data.txt" #raw_input() dxstep = 0.01 xx = arange(0,20, dxstep) # we'll create an x-axis from 0 to 2 pi temp, = plot(xx,xx*0) # this is our initial plot, and does nothing `````` Matheus Müller committed Oct 21, 2011 20 `````` `````` 21 22 23 ``````####################################################################################################### # Add square `````` Matheus Müller committed Oct 24, 2011 24 ``````def add_square(x, y): `````` 25 `````` global listay, drs, ax `````` Matheus Müller committed Oct 21, 2011 26 `````` rect = ax.bar(x, 0.02, width=0.3) `````` 27 28 29 30 `````` for rec in rect: dr = DraggableRectangle(rec) dr.connect() drs.append(dr) `````` Matheus Müller committed Oct 24, 2011 31 32 33 34 35 36 37 38 39 40 41 42 43 44 `````` rec.set_y(y-0.01) listay.append(dr.rect.xy[1]) listax.append(dr.rect.xy[0]) dr.updateplot() ylim([0,1]) xlim([0,20]) def add_immovable_square(x, y): global listay, drs, ax rect = ax.bar(x, 0.02, width=0.3) for rec in rect: dr = DraggableRectangle(rec) drs.append(dr) rec.set_y(y-0.01) `````` 45 `````` listay.append(dr.rect.xy[1]) `````` Matheus Müller committed Oct 21, 2011 46 `````` listax.append(dr.rect.xy[0]) `````` 47 48 49 50 51 52 53 54 55 56 57 `````` ylim([0,1]) xlim([0,20]) def remove_square(): global listay, drs listay.pop() drs.pop() print listay, drs ####################################################################################################### `````` Matheus Müller committed Oct 21, 2011 58 ``````class voidfunctions: `````` 59 60 61 62 63 64 65 66 67 68 69 70 71 `````` def __init__(self, rect): self.rect = rect def connect(self): 'connect to all the events we need' self.cida = self.rect.figure.canvas.mpl_connect( 'key_release_event', self.on_a) self.cidr = self.rect.figure.canvas.mpl_connect( 'key_release_event', self.on_r) def on_a(self, event): if event.key=='a': if not event.inaxes: return `````` Matheus Müller committed Oct 24, 2011 72 `````` add_square(float(event.xdata), float(event.ydata)) `````` 73 `````` `````` Matheus Müller committed Oct 21, 2011 74 `````` # NAO FUNCIONA (DELETA TUDO) `````` 75 76 77 78 79 `````` def on_r(self, event): if event.key=='r': if not event.inaxes: return remove_square() `````` Matheus Müller committed Oct 21, 2011 80 81 82 83 84 `````` def disconnect(self): 'disconnect all the stored connection ids' self.rect.figure.canvas.mpl_disconnect(self.cida) self.rect.figure.canvas.mpl_disconnect(self.cidr) `````` Matheus Müller committed Oct 24, 2011 85 ``````####################################################################################################### `````` Matheus Müller committed Oct 21, 2011 86 87 88 89 90 91 92 `````` class DraggableRectangle: lock = None # only one can be animated at a time def __init__(self, rect): self.rect = rect self.press = None self.background = None `````` Matheus Müller committed Oct 24, 2011 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 `````` def updateplot(self): global listax, listay, temp, drs, dxstep xytemp = list(drs) for i in range(len(drs)): xytemp[i] = drs[i].rect.xy xytemp.sort(key=lambda s: s[0]) for i in range(len(listay)): (listax[i], listay[i]) = xytemp[i] data = open(filename , 'w') for i in range(len(listay)): print>>data, listax[i], listay[i] data.close() xydata = np.loadtxt(filename) modo = 'step' modo2 = 'reta' temp.set_ydata(montaperfil( xydata, dxstep, modo)) # update the plot data draw() # redraw the canvas ylim([0,1]) xlim([0,20]) `````` Matheus Müller committed Oct 21, 2011 116 117 118 119 120 121 122 123 124 125 `````` def connect(self): 'connect to all the events we need' self.cidpress = self.rect.figure.canvas.mpl_connect( 'button_press_event', self.on_press) self.cidrelease = self.rect.figure.canvas.mpl_connect( 'button_release_event', self.on_release) self.cidmotion = self.rect.figure.canvas.mpl_connect( 'motion_notify_event', self.on_motion) `````` 126 127 128 129 130 131 `````` def on_press(self, event): 'on button press we will see if the mouse is over us and store some data' if event.inaxes != self.rect.axes: return if DraggableRectangle.lock is not None: return contains, attrd = self.rect.contains(event) if not contains: return `````` Matheus Müller committed Oct 21, 2011 132 `````` `````` 133 134 135 136 137 138 139 140 141 142 143 144 145 146 147 148 149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 `````` #print 'event contains', self.rect.xy x0, y0 = self.rect.xy self.press = x0, y0, event.xdata, event.ydata DraggableRectangle.lock = self # draw everything but the selected rectangle and store the pixel buffer canvas = self.rect.figure.canvas axes = self.rect.axes self.rect.set_animated(True) canvas.draw() self.background = canvas.copy_from_bbox(self.rect.axes.bbox) # now redraw just the rectangle axes.draw_artist(self.rect) # and blit just the redrawn area canvas.blit(axes.bbox) def on_motion(self, event): 'on motion we will move the rect if the mouse is over us' if DraggableRectangle.lock is not self: return if event.inaxes != self.rect.axes: return x0, y0, xpress, ypress = self.press dx = event.xdata - xpress dy = event.ydata - ypress self.rect.set_x(x0+dx) self.rect.set_y(y0+dy) canvas = self.rect.figure.canvas axes = self.rect.axes # restore the background region canvas.restore_region(self.background) # redraw just the current rectangle axes.draw_artist(self.rect) # blit just the redrawn area canvas.blit(axes.bbox) `````` Matheus Müller committed Oct 24, 2011 170 171 `````` #self.updateplot() `````` 172 173 `````` def on_release(self, event): 'on release we reset the press data' `````` Matheus Müller committed Oct 21, 2011 174 `````` if DraggableRectangle.lock is not self: return `````` 175 176 177 178 179 180 181 182 183 184 `````` self.press = None DraggableRectangle.lock = None # turn off the rect animation property and reset the background self.rect.set_animated(False) self.background = None # redraw the full figure self.rect.figure.canvas.draw() `````` Matheus Müller committed Oct 24, 2011 185 `````` self.updateplot() `````` 186 187 188 189 190 191 192 `````` def disconnect(self): 'disconnect all the stored connection ids' self.rect.figure.canvas.mpl_disconnect(self.cidpress) self.rect.figure.canvas.mpl_disconnect(self.cidrelease) self.rect.figure.canvas.mpl_disconnect(self.cidmotion) `````` Matheus Müller committed Oct 24, 2011 193 194 ``````####################################################################################################### `````` Matheus Müller committed Oct 21, 2011 195 196 197 198 ``````voidrect = ax.bar(0, 0, width=0) for void in voidrect: vf = voidfunctions(void) vf.connect() `````` Matheus Müller committed Oct 24, 2011 199 200 201 202 ``````add_immovable_square(0,0) add_immovable_square(20,0) ####################################################################################################### `````` 203 204 205 206 207 `````` plt.show() ``````