1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
|
"""
Thread that runs to create QImages for MainWindow's preview label.
Processes a queue of component lists.
"""
from PyQt6 import QtCore, QtGui, uic
from PyQt6.QtCore import pyqtSignal, pyqtSlot
from PIL import Image
from PIL.ImageQt import ImageQt
from queue import Queue, Empty
import os
import logging
from ..toolkit.frame import Checkerboard
from ..toolkit import disableWhenOpeningProject
log = logging.getLogger("AVP.Gui.PreviewThread")
class Worker(QtCore.QObject):
imageCreated = pyqtSignal(QtGui.QImage)
error = pyqtSignal(str)
def __init__(self, core, settings, queue):
super().__init__()
self.core = core
self.settings = settings
width = int(self.settings.value("outputWidth"))
height = int(self.settings.value("outputHeight"))
self.queue = queue
self.background = Checkerboard(width, height)
@disableWhenOpeningProject
@pyqtSlot(list)
def createPreviewImage(self, components):
dic = {
"components": components,
}
self.queue.put(dic)
log.debug("Preview thread id: {}".format(int(QtCore.QThread.currentThreadId())))
@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.settings.value("outputWidth"))
height = int(self.settings.value("outputHeight"))
if self.background.width != width or self.background.height != height:
self.background = Checkerboard(width, height)
frame = self.background.copy()
log.info("Creating new preview frame")
components = nextPreviewInformation["components"]
for component in reversed(components):
try:
component.lockSize(width, height)
if "composite" in component.properties():
newFrame = component.previewRender(frame)
else:
newFrame = component.previewRender()
component.unlockSize()
frame = Image.alpha_composite(frame, newFrame)
except (AttributeError, ValueError) as e:
errMsg = (
"Bad frame returned by %s's preview renderer. "
"%s. New frame %s."
% (
str(component),
str(e).capitalize(),
(
"is None"
if newFrame is None
else "size was %s*%s; should be %s*%s"
% (newFrame.width, newFrame.height, width, height)
),
)
)
log.critical(errMsg)
self.error.emit(errMsg)
break
except RuntimeError as e:
log.error(str(e))
else:
# We must store a reference to this QImage
# or else Qt will garbage-collect it on the C++ side
self.frame = ImageQt(frame)
self.imageCreated.emit(QtGui.QImage(self.frame))
except Empty:
True
|
