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