From 946febaf8d16a7d87021ff53416a7c1cb2cae497 Mon Sep 17 00:00:00 2001
From: Linloir <3145078758@qq.com>
Date: Mon, 19 Dec 2022 17:06:21 +0800
Subject: [PATCH] [UI][FIX] Push button 'setEnabled' func - Add API
 'setEnabled' - Fix 'setEnabled' func

---
 FinalProject/pushbutton.cpp | 55 +++++++++++++++++++++++++++++++++++++
 FinalProject/pushbutton.h   |  3 ++
 2 files changed, 58 insertions(+)

diff --git a/FinalProject/pushbutton.cpp b/FinalProject/pushbutton.cpp
index 0470b86..5ed448d 100644
--- a/FinalProject/pushbutton.cpp
+++ b/FinalProject/pushbutton.cpp
@@ -84,6 +84,10 @@ void PushButton::generateColor(QColor colorScheme) {
 }
 
 void PushButton::enterEvent(QEnterEvent* event) {
+    if (!_enabled) {
+        return;
+    }
+
     setCursor(Qt::PointingHandCursor);
 
     _backgroundWidget->setStyleSheet("QWidget#backgroundWidget{background-color:" + _hoverColor.name(QColor::HexArgb) + ";border-radius:" + QString::number(_radius) + "px;}");
@@ -141,6 +145,10 @@ void PushButton::enterEvent(QEnterEvent* event) {
 }
 
 void PushButton::leaveEvent(QEvent* event) {
+    if (!_enabled) {
+        return;
+    }
+
     setCursor(Qt::ArrowCursor);
 
     if (_selected) {
@@ -203,6 +211,10 @@ void PushButton::leaveEvent(QEvent* event) {
 }
 
 void PushButton::mousePressEvent(QMouseEvent* event) {
+    if (!_enabled) {
+        return;
+    }
+
     _backgroundWidget->setStyleSheet("QWidget#backgroundWidget{background-color:" + _pressedColor.name(QColor::HexArgb) + ";border-radius:" + QString::number(_radius) + "px;}");
     
     QPropertyAnimation* indicatorShrinkLength = new QPropertyAnimation(_indicator, "geometry", this);
@@ -601,6 +613,49 @@ void PushButton::deselect() {
     _selected = false;
 }
 
+void PushButton::setEnabled(bool enabled) {
+    if (enabled == _enabled) {
+        return;
+    }
+
+    if (enabled) {
+        _enabled = true;
+        // Restore colors
+        _backgroundColor = _restoredColor[0];
+        _hoverColor = _restoredColor[1];
+        _pressedColor = _restoredColor[2];
+        _selectedColor = _restoredColor[3];
+        _indicatorColor = _restoredColor[4];
+        if (_pressed) {
+            _backgroundWidget->setStyleSheet("QWidget#backgroundWidget{background-color:" + _pressedColor.name(QColor::HexArgb) + ";border-radius:" + QString::number(_radius) + "px;}");
+        }
+        else if (_hovered) {
+            _backgroundWidget->setStyleSheet("QWidget#backgroundWidget{background-color:" + _hoverColor.name(QColor::HexArgb) + ";border-radius:" + QString::number(_radius) + "px;}");
+        }
+        else if (_selected) {
+            _backgroundWidget->setStyleSheet("QWidget#backgroundWidget{background-color:" + _selectedColor.name(QColor::HexArgb) + ";border-radius:" + QString::number(_radius) + "px;}");
+        }
+        else {
+            _backgroundWidget->setStyleSheet("QWidget#backgroundWidget{background-color:" + _backgroundColor.name(QColor::HexArgb) + ";border-radius:" + QString::number(_radius) + "px;}");
+        }
+        _indicator->setStyleSheet("QWidget#indicator{background-color:" + _indicatorColor.name(QColor::HexArgb) + ";"
+            "border-radius:" + QString::number((float)_indicatorWidth / 2) + "px;}");
+    }
+    else {
+        _enabled = false;
+        _pressed = false;
+        _hovered = false;
+        // Store color
+        _restoredColor[0] = _backgroundColor;
+        _restoredColor[1] = _hoverColor;
+        _restoredColor[2] = _pressedColor;
+        _restoredColor[3] = _selectedColor;
+        _restoredColor[4] = _indicatorColor;
+        // Set disabled colors
+        setColorScheme(QColor(200, 200, 200));
+    }
+}
+
 void PushButton::setRadius(int radius) {
     // get current style sheet
     QString styleSheet = _backgroundWidget->styleSheet();
diff --git a/FinalProject/pushbutton.h b/FinalProject/pushbutton.h
index 8ac2f6d..ab96288 100644
--- a/FinalProject/pushbutton.h
+++ b/FinalProject/pushbutton.h
@@ -39,6 +39,7 @@ private:
     QColor _hoverColor;
     QColor _pressedColor;
     QColor _selectedColor;
+    QColor _restoredColor[5];
     
     QWidget* _indicator;
     LUI_BTN_INDICATOR_POS _indicatorPosition = LUI_BTN_POS_LEFT;
@@ -51,6 +52,7 @@ private:
     QGraphicsOpacityEffect* _indicatorEffect;
 
     // Button state
+    bool _enabled = true;
     bool _hovered = false;
     bool _pressed = false;
     bool _selected = false;
@@ -72,6 +74,7 @@ public:
     // Operation APIs
     void select();
     void deselect();
+    void setEnabled(bool enabled = true);
     
     // Attribute setter APIs
     void setRadius(int radius);