232 lines
6.5 KiB
C++

// vim: set tabstop=4 shiftwidth=4 expandtab:
/*
Gwenview: an image viewer
Copyright 2007 Aurélien Gâteau <agateau@kde.org>
This program is free software; you can redistribute it and/or
modify it under the terms of the GNU General Public License
as published by the Free Software Foundation; either version 2
of the License, or (at your option) any later version.
This program is distributed in the hope that it will be useful,
but WITHOUT ANY WARRANTY; without even the implied warranty of
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
GNU General Public License for more details.
You should have received a copy of the GNU General Public License
along with this program; if not, write to the Free Software
Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
*/
// Self
#include "redeyereductiontool.h"
// Qt
#include <QDialogButtonBox>
#include <QGraphicsSceneMouseEvent>
#include <QPainter>
#include <QPushButton>
#include <QRect>
// KF
// Local
#include "gwenview_lib_debug.h"
#include "gwenviewconfig.h"
#include "redeyereductionimageoperation.h"
#include "ui_redeyereductionwidget.h"
#include <lib/documentview/rasterimageview.h>
namespace Gwenview
{
struct RedEyeReductionWidget : public QWidget, public Ui_RedEyeReductionWidget {
RedEyeReductionWidget()
{
setupUi(this);
QPushButton *okButton = mainDialogButtonBox->button(QDialogButtonBox::Ok);
okButton->setIcon(QIcon::fromTheme(QStringLiteral("redeyes")));
okButton->setText(i18n("Reduce Red Eye"));
}
void showNotSetPage()
{
// Prevent Close button from turning blue upon accepting
helpTextLabel->setFocus();
stackedWidget->setCurrentWidget(notSetPage);
}
void showMainPage()
{
stackedWidget->setCurrentWidget(mainPage);
}
};
struct RedEyeReductionToolPrivate {
RedEyeReductionTool *q = nullptr;
RedEyeReductionTool::Status mStatus;
QPointF mCenter;
int mDiameter;
RedEyeReductionWidget *mToolWidget = nullptr;
void setupToolWidget()
{
mToolWidget = new RedEyeReductionWidget;
mToolWidget->showNotSetPage();
QObject::connect(mToolWidget->diameterSpinBox, SIGNAL(valueChanged(int)), q, SLOT(setDiameter(int)));
QObject::connect(mToolWidget->mainDialogButtonBox, &QDialogButtonBox::accepted, q, &RedEyeReductionTool::slotApplyClicked);
QObject::connect(mToolWidget->mainDialogButtonBox, &QDialogButtonBox::rejected, q, &RedEyeReductionTool::done);
QObject::connect(mToolWidget->helpDialogButtonBox, &QDialogButtonBox::rejected, q, &RedEyeReductionTool::done);
}
QRectF rectF() const
{
if (mStatus == RedEyeReductionTool::NotSet) {
return QRectF();
}
return QRectF(mCenter.x() - mDiameter / 2, mCenter.y() - mDiameter / 2, mDiameter, mDiameter);
}
};
RedEyeReductionTool::RedEyeReductionTool(RasterImageView *view)
: AbstractRasterImageViewTool(view)
, d(new RedEyeReductionToolPrivate)
{
d->q = this;
d->mDiameter = GwenviewConfig::redEyeReductionDiameter();
d->mStatus = NotSet;
d->setupToolWidget();
view->document()->startLoadingFullImage();
}
RedEyeReductionTool::~RedEyeReductionTool()
{
GwenviewConfig::setRedEyeReductionDiameter(d->mDiameter);
delete d->mToolWidget;
delete d;
}
void RedEyeReductionTool::paint(QPainter *painter)
{
if (d->mStatus == NotSet) {
return;
}
const QRectF docRectF = d->rectF();
imageView()->document()->waitUntilLoaded();
const QRect docRect = docRectF.toAlignedRect();
QImage img = imageView()->document()->image().copy(docRect);
const QRectF imgRectF(docRectF.left() - docRect.left(), docRectF.top() - docRect.top(), docRectF.width(), docRectF.height());
RedEyeReductionImageOperation::apply(&img, imgRectF);
const QRectF viewRectF = imageView()->mapToView(docRectF);
painter->drawImage(viewRectF, img, imgRectF);
}
void RedEyeReductionTool::mousePressEvent(QGraphicsSceneMouseEvent *event)
{
if (event->buttons() != Qt::LeftButton || event->modifiers() & Qt::ControlModifier) {
event->ignore();
return;
}
event->accept();
if (d->mStatus == NotSet) {
d->mToolWidget->diameterSpinBox->setValue(d->mDiameter);
d->mToolWidget->showMainPage();
d->mStatus = Adjusting;
}
d->mCenter = imageView()->mapToImage(event->pos());
imageView()->update();
}
void RedEyeReductionTool::mouseMoveEvent(QGraphicsSceneMouseEvent *event)
{
event->accept();
if (event->buttons() == Qt::NoButton) {
return;
}
d->mCenter = imageView()->mapToImage(event->pos());
imageView()->update();
}
void RedEyeReductionTool::mouseReleaseEvent(QGraphicsSceneMouseEvent *event)
{
// Just prevent the event from reaching the image view
event->accept();
}
void RedEyeReductionTool::mouseDoubleClickEvent(QGraphicsSceneMouseEvent *event)
{
if (event->buttons() != Qt::LeftButton) {
event->ignore();
return;
}
event->accept();
Q_EMIT d->mToolWidget->mainDialogButtonBox->accepted();
}
void RedEyeReductionTool::keyPressEvent(QKeyEvent *event)
{
QDialogButtonBox *buttons;
if (d->mStatus == Adjusting) {
buttons = d->mToolWidget->mainDialogButtonBox;
} else {
buttons = d->mToolWidget->helpDialogButtonBox;
}
switch (event->key()) {
case Qt::Key_Escape:
event->accept();
Q_EMIT buttons->rejected();
break;
case Qt::Key_Return:
case Qt::Key_Enter: {
event->accept();
auto focusButton = static_cast<QPushButton *>(buttons->focusWidget());
if (focusButton && buttons->buttonRole(focusButton) == QDialogButtonBox::RejectRole) {
Q_EMIT buttons->rejected();
} else {
Q_EMIT buttons->accepted();
}
break;
}
default:
break;
}
}
void RedEyeReductionTool::toolActivated()
{
imageView()->setCursor(Qt::CrossCursor);
}
void RedEyeReductionTool::slotApplyClicked()
{
QRectF docRectF = d->rectF();
if (!docRectF.isValid()) {
qCWarning(GWENVIEW_LIB_LOG) << "invalid rect";
return;
}
auto op = new RedEyeReductionImageOperation(docRectF);
Q_EMIT imageOperationRequested(op);
d->mStatus = NotSet;
d->mToolWidget->showNotSetPage();
}
void RedEyeReductionTool::setDiameter(int value)
{
d->mDiameter = value;
imageView()->update();
}
QWidget *RedEyeReductionTool::widget() const
{
return d->mToolWidget;
}
} // namespace
#include "moc_redeyereductiontool.cpp"