1 /*
   2  * Copyright (c) 2007, 2016, Oracle and/or its affiliates. All rights reserved.
   3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
   4  *
   5  * This code is free software; you can redistribute it and/or modify it
   6  * under the terms of the GNU General Public License version 2 only, as
   7  * published by the Free Software Foundation.
   8  *
   9  * This code is distributed in the hope that it will be useful, but WITHOUT
  10  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  11  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  12  * version 2 for more details (a copy is included in the LICENSE file that
  13  * accompanied this code).
  14  *
  15  * You should have received a copy of the GNU General Public License version
  16  * 2 along with this work; if not, write to the Free Software Foundation,
  17  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  18  *
  19  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  20  * or visit www.oracle.com if you need additional information or have any
  21  * questions.
  22  */
  23 
  24 /*
  25  * @test
  26  * @key headful
  27  *
  28  * @bug 6608456
  29  * @author Igor Kushnirskiy
  30  * @modules java.desktop/com.sun.java.swing
  31  * @summary tests if delegate RepaintManager gets invoked.
  32  */
  33 
  34 import java.awt.*;
  35 import java.lang.reflect.Method;
  36 import java.util.concurrent.Callable;
  37 import java.util.concurrent.FutureTask;
  38 import java.util.concurrent.TimeUnit;
  39 
  40 import javax.swing.JComponent;
  41 import javax.swing.JButton;
  42 import javax.swing.JFrame;
  43 import javax.swing.RepaintManager;
  44 import javax.swing.SwingUtilities;
  45 
  46 
  47 
  48 public class bug6608456 {
  49     private static final TestFuture testFuture = new TestFuture();
  50     public static void main(String[] args) throws Exception {
  51         final JComponent component = invokeAndWait(
  52             new Callable<JComponent>() {
  53                 public JComponent call() throws Exception {
  54                     RepaintManager.setCurrentManager(new TestRepaintManager());
  55                     JFrame frame = new JFrame("test");
  56                     frame.setLayout(new FlowLayout());
  57                     JButton button = new JButton("default");
  58 
  59                     frame.add(button);
  60                     button = new JButton("delegate");
  61                     if ( ! registerDelegate(
  62                              button, new TestRepaintManager())) {
  63                         return null;
  64                     }
  65                     frame.add(button);
  66                     frame.pack();
  67                     frame.setVisible(true);
  68                     return button;
  69                 }
  70             });
  71         if (component == null) {
  72             throw new RuntimeException("failed. can not register delegate");
  73         }
  74         blockTillDisplayed(component);
  75         // trigger repaint for delegate RepaintManager
  76         invokeAndWait(
  77             new Callable<Void>() {
  78                 public Void call() {
  79                     component.repaint();
  80                     return null;
  81                 }
  82         });
  83         try {
  84             if (testFuture.get(10, TimeUnit.SECONDS)) {
  85                 // passed
  86             }
  87         } catch (Exception e) {
  88             throw new RuntimeException("failed", e);
  89         } finally {
  90             JFrame frame = (JFrame) SwingUtilities
  91                 .getAncestorOfClass(JFrame.class, component);
  92             if (frame != null) {
  93                 frame.dispose();
  94             }
  95         }
  96     }
  97     static class TestRepaintManager extends RepaintManager {
  98         @Override
  99         public void addDirtyRegion(JComponent c, int x, int y, int w, int h) {
 100             if (RepaintManager.currentManager(c) == this) {
 101                 testFuture.defaultCalled();
 102             } else {
 103                 testFuture.delegateCalled();
 104             }
 105             super.addDirtyRegion(c, x, y, w, h);
 106         }
 107     }
 108     static class TestFuture extends FutureTask<Boolean> {
 109         private volatile boolean defaultCalled = false;
 110         private volatile boolean delegateCalled = false;
 111         public TestFuture() {
 112             super(new Callable<Boolean>() {
 113                 public Boolean call() {
 114                     return null;
 115                 }
 116             });
 117         }
 118         public void defaultCalled() {
 119             defaultCalled = true;
 120             updateState();
 121         }
 122         public void delegateCalled() {
 123             delegateCalled = true;
 124             updateState();
 125         }
 126         private void updateState() {
 127             if (defaultCalled && delegateCalled) {
 128                 set(Boolean.TRUE);
 129             }
 130         }
 131     }
 132 
 133     private static boolean registerDelegate(JComponent c,
 134             RepaintManager repaintManager) {
 135         boolean rv = false;
 136         try {
 137             Class<?> clazz = Class.forName("com.sun.java.swing.SwingUtilities3");
 138             Method method = clazz.getMethod("setDelegateRepaintManager",
 139                 JComponent.class, RepaintManager.class);
 140             method.invoke(clazz, c, repaintManager);
 141             rv = true;
 142         } catch (Exception ignore) {
 143         }
 144         return rv;
 145     }
 146     static <T> T invokeAndWait(Callable<T> callable) throws Exception {
 147         FutureTask<T> future = new FutureTask<T>(callable);
 148         SwingUtilities.invokeLater(future);
 149         return future.get();
 150     }
 151 
 152     public static void blockTillDisplayed(Component comp) {
 153         Point p = null;
 154         while (p == null) {
 155             try {
 156                 p = comp.getLocationOnScreen();
 157             } catch (IllegalStateException e) {
 158                 try {
 159                     Thread.sleep(100);
 160                 } catch (InterruptedException ie) {
 161                 }
 162             }
 163         }
 164     }
 165 }