Yolo_Label/label_img.cpp
2018-10-29 13:26:06 +09:00

269 lines
7.3 KiB
C++

#include "label_img.h"
#include <QPainter>
using std::ifstream;
QColor label_img::BOX_COLORS[10] ={ Qt::green,
Qt::darkGreen,
Qt::blue,
Qt::darkBlue,
Qt::yellow,
Qt::darkYellow,
Qt::red,
Qt::darkRed,
Qt::cyan,
Qt::darkCyan};
label_img::label_img(QWidget *parent)
:QLabel(parent),
m_bMouseLeftButtonClicked(false),
m_focusedObjectLabel(0)
{
}
void label_img::mouseMoveEvent(QMouseEvent *ev)
{
std::cout<< "moved"<< std::endl;
setMousePosition(ev->x(), ev->y());
showImage();
emit Mouse_Moved();
}
void label_img::mousePressEvent(QMouseEvent *ev)
{
std::cout<< "clicked"<< std::endl;
setMousePosition(ev->x(), ev->y());
if(ev->button() == Qt::RightButton)
{
removeFocusedObjectBox(m_mouse_pos_in_image_coordinate);
showImage();
}
else if(ev->button() == Qt::LeftButton)
{
m_bMouseLeftButtonClicked = true;
m_objStartPoint = m_mouse_pos_in_image_coordinate;
}
emit Mouse_Pressed();
}
void label_img::mouseReleaseEvent(QMouseEvent *ev)
{
std::cout<< "released"<< std::endl;
if(ev->button() == Qt::LeftButton)
{
m_bMouseLeftButtonClicked = false;
setMousePosition(ev->x(), ev->y());
m_objEndPoint = m_mouse_pos_in_image_coordinate;
ObjectLabelingBox objBoundingbox;
objBoundingbox.label = m_focusedObjectLabel;
objBoundingbox.box = getRectFromTwoPoints(m_objEndPoint, m_objStartPoint);
bool width_is_too_small = objBoundingbox.box.width() < m_inputImg.width() * 0.01;
bool height_is_too_small = objBoundingbox.box.height() < m_inputImg.height() * 0.01;
if(!width_is_too_small && !height_is_too_small)
{
m_objBoundingBoxes.push_back(objBoundingbox);
}
showImage();
}
emit Mouse_Release();
}
void label_img::setMousePosition(int x, int y)
{
if(x < 0) x = 0;
if(y < 0) y = 0;
if(x > this->width()) x = this->width() - 1;
if(y > this->height()) y = this->height() - 1;
m_mouse_pos_in_ui_coordinate = QPoint(x, y);
m_mouse_pos_in_image_coordinate = QPoint(static_cast<int>(m_mouse_pos_in_ui_coordinate.x() * m_aspectRatioWidth),
static_cast<int>(m_mouse_pos_in_ui_coordinate.y() * m_aspectRatioHeight));
}
void label_img::openImage(const QString &qstrImg)
{
m_inputImg = QImage(qstrImg);
m_inputImgCopy = m_inputImg;
m_aspectRatioWidth = static_cast<double>(m_inputImg.width()) / this->width();
m_aspectRatioHeight = static_cast<double>(m_inputImg.height()) / this->height();
m_objBoundingBoxes.clear();
m_objStartPoint = QPoint();
QPoint curMousePos = this->mapFromGlobal(QCursor::pos());
bool mouse_is_not_in_image = (curMousePos.x() < 0 || curMousePos.x() > this->width() - 1)
|| (curMousePos.y() < 0 || curMousePos.y() > this->height() - 1);
if (mouse_is_not_in_image)
{
m_objEndPoint = QPoint();
}
else
{
setMousePosition(curMousePos.x(), curMousePos.y());
m_objEndPoint = m_mouse_pos_in_image_coordinate;
}
}
void label_img::showImage()
{
if(m_inputImg.isNull()) return;
m_inputImg = m_inputImgCopy;
QPainter painter(&m_inputImg);
int penThick = (m_inputImg.width() + m_inputImg.height())/400;
QColor crossLineColor(255, 187, 0);
drawCrossLine(painter, crossLineColor, penThick);
drawFocusedObjectBox(painter, Qt::magenta, penThick);
drawObjectBoxes(painter, penThick);
this->setPixmap(QPixmap::fromImage(m_inputImg));
}
void label_img::loadLabelData(const QString& labelFilePath)
{
ifstream inputFile(labelFilePath.toStdString());
if(inputFile.is_open()){
double inputFileValue;
QVector<double> inputFileValues;
while(inputFile >> inputFileValue)
inputFileValues.push_back(inputFileValue);
for(int i = 0; i < inputFileValues.size(); i += 5)
{
try {
ObjectLabelingBox objBox;
objBox.label = static_cast<int>(inputFileValues[i]);
double midX = inputFileValues.at(i + 1) * m_inputImg.width();
double midY = inputFileValues.at(i + 2) * m_inputImg.height();
double width = inputFileValues.at(i + 3) * m_inputImg.width();
double height = inputFileValues.at(i + 4) * m_inputImg.height();
objBox.box.setX(static_cast<int>(midX - width/2));
objBox.box.setY(static_cast<int>(midY - height/2));
objBox.box.setWidth(static_cast<int>(width));
objBox.box.setHeight(static_cast<int>(height));
m_objBoundingBoxes.push_back(objBox);
}
catch (const std::out_of_range& e) {
std::cout << "loadLabelData: Out of Range error.";
}
}
}
}
void label_img::setFocusObjectLabel(int nLabel)
{
m_focusedObjectLabel = nLabel;
}
void label_img::setFocusObjectName(QString qstrName)
{
m_foucsedObjectName = qstrName;
}
void label_img::drawCrossLine(QPainter& painter, QColor color, int thickWidth)
{
if(m_objEndPoint == QPoint()) return;
QPen pen;
pen.setWidth(thickWidth);
pen.setColor(color);
painter.setPen(pen);
//draw cross line
painter.drawLine(QPoint(m_mouse_pos_in_image_coordinate.x(),0), QPoint(m_mouse_pos_in_image_coordinate.x(),m_inputImg.height() - 1));
painter.drawLine(QPoint(0,m_mouse_pos_in_image_coordinate.y()), QPoint(m_inputImg.width() - 1, m_mouse_pos_in_image_coordinate.y()));
}
void label_img::drawFocusedObjectBox(QPainter& painter, Qt::GlobalColor color, int thickWidth)
{
if(m_bMouseLeftButtonClicked)
{
QPen pen;
pen.setWidth(thickWidth);
pen.setColor(color);
painter.setPen(pen);
painter.drawRect(QRect(m_objStartPoint, m_mouse_pos_in_image_coordinate));
}
}
void label_img::drawObjectBoxes(QPainter& painter, int thickWidth)
{
QPen pen;
pen.setWidth(thickWidth);
for(ObjectLabelingBox boundingbox: m_objBoundingBoxes)
{
pen.setColor(m_drawObjectBoxColor.at(boundingbox.label));
painter.setPen(pen);
painter.drawRect(boundingbox.box);
}
}
void label_img::removeFocusedObjectBox(QPoint point)
{
int removeBoxIdx = -1;
double nearestBoxDistance = 99999999999999.;
for(int i = 0; i < m_objBoundingBoxes.size(); i++)
{
QRect objBox = m_objBoundingBoxes.at(i).box;
if(objBox.contains(point))
{
double distance = sqrt(pow(objBox.center().x() - point.x(), 2)+ pow(objBox.center().y() - point.y(), 2));
if(distance < nearestBoxDistance)
{
nearestBoxDistance = distance;
removeBoxIdx = i;
}
}
}
if(removeBoxIdx != -1)
{
m_objBoundingBoxes.remove(removeBoxIdx);
}
}
QRect label_img::getRectFromTwoPoints(QPoint p1, QPoint p2)
{
int midX = (p1.x() + p2.x()) / 2;
int midY = (p1.y() + p2.y()) / 2;
int width = abs(p1.x() - p2.x());
int height = abs(p1.y() - p2.y());
QPoint topLeftPoint(midX - width/2, midY - height/2);
QPoint bottomRightPoint(midX + width/2, midY + height/2);
return QRect(topLeftPoint, bottomRightPoint);
}