def draw_image(self, painter, canvasRect, srcRect, dstRect, xMap, yMap):
# normally we use this method indirectly from quiqwt which takes the burden of constructing
# the right parameters. if we want to call this method manually, eg for painting on on a
# QPixmap for saving the image, we just use the last set of parmeters passed to this
# method, this is much easier than constructing the params seperatly, and so we get the
# exact same result as we see on screen:
self.last_canvas_rect = canvasRect
self.last_src_rect = srcRect
self.last_dst_rect = dstRect
self.last_xmap = xMap
self.last_ymap = yMap
x1, y1 = canvasRect.left(), canvasRect.top()
x2, y2 = canvasRect.right(), canvasRect.bottom()
NX = x2 - x1
NY = y2 - y1
rtmin, mzmax, rtmax, mzmin = srcRect
self.data = self.compute_image(0, NX, NY, rtmin, rtmax, mzmin, mzmax)
# draw
srcRect = (0, 0, NX, NY)
x1, y1, x2, y2 = canvasRect.getCoords()
RawImageItem.draw_image(self, painter, canvasRect, srcRect, (x1, y1, x2, y2), xMap, yMap)
def draw_image(self, painter, canvasRect, srcRect, dstRect, xMap, yMap):
# normally we use this method indirectly from quiqwt which takes the burden of constructing
# the right parameters. if we want to call this method manually, eg for painting on on a
# QPixmap for saving the image, we just use the last set of parmeters passed to this
# method, this is much easier than constructing the params seperatly, and so we get the
# exact same result as we see on screen:
self.last_canvas_rect = canvasRect
self.last_src_rect = srcRect
self.last_dst_rect = dstRect
self.last_xmap = xMap
self.last_ymap = yMap
x1, y1 = canvasRect.left(), canvasRect.top()
x2, y2 = canvasRect.right(), canvasRect.bottom()
NX = x2 - x1
NY = y2 - y1
rtmin, mzmax, rtmax, mzmin = srcRect
self.data = self.compute_image(0, NX, NY, rtmin, rtmax, mzmin, mzmax)
# draw
srcRect = (0, 0, NX, NY)
x1, y1, x2, y2 = canvasRect.getCoords()
RawImageItem.draw_image(self, painter, canvasRect, srcRect,
(x1, y1, x2, y2), xMap, yMap)
def draw_image(self, painter, canvasRect, srcRect, dstRect, xMap, yMap):
x1, y1 = canvasRect.left(), canvasRect.top()
x2, y2 = canvasRect.right(), canvasRect.bottom()
i1, j1, i2, j2 = srcRect
NX = x2-x1
NY = y2-y1
if self.data.shape != (NX, NY):
self.data = np.zeros((NY, NX), np.uint8)
mandelbrot(i1, j1, i2, j2, self.data, self.IMAX)
srcRect = (0, 0, NX, NY)
x1, y1, x2, y2 = canvasRect.getCoords()
RawImageItem.draw_image(self, painter, canvasRect,
srcRect, (x1, y1, x2, y2), xMap, yMap)
def draw_image(self, painter, canvasRect, srcRect, dstRect, xMap, yMap):
x1, y1 = canvasRect.left(), canvasRect.top()
x2, y2 = canvasRect.right(), canvasRect.bottom()
i1, j1, i2, j2 = srcRect
NX = x2 - x1
NY = y2 - y1
if self.data.shape != (NX, NY):
self.data = np.zeros((NY, NX), np.uint8)
mandelbrot(i1, j1, i2, j2, self.data, self.IMAX)
srcRect = (0, 0, NX, NY)
x1, y1, x2, y2 = canvasRect.getCoords()
RawImageItem.draw_image(self, painter, canvasRect, srcRect,
(x1, y1, x2, y2), xMap, yMap)
