276 lines
9.7 KiB
Java
276 lines
9.7 KiB
Java
/*
|
|
* 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();
|
|
}
|
|
}
|