123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306 |
- /* ====================================================================
- 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 org.apache.poi.sl.draw;
-
- import java.awt.AlphaComposite;
- import java.awt.Dimension;
- import java.awt.Graphics;
- import java.awt.Graphics2D;
- import java.awt.Insets;
- import java.awt.Shape;
- import java.awt.geom.AffineTransform;
- import java.awt.geom.Rectangle2D;
- import java.awt.image.AffineTransformOp;
- import java.awt.image.BufferedImage;
- import java.awt.image.RescaleOp;
- import java.io.ByteArrayInputStream;
- import java.io.IOException;
- import java.io.InputStream;
- import java.util.Iterator;
-
- import javax.imageio.ImageIO;
- import javax.imageio.ImageReadParam;
- import javax.imageio.ImageReader;
- import javax.imageio.ImageTypeSpecifier;
- import javax.imageio.stream.ImageInputStream;
- import javax.imageio.stream.MemoryCacheImageInputStream;
-
- import org.apache.poi.util.POILogFactory;
- import org.apache.poi.util.POILogger;
-
- /**
- * For now this class renders only images supported by the javax.imageio.ImageIO framework.
- **/
- public class BitmapImageRenderer implements ImageRenderer {
- private final static POILogger LOG = POILogFactory.getLogger(ImageRenderer.class);
-
- protected BufferedImage img;
-
- @Override
- public void loadImage(InputStream data, String contentType) throws IOException {
- img = readImage(data, contentType);
- }
-
- @Override
- public void loadImage(byte data[], String contentType) throws IOException {
- img = readImage(new ByteArrayInputStream(data), contentType);
- }
-
- /**
- * Read the image data via ImageIO and optionally try to workaround metadata errors.
- * The resulting image is of image type {@link BufferedImage#TYPE_INT_ARGB}
- *
- * @param data the data stream
- * @param contentType the content type
- * @return the bufferedImage or null, if there was no image reader for this content type
- * @throws IOException thrown if there was an error while processing the image
- */
- private static BufferedImage readImage(InputStream data, String contentType) throws IOException {
- IOException lastException = null;
- BufferedImage img = null;
- if (data.markSupported()) {
- data.mark(data.available());
- }
-
- // currently don't use FileCacheImageInputStream,
- // because of the risk of filling the file handles (see #59166)
- ImageInputStream iis = new MemoryCacheImageInputStream(data);
- try {
- iis = new MemoryCacheImageInputStream(data);
- iis.mark();
-
- Iterator<ImageReader> iter = ImageIO.getImageReaders(iis);
- while (img==null && iter.hasNext()) {
- ImageReader reader = iter.next();
- ImageReadParam param = reader.getDefaultReadParam();
- // 0:default mode, 1:fallback mode
- for (int mode=0; img==null && mode<3; mode++) {
- lastException = null;
- try {
- iis.reset();
- } catch (IOException e) {
- if (data.markSupported()) {
- data.reset();
- data.mark(data.available());
- iis.close();
- iis = new MemoryCacheImageInputStream(data);
- } else {
- // can't restore the input stream, so we need to stop processing here
- lastException = e;
- break;
- }
- }
- iis.mark();
-
- try {
-
- switch (mode) {
- case 0:
- reader.setInput(iis, false, true);
- img = reader.read(0, param);
- break;
- case 1: {
- // try to load picture in gray scale mode
- // fallback mode for invalid image band metadata
- // see http://stackoverflow.com/questions/10416378
- Iterator<ImageTypeSpecifier> imageTypes = reader.getImageTypes(0);
- while (imageTypes.hasNext()) {
- ImageTypeSpecifier imageTypeSpecifier = imageTypes.next();
- int bufferedImageType = imageTypeSpecifier.getBufferedImageType();
- if (bufferedImageType == BufferedImage.TYPE_BYTE_GRAY) {
- param.setDestinationType(imageTypeSpecifier);
- break;
- }
- }
- reader.setInput(iis, false, true);
- img = reader.read(0, param);
- break;
- }
- case 2: {
- // try to load truncated pictures by supplying a BufferedImage
- // and use the processed data up till the point of error
- reader.setInput(iis, false, true);
- int height = reader.getHeight(0);
- int width = reader.getWidth(0);
-
- Iterator<ImageTypeSpecifier> imageTypes = reader.getImageTypes(0);
- if (imageTypes.hasNext()) {
- ImageTypeSpecifier imageTypeSpecifier = imageTypes.next();
- img = imageTypeSpecifier.createBufferedImage(width, height);
- param.setDestination(img);
- } else {
- lastException = new IOException("unable to load even a truncated version of the image.");
- break;
- }
-
- try {
- reader.read(0, param);
- } finally {
- if (img.getType() != BufferedImage.TYPE_INT_ARGB) {
- int y = findTruncatedBlackBox(img, width, height);
- if (y < height) {
- BufferedImage argbImg = new BufferedImage(width, height, BufferedImage.TYPE_INT_ARGB);
- Graphics2D g = argbImg.createGraphics();
- g.clipRect(0, 0, width, y);
- g.drawImage(img, 0, 0, null);
- g.dispose();
- img.flush();
- img = argbImg;
- }
- }
- }
- break;
- }
- }
-
- } catch (IOException e) {
- if (mode < 2) {
- lastException = e;
- }
- } catch (RuntimeException e) {
- if (mode < 2) {
- lastException = new IOException("ImageIO runtime exception - "+(mode==0 ? "normal" : "fallback"), e);
- }
- }
- }
- reader.dispose();
- }
- } finally {
- iis.close();
- }
-
- // If you don't have an image at the end of all readers
- if (img == null) {
- if (lastException != null) {
- // rethrow exception - be aware that the exception source can be in
- // multiple locations above ...
- throw lastException;
- }
- LOG.log(POILogger.WARN, "Content-type: "+contentType+" is not support. Image ignored.");
- return null;
- }
-
- // add alpha channel
- if (img.getType() != BufferedImage.TYPE_INT_ARGB) {
- BufferedImage argbImg = new BufferedImage(img.getWidth(), img.getHeight(), BufferedImage.TYPE_INT_ARGB);
- Graphics g = argbImg.getGraphics();
- g.drawImage(img, 0, 0, null);
- g.dispose();
- return argbImg;
- }
-
- return img;
- }
-
- private static int findTruncatedBlackBox(BufferedImage img, int width, int height) {
- // scan through the image to find the black box after the truncated data
- int h = height-1;
- for (; h > 0; h--) {
- for (int w = width-1; w > 0; w-=width/10) {
- int p = img.getRGB(w, h);
- if (p != 0xff000000) {
- return h+1;
- }
- }
- }
- return 0;
- }
-
-
- @Override
- public BufferedImage getImage() {
- return img;
- }
-
- @Override
- public BufferedImage getImage(Dimension dim) {
- double w_old = img.getWidth();
- double h_old = img.getHeight();
- BufferedImage scaled = new BufferedImage((int)w_old, (int)h_old, BufferedImage.TYPE_INT_ARGB);
- double w_new = dim.getWidth();
- double h_new = dim.getHeight();
- AffineTransform at = new AffineTransform();
- at.scale(w_new/w_old, h_new/h_old);
- AffineTransformOp scaleOp = new AffineTransformOp(at, AffineTransformOp.TYPE_BILINEAR);
- scaleOp.filter(img, scaled);
- return scaled;
- }
-
- @Override
- public Dimension getDimension() {
- return (img == null)
- ? new Dimension(0,0)
- : new Dimension(img.getWidth(),img.getHeight());
- }
-
- @Override
- public void setAlpha(double alpha) {
- if (img == null) return;
-
- Dimension dim = getDimension();
- BufferedImage newImg = new BufferedImage((int)dim.getWidth(), (int)dim.getHeight(), BufferedImage.TYPE_INT_ARGB);
- Graphics2D g = newImg.createGraphics();
- RescaleOp op = new RescaleOp(new float[]{1.0f, 1.0f, 1.0f, (float)alpha}, new float[]{0,0,0,0}, null);
- g.drawImage(img, op, 0, 0);
- g.dispose();
-
- img = newImg;
- }
-
-
- @Override
- public boolean drawImage(
- Graphics2D graphics,
- Rectangle2D anchor) {
- return drawImage(graphics, anchor, null);
- }
-
- @Override
- public boolean drawImage(
- Graphics2D graphics,
- Rectangle2D anchor,
- Insets clip) {
- if (img == null) return false;
-
- boolean isClipped = true;
- if (clip == null) {
- isClipped = false;
- clip = new Insets(0,0,0,0);
- }
-
- int iw = img.getWidth();
- int ih = img.getHeight();
-
-
- double cw = (100000-clip.left-clip.right) / 100000.0;
- double ch = (100000-clip.top-clip.bottom) / 100000.0;
- double sx = anchor.getWidth()/(iw*cw);
- double sy = anchor.getHeight()/(ih*ch);
- double tx = anchor.getX()-(iw*sx*clip.left/100000.0);
- double ty = anchor.getY()-(ih*sy*clip.top/100000.0);
-
- AffineTransform at = new AffineTransform(sx, 0, 0, sy, tx, ty) ;
-
- Shape clipOld = graphics.getClip();
- if (isClipped) graphics.clip(anchor.getBounds2D());
- graphics.drawRenderedImage(img, at);
- graphics.setClip(clipOld);
-
- return true;
- }
- }
|