diff options
Diffstat (limited to 'src/imageplugins/antivignetting/antivignettingtool.cpp')
| -rw-r--r-- | src/imageplugins/antivignetting/antivignettingtool.cpp | 378 |
1 files changed, 378 insertions, 0 deletions
diff --git a/src/imageplugins/antivignetting/antivignettingtool.cpp b/src/imageplugins/antivignetting/antivignettingtool.cpp new file mode 100644 index 00000000..d8f211c7 --- /dev/null +++ b/src/imageplugins/antivignetting/antivignettingtool.cpp @@ -0,0 +1,378 @@ +/* ============================================================ + * + * 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 <tdelocale.h> +#include <tdeaboutdata.h> +#include <kiconloader.h> +#include <tdeapplication.h> +#include <kstandarddirs.h> +#include <kseparator.h> + +// LibKDcraw includes. + +#include <libkdcraw/rnuminput.h> + +// Local includes. + +#include "daboutdata.h" +#include "ddebug.h" +#include "bcgmodifier.h" +#include "imageiface.h" +#include "imagewidget.h" +#include "editortoolsettings.h" +#include "dimgimagefilters.h" +#include "antivignetting.h" +#include "antivignettingtool.h" +#include "antivignettingtool.moc" + +using namespace KDcrawIface; +using namespace Digikam; + +namespace DigikamAntiVignettingImagesPlugin +{ + +AntiVignettingTool::AntiVignettingTool(TQObject* parent) + : EditorToolThreaded(parent) +{ + setName("antivignettings"); + setToolName(i18n("Vignetting Correction")); + setToolIcon(SmallIcon("antivignetting")); + + m_previewWidget = new ImageWidget("antivignetting Tool", 0, TQString(), + false, ImageGuideWidget::HVGuideMode, false); + + setToolView(m_previewWidget); + + // ------------------------------------------------------------- + + m_gboxSettings = new EditorToolSettings(EditorToolSettings::Default| + EditorToolSettings::Ok| + EditorToolSettings::Cancel); + TQGridLayout* grid = new TQGridLayout(m_gboxSettings->plainPage(), 14, 2); + + m_maskPreviewLabel = new TQLabel( m_gboxSettings->plainPage() ); + 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.") ); + + // ------------------------------------------------------------- + + TQLabel *label1 = new TQLabel(i18n("Density:"), m_gboxSettings->plainPage()); + + m_densityInput = new RDoubleNumInput(m_gboxSettings->plainPage()); + m_densityInput->setPrecision(1); + m_densityInput->setRange(1.0, 20.0, 0.1); + m_densityInput->setDefaultValue(2.0); + TQWhatsThis::add( m_densityInput, i18n("<p>This value controls the degree of intensity attenuation by the filter " + "at its point of maximum density.")); + + // ------------------------------------------------------------- + + TQLabel *label2 = new TQLabel(i18n("Power:"), m_gboxSettings->plainPage()); + + m_powerInput = new RDoubleNumInput(m_gboxSettings->plainPage()); + m_powerInput->setPrecision(1); + m_powerInput->setRange(0.1, 2.0, 0.1); + m_powerInput->setDefaultValue(1.0); + 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.")); + + // ------------------------------------------------------------- + + TQLabel *label3 = new TQLabel(i18n("Radius:"), m_gboxSettings->plainPage()); + + m_radiusInput = new RDoubleNumInput(m_gboxSettings->plainPage()); + m_radiusInput->setPrecision(1); + m_radiusInput->setRange(-100.0, 100.0, 0.1); + m_radiusInput->setDefaultValue(1.0); + 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.")); + + KSeparator *line = new KSeparator(Horizontal, m_gboxSettings->plainPage()); + + // ------------------------------------------------------------- + + TQLabel *label4 = new TQLabel(i18n("Brightness:"), m_gboxSettings->plainPage()); + + m_brightnessInput = new RIntNumInput(m_gboxSettings->plainPage()); + m_brightnessInput->setRange(0, 100, 1); + m_brightnessInput->setDefaultValue(0); + TQWhatsThis::add( m_brightnessInput, i18n("<p>Set here the brightness re-adjustment of the target image.")); + + // ------------------------------------------------------------- + + TQLabel *label5 = new TQLabel(i18n("Contrast:"), m_gboxSettings->plainPage()); + + m_contrastInput = new RIntNumInput(m_gboxSettings->plainPage()); + m_contrastInput->setRange(0, 100, 1); + m_contrastInput->setDefaultValue(0); + TQWhatsThis::add( m_contrastInput, i18n("<p>Set here the contrast re-adjustment of the target image.")); + + // ------------------------------------------------------------- + + TQLabel *label6 = new TQLabel(i18n("Gamma:"), m_gboxSettings->plainPage()); + + m_gammaInput = new RDoubleNumInput(m_gboxSettings->plainPage()); + m_gammaInput->setPrecision(2); + m_gammaInput->setRange(0.1, 3.0, 0.01); + m_gammaInput->setDefaultValue(1.0); + TQWhatsThis::add( m_gammaInput, i18n("<p>Set here the gamma re-adjustment of the target image.")); + + grid->addMultiCellWidget(m_maskPreviewLabel, 0, 0, 0, 2); + grid->addMultiCellWidget(label1, 1, 1, 0, 2); + grid->addMultiCellWidget(m_densityInput, 2, 2, 0, 2); + grid->addMultiCellWidget(label2, 3, 3, 0, 2); + grid->addMultiCellWidget(m_powerInput, 4, 4, 0, 2); + grid->addMultiCellWidget(label3, 5, 5, 0, 2); + grid->addMultiCellWidget(m_radiusInput, 6, 6, 0, 2); + grid->addMultiCellWidget(line, 7, 7, 0, 2); + grid->addMultiCellWidget(label4, 8, 8, 0, 2); + grid->addMultiCellWidget(m_brightnessInput, 9, 9, 0, 2); + grid->addMultiCellWidget(label5, 10, 10, 0, 2); + grid->addMultiCellWidget(m_contrastInput, 11, 11, 0, 2); + grid->addMultiCellWidget(label6, 12, 12, 0, 2); + grid->addMultiCellWidget(m_gammaInput, 13, 13, 0, 2); + grid->setRowStretch(14, 10); + grid->setMargin(m_gboxSettings->spacingHint()); + grid->setSpacing(m_gboxSettings->spacingHint()); + + setToolSettings(m_gboxSettings); + init(); + + // ------------------------------------------------------------- + + connect(m_densityInput, TQ_SIGNAL(valueChanged(double)), + this, TQ_SLOT(slotTimer())); + + connect(m_powerInput, TQ_SIGNAL(valueChanged(double)), + this, TQ_SLOT(slotTimer())); + + connect(m_radiusInput, TQ_SIGNAL(valueChanged(double)), + this, TQ_SLOT(slotTimer())); + + connect(m_brightnessInput, TQ_SIGNAL(valueChanged(int)), + this, TQ_SLOT(slotTimer())); + + connect(m_contrastInput, TQ_SIGNAL(valueChanged(int)), + this, TQ_SLOT(slotTimer())); + + connect(m_gammaInput, TQ_SIGNAL(valueChanged(double)), + this, TQ_SLOT(slotTimer())); +} + +AntiVignettingTool::~AntiVignettingTool() +{ +} + +void AntiVignettingTool::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 AntiVignettingTool::readSettings() +{ + TDEConfig* config = kapp->config(); + config->setGroup("antivignettings Tool"); + + 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", m_densityInput->defaultValue())); + m_powerInput->setValue(config->readDoubleNumEntry("PowerAjustment", m_powerInput->defaultValue())); + m_radiusInput->setValue(config->readDoubleNumEntry("RadiusAjustment", m_radiusInput->defaultValue())); + m_brightnessInput->setValue(config->readNumEntry("BrightnessAjustment", m_brightnessInput->defaultValue())); + m_contrastInput->setValue(config->readNumEntry("ContrastAjustment", m_contrastInput->defaultValue())); + m_gammaInput->setValue(config->readDoubleNumEntry("GammaAjustment", m_gammaInput->defaultValue())); + + 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 AntiVignettingTool::writeSettings() +{ + TDEConfig* config = kapp->config(); + config->setGroup("antivignettings Tool"); + 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()); + m_previewWidget->writeSettings(); + config->sync(); +} + +void AntiVignettingTool::slotResetSettings() +{ + 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->slotReset(); + m_powerInput->slotReset(); + m_radiusInput->slotReset(); + m_brightnessInput->slotReset(); + m_contrastInput->slotReset(); + m_gammaInput->slotReset(); + + 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 AntiVignettingTool::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(); + + ImageIface* iface = m_previewWidget->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. + 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 ); + + DImg orgImage(orgWidth, orgHeight, iface->originalSixteenBit(), + iface->originalHasAlpha(), data); + delete [] data; + + setFilter(dynamic_cast<DImgThreadedFilter*>(new AntiVignetting(&orgImage, this, d, p, r, 0, 0, true))); +} + +void AntiVignettingTool::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(); + + ImageIface iface(0, 0); + + uchar *data = iface.getOriginalImage(); + DImg orgImage(iface.originalWidth(), iface.originalHeight(), iface.originalSixteenBit(), + iface.originalHasAlpha(), data); + delete [] data; + + setFilter(dynamic_cast<DImgThreadedFilter*>(new AntiVignetting(&orgImage, this, d, p, r, 0, 0, true))); +} + +void AntiVignettingTool::putPreviewData(void) +{ + ImageIface* iface = m_previewWidget->imageIface(); + DImg imDest = filter()->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(); + + BCGModifier cmod; + cmod.setGamma(g); + cmod.setBrightness(b); + cmod.setContrast(c); + cmod.applyBCG(imDest); + + iface->putPreviewImage((imDest.smoothScale(iface->previewWidth(), + iface->previewHeight())).bits()); + m_previewWidget->updatePreview(); +} + +void AntiVignettingTool::putFinalData(void) +{ + ImageIface iface(0, 0); + + iface.putOriginalImage(i18n("Vignetting Correction"), + filter()->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 |