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