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 final long serialVersionUID = -4156533490858298387L;
  36
  37         private static void ensureNonzeroImageArea(final java.awt.Dimension result) {
  38                 if (result.width < 1)
  39                         result.width = 1;
  40                 if (result.height < 1)
  41                         result.height = 1;
  42         }
  43
  44         /**
  45          * Actual returned target thumbnail size will be adjusted from desired one
  46          * by attempting to generate as large as possible thumbnail, while not
  47          * exceeding provided thumbnail width and height, and preserving aspect
  48          * ratio of original image.
  49          */
  50         public static java.awt.Dimension getTargetThumbnailDimension(
  51                         final java.awt.Dimension originalImageDimension,
  52                         final java.awt.Dimension desiredTargetDimension) {
  53
  54                 final double thumbWidthToHeightRatio = (double) desiredTargetDimension.width
  55                                 / (double) desiredTargetDimension.height;
  56                 final double inputImageWidthToHeightRatio = (double) originalImageDimension.width
  57                                 / (double) originalImageDimension.height;
  58
  59                 if (thumbWidthToHeightRatio < inputImageWidthToHeightRatio) {
  60
  61                         final java.awt.Dimension result = new java.awt.Dimension(
  62                                         desiredTargetDimension.width,
  63                                         (int) (desiredTargetDimension.width / inputImageWidthToHeightRatio));
  64
  65                         ensureNonzeroImageArea(result);
  66
  67                         return result;
  68                 } else {
  69
  70                         final java.awt.Dimension result = new java.awt.Dimension(
  71                                         (int) (desiredTargetDimension.height * inputImageWidthToHeightRatio),
  72                                         desiredTargetDimension.height);
  73
  74                         ensureNonzeroImageArea(result);
  75
  76                         return result;
  77                 }
  78         }
  79
  80         /**
  81          * Propose list of thumbnail dimensions.
  82          */
  83         public static List<Dimension> getThumbnailDimensions(Dimension current) {
  84                 final ArrayList<Dimension> result = new ArrayList<Dimension>();
  85                 result.add(current);
  86
  87                 while (current.getArea() > 1000000) {
  88                         current = current.getScaled(0.5d);
  89                         result.add(current);
  90                 }
  91
  92                 return result;
  93
  94         }
  95
  96         public static void makeThumbnail(final File inputFile,
  97                         final File outputFile,
  98                         final java.awt.Dimension preferredTargetDimensions) {
  99
 100                 try {
 101
 102                         final BufferedImage inputImage = Utils.getBufferedImage(inputFile);
 103
 104                         final java.awt.Dimension sourceImageDimension = new java.awt.Dimension(
 105                                         inputImage.getWidth(), inputImage.getHeight());
 106
 107                         final java.awt.Dimension targetDimensions = getTargetThumbnailDimension(
 108                                         sourceImageDimension, preferredTargetDimensions);
 109
 110                         final OutputStream out = new FileOutputStream(outputFile);
 111
 112                         final Image scaledImage = scaleImage(inputImage,
 113                                         targetDimensions.width, targetDimensions.height);
 114
 115                         final BufferedImage bufferedImage = new BufferedImage(
 116                                         scaledImage.getWidth(null), scaledImage.getHeight(null),
 117                                         BufferedImage.TYPE_INT_RGB);
 118
 119                         final Graphics2D g = bufferedImage.createGraphics();
 120                         g.drawImage(scaledImage, 0, 0, null);
 121                         g.dispose();
 122
 123                         ImageIO.write(bufferedImage, "jpg", out);
 124                         out.close();
 125
 126                 } catch (final Exception exception) {
 127                         System.out.println(exception.toString());
 128                         exception.printStackTrace();
 129                 }
 130         }
 131
 132         /**
 133          * @return new image scaled to desired dimensions
 134          */
 135         public static Image scaleImage(final Image srcImage, final int width,
 136                         final int height) {
 137
 138                 final ImageFilter filter = new java.awt.image.AreaAveragingScaleFilter(
 139                                 width, height);
 140
 141                 final ImageProducer prod = new FilteredImageSource(
 142                                 srcImage.getSource(), filter);
 143
 144                 final Image newImage = Toolkit.getDefaultToolkit().createImage(prod);
 145
 146                 return new ImageIcon(newImage).getImage();
 147         };
 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 }