imagetransparentcolordialog.h

Go to the documentation of this file.

/*
        Copyright 2006-2026 The QElectroTech Team
        This file is part of QElectroTech.
 
        QElectroTech 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.
 
        QElectroTech 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 QElectroTech.  If not, see <http://www.gnu.org/licenses/>.
*/
#ifndef IMAGE_TRANSPARENT_COLOR_DIALOG_H
#define IMAGE_TRANSPARENT_COLOR_DIALOG_H
 
#include <QDialog>
#include <QImage>
#include <QColor>
#include <QLabel>
#include <QList>
 
class QVBoxLayout;
class QPushButton;
class QMouseEvent;

class ClickableImageLabel : public QLabel
{
        Q_OBJECT
 
        public:
                explicit ClickableImageLabel(const QImage &sourceImage, QWidget *parent = nullptr);

                QImage displayImage() const { return m_displayImage; }
 
        signals:
                void colorPicked(const QColor &color);
 
        protected:
                void mousePressEvent(QMouseEvent *event) override;
 
        private:
                QImage m_source;
                QImage m_displayImage;
                qreal m_displayScale = 1.0;
};

class ImageTransparentColorDialog : public QDialog
{
        Q_OBJECT
 
        public:
                struct PickedColor
                {
                        QColor color;
                        int tolerance;
                };

                explicit ImageTransparentColorDialog(const QPixmap &basePixmap,
                                const QList<PickedColor> &existingColors = {}, QWidget *parent = nullptr);

                QPixmap resultPixmap() const;
                QList<PickedColor> pickedColors() const { return m_pickedColors; }

                static QImage applyColorKey(const QImage &source, const QList<PickedColor> &keyColors);
 
        private slots:
                void onColorPicked(const QColor &color);
 
        private:
                void setToleranceForIndex(int index, int value);
                void removeColor(int index);
                void rebuildSwatches();
                void updatePreview();
                void updateOkEnabled();
                static QPixmap onCheckerboard(const QImage &image);
 
                QImage              m_sourceImage;
                QImage              m_previewSourceImage;   // downsampled -- see ClickableImageLabel::displayImage()'s comment for why
                QList<PickedColor>  m_pickedColors;
                bool                m_startedWithColors = false;   // whether existingColors was non-empty -- see updateOkEnabled()'s comment for why this matters
                int                 m_lastToleranceUsed = 10;   // seeds a newly-picked colour's own tolerance, so successive picks in one session feel consistent rather than each resetting to some fixed default
 
                ClickableImageLabel *m_sourceLabel;
                QLabel              *m_previewLabel;
                QVBoxLayout         *m_swatchesLayout;   // one row per picked colour, stacked vertically -- each row now carries its own slider, too wide to lay out side by side the way plain swatches once were
                QLabel              *m_hintLabel;
                QPushButton         *m_okButton;
};
 
#endif // IMAGE_TRANSPARENT_COLOR_DIALOG_H