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/coreplugin/imageeffect_rgb.cpp

420 lines
16 KiB

/* ============================================================
*
* This file is a part of digiKam project
* http://www.digikam.org
*
* Date : 2004-07-11
* Description : digiKam image editor Color Balance tool.
*
* Copyright (C) 2004-2007 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 <tqspinbox.h>
#include <tqslider.h>
#include <tqcolor.h>
#include <tqgroupbox.h>
#include <tqhgroupbox.h>
#include <tqvgroupbox.h>
#include <tqhbuttongroup.h>
#include <tqlabel.h>
#include <tqlayout.h>
#include <tqframe.h>
#include <tqvbox.h>
#include <tqlabel.h>
#include <tqpushbutton.h>
#include <tqcheckbox.h>
#include <tqcombobox.h>
#include <tqwhatsthis.h>
#include <tqtooltip.h>
// KDE includes.
#include <tdeconfig.h>
#include <tdelocale.h>
#include <tdeapplication.h>
#include <kcursor.h>
#include <kstandarddirs.h>
// Digikam includes.
#include "imageiface.h"
#include "imagewidget.h"
#include "histogramwidget.h"
#include "colorgradientwidget.h"
#include "colormodifier.h"
#include "dimg.h"
// Local includes.
#include "imageeffect_rgb.h"
#include "imageeffect_rgb.moc"
namespace DigikamImagesPluginCore
{
ImageEffect_RGB::ImageEffect_RGB(TQWidget* parent)
: Digikam::ImageDlgBase(parent, i18n("Color Balance"), "colorbalance", false)
{
m_destinationPreviewData = 0L;
setHelp("colorbalancetool.anchor", "digikam");
m_previewWidget = new Digikam::ImageWidget("colorbalance Tool Dialog", plainPage(),
i18n("<p>Here you can see the image "
"color-balance adjustments preview. "
"You can pick color on image "
"to see the color level corresponding on histogram."));
setPreviewAreaWidget(m_previewWidget);
// -------------------------------------------------------------
TQWidget *gboxSettings = new TQWidget(plainPage());
TQGridLayout* gridSettings = new TQGridLayout( gboxSettings, 7, 4, spacingHint());
TQLabel *label1 = new TQLabel(i18n("Channel:"), gboxSettings);
label1->setAlignment ( TQt::AlignRight | TQt::AlignVCenter );
m_channelCB = new TQComboBox( false, gboxSettings );
m_channelCB->insertItem( i18n("Luminosity") );
m_channelCB->insertItem( i18n("Red") );
m_channelCB->insertItem( i18n("Green") );
m_channelCB->insertItem( i18n("Blue") );
TQWhatsThis::add( m_channelCB, i18n("<p>Select the histogram channel to display here:<p>"
"<b>Luminosity</b>: display the image's luminosity values.<p>"
"<b>Red</b>: display the red image-channel values.<p>"
"<b>Green</b>: display the green image-channel values.<p>"
"<b>Blue</b>: display the blue image-channel values.<p>"));
m_scaleBG = new TQHButtonGroup(gboxSettings);
m_scaleBG->setExclusive(true);
m_scaleBG->setFrameShape(TQFrame::NoFrame);
m_scaleBG->setInsideMargin( 0 );
TQWhatsThis::add( m_scaleBG, i18n("<p>Select the histogram scale here.<p>"
"If the image's maximal counts are small, you can use the linear scale.<p>"
"Logarithmic scale can be used when the maximal counts are big; "
"if it is used, all values (small and large) will be visible on the graph."));
TQPushButton *linHistoButton = new TQPushButton( m_scaleBG );
TQToolTip::add( linHistoButton, i18n( "<p>Linear" ) );
m_scaleBG->insert(linHistoButton, Digikam::HistogramWidget::LinScaleHistogram);
TDEGlobal::dirs()->addResourceType("histogram-lin", TDEGlobal::dirs()->kde_default("data") + "digikam/data");
TQString directory = TDEGlobal::dirs()->findResourceDir("histogram-lin", "histogram-lin.png");
linHistoButton->setPixmap( TQPixmap( directory + "histogram-lin.png" ) );
linHistoButton->setToggleButton(true);
TQPushButton *logHistoButton = new TQPushButton( m_scaleBG );
TQToolTip::add( logHistoButton, i18n( "<p>Logarithmic" ) );
m_scaleBG->insert(logHistoButton, Digikam::HistogramWidget::LogScaleHistogram);
TDEGlobal::dirs()->addResourceType("histogram-log", TDEGlobal::dirs()->kde_default("data") + "digikam/data");
directory = TDEGlobal::dirs()->findResourceDir("histogram-log", "histogram-log.png");
logHistoButton->setPixmap( TQPixmap( directory + "histogram-log.png" ) );
logHistoButton->setToggleButton(true);
TQHBoxLayout* l1 = new TQHBoxLayout();
l1->addWidget(label1);
l1->addWidget(m_channelCB);
l1->addStretch(10);
l1->addWidget(m_scaleBG);
gridSettings->addMultiCellLayout(l1, 0, 0, 0, 4);
// -------------------------------------------------------------
TQVBox *histoBox = new TQVBox(gboxSettings);
m_histogramWidget = new Digikam::HistogramWidget(256, 140, histoBox, false, true, true);
TQWhatsThis::add( m_histogramWidget, i18n("<p>Here you can see the target preview image histogram drawing "
"of the selected image channel. This one is re-computed at any "
"settings changes."));
TQLabel *space = new TQLabel(histoBox);
space->setFixedHeight(1);
m_hGradient = new Digikam::ColorGradientWidget( Digikam::ColorGradientWidget::Horizontal, 10, histoBox );
m_hGradient->setColors( TQColor( "black" ), TQColor( "white" ) );
gridSettings->addMultiCellWidget(histoBox, 1, 2, 0, 4);
// -------------------------------------------------------------
TQLabel *labelLeft = new TQLabel(i18n("Cyan"), gboxSettings);
labelLeft->setAlignment ( TQt::AlignRight | TQt::AlignVCenter );
m_rSlider = new TQSlider(-100, 100, 1, 0, Qt::Horizontal, gboxSettings, "m_rSlider");
m_rSlider->setTickmarks(TQSlider::Below);
m_rSlider->setTickInterval(20);
TQWhatsThis::add( m_rSlider, i18n("<p>Set here the cyan/red color adjustment of the image."));
TQLabel *labelRight = new TQLabel(i18n("Red"), gboxSettings);
labelRight->setAlignment ( TQt::AlignLeft | TQt::AlignVCenter );
m_rInput = new TQSpinBox(-100, 100, 1, gboxSettings, "m_rInput");
gridSettings->addMultiCellWidget(labelLeft, 3, 3, 0, 0);
gridSettings->addMultiCellWidget(m_rSlider, 3, 3, 1, 1);
gridSettings->addMultiCellWidget(labelRight, 3, 3, 2, 2);
gridSettings->addMultiCellWidget(m_rInput, 3, 3, 3, 3);
// -------------------------------------------------------------
labelLeft = new TQLabel(i18n("Magenta"), gboxSettings);
labelLeft->setAlignment ( TQt::AlignRight | TQt::AlignVCenter );
m_gSlider = new TQSlider(-100, 100, 1, 0, Qt::Horizontal, gboxSettings, "m_gSlider");
m_gSlider->setTickmarks(TQSlider::Below);
m_gSlider->setTickInterval(20);
TQWhatsThis::add( m_gSlider, i18n("<p>Set here the magenta/green color adjustment of the image."));
labelRight = new TQLabel(i18n("Green"), gboxSettings);
labelRight->setAlignment ( TQt::AlignLeft | TQt::AlignVCenter );
m_gInput = new TQSpinBox(-100, 100, 1, gboxSettings, "m_gInput");
gridSettings->addMultiCellWidget(labelLeft, 4, 4, 0, 0);
gridSettings->addMultiCellWidget(m_gSlider, 4, 4, 1, 1);
gridSettings->addMultiCellWidget(labelRight, 4, 4, 2, 2);
gridSettings->addMultiCellWidget(m_gInput, 4, 4, 3, 3);
// -------------------------------------------------------------
labelLeft = new TQLabel(i18n("Yellow"), gboxSettings);
labelLeft->setAlignment ( TQt::AlignRight | TQt::AlignVCenter );
m_bSlider = new TQSlider(-100, 100, 1, 0, Qt::Horizontal, gboxSettings, "m_bSlider");
m_bSlider->setTickmarks(TQSlider::Below);
m_bSlider->setTickInterval(20);
TQWhatsThis::add( m_bSlider, i18n("<p>Set here the yellow/blue color adjustment of the image."));
labelRight = new TQLabel(i18n("Blue"), gboxSettings);
labelRight->setAlignment ( TQt::AlignLeft | TQt::AlignVCenter );
m_bInput = new TQSpinBox(-100, 100, 1, gboxSettings, "m_bInput");
gridSettings->addMultiCellWidget(labelLeft, 5, 5, 0, 0);
gridSettings->addMultiCellWidget(m_bSlider, 5, 5, 1, 1);
gridSettings->addMultiCellWidget(labelRight, 5, 5, 2, 2);
gridSettings->addMultiCellWidget(m_bInput, 5, 5, 3, 3);
m_rInput->setValue(0);
m_gInput->setValue(0);
m_bInput->setValue(0);
gridSettings->setRowStretch(6, 10);
setUserAreaWidget(gboxSettings);
// -------------------------------------------------------------
connect(m_channelCB, TQT_SIGNAL(activated(int)),
this, TQT_SLOT(slotChannelChanged(int)));
connect(m_scaleBG, TQT_SIGNAL(released(int)),
this, TQT_SLOT(slotScaleChanged(int)));
connect(m_previewWidget, TQT_SIGNAL(spotPositionChangedFromTarget( const Digikam::DColor &, const TQPoint & )),
this, TQT_SLOT(slotColorSelectedFromTarget( const Digikam::DColor & )));
connect(m_rSlider, TQT_SIGNAL(valueChanged(int)),
m_rInput, TQT_SLOT(setValue(int)));
connect(m_rInput, TQT_SIGNAL(valueChanged (int)),
m_rSlider, TQT_SLOT(setValue(int)));
connect(m_rInput, TQT_SIGNAL(valueChanged (int)),
this, TQT_SLOT(slotTimer()));
connect(m_gSlider, TQT_SIGNAL(valueChanged(int)),
m_gInput, TQT_SLOT(setValue(int)));
connect(m_gInput, TQT_SIGNAL(valueChanged (int)),
m_gSlider, TQT_SLOT(setValue(int)));
connect(m_gInput, TQT_SIGNAL(valueChanged (int)),
this, TQT_SLOT(slotTimer()));
connect(m_bSlider, TQT_SIGNAL(valueChanged(int)),
m_bInput, TQT_SLOT(setValue(int)));
connect(m_bInput, TQT_SIGNAL(valueChanged (int)),
m_bSlider, TQT_SLOT(setValue(int)));
connect(m_bInput, TQT_SIGNAL(valueChanged (int)),
this, TQT_SLOT(slotTimer()));
connect(m_previewWidget, TQT_SIGNAL(signalResized()),
this, TQT_SLOT(slotEffect()));
// -------------------------------------------------------------
enableButtonOK( false );
}
ImageEffect_RGB::~ImageEffect_RGB()
{
m_histogramWidget->stopHistogramComputation();
if (m_destinationPreviewData)
delete [] m_destinationPreviewData;
delete m_histogramWidget;
delete m_previewWidget;
}
void ImageEffect_RGB::slotChannelChanged(int channel)
{
switch(channel)
{
case LuminosityChannel:
m_histogramWidget->m_channelType = Digikam::HistogramWidget::ValueHistogram;
m_hGradient->setColors( TQColor( "black" ), TQColor( "white" ) );
break;
case RedChannel:
m_histogramWidget->m_channelType = Digikam::HistogramWidget::RedChannelHistogram;
m_hGradient->setColors( TQColor( "black" ), TQColor( "red" ) );
break;
case GreenChannel:
m_histogramWidget->m_channelType = Digikam::HistogramWidget::GreenChannelHistogram;
m_hGradient->setColors( TQColor( "black" ), TQColor( "green" ) );
break;
case BlueChannel:
m_histogramWidget->m_channelType = Digikam::HistogramWidget::BlueChannelHistogram;
m_hGradient->setColors( TQColor( "black" ), TQColor( "blue" ) );
break;
}
m_histogramWidget->repaint(false);
}
void ImageEffect_RGB::slotScaleChanged(int scale)
{
m_histogramWidget->m_scaleType = scale;
m_histogramWidget->repaint(false);
}
void ImageEffect_RGB::slotColorSelectedFromTarget( const Digikam::DColor &color )
{
m_histogramWidget->setHistogramGuideByColor(color);
}
void ImageEffect_RGB::readUserSettings()
{
TDEConfig* config = kapp->config();
config->setGroup("colorbalance Tool Dialog");
m_channelCB->setCurrentItem(config->readNumEntry("Histogram Channel", 0)); // Luminosity.
m_scaleBG->setButton(config->readNumEntry("Histogram Scale", Digikam::HistogramWidget::LogScaleHistogram));
int r = config->readNumEntry("RedAjustment", 0);
int g = config->readNumEntry("GreenAjustment", 0);
int b = config->readNumEntry("BlueAjustment", 0);
adjustSliders(r, g, b);
slotChannelChanged(m_channelCB->currentItem());
slotScaleChanged(m_scaleBG->selectedId());
}
void ImageEffect_RGB::writeUserSettings()
{
TDEConfig* config = kapp->config();
config->setGroup("colorbalance Tool Dialog");
config->writeEntry("Histogram Channel", m_channelCB->currentItem());
config->writeEntry("Histogram Scale", m_scaleBG->selectedId());
config->writeEntry("RedAjustment", m_rSlider->value());
config->writeEntry("GreenAjustment", m_gInput->value());
config->writeEntry("BlueAjustment", m_bInput->value());
config->sync();
}
void ImageEffect_RGB::resetValues()
{
adjustSliders(0, 0, 0);
}
void ImageEffect_RGB::adjustSliders(int r, int g, int b)
{
m_rSlider->blockSignals(true);
m_gSlider->blockSignals(true);
m_bSlider->blockSignals(true);
m_rInput->blockSignals(true);
m_gInput->blockSignals(true);
m_bInput->blockSignals(true);
m_rSlider->setValue(r);
m_gSlider->setValue(g);
m_bSlider->setValue(b);
m_rInput->setValue(r);
m_gInput->setValue(g);
m_bInput->setValue(b);
m_rSlider->blockSignals(false);
m_gSlider->blockSignals(false);
m_bSlider->blockSignals(false);
m_rInput->blockSignals(false);
m_gInput->blockSignals(false);
m_bInput->blockSignals(false);
slotEffect();
}
void ImageEffect_RGB::slotEffect()
{
kapp->setOverrideCursor( KCursor::waitCursor() );
enableButtonOK(m_rInput->value() != 0 ||
m_gInput->value() != 0 ||
m_bInput->value() != 0);
m_histogramWidget->stopHistogramComputation();
if (m_destinationPreviewData)
delete [] m_destinationPreviewData;
Digikam::ImageIface* iface = m_previewWidget->imageIface();
m_destinationPreviewData = iface->getPreviewImage();
int w = iface->previewWidth();
int h = iface->previewHeight();
bool alpha = iface->previewHasAlpha();
bool sixteenBit = iface->previewSixteenBit();
double r = ((double)m_rInput->value() + 100.0)/100.0;
double g = ((double)m_gInput->value() + 100.0)/100.0;
double b = ((double)m_bInput->value() + 100.0)/100.0;
double a = 1.0;
Digikam::DImg preview(w, h, sixteenBit, alpha, m_destinationPreviewData);
Digikam::ColorModifier cmod;
cmod.applyColorModifier(preview, r, g, b, a);
iface->putPreviewImage(preview.bits());
m_previewWidget->updatePreview();
// Update histogram.
memcpy(m_destinationPreviewData, preview.bits(), preview.numBytes());
m_histogramWidget->updateData(m_destinationPreviewData, w, h, sixteenBit, 0, 0, 0, false);
kapp->restoreOverrideCursor();
}
void ImageEffect_RGB::finalRendering()
{
kapp->setOverrideCursor( KCursor::waitCursor() );
double r = ((double)m_rInput->value() + 100.0)/100.0;
double g = ((double)m_gInput->value() + 100.0)/100.0;
double b = ((double)m_bInput->value() + 100.0)/100.0;
double a = 1.0;
Digikam::ImageIface* iface = m_previewWidget->imageIface();
uchar *data = iface->getOriginalImage();
int w = iface->originalWidth();
int h = iface->originalHeight();
bool alpha = iface->originalHasAlpha();
bool sixteenBit = iface->originalSixteenBit();
Digikam::DImg original(w, h, sixteenBit, alpha, data);
delete [] data;
Digikam::ColorModifier cmod;
cmod.applyColorModifier(original, r, g, b, a);
iface->putOriginalImage(i18n("Color Balance"), original.bits());
kapp->restoreOverrideCursor();
accept();
}
} // NameSpace DigikamImagesPluginCore