Changeset View
Changeset View
Standalone View
Standalone View
plugins/dockers/gamutmask/gamutmask_dock.cpp
- This file was added.
1 | /* | ||||
---|---|---|---|---|---|
2 | * Copyright (c) 2018 Anna Medonosova <anna.medonosova@gmail.com> | ||||
3 | * | ||||
4 | * This library is free software; you can redistribute it and/or modify | ||||
5 | * it under the terms of the GNU Lesser General Public License as published by | ||||
6 | * the Free Software Foundation; version 2.1 of the License. | ||||
7 | * | ||||
8 | * This library is distributed in the hope that it will be useful, | ||||
9 | * but WITHOUT ANY WARRANTY; without even the implied warranty of | ||||
10 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the | ||||
11 | * GNU Lesser General Public License for more details. | ||||
12 | * | ||||
13 | * You should have received a copy of the GNU Lesser General Public License | ||||
14 | * along with this program; if not, write to the Free Software | ||||
15 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA. | ||||
16 | */ | ||||
17 | | ||||
18 | #include <kis_debug.h> | ||||
19 | | ||||
20 | #include <klocalizedstring.h> | ||||
21 | #include <KoCanvasResourceManager.h> | ||||
22 | #include <KoResourceServerProvider.h> | ||||
23 | #include <KoResourceServerObserver.h> | ||||
24 | #include <KoResourceServerAdapter.h> | ||||
25 | #include <KoCanvasBase.h> | ||||
26 | #include <KoColor.h> | ||||
27 | #include <resources/KoGamutMask.h> | ||||
28 | #include <kis_icon_utils.h> | ||||
29 | #include <KisPart.h> | ||||
30 | #include <kis_shape_layer.h> | ||||
31 | #include <kis_types.h> | ||||
32 | #include <KisDocument.h> | ||||
33 | #include <kis_node_selection_adapter.h> | ||||
34 | #include <kis_group_layer.h> | ||||
35 | #include <KisView.h> | ||||
36 | #include <KoResourceItemChooser.h> | ||||
37 | | ||||
38 | #include <QWidget> | ||||
39 | #include <QMenu> | ||||
40 | #include <QButtonGroup> | ||||
41 | #include <QRegExpValidator> | ||||
42 | #include <QRegExp> | ||||
43 | #include <QFileInfo> | ||||
44 | | ||||
45 | #include "gamutmask_dock.h" | ||||
46 | #include <KisViewManager.h> | ||||
47 | #include <kis_canvas_resource_provider.h> | ||||
48 | | ||||
49 | #include "ui_wdgGamutMaskChooser.h" | ||||
50 | | ||||
51 | class KisMainWindow; | ||||
52 | | ||||
53 | struct GamutMaskChooserUI: public QWidget, public Ui_wdgGamutMaskChooser | ||||
54 | { | ||||
55 | GamutMaskChooserUI() { | ||||
56 | setupUi(this); | ||||
57 | } | ||||
58 | }; | ||||
59 | | ||||
60 | | ||||
61 | GamutMaskDock::GamutMaskDock() | ||||
62 | : QDockWidget(i18n("Gamut Masks")) | ||||
63 | , m_resourceProvider(0) | ||||
64 | { | ||||
65 | m_dockerUI = new GamutMaskChooserUI(); | ||||
66 | | ||||
67 | m_dockerUI->bnMaskEditor->setIcon(KisIconUtils::loadIcon("dirty-preset")); | ||||
68 | m_dockerUI->bnMaskNew->setIcon(KisIconUtils::loadIcon("duplicatelayer")); | ||||
69 | m_dockerUI->bnMaskSet->setIcon(KisIconUtils::loadIcon("dialog-ok")); | ||||
70 | m_dockerUI->bnMaskDelete->setIcon(KisIconUtils::loadIcon("edit-delete")); | ||||
71 | m_dockerUI->maskPropertiesBox->setVisible(false); | ||||
72 | | ||||
73 | QRegExp maskTitleRegex("^[A-Za-z0-9]+$"); | ||||
74 | QRegExpValidator* m_maskTitleValidator = new QRegExpValidator(maskTitleRegex, this); | ||||
75 | m_dockerUI->maskTitleEdit->setValidator(m_maskTitleValidator); | ||||
76 | | ||||
77 | KoResourceServer<KoGamutMask>* rServer = KoResourceServerProvider::instance()->gamutMaskServer(); | ||||
78 | rServer->addObserver(this); | ||||
79 | | ||||
80 | // gamut mask connections | ||||
81 | connect(m_dockerUI->bnSaveMask , SIGNAL(clicked()) , SLOT(slotGamutMaskSave())); | ||||
82 | connect(m_dockerUI->bnCancelMaskEdit , SIGNAL(clicked()) , SLOT(slotGamutMaskCancelEdit())); | ||||
83 | connect(m_dockerUI->bnPreviewMask , SIGNAL(clicked()) , SLOT(slotGamutMaskPreview())); | ||||
84 | | ||||
85 | connect(m_dockerUI->bnMaskEditor , SIGNAL(clicked()) , SLOT(slotGamutMaskEdit())); | ||||
86 | connect(m_dockerUI->maskChooser, SIGNAL(sigGamutMaskSelected(KoGamutMask*)), SLOT(slotGamutMaskSelected(KoGamutMask*))); | ||||
87 | connect(m_dockerUI->bnMaskNew , SIGNAL(clicked()) , SLOT(slotGamutMaskNew())); | ||||
88 | connect(m_dockerUI->bnMaskSet , SIGNAL(clicked()) , SLOT(slotGamutMaskSet())); | ||||
89 | connect(m_dockerUI->bnMaskDelete , SIGNAL(clicked()) , SLOT(slotGamutMaskDelete())); | ||||
90 | | ||||
91 | setWidget(m_dockerUI); | ||||
92 | } | ||||
93 | | ||||
94 | GamutMaskDock::~GamutMaskDock() | ||||
95 | { | ||||
96 | KoResourceServer<KoGamutMask>* rServer = KoResourceServerProvider::instance()->gamutMaskServer(); | ||||
97 | rServer->removeObserver(this); | ||||
98 | } | ||||
99 | | ||||
100 | void GamutMaskDock::setMainWindow(KisViewManager* kisview) | ||||
101 | { | ||||
102 | m_resourceProvider = kisview->resourceProvider(); | ||||
103 | | ||||
104 | m_selectedMask = m_resourceProvider->currentGamutMask(); | ||||
105 | | ||||
106 | connect(this, SIGNAL(sigGamutMaskSet(KoGamutMask*)), m_resourceProvider, SLOT(slotGamutMaskActivated(KoGamutMask*))); | ||||
107 | connect(this, SIGNAL(sigGamutMaskChanged(KoGamutMask*)), m_resourceProvider, SLOT(slotGamutMaskActivated(KoGamutMask*))); | ||||
108 | connect(this, SIGNAL(sigGamutMaskUnset()), m_resourceProvider, SLOT(slotGamutMaskUnset())); | ||||
109 | } | ||||
110 | | ||||
111 | void GamutMaskDock::slotGamutMaskEdit() | ||||
112 | { | ||||
113 | if (m_selectedMask) { | ||||
114 | openMaskEditor(false); | ||||
115 | } | ||||
116 | } | ||||
117 | | ||||
118 | void GamutMaskDock::openMaskEditor(bool creatingNewMask) | ||||
119 | { | ||||
120 | m_dockerUI->maskPropertiesBox->setVisible(true); | ||||
121 | | ||||
122 | //shouldnt be nullptr here | ||||
123 | if (!m_selectedMask) { | ||||
124 | return; | ||||
125 | } | ||||
126 | | ||||
127 | m_dockerUI->maskTitleEdit->setText(m_selectedMask->title()); | ||||
128 | m_dockerUI->maskDescriptionEdit->setPlainText(m_selectedMask->description()); | ||||
129 | | ||||
130 | // open gamut mask template in the application | ||||
131 | QString maskTemplateFile = KoResourcePaths::findResource("data", "gamutmasks/GamutMaskTemplate.kra"); | ||||
132 | dbgPlugins << "GamutMaskDock::slotGamutMaskEdit: maskTemplateFile" | ||||
133 | << maskTemplateFile; | ||||
134 | | ||||
135 | // open template document | ||||
136 | m_maskDocument = KisPart::instance()->createDocument(); | ||||
137 | m_maskDocument->setFileBatchMode(false); | ||||
138 | KisPart::instance()->addDocument(m_maskDocument); | ||||
139 | m_maskDocument->openUrl(QUrl::fromLocalFile(maskTemplateFile), KisDocument::DontAddToRecent); | ||||
140 | m_maskDocument->setFileBatchMode(false); | ||||
141 | // // TODO: change file name, so the template wont be overwritten by accident (mask name?) | ||||
142 | | ||||
143 | | ||||
144 | // create shape layer and load shapes from mask | ||||
145 | KisShapeLayerSP shapeLayer = new KisShapeLayer(m_maskDocument->shapeController() | ||||
146 | , m_maskDocument->image() | ||||
147 | , "maskShapesLayer" | ||||
148 | , OPACITY_OPAQUE_U8); | ||||
149 | | ||||
150 | for (KoShape *shape: m_selectedMask->koShapes()) { | ||||
151 | shapeLayer->addShape(shape); | ||||
152 | } | ||||
153 | | ||||
154 | m_maskDocument->image()->addNode(shapeLayer); | ||||
155 | | ||||
156 | m_maskDocument->setPreActivatedNode(shapeLayer); | ||||
157 | | ||||
158 | // set document as active | ||||
159 | KisMainWindow* mainWindow = KisPart::instance()->currentMainwindow(); | ||||
160 | Q_ASSERT(mainWindow); | ||||
161 | | ||||
162 | KisView* view = mainWindow->newView(m_maskDocument); | ||||
163 | Q_ASSERT(view); | ||||
164 | | ||||
165 | for(KisView *view: KisPart::instance()->views()) { | ||||
166 | if (view->document() == m_maskDocument) { | ||||
167 | view->activateWindow(); | ||||
168 | break; | ||||
169 | } | ||||
170 | } | ||||
171 | } | ||||
172 | | ||||
173 | bool GamutMaskDock::closeMaskDocument() | ||||
174 | { | ||||
175 | bool retval = m_maskDocument->closeUrl(false); | ||||
176 | | ||||
177 | for(KisView *view: KisPart::instance()->views()) { | ||||
178 | if (view->document() == m_maskDocument) { | ||||
179 | view->close(); | ||||
180 | view->closeView(); | ||||
181 | view->deleteLater(); | ||||
182 | } | ||||
183 | } | ||||
184 | | ||||
185 | KisPart::instance()->removeDocument(m_maskDocument); | ||||
186 | m_maskDocument = 0; | ||||
187 | | ||||
188 | return retval; | ||||
189 | } | ||||
190 | | ||||
191 | QList<KoShape*> GamutMaskDock::getShapesFromLayer() | ||||
192 | { | ||||
193 | KisNodeSP node = m_maskDocument->image()->rootLayer()->findChildByName("maskShapesLayer"); | ||||
194 | KisShapeLayerSP shapeLayer = KisShapeLayerSP(dynamic_cast<KisShapeLayer*>(node.data())); | ||||
195 | | ||||
196 | // create a deep copy of the shapes to save in the mask, | ||||
197 | // otherwise they vanish when the template closes | ||||
198 | // TODO: they vanish even if copied | ||||
199 | QList<KoShape*> newShapes; | ||||
200 | for (KoShape* sh: shapeLayer->shapes()) { | ||||
201 | KoShape* newShape = sh->cloneShape(); | ||||
202 | newShapes.append(newShape); | ||||
203 | } | ||||
204 | | ||||
205 | return newShapes; | ||||
206 | } | ||||
207 | | ||||
208 | void GamutMaskDock::slotGamutMaskSave() | ||||
209 | { | ||||
210 | if (m_selectedMask) { | ||||
211 | m_selectedMask->setTitle(m_dockerUI->maskTitleEdit->text()); | ||||
212 | m_selectedMask->setDescription(m_dockerUI->maskDescriptionEdit->toPlainText()); | ||||
213 | | ||||
214 | m_selectedMask->setMaskShapes(getShapesFromLayer()); | ||||
215 | m_selectedMask->save(); | ||||
216 | | ||||
217 | closeMaskDocument(); | ||||
218 | m_dockerUI->maskPropertiesBox->setVisible(false); | ||||
219 | | ||||
220 | if (m_resourceProvider->currentGamutMask() == m_selectedMask) { | ||||
221 | emit sigGamutMaskChanged(m_selectedMask); | ||||
222 | } | ||||
223 | } | ||||
224 | } | ||||
225 | | ||||
226 | void GamutMaskDock::slotGamutMaskCancelEdit() | ||||
227 | { | ||||
228 | // m_selectedMask->resetMaskShapes(); | ||||
229 | closeMaskDocument(); | ||||
230 | m_dockerUI->maskPropertiesBox->setVisible(false); | ||||
231 | } | ||||
232 | | ||||
233 | void GamutMaskDock::slotGamutMaskPreview() | ||||
234 | { | ||||
235 | // if (!m_selectedMask) { | ||||
236 | // return; | ||||
237 | // } | ||||
238 | // m_selectedMask->setPreviewMaskShapes(getShapesFromLayer()); | ||||
239 | // emit sigGamutMaskPreviewUpdate(m_selectedMask); | ||||
240 | } | ||||
241 | | ||||
242 | void GamutMaskDock::slotGamutMaskSelected(KoGamutMask *mask) | ||||
243 | { | ||||
244 | m_selectedMask = mask; | ||||
245 | | ||||
246 | if (m_selectedMask) { | ||||
247 | m_dockerUI->labelMaskName->setText(m_selectedMask->title()); | ||||
248 | } else { | ||||
249 | m_dockerUI->labelMaskName->setText("None"); | ||||
250 | } | ||||
251 | } | ||||
252 | | ||||
253 | void GamutMaskDock::setCanvas(KoCanvasBase *canvas) | ||||
254 | { | ||||
255 | setEnabled(canvas != 0); | ||||
256 | } | ||||
257 | | ||||
258 | void GamutMaskDock::unsetCanvas() | ||||
259 | { | ||||
260 | setEnabled(false); | ||||
261 | } | ||||
262 | | ||||
263 | | ||||
264 | void GamutMaskDock::unsetResourceServer() | ||||
265 | { | ||||
266 | KoResourceServer<KoGamutMask>* rServer = KoResourceServerProvider::instance()->gamutMaskServer(); | ||||
267 | rServer->removeObserver(this); | ||||
268 | } | ||||
269 | | ||||
270 | void GamutMaskDock::removingResource(KoGamutMask *resource) | ||||
271 | { | ||||
272 | } | ||||
273 | | ||||
274 | void GamutMaskDock::resourceChanged(KoGamutMask *resource) | ||||
275 | { | ||||
276 | } | ||||
277 | | ||||
278 | void GamutMaskDock::slotGamutMaskNew() | ||||
279 | { | ||||
280 | KoResourceServer<KoGamutMask>* rServer = KoResourceServerProvider::instance()->gamutMaskServer(); | ||||
281 | | ||||
282 | // construct a copy | ||||
283 | KoGamutMask* newMask = new KoGamutMask(m_selectedMask); | ||||
284 | | ||||
285 | QString title = newMask->title(); | ||||
286 | | ||||
287 | QString saveLocation = rServer->saveLocation(); | ||||
288 | QFileInfo fileInfo(newMask->filename()); | ||||
289 | | ||||
290 | int i = 1; | ||||
291 | int lastIndex = 0; | ||||
292 | while (fileInfo.exists()) { | ||||
293 | fileInfo.setFile(saveLocation + title + QString("%1").arg(i) + ".kgm"); | ||||
294 | lastIndex = i; | ||||
295 | i++; | ||||
296 | } | ||||
297 | | ||||
298 | QString maskTitle = (lastIndex > 0) ? (title + QString("%1").arg(lastIndex)) : title; | ||||
299 | newMask->setTitle(maskTitle); | ||||
300 | newMask->setFilename(fileInfo.filePath()); | ||||
301 | | ||||
302 | newMask->setValid(true); | ||||
303 | rServer->addResource(newMask, false); | ||||
304 | | ||||
305 | openMaskEditor(false); | ||||
306 | } | ||||
307 | | ||||
308 | void GamutMaskDock::slotGamutMaskDelete() | ||||
309 | { | ||||
310 | // if deleting previously set mask, notify selectors to unset their mask | ||||
311 | if (m_resourceProvider->currentGamutMask() == m_selectedMask) { | ||||
312 | emit sigGamutMaskUnset(); | ||||
313 | } | ||||
314 | | ||||
315 | // TODO: dialog to confirm del | ||||
316 | QString resourceFile = m_selectedMask->filename(); | ||||
317 | | ||||
318 | KoResourceServer<KoGamutMask>* rServer = KoResourceServerProvider::instance()->gamutMaskServer(); | ||||
319 | rServer->removeResourceFromServer(m_selectedMask); | ||||
320 | // TODO: isnt there a function to do that? what about permanent resources? | ||||
321 | QFile::remove(resourceFile); | ||||
322 | } | ||||
323 | | ||||
324 | void GamutMaskDock::slotGamutMaskSet() | ||||
325 | { | ||||
326 | emit sigGamutMaskSet(m_selectedMask); | ||||
327 | } |