You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
318 lines
12 KiB
318 lines
12 KiB
/* ============================================================
|
|
*
|
|
* This file is a part of digiKam project
|
|
* http://www.digikam.org
|
|
*
|
|
* Date : 2004-12-27
|
|
* Description : a plugin to reduce lens distorsions to an image.
|
|
*
|
|
* Copyright (C) 2004-2008 by Gilles Caulier <caulier dot gilles at gmail dot com>
|
|
* Copyright (C) 2006-2008 by Marcel Wiesweg <marcel dot wiesweg at gmx dot de>
|
|
*
|
|
* 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, 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.
|
|
*
|
|
* ============================================================ */
|
|
|
|
// C++ include.
|
|
|
|
#include <cstring>
|
|
#include <cmath>
|
|
#include <cstdlib>
|
|
|
|
// TQt includes.
|
|
|
|
#include <tqlabel.h>
|
|
#include <tqwhatsthis.h>
|
|
#include <tqlayout.h>
|
|
#include <tqpixmap.h>
|
|
#include <tqpainter.h>
|
|
#include <tqbrush.h>
|
|
#include <tqpen.h>
|
|
|
|
// KDE includes.
|
|
|
|
#include <klocale.h>
|
|
#include <tdeconfig.h>
|
|
#include <kaboutdata.h>
|
|
#include <kiconloader.h>
|
|
#include <kapplication.h>
|
|
#include <kstandarddirs.h>
|
|
#include <knuminput.h>
|
|
|
|
// Local includes.
|
|
|
|
#include "version.h"
|
|
#include "ddebug.h"
|
|
#include "imageiface.h"
|
|
#include "imagewidget.h"
|
|
#include "lensdistortion.h"
|
|
#include "imageeffect_lensdistortion.h"
|
|
#include "imageeffect_lensdistortion.moc"
|
|
|
|
namespace DigikamLensDistortionImagesPlugin
|
|
{
|
|
|
|
ImageEffect_LensDistortion::ImageEffect_LensDistortion(TQWidget* parent)
|
|
: Digikam::ImageGuideDlg(parent, i18n("Lens Distortion Correction"),
|
|
"lensdistortion", false, true, true,
|
|
Digikam::ImageGuideWidget::HVGuideMode)
|
|
{
|
|
TQString whatsThis;
|
|
|
|
TDEAboutData* about = new TDEAboutData("digikam",
|
|
I18N_NOOP("Lens Distortion Correction"),
|
|
digikam_version,
|
|
I18N_NOOP("A digiKam image plugin to reduce spherical aberration caused "
|
|
"by a lens to an image."),
|
|
TDEAboutData::License_GPL,
|
|
"(c) 2004-2006, Gilles Caulier\n"
|
|
"(c) 2006-2008, Gilles Caulier and Marcel Wiesweg",
|
|
0,
|
|
"http://www.digikam.org");
|
|
|
|
about->addAuthor("Gilles Caulier", I18N_NOOP("Author and maintainer"),
|
|
"caulier dot gilles at gmail dot com");
|
|
|
|
about->addAuthor("Marcel Wiesweg", I18N_NOOP("Developer"),
|
|
"marcel dot wiesweg at gmx dot de");
|
|
|
|
about->addAuthor("David Hodson", I18N_NOOP("Lens distortion correction algorithm."),
|
|
"hodsond at acm dot org");
|
|
|
|
setAboutData(about);
|
|
|
|
// -------------------------------------------------------------
|
|
|
|
TQWidget *gboxSettings = new TQWidget(plainPage());
|
|
TQGridLayout* gridSettings = new TQGridLayout( gboxSettings, 8, 1, spacingHint());
|
|
|
|
m_maskPreviewLabel = new TQLabel( gboxSettings );
|
|
m_maskPreviewLabel->setAlignment ( TQt::AlignHCenter | TQt::AlignVCenter );
|
|
TQWhatsThis::add( m_maskPreviewLabel, i18n("<p>You can see here a thumbnail preview of the distortion correction "
|
|
"applied to a cross pattern.") );
|
|
gridSettings->addMultiCellWidget(m_maskPreviewLabel, 0, 0, 0, 1);
|
|
|
|
// -------------------------------------------------------------
|
|
|
|
TQLabel *label1 = new TQLabel(i18n("Main:"), gboxSettings);
|
|
|
|
m_mainInput = new KDoubleNumInput(gboxSettings);
|
|
m_mainInput->setPrecision(1);
|
|
m_mainInput->setRange(-100.0, 100.0, 0.1, true);
|
|
TQWhatsThis::add( m_mainInput, i18n("<p>This value controls the amount of distortion. Negative values correct lens barrel "
|
|
"distortion, while positive values correct lens pincushion distortion."));
|
|
|
|
gridSettings->addMultiCellWidget(label1, 1, 1, 0, 1);
|
|
gridSettings->addMultiCellWidget(m_mainInput, 2, 2, 0, 1);
|
|
|
|
// -------------------------------------------------------------
|
|
|
|
TQLabel *label2 = new TQLabel(i18n("Edge:"), gboxSettings);
|
|
|
|
m_edgeInput = new KDoubleNumInput(gboxSettings);
|
|
m_edgeInput->setPrecision(1);
|
|
m_edgeInput->setRange(-100.0, 100.0, 0.1, true);
|
|
TQWhatsThis::add( m_edgeInput, i18n("<p>This value controls in the same manner as the Main control, but has more effect "
|
|
"at the edges of the image than at the center."));
|
|
|
|
gridSettings->addMultiCellWidget(label2, 3, 3, 0, 1);
|
|
gridSettings->addMultiCellWidget(m_edgeInput, 4, 4, 0, 1);
|
|
|
|
// -------------------------------------------------------------
|
|
|
|
TQLabel *label3 = new TQLabel(i18n("Zoom:"), gboxSettings);
|
|
|
|
m_rescaleInput = new KDoubleNumInput(gboxSettings);
|
|
m_rescaleInput->setPrecision(1);
|
|
m_rescaleInput->setRange(-100.0, 100.0, 0.1, true);
|
|
TQWhatsThis::add( m_rescaleInput, i18n("<p>This value rescales the overall image size."));
|
|
|
|
gridSettings->addMultiCellWidget(label3, 5, 5, 0, 1);
|
|
gridSettings->addMultiCellWidget(m_rescaleInput, 6, 6, 0, 1);
|
|
|
|
// -------------------------------------------------------------
|
|
|
|
TQLabel *label4 = new TQLabel(i18n("Brighten:"), gboxSettings);
|
|
|
|
m_brightenInput = new KDoubleNumInput(gboxSettings);
|
|
m_brightenInput->setPrecision(1);
|
|
m_brightenInput->setRange(-100.0, 100.0, 0.1, true);
|
|
TQWhatsThis::add( m_brightenInput, i18n("<p>This value adjusts the brightness in image corners."));
|
|
|
|
gridSettings->addMultiCellWidget(label4, 7, 7, 0, 1);
|
|
gridSettings->addMultiCellWidget(m_brightenInput, 8, 8, 0, 1);
|
|
|
|
setUserAreaWidget(gboxSettings);
|
|
|
|
// -------------------------------------------------------------
|
|
|
|
connect(m_mainInput, TQT_SIGNAL(valueChanged (double)),
|
|
this, TQT_SLOT(slotTimer()));
|
|
|
|
connect(m_edgeInput, TQT_SIGNAL(valueChanged (double)),
|
|
this, TQT_SLOT(slotTimer()));
|
|
|
|
connect(m_rescaleInput, TQT_SIGNAL(valueChanged (double)),
|
|
this, TQT_SLOT(slotTimer()));
|
|
|
|
connect(m_brightenInput, TQT_SIGNAL(valueChanged (double)),
|
|
this, TQT_SLOT(slotTimer()));
|
|
|
|
// -------------------------------------------------------------
|
|
|
|
/* Calc transform preview.
|
|
We would like a checkered area to demonstrate the effect.
|
|
We do not have any drawing support in DImg, so we let TQt draw.
|
|
First we create a white TQImage. We convert this to a TQPixmap,
|
|
on which we can draw. Then we convert back to TQImage,
|
|
convert the TQImage to a DImg which we only need to create once, here.
|
|
Later, we apply the effect on a copy and convert the DImg to TQPixmap.
|
|
Longing for TQt4 where we can paint directly on the TQImage...
|
|
*/
|
|
|
|
TQImage preview(120, 120, 32);
|
|
memset(preview.bits(), 255, preview.numBytes());
|
|
TQPixmap pix (preview);
|
|
TQPainter pt(&pix);
|
|
pt.setPen( TQPen(TQt::black, 1) );
|
|
pt.fillRect( 0, 0, pix.width(), pix.height(), TQBrush(TQt::black, TQt::CrossPattern) );
|
|
pt.drawRect( 0, 0, pix.width(), pix.height() );
|
|
pt.end();
|
|
TQImage preview2(pix.convertToImage());
|
|
m_previewRasterImage = Digikam::DImg(preview2.width(), preview2.height(), false, false, preview2.bits());
|
|
}
|
|
|
|
ImageEffect_LensDistortion::~ImageEffect_LensDistortion()
|
|
{
|
|
}
|
|
|
|
void ImageEffect_LensDistortion::readUserSettings(void)
|
|
{
|
|
TDEConfig *config = kapp->config();
|
|
config->setGroup("lensdistortion Tool Dialog");
|
|
|
|
m_mainInput->blockSignals(true);
|
|
m_edgeInput->blockSignals(true);
|
|
m_rescaleInput->blockSignals(true);
|
|
m_brightenInput->blockSignals(true);
|
|
|
|
m_mainInput->setValue(config->readDoubleNumEntry("2nd Order Distortion", 0.0));
|
|
m_edgeInput->setValue(config->readDoubleNumEntry("4th Order Distortion",0.0));
|
|
m_rescaleInput->setValue(config->readDoubleNumEntry("Zoom Factor", 0.0));
|
|
m_brightenInput->setValue(config->readDoubleNumEntry("Brighten", 0.0));
|
|
|
|
m_mainInput->blockSignals(false);
|
|
m_edgeInput->blockSignals(false);
|
|
m_rescaleInput->blockSignals(false);
|
|
m_brightenInput->blockSignals(false);
|
|
|
|
slotEffect();
|
|
}
|
|
|
|
void ImageEffect_LensDistortion::writeUserSettings(void)
|
|
{
|
|
TDEConfig *config = kapp->config();
|
|
config->setGroup("lensdistortion Tool Dialog");
|
|
config->writeEntry("2nd Order Distortion", m_mainInput->value());
|
|
config->writeEntry("4th Order Distortion", m_edgeInput->value());
|
|
config->writeEntry("Zoom Factor", m_rescaleInput->value());
|
|
config->writeEntry("Brighten", m_brightenInput->value());
|
|
config->sync();
|
|
}
|
|
|
|
void ImageEffect_LensDistortion::resetValues()
|
|
{
|
|
m_mainInput->blockSignals(true);
|
|
m_edgeInput->blockSignals(true);
|
|
m_rescaleInput->blockSignals(true);
|
|
m_brightenInput->blockSignals(true);
|
|
|
|
m_mainInput->setValue(0.0);
|
|
m_edgeInput->setValue(0.0);
|
|
m_rescaleInput->setValue(0.0);
|
|
m_brightenInput->setValue(0.0);
|
|
|
|
m_mainInput->blockSignals(false);
|
|
m_edgeInput->blockSignals(false);
|
|
m_rescaleInput->blockSignals(false);
|
|
m_brightenInput->blockSignals(false);
|
|
}
|
|
|
|
void ImageEffect_LensDistortion::prepareEffect()
|
|
{
|
|
m_mainInput->setEnabled(false);
|
|
m_edgeInput->setEnabled(false);
|
|
m_rescaleInput->setEnabled(false);
|
|
m_brightenInput->setEnabled(false);
|
|
|
|
double m = m_mainInput->value();
|
|
double e = m_edgeInput->value();
|
|
double r = m_rescaleInput->value();
|
|
double b = m_brightenInput->value();
|
|
|
|
LensDistortion transformPreview(&m_previewRasterImage, 0L, m, e, r, b, 0, 0);
|
|
m_maskPreviewLabel->setPixmap(TQPixmap::TQPixmap(transformPreview.getTargetImage().convertToPixmap()));
|
|
|
|
Digikam::ImageIface* iface = m_imagePreviewWidget->imageIface();
|
|
|
|
m_threadedFilter = dynamic_cast<Digikam::DImgThreadedFilter *>(
|
|
new LensDistortion(iface->getOriginalImg(), this, m, e, r, b, 0, 0));
|
|
}
|
|
|
|
void ImageEffect_LensDistortion::prepareFinal()
|
|
{
|
|
m_mainInput->setEnabled(false);
|
|
m_edgeInput->setEnabled(false);
|
|
m_rescaleInput->setEnabled(false);
|
|
m_brightenInput->setEnabled(false);
|
|
|
|
double m = m_mainInput->value();
|
|
double e = m_edgeInput->value();
|
|
double r = m_rescaleInput->value();
|
|
double b = m_brightenInput->value();
|
|
|
|
Digikam::ImageIface iface(0, 0);
|
|
|
|
m_threadedFilter = dynamic_cast<Digikam::DImgThreadedFilter *>(
|
|
new LensDistortion(iface.getOriginalImg(), this, m, e, r, b, 0, 0));
|
|
}
|
|
|
|
void ImageEffect_LensDistortion::putPreviewData(void)
|
|
{
|
|
Digikam::ImageIface* iface = m_imagePreviewWidget->imageIface();
|
|
|
|
Digikam::DImg imDest = m_threadedFilter->getTargetImage()
|
|
.smoothScale(iface->previewWidth(), iface->previewHeight());
|
|
iface->putPreviewImage(imDest.bits());
|
|
|
|
m_imagePreviewWidget->updatePreview();
|
|
}
|
|
|
|
void ImageEffect_LensDistortion::putFinalData(void)
|
|
{
|
|
Digikam::ImageIface iface(0, 0);
|
|
|
|
iface.putOriginalImage(i18n("Lens Distortion"),
|
|
m_threadedFilter->getTargetImage().bits());
|
|
}
|
|
|
|
void ImageEffect_LensDistortion::renderingFinished()
|
|
{
|
|
m_mainInput->setEnabled(true);
|
|
m_edgeInput->setEnabled(true);
|
|
m_rescaleInput->setEnabled(true);
|
|
m_brightenInput->setEnabled(true);
|
|
}
|
|
|
|
} // NameSpace DigikamLensDistortionImagesPlugin
|
|
|