1 /*
   2  * Copyright (c) 2001, 2017, 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  * @bug 4420342 4421831
  27  * @summary Checks that IIOWriteProgressListener methods are called in proper
  28  *          sequence for the JPEG and PNG writers
  29  */
  30 
  31 import java.awt.image.BufferedImage;
  32 import java.io.File;
  33 import java.io.IOException;
  34 import java.util.Iterator;
  35 import javax.imageio.ImageIO;
  36 import javax.imageio.ImageWriter;
  37 import javax.imageio.event.IIOWriteProgressListener;
  38 import javax.imageio.stream.ImageOutputStream;
  39 
  40 public class WriteProgressListenerTest implements IIOWriteProgressListener {
  41 
  42     final static int UNINITIALIZED = 0;
  43     final static int IMAGE_STARTED = 1;
  44     final static int IMAGE_COMPLETE = 2;
  45 
  46     int state = UNINITIALIZED;
  47     float prevPercentageDone = 0.0F;
  48     File tempFile = null;
  49 
  50     public WriteProgressListenerTest(String format) throws IOException {
  51         ImageWriter writer = null;
  52         Iterator witer = ImageIO.getImageWritersByFormatName(format);
  53         if (!witer.hasNext()) {
  54             error("No writer for format " + format + "!");
  55         }
  56         writer = (ImageWriter)witer.next();
  57 
  58         System.out.println("Got writer " + writer);
  59         writer.addIIOWriteProgressListener(this);
  60 
  61         this.tempFile = File.createTempFile("imageio", ".tmp");
  62         tempFile.deleteOnExit();
  63         ImageOutputStream stream = ImageIO.createImageOutputStream(tempFile);
  64         writer.setOutput(stream);
  65 
  66         BufferedImage im =
  67             new BufferedImage(100, 100, BufferedImage.TYPE_3BYTE_BGR);
  68 
  69         this.state = UNINITIALIZED;
  70 
  71         writer.write(im);
  72 
  73         if (this.state == UNINITIALIZED) {
  74             error("imageStarted never called!");
  75         }
  76         if (this.state != IMAGE_COMPLETE) {
  77             error("imageComplete not called!");
  78         }
  79 
  80         print("Passed!");
  81     }
  82 
  83     private void error(String s) {
  84         if (tempFile != null) {
  85             tempFile.delete();
  86         }
  87         throw new RuntimeException(s);
  88     }
  89 
  90     private void print(String s) {
  91         System.out.println(s);
  92     }
  93 
  94     public void sequenceStarted(ImageWriter source) {
  95         error("Obsolete method sequenceStarted was called!");
  96     }
  97 
  98     public void sequenceComplete(ImageWriter source) {
  99         error("Obsolete method sequenceComplete was called!");
 100     }
 101 
 102     public void imageStarted(ImageWriter source, int imageIndex) {
 103         print("imageStarted: imageIndex = " + imageIndex);
 104 
 105         if (state != UNINITIALIZED) {
 106             error("imageStarted not called first!");
 107         }
 108         state = IMAGE_STARTED;
 109         prevPercentageDone = 0.0F;
 110     }
 111 
 112     public void imageProgress(ImageWriter source,
 113                               float percentageDone) {
 114         print("imageProgress: percentageDone = " + percentageDone);
 115 
 116         if (state != IMAGE_STARTED) {
 117             error("imageProgress called without prior imageStarted!");
 118         }
 119         if (percentageDone < prevPercentageDone) {
 120             error("percentageDone did not increase!");
 121         }
 122         prevPercentageDone = percentageDone;
 123     }
 124 
 125     public void imageComplete(ImageWriter source) {
 126         print("imageComplete");
 127 
 128         if (state != IMAGE_STARTED) {
 129             error("imageComplete called without imageStarted!");
 130         }
 131         if (prevPercentageDone == 0.0F) {
 132             error("percentageDone was never updated!");
 133         }
 134         state = IMAGE_COMPLETE;
 135     }
 136 
 137     public void thumbnailStarted(ImageWriter source,
 138                                  int imageIndex, int thumbnailIndex) {
 139     }
 140 
 141     public void thumbnailProgress(ImageWriter source, float percentageDone) {
 142     }
 143 
 144     public void thumbnailComplete(ImageWriter source) {
 145     }
 146 
 147     public void writeAborted(ImageWriter source) {
 148     }
 149 
 150     public static void main(String[] args) throws IOException {
 151         new WriteProgressListenerTest("jpeg");
 152         new WriteProgressListenerTest("png");
 153     }
 154 }