Change label text colors for more readable ones

Author: dwlsalmeida

src/label_widget.cpp

@@ -19,7 +19,7 @@ void LabelWidget::setNewLabel(const LabelInfo &label) {
 
 	setText(text);
 	setAlignment(Qt::AlignHCenter);
-	setStyleSheet("QLabel { background-color : " + label.color.name() + "; color : " + invColor(_label.color).name() + ";  font: bold 14px; }");
+	setStyleSheet("QLabel { background-color : " + label.color.name() + "; color : " + readableColor(_label.color).name() + ";  font:  14px; }");
 }
 
 void LabelWidget::setSelected(bool s) {
@@ -40,4 +40,8 @@ void LabelWidget::paintEvent(QPaintEvent *event) {
 		p.drawRect(rect());
 		p.end();
 	}
+}
+
+QString LabelWidget::getName(){
+	return this->_label.name;
 }

src/label_widget.h

@@ -18,6 +18,7 @@ class LabelWidget : public QLabel {
 
 	void setNewLabel(const LabelInfo &label);
 	void setSelected(bool s);
+	QString getName();
 
 public : // override
 	void paintEvent(QPaintEvent *event) override;

src/main_window.cpp

@@ -177,7 +177,7 @@ void MainWindow::changeLabel(QListWidgetItem* current, QListWidgetItem* previous
 	label->setSelected(true);
 
 	QString str;
-	QString key = label->text();
+	QString key = label->getName();
 	QTextStream sstr(&str);
 	sstr <<"label=["<< key <<"] id=[" << labels[key].id << "] categorie=[" << labels[key].categorie << "] color=[" << labels[key].color.name() << "]" ;
 	statusBar()->showMessage(str);

src/utils.cpp

@@ -47,6 +47,17 @@ void idToColor(const QImage &image_id, const Id2Labels& id_label, QImage *result
 	}
 }
 
+QColor readableColor(const QColor & color)
+{
+	int r, g, b;
+	color.getRgb(&r, &g, &b);
+
+	if ((r*0.299 + g*0.587 + b*0.114) > 150)
+		return QColor(0, 0, 0);
+
+	return QColor(255, 255, 255);
+
+}
 QColor invColor(const QColor& color) {
 	int h, s, v;
 	color.getHsv(&h, &s, &v);

src/utils.h

@@ -13,6 +13,7 @@ QImage idToColor(const QImage &image_id, const Id2Labels& id_label);
 void idToColor(const QImage &image_id, const Id2Labels& id_label, QImage *result);
 inline bool operator<(const QColor & a, const QColor & b) { return a.rgb() < b.rgb(); }
 QColor invColor(const QColor & color);
+QColor readableColor(const QColor & color);
 QVector<QColor> colorMap(int size);
 cv::Mat convertMat32StoRGBC3(const cv::Mat &mat);
 QImage watershed(const QImage& qimage, const QImage & qmarkers_mask);