Differential D13525 Diff 36302 libs/flake/resources/KoGamutMask.cpp

Changeset View

Standalone View

View Options

libs/flake/resources/KoGamutMask.cpp

  • This file was added.
1​#include "KoGamutMask.h"
2​
3​#include <cstring>
4​
5​#include <QVector>
6​#include <QString>
7​#include <QFile>
8​#include <QList>
9​#include <QDomDocument>
10​#include <QDomElement>
11​#include <QByteArray>
12​#include <QBuffer>
13​
14​#include <FlakeDebug.h>
15​
16​#include <KoStore.h>
17​#include <KoStoreDevice.h>
18​#include <KoDocumentResourceManager.h>
19​#include <SvgParser.h>
20​#include <SvgWriter.h>
21​#include <KoShape.h>
22​#include <KisGamutMaskViewConverter.h>
23​#include <kis_assert.h>
24​
25​
26​KoGamutMaskShape::KoGamutMaskShape(KoShape* shape)
27​ : m_maskShape(shape)
28​ , m_shapePaintingContext(KoShapePaintingContext())
29​{
30​}
31​
32​KoGamutMaskShape::KoGamutMaskShape() {};
33​KoGamutMaskShape::~KoGamutMaskShape() {};
34​
35​KoShape* KoGamutMaskShape::koShape()
36​{
37​ return m_maskShape;
38​}
39​
40​bool KoGamutMaskShape::coordIsClear(const QPointF& coord, const KoViewConverter& viewConverter) const
41​{
42​ QPointF translatedPoint = viewConverter.viewToDocument(coord);
43​
44​ bool isClear = m_maskShape->hitTest(translatedPoint);
45​
46​ return isClear;
47​}
48​
49​void KoGamutMaskShape::paint(QPainter &painter, const KoViewConverter& viewConverter)
50​{
51​ painter.save();
52​ painter.setTransform(m_maskShape->absoluteTransformation(&viewConverter) * painter.transform());
53​ m_maskShape->paint(painter, viewConverter, m_shapePaintingContext);
54​ painter.restore();
55​}
56​
57​
58​struct Q_DECL_HIDDEN KoGamutMask::Private {
59​ QString name;
60​ QString title;
61​ QString description;
62​ QByteArray data;
63​ QVector<KoGamutMaskShape*> maskShapes;
64​ QVector<KoGamutMaskShape*> originalShapes; // stores original loaded shapes, when preview is activated
65​ KisGamutMaskViewConverter* viewConverter;
66​ QSizeF maskSize; // at 100DPI
67​};
68​
69​KoGamutMask::KoGamutMask(const QString& filename)
70​ : KoResource(filename)
71​ , d(new Private())
72​{
73​ d->viewConverter = new KisGamutMaskViewConverter();
74​}
75​
76​KoGamutMask::KoGamutMask()
77​ : KoResource(QString())
78​ , d(new Private())
79​{
80​}
81​
82​KoGamutMask::KoGamutMask(KoGamutMask* rhs)
83​ : QObject(0)
84​ , KoResource(QString())
85​ , d(new Private())
86​{
87​ setFilename(rhs->filename());
88​ setTitle(rhs->title());
89​ setDescription(rhs->description());
90​ d->maskSize = rhs->d->maskSize;
91​ d->viewConverter = rhs->d->viewConverter;
92​
93​ QVector<KoGamutMaskShape*> shapesToCopy;
94​ if (!rhs->d->originalShapes.isEmpty()) {
95​ shapesToCopy = rhs->d->originalShapes;
96​ } else {
97​ shapesToCopy = rhs->d->maskShapes;
98​ }
99​
100​ QList<KoShape*> newShapes;
101​ for(KoGamutMaskShape* sh: shapesToCopy) {
102​ KoShape* shape = sh->koShape();
103​ newShapes.append(shape);
104​ }
105​
106​ setMaskShapes(newShapes);
107​
108​ setValid(true);
109​}
110​
111​
112​bool KoGamutMask::coordIsClear(const QPointF& coord)
113​{
114​ Q_FOREACH(const KoGamutMaskShape* shape, d->maskShapes) {
115​ if (shape->coordIsClear(coord, *d->viewConverter) == true) {
116​ return true;
117​ }
118​ }
119​
120​ return false;
121​}
122​
123​void KoGamutMask::paint(QPainter &painter)
124​{
125​ for(KoGamutMaskShape* shape: d->maskShapes) {
126​ shape->paint(painter, *d->viewConverter);
127​ }
128​}
129​
130​bool KoGamutMask::load()
131​{
132​ QFile file(filename());
133​ if (file.size() == 0) return false;
134​ if (!file.open(QIODevice::ReadOnly)) {
135​ warnFlake << "Can't open file " << filename();
136​ return false;
137​ }
138​ bool res = loadFromDevice(&file);
139​ setValid(res);
140​ file.close();
141​ return res;
142​}
143​
144​bool KoGamutMask::loadFromDevice(QIODevice *dev)
145​{
146​ if (!dev->isOpen()) dev->open(QIODevice::ReadOnly);
147​
148​ d->data = dev->readAll();
149​
150​ Q_ASSERT(d->data.size() != 0);
151​
152​ if (filename().isNull()) {
153​ warnFlake << "Cannot load gamut mask" << name() << "there is no filename set";
154​ return false;
155​ }
156​
157​ if (d->data.isNull()) {
158​ QFile file(filename());
dkazakovUnsubmitted
Not Done

Please use KIS_ASSERT or KIS_SAFE_ASSERT_RECOVER or KIS_SAFE_ASSERT_RECOVER_RETURN here.

Q_ASSERTS's are disabled on 99% of the systems, so we prefer not to use them.

dkazakov: Please use KIS_ASSERT or KIS_SAFE_ASSERT_RECOVER or KIS_SAFE_ASSERT_RECOVER_RETURN here.
159​ if (file.size() == 0) {
160​ warnFlake << "Cannot load gamut mask" << name() << "there is no data available";
161​ return false;
162​ }
163​
164​ file.open(QIODevice::ReadOnly);
165​ d->data = file.readAll();
166​ file.close();
167​ }
168​
169​ QBuffer buf(&d->data);
170​ buf.open(QBuffer::ReadOnly);
171​
172​ QScopedPointer<KoStore> store(KoStore::createStore(&buf, KoStore::Read, "application/x-krita-gamutmask", KoStore::Zip));
173​ if (!store || store->bad()) return false;
174​
175​ bool storeOpened = store->open("gamutmask.svg");
176​ if (!storeOpened) { return false; }
177​
178​ QByteArray data;
179​ data.resize(store->size());
180​ QByteArray ba = store->read(store->size());
181​ store->close();
182​
183​ KoXmlDocument xmlDocument;
184​ QString errorMsg;
185​ int errorLine = 0;
186​ int errorColumn;
187​
188​ bool ok = xmlDocument.setContent(ba, false, &errorMsg, &errorLine, &errorColumn);
189​ if (!ok) {
190​
191​ errorFlake << "Parsing error in " << filename() << "! Aborting!" << endl
192​ << " In line: " << errorLine << ", column: " << errorColumn << endl
193​ << " Error message: " << errorMsg << endl;
194​ errorFlake << "Parsing error in the main document at line" << errorLine
195​ << ", column" << errorColumn << endl
196​ << "Error message: " << errorMsg;
197​
198​ return false;
199​ }
200​
201​ KoDocumentResourceManager manager;
202​ SvgParser parser(&manager);
203​ parser.setResolution(QRectF(0,0,100,100), 72); // initialize with default values
204​ QSizeF fragmentSize;
205​
206​ QList<KoShape*> shapes = parser.parseSvg(xmlDocument.documentElement(), &fragmentSize);
207​
208​ d->maskSize = fragmentSize;
209​ d->viewConverter->setMaskSize(d->maskSize);
210​
211​ d->title = parser.documentTitle();
212​ setName(d->title);
213​ d->description = parser.documentDescription();
214​
215​ setMaskShapes(shapes);
216​
217​ buf.close();
218​
219​ setImage(generateIcon());
220​ return true;
221​}
222​
223​void KoGamutMask::setMaskShapes(QList<KoShape*> shapes)
224​{
225​ d->maskShapes.clear();
226​
227​ for(KoShape* sh: shapes) {
228​ KoGamutMaskShape* maskShape = new KoGamutMaskShape(sh);
229​ d->maskShapes.append(maskShape);
230​ }
231​}
232​
233​
234​bool KoGamutMask::save()
235​{
236​ QFile file(filename());
237​ if (!file.open(QIODevice::WriteOnly | QIODevice::Truncate)) {
238​ return false;
239​ }
240​ saveToDevice(&file);
241​ file.close();
242​
243​ setImage(generateIcon());
244​
245​ return true;
246​}
247​
248​QList<KoShape*> KoGamutMask::koShapes() const
249​{
250​ QList<KoShape*> shapes;
251​ for(KoGamutMaskShape* maskShape: d->maskShapes) {
252​ shapes.append(maskShape->koShape());
253​ }
254​
255​ return shapes;
256​}
257​
258​bool KoGamutMask::saveToDevice(QIODevice *dev) const
259​{
260​ KoStore* store(KoStore::createStore(dev, KoStore::Write, "application/x-krita-gamutmask", KoStore::Zip));
261​ if (!store || store->bad()) return false;
262​
263​ QList<KoShape*> shapes = koShapes();
264​
265​ std::sort(shapes.begin(), shapes.end(), KoShape::compareShapeZIndex);
266​
267​ if (!store->open("gamutmask.svg")) {
268​ return false;
269​ }
270​
271​ KoStoreDevice storeDev(store);
272​ storeDev.open(QIODevice::WriteOnly);
273​
274​ SvgWriter writer(shapes);
275​ writer.setDocumentTitle(d->title);
276​ writer.setDocumentDescription(d->description);
277​
278​ writer.save(storeDev, d->maskSize);
279​
280​ if (!store->close()) { return false; }
281​ return store->finalize();
282​
283​ return true;
284​}
285​
286​void KoGamutMask::setViewSize(int size){
287​ d->viewConverter->setViewSize(size);
288​}
289​
290​QString KoGamutMask::title()
291​{
292​ return d->title;
293​}
294​
295​void KoGamutMask::setTitle(QString title)
296​{
297​ d->title = title;
298​ setName(title);
299​}
300​
301​QString KoGamutMask::description()
302​{
303​ return d->description;
304​}
305​
306​void KoGamutMask::setDescription(QString description)
307​{
308​ d->description = description;
309​}
310​
311​// TODO: rethink preview
312​void KoGamutMask::setPreviewMaskShapes(QList<KoShape*> shapes)
313​{
314​// d->originalShapes = QVector<KoGamutMaskShape*>(d->maskShapes);
315​// setMaskShapes(shapes);
316​}
317​
318​// switch back to loaded shapes when ending mask preview
319​void KoGamutMask::resetMaskShapes()
320​{
321​// if (!d->originalShapes.isEmpty()) {
322​// d->maskShapes.clear();
323​// d->maskShapes = d->originalShapes;
324​// d->originalShapes.clear();
325​// }
326​}
327​
328​QImage KoGamutMask::generateIcon()
329​{
330​ QImage image(d->maskSize.width(), d->maskSize.height(), QImage::Format_ARGB32_Premultiplied);
331​ QPainter painter(&image);
332​ image.fill(Qt::gray);
333​
334​ KoViewConverter viewConverter;
335​
336​ for(KoGamutMaskShape* maskShape: d->maskShapes) {
337​ maskShape->paint(painter, viewConverter);
338​ }
339​
340​ painter.end();
341​ image = image.scaled(128, 128, Qt::KeepAspectRatio);
342​ return image;
343​}