Differential D13525 Diff 36869 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*> previewShapes;
65​ QSizeF maskSize; // at 100DPI
66​};
67​
68​KoGamutMask::KoGamutMask(const QString& filename)
69​ : KoResource(filename)
70​ , d(new Private())
71​{
72​
73​}
74​
75​KoGamutMask::KoGamutMask()
76​ : KoResource(QString())
77​ , d(new Private())
78​{
79​}
80​
81​KoGamutMask::KoGamutMask(KoGamutMask* rhs)
82​ : QObject(0)
83​ , KoResource(QString())
84​ , d(new Private())
85​{
86​ setFilename(rhs->filename());
87​ setTitle(rhs->title());
88​ setDescription(rhs->description());
89​ d->maskSize = rhs->d->maskSize;
90​
91​ QList<KoShape*> newShapes;
92​ for(KoGamutMaskShape* sh: d->maskShapes) {
93​ KoShape* shape = sh->koShape();
94​ newShapes.append(shape);
95​ }
96​
97​ setMaskShapes(newShapes);
98​
99​ setValid(true);
100​}
101​
102​
103​bool KoGamutMask::coordIsClear(const QPointF& coord, KoViewConverter &viewConverter, bool preview)
104​{
105​ QVector<KoGamutMaskShape*>* shapeVector;
106​
107​ if (preview && !d->previewShapes.isEmpty()) {
108​ shapeVector = &d->previewShapes;
109​ } else {
110​ shapeVector = &d->maskShapes;
111​ }
112​
113​ for(KoGamutMaskShape* shape: *shapeVector) {
114​ if (shape->coordIsClear(coord, viewConverter) == true) {
115​ return true;
116​ }
117​ }
118​
119​ return false;
120​}
121​
122​void KoGamutMask::paint(QPainter &painter, KoViewConverter& viewConverter, bool preview)
123​{
124​ QVector<KoGamutMaskShape*>* shapeVector;
125​
126​ if (preview && !d->previewShapes.isEmpty()) {
127​ shapeVector = &d->previewShapes;
128​ } else {
129​ shapeVector = &d->maskShapes;
130​ }
131​
132​ for(KoGamutMaskShape* shape: *shapeVector) {
133​ shape->paint(painter, viewConverter);
134​ }
135​}
136​
137​bool KoGamutMask::load()
138​{
139​ QFile file(filename());
140​ if (file.size() == 0) return false;
141​ if (!file.open(QIODevice::ReadOnly)) {
142​ warnFlake << "Can't open file " << filename();
143​ return false;
144​ }
145​ bool res = loadFromDevice(&file);
146​ setValid(res);
147​ file.close();
148​ return res;
149​}
150​
151​bool KoGamutMask::loadFromDevice(QIODevice *dev)
152​{
153​ if (!dev->isOpen()) dev->open(QIODevice::ReadOnly);
154​
155​ d->data = dev->readAll();
156​
157​ Q_ASSERT(d->data.size() != 0);
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.
158​
159​ if (filename().isNull()) {
160​ warnFlake << "Cannot load gamut mask" << name() << "there is no filename set";
161​ return false;
162​ }
163​
164​ if (d->data.isNull()) {
165​ QFile file(filename());
166​ if (file.size() == 0) {
167​ warnFlake << "Cannot load gamut mask" << name() << "there is no data available";
168​ return false;
169​ }
170​
171​ file.open(QIODevice::ReadOnly);
172​ d->data = file.readAll();
173​ file.close();
174​ }
175​
176​ QBuffer buf(&d->data);
177​ buf.open(QBuffer::ReadOnly);
178​
179​ KoStore* store(KoStore::createStore(&buf, KoStore::Read, "application/x-krita-gamutmask", KoStore::Zip));
180​ if (!store || store->bad()) return false;
181​
182​ bool storeOpened = store->open("gamutmask.svg");
183​ if (!storeOpened) { return false; }
184​
185​ QByteArray data;
186​ data.resize(store->size());
187​ QByteArray ba = store->read(store->size());
188​ store->close();
189​
190​ KoXmlDocument xmlDocument;
191​ QString errorMsg;
192​ int errorLine = 0;
193​ int errorColumn;
194​
195​ bool ok = xmlDocument.setContent(ba, false, &errorMsg, &errorLine, &errorColumn);
196​ if (!ok) {
197​
198​ errorFlake << "Parsing error in " << filename() << "! Aborting!" << endl
199​ << " In line: " << errorLine << ", column: " << errorColumn << endl
200​ << " Error message: " << errorMsg << endl;
201​ errorFlake << "Parsing error in the main document at line" << errorLine
202​ << ", column" << errorColumn << endl
203​ << "Error message: " << errorMsg;
204​
205​ return false;
206​ }
207​
208​ KoDocumentResourceManager manager;
209​ SvgParser parser(&manager);
210​ parser.setResolution(QRectF(0,0,100,100), 72); // initialize with default values
211​ QSizeF fragmentSize;
212​
213​ QList<KoShape*> shapes = parser.parseSvg(xmlDocument.documentElement(), &fragmentSize);
214​
215​ d->maskSize = fragmentSize;
216​
217​ d->title = parser.documentTitle();
218​ setName(d->title);
219​ d->description = parser.documentDescription();
220​
221​ setMaskShapes(shapes);
222​
223​ if (store->open("preview.png")) {
224​ KoStoreDevice previewDev(store);
225​ previewDev.open(QIODevice::ReadOnly);
226​
227​ QImage preview = QImage();
228​ preview.load(&previewDev, "PNG");
229​ setImage(preview);
230​
231​ (void)store->close();
232​ }
233​
234​ buf.close();
235​
236​ return true;
237​}
238​
239​void KoGamutMask::setMaskShapes(QList<KoShape*> shapes)
240​{
241​ setMaskShapesToVector(shapes, d->maskShapes);
242​}
243​
244​bool KoGamutMask::save()
245​{
246​ QFile file(filename());
247​ if (!file.open(QIODevice::WriteOnly | QIODevice::Truncate)) {
248​ return false;
249​ }
250​ saveToDevice(&file);
251​ file.close();
252​
253​ return true;
254​}
255​
256​QList<KoShape*> KoGamutMask::koShapes() const
257​{
258​ QList<KoShape*> shapes;
259​ for(KoGamutMaskShape* maskShape: d->maskShapes) {
260​ shapes.append(maskShape->koShape());
261​ }
262​
263​ return shapes;
264​}
265​
266​bool KoGamutMask::saveToDevice(QIODevice *dev) const
267​{
268​ KoStore* store(KoStore::createStore(dev, KoStore::Write, "application/x-krita-gamutmask", KoStore::Zip));
269​ if (!store || store->bad()) return false;
270​
271​ QList<KoShape*> shapes = koShapes();
272​
273​ std::sort(shapes.begin(), shapes.end(), KoShape::compareShapeZIndex);
274​
275​ if (!store->open("gamutmask.svg")) {
276​ return false;
277​ }
278​
279​ KoStoreDevice storeDev(store);
280​ storeDev.open(QIODevice::WriteOnly);
281​
282​ SvgWriter writer(shapes);
283​ writer.setDocumentTitle(d->title);
284​ writer.setDocumentDescription(d->description);
285​
286​ writer.save(storeDev, d->maskSize);
287​
288​ if (!store->close()) { return false; }
289​
290​
291​ if (!store->open("preview.png")) {
292​ return false;
293​ }
294​
295​ KoStoreDevice previewDev(store);
296​ previewDev.open(QIODevice::WriteOnly);
297​
298​ image().save(&previewDev, "PNG");
299​ if (!store->close()) { return false; }
300​
301​ return store->finalize();
302​}
303​
304​QString KoGamutMask::title()
305​{
306​ return d->title;
307​}
308​
309​void KoGamutMask::setTitle(QString title)
310​{
311​ d->title = title;
312​ setName(title);
313​}
314​
315​QString KoGamutMask::description()
316​{
317​ return d->description;
318​}
319​
320​void KoGamutMask::setDescription(QString description)
321​{
322​ d->description = description;
323​}
324​
325​QSizeF KoGamutMask::maskSize()
326​{
327​ return d->maskSize;
328​}
329​
330​// TODO: rethink preview
331​void KoGamutMask::setPreviewMaskShapes(QList<KoShape*> shapes)
332​{
333​ setMaskShapesToVector(shapes, d->previewShapes);
334​}
335​
336​void KoGamutMask::setMaskShapesToVector(QList<KoShape *> shapes, QVector<KoGamutMaskShape *> &targetVector)
337​{
338​ targetVector.clear();
339​
340​ for(KoShape* sh: shapes) {
341​ KoGamutMaskShape* maskShape = new KoGamutMaskShape(sh);
342​ targetVector.append(maskShape);
343​ }
344​}
345​
346​// clean up when ending mask preview
347​void KoGamutMask::clearPreview()
348​{
349​ d->previewShapes.clear();
350​}