plugin-event-manager/src/main/java/com/fr/plugin/event/manager/ui/CellWidgetPane.java

package com.fr.plugin.event.manager.ui;

import com.fr.base.GraphHelper;
import com.fr.design.constants.UIConstants;
import com.fr.design.dialog.BasicPane;
import com.fr.design.gui.ibutton.UIButton;
import com.fr.design.gui.ibutton.UIButtonUI;
import com.fr.design.gui.ibutton.UIToggleButton;
import com.fr.design.layout.FRGUIPaneFactory;
import com.fr.design.utils.gui.GUIPaintUtils;
import com.fr.general.IOUtils;
import com.fr.plugin.event.manager.data.MyTree;

import javax.swing.BorderFactory;
import javax.swing.Icon;
import javax.swing.JPanel;
import java.awt.BorderLayout;

import java.awt.Dimension;
import java.awt.FlowLayout;
import java.awt.Graphics2D;
import java.awt.event.ActionEvent;
import java.awt.event.ActionListener;
import java.awt.event.MouseAdapter;
import java.awt.event.MouseEvent;
import java.awt.event.MouseListener;
import java.util.ArrayList;
import java.util.List;

/**
 * @author Joe
 * Created by Joe on 12/13/2020
 */
public class CellWidgetPane extends BasicPane {
    private static final Icon LEFT_ICON = IOUtils.readIcon("com/fr/design/images/sheet/left_normal@1x.png");
    private static final Icon RIGHT_ICON = IOUtils.readIcon("com/fr/design/images/sheet/right_normal@1x.png");
    private static final Icon DISABLED_LEFT_ICON = IOUtils.readIcon("com/fr/design/images/sheet/left_hover@1x.png");
    private static final Icon DISABLED_RIGHT_ICON = IOUtils.readIcon("com/fr/design/images/sheet/right_hover@1x.png");

    private static final String ELLIPSIS = "...";
    private static final int MAX_WIDTH = 810;
    private static final int PREFERRED_HEIGHT = 24;
    private static final int FLOW_GAP = 5;
    private static final int BUTTON_EXTRA_WIDTH = 20;

    private List<MyTree> cellWidgetTrees;
    private EventConfigPane eventConfigPane;
    private SheetChoosePane sheetChoosePane;

    public CellWidgetPane(List<MyTree> cellWidgetTrees) {
        this.cellWidgetTrees = cellWidgetTrees;
        initComponentPane();
    }

    private void initComponentPane() {
        setLayout(new BorderLayout(4, 0));
        eventConfigPane = new EventConfigPane(new MyTree());
        sheetChoosePane = new SheetChoosePane();
        populate(cellWidgetTrees.size() > 0 ? cellWidgetTrees.get(0) : new MyTree());
    }

    public String tabTitle() {
        return "Cell Widget";
    }

    @Override
    protected String title4PopupWindow() {
        return "cell widget";
    }

    public void populate(MyTree myTree) {
        removeAll();
        eventConfigPane = new EventConfigPane(myTree);
        add(eventConfigPane, BorderLayout.CENTER);
        add(sheetChoosePane, BorderLayout.NORTH);
        repaint();
        revalidate();
    }

    public void update() {
        eventConfigPane.update();
    }

    private class SheetChoosePane extends JPanel {
        // 总sheet按钮数
        private int totalCount;

        // 展示的sheet按钮数
        private int showCount = 0;

        // sheet宽度数组
        private int[] widthArray;

        // 选中的sheet的index
        private int selectedIndex = 0;

        // 最左边的sheet的index
        private int scrollIndex = 0;

        // button数组
        private List<UIToggleButton> buttonList;

        // 左边的panel
        private JPanel sheetFlowPane;

        /**
         * 左移和右移按钮
         */
        private UIButton leftButton;
        private UIButton rightButton;

        public SheetChoosePane() {
            totalCount = cellWidgetTrees.size();
            buttonList = new ArrayList<>();
            initComponentPane();
        }

        private void initComponentPane() {
            setLayout(FRGUIPaneFactory.createBorderLayout());
            setBorder(BorderFactory.createEmptyBorder(4, 8, 0, 8));
            // 初始化右边按钮
            add(createButtonPane(), BorderLayout.EAST);
            // 初始化左边面板
            initSheetFlowPane();
            add(sheetFlowPane, BorderLayout.CENTER);
        }

        private void initSheetFlowPane() {
            sheetFlowPane = new JPanel(new FlowLayout(FlowLayout.LEFT));
            initWidthAndButtonArray();
            addButton();
            checkButton(showCount < widthArray.length);
        }

        private JPanel createButtonPane() {
            leftButton = new UIButton(LEFT_ICON) {
                @Override
                public Dimension getPreferredSize() {
                    return new Dimension(super.getPreferredSize().width, PREFERRED_HEIGHT);
                }
            };
            leftButton.setUI(new UIButtonUI() {
                @Override
                protected void doExtraPainting(UIButton b, Graphics2D g2d, int w, int h, String selectedRoles) {
                    if (isPressed(b) && b.isPressedPainted()) {
                        GUIPaintUtils.fillPressed(g2d, 0, 0, w, h, b.isRoundBorder(), b.getRectDirection(), b.isDoneAuthorityEdited(selectedRoles), UIConstants.PROPERTY_PANE_BACKGROUND);
                    } else if (isRollOver(b)) {
                        GUIPaintUtils.fillRollOver(g2d, 0, 0, w, h, b.isRoundBorder(), b.getRectDirection(), b.isDoneAuthorityEdited(selectedRoles), b.isPressedPainted(), UIConstants.PROPERTY_PANE_BACKGROUND);
                    } else if (b.isNormalPainted()) {
                        GUIPaintUtils.fillNormal(g2d, 0, 0, w, h, b.isRoundBorder(), b.getRectDirection(), b.isDoneAuthorityEdited(selectedRoles), b.isPressedPainted());
                    }
                }
            });
            leftButton.set4ToolbarButton();
            leftButton.setDisabledIcon(DISABLED_LEFT_ICON);
            leftButton.addActionListener(new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent e) {
                    moveLeft();
                }
            });
            rightButton = new UIButton(RIGHT_ICON) {
                @Override
                public Dimension getPreferredSize() {
                    return new Dimension(super.getPreferredSize().width, PREFERRED_HEIGHT);
                }
            };
            rightButton.setUI(new UIButtonUI() {
                @Override
                protected void doExtraPainting(UIButton b, Graphics2D g2d, int w, int h, String selectedRoles) {
                    if (isPressed(b) && b.isPressedPainted()) {
                        GUIPaintUtils.fillPressed(g2d, 0, 0, w, h, b.isRoundBorder(), b.getRectDirection(), b.isDoneAuthorityEdited(selectedRoles), UIConstants.PROPERTY_PANE_BACKGROUND);
                    } else if (isRollOver(b)) {
                        GUIPaintUtils.fillRollOver(g2d, 0, 0, w, h, b.isRoundBorder(), b.getRectDirection(), b.isDoneAuthorityEdited(selectedRoles), b.isPressedPainted(), UIConstants.PROPERTY_PANE_BACKGROUND);
                    } else if (b.isNormalPainted()) {
                        GUIPaintUtils.fillNormal(g2d, 0, 0, w, h, b.isRoundBorder(), b.getRectDirection(), b.isDoneAuthorityEdited(selectedRoles), b.isPressedPainted());
                    }
                }
            });
            rightButton.set4ToolbarButton();
            rightButton.setDisabledIcon(DISABLED_RIGHT_ICON);
            rightButton.addActionListener(new ActionListener() {
                @Override
                public void actionPerformed(ActionEvent e) {
                    moveRight();
                }
            });
            JPanel buttonPane = new JPanel(new BorderLayout(0, 0));
            buttonPane.add(rightButton, BorderLayout.EAST);
            buttonPane.add(leftButton, BorderLayout.CENTER);
            return buttonPane;
        }

        private void moveLeft() {
            if (scrollIndex > 0) {
                scrollIndex--;
                refreshSheetFlowPane();
            }
        }

        private void moveRight() {
            if (scrollIndex < totalCount - showCount) {
                scrollIndex++;
                refreshSheetFlowPane();
            }
        }

        private void refreshSheetFlowPane() {
            sheetFlowPane.removeAll();
            sheetFlowPane.repaint();
            addButton();
            sheetFlowPane.revalidate();
        }

        private void addButton() {
            showCount = 0;
            int currentWidth = FLOW_GAP;
            for (int i = scrollIndex; i < totalCount; i++) {
                currentWidth += widthArray[i] + FLOW_GAP;
                if (currentWidth > MAX_WIDTH) {
                    break;
                } else {
                    UIToggleButton button = buttonList.get(i);
                    if (selectedIndex == i) {
                        // 如果被选中
                        button.setSelected(true);
                    }
                    sheetFlowPane.add(button);
                    showCount++;
                }
            }
        }

        private void checkButton(boolean buttonEnabled) {
            leftButton.setEnabled(buttonEnabled);
            rightButton.setEnabled(buttonEnabled);
        }

        /**
         * 根据名字和个数计算出所有tab的宽度和名称
         *
         * @return
         */
        private void initWidthAndButtonArray() {
            widthArray = new int[totalCount];
            for (int i = 0; i < totalCount; i++) {
                final int index = i;
                String sheetName = cellWidgetTrees.get(i).getData().getNodeName();
                if (getStringWidth(sheetName) > MAX_WIDTH - 2 * FLOW_GAP) {
                    sheetName = getEllipsisName(sheetName, MAX_WIDTH - 2 * FLOW_GAP);
                }
                // 按钮除了字符串还有20的宽度
                widthArray[i] = getStringWidth(sheetName) + BUTTON_EXTRA_WIDTH;
                UIToggleButton button = new UIToggleButton(sheetName) {
                    @Override
                    protected MouseListener getMouseListener() {
                        return new MouseAdapter() {
                            @Override
                            public void mousePressed(MouseEvent e) {
                                setSelectedStatus(index);
                                eventConfigPane.update();
                                populate(cellWidgetTrees.get(index));
                            }
                        };
                    }
                };
                buttonList.add(button);
            }
        }

        private void setSelectedStatus(int index) {
            selectedIndex = index;
            for (int i = 0; i < buttonList.size(); i++) {
                if (i == index) {
                    buttonList.get(i).setSelected(true);
                } else {
                    buttonList.get(i).setSelected(false);
                }
            }
        }

        private int getStringWidth(String str) {
            return GraphHelper.getFontMetrics(this.getFont()).stringWidth(str);
        }

        /**
         * 判断tab文字的长度大于能装下的最大文字长度，要用省略号
         *
         * @param name
         * @param maxStringlength
         * @return
         */
        private String getEllipsisName(String name, int maxStringlength) {

            //若是名字长度大于能显示的长度，那能显示的文字的最大长度还要减去省略号的最大长度
            int ellipsisWidth = getStringWidth(ELLIPSIS);
            int leftKeyPoint = 0;
            int rightKeyPoint = name.length() - 1;
            int leftStrWidth = 0;
            int rightStrWidth = 0;
            while (leftStrWidth + rightStrWidth + ellipsisWidth < maxStringlength) {
                if (leftStrWidth <= rightStrWidth) {
                    leftKeyPoint++;
                } else {
                    rightKeyPoint--;
                }
                leftStrWidth = getStringWidth(name.substring(0, leftKeyPoint));
                rightStrWidth = getStringWidth(name.substring(rightKeyPoint));

                if (leftStrWidth + rightStrWidth + ellipsisWidth > maxStringlength) {
                    if (leftStrWidth <= rightStrWidth) {
                        rightKeyPoint++;
                    }
                    break;
                }
            }

            return name.substring(0, leftKeyPoint) + ELLIPSIS + name.substring(rightKeyPoint);
        }
    }
}