如何将图标设置为JButton而不丢失按下时的动画？

import javax.swing.*;
import javax.swing.plaf.basic.BasicButtonUI;
import java.awt.*;

public class CustomButtonUI extends BasicButtonUI {
    @Override
    protected void paintIcon(Graphics g, JComponent c, Rectangle iconRect) {
        AbstractButton button = (AbstractButton) c;
        ButtonModel model = button.getModel();
        Icon icon = button.getIcon();

        if (icon != null) {
            if (model.isPressed() && model.isArmed()) {
                // 绘制按下时的图标
                icon.paintIcon(c, g, iconRect.x + 1, iconRect.y + 1);
            } else {
                // 绘制正常状态下的图标
                icon.paintIcon(c, g, iconRect.x, iconRect.y);
            }
        }
    }

    @Override
    protected void paintButtonPressed(Graphics g, AbstractButton b) {
        // 绘制按下时的背景色
        g.setColor(b.getBackground().darker());
        g.fillRect(0, 0, b.getWidth(), b.getHeight());
    }
}

JButton button = new JButton("Button");
button.setIcon(new ImageIcon("icon.png"));
button.setUI(new CustomButtonUI());