diff options
Diffstat (limited to 'src/components/color.py')
| -rw-r--r-- | src/components/color.py | 176 |
1 files changed, 0 insertions, 176 deletions
diff --git a/src/components/color.py b/src/components/color.py deleted file mode 100644 index 1f32c23..0000000 --- a/src/components/color.py +++ /dev/null @@ -1,176 +0,0 @@ -from PyQt6 import QtGui -import logging - -from ..component import Component -from ..toolkit.frame import BlankFrame, FloodFrame, FramePainter, PaintColor - - -log = logging.getLogger("AVP.Components.Color") - - -class Component(Component): - name = "Color" - version = "1.0.0" - - def widget(self, *args): - self.x = 0 - self.y = 0 - super().widget(*args) - - # disable color #2 until non-default 'fill' option gets changed - self.page.lineEdit_color2.setDisabled(True) - self.page.pushButton_color2.setDisabled(True) - self.page.spinBox_width.setValue(int(self.settings.value("outputWidth"))) - self.page.spinBox_height.setValue(int(self.settings.value("outputHeight"))) - - self.fillLabels = [ - "Solid", - "Linear Gradient", - "Radial Gradient", - ] - for label in self.fillLabels: - self.page.comboBox_fill.addItem(label) - self.page.comboBox_fill.setCurrentIndex(0) - - self.trackWidgets( - { - "x": self.page.spinBox_x, - "y": self.page.spinBox_y, - "sizeWidth": self.page.spinBox_width, - "sizeHeight": self.page.spinBox_height, - "trans": self.page.checkBox_trans, - "spread": self.page.comboBox_spread, - "stretch": self.page.checkBox_stretch, - "RG_start": self.page.spinBox_radialGradient_start, - "LG_start": self.page.spinBox_linearGradient_start, - "RG_end": self.page.spinBox_radialGradient_end, - "LG_end": self.page.spinBox_linearGradient_end, - "RG_centre": self.page.spinBox_radialGradient_spread, - "fillType": self.page.comboBox_fill, - "color1": self.page.lineEdit_color1, - "color2": self.page.lineEdit_color2, - }, - presetNames={ - "sizeWidth": "width", - "sizeHeight": "height", - }, - colorWidgets={ - "color1": self.page.pushButton_color1, - "color2": self.page.pushButton_color2, - }, - relativeWidgets=[ - "x", - "y", - "sizeWidth", - "sizeHeight", - "LG_start", - "LG_end", - "RG_start", - "RG_end", - "RG_centre", - ], - ) - - def update(self): - fillType = self.page.comboBox_fill.currentIndex() - if fillType == 0: - self.page.lineEdit_color2.setEnabled(False) - self.page.pushButton_color2.setEnabled(False) - self.page.checkBox_trans.setEnabled(False) - self.page.checkBox_stretch.setEnabled(False) - self.page.comboBox_spread.setEnabled(False) - else: - self.page.lineEdit_color2.setEnabled(True) - self.page.pushButton_color2.setEnabled(True) - self.page.checkBox_trans.setEnabled(True) - self.page.checkBox_stretch.setEnabled(True) - self.page.comboBox_spread.setEnabled(True) - if self.page.checkBox_trans.isChecked(): - self.page.lineEdit_color2.setEnabled(False) - self.page.pushButton_color2.setEnabled(False) - self.page.fillWidget.setCurrentIndex(fillType) - - def previewRender(self): - return self.drawFrame(self.width, self.height) - - def properties(self): - return ["static"] - - def frameRender(self, frameNo): - log.debug("Color component is drawing frame #%s", frameNo) - return self.drawFrame(self.width, self.height) - - def drawFrame(self, width, height): - r, g, b = self.color1 - shapeSize = (self.sizeWidth, self.sizeHeight) - # in default state, skip all this logic and return a plain fill - if ( - self.fillType == 0 - and shapeSize == (width, height) - and self.x == 0 - and self.y == 0 - ): - return FloodFrame(width, height, (r, g, b, 255)) - - # Return a solid image at x, y - if self.fillType == 0: - frame = BlankFrame(width, height) - image = FloodFrame(self.sizeWidth, self.sizeHeight, (r, g, b, 255)) - frame.paste(image, box=(self.x, self.y)) - return frame - - # Now fills that require using Qt... - elif self.fillType > 0: - image = FramePainter(width, height) - - if self.stretch: - w = width - h = height - else: - w = self.sizeWidth - h = self.sizeWidth - - if self.fillType == 1: # Linear Gradient - brush = QtGui.QLinearGradient( - float(self.LG_start), - float(self.LG_start), - float(self.LG_end + width / 3), - float(self.LG_end), - ) - - elif self.fillType == 2: # Radial Gradient - brush = QtGui.QRadialGradient( - float(self.RG_start), - float(self.RG_end), - float(w), - float(h), - float(self.RG_centre), - ) - spread = QtGui.QGradient.Spread.PadSpread - if self.spread == 1: - spread = QtGui.QGradient.Spread.ReflectSpread - elif self.spread == 2: - spread = QtGui.QGradient.Spread.RepeatSpread - brush.setSpread(spread) - brush.setColorAt(0.0, PaintColor(*self.color1)) - if self.trans: - brush.setColorAt(1.0, PaintColor(0, 0, 0, 0)) - elif self.fillType == 1 and self.stretch: - brush.setColorAt(0.2, PaintColor(*self.color2)) - else: - brush.setColorAt(1.0, PaintColor(*self.color2)) - image.setBrush(brush) - image.drawRect(self.x, self.y, self.sizeWidth, self.sizeHeight) - - return image.finalize() - - def commandHelp(self): - print("Specify a color:\n color=255,255,255") - - def command(self, arg): - if "=" in arg: - key, arg = arg.split("=", 1) - if key == "color": - self.page.lineEdit_color1.setText(arg) - return - super().command(arg) |