| /* |
| * 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(); |
| } |
| } |