1 files changed, 380 insertions, 0 deletions

diff --git a/src/imageplugins/antivignetting/imageeffect_antivignetting.cpp b/src/imageplugins/antivignetting/imageeffect_antivignetting.cpp
new file mode 100644
index 00000000..7d7a00c6
--- /dev/null
+++ b/src/imageplugins/antivignetting/imageeffect_antivignetting.cpp
@@ -0,0 +1,380 @@
+/* ============================================================
+ *
+ * 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 <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, 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()));      
+}
+
+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
+