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.
digikam/digikam/imageplugins/antivignetting/imageeffect_antivignetting.cpp

381 lines
14 KiB

/* ============================================================
*
* This file is a part of digiKam project
* http://www.digikam.org
*
* Date : 2004-12-25
* Description : a digiKam image plugin to reduce
* vignetting on an image.
*
* Copyright (C) 2004-2008 by Gilles Caulier <caulier dot gilles at gmail dot com>
*
* 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.
*
* ============================================================ */
// TQt includes.
#include <tqlabel.h>
#include <tqwhatsthis.h>
#include <tqlayout.h>
#include <tqimage.h>
#include <tqpixmap.h>
#include <tqpainter.h>
#include <tqpen.h>
#include <tqtabwidget.h>
// KDE includes.
#include <tdeconfig.h>
#include <klocale.h>
#include <kaboutdata.h>
#include <kiconloader.h>
#include <kapplication.h>
#include <kstandarddirs.h>
#include <knuminput.h>
#include <kseparator.h>
// Local includes.
#include "version.h"
#include "ddebug.h"
#include "bcgmodifier.h"
#include "imageiface.h"
#include "imagewidget.h"
#include "dimgimagefilters.h"
#include "antivignetting.h"
#include "imageeffect_antivignetting.h"
#include "imageeffect_antivignetting.moc"
namespace DigikamAntiVignettingImagesPlugin
{
ImageEffect_AntiVignetting::ImageEffect_AntiVignetting(TQWidget* parent)
: Digikam::ImageGuideDlg(parent, i18n("Vignetting Correction"),
"antivignettings", false, true, false,
Digikam::ImageGuideWidget::HVGuideMode, 0, true)
{
TQString whatsThis;
TDEAboutData* about = new TDEAboutData("digikam",
I18N_NOOP("Vignetting Correction"),
digikam_version,
I18N_NOOP("A digiKam image plugin to reduce image vignetting."),
TDEAboutData::License_GPL,
"(c) 2004-2008, Gilles Caulier",
0,
"http://www.digikam.org");
about->addAuthor("Gilles Caulier", I18N_NOOP("Author and maintainer"),
"caulier dot gilles at gmail dot com");
about->addAuthor("John Walker", I18N_NOOP("Anti Vignetting algorithm"), 0,
"http://www.fourmilab.ch/netpbm/pnmctrfilt");
setAboutData(about);
// -------------------------------------------------------------
TQWidget *gboxSettings = new TQWidget(plainPage());
TQGridLayout* gridSettings = new TQGridLayout( gboxSettings, 13, 2, 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 anti-vignetting "
"mask applied to the image.") );
gridSettings->addMultiCellWidget(m_maskPreviewLabel, 0, 0, 0, 2);
// -------------------------------------------------------------
TQLabel *label1 = new TQLabel(i18n("Density:"), gboxSettings);
m_densityInput = new KDoubleNumInput(gboxSettings);
m_densityInput->setPrecision(1);
m_densityInput->setRange(1.0, 20.0, 0.1, true);
TQWhatsThis::add( m_densityInput, i18n("<p>This value controls the degree of intensity attenuation by the filter "
"at its point of maximum density."));
gridSettings->addMultiCellWidget(label1, 1, 1, 0, 2);
gridSettings->addMultiCellWidget(m_densityInput, 2, 2, 0, 2);
// -------------------------------------------------------------
TQLabel *label2 = new TQLabel(i18n("Power:"), gboxSettings);
m_powerInput = new KDoubleNumInput(gboxSettings);
m_powerInput->setPrecision(1);
m_powerInput->setRange(0.1, 2.0, 0.1, true);
TQWhatsThis::add( m_powerInput, i18n("<p>This value is used as the exponent controlling the fall-off in density "
"from the center of the filter to the periphery."));
gridSettings->addMultiCellWidget(label2, 3, 3, 0, 2);
gridSettings->addMultiCellWidget(m_powerInput, 4, 4, 0, 2);
// -------------------------------------------------------------
TQLabel *label3 = new TQLabel(i18n("Radius:"), gboxSettings);
m_radiusInput = new KDoubleNumInput(gboxSettings);
m_radiusInput->setPrecision(1);
m_radiusInput->setRange(-100.0, 100.0, 0.1, true);
TQWhatsThis::add( m_radiusInput, i18n("<p>This value is the radius of the center filter. It is a multiple of the "
"half-diagonal measure of the image, at which the density of the filter falls "
"to zero."));
gridSettings->addMultiCellWidget(label3, 5, 5, 0, 2);
gridSettings->addMultiCellWidget(m_radiusInput, 6, 6, 0, 2);
KSeparator *line = new KSeparator(Horizontal, gboxSettings);
gridSettings->addMultiCellWidget(line, 7, 7, 0, 2);
// -------------------------------------------------------------
TQLabel *label4 = new TQLabel(i18n("Brightness:"), gboxSettings);
m_brightnessInput = new KIntNumInput(gboxSettings);
m_brightnessInput->setRange(0, 100, 1, true);
TQWhatsThis::add( m_brightnessInput, i18n("<p>Set here the brightness re-adjustment of the target image."));
gridSettings->addMultiCellWidget(label4, 8, 8, 0, 2);
gridSettings->addMultiCellWidget(m_brightnessInput, 9, 9, 0, 2);
// -------------------------------------------------------------
TQLabel *label5 = new TQLabel(i18n("Contrast:"), gboxSettings);
m_contrastInput = new KIntNumInput(gboxSettings);
m_contrastInput->setRange(0, 100, 1, true);
TQWhatsThis::add( m_contrastInput, i18n("<p>Set here the contrast re-adjustment of the target image."));
gridSettings->addMultiCellWidget(label5, 10, 10, 0, 2);
gridSettings->addMultiCellWidget(m_contrastInput, 11, 11, 0, 2);
// -------------------------------------------------------------
TQLabel *label6 = new TQLabel(i18n("Gamma:"), gboxSettings);
m_gammaInput = new KDoubleNumInput(gboxSettings);
m_gammaInput->setPrecision(2);
m_gammaInput->setRange(0.1, 3.0, 0.01, true);
m_gammaInput->setValue(1.0);
TQWhatsThis::add( m_gammaInput, i18n("<p>Set here the gamma re-adjustment of the target image."));
gridSettings->addMultiCellWidget(label6, 12, 12, 0, 2);
gridSettings->addMultiCellWidget(m_gammaInput, 13, 13, 0, 2);
setUserAreaWidget(gboxSettings);
// -------------------------------------------------------------
connect(m_densityInput, TQT_SIGNAL(valueChanged (double)),
this, TQT_SLOT(slotTimer()));
connect(m_powerInput, TQT_SIGNAL(valueChanged (double)),
this, TQT_SLOT(slotTimer()));
connect(m_radiusInput, TQT_SIGNAL(valueChanged (double)),
this, TQT_SLOT(slotTimer()));
connect(m_brightnessInput, TQT_SIGNAL(valueChanged (int)),
this, TQT_SLOT(slotTimer()));
connect(m_contrastInput, TQT_SIGNAL(valueChanged (int)),
this, TQT_SLOT(slotTimer()));
connect(m_gammaInput, TQT_SIGNAL(valueChanged (double)),
this, TQT_SLOT(slotTimer()));
}
ImageEffect_AntiVignetting::~ImageEffect_AntiVignetting()
{
}
void ImageEffect_AntiVignetting::renderingFinished()
{
m_densityInput->setEnabled(true);
m_powerInput->setEnabled(true);
m_radiusInput->setEnabled(true);
m_brightnessInput->setEnabled(true);
m_contrastInput->setEnabled(true);
m_gammaInput->setEnabled(true);
}
void ImageEffect_AntiVignetting::readUserSettings()
{
TDEConfig* config = kapp->config();
config->setGroup("antivignettings Tool Dialog");
m_densityInput->blockSignals(true);
m_powerInput->blockSignals(true);
m_radiusInput->blockSignals(true);
m_brightnessInput->blockSignals(true);
m_contrastInput->blockSignals(true);
m_gammaInput->blockSignals(true);
m_densityInput->setValue(config->readDoubleNumEntry("DensityAjustment", 2.0));
m_powerInput->setValue(config->readDoubleNumEntry("PowerAjustment", 1.0));
m_radiusInput->setValue(config->readDoubleNumEntry("RadiusAjustment", 1.0));
m_brightnessInput->setValue(config->readNumEntry("BrightnessAjustment", 0));
m_contrastInput->setValue(config->readNumEntry("ContrastAjustment", 0));
m_gammaInput->setValue(config->readDoubleNumEntry("GammaAjustment", 1.0));
m_densityInput->blockSignals(false);
m_powerInput->blockSignals(false);
m_radiusInput->blockSignals(false);
m_brightnessInput->blockSignals(false);
m_contrastInput->blockSignals(false);
m_gammaInput->blockSignals(false);
slotEffect();
}
void ImageEffect_AntiVignetting::writeUserSettings()
{
TDEConfig* config = kapp->config();
config->setGroup("antivignettings Tool Dialog");
config->writeEntry("DensityAjustment", m_densityInput->value());
config->writeEntry("PowerAjustment", m_powerInput->value());
config->writeEntry("RadiusAjustment", m_radiusInput->value());
config->writeEntry("BrightnessAjustment", m_brightnessInput->value());
config->writeEntry("ContrastAjustment", m_contrastInput->value());
config->writeEntry("GammaAjustment", m_gammaInput->value());
config->sync();
}
void ImageEffect_AntiVignetting::resetValues()
{
m_densityInput->blockSignals(true);
m_powerInput->blockSignals(true);
m_radiusInput->blockSignals(true);
m_brightnessInput->blockSignals(true);
m_contrastInput->blockSignals(true);
m_gammaInput->blockSignals(true);
m_densityInput->setValue(2.0);
m_powerInput->setValue(1.0);
m_radiusInput->setValue(1.0);
m_brightnessInput->setValue(0);
m_contrastInput->setValue(0);
m_gammaInput->setValue(1.0);
m_densityInput->blockSignals(false);
m_powerInput->blockSignals(false);
m_radiusInput->blockSignals(false);
m_brightnessInput->blockSignals(false);
m_contrastInput->blockSignals(false);
m_gammaInput->blockSignals(false);
}
void ImageEffect_AntiVignetting::prepareEffect()
{
m_densityInput->setEnabled(false);
m_powerInput->setEnabled(false);
m_radiusInput->setEnabled(false);
m_brightnessInput->setEnabled(false);
m_contrastInput->setEnabled(false);
m_gammaInput->setEnabled(false);
double d = m_densityInput->value();
double p = m_powerInput->value();
double r = m_radiusInput->value();
Digikam::ImageIface* iface = m_imagePreviewWidget->imageIface();
uchar *data = iface->getOriginalImage();
int orgWidth = iface->originalWidth();
int orgHeight = iface->originalHeight();
TQSize ps(orgWidth, orgHeight);
ps.scale( TQSize(120, 120), TQSize::ScaleMin );
// Calc mask preview.
Digikam::DImg preview(ps.width(), ps.height(), false);
memset(preview.bits(), 255, preview.numBytes());
AntiVignetting maskPreview(&preview, 0L, d, p, r, 0, 0, false);
TQPixmap pix = maskPreview.getTargetImage().convertToPixmap();
TQPainter pt(&pix);
pt.setPen( TQPen(TQt::black, 1) );
pt.drawRect( 0, 0, pix.width(), pix.height() );
pt.end();
m_maskPreviewLabel->setPixmap( pix );
Digikam::DImg orgImage(orgWidth, orgHeight, iface->originalSixteenBit(),
iface->originalHasAlpha(), data);
delete [] data;
m_threadedFilter = dynamic_cast<Digikam::DImgThreadedFilter *>(
new AntiVignetting(&orgImage, this, d, p, r, 0, 0, true));
}
void ImageEffect_AntiVignetting::prepareFinal()
{
m_densityInput->setEnabled(false);
m_powerInput->setEnabled(false);
m_radiusInput->setEnabled(false);
m_brightnessInput->setEnabled(false);
m_contrastInput->setEnabled(false);
m_gammaInput->setEnabled(false);
double d = m_densityInput->value();
double p = m_powerInput->value();
double r = m_radiusInput->value();
Digikam::ImageIface iface(0, 0);
uchar *data = iface.getOriginalImage();
Digikam::DImg orgImage(iface.originalWidth(), iface.originalHeight(), iface.originalSixteenBit(),
iface.originalHasAlpha(), data);
delete [] data;
m_threadedFilter = dynamic_cast<Digikam::DImgThreadedFilter *>(
new AntiVignetting(&orgImage, this, d, p, r, 0, 0, true));
}
void ImageEffect_AntiVignetting::putPreviewData(void)
{
Digikam::ImageIface* iface = m_imagePreviewWidget->imageIface();
Digikam::DImg imDest = m_threadedFilter->getTargetImage();
// Adjust Image BCG.
double b = (double)(m_brightnessInput->value() / 100.0);
double c = (double)(m_contrastInput->value() / 100.0) + (double)(1.00);
double g = m_gammaInput->value();
Digikam::BCGModifier cmod;
cmod.setGamma(g);
cmod.setBrightness(b);
cmod.setContrast(c);
cmod.applyBCG(imDest);
iface->putPreviewImage((imDest.smoothScale(iface->previewWidth(),
iface->previewHeight())).bits());
m_imagePreviewWidget->updatePreview();
}
void ImageEffect_AntiVignetting::putFinalData(void)
{
Digikam::ImageIface iface(0, 0);
iface.putOriginalImage(i18n("Vignetting Correction"),
m_threadedFilter->getTargetImage().bits());
double b = (double)(m_brightnessInput->value() / 100.0);
double c = (double)(m_contrastInput->value() / 100.0) + (double)(1.00);
double g = m_gammaInput->value();
// Adjust Image BCG.
iface.setOriginalBCG(b, c, g);
}
} // NameSpace DigikamAntiVignettingImagesPlugin