import wx

from .backend_agg import FigureCanvasAgg

from .backend_wx import _BackendWx, _FigureCanvasWxBase

from .backend_wx import ( # noqa: F401 # pylint: disable=W0611

NavigationToolbar2Wx as NavigationToolbar2WxAgg)

class FigureCanvasWxAgg(FigureCanvasAgg, _FigureCanvasWxBase):

def draw(self, drawDC=None):

"""

Render the figure using agg.

"""

FigureCanvasAgg.draw(self)

self.bitmap = self._create_bitmap()

self._isDrawn = True

self.gui_repaint(drawDC=drawDC)

def blit(self, bbox=None):

# docstring inherited

bitmap = self._create_bitmap()

if bbox is None:

self.bitmap = bitmap

else:

srcDC = wx.MemoryDC(bitmap)

destDC = wx.MemoryDC(self.bitmap)

x = int(bbox.x0)

y = int(self.bitmap.GetHeight() - bbox.y1)

destDC.Blit(x, y, int(bbox.width), int(bbox.height), srcDC, x, y)

destDC.SelectObject(wx.NullBitmap)

srcDC.SelectObject(wx.NullBitmap)

self.gui_repaint()

def _create_bitmap(self):

"""Create a wx.Bitmap from the renderer RGBA buffer"""

rgba = self.get_renderer().buffer_rgba()

h, w, _ = rgba.shape

bitmap = wx.Bitmap.FromBufferRGBA(w, h, rgba)

bitmap.SetScaleFactor(self.GetDPIScaleFactor())

return bitmap

@_BackendWx.export

class _BackendWxAgg(_BackendWx):

FigureCanvas = FigureCanvasWxAgg