File indexing completed on 2024-05-26 04:33:03

 /*
  * This file is part of Krita
  *
  * SPDX-FileCopyrightText: 2004 Cyrille Berger <cberger@cberger.net>
  *
  *  SPDX-License-Identifier: GPL-2.0-or-later
  */
 
 #include "colorsfilters.h"
 
 
 #include <math.h>
 
 #include <stdlib.h>
 #include <string.h>
 
 #include <QSlider>
 #include <QPoint>
 #include <QColor>
 
 #include <klocalizedstring.h>
 
 #include <kis_debug.h>
 #include <kpluginfactory.h>
 
 #include "KoBasicHistogramProducers.h"
 #include <KoColorSpace.h>
 #include <KoColorTransformation.h>
 #include <filter/kis_filter_category_ids.h>
 #include <filter/kis_filter_configuration.h>
 #include <kis_paint_device.h>
 #include <kis_processing_information.h>
 #include <KisDocument.h>
 #include <kis_image.h>
 #include <kis_layer.h>
 #include <kis_global.h>
 #include <kis_types.h>
 #include <kis_selection.h>
 #include <kis_histogram.h>
 #include <filter/kis_filter_registry.h>
 #include <kis_painter.h>
 #include <KoUpdater.h>
 #include <KoColorSpaceConstants.h>
 #include <KoCompositeOp.h>
 #include <KisSequentialIteratorProgress.h>
 
 
 #include "kis_hsv_adjustment_filter.h"
 #include "kis_perchannel_filter.h"
 #include "kis_cross_channel_filter.h"
 #include "kis_color_balance_filter.h"
 #include "kis_desaturate_filter.h"
 
 K_PLUGIN_FACTORY_WITH_JSON(ColorsFiltersFactory, "kritacolorsfilter.json", registerPlugin<ColorsFilters>();)
 
 ColorsFilters::ColorsFilters(QObject *parent, const QVariantList &)
         : QObject(parent)
 {
     KisFilterRegistry * manager = KisFilterRegistry::instance();
     manager->add(new KisAutoContrast());
     manager->add(new KisPerChannelFilter());
     manager->add(new KisCrossChannelFilter());
     manager->add(new KisDesaturateFilter());
     manager->add(new KisHSVAdjustmentFilter());
     manager->add(new KisColorBalanceFilter());
 
 }
 
 ColorsFilters::~ColorsFilters()
 {
 }
 
 
 //==================================================================
 
 
 KisAutoContrast::KisAutoContrast() : KisFilter(id(), FiltersCategoryAdjustId, i18n("&Auto Contrast"))
 {
     setSupportsPainting(false);
     setSupportsThreading(false);
     setSupportsAdjustmentLayers(false);
     setColorSpaceIndependence(TO_LAB16);
     setShowConfigurationWidget(false);
 }
 
 void KisAutoContrast::processImpl(KisPaintDeviceSP device,
                                   const QRect& applyRect,
                                   const KisFilterConfigurationSP config,
                                   KoUpdater* progressUpdater) const
 {
     Q_ASSERT(device != 0);
     Q_UNUSED(config);
     // initialize
     KoHistogramProducer *producer = new KoGenericLabHistogramProducer();
     KisHistogram histogram(device, applyRect, producer, LINEAR);
     int minvalue = int(255 * histogram.calculations().getMin() + 0.5);
     int maxvalue = int(255 * histogram.calculations().getMax() + 0.5);
 
     if (maxvalue > 255)
         maxvalue = 255;
 
     histogram.setChannel(0);
     int twoPercent = int(0.005 * histogram.calculations().getCount());
     int pixCount = 0;
     int binnum = 0;
 
     while (binnum < histogram.producer()->numberOfBins()) {
         pixCount += histogram.getValue(binnum);
         if (pixCount > twoPercent) {
             minvalue = binnum;
             break;
         }
         binnum++;
     }
     pixCount = 0;
     binnum = histogram.producer()->numberOfBins() - 1;
     while (binnum > 0) {
         pixCount += histogram.getValue(binnum);
         if (pixCount > twoPercent) {
             maxvalue = binnum;
             break;
         }
         binnum--;
     }
     // build the transferfunction
     int diff = maxvalue - minvalue;
 
     QScopedArrayPointer<quint16> transfer(new quint16[256]);
     for (int i = 0; i < 255; i++)
         transfer[i] = 0xFFFF;
 
     if (diff != 0) {
         for (int i = 0; i < minvalue; i++)
             transfer[i] = 0x0;
         for (int i = minvalue; i < maxvalue; i++) {
             qint32 val = int((0xFFFF * (i - minvalue)) / diff);
             if (val > 0xFFFF)
                 val = 0xFFFF;
             if (val < 0)
                 val = 0;
 
             transfer[i] = val;
         }
         for (int i = maxvalue; i < 256; i++)
             transfer[i] = 0xFFFF;
     }
     // apply
     QScopedPointer<KoColorTransformation> adj(device->colorSpace()->createBrightnessContrastAdjustment(transfer.data()));
     KIS_SAFE_ASSERT_RECOVER_RETURN(adj);
 
     KisSequentialIteratorProgress it(device, applyRect, progressUpdater);
 
     quint32 npix = it.nConseqPixels();
     while(it.nextPixels(npix)) {
 
         // adjust
         npix = it.nConseqPixels();
         adj->transform(it.oldRawData(), it.rawData(), npix);
     }
 }
 
 #include "colorsfilters.moc"