1 /*
2  *  Copyright (c) 2007 Boudewijn Rempt <boud@valdyas.org>
3  *
4  *  This program is free software; you can redistribute it and/or modify
5  *  it under the terms of the GNU General Public License as published by
6  *  the Free Software Foundation; either version 2 of the License, or
7  *  (at your option) any later version.
8  *
9  *  This program is distributed in the hope that it will be useful,
10  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
11  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
12  *  GNU General Public License for more details.
13  *
14  *  You should have received a copy of the GNU General Public License
15  *  along with this program; if not, write to the Free Software
16  *  Foundation, Inc., 51 Franklin Street, Fifth Floor,
17  * Boston, MA 02110-1301, USA.
18  */
19 
20 #include "kis_qimageio_export.h"
21 #include "ui_kis_wdg_options_qimageio.h"
22 
23 #include <QCheckBox>
24 #include <QSlider>
25 
26 #include <kpluginfactory.h>
27 #include <QFileInfo>
28 #include <QApplication>
29 
30 #include <KisMimeDatabase.h>
31 #include <KisExportCheckRegistry.h>
32 #include <kis_paint_device.h>
33 #include <KisDocument.h>
34 #include <kis_image.h>
35 #include <kis_paint_layer.h>
36 #include <kis_config_widget.h>
37 
38 K_PLUGIN_FACTORY_WITH_JSON(KisQImageIOExportFactory, "krita_qimageio_export.json", registerPlugin<KisQImageIOExport>();)
39 
40 class KisWdgOptionsQImageIO : public KisConfigWidget, public Ui::KisWdgOptionsQImageIO
41 {
42     Q_OBJECT
43 
44 public:
45     KisWdgOptionsQImageIO(QWidget *parent);
46 
47     void setConfiguration(const KisPropertiesConfigurationSP  cfg) override;
48     KisPropertiesConfigurationSP configuration() const override;
49 };
50 
KisWdgOptionsQImageIO(QWidget * parent)51 KisWdgOptionsQImageIO::KisWdgOptionsQImageIO(QWidget *parent)
52     : KisConfigWidget(parent)
53 {
54     setupUi(this);
55     imageQuality->setRange(1, 100, 0);
56 }
57 
configuration() const58 KisPropertiesConfigurationSP KisWdgOptionsQImageIO::configuration() const
59 {
60     KisPropertiesConfigurationSP cfg(new KisPropertiesConfiguration());
61 
62     cfg->setProperty("quality", (int)imageQuality->value());
63     return cfg;
64 }
65 
setConfiguration(const KisPropertiesConfigurationSP cfg)66 void KisWdgOptionsQImageIO::setConfiguration(const KisPropertiesConfigurationSP cfg)
67 {
68     imageQuality->setValue(cfg->getInt("quality", 75));
69 }
70 
KisQImageIOExport(QObject * parent,const QVariantList &)71 KisQImageIOExport::KisQImageIOExport(QObject *parent, const QVariantList &) : KisImportExportFilter(parent)
72 {
73 }
74 
~KisQImageIOExport()75 KisQImageIOExport::~KisQImageIOExport()
76 {
77 }
78 
convert(KisDocument * document,QIODevice * io,KisPropertiesConfigurationSP configuration)79 KisImportExportErrorCode KisQImageIOExport::convert(KisDocument *document, QIODevice *io, KisPropertiesConfigurationSP configuration)
80 {
81     QRect rc = document->savingImage()->bounds();
82     QImage image = document->savingImage()->projection()->convertToQImage(0, 0, 0, rc.width(), rc.height(), KoColorConversionTransformation::internalRenderingIntent(), KoColorConversionTransformation::internalConversionFlags());
83     int quality = -1;
84     if (configuration) {
85         quality = configuration->getInt("quality", -1);
86     }
87 
88     bool result = image.save(io, QFileInfo(filename()).suffix().toLatin1(), quality);
89     if (result) {
90         return ImportExportCodes::OK;
91     }
92     else {
93         return ImportExportCodes::FileFormatIncorrect;
94     }
95 }
96 
initializeCapabilities()97 void KisQImageIOExport::initializeCapabilities()
98 {
99     QList<QPair<KoID, KoID> > supportedColorModels;
100     supportedColorModels << QPair<KoID, KoID>()
101             << QPair<KoID, KoID>(RGBAColorModelID, Integer8BitsColorDepthID);
102     addSupportedColorModels(supportedColorModels, KisMimeDatabase::descriptionForMimeType(mimeType()));
103     addCapability(KisExportCheckRegistry::instance()->get("ColorModelPerLayerCheck/" + RGBAColorModelID.id() + "/" + Integer8BitsColorDepthID.id())->create(KisExportCheckBase::SUPPORTED));
104 }
105 
createConfigurationWidget(QWidget * parent,const QByteArray &,const QByteArray &) const106 KisConfigWidget *KisQImageIOExport::createConfigurationWidget(QWidget *parent, const QByteArray& /*from*/, const QByteArray& /*to*/) const
107 {
108     if (mimeType() == "image/webp") {
109         return new KisWdgOptionsQImageIO(parent);
110     }
111 
112     return 0;
113 }
114 
defaultConfiguration(const QByteArray &,const QByteArray &) const115 KisPropertiesConfigurationSP KisQImageIOExport::defaultConfiguration(const QByteArray &, const QByteArray &) const
116 {
117     if (mimeType() == "image/webp") {
118         KisPropertiesConfigurationSP cfg = new KisPropertiesConfiguration();
119         // the Qt webp plugin chooses 75 when passing -1 (default argument) to QImage::save()
120         cfg->setProperty("quality", 75);
121         return cfg;
122     }
123     return 0;
124 }
125 
126 #include "kis_qimageio_export.moc"
127 
128