src/main/java/eu/svjatoslav/meviz/htmlindexer/metadata/fileTypes/Picture.java

   1 /*
   2  * Meviz - Various tools collection to work with multimedia.
   3  * Copyright (C) 2012, Svjatoslav Agejenko, svjatoslav@svjatoslav.eu
   4  *
   5  * This program is free software; you can redistribute it and/or
   6  * modify it under the terms of version 2 of the GNU General Public License
   7  * as published by the Free Software Foundation.
   8  */
   9
  10 package eu.svjatoslav.meviz.htmlindexer.metadata.fileTypes;
  11
  12 import java.awt.Graphics2D;
  13 import java.awt.Image;
  14 import java.awt.Toolkit;
  15 import java.awt.image.BufferedImage;
  16 import java.awt.image.FilteredImageSource;
  17 import java.awt.image.ImageFilter;
  18 import java.awt.image.ImageProducer;
  19 import java.io.File;
  20 import java.io.FileOutputStream;
  21 import java.io.IOException;
  22 import java.io.OutputStream;
  23 import java.util.ArrayList;
  24 import java.util.List;
  25
  26 import javax.imageio.ImageIO;
  27 import javax.swing.ImageIcon;
  28
  29 import eu.svjatoslav.commons.file.FilePathParser;
  30 import eu.svjatoslav.meviz.htmlindexer.Constants;
  31 import eu.svjatoslav.meviz.htmlindexer.ImageFormatError;
  32 import eu.svjatoslav.meviz.htmlindexer.Utils;
  33 import eu.svjatoslav.meviz.htmlindexer.metadata.Dimension;
  34
  35 public class Picture extends AbstractFile {
  36
  37         private static final long serialVersionUID = -4156533490858298387L;
  38
  39         /**
  40          * Picture dimensions.
  41          */
  42         private Dimension dimensions;
  43
  44         public Picture(final File parentDirectory, final String fileName)
  45                         throws IOException, ImageFormatError {
  46                 super(parentDirectory, fileName);
  47         }
  48
  49         private String computeThumbnailHash(final Dimension targetDimension) {
  50
  51                 // compute new thumbnails hash number
  52                 final String forMagicHash = fileName + " "
  53                                 + String.valueOf(targetDimension.width) + " "
  54                                 + String.valueOf(targetDimension.height) + " "
  55                                 + String.valueOf(getFileLength()) + " "
  56                                 + Constants.THUMBNAIL_VERSION;
  57
  58                 // System.out.println("Computing hash from: " + forMagicHash);
  59                 return Utils.getStringCrcAsHex(forMagicHash);
  60         }
  61
  62         public Dimension getDimensions() {
  63                 return dimensions;
  64         };
  65
  66         /**
  67          * Get thumbnail file name for this image and desired thumbnail dimensions
  68          * relative to designated thumbnails directory within parent directory.
  69          */
  70         public String getRelativeThumbnailFileName(final Dimension targetDimension) {
  71                 return FilePathParser.getFileNameWithoutExtension(fileName) + " ("
  72                                 + computeThumbnailHash(targetDimension) + ").jpeg";
  73         }
  74
  75         @Override
  76         public void updateFileMetainfo(final File parentDirectory)
  77                         throws IOException {
  78                 BufferedImage image;
  79                 try {
  80                         image = Utils.getBufferedImage(getFile(parentDirectory));
  81                 } catch (final ImageFormatError e) {
  82                         throw new RuntimeException(e);
  83                 }
  84                 dimensions = new Dimension(image.getWidth(), image.getHeight());
  85         }
  86
  87         /**
  88          * Actual returned target thumbnail size will be adjusted from desired one
  89          * by attempting to generate as large as possible thumbnail, while not
  90          * exceeding provided thumbnail width and height, and preserving aspect
  91          * ratio of original image.
  92          */
  93         public static java.awt.Dimension getTargetThumbnailDimension(
  94                         final java.awt.Dimension originalImageDimension,
  95                         final java.awt.Dimension desiredTargetDimension) {
  96
  97                 final double thumbWidthToHeightRatio = (double) desiredTargetDimension.width
  98                                 / (double) desiredTargetDimension.height;
  99                 final double inputImageWidthToHeightRatio = (double) originalImageDimension.width
 100                                 / (double) originalImageDimension.height;
 101
 102                 if (thumbWidthToHeightRatio < inputImageWidthToHeightRatio) {
 103
 104                         final java.awt.Dimension result = new java.awt.Dimension(
 105                                         desiredTargetDimension.width,
 106                                         (int) (desiredTargetDimension.width / inputImageWidthToHeightRatio));
 107
 108                         return result;
 109                 } else {
 110
 111                         final java.awt.Dimension result = new java.awt.Dimension(
 112                                         (int) (desiredTargetDimension.height * inputImageWidthToHeightRatio),
 113                                         desiredTargetDimension.height);
 114
 115                         return result;
 116                 }
 117         }
 118
 119         /**
 120          * Propose list of thumbnail dimensions.
 121          */
 122         public static List<Dimension> getThumbnailDimensions(Dimension current) {
 123                 final ArrayList<Dimension> result = new ArrayList<Dimension>();
 124                 result.add(current);
 125
 126                 while (current.getArea() > 100000) {
 127                         current = current.getScaled(0.5d);
 128                         result.add(current);
 129                 }
 130
 131                 return result;
 132
 133         }
 134
 135         public static void makeThumbnail(final File inputFile,
 136                         final File outputFile,
 137                         final java.awt.Dimension preferredTargetDimensions) {
 138
 139                 try {
 140
 141                         final BufferedImage inputImage = Utils.getBufferedImage(inputFile);
 142
 143                         final java.awt.Dimension sourceImageDimension = new java.awt.Dimension(
 144                                         inputImage.getWidth(), inputImage.getHeight());
 145
 146                         final java.awt.Dimension targetDimensions = getTargetThumbnailDimension(
 147                                         sourceImageDimension, preferredTargetDimensions);
 148
 149                         final OutputStream out = new FileOutputStream(outputFile);
 150
 151                         final Image scaledImage = scaleImage(inputImage,
 152                                         targetDimensions.width, targetDimensions.height);
 153
 154                         final BufferedImage bufferedImage = new BufferedImage(
 155                                         scaledImage.getWidth(null), scaledImage.getHeight(null),
 156                                         BufferedImage.TYPE_INT_RGB);
 157
 158                         final Graphics2D g = bufferedImage.createGraphics();
 159                         g.drawImage(scaledImage, 0, 0, null);
 160                         g.dispose();
 161
 162                         ImageIO.write(bufferedImage, "jpg", out);
 163                         out.close();
 164
 165                 } catch (final Exception exception) {
 166                         System.out.println(exception.toString());
 167                         exception.printStackTrace();
 168                 }
 169         }
 170
 171         /**
 172          * @return new image scaled to desired dimensions
 173          */
 174         public static Image scaleImage(final Image srcImage, final int width,
 175                         final int height) {
 176
 177                 ImageFilter filter;
 178                 filter = new java.awt.image.AreaAveragingScaleFilter(width, height);
 179
 180                 final ImageProducer prod = new FilteredImageSource(
 181                                 srcImage.getSource(), filter);
 182
 183                 final Image newImage = Toolkit.getDefaultToolkit().createImage(prod);
 184
 185                 final ImageIcon imageIcon = new ImageIcon(newImage);
 186                 return imageIcon.getImage();
 187         }
 188
 189 }