Differential D241 Diff 691 krita/plugins/tools/selectiontools/kis_tool_select_magnetic.h

Changeset View

Standalone View

View Options

krita/plugins/tools/selectiontools/kis_tool_select_magnetic.h

  • This file was added.
1​/*
2​ * Copyright (c) 2004 Boudewijn Rempt (boud@valdyas.org)
3​ * Copyright (c) 2010 Adam Celarek <kdedev at xibo dot at>
4​ * Copyright (c) 2015 Michael Abrahams <miabraha@gmail.com>
5​ *
6​ * This program is free software; you can redistribute it and/or modify
7​ * it under the terms of the GNU General Public License as published by
8​ * the Free Software Foundation; either version 2 of the License, or
9​ * (at your option) any later version.
10​ *
11​ * This program is distributed in the hope that it will be useful,
12​ * but WITHOUT ANY WARRANTY; without even the implied warranty of
13​ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14​ * GNU General Public License for more details.
15​ *
16​ * You should have received a copy of the GNU General Public License
17​ * along with this program; if not, write to the Free Software
18​ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
19​ */
20​#ifndef KIS_TOOL_SELECT_MAGNETIC_H_
21​#define KIS_TOOL_SELECT_MAGNETIC_H_
22​
23​/**
24​ * Tool to select colors by pointing at a color on the image.
25​ */
26​
27​#include <QPainter>
28​#include <QPolygonF>
29​#include <QTransform>
30​#include <QDebug>
31​
32​#include <KoToolFactoryBase.h>
33​#include <KoIcon.h>
34​#include <KoCreatePathTool.h>
35​#include <kshortcut.h>
36​#include "flake/kis_node_shape.h"
37​#include "kis_tool_select_base.h"
38​#include "kis_delegated_tool.h"
39​#include "kis_tool_select_magnetic_option_widget.h"
40​namespace Ui {
41​ class KisToolSelectMagneticOptionWidget;
42​}
43​
44​class QWidget;
45​class KoCanvasBase;
46​class KisSelectionOptions;
47​class KisRandomConstAccessor;
48​class KoColorTransformation;
49​class KoColorSpace;
50​
51​#include <Eigen/Core>
52​using namespace Eigen;
53​
54​
55​#include <config-gsl.h>
56​#ifdef HAVE_GSL
57​#include <gsl/gsl_min.h>
58​#endif
59​
60​class KisToolSelectMagnetic;
61​
62​class KisSelectMagneticLocalTool : public KoCreatePathTool {
63​public:
64​ KisSelectMagneticLocalTool(KoCanvasBase * canvas, KisToolSelectMagnetic* parentTool);
65​ ~KisSelectMagneticLocalTool();
66​ virtual void paintPath(KoPathShape &path, QPainter &painter, const KoViewConverter &converter);
67​ virtual void addPathShape(KoPathShape* pathShape);
68​
69​ void activate(ToolActivation toolActivation, const QSet<KoShape*> &shapes);
70​ void deactivate();
71​ void mouseReleaseEvent(KoPointerEvent *event);
72​ void cancelPath();
73​ using KoCreatePathTool::createOptionWidgets;
74​ using KoCreatePathTool::endPathWithoutLastPoint;
75​ using KoCreatePathTool::endPath;
76​ QVector2D getEdgeGradient(QVector2D& currentPoint, KisRandomConstAccessorSP& pixelAccessor ) const;
77​
78​protected:
79​ friend class KisToolSelectMagnetic;
80​ void paintOutline(QPainter* painter, const QPainterPath &path, qreal zoom);
81​ void computeSections(const QPainterPath &pixelPath);
82​ void computeOutline(QPolygonF points, KisRandomConstAccessorSP pixelAccessor, bool fromLeft);
83​ void computeEdge (const QPolygonF &points, int i, const KisRandomConstAccessorSP &pixelAccessor);
84​ void optimizeEdge(const QPolygonF &points, int i, const KisRandomConstAccessorSP &pixelAccessor);
85​ Matrix3d getNearbyPixels(const QVector2D &point, KisRandomConstAccessorSP pixelAccessor) const;
86​
87​ /// this function removes unneeded points in m_detectedBorder, so that it is faster to paint
88​ void cleanDetectedBorder();
89​
90​private:
91​ KisToolSelectMagnetic* const m_selectionTool;
92​
93​ KoShapeStroke* m_borderBackup;
94​ QPolygonF m_outline;
95​ const KoColorSpace* m_colorSpace;
96​ KoColorTransformation* m_colorTransformation;
97​ int m_accuracy;
98​
99​
100​ //Data for minimization
101​ //TODO: move this stuff to a private class
102​ QPolygonF m_detectedBorder; // Our estimate of the border
103​ QList<qreal> m_detectedOffset; // Results from optimization, offset from center path to m_detectedBorder
104​ QPolygonF m_lastPoints; // Previously computed points from the polyline
105​ QPolygonF m_debugScannedPoints; // List of points we have scanned as edge candidates
106​ QList<bool> m_converged; // Remember whether we have converged
107​ void cleanEstimates(); // Zero out all the above vectors
108​ gsl_min_fminimizer *minimizer; // Memory address for GSL minimizer
109​};
110​
111​
112​
113​typedef KisDelegatedTool<KisTool, KisSelectMagneticLocalTool,
114​ DeselectShapesActivationPolicy> DelegatedSelectMagneticTool;
115​
116​
117​struct KisDelegatedSelectMagneticWrapper : public DelegatedSelectMagneticTool {
118​ KisDelegatedSelectMagneticWrapper(KoCanvasBase *canvas, const QCursor &cursor, KisTool* delegateTool)
119​ : DelegatedSelectMagneticTool(canvas, cursor, (KisSelectMagneticLocalTool*) delegateTool)
120​ {
121​ }
122​ bool listeningToModifiers() { return m_localTool->listeningToModifiers(); };
123​
124​ // If an event is explicitly forwarded only as an action (e.g. shift-click is captured by "change size")
125​ // we will receive a primary action but no mousePressEvent. Thus these events must be explicitly forwarded.
126​ void beginPrimaryAction(KoPointerEvent *event) { mousePressEvent(event); };
127​ void continuePrimaryAction(KoPointerEvent *event) { mouseMoveEvent(event); };
128​ void endPrimaryAction(KoPointerEvent *event) { mouseReleaseEvent(event); };
129​};
130​
131​
132​class KisToolSelectMagnetic : public SelectionActionHandler<KisDelegatedSelectMagneticWrapper>
133​{
134​ Q_OBJECT
135​public:
136​ KisToolSelectMagnetic(KoCanvasBase * canvas);
137​ void mousePressEvent(KoPointerEvent* event);
138​
139​ int radius() const;
140​ int threshold() const;
141​ bool searchFromLeft() const;
142​ int colorLimitation() const;
143​ bool limitToCurrentLayer() const;
144​
145​protected:
146​ void requestStrokeCancellation();
147​ void requestStrokeEnd();
148​ void setAlternateSelectionAction(SelectionAction action);
149​ friend class KisSelectMagneticLocalTool;
150​private:
151​ virtual QWidget* createOptionWidget();
152​ Ui::KisToolSelectMagneticOptionWidget *m_magneticOptions;
153​};
154​
155​
156​
157​class KisToolSelectMagneticFactory : public KoToolFactoryBase
158​{
159​public:
160​ KisToolSelectMagneticFactory(const QStringList&)
161​ : KoToolFactoryBase("KisToolSelectMagnetic") {
162​ setToolTip(i18n("Magnetic Selection Tool"));
163​ setToolType(TOOL_TYPE_SELECTED);
164​ setActivationShapeId(KRITA_TOOL_ACTIVATION_ID);
165​ setIconName(koIconNameCStr("tool_magnetic_selection"));
166​ setShortcut(KShortcut(Qt::CTRL + Qt::Key_E));
dkazakovUnsubmitted
Not Done

Here instead of koIconNameCStr("tool_magnetic_selection"), please use KisIconUtils::loadIcon("<icon_name>"). See [0] for more info.

If will also allow you to avoid duplicating icons in the selectiontools/ directly and int pics/tools/16. I guess only one copy of the icons should be kept :)

[0] - https://community.kde.org/Krita/docs/Icons_API_Roadmap

dkazakov: Here instead of koIconNameCStr("tool_magnetic_selection"), please use KisIconUtils::loadIcon…
167​ qDebug("MagneticSelection: Creating Magnetic Selection Tool Factory");
168​ setPriority(80);
169​ }
170​
171​ virtual ~KisToolSelectMagneticFactory() {}
172​
173​ virtual KoToolBase * createTool(KoCanvasBase *canvas) {
174​ return new KisToolSelectMagnetic(canvas);
175​ }
176​
177​};
178​
179​
180​#endif // KIS_TOOL_SELECT_MAGNETIC_H_