Image Magick Thumbnail Creator

dspace-api/pom.xml

@@ -531,6 +531,14 @@
             <artifactId>jdbm</artifactId>
             <version>1.0</version>
         </dependency>
+
+        <!-- For ImageMagick MediaFilters -->
+        <dependency>
+            <groupId>org.im4java</groupId>
+            <artifactId>im4java</artifactId>
+            <version>1.4.0</version>
+        </dependency>
+
     </dependencies>
 
 </project>

dspace-api/src/main/java/org/dspace/app/mediafilter/LitImageMagickThumbnailFilter.java

@@ -0,0 +1,157 @@
+package org.dspace.app.mediafilter;
+
+import java.io.File;
+import java.io.FileOutputStream;
+import java.io.IOException;
+import java.io.InputStream;
+
+import javax.imageio.ImageIO;
+
+import org.dspace.app.mediafilter.MediaFilter;
+import org.dspace.app.mediafilter.SelfRegisterInputFormats;
+import org.dspace.content.Bitstream;
+import org.dspace.content.Bundle;
+import org.dspace.content.Item;
+import org.dspace.core.Context;
+import org.im4java.core.ConvertCmd;
+import org.im4java.core.IM4JavaException;
+import org.im4java.core.IMOperation;
+import org.im4java.process.ProcessStarter;
+
+import org.dspace.core.ConfigurationManager;
+
+/**
+ * Filter image bitstreams, scaling the image to be within the bounds of
+ * thumbnail.maxwidth, thumbnail.maxheight, the size we want our thumbnail to be
+ * no bigger than. Creates only JPEGs.
+ */
+public abstract class LitImageMagickThumbnailFilter extends MediaFilter implements SelfRegisterInputFormats
+{
+	int width = 180;
+	int height = 120;
+	public LitImageMagickThumbnailFilter() {
+		String pre = LitImageMagickThumbnailFilter.class.getName();
+		//System.out.println(pre+ ".ProcessStarter");
+		String s = ConfigurationManager.getProperty(pre + ".ProcessStarter");
+		ProcessStarter.setGlobalSearchPath(s);
+		width = ConfigurationManager.getIntProperty(pre + ".ThumbnailWidth", width);
+		height = ConfigurationManager.getIntProperty(pre + ".ThumbnailHeight", height);
+		//System.out.println(s+ " "+ width+" "+height);
+	}
+	
+	
+    public String getFilteredName(String oldFilename)
+    {
+        return oldFilename + ".jpg";
+    }
+
+    /**
+     * @return String bundle name
+     *  
+     */
+    public String getBundleName()
+    {
+        return "THUMBNAIL";
+    }
+
+    /**
+     * @return String bitstreamformat
+     */
+    public String getFormatString()
+    {
+        return "JPEG";
+    }
+
+    /**
+     * @return String description
+     */
+    public String getDescription()
+    {
+        return "LIT Thumbnail";
+    }
+
+    public File inputStreamToTempFile(InputStream source, String prefix, String suffix) throws IOException {
+		File f = File.createTempFile(prefix, suffix);
+		f.deleteOnExit();
+    	FileOutputStream fos = new FileOutputStream(f);
+    	
+    	byte[] buffer = new byte[1024];
+    	int len = source.read(buffer);
+    	while (len != -1) {
+    		fos.write(buffer, 0, len);
+    		len = source.read(buffer);
+    	}
+    	fos.close();
+		return f;
+    }
+    
+    public File getThumbnailFile(File f) throws IOException, InterruptedException, IM4JavaException {
+    	File f2 = new File(f.getParentFile(), f.getName() + ".jpg");
+    	f2.deleteOnExit();
+    	ConvertCmd cmd = new ConvertCmd();
+		IMOperation op = new IMOperation();
+		op.addImage(f.getAbsolutePath());
+		op.thumbnail(width, height);
+		op.addImage(f2.getAbsolutePath());
+		//System.out.println("THUMB: "+op);
+		cmd.run(op);
+		return f2;
+    }
+    
+    public File getImageFile(File f, int page) throws IOException, InterruptedException, IM4JavaException {
+    	File f2 = new File(f.getParentFile(), f.getName() + ".jpg");
+    	f2.deleteOnExit();
+    	ConvertCmd cmd = new ConvertCmd();
+		IMOperation op = new IMOperation();
+		String s = "[" + page + "]";
+		op.addImage(f.getAbsolutePath()+s);
+		op.addImage(f2.getAbsolutePath());
+		//System.out.println("IMAGE: "+op);
+		cmd.run(op);
+		return f2;
+    }
+    
+    public boolean preProcessBitstream(Context c, Item item, Bitstream source)
+            throws Exception
+    {
+    	String nsrc = source.getName();
+    	for(Bundle b: item.getBundles("THUMBNAIL")) {
+    		for(Bitstream bit: b.getBitstreams()) {
+    			String n = bit.getName();
+    			if (n != null) {
+    				if (nsrc != null) {
+    					if (!n.startsWith(nsrc)) continue;
+    				}
+    			}
+     	    	String description = bit.getDescription();
+    	    	//If anything other than a generated thumbnail is found, halt processing
+    	    	if (description != null) {
+        	    	if (description.endsWith("Generated Thumbnail")) continue;
+        	    	if (description.endsWith("LIT Thumbnail")) continue;    	    		
+    	    	}
+    			System.out.println("Custom Thumbnail exists for " + nsrc + " for item " + item.getHandle() + ".  Thumbnail will not be generated. ");
+    			return false;
+    		}
+    	}
+    	
+    	
+        return true; //assume that the thumbnail is a custom one
+    }
+
+     public String[] getInputMIMETypes()
+    {
+        return ImageIO.getReaderMIMETypes();
+    }
+
+    public String[] getInputDescriptions()
+    {
+        return null;
+    }
+
+    public String[] getInputExtensions()
+    {
+        // Temporarily disabled as JDK 1.6 only
+        // return ImageIO.getReaderFileSuffixes();
+        return null;
+    }
+}

dspace-api/src/main/java/org/dspace/app/mediafilter/LitImageThumbnailFilter.java

@@ -0,0 +1,31 @@
+package org.dspace.app.mediafilter;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.InputStream;
+
+
+/**
+ * Filter image bitstreams, scaling the image to be within the bounds of
+ * thumbnail.maxwidth, thumbnail.maxheight, the size we want our thumbnail to be
+ * no bigger than. Creates only JPEGs.
+ */
+public class LitImageThumbnailFilter extends LitImageMagickThumbnailFilter 
+{
+
+    /**
+     * @param source
+     *            source input stream
+     * 
+     * @return InputStream the resulting input stream
+     */
+    public InputStream getDestinationStream(InputStream source)
+            throws Exception
+    {
+		File f = inputStreamToTempFile(source, "litthumb", ".tmp");
+    	File f2 = getThumbnailFile(f);
+		return new FileInputStream(f2);
+    }
+
+
+}

dspace-api/src/main/java/org/dspace/app/mediafilter/LitPdfThumbnailFilter.java

@@ -0,0 +1,23 @@
+package org.dspace.app.mediafilter;
+
+import java.io.File;
+import java.io.FileInputStream;
+import java.io.InputStream;
+
+public class LitPdfThumbnailFilter extends LitImageMagickThumbnailFilter {
+   public InputStream getDestinationStream(InputStream source)
+        throws Exception
+    {
+		File f = inputStreamToTempFile(source, "litpdfthumb", ".pdf");
+		File f2 = getImageFile(f, 0);
+    	File f3 = getThumbnailFile(f2);
+    	return new FileInputStream(f3);
+    }
+
+   public static final String[] PDF = {"Adobe PDF"};
+   public String[] getInputMIMETypes()
+  {
+      return PDF;
+  }
+
+}