From 770342625d320659d2c5c406b1a78a809be086b0 Mon Sep 17 00:00:00 2001
From: qixinbo <qixinbo@gmail.com>
Date: Sun, 22 Mar 2020 18:08:29 +0800
Subject: [PATCH] add save_buffer to canvas

---
 imagepy/ui/canvas/canvas.py | 38 +++++++++++++++++++++++++++++++++++++
 1 file changed, 38 insertions(+)

diff --git a/imagepy/ui/canvas/canvas.py b/imagepy/ui/canvas/canvas.py
index 00a6b696..58fc0c4c 100644
--- a/imagepy/ui/canvas/canvas.py
+++ b/imagepy/ui/canvas/canvas.py
@@ -187,6 +187,9 @@ def update(self, counter = [0,0]):
             else:
                 drawmark(dc, self.to_panel_coor, self.marks[i], k=self.scale)
         dc.UnMask()
+
+        self.dc = dc
+
         counter[1] += time()-start
         if counter[0] == 10:
             print('frame rate:',int(10/max(0.001,counter[1])))
@@ -233,6 +236,41 @@ def to_panel_coor(self, x, y):
         y = y * self.scale + self.conbox[1]
         return x, y
 
+    def save_buffer(self, path):
+        dcSource = self.dc
+        # based largely on code posted to wxpython-users by Andrea Gavana 2006-11-08
+        size = dcSource.Size
+
+        # Create a Bitmap that will later on hold the screenshot image
+        # Note that the Bitmap must have a size big enough to hold the screenshot
+        # -1 means using the current default colour depth
+        bmp = wx.Bitmap(size.width, size.height)
+
+        # Create a memory DC that will be used for actually taking the screenshot
+        memDC = wx.MemoryDC()
+
+        # Tell the memory DC to use our Bitmap
+        # all drawing action on the memory DC will go to the Bitmap now
+        memDC.SelectObject(bmp)
+
+        # Blit (in this case copy) the actual screen on the memory DC
+        # and thus the Bitmap
+        memDC.Blit( 0, # Copy to this X coordinate
+            0, # Copy to this Y coordinate
+            size.width, # Copy this width
+            size.height, # Copy this height
+            dcSource, # From where do we copy?
+            0, # What's the X offset in the original DC?
+            0  # What's the Y offset in the original DC?
+            )
+
+        # Select the Bitmap out of the memory DC by selecting a new
+        # uninitialized Bitmap
+        memDC.SelectObject(wx.NullBitmap)
+
+        img = bmp.ConvertToImage()
+        img.SaveFile(path, wx.BITMAP_TYPE_PNG)
+
     def __del__(self):
         self.img = self.back = None
         print('========== canvas del')