PixelAnnotationTool/src/image_mask.cpp at c25a25af08afb81b38b3a200de01fbb16720b283 · abreheret/PixelAnnotationTool · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
#include "image_mask.h"
#include "utils.h"

#include <QPainter>

ImageMask::ImageMask() {}
ImageMask::ImageMask(const QString &file, Id2Labels id_labels) {
	id = mat2QImage(cv::imread(file.toStdString()));
	color = idToColor(id, id_labels);
}
ImageMask::ImageMask(QSize s) {
	id = QImage(s, QImage::Format_RGB888);
	color = QImage(s, QImage::Format_RGB888);
	id.fill(QColor(0, 0, 0));
	color.fill(QColor(0, 0, 0));
}

void ImageMask::drawLine(QPoint p1, QPoint p2, int pen_size, ColorMask cm) {
	QPainter painter_id(&id);
    painter_id.setRenderHint(QPainter::Antialiasing, false);
    painter_id.setPen(QPen(cm.id, pen_size, Qt::SolidLine, Qt::RoundCap,
                        Qt::RoundJoin));
    painter_id.drawLine(p1, p2);
	painter_id.end();

    QPainter painter_color(&color);
    painter_color.setRenderHint(QPainter::Antialiasing, false);
    painter_color.setPen(QPen(cm.color, pen_size, Qt::SolidLine, Qt::RoundCap,
                        Qt::RoundJoin));
    painter_color.drawLine(p1, p2);
    painter_color.end();
}
void ImageMask::drawFillCircle(QPoint point, int pen_size, ColorMask cm) {
    if(pen_size<=1) {
        id.setPixelColor(point, cm.id);
        color.setPixelColor(point, cm.color);
    }
    else {
        pen_size = (pen_size-1)/2;
        QPen pen(QBrush(cm.id), 1.0);
        QPainter painter_id(&id);
        painter_id.setRenderHint(QPainter::Antialiasing, false);
        painter_id.setPen(pen);
        painter_id.setBrush(QBrush(cm.id));
        painter_id.drawEllipse(point, pen_size, pen_size);
        painter_id.end();

        QPainter painter_color(&color);
        QPen pen_color(QBrush(cm.color), 1.0);
        painter_color.setRenderHint(QPainter::Antialiasing, false);
        painter_color.setPen(pen_color);
        painter_color.setBrush(QBrush(cm.color));
        painter_color.drawEllipse(point, pen_size, pen_size);
        painter_color.end();
    }
}
void ImageMask::updateColor(const Id2Labels & labels) {
	idToColor(id, labels, &color);
}

void ImageMask::exchangeLabel(int x, int y, const Id2Labels& id_labels, ColorMask cm) {

	QColor current_id = id.pixelColor(QPoint(x, y));
	if (current_id.red() == 0 || current_id.green() == 0 || current_id.blue() == 0)
		return;

	cv::Mat id_mat = qImage2Mat(id);
	cv::floodFill(id_mat, cv::Point(x, y), cv::Scalar(cm.id.red(), cm.id.green(), cm.id.blue()), 0, cv::Scalar(0, 0, 0), cv::Scalar(0, 0, 0));

	id = mat2QImage(id_mat);
	color = idToColor(id, id_labels);

}