8149028: [TEST] add test for TIFFDirectory

Reviewed-by: ssadetsky, yan
This commit is contained in:
Alexander Stepanov 2016-04-01 13:52:04 +03:00
parent a7b8a6a5cc
commit fbd868d77c
3 changed files with 796 additions and 0 deletions

View File

@ -0,0 +1,266 @@
/*
* 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 some simple checks for TIFFDirectory
* @run main TIFFDirectoryTest
*/
import java.util.List;
import java.util.ArrayList;
import javax.imageio.metadata.*;
import javax.imageio.plugins.tiff.*;
public class TIFFDirectoryTest {
private static void check(boolean ok, String msg) {
if (!ok) { throw new RuntimeException(msg); }
}
private void run() {
int type = TIFFTag.TIFF_LONG, dt = 1 << type;
int n0 = 1000, n1 = 1001, n2 = 1002, n3 = 1003;
TIFFTag tag1 = new TIFFTag(Integer.toString(n1), n1, dt);
TIFFTag tag2 = new TIFFTag(Integer.toString(n2), n2, dt);
TIFFTag tag3 = new TIFFTag(Integer.toString(n3), n3, dt);
TIFFTag parent = new TIFFTag(Integer.toString(n0), n0, dt);
// tag sets array must not be null
boolean ok = false;
try { new TIFFDirectory(null, parent); }
catch (NullPointerException e) { ok = true; }
check(ok, "can construct TIFFDirectory with null tagsets array");
// but can be empty
TIFFTagSet emptySets[] = {};
TIFFDirectory d = new TIFFDirectory(emptySets, parent);
check(d.getTagSets().length == 0, "invalid number of tag sets");
check(d.getParentTag().getName().equals(Integer.toString(n0)) &&
(d.getParentTag().getNumber() == n0), "invalid parent tag");
// add tags
List<TIFFTag> tags = new ArrayList<>();
tags.add(tag1);
tags.add(tag2);
TIFFTagSet ts1 = new TIFFTagSet(tags);
tags.clear();
tags.add(tag3);
TIFFTagSet ts2 = new TIFFTagSet(tags);
TIFFTagSet sets[] = {ts1, ts2};
d = new TIFFDirectory(sets, parent);
check(d.getTagSets().length == sets.length, "invalid number of tag sets");
// check getTag()
for (int i = n1; i <= n3; i++) {
TIFFTag t = d.getTag(i);
check(t.getNumber() == i, "invalid tag number");
check(t.getName().equals(Integer.toString(i)), "invalid tag name");
check(t.getDataTypes() == dt, "invalid tag data types");
}
TIFFDirectory d2;
try { d2 = d.clone(); }
catch (CloneNotSupportedException e) { throw new RuntimeException(e); }
// check removeTagSet()
d.removeTagSet(ts2);
check(d.getTagSets().length == 1, "invalid number of tag sets");
check(d.getTagSets()[0].getTag(n1).getName().equals(Integer.toString(n1)),
"invalid tag name");
check(d.getTagSets()[0].getTag(n2).getName().equals(Integer.toString(n2)),
"invalid tag name");
d.removeTagSet(ts1);
check(d.getTagSets().length == 0, "invalid number of tag sets");
// check cloned data
check(d2.getTagSets().length == sets.length,
"invalid number of tag sets");
TIFFTagSet sets2[] = d2.getTagSets();
check(sets2.length == sets.length, "invalid number of tag sets");
check(
(sets2[0].getTag(Integer.toString(n1)).getNumber() == n1) &&
(sets2[0].getTag(Integer.toString(n2)).getNumber() == n2) &&
(sets2[0].getTag(Integer.toString(n0)) == null) &&
(sets2[1].getTag(Integer.toString(n3)).getNumber() == n3) &&
(sets2[1].getTag(Integer.toString(n0)) == null), "invalid data");
check(
(sets2[0].getTag(Integer.toString(n1)).getDataTypes() == dt) &&
(sets2[0].getTag(Integer.toString(n2)).getDataTypes() == dt) &&
(sets2[1].getTag(Integer.toString(n3)).getDataTypes() == dt),
"invalid data type");
// must not be able to call removeTagSet with null argument
ok = false;
try { d.removeTagSet(null); }
catch (NullPointerException e) { ok = true; }
check(ok, "must not be able to use null as an argument for remove");
// check parent tag
check( d.getParentTag().getName().equals(Integer.toString(n0)) &&
d2.getParentTag().getName().equals(Integer.toString(n0)),
"invalid parent tag name");
check(( d.getParentTag().getNumber() == n0) &&
(d2.getParentTag().getNumber() == n0),
"invalid parent tag number");
check(( d.getParentTag().getDataTypes() == dt) &&
(d2.getParentTag().getDataTypes() == dt),
"invalid parent data type");
d.addTagSet(ts1);
d.addTagSet(ts2);
// add the same tag set twice and check that nothing changed
d.addTagSet(ts2);
check(d.getTagSets().length == 2, "invalid number of tag sets");
// check field operations
check(d.getNumTIFFFields() == 0, "invalid TIFFFields number");
check(d.getTIFFField(Integer.MAX_VALUE) == null,
"must return null TIFFField");
long offset = 4L;
long a[] = {Long.MIN_VALUE, 0, Long.MAX_VALUE};
int v = 100500;
TIFFField
f1 = new TIFFField(tag1, type, offset, d),
f2 = new TIFFField(tag2, v),
f3 = new TIFFField(tag3, type, a.length, a);
d.addTIFFField(f1);
d.addTIFFField(f2);
d.addTIFFField(f3);
check(d.containsTIFFField(n1) &&
d.containsTIFFField(n2) &&
d.containsTIFFField(n3) &&
!d.containsTIFFField(n0), "invalid containsTIFFField() results");
check(d.getTIFFField(n0) == null, "can get unadded field");
check(d.getNumTIFFFields() == 3, "invalid TIFFFields number");
check(d.getTIFFField(n1).getCount() == 1, "invalid TIFFField count");
check(d.getTIFFField(n1).getAsLong(0) == offset, "invalid offset");
check(d.getTIFFField(n2).getCount() == 1, "invalid TIFFField count");
check(d.getTIFFField(n2).getAsInt(0) == v, "invalid TIFFField value");
check(d.getTIFFField(n3).getCount() == a.length,
"invalid TIFFField count");
for (int i = 0; i < a.length; ++i) {
check(d.getTIFFField(n3).getAsLong(i) == a[i],
"invalid TIFFField value");
}
TIFFField nested = d.getTIFFField(n1).getDirectory().getTIFFField(n1);
check(nested.getTag().getNumber() == n1, "invalid tag number");
check(nested.getCount() == 1, "invalid field count");
check(nested.getAsLong(0) == offset, "invalid offset");
// check that the field is overwritten correctly
int v2 = 1 << 16;
d.addTIFFField(new TIFFField(tag3, v2));
check(d.getTIFFField(n3).getCount() == 1, "invalid TIFFField count");
check(d.getTIFFField(n3).getAsInt(0)== v2, "invalid TIFFField value");
check(d.getNumTIFFFields() == 3, "invalid TIFFFields number");
// check removeTIFFField()
d.removeTIFFField(n3);
check(d.getNumTIFFFields() == 2, "invalid TIFFFields number");
check(d.getTIFFField(n3) == null, "can get removed field");
d.removeTIFFFields();
check((d.getTIFFField(n1) == null) && (d.getTIFFField(n2) == null),
"can get removed field");
check((d.getNumTIFFFields() == 0) && (d.getTIFFFields().length == 0),
"invalid TIFFFields number");
// check that array returned by getTIFFFields() is sorted
// by tag number (as it stated in the docs)
d.addTIFFField(f3);
d.addTIFFField(f1);
d.addTIFFField(f2);
TIFFField fa[] = d.getTIFFFields();
check(fa.length == 3, "invalid number of fields");
check((fa[0].getTagNumber() == n1) &&
(fa[1].getTagNumber() == n2) &&
(fa[2].getTagNumber() == n3),
"array of the fields must be sorted by tag number");
d.removeTIFFFields();
d.addTIFFField(f2);
// test getAsMetaData / createFromMetadata
try {
d2 = TIFFDirectory.createFromMetadata(d.getAsMetadata());
} catch (IIOInvalidTreeException e) {
throw new RuntimeException(e);
}
// check new data
check(d2.getTagSets().length == sets.length,
"invalid number of tag sets");
sets2 = d2.getTagSets();
check(sets2.length == sets.length, "invalid number of tag sets");
check(
(sets2[0].getTag(Integer.toString(n1)).getNumber() == n1) &&
(sets2[0].getTag(Integer.toString(n2)).getNumber() == n2) &&
(sets2[0].getTag(Integer.toString(n0)) == null) &&
(sets2[1].getTag(Integer.toString(n3)).getNumber() == n3) &&
(sets2[1].getTag(Integer.toString(n0)) == null), "invalid data");
check(
(sets2[0].getTag(Integer.toString(n1)).getDataTypes() == dt) &&
(sets2[0].getTag(Integer.toString(n2)).getDataTypes() == dt) &&
(sets2[1].getTag(Integer.toString(n3)).getDataTypes() == dt),
"invalid data type");
check(!d2.containsTIFFField(n1) &&
d2.containsTIFFField(n2) &&
!d2.containsTIFFField(n3), "invalid containsTIFFField() results");
check(d2.getTIFFField(n2).getCount() == 1, "invalid TIFFField count");
check(d2.getTIFFField(n2).getAsInt(0) == v, "invalid TIFFField value");
check((d2.getParentTag().getNumber() == n0) &&
d2.getParentTag().getName().equals(Integer.toString(n0)),
"invalid parent tag");
}
public static void main(String[] args) { (new TIFFDirectoryTest()).run(); }
}

View File

@ -0,0 +1,255 @@
/*
* 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 a simple write-read test for TIFFDirectory
* @run main TIFFDirectoryWriteReadTest
*/
import java.awt.*;
import java.awt.color.*;
import java.awt.image.BufferedImage;
import java.io.*;
import javax.imageio.*;
import javax.imageio.metadata.*;
import javax.imageio.stream.*;
import javax.imageio.plugins.tiff.*;
public class TIFFDirectoryWriteReadTest {
private final static String FILENAME = "test.tiff";
private final static int SZ = 100;
private final static Color C = Color.RED;
private static final String COPYRIGHT[] = {"Copyright 123ABC.."};
private static final String DESCRIPTION[] = {"Test Image", "Description"};
private static final String SOFTWARE[] = {"test", "software", "123"};
private static final long RES_X[][] = {{2, 1}}, RES_Y[][] = {{1, 1}};
private static final byte[] ICC_PROFILE =
ICC_ProfileRGB.getInstance(ColorSpace.CS_sRGB).getData();
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 addASCIIField(TIFFDirectory d,
String name,
String data[],
int num) {
d.addTIFFField(new TIFFField(
new TIFFTag(name, num, 1 << TIFFTag.TIFF_ASCII),
TIFFTag.TIFF_ASCII, data.length, data));
}
private void checkASCIIField(TIFFDirectory d,
String what,
String data[],
int num) {
String notFound = what + " field was not found";
check(d.containsTIFFField(num), notFound);
TIFFField f = d.getTIFFField(num);
check(f.getType() == TIFFTag.TIFF_ASCII, "field type != ASCII");
check(f.getCount() == data.length, "invalid " + what + " data count");
for (int i = 0; i < data.length; i++) {
check(f.getValueAsString(i).equals(data[i]),
"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);
addASCIIField(dir, "Copyright",
COPYRIGHT, BaselineTIFFTagSet.TAG_COPYRIGHT);
addASCIIField(dir, "ImageDescription",
DESCRIPTION, BaselineTIFFTagSet.TAG_IMAGE_DESCRIPTION);
addASCIIField(dir, "Software",
SOFTWARE, BaselineTIFFTagSet.TAG_SOFTWARE);
dir.addTIFFField(new TIFFField(
new TIFFTag("XResolution", BaselineTIFFTagSet.TAG_X_RESOLUTION,
1 << TIFFTag.TIFF_RATIONAL), TIFFTag.TIFF_RATIONAL, 1, RES_X));
dir.addTIFFField(new TIFFField(
new TIFFTag("YResolution", BaselineTIFFTagSet.TAG_Y_RESOLUTION,
1 << TIFFTag.TIFF_RATIONAL), TIFFTag.TIFF_RATIONAL, 1, RES_Y));
dir.addTIFFField(new TIFFField(
new TIFFTag("ICC Profile", BaselineTIFFTagSet.TAG_ICC_PROFILE,
1 << TIFFTag.TIFF_UNDEFINED),
TIFFTag.TIFF_UNDEFINED, ICC_PROFILE.length, ICC_PROFILE));
IIOMetadata data = dir.getAsMetadata();
writer.write(new IIOImage(img, null, data));
ios.flush();
writer.dispose();
}
s.close();
}
private void readAndCheckImage() throws Exception {
ImageReader reader = getTIFFReader();
ImageInputStream s = ImageIO.createImageInputStream(new File(FILENAME));
reader.setInput(s);
int ni = reader.getNumImages(true);
check(ni == 1, "invalid number of images");
// check image
BufferedImage img = reader.read(0);
check(img.getWidth() == SZ && img.getHeight() == SZ,
"invalid image size");
Color c = new Color(img.getRGB(SZ / 2, SZ / 2));
check(C.equals(c), "invalid image color");
IIOMetadata metadata = reader.readAll(0, null).getMetadata();
TIFFDirectory dir = TIFFDirectory.createFromMetadata(metadata);
reader.dispose();
s.close();
// ===== perform tag checks =====
checkASCIIField(dir, "copyright", COPYRIGHT,
BaselineTIFFTagSet.TAG_COPYRIGHT);
checkASCIIField(dir, "description", DESCRIPTION,
BaselineTIFFTagSet.TAG_IMAGE_DESCRIPTION);
checkASCIIField(dir, "software", SOFTWARE,
BaselineTIFFTagSet.TAG_SOFTWARE);
TIFFField f = dir.getTIFFField(BaselineTIFFTagSet.TAG_IMAGE_WIDTH);
check(f.getCount() == 1, "invalid width field count");
int w = f.getAsInt(0);
check(w == SZ, "invalid width");
f = dir.getTIFFField(BaselineTIFFTagSet.TAG_IMAGE_LENGTH);
check(f.getCount() == 1, "invalid height field count");
int h = f.getAsInt(0);
check(h == SZ, "invalid height");
f = dir.getTIFFField(BaselineTIFFTagSet.TAG_BITS_PER_SAMPLE);
// RGB: 3 x 8 bits for R, G and B components
int bps[] = f.getAsInts();
check((f.getCount() == 3) && (bps.length == 3), "invalid BPS count");
for (int b: bps) { check(b == 8, "invalid bits per sample"); }
// RGB: PhotometricInterpretation = 2
f = dir.getTIFFField(BaselineTIFFTagSet.TAG_PHOTOMETRIC_INTERPRETATION);
check(f.getCount() == 1, "invalid count");
check(f.getAsInt(0) == 2, "invalid photometric interpretation for RGB");
String rat = " resolution must be rational";
f = dir.getTIFFField(BaselineTIFFTagSet.TAG_X_RESOLUTION);
check(f.getType() == TIFFTag.TIFF_RATIONAL, "x" + rat);
check(f.getCount() == 1 &&
f.getAsInt(0) == (int) (RES_X[0][0] / RES_X[0][1]),
"invalid x resolution");
f = dir.getTIFFField(BaselineTIFFTagSet.TAG_Y_RESOLUTION);
check(f.getType() == TIFFTag.TIFF_RATIONAL, "y" + rat);
check(f.getCount() == 1 &&
f.getAsInt(0) == (int) (RES_Y[0][0] / RES_Y[0][1]),
"invalid y resolution");
f = dir.getTIFFField(BaselineTIFFTagSet.TAG_ICC_PROFILE);
check(f.getType() == TIFFTag.TIFF_UNDEFINED,
"invalid ICC profile field type");
int cnt = f.getCount();
byte icc[] = f.getAsBytes();
check((cnt == ICC_PROFILE.length) && (cnt == icc.length),
"invalid ICC profile");
for (int i = 0; i < cnt; i++) {
check(icc[i] == ICC_PROFILE[i], "invalid ICC profile");
}
}
public void run() {
try {
writeImage();
readAndCheckImage();
} catch (Exception e) {
throw new RuntimeException(e);
}
}
private void check(boolean ok, String msg) {
if (!ok) { throw new RuntimeException(msg); }
}
public static void main(String[] args) {
(new TIFFDirectoryWriteReadTest()).run();
}
}

View File

@ -0,0 +1,275 @@
/*
* 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, true);
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();
}
}