Backport recent changes from mainline.

[pulseview.git] / pv / widgets / colorbutton.cpp

diff --git a/pv/widgets/colorbutton.cpp b/pv/widgets/colorbutton.cpp
new file mode 100644 (file)
index 0000000..702fdde
--- /dev/null
+++ b/pv/widgets/colorbutton.cpp
@@ -0,0 +1,134 @@
+/*
+ * This file is part of the PulseView project.
+ *
+ * Copyright (C) 2013 Joel Holdsworth <joel@airwebreathe.org.uk>
+ *
+ * This program 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 2 of the License, or
+ * (at your option) any later version.
+ *
+ * This program 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 this program; if not, see <http://www.gnu.org/licenses/>.
+ */
+
+#include "colorbutton.hpp"
+
+#include <cassert>
+
+#include <QApplication>
+#include <QColorDialog>
+#include <QPainter>
+
+namespace pv {
+namespace widgets {
+
+const int ColorButton::SwatchMargin = 7;
+
+ColorButton::ColorButton(QWidget *parent) :
+       QPushButton("", parent),
+       popup_(nullptr)
+{
+       connect(this, SIGNAL(clicked(bool)), this, SLOT(on_clicked(bool)));
+}
+
+ColorButton::ColorButton(int rows, int cols, QWidget *parent) :
+       QPushButton("", parent),
+       popup_(new ColorPopup(rows, cols, this))
+{
+       connect(this, SIGNAL(clicked(bool)), this, SLOT(on_clicked(bool)));
+       connect(popup_, SIGNAL(selected(int, int)),
+               this, SLOT(on_selected(int, int)));
+}
+
+ColorPopup* ColorButton::popup()
+{
+       return popup_;
+}
+
+const QColor& ColorButton::color() const
+{
+       return cur_color_;
+}
+
+void ColorButton::set_color(QColor color)
+{
+       cur_color_ = color;
+
+       if (popup_) {
+               const unsigned int rows = popup_->well_array().numRows();
+               const unsigned int cols = popup_->well_array().numCols();
+
+               for (unsigned int r = 0; r < rows; r++)
+                       for (unsigned int c = 0; c < cols; c++)
+                               if (popup_->well_array().cellBrush(r, c).color() == color) {
+                                       popup_->well_array().setSelected(r, c);
+                                       popup_->well_array().setCurrent(r, c);
+                                       return;
+                               }
+       }
+}
+
+void ColorButton::set_palette(const QColor *const palette)
+{
+       assert(palette);
+       assert(popup_);
+
+       const unsigned int rows = popup_->well_array().numRows();
+       const unsigned int cols = popup_->well_array().numCols();
+
+       for (unsigned int r = 0; r < rows; r++)
+               for (unsigned int c = 0; c < cols; c++)
+                       popup_->well_array().setCellBrush(r, c, QBrush(palette[r * cols + c]));
+}
+
+void ColorButton::on_clicked(bool)
+{
+       if (popup_) {
+               popup_->set_position(mapToGlobal(rect().center()), Popup::Bottom);
+               popup_->show();
+       } else {
+               QColorDialog dlg(this);
+               dlg.setOption(QColorDialog::ShowAlphaChannel);
+               dlg.setOption(QColorDialog::DontUseNativeDialog);
+               connect(&dlg, SIGNAL(colorSelected(const QColor)),
+                       this, SLOT(on_color_selected(const QColor)));
+               dlg.setCurrentColor(cur_color_);
+               dlg.exec();
+       }
+}
+
+void ColorButton::on_selected(int row, int col)
+{
+       assert(popup_);
+
+       cur_color_ = popup_->well_array().cellBrush(row, col).color();
+       selected(cur_color_);
+}
+
+void ColorButton::on_color_selected(const QColor& color)
+{
+       cur_color_ = color;
+       selected(cur_color_);
+}
+
+void ColorButton::paintEvent(QPaintEvent *event)
+{
+       QPushButton::paintEvent(event);
+
+       QPainter p(this);
+
+       const QRect r = rect().adjusted(SwatchMargin, SwatchMargin,
+               -SwatchMargin, -SwatchMargin);
+       p.setPen(QApplication::palette().color(QPalette::Dark));
+       p.setBrush(QBrush(cur_color_));
+       p.drawRect(r);
+}
+
+}  // namespace widgets
+}  // namespace pv