Differential D20103 Diff 55095 plugins/python/mixer_slider_docker/mixer_slider_docker.py

Changeset View

Standalone View

plugins/python/mixer_slider_docker/mixer_slider_docker.py

This file was added.

1		'''
2		Copyright (C) 2019 Tusooa Zhu <tusooa@vista.aero>
3
4		This file is part of Krita-docker-color-slider.
5
6		Krita-docker-color-slider is free software: you can redistribute it and/or modify
7		it under the terms of the GNU General Public License as published by
8		the Free Software Foundation, either version 3 of the License, or
9		(at your option) any later version.
10
11		Krita-docker-color-slider is distributed in the hope that it will be useful,
12		but WITHOUT ANY WARRANTY; without even the implied warranty of
13		MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14		GNU General Public License for more details.
15
16		You should have received a copy of the GNU General Public License
17		along with Krita-docker-color-slider. If not, see <https://www.gnu.org/licenses/>.
18		'''
19		from PyQt5.QtGui import QColor
20		from PyQt5.QtWidgets import QWidget, QVBoxLayout, QHBoxLayout, QPushButton
21
22		from krita import Krita, DockWidget, ManagedColor, DockWidgetFactory, DockWidgetFactoryBase
23
24		from .slider_line import SliderLine
25		from .ui_mixer_slider_docker import UIMixerSliderDocker
26
27
28		class MixerSliderDocker(DockWidget):
29		# Init the docker
30
31		def __init__(self):
32		super(MixerSliderDocker, self).__init__()
33
34		main_program = Krita.instance()
35		settings = main_program.readSetting("", "MixerSliderColors",
36		"RGBA,U8,sRGB-elle-V2-srgbtrc.icc,1,0.8,0.4,1," +
37		"RGBA,U8,sRGB-elle-V2-srgbtrc.icc,0,0,0,1") # alpha=1 == non-transparent
38
39		self.default_left_color = self.qcolor_to_managedcolor(QColor.fromRgbF(0.4, 0.8, 1, 1))
40		self.default_right_color = self.qcolor_to_managedcolor(QColor.fromRgbF(0, 0, 0, 1))
41
42		# make base-widget and layout
43		self.widget = QWidget()
44		self.sliders = []
45		self.top_layout = QVBoxLayout()
46		self.main_layout = QHBoxLayout()
47		self.top_layout.addLayout(self.main_layout)
48		self.top_layout.addStretch(0)
49		# The text on the button for settings
50		self.settings_button = QPushButton(i18n('S'))
51		self.settings_button.setToolTip(i18n('Change settings'))
52		self.settings_button.setMaximumSize(30, 30)
53		self.main_layout.addWidget(self.settings_button)
54		self.layout = QVBoxLayout()
55		self.layout.setSpacing(0)
56		self.main_layout.addLayout(self.layout)
57		for line in settings.split(";"):
58		colors = line.split(',')
59		left_color = self.parse_color(colors[0:7])
60		right_color = self.parse_color(colors[7:])
61		widget = SliderLine(left_color, right_color, self)
62		self.sliders.append(widget)
63		self.layout.addWidget(widget)
64
65		self.widget.setLayout(self.top_layout)
66		self.setWindowTitle(i18n("Mixer Slider Docker"))
67		self.setWidget(self.widget)
68		[x.show() for x in self.sliders]
69
70		self.settings_button.clicked.connect(self.init_ui)
71
72		def settings_changed(self):
73		if self.ui.line_edit is not None:
74		num_sliders = int(self.ui.line_edit.text())
75		if len(self.sliders) > num_sliders:
76		for extra_line in self.sliders[num_sliders:]:
77		self.layout.removeWidget(extra_line)
78		extra_line.setParent(None)
79
80		self.sliders = self.sliders[0:num_sliders]
81		elif len(self.sliders) < num_sliders:
82		for i in range(num_sliders - len(self.sliders)):
83		widget = SliderLine(self.default_left_color, self.default_right_color, self)
84		self.sliders.append(widget)
85		self.layout.addWidget(widget)
86		self.write_settings()
87
88		def get_color_space(self):
89		if self.canvas() is not None:
90		if self.canvas().view() is not None:
91		canvas_color = self.canvas().view().foregroundColor()
92		return ManagedColor(canvas_color.colorModel(), canvas_color.colorDepth(), canvas_color.colorProfile())
93		return ManagedColor('RGBA', 'U8', 'sRGB-elle-V2-srgbtrc.icc')
94
95		def init_ui(self):
96		self.ui = UIMixerSliderDocker()
97		self.ui.initialize(self)
98
99		def write_settings(self):
100		main_program = Krita.instance()
101		setting = ';'.join(
102		[self.color_to_settings(line.left) + ',' + self.color_to_settings(line.right)
103		for line in self.sliders])
104
105		main_program.writeSetting("", "MixerSliderColors", setting)
106
107		def color_to_settings(self, managedcolor):
108		return ','.join([managedcolor.colorModel(),
109		managedcolor.colorDepth(),
110		managedcolor.colorProfile()] +
111		[str(c) for c in managedcolor.components()])
112
	rbreuUnsubmitted Done I think this were easier to read if you build a list of all the strings first and then use join once, instead of this concatenation of two joins. Took me a bit to understand where all the commas end up. rbreu: I think this were easier to read if you build a list of all the strings first and then use join…
113		def parse_color(self, array):
114		color = ManagedColor(array[0], array[1], array[2])
115		color.setComponents([float(x) for x in array[3:]])
116		return color
117
118		def canvasChanged(self, canvas):
119		pass
120
121		def qcolor_to_managedcolor(self, qcolor):
122		mc = self.get_color_space()
123		mc.setComponents([qcolor.blueF(), qcolor.greenF(), qcolor.redF(), qcolor.alphaF()])
124		return mc
125
126		def managedcolor_to_qcolor(self, managedcolor):
127		return managedcolor.colorForCanvas(self.canvas())
128
129		Application.addDockWidgetFactory(DockWidgetFactory("mixer_slider_docker", DockWidgetFactoryBase.DockRight, MixerSliderDocker))