1 files changed, 156 insertions, 0 deletions
diff --git a/src/main/java/org/pdfbox/pdmodel/graphics/xobject/PDJpeg.java b/src/main/java/org/pdfbox/pdmodel/graphics/xobject/PDJpeg.java
new file mode 100644
index 0000000..79ee8e8
--- /dev/null
+++ b/src/main/java/org/pdfbox/pdmodel/graphics/xobject/PDJpeg.java
@@ -0,0 +1,156 @@
+/**
+ * Copyright (c) 2004, www.pdfbox.org
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions are met:
+ *
+ * 1. Redistributions of source code must retain the above copyright notice,
+ *    this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright notice,
+ *    this list of conditions and the following disclaimer in the documentation
+ *    and/or other materials provided with the distribution.
+ * 3. Neither the name of pdfbox; nor the names of its
+ *    contributors may be used to endorse or promote products derived from this
+ *    software without specific prior written permission.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
+ * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
+ * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE
+ * DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE FOR ANY
+ * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
+ * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
+ * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON
+ * ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
+ * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ *
+ * http://www.pdfbox.org
+ *
+ */
+package org.pdfbox.pdmodel.graphics.xobject;
+
+import java.awt.image.BufferedImage;
+import java.io.InputStream;
+import java.io.IOException;
+import java.io.OutputStream;
+
+import java.util.ArrayList;
+import java.util.List;
+
+import javax.imageio.ImageIO;
+
+import org.pdfbox.cos.COSDictionary;
+import org.pdfbox.cos.COSName;
+
+import org.pdfbox.pdmodel.PDDocument;
+import org.pdfbox.pdmodel.common.PDStream;
+import org.pdfbox.pdmodel.graphics.color.PDDeviceRGB;
+
+/**
+ * An image class for JPegs. 
+ * 
+ * @author mathiak
+ * @version $Revision: 1.4 $
+ */
+public class PDJpeg extends PDXObjectImage 
+{
+    
+    private static final List DCT_FILTERS = new ArrayList();
+    
+    static
+    {
+        DCT_FILTERS.add( COSName.DCT_DECODE.getName() );
+        DCT_FILTERS.add( COSName.DCT_DECODE_ABBREVIATION.getName() );
+    }
+    
+    /**
+     * Standard constructor.
+     * 
+     * @param jpeg The COSStream from which to extract the JPeg
+     */
+    public PDJpeg(PDStream jpeg) 
+    {
+        super(jpeg, "jpg");
+    }
+    
+    /**
+     * Construct from a stream.
+     * 
+     * @param doc The document to create the image as part of.
+     * @param is The stream that contains the jpeg data.
+     * @throws IOException If there is an error reading the jpeg data.
+     */
+    public PDJpeg( PDDocument doc, InputStream is ) throws IOException
+    {
+        super( new PDStream( doc, is, true ), "jpg" );
+        COSDictionary dic = getCOSStream();
+        dic.setItem( COSName.FILTER, COSName.DCT_DECODE );
+        dic.setItem( COSName.SUBTYPE, COSName.IMAGE);
+        dic.setItem( COSName.TYPE, COSName.getPDFName( "XObject" ) );
+        
+        BufferedImage image = getRGBImage();
+        setBitsPerComponent( 8 );
+        setColorSpace( PDDeviceRGB.INSTANCE );
+        setHeight( image.getHeight() );
+        setWidth( image.getWidth() );
+        
+    }
+    
+    /**
+     * Construct from a buffered image.
+     * 
+     * @param doc The document to create the image as part of.
+     * @param bi The image to convert to a jpeg
+     * @throws IOException If there is an error processing the jpeg data.
+     */
+    public PDJpeg( PDDocument doc, BufferedImage bi ) throws IOException
+    {
+        super( new PDStream( doc ), "jpg" );
+        
+        java.io.OutputStream os = getCOSStream().createFilteredStream();
+        try 
+        {
+            
+            ImageIO.write(bi,"jpeg",os);
+            
+            COSDictionary dic = getCOSStream();
+            dic.setItem( COSName.FILTER, COSName.DCT_DECODE );
+            dic.setItem( COSName.SUBTYPE, COSName.IMAGE);
+            dic.setItem( COSName.TYPE, COSName.getPDFName( "XObject" ) );
+        
+            setBitsPerComponent( 8 );
+            setColorSpace( PDDeviceRGB.INSTANCE );
+            setHeight( bi.getHeight() );
+            setWidth( bi.getWidth() );
+        }
+        finally 
+        {
+            os.close();
+        }
+    }
+    
+    /**
+     * Returns an image of the JPeg, or null if JPegs are not supported. (They should be. )
+     * @see org.pdfbox.pdmodel.graphics.xobject.PDXObjectImage#getRGBImage()
+     */
+    public BufferedImage getRGBImage() throws IOException
+    {
+        return ImageIO.read(getPDStream().getPartiallyFilteredStream( DCT_FILTERS ));
+    }
+    
+    /**
+     * This writes the JPeg to out. 
+     * @see org.pdfbox.pdmodel.graphics.xobject.PDXObjectImage#write2OutputStream(java.io.OutputStream)
+     */
+    public void write2OutputStream(OutputStream out) throws IOException
+    {
+        InputStream data = getPDStream().getPartiallyFilteredStream( DCT_FILTERS );
+        byte[] buf = new byte[1024];
+        int amountRead = -1;
+        while( (amountRead = data.read( buf )) != -1 )
+        {
+            out.write( buf, 0, amountRead );
+        }
+    }
+}