QT5 Conversion + Directory Structure

1 files changed, 59 insertions, 0 deletions

diff --git a/src/preview_thread.py b/src/preview_thread.py
new file mode 100644
index 0000000..4a46d51
--- /dev/null
+++ b/src/preview_thread.py
@@ -0,0 +1,59 @@
+from PyQt5 import QtCore, QtGui, uic
+from PyQt5.QtCore import pyqtSignal, pyqtSlot
+from PIL import Image
+from PIL.ImageQt import ImageQt
+import core
+from queue import Queue, Empty
+import os
+from copy import copy
+
+
+class Worker(QtCore.QObject):
+
+    imageCreated = pyqtSignal(['QImage'])
+
+    def __init__(self, parent=None, queue=None):
+        QtCore.QObject.__init__(self)
+        parent.newTask.connect(self.createPreviewImage)
+        parent.processTask.connect(self.process)
+        self.parent = parent
+        self.core = core.Core()
+        self.queue = queue
+        self.core.settings = parent.settings
+        self.stackedWidget = parent.window.stackedWidget
+        self.background = Image.new("RGBA", (1920, 1080), (0, 0, 0, 0))
+        self.background.paste(Image.open(os.path.join(
+            self.core.wd, "background.png")))
+
+    @pyqtSlot(list)
+    def createPreviewImage(self, components):
+        dic = {
+          "components": components,
+        }
+        self.queue.put(dic)
+
+    @pyqtSlot()
+    def process(self):
+        try:
+            nextPreviewInformation = self.queue.get(block=False)
+            while self.queue.qsize() >= 2:
+                try:
+                    self.queue.get(block=False)
+                except Empty:
+                    continue
+
+            width = int(self.core.settings.value('outputWidth'))
+            height = int(self.core.settings.value('outputHeight'))
+            frame = copy(self.background)
+            frame = frame.resize((width, height))
+
+            components = nextPreviewInformation["components"]
+            for component in reversed(components):
+                frame = Image.alpha_composite(
+                    frame, component.previewRender(self))
+
+            self._image = ImageQt(frame)
+            self.imageCreated.emit(QtGui.QImage(self._image))
+
+        except Empty:
+            True