RibbonToolbar.java

package edu.stanford.hci.r3.util.components.ribbons;

import java.awt.BorderLayout;
import java.awt.CardLayout;
import java.awt.Color;
import java.awt.Component;
import java.awt.GradientPaint;
import java.awt.Graphics;
import java.awt.Graphics2D;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;

import javax.swing.BoxLayout;
import javax.swing.JPanel;

import edu.stanford.hci.r3.util.graphics.GraphicsUtils;
import edu.stanford.hci.r3.util.layout.StackedLayout;

public class RibbonToolbar extends JPanel implements RibbonConstants {

        private final CardLayout cardLayout = new CardLayout();

        private JPanel categories;

        private HashMap<String, JPanel> categoriesToPanels = new HashMap<String, JPanel>();

        private List<RibbonCategoryLabel> categoryTabList = new ArrayList<RibbonCategoryLabel>();

        private List<RibbonPanel> tools = new ArrayList<RibbonPanel>();

        private JPanel toolsPanel;

        public RibbonToolbar() {
                setLayout(new BorderLayout());
                add(getCategories(), BorderLayout.NORTH);
                add(getTools(), BorderLayout.CENTER);
        }

        public void addCategories(String... categoryNames) {
                for (String name : categoryNames) {
                        addCategory(name);
                }
        }

        public void addCategory(String categoryName) {
                // the clickable label at the top
                final RibbonCategoryLabel labelledTab = new RibbonCategoryLabel(categoryName, this);
                categories.add(labelledTab, "LeftTall");
                categoryTabList.add(labelledTab);

                // select the first one!
                if (categoryTabList.size() == 1) {
                        labelledTab.setSelected(true);
                } else {
                        labelledTab.setSelected(false);
                }

                final JPanel categoryPanel = getNewCategoryPanel();

                // put it into the card panel
                getTools().add(categoryName, categoryPanel);
                labelledTab.setCardLayoutInfo(cardLayout, getTools());

                // the corresponding panel of tools
                categoriesToPanels.put(categoryName, categoryPanel);
        }

        public void addToolsToCategoryPanel(String panelName, List<RibbonPanel> panels) {
                for (RibbonPanel rp : panels) {
                        addToolToCategoryPanel(panelName, rp);
                }
        }

        private void addToolToCategoryPanel(String categoryName, RibbonPanel rp) {
                final JPanel panel = categoriesToPanels.get(categoryName);
                // System.out.println(categoriesToPanels.keySet());
                panel.add(rp);
                panel.invalidate();
        }

        private Component getCategories() {
                if (categories == null) {
                        categories = new JPanel() {
                                final GradientPaint gradient = new GradientPaint(0, 0, new Color(202, 198, 195),
                                                getWidth(), 0, new Color(234, 234, 234), false);

                                protected void paintComponent(Graphics g) {
                                        super.paintComponent(g);
                                        final Graphics2D g2d = (Graphics2D) g;
                                        g2d.setRenderingHints(GraphicsUtils.getBestRenderingHints());
                                        g2d.setPaint(gradient);
                                        g2d.fillRect(0, 0, getWidth(), getHeight());
                                }

                        };
                        categories.setLayout(new StackedLayout(StackedLayout.HORIZONTAL));
                }
                return categories;
        }

        private JPanel getNewCategoryPanel() {
                final JPanel jp = new JPanel() {

                        protected void paintComponent(Graphics g) {
                                super.paintComponent(g);

                                final Graphics2D g2d = (Graphics2D) g;

                                g2d.setColor(HEADER_COLOR);
                                g2d.fillRect(0, 0, getWidth(), HEADER_HEIGHT);

                                // paint the background panel gradient
                                final GradientPaint midToBottomGradient = new GradientPaint(0, HEADER_HEIGHT,
                                                PANEL_BEGIN_COLOR, 0, getHeight(), PANEL_END_COLOR, false);

                                g2d.setPaint(midToBottomGradient);
                                g2d.fillRect(0, HEADER_HEIGHT, getWidth(), getHeight());
                        }
                };

                jp.setLayout(new BoxLayout(jp, BoxLayout.X_AXIS));
                return jp;
        }

        private JPanel getTools() {
                if (toolsPanel == null) {
                        toolsPanel = new JPanel();
                        toolsPanel.setLayout(cardLayout);
                }
                return toolsPanel;
        }

        public void registerTool(RibbonPanel rp) {
                tools.add(rp);
        }

        public void setSelectedLabel(RibbonCategoryLabel labelToSelect) {
                for (RibbonCategoryLabel tab : categoryTabList) {
                        if (tab.equals(labelToSelect)) {
                                tab.setSelected(true);
                        } else {
                                tab.setSelected(false);
                        }
                }
        }
}

---