flameshot/src/capture/tools/linetool.cpp

// Copyright 2017 Alejandro Sirgo Rica
//
// This file is part of Flameshot.
//
//     Flameshot 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 3 of the License, or
//     (at your option) any later version.
//
//     Flameshot 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 Flameshot.  If not, see <http://www.gnu.org/licenses/>.

#include "linetool.h"
#include <QPainter>

#define ADJ_VALUE 13

LineTool::LineTool(QObject *parent) : CaptureTool(parent) {

}

int LineTool::id() const {
    return 0;
}

bool LineTool::isSelectable() const {
    return true;
}

QString LineTool::iconName() const {
    return "line.png";
}

QString LineTool::name() const {
    return tr("Line");
}

QString LineTool::description() const {
    return tr("Sets the Line as the paint tool");
}

CaptureTool::ToolWorkType LineTool::toolType() const {
    return TYPE_LINE_DRAWER;
}

void LineTool::processImage(
        QPainter &painter,
        const QVector<QPoint> &points,
        const QColor &color,
        const int thickness)
{
    QPoint p0 = points[0];
    QPoint p1 = points[1];
    if (needsAdjustment(p0, p1)) {
        p1.setY(p0.y());
    }
    painter.setPen(QPen(color, 2 + thickness));
    painter.drawLine(p0, p1);
}

void LineTool::onPressed() {
}

bool LineTool::needsAdjustment(const QPoint &p0, const QPoint &p1) const {
    return (p1.y() >= p0.y() - ADJ_VALUE) && (p1.y() <= p0.y() + ADJ_VALUE);
}