/*
* Licensed to the Apache Software Foundation (ASF) under one or more
* contributor license agreements. See the NOTICE file distributed with
* this work for additional information regarding copyright ownership.
* The ASF licenses this file to You under the Apache License, Version 2.0
* (the "License"); you may not use this file except in compliance with
* the License. You may obtain a copy of the License at
*
* http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
package com.google.code.appengine.imageio;
import java.io.File;
import java.io.IOException;
import java.io.InputStream;
import java.io.OutputStream;
import java.net.URL;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.Iterator;
import java.util.List;
import org.apache.harmony.luni.util.NotImplementedException;
import org.apache.harmony.x.imageio.internal.nls.Messages;
import org.apache.sanselan.ImageReadException;
import org.apache.sanselan.common.byteSources.ByteSourceInputStream;
import org.apache.sanselan.formats.png.PngImageParser;
import com.google.code.appengine.awt.image.BufferedImage;
import com.google.code.appengine.awt.image.RenderedImage;
import com.google.code.appengine.imageio.spi.IIORegistry;
import com.google.code.appengine.imageio.spi.ImageInputStreamSpi;
import com.google.code.appengine.imageio.spi.ImageOutputStreamSpi;
import com.google.code.appengine.imageio.spi.ImageReaderSpi;
import com.google.code.appengine.imageio.spi.ImageReaderWriterSpi;
import com.google.code.appengine.imageio.spi.ImageWriterSpi;
import com.google.code.appengine.imageio.spi.ServiceRegistry;
import com.google.code.appengine.imageio.stream.ImageInputStream;
import com.google.code.appengine.imageio.stream.ImageOutputStream;
public final class ImageIO {
private static final IIORegistry registry = IIORegistry.getDefaultInstance();
private static final Cache cacheInfo = new Cache();
private ImageIO() {}
public static void scanForPlugins() {
registry.registerApplicationClasspathSpis();
}
public static void setUseCache(boolean useCache) {
cacheInfo.setUseCache(useCache);
}
public static boolean getUseCache() {
return cacheInfo.getUseCache();
}
public static void setCacheDirectory(File cacheDirectory) {
cacheInfo.setCacheDirectory(cacheDirectory);
}
public static File getCacheDirectory() {
return cacheInfo.getCacheDirectory();
}
public static ImageInputStream createImageInputStream(Object input)
throws IOException {
if (input == null) {
throw new IllegalArgumentException(Messages.getString("imageio.4C"));
}
Iterator<ImageInputStreamSpi> it = registry.getServiceProviders(ImageInputStreamSpi.class, true);
while (it.hasNext()) {
ImageInputStreamSpi spi = it.next();
if (spi.getInputClass().isInstance(input)) {
return getUseCache() ?
spi.createInputStreamInstance(input, true, getCacheDirectory()) :
spi.createInputStreamInstance(input);
}
}
return null;
}
public static ImageOutputStream createImageOutputStream(Object output)
throws IOException {
if (output == null) {
throw new IllegalArgumentException(Messages.getString("imageio.4D"));
}
Iterator<ImageOutputStreamSpi> it = registry.getServiceProviders(ImageOutputStreamSpi.class, true);
while (it.hasNext()) {
ImageOutputStreamSpi spi = it.next();
if (spi.getOutputClass().isInstance(output)) {
return getUseCache() ?
spi.createOutputStreamInstance(output, true, getCacheDirectory()) :
spi.createOutputStreamInstance(output);
}
}
return null;
}
public static String[] getReaderFormatNames() {
List<String> FormatNames = new ArrayList<String>();
Iterator<ImageReaderSpi> it = registry.getServiceProviders(ImageReaderSpi.class, true);
while (it.hasNext()) {
ImageReaderSpi spi = it.next();
FormatNames.addAll(Arrays.asList(spi.getFormatNames()));
}
return (String[])FormatNames.toArray(new String[FormatNames.size()]);
}
public static String[] getReaderMIMETypes() {
List<String> MIMETypes = new ArrayList<String>();
Iterator<ImageReaderSpi> it = registry.getServiceProviders(ImageReaderSpi.class, true);
while (it.hasNext()) {
ImageReaderSpi spi = it.next();
MIMETypes.addAll(Arrays.asList(spi.getMIMETypes()));
}
return (String[])MIMETypes.toArray(new String[MIMETypes.size()]);
}
public static Iterator<ImageReader> getImageReaders(Object input) {
if (input == null) {
throw new NullPointerException(Messages.getString("imageio.4E"));
}
Iterator<ImageReaderSpi> it = registry.getServiceProviders(ImageReaderSpi.class,
new CanReadFilter(input), true);
return new SpiIteratorToReadersIteratorWrapper(it);
}
public static Iterator<ImageReader> getImageReadersByFormatName(String formatName) {
if (formatName == null) {
throw new NullPointerException(Messages.getString("imageio.4F"));
}
Iterator<ImageReaderSpi> it = registry.getServiceProviders(ImageReaderSpi.class,
new FormatFilter(formatName), true);
return new SpiIteratorToReadersIteratorWrapper(it);
}
public static Iterator<ImageReader> getImageReadersBySuffix(String fileSuffix) {
if (fileSuffix == null) {
throw new NullPointerException(Messages.getString("imageio.50"));
}
Iterator<ImageReaderSpi> it = registry.getServiceProviders(ImageReaderSpi.class,
new SuffixFilter(fileSuffix), true);
return new SpiIteratorToReadersIteratorWrapper(it);
}
public static Iterator<ImageReader> getImageReadersByMIMEType(
final String MIMEType) {
return new SpiIteratorToReadersIteratorWrapper(
registry.getServiceProviders(ImageReaderSpi.class,
new MIMETypeFilter(MIMEType), true));
}
public static String[] getWriterFormatNames() {
ArrayList<String> FormatNames = new ArrayList<String>();
Iterator<ImageWriterSpi> it = registry.getServiceProviders(ImageWriterSpi.class, true);
while (it.hasNext()) {
ImageWriterSpi spi = it.next();
FormatNames.addAll(Arrays.asList(spi.getFormatNames()));
}
return (String[])FormatNames.toArray(new String[FormatNames.size()]);
}
public static String[] getWriterMIMETypes() {
ArrayList<String> MIMETypes = new ArrayList<String>();
Iterator<ImageWriterSpi> it = registry.getServiceProviders(ImageWriterSpi.class, true);
while (it.hasNext()) {
ImageWriterSpi spi = it.next();
MIMETypes.addAll(Arrays.asList(spi.getMIMETypes()));
}
return (String[])MIMETypes.toArray(new String[MIMETypes.size()]);
}
public static Iterator<ImageWriter> getImageWritersByFormatName(String formatName) {
if (formatName == null) {
throw new NullPointerException(Messages.getString("imageio.4F"));
}
Iterator<ImageWriterSpi> it = registry.getServiceProviders(ImageWriterSpi.class,
new FormatFilter(formatName), true);
return new SpiIteratorToWritersIteratorWrapper(it);
}
public static Iterator<ImageWriter> getImageWritersBySuffix(String fileSuffix) {
if (fileSuffix == null) {
throw new NullPointerException(Messages.getString("imageio.50"));
}
Iterator<ImageWriterSpi> it = registry.getServiceProviders(ImageWriterSpi.class,
new SuffixFilter(fileSuffix), true);
return new SpiIteratorToWritersIteratorWrapper(it);
}
public static Iterator<ImageWriter> getImageWritersByMIMEType(
final String MIMEType) {
return new SpiIteratorToWritersIteratorWrapper(
registry.getServiceProviders(ImageWriterSpi.class,
new MIMETypeFilter(MIMEType), true));
}
public static ImageWriter getImageWriter(ImageReader reader) {
if (reader == null) {
// imageio.97=Reader cannot be null
throw new IllegalArgumentException(Messages.getString("imageio.97")); //$NON-NLS-1$
}
ImageReaderSpi readerSpi = reader.getOriginatingProvider();
if (readerSpi.getImageWriterSpiNames() == null) {
return null;
}
String writerSpiName = readerSpi.getImageWriterSpiNames()[0];
Iterator<ImageWriterSpi> writerSpis;
writerSpis = registry.getServiceProviders(ImageWriterSpi.class, true);
try {
while (writerSpis.hasNext()) {
ImageWriterSpi writerSpi = writerSpis.next();
if (writerSpi.getClass().getName().equals(writerSpiName)) {
return writerSpi.createWriterInstance();
}
}
} catch (IOException e) {
// Ignored
}
return null;
}
public static ImageReader getImageReader(ImageWriter writer) {
if (writer == null) {
// imageio.96=Writer cannot be null
throw new IllegalArgumentException(Messages.getString("imageio.96")); //$NON-NLS-1$
}
ImageWriterSpi writerSpi = writer.getOriginatingProvider();
if (writerSpi.getImageReaderSpiNames() == null) {
return null;
}
String readerSpiName = writerSpi.getImageReaderSpiNames()[0];
Iterator<ImageReaderSpi> readerSpis;
readerSpis = registry.getServiceProviders(ImageReaderSpi.class, true);
try {
while (readerSpis.hasNext()) {
ImageReaderSpi readerSpi = readerSpis.next();
if (readerSpi.getClass().getName().equals(readerSpiName)) {
return readerSpi.createReaderInstance();
}
}
} catch (IOException e) {
// Ignored
}
return null;
}
public static Iterator<ImageWriter> getImageWriters(ImageTypeSpecifier type,
String formatName) {
if (type == null) {
throw new NullPointerException(Messages.getString("imageio.51"));
}
if (formatName == null) {
throw new NullPointerException(Messages.getString("imageio.4F"));
}
Iterator<ImageWriterSpi> it = registry.getServiceProviders(ImageWriterSpi.class,
new FormatAndEncodeFilter(type, formatName), true);
return new SpiIteratorToWritersIteratorWrapper(it);
}
public static Iterator<ImageTranscoder> getImageTranscoders(ImageReader reader,
ImageWriter writer) throws NotImplementedException {
// TODO: implement
throw new NotImplementedException();
}
public static BufferedImage read(File input) throws IOException {
if (input == null) {
throw new IllegalArgumentException(Messages.getString("imageio.52"));
}
if(input.getName().endsWith(".png")) {
PngImageParser parser = new PngImageParser();
try {
return parser.getBufferedImage(input, new HashMap());
} catch (ImageReadException e) {
throw new IOException(e);
}
}
ImageInputStream stream = createImageInputStream(input);
return read(stream);
}
public static BufferedImage read(InputStream input) throws IOException {
if (input == null) {
throw new IllegalArgumentException(Messages.getString("imageio.52"));
}
PngImageParser parser = new PngImageParser();
try {
return parser.getBufferedImage(new ByteSourceInputStream(input, "file.png"), Collections.emptyMap());
} catch (ImageReadException e) {
throw new IOException(e);
}
}
public static BufferedImage read(URL input) throws IOException {
if (input == null) {
throw new IllegalArgumentException(Messages.getString("imageio.52"));
}
InputStream stream = input.openStream();
BufferedImage res = read(stream);
stream.close();
return res;
}
public static BufferedImage read(ImageInputStream stream) throws IOException {
if (stream == null) {
throw new IllegalArgumentException(Messages.getString("imageio.0A"));
}
Iterator<ImageReader> imageReaders = getImageReaders(stream);
if (!imageReaders.hasNext()) {
return null;
}
ImageReader reader = imageReaders.next();
reader.setInput(stream, false, true);
BufferedImage res = reader.read(0);
reader.dispose();
try {
stream.close();
} catch (IOException e) {
// Stream could be already closed, proceed silently in this case
}
return res;
}
public static boolean write(RenderedImage im,
String formatName,
ImageOutputStream output)
throws IOException {
if (im == null) {
throw new IllegalArgumentException(Messages.getString("imageio.53"));
}
if (formatName == null) {
throw new IllegalArgumentException(Messages.getString("imageio.4F"));
}
if (output == null) {
throw new IllegalArgumentException(Messages.getString("imageio.54"));
}
Iterator<ImageWriter> it = getImageWriters(ImageTypeSpecifier.createFromRenderedImage(im), formatName);
if (it.hasNext()) {
ImageWriter writer = it.next();
writer.setOutput(output);
writer.write(im);
output.flush();
writer.dispose();
return true;
}
return false;
}
public static boolean write(RenderedImage im,
String formatName,
File output)
throws IOException {
if (output == null) {
throw new IllegalArgumentException(Messages.getString("imageio.54"));
}
if (output.exists()) {
output.delete();
}
ImageOutputStream ios = createImageOutputStream(output);
boolean rt = write(im, formatName, ios);
ios.close();
return rt;
}
public static boolean write(RenderedImage im,
String formatName,
OutputStream output)
throws IOException {
if (output == null) {
throw new IllegalArgumentException(Messages.getString("imageio.54"));
}
ImageOutputStream ios = createImageOutputStream(output);
boolean rt = write(im, formatName, ios);
ios.close();
return rt;
}
private static class Cache {
// we can't use a cache in AppEngine
private boolean useCache = false;
private File cacheDirectory = null;
public Cache() {
}
public File getCacheDirectory() {
return cacheDirectory;
}
public void setCacheDirectory(File cacheDirectory) {
if ((cacheDirectory != null) && (!cacheDirectory.isDirectory())) {
throw new IllegalArgumentException(Messages.getString("imageio.0B"));
}
SecurityManager security = System.getSecurityManager();
if (security != null) {
String filepath;
if (cacheDirectory == null) {
filepath = System.getProperty("java.io.tmpdir");
} else {
filepath = cacheDirectory.getPath();
}
security.checkWrite(filepath);
}
this.cacheDirectory = cacheDirectory;
}
public boolean getUseCache() {
return useCache;
}
public void setUseCache(boolean useCache) {
this.useCache = useCache;
}
}
/**
* Filter to match spi by format name
*/
static class FormatFilter implements ServiceRegistry.Filter {
private String name;
public FormatFilter(String name) {
this.name = name;
}
public boolean filter(Object provider) {
ImageReaderWriterSpi spi = (ImageReaderWriterSpi) provider;
return Arrays.asList(spi.getFormatNames()).contains(name);
}
}
/**
* Filter to match spi by format name and encoding possibility
*/
static class FormatAndEncodeFilter extends FormatFilter {
private ImageTypeSpecifier type;
public FormatAndEncodeFilter(ImageTypeSpecifier type, String name) {
super(name);
this.type = type;
}
@Override
public boolean filter(Object provider) {
ImageWriterSpi spi = (ImageWriterSpi) provider;
return super.filter(provider) && spi.canEncodeImage(type);
}
}
/**
* Filter to match spi by suffix
*/
static class SuffixFilter implements ServiceRegistry.Filter {
private String suf;
public SuffixFilter(String suf) {
this.suf = suf;
}
public boolean filter(Object provider) {
ImageReaderWriterSpi spi = (ImageReaderWriterSpi) provider;
return Arrays.asList(spi.getFileSuffixes()).contains(suf);
}
}
/**
* Filter to match spi by MIMEType
*/
static class MIMETypeFilter implements ServiceRegistry.Filter {
private final String mimeType;
public MIMETypeFilter(final String mimeType) {
if (mimeType == null) {
throw new NullPointerException(Messages.getString("imageio.55"));
}
this.mimeType = mimeType;
}
public boolean filter(final Object provider) {
final String[] types = ((ImageReaderWriterSpi) provider).getMIMETypes();
return (types != null) && Arrays.asList(types).contains(mimeType);
}
}
/**
* Filter to match spi by decoding possibility
*/
static class CanReadFilter implements ServiceRegistry.Filter {
private Object input;
public CanReadFilter(Object input) {
this.input = input;
}
public boolean filter(Object provider) {
ImageReaderSpi spi = (ImageReaderSpi) provider;
try {
return spi.canDecodeInput(input);
} catch (IOException e) {
return false;
}
}
}
/**
* Wraps Spi's iterator to ImageWriter iterator
*/
static class SpiIteratorToWritersIteratorWrapper implements Iterator<ImageWriter> {
private Iterator<ImageWriterSpi> backend;
public SpiIteratorToWritersIteratorWrapper(Iterator<ImageWriterSpi> backend) {
this.backend = backend;
}
public ImageWriter next() {
try {
return backend.next().createWriterInstance();
} catch (IOException e) {
e.printStackTrace();
return null;
}
}
public boolean hasNext() {
return backend.hasNext();
}
public void remove() {
throw new UnsupportedOperationException(Messages.getString("imageio.56"));
}
}
/**
* Wraps spi's iterator to ImageReader iterator
*/
static class SpiIteratorToReadersIteratorWrapper implements Iterator<ImageReader> {
private Iterator<ImageReaderSpi> backend;
public SpiIteratorToReadersIteratorWrapper(Iterator<ImageReaderSpi> backend) {
this.backend = backend;
}
public ImageReader next() {
try {
return backend.next().createReaderInstance();
} catch (IOException e) {
e.printStackTrace();
return null;
}
}
public boolean hasNext() {
return backend.hasNext();
}
public void remove() {
throw new UnsupportedOperationException(Messages.getString("imageio.56"));
}
}
}