From 243723a32128d0e57085f6f6c02b59ee1e3647f4 Mon Sep 17 00:00:00 2001
From: falkTX <falktx@falktx.com>
Date: Mon, 17 Aug 2020 10:16:58 +0100
Subject: [PATCH] Compress canvas clipboard image saving (as PNG) Closes #1078

Signed-off-by: falkTX <falktx@falktx.com>
---
 source/frontend/carla_host.py | 69 +++++++++++++++++++++--------------
 1 file changed, 41 insertions(+), 28 deletions(-)

diff --git a/source/frontend/carla_host.py b/source/frontend/carla_host.py
index b2831e2fb..b6dc10a14 100644
--- a/source/frontend/carla_host.py
+++ b/source/frontend/carla_host.py
@@ -37,9 +37,15 @@ try:
 except ImportError:
     PYQT_VERSION = 0x50600
 
-from PyQt5.QtCore import QT_VERSION, qCritical, QEventLoop, QFileInfo, QModelIndex, QPointF, QTimer, QEvent
-from PyQt5.QtGui import QImage, QImageWriter, QPainter, QPalette, QBrush
-from PyQt5.QtWidgets import QAction, QApplication, QInputDialog, QFileSystemModel, QListWidgetItem, QGraphicsView, QMainWindow
+from PyQt5.QtCore import (
+    QT_VERSION, qCritical, QBuffer, QEventLoop, QFileInfo, QIODevice, QMimeData, QModelIndex, QPointF, QTimer, QEvent
+)
+from PyQt5.QtGui import (
+    QImage, QImageWriter, QPainter, QPalette, QBrush
+)
+from PyQt5.QtWidgets import (
+    QAction, QApplication, QInputDialog, QFileSystemModel, QListWidgetItem, QGraphicsView, QMainWindow
+)
 
 # ------------------------------------------------------------------------------------------------------------
 # Imports (Custom)
@@ -1570,6 +1576,24 @@ class HostWindow(QMainWindow):
     def slot_canvasZoomReset(self):
         self.scene.zoom_reset()
 
+    def _canvasImageRender(self, zoom = 1.0):
+        image   = QImage(self.scene.width()*zoom, self.scene.height()*zoom, QImage.Format_RGB32)
+        painter = QPainter(image)
+        painter.save()
+        painter.setRenderHints(painter.renderHints() | QPainter.Antialiasing | QPainter.TextAntialiasing)
+        self.scene.clearSelection()
+        self.scene.render(painter)
+        painter.restore()
+        del painter
+        return image
+
+    def _canvasImageWrite(self, iw: QImageWriter, imgFormat: bytes, image: QImage):
+        iw.setFormat(imgFormat)
+        iw.setCompression(-1)
+        if QT_VERSION >= 0x50500:
+            iw.setOptimizedWrite(True)
+        iw.write(image)
+
     @pyqtSlot()
     def slot_canvasSaveImage(self):
         if self.fProjectFilename:
@@ -1617,36 +1641,25 @@ class HostWindow(QMainWindow):
         else:
             zoom = 1.0
 
-        image   = QImage(self.scene.width()*zoom, self.scene.height()*zoom, QImage.Format_RGB32)
-        painter = QPainter(image)
-        painter.save()
-        painter.setRenderHints(painter.renderHints() | QPainter.Antialiasing | QPainter.TextAntialiasing)
-        self.scene.clearSelection()
-        self.scene.render(painter)
-        painter.restore()
-        del painter
-
+        image = self._canvasImageRender(zoom)
         iw = QImageWriter(newPath)
-        iw.setFormat(imgFormat)
-        iw.setCompression(-1)
-
-        if QT_VERSION >= 0x50500:
-            iw.setOptimizedWrite(True)
-
-        iw.write(image)
+        self._canvasImageWrite(iw, imgFormat, image)
 
     @pyqtSlot()
     def slot_canvasCopyToClipboard(self):
-        image   = QImage(self.scene.width(), self.scene.height(), QImage.Format_RGB32)
-        painter = QPainter(image)
-        painter.save()
-        painter.setRenderHints(painter.renderHints() | QPainter.Antialiasing | QPainter.TextAntialiasing)
-        self.scene.clearSelection()
-        self.scene.render(painter)
-        painter.restore()
-        del painter
+        buffer = QBuffer()
+        buffer.open(QIODevice.WriteOnly)
+
+        image = self._canvasImageRender()
+        iw = QImageWriter(buffer, b"PNG")
+        self._canvasImageWrite(iw, b"PNG", image)
+
+        buffer.close()
+
+        mimeData = QMimeData()
+        mimeData.setData("image/png", buffer.buffer());
 
-        QApplication.clipboard().setImage(image)
+        QApplication.clipboard().setMimeData(mimeData)
 
     # --------------------------------------------------------------------------------------------------------
     # Canvas (canvas callbacks)