Quickly hacked together capability to thumbnail animated gifs based on code snippets...

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

diff --git a/src/main/java/eu/svjatoslav/meviz/htmlindexer/metadata/fileTypes/Picture.java b/src/main/java/eu/svjatoslav/meviz/htmlindexer/metadata/fileTypes/Picture.java
index 9f35743..7a1ba39 100755 (executable)
--- a/src/main/java/eu/svjatoslav/meviz/htmlindexer/metadata/fileTypes/Picture.java
+++ b/src/main/java/eu/svjatoslav/meviz/htmlindexer/metadata/fileTypes/Picture.java
@@ -10,20 +10,25 @@
 package eu.svjatoslav.meviz.htmlindexer.metadata.fileTypes;
 
 import eu.svjatoslav.commons.file.FilePathParser;
+import eu.svjatoslav.commons.file.IOHelper;
 import eu.svjatoslav.meviz.htmlindexer.Constants;
+import eu.svjatoslav.meviz.htmlindexer.GifSequenceWriter;
+import eu.svjatoslav.meviz.htmlindexer.ImageFormatError;
 import eu.svjatoslav.meviz.htmlindexer.Utils;
 import eu.svjatoslav.meviz.htmlindexer.metadata.Dimension;
 
 import javax.imageio.ImageIO;
+import javax.imageio.ImageReader;
+import javax.imageio.stream.FileImageOutputStream;
+import javax.imageio.stream.ImageInputStream;
+import javax.imageio.stream.ImageOutputStream;
 import javax.swing.*;
 import java.awt.*;
 import java.awt.image.BufferedImage;
 import java.awt.image.FilteredImageSource;
 import java.awt.image.ImageFilter;
 import java.awt.image.ImageProducer;
-import java.io.File;
-import java.io.FileOutputStream;
-import java.io.OutputStream;
+import java.io.*;
 import java.util.ArrayList;
 import java.util.List;
 
@@ -106,7 +111,21 @@ public class Picture extends AbstractFile {
                                      final File outputFile,
                                      final java.awt.Dimension preferredTargetDimensions) {
 
+        String fileExtension = FilePathParser.getFileExtension(inputFile.getName());
+
         try {
+            if ("gif".equalsIgnoreCase(fileExtension))
+                makeGifThumbnail(inputFile, outputFile, preferredTargetDimensions);
+            else
+                makeJpegThumbnail(inputFile, outputFile, preferredTargetDimensions);
+        } catch (final Exception exception) {
+            System.out.println(exception.toString());
+            exception.printStackTrace();
+        }
+    }
+
+    private static void makeJpegThumbnail(File inputFile, File outputFile, java.awt.Dimension preferredTargetDimensions)
+            throws IOException, ImageFormatError {
 
             final BufferedImage inputImage = getBufferedImage(inputFile);
 
@@ -131,11 +150,52 @@ public class Picture extends AbstractFile {
 
             ImageIO.write(bufferedImage, "jpg", out);
             out.close();
+    }
 
-        } catch (final Exception exception) {
-            System.out.println(exception.toString());
-            exception.printStackTrace();
+    private static void makeGifThumbnail(
+            File inputFile, File outputFile, java.awt.Dimension preferredTargetDimensions) throws IOException {
+        ImageIcon imageIcon = new ImageIcon(IOHelper.getFileContents(inputFile));
+
+        final java.awt.Dimension sourceImageDimension = new java.awt.Dimension(
+                imageIcon.getIconWidth(), imageIcon.getIconHeight());
+
+        System.out.println("Source image dimensions:" + sourceImageDimension);
+
+        final java.awt.Dimension targetDimensions = getTargetThumbnailDimension(
+                sourceImageDimension, preferredTargetDimensions);
+
+        System.out.println("Desired target image dimensions:" + targetDimensions);
+
+        FileInputStream fiStream = new FileInputStream( inputFile );
+
+        ImageReader reader = ImageIO.getImageReadersByFormatName("gif").next();
+        ImageInputStream stream = ImageIO.createImageInputStream(inputFile);
+        reader.setInput(stream);
+
+        Utils.ImageFrame[] frames = Utils.readGIF(reader);
+        for (Utils.ImageFrame frame : frames) {
+            Image scaleImage = scaleImage(frame.image, targetDimensions.width, targetDimensions.height);
+            BufferedImage bimage = new BufferedImage(
+                    targetDimensions.width, targetDimensions.height, BufferedImage.TYPE_INT_ARGB);
+            Graphics2D bGr = bimage.createGraphics();
+            bGr.drawImage(scaleImage, 0, 0, null);
+            bGr.dispose();
+            frame.image = bimage;
         }
+
+        ImageOutputStream output = new FileImageOutputStream(outputFile );
+
+        GifSequenceWriter writer =
+                new GifSequenceWriter( output, frames[0].image.getType(), frames[0].getDelay(), true );
+
+        writer.writeToSequence( frames[0].image );
+        for ( int i = 1; i < frames.length; i++ ) {
+            BufferedImage nextImage = frames[i].image;
+            writer.writeToSequence( nextImage );
+        }
+
+        writer.close();
+        output.close();
     }
 
     /**