Java tutorial
// Copyright 2000-2019 JetBrains s.r.o. Use of this source code is governed by the Apache 2.0 license that can be found in the LICENSE file. package com.intellij.openapi.ui; import com.intellij.icons.AllIcons; import com.intellij.ui.ColorUtil; import com.intellij.ui.ComponentUtil; import com.intellij.ui.JBColor; import com.intellij.ui.ScreenUtil; import com.intellij.util.ui.GraphicsUtil; import com.intellij.util.ui.TimerUtil; import org.jetbrains.annotations.NotNull; import javax.swing.*; import javax.swing.event.PopupMenuEvent; import javax.swing.event.PopupMenuListener; import javax.swing.plaf.basic.DefaultMenuLayout; import java.awt.*; import java.awt.event.ActionEvent; import java.awt.event.ActionListener; import java.awt.event.MouseEvent; import java.awt.event.MouseWheelEvent; /** * @author ignatov */ public class JBPopupMenu extends JPopupMenu { private final MyLayout myLayout; public JBPopupMenu() { this(null); } public JBPopupMenu(String label) { super(label); enableEvents(AWTEvent.MOUSE_EVENT_MASK | AWTEvent.MOUSE_WHEEL_EVENT_MASK); myLayout = new MyLayout(this); setLayout(myLayout); setLightWeightPopupEnabled(false); } @Override public void processMouseWheelEvent(MouseWheelEvent e) { if (!isShowing()) return; int rotation = e.getWheelRotation(); if (rotation == 0) return; if (e.getComponent() != this) { e = (MouseWheelEvent) SwingUtilities.convertMouseEvent(e.getComponent(), e, this); } Point p = e.getPoint(); SwingUtilities.convertPointToScreen(p, this); Point tPoint = getLocationOnScreen(); if (p.x >= tPoint.x && p.x <= tPoint.x + getWidth() && p.y >= tPoint.y && p.y <= tPoint.y + getHeight()) { myLayout.updateShift(rotation * 10); } } @Override public void setLayout(LayoutManager mgr) { if (!(mgr instanceof MyLayout)) return; super.setLayout(mgr); } @Override public void paint(Graphics g) { GraphicsUtil.setupAntialiasing(g); super.paint(g); LayoutManager layout = getLayout(); if (layout instanceof MyLayout) { ((MyLayout) layout).paintIfNeeded(g); } } private static class MyLayout extends DefaultMenuLayout implements ActionListener { private final JPopupMenu myTarget; int myShift = 0; int myScrollDirection = 0; Timer myTimer; MyLayout(final JPopupMenu target) { super(target, BoxLayout.PAGE_AXIS); myTarget = target; myTimer = TimerUtil.createNamedTimer("PopupTimer", 40, this); myTarget.addPopupMenuListener(new PopupMenuListener() { @Override public void popupMenuWillBecomeVisible(PopupMenuEvent e) { switchTimer(true); } @Override public void popupMenuWillBecomeInvisible(PopupMenuEvent e) { switchTimer(false); JRootPane rootPane = SwingUtilities.getRootPane(target); if (rootPane != null) { rootPane.putClientProperty("apple.awt._windowFadeDelegate", null); } } @Override public void popupMenuCanceled(PopupMenuEvent e) { switchTimer(false); } }); if (myTarget.isVisible()) { switchTimer(true); } } private void switchTimer(boolean on) { if (on && !myTimer.isRunning()) { myTimer.start(); } if (!on && myTimer.isRunning()) { myTimer.stop(); } } @Override public void actionPerformed(ActionEvent e) { if (!myTarget.isShowing()) return; PointerInfo info = MouseInfo.getPointerInfo(); if (info == null) return; Point mouseLocation = info.getLocation(); Point targetLocation = myTarget.getLocationOnScreen(); if (mouseLocation.x < targetLocation.x || mouseLocation.x > targetLocation.x + myTarget.getWidth()) { return; } if (Math.abs(mouseLocation.y - targetLocation.y - getMaxHeight()) < 10) { myScrollDirection = 1; } else if (Math.abs(mouseLocation.y - targetLocation.y) < 10) { myScrollDirection = -1; } else { myScrollDirection = 0; } if (myScrollDirection == 0) { myTarget.revalidate(); myTarget.repaint(); return; } SwingUtilities.convertPointFromScreen(mouseLocation, myTarget); myTarget.dispatchEvent(new MouseEvent(myTarget, MouseEvent.MOUSE_ENTERED, System.currentTimeMillis(), 0, mouseLocation.x, mouseLocation.y, 0, false)); updateShift(5 * myScrollDirection); } private void updateShift(int increment) { int maxHeight = super.preferredLayoutSize(myTarget).height - getMaxHeight(); int newShift = Math.max(0, Math.min(maxHeight, myShift + increment)); if (newShift != myShift) { myShift = newShift; myTarget.revalidate(); myTarget.repaint(); Window w = ComponentUtil.getWindow(myTarget.getComponent()); if (w != null) { for (Window window : w.getOwnedWindows()) { window.dispose(); } } } } private final Color[] dim = new Color[] { JBColor.background(), ColorUtil.withAlpha(JBColor.background(), .9), ColorUtil.withAlpha(JBColor.background(), .8), ColorUtil.withAlpha(JBColor.background(), .7), ColorUtil.withAlpha(JBColor.background(), .6), ColorUtil.withAlpha(JBColor.background(), .5), ColorUtil.withAlpha(JBColor.background(), .4), ColorUtil.withAlpha(JBColor.background(), .3), ColorUtil.withAlpha(JBColor.background(), .2), ColorUtil.withAlpha(JBColor.background(), .1), }; public void paintIfNeeded(Graphics g) { if (myShift > 0) { for (int i = 0; i < dim.length; i++) { g.setColor(dim[i]); g.drawLine(0, i, myTarget.getWidth(), i); } AllIcons.General.ArrowUp.paintIcon(myTarget, g, myTarget.getWidth() / 2 - AllIcons.General.ArrowUp.getIconWidth() / 2, 0); } if (super.preferredLayoutSize(myTarget).height - getMaxHeight() - myShift > 0) { for (int i = 0; i < dim.length; i++) { g.setColor(dim[i]); g.drawLine(0, myTarget.getHeight() - i, myTarget.getWidth(), myTarget.getHeight() - i); } AllIcons.General.ArrowDown.paintIcon(myTarget, g, myTarget.getWidth() / 2 - AllIcons.General.ArrowDown.getIconWidth() / 2, myTarget.getHeight() - AllIcons.General.ArrowDown.getIconHeight()); } } @Override public void layoutContainer(Container target) { Insets insets = target.getInsets(); int width = target.getWidth() - insets.left - insets.right; Component[] components = target.getComponents(); int y = -myShift + insets.top; for (Component component : components) { int height = component.getPreferredSize().height; component.setBounds(insets.left, y, width, height); y += height; } } private int getMaxHeight() { GraphicsConfiguration configuration = myTarget.getGraphicsConfiguration(); if (configuration == null && myTarget.getInvoker() != null) { configuration = myTarget.getInvoker().getGraphicsConfiguration(); } if (configuration == null) return Short.MAX_VALUE; Rectangle screenRectangle = ScreenUtil.getScreenRectangle(configuration); return screenRectangle.height; } @NotNull @Override public Dimension preferredLayoutSize(Container target) { Dimension dimension = super.preferredLayoutSize(target); dimension.height = Math.min(getMaxHeight(), dimension.height); return dimension; } } }