code cleanup

ui/colorbutton.cpp

@@ -0,0 +1,48 @@
+#include "colorbutton.h"
+
+#include "project/undo.h"
+
+#include <QColorDialog>
+
+ColorButton::ColorButton(QWidget *parent)
+    : QPushButton(parent), color(Qt::white) {
+    set_button_color();
+    connect(this, SIGNAL(clicked(bool)), this, SLOT(open_dialog()));
+}
+
+QColor ColorButton::get_color() {
+    return color;
+}
+
+void ColorButton::set_color(QColor c) {
+    color = c;
+}
+
+void ColorButton::set_button_color() {
+    QPalette pal = palette();
+    pal.setColor(QPalette::Button, color);
+    setPalette(pal);
+}
+
+void ColorButton::open_dialog() {
+    QColor old_color = color;
+    QColor new_color = QColorDialog::getColor(color, NULL);
+    if (old_color != new_color) {
+        ColorCommand* command = new ColorCommand(this, old_color, new_color);
+        undo_stack.push(command);
+
+        set_button_color();
+        emit color_changed();
+    }
+}
+
+ColorCommand::ColorCommand(ColorButton* s, QColor o, QColor n)
+    : sender(s), old_color(o), new_color(n) {}
+
+void ColorCommand::undo() {
+    sender->set_color(old_color);
+}
+
+void ColorCommand::redo() {
+    sender->set_color(new_color);
+}