test/jdk/javax/imageio/plugins/tiff/TIFFImageReadParamTest.java - platform/libcore - Git at Google

 /*
  * Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
  *
  * This code is free software; you can redistribute it and/or modify it
  * under the terms of the GNU General Public License version 2 only, as
  * published by the Free Software Foundation.
  *
  * This code is distributed in the hope that it will be useful, but WITHOUT
  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  * version 2 for more details (a copy is included in the LICENSE file that
  * accompanied this code).
  *
  * You should have received a copy of the GNU General Public License version
  * 2 along with this work; if not, write to the Free Software Foundation,
  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
  *
  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
  * or visit www.oracle.com if you need additional information or have any
  * questions.
  */

 /**
  * @test
  * @bug     8149028
  * @author  a.stepanov
  * @summary check TIFFDirectory manipulation
  *          by means of TIFFImageReadParam
  * @run     main TIFFImageReadParamTest
  */


 import java.awt.Color;
 import java.awt.Graphics;
 import java.awt.image.BufferedImage;
 import java.io.*;
 import javax.imageio.*;
 import javax.imageio.metadata.IIOMetadata;
 import javax.imageio.plugins.tiff.*;
 import javax.imageio.stream.ImageInputStream;
 import javax.imageio.stream.ImageOutputStream;

 public class TIFFImageReadParamTest {

     private final static String FILENAME = "test.tiff";
     private final static int SZ = 100;
     private final static Color C = Color.RED;

     private final static String GEO_DATA = "test params";
     private final static int GEO_N = GeoTIFFTagSet.TAG_GEO_ASCII_PARAMS;

     private final static String EXIF_DATA = "2000:01:01 00:00:01";
     private final static int EXIF_N = ExifTIFFTagSet.TAG_DATE_TIME_ORIGINAL;

     private final static String GPS_DATA =
         ExifGPSTagSet.STATUS_MEASUREMENT_IN_PROGRESS;
     private final static int GPS_N = ExifGPSTagSet.TAG_GPS_STATUS;

     private final static short FAX_DATA =
         FaxTIFFTagSet.CLEAN_FAX_DATA_ERRORS_UNCORRECTED;
     private final static int FAX_N = FaxTIFFTagSet.TAG_CLEAN_FAX_DATA;

     private ImageWriter getTIFFWriter() {

         java.util.Iterator<ImageWriter> writers =
             ImageIO.getImageWritersByFormatName("TIFF");
         if (!writers.hasNext()) {
             throw new RuntimeException("No writers available for TIFF format");
         }
         return writers.next();
     }

     private ImageReader getTIFFReader() {

         java.util.Iterator<ImageReader> readers =
             ImageIO.getImageReadersByFormatName("TIFF");
         if (!readers.hasNext()) {
             throw new RuntimeException("No readers available for TIFF format");
         }
         return readers.next();
     }

     private void check(boolean ok, String msg) {
         if (!ok) { throw new RuntimeException(msg); }
     }

     private void addASCIIField(TIFFDirectory d,
                                String        name,
                                String        data,
                                int           num) {

         String a[] = {data};
         d.addTIFFField(new TIFFField(
             new TIFFTag(name, num, 1 << TIFFTag.TIFF_ASCII),
                 TIFFTag.TIFF_ASCII, 1, a));
     }

     private void checkASCIIValue(TIFFDirectory d,
                                  String        what,
                                  String        data,
                                  int           num) {

         TIFFField f = d.getTIFFField(num);
         check(f.getType() == TIFFTag.TIFF_ASCII, "field type != ASCII");
         check(f.getCount() == 1, "invalid " + what + " data count");
         check(f.getValueAsString(0).equals(data),
             "invalid " + what + " data");
     }


     private void writeImage() throws Exception {

         OutputStream s = new BufferedOutputStream(new FileOutputStream(FILENAME));
         try (ImageOutputStream ios = ImageIO.createImageOutputStream(s)) {
             ImageWriter writer = getTIFFWriter();
             writer.setOutput(ios);

             BufferedImage img =
                 new BufferedImage(SZ, SZ, BufferedImage.TYPE_INT_RGB);
             Graphics g = img.getGraphics();
             g.setColor(C);
             g.fillRect(0, 0, SZ, SZ);
             g.dispose();

             IIOMetadata metadata = writer.getDefaultImageMetadata(
                 new ImageTypeSpecifier(img), writer.getDefaultWriteParam());

             TIFFDirectory dir = TIFFDirectory.createFromMetadata(metadata);

             // add some extension tags
             addASCIIField(dir, "GeoAsciiParamsTag", GEO_DATA, GEO_N);
             addASCIIField(dir, "DateTimeOriginal", EXIF_DATA, EXIF_N);
             addASCIIField(dir, "GPSStatus", GPS_DATA, GPS_N);

             dir.addTIFFField(new TIFFField(new TIFFTag(
                 "CleanFaxData", FAX_N, 1 << TIFFTag.TIFF_SHORT), FAX_DATA));

             IIOMetadata data = dir.getAsMetadata();

             writer.write(new IIOImage(img, null, data));

             ios.flush();
             writer.dispose();
         }
     }

     private void checkImage(BufferedImage img) {

         check(img.getWidth() == SZ, "invalid image width");
         check(img.getHeight() == SZ, "invalid image height");
         Color c = new Color(img.getRGB(SZ / 2, SZ / 2));
         check(c.equals(C), "invalid image color");
     }

     private TIFFDirectory getDir(TIFFTagSet[] add,
                                  TIFFTagSet[] remove) throws Exception {

         ImageReader reader = getTIFFReader();

         ImageInputStream s = ImageIO.createImageInputStream(new File(FILENAME));
         reader.setInput(s, false, false);

         int ni = reader.getNumImages(true);
         check(ni == 1, "invalid number of images: " + ni);

         TIFFImageReadParam param = new TIFFImageReadParam();
         for (TIFFTagSet ts: add) { param.addAllowedTagSet(ts); }
         for (TIFFTagSet ts: remove) { param.removeAllowedTagSet(ts); }

         IIOImage img = reader.readAll(0, param);

         // just in case, check image
         checkImage((BufferedImage) img.getRenderedImage());

         IIOMetadata metadata = img.getMetadata();
         TIFFDirectory dir = TIFFDirectory.createFromMetadata(metadata);

         reader.dispose();
         s.close();

         return dir;
     }

     private void simpleChecks() {

         TIFFImageReadParam param = new TIFFImageReadParam();

         java.util.List<TIFFTagSet> allowed = param.getAllowedTagSets();

         // see docs
         check(allowed.contains(BaselineTIFFTagSet.getInstance()),
             "must contain BaselineTIFFTagSet");
         check(allowed.contains(FaxTIFFTagSet.getInstance()),
             "must contain FaxTIFFTagSet");
         check(allowed.contains(ExifParentTIFFTagSet.getInstance()),
             "must contain ExifParentTIFFTagSet");
         check(allowed.contains(GeoTIFFTagSet.getInstance()),
             "must contain GeoTIFFTagSet");

         TIFFTagSet gps = ExifGPSTagSet.getInstance();
         param.addAllowedTagSet(gps);
         check(param.getAllowedTagSets().contains(gps),
             "must contain ExifGPSTagSet");

         param.removeAllowedTagSet(gps);
         check(!param.getAllowedTagSets().contains(gps),
             "must not contain ExifGPSTagSet");

         // check that repeating remove goes properly
         param.removeAllowedTagSet(gps);

         boolean ok = false;
         try { param.addAllowedTagSet(null); }
         catch (IllegalArgumentException e) { ok = true; }
         check(ok, "must not be able to add null tag set");

         ok = false;
         try { param.removeAllowedTagSet(null); }
         catch (IllegalArgumentException e) { ok = true; }
         check(ok, "must not be able to remove null tag set");
     }

     private void run() {

         simpleChecks();

         try {

             writeImage();

             TIFFTagSet
                 empty[] = {},
                 geo[]   = {  GeoTIFFTagSet.getInstance() },
                 exif[]  = { ExifTIFFTagSet.getInstance() },
                 gps[]   = {  ExifGPSTagSet.getInstance() },
                 fax[]   = {  FaxTIFFTagSet.getInstance() };

             // default param state
             TIFFDirectory dir = getDir(empty, empty);
             // Geo and Fax are default allowed tag sets
             check(dir.containsTIFFField(GEO_N), "must contain Geo field");
             checkASCIIValue(dir, "Geo", GEO_DATA, GEO_N);
             check(dir.containsTIFFField(FAX_N), "must contain Fax field");
             check(
                 (dir.getTIFFField(FAX_N).getCount() == 1) &&
                 (dir.getTIFFField(FAX_N).getAsInt(0) == FAX_DATA),
                 "invalid Fax field value");

             // corresponding tag sets are non-default
             check(!dir.containsTIFFField(EXIF_N), "must not contain Geo field");
             check(!dir.containsTIFFField(GPS_N), "must not contain GPS field");

             // remove Fax
             dir = getDir(empty, fax);
             check(!dir.containsTIFFField(FAX_N), "must not contain Fax field");

             // add EXIF, remove Geo
             dir = getDir(exif, geo);
             check(dir.containsTIFFField(EXIF_N), "must contain EXIF field");
             checkASCIIValue(dir, "EXIF", EXIF_DATA, EXIF_N);
             check(!dir.containsTIFFField(GEO_N), "must not contain Geo field");

             // add GPS
             dir = getDir(gps, empty);
             check(dir.containsTIFFField(GPS_N), "must contain GPS field");
             checkASCIIValue(dir, "GPS", GPS_DATA, GPS_N);

         } catch (Exception e) { throw new RuntimeException(e); }
     }

     public static void main(String[] args) {
         (new TIFFImageReadParamTest()).run();
     }
 }
	/*
	* Copyright (c) 2016, Oracle and/or its affiliates. All rights reserved.
	* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
	*
	* This code is free software; you can redistribute it and/or modify it
	* under the terms of the GNU General Public License version 2 only, as
	* published by the Free Software Foundation.
	*
	* This code is distributed in the hope that it will be useful, but WITHOUT
	* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
	* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
	* version 2 for more details (a copy is included in the LICENSE file that
	* accompanied this code).
	*
	* You should have received a copy of the GNU General Public License version
	* 2 along with this work; if not, write to the Free Software Foundation,
	* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
	*
	* Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA
	* or visit www.oracle.com if you need additional information or have any
	* questions.
	*/

	/**
	* @test
	* @bug 8149028
	* @author a.stepanov
	* @summary check TIFFDirectory manipulation
	* by means of TIFFImageReadParam
	* @run main TIFFImageReadParamTest
	*/


	import java.awt.Color;
	import java.awt.Graphics;
	import java.awt.image.BufferedImage;
	import java.io.*;
	import javax.imageio.*;
	import javax.imageio.metadata.IIOMetadata;
	import javax.imageio.plugins.tiff.*;
	import javax.imageio.stream.ImageInputStream;
	import javax.imageio.stream.ImageOutputStream;

	public class TIFFImageReadParamTest {

	private final static String FILENAME = "test.tiff";
	private final static int SZ = 100;
	private final static Color C = Color.RED;

	private final static String GEO_DATA = "test params";
	private final static int GEO_N = GeoTIFFTagSet.TAG_GEO_ASCII_PARAMS;

	private final static String EXIF_DATA = "2000:01:01 00:00:01";
	private final static int EXIF_N = ExifTIFFTagSet.TAG_DATE_TIME_ORIGINAL;

	private final static String GPS_DATA =
	ExifGPSTagSet.STATUS_MEASUREMENT_IN_PROGRESS;
	private final static int GPS_N = ExifGPSTagSet.TAG_GPS_STATUS;

	private final static short FAX_DATA =
	FaxTIFFTagSet.CLEAN_FAX_DATA_ERRORS_UNCORRECTED;
	private final static int FAX_N = FaxTIFFTagSet.TAG_CLEAN_FAX_DATA;

	private ImageWriter getTIFFWriter() {

	java.util.Iterator<ImageWriter> writers =
	ImageIO.getImageWritersByFormatName("TIFF");
	if (!writers.hasNext()) {
	throw new RuntimeException("No writers available for TIFF format");
	}
	return writers.next();
	}

	private ImageReader getTIFFReader() {

	java.util.Iterator<ImageReader> readers =
	ImageIO.getImageReadersByFormatName("TIFF");
	if (!readers.hasNext()) {
	throw new RuntimeException("No readers available for TIFF format");
	}
	return readers.next();
	}

	private void check(boolean ok, String msg) {
	if (!ok) { throw new RuntimeException(msg); }
	}

	private void addASCIIField(TIFFDirectory d,
	String name,
	String data,
	int num) {

	String a[] = {data};
	d.addTIFFField(new TIFFField(
	new TIFFTag(name, num, 1 << TIFFTag.TIFF_ASCII),
	TIFFTag.TIFF_ASCII, 1, a));
	}

	private void checkASCIIValue(TIFFDirectory d,
	String what,
	String data,
	int num) {

	TIFFField f = d.getTIFFField(num);
	check(f.getType() == TIFFTag.TIFF_ASCII, "field type != ASCII");
	check(f.getCount() == 1, "invalid " + what + " data count");
	check(f.getValueAsString(0).equals(data),
	"invalid " + what + " data");
	}


	private void writeImage() throws Exception {

	OutputStream s = new BufferedOutputStream(new FileOutputStream(FILENAME));
	try (ImageOutputStream ios = ImageIO.createImageOutputStream(s)) {
	ImageWriter writer = getTIFFWriter();
	writer.setOutput(ios);

	BufferedImage img =
	new BufferedImage(SZ, SZ, BufferedImage.TYPE_INT_RGB);
	Graphics g = img.getGraphics();
	g.setColor(C);
	g.fillRect(0, 0, SZ, SZ);
	g.dispose();

	IIOMetadata metadata = writer.getDefaultImageMetadata(
	new ImageTypeSpecifier(img), writer.getDefaultWriteParam());

	TIFFDirectory dir = TIFFDirectory.createFromMetadata(metadata);

	// add some extension tags
	addASCIIField(dir, "GeoAsciiParamsTag", GEO_DATA, GEO_N);
	addASCIIField(dir, "DateTimeOriginal", EXIF_DATA, EXIF_N);
	addASCIIField(dir, "GPSStatus", GPS_DATA, GPS_N);

	dir.addTIFFField(new TIFFField(new TIFFTag(
	"CleanFaxData", FAX_N, 1 << TIFFTag.TIFF_SHORT), FAX_DATA));

	IIOMetadata data = dir.getAsMetadata();

	writer.write(new IIOImage(img, null, data));

	ios.flush();
	writer.dispose();
	}
	}

	private void checkImage(BufferedImage img) {

	check(img.getWidth() == SZ, "invalid image width");
	check(img.getHeight() == SZ, "invalid image height");
	Color c = new Color(img.getRGB(SZ / 2, SZ / 2));
	check(c.equals(C), "invalid image color");
	}

	private TIFFDirectory getDir(TIFFTagSet[] add,
	TIFFTagSet[] remove) throws Exception {

	ImageReader reader = getTIFFReader();

	ImageInputStream s = ImageIO.createImageInputStream(new File(FILENAME));
	reader.setInput(s, false, false);

	int ni = reader.getNumImages(true);
	check(ni == 1, "invalid number of images: " + ni);

	TIFFImageReadParam param = new TIFFImageReadParam();
	for (TIFFTagSet ts: add) { param.addAllowedTagSet(ts); }
	for (TIFFTagSet ts: remove) { param.removeAllowedTagSet(ts); }

	IIOImage img = reader.readAll(0, param);

	// just in case, check image
	checkImage((BufferedImage) img.getRenderedImage());

	IIOMetadata metadata = img.getMetadata();
	TIFFDirectory dir = TIFFDirectory.createFromMetadata(metadata);

	reader.dispose();
	s.close();

	return dir;
	}

	private void simpleChecks() {

	TIFFImageReadParam param = new TIFFImageReadParam();

	java.util.List<TIFFTagSet> allowed = param.getAllowedTagSets();

	// see docs
	check(allowed.contains(BaselineTIFFTagSet.getInstance()),
	"must contain BaselineTIFFTagSet");
	check(allowed.contains(FaxTIFFTagSet.getInstance()),
	"must contain FaxTIFFTagSet");
	check(allowed.contains(ExifParentTIFFTagSet.getInstance()),
	"must contain ExifParentTIFFTagSet");
	check(allowed.contains(GeoTIFFTagSet.getInstance()),
	"must contain GeoTIFFTagSet");

	TIFFTagSet gps = ExifGPSTagSet.getInstance();
	param.addAllowedTagSet(gps);
	check(param.getAllowedTagSets().contains(gps),
	"must contain ExifGPSTagSet");

	param.removeAllowedTagSet(gps);
	check(!param.getAllowedTagSets().contains(gps),
	"must not contain ExifGPSTagSet");

	// check that repeating remove goes properly
	param.removeAllowedTagSet(gps);

	boolean ok = false;
	try { param.addAllowedTagSet(null); }
	catch (IllegalArgumentException e) { ok = true; }
	check(ok, "must not be able to add null tag set");

	ok = false;
	try { param.removeAllowedTagSet(null); }
	catch (IllegalArgumentException e) { ok = true; }
	check(ok, "must not be able to remove null tag set");
	}

	private void run() {

	simpleChecks();

	try {

	writeImage();

	TIFFTagSet
	empty[] = {},
	geo[] = { GeoTIFFTagSet.getInstance() },
	exif[] = { ExifTIFFTagSet.getInstance() },
	gps[] = { ExifGPSTagSet.getInstance() },
	fax[] = { FaxTIFFTagSet.getInstance() };

	// default param state
	TIFFDirectory dir = getDir(empty, empty);
	// Geo and Fax are default allowed tag sets
	check(dir.containsTIFFField(GEO_N), "must contain Geo field");
	checkASCIIValue(dir, "Geo", GEO_DATA, GEO_N);
	check(dir.containsTIFFField(FAX_N), "must contain Fax field");
	check(
	(dir.getTIFFField(FAX_N).getCount() == 1) &&
	(dir.getTIFFField(FAX_N).getAsInt(0) == FAX_DATA),
	"invalid Fax field value");

	// corresponding tag sets are non-default
	check(!dir.containsTIFFField(EXIF_N), "must not contain Geo field");
	check(!dir.containsTIFFField(GPS_N), "must not contain GPS field");

	// remove Fax
	dir = getDir(empty, fax);
	check(!dir.containsTIFFField(FAX_N), "must not contain Fax field");

	// add EXIF, remove Geo
	dir = getDir(exif, geo);
	check(dir.containsTIFFField(EXIF_N), "must contain EXIF field");
	checkASCIIValue(dir, "EXIF", EXIF_DATA, EXIF_N);
	check(!dir.containsTIFFField(GEO_N), "must not contain Geo field");

	// add GPS
	dir = getDir(gps, empty);
	check(dir.containsTIFFField(GPS_N), "must contain GPS field");
	checkASCIIValue(dir, "GPS", GPS_DATA, GPS_N);

	} catch (Exception e) { throw new RuntimeException(e); }
	}

	public static void main(String[] args) {
	(new TIFFImageReadParamTest()).run();
	}
	}