Java Swing自定义BasicButtonUI如何获取按钮自定义属性？

问题背景 #

实现自定义NumberButton（继承JButton）时，希望将圆形绘制逻辑封装为通用的RoundButtonUI类，但最初通过构造方法传入颜色的方式无法动态获取按钮更新后的颜色，重新设置UI的方案不够优雅。

解决方案核心思路 #

在RoundButtonUI的update方法中，直接从传入的JComponent对象（即目标按钮实例）获取当前颜色属性，而非存储固定颜色值。利用Java的类型转换，将JComponent转为NumberButton后调用其getColor()方法，即可每次重绘时拿到最新的颜色。

修改后的代码实现 #

1. 优化RoundButtonUI类

public class RoundButtonUI extends BasicButtonUI {
    @Override
    public void update(Graphics g, JComponent c) {
        // 确保组件是NumberButton且不透明时绘制圆形背景
        if (c.isOpaque() && c instanceof NumberButton) {
            NumberButton targetBtn = (NumberButton) c;
            g.setColor(targetBtn.getColor());
            g.fillRoundRect(0, 0, c.getWidth(), c.getHeight(), 100, 100);
        }
        paint(g, c);
    }
}

2. 简化UI调用代码

无需传递颜色，直接实例化RoundButtonUI并设置：

NumberButton testbutton = new NumberButton("0");
testbutton.setUI(new RoundButtonUI());
mainPanel.add(testbutton);

3. 可选优化：颜色变更立即触发重绘

在NumberButton的setColor方法中添加repaint()，确保颜色修改后立刻更新界面：

public void setColor(Color color) {
    this.color = color;
    repaint(); // 触发组件重绘，即时显示新颜色
}

原理说明 #

每次Swing组件需要重绘时，update方法会被调用，此时直接从按钮实例获取当前的color属性，确保绘制的是最新颜色值。无需重新设置UI，即可实现颜色的动态更新，代码更简洁且符合Swing组件的设计规范。

内容的提问来源于stack exchange，提问作者afkfish