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.
131 lines
4.3 KiB
131 lines
4.3 KiB
/*
|
|
* This file is part of the KDE project
|
|
*
|
|
* Copyright (c) 2005 Cyrille Berger <cberger@cberger.net>
|
|
* 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 of the License, 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.
|
|
*
|
|
* You should have received a copy of the GNU General Public License
|
|
* along with this program; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
|
|
*/
|
|
|
|
#include "kis_wavelet_noise_reduction.h"
|
|
|
|
#include <cmath>
|
|
|
|
#include <kis_iterators_pixel.h>
|
|
#include <kis_layer.h>
|
|
#include <kis_math_toolbox.h>
|
|
#include <kis_meta_registry.h>
|
|
#include <kis_multi_double_filter_widget.h>
|
|
#include <kis_paint_device.h>
|
|
|
|
KisWaveletNoiseReduction::KisWaveletNoiseReduction()
|
|
: KisFilter(id(), "enhance", i18n("&Wavelet Noise Reduction..."))
|
|
{
|
|
}
|
|
|
|
|
|
KisWaveletNoiseReduction::~KisWaveletNoiseReduction()
|
|
{
|
|
}
|
|
|
|
KisFilterConfigWidget * KisWaveletNoiseReduction::createConfigurationWidget(TQWidget* parent, KisPaintDeviceSP )
|
|
{
|
|
vKisDoubleWidgetParam param;
|
|
param.push_back( KisDoubleWidgetParam( 0.0, 256.0, BEST_WAVELET_THRESHOLD_VALUE, i18n("Threshold"), "threshold" ) );
|
|
return new KisMultiDoubleFilterWidget(parent, id().id().ascii(), id().id().ascii(), param );
|
|
}
|
|
|
|
KisFilterConfiguration* KisWaveletNoiseReduction::configuration(TQWidget* nwidget )
|
|
{
|
|
KisMultiDoubleFilterWidget* widget = (KisMultiDoubleFilterWidget*) nwidget;
|
|
if( widget == 0 )
|
|
{
|
|
return new KisWaveletNoiseReductionConfiguration( BEST_WAVELET_THRESHOLD_VALUE );
|
|
} else {
|
|
return new KisWaveletNoiseReductionConfiguration( widget->valueAt( 0 ) );
|
|
}
|
|
}
|
|
|
|
void KisWaveletNoiseReduction::process(KisPaintDeviceSP src, KisPaintDeviceSP dst, KisFilterConfiguration* config, const TQRect& rect)
|
|
{
|
|
|
|
float threshold = 1.0;
|
|
|
|
if(config !=0)
|
|
{
|
|
KisWaveletNoiseReductionConfiguration* configWNRC = (KisWaveletNoiseReductionConfiguration*)config;
|
|
kdDebug() << "threshold: " << configWNRC->threshold() << endl;
|
|
threshold = configWNRC->threshold();
|
|
}
|
|
|
|
|
|
TQ_INT32 depth = src->colorSpace()->nColorChannels();
|
|
|
|
int size;
|
|
int maxrectsize = (rect.height() < rect.width()) ? rect.width() : rect.height();
|
|
for(size = 2; size < maxrectsize; size *= 2) ;
|
|
|
|
KisMathToolbox* mathToolbox = KisMetaRegistry::instance()->mtRegistry()->get( src->colorSpace()->mathToolboxID() );
|
|
setProgressTotalSteps(mathToolbox->fastWaveletTotalSteps(rect) * 2 + size*size*depth );
|
|
connect(mathToolbox, TQT_SIGNAL(nextStep()), this, TQT_SLOT(incProgress()));
|
|
|
|
|
|
kdDebug(41005) << size << " " << maxrectsize << " " << rect.x() << " " << rect.y() << endl;
|
|
|
|
kdDebug(41005) << "Transforming..." << endl;
|
|
setProgressStage( i18n("Fast wavelet transformation") ,progress());
|
|
KisMathToolbox::KisWavelet* buff = 0;
|
|
KisMathToolbox::KisWavelet* wav = 0;
|
|
try {
|
|
buff = mathToolbox->initWavelet(src, rect);
|
|
} catch(std::bad_alloc)
|
|
{
|
|
if(buff) delete buff;
|
|
return;
|
|
}
|
|
try {
|
|
wav = mathToolbox->fastWaveletTransformation(src, rect, buff);
|
|
} catch(std::bad_alloc)
|
|
{
|
|
if(wav) delete wav;
|
|
return;
|
|
}
|
|
|
|
kdDebug(41005) << "Thresholding..." << endl;
|
|
float* fin = wav->coeffs + wav->depth*wav->size*wav->size;
|
|
setProgressStage( i18n("Thresholding") ,progress());
|
|
for(float* it = wav->coeffs + wav->depth; it < fin; it++)
|
|
{
|
|
if( *it > threshold)
|
|
{
|
|
*it -= threshold;
|
|
} else if( *it < -threshold ) {
|
|
*it += threshold;
|
|
} else {
|
|
*it = 0.;
|
|
}
|
|
incProgress();
|
|
}
|
|
|
|
kdDebug(41005) << "Untransforming..." << endl;
|
|
|
|
setProgressStage( i18n("Fast wavelet untransformation") ,progress());
|
|
mathToolbox->fastWaveletUntransformation( dst, rect, wav, buff);
|
|
|
|
delete wav;
|
|
delete buff;
|
|
disconnect(mathToolbox, TQT_SIGNAL(nextStep()), this, TQT_SLOT(incProgress()));
|
|
|
|
setProgressDone(); // Must be called even if you don't really support progression
|
|
}
|