JPEGImageEncoder.java
上传用户:btjssb159
上传日期:2018-01-04
资源大小:241k
文件大小:8k
源码类别:

DNA

开发平台:

Java

  1. /*
  2.  * Copyright (c) 2001 Sun Microsystems, Inc. All Rights Reserved.
  3.  *
  4.  * Redistribution and use in source and binary forms, with or without 
  5.  * modification, are permitted provided that the following conditions are met:
  6.  * 
  7.  * -Redistributions of source code must retain the above copyright notice, this 
  8.  * list of conditions and the following disclaimer.
  9.  *
  10.  * -Redistribution in binary form must reproduct the above copyright notice,
  11.  * this list of conditions and the following disclaimer in the documentation
  12.  * and/or other materials provided with the distribution.
  13.  * 
  14.  * Neither the name of Sun Microsystems, Inc. or the names of contributors may
  15.  * be used to endorse or promote products derived from this software without
  16.  * specific prior written permission.
  17.  * 
  18.  * This software is provided "AS IS," without a warranty of any kind. ALL
  19.  * EXPRESS OR IMPLIED CONDITIONS, REPRESENTATIONS AND WARRANTIES, INCLUDING ANY
  20.  * IMPLIED WARRANTY OF MERCHANTABILITY, FITNESS FOR A PARTICULAR PURPOSE OR
  21.  * NON-INFRINGEMENT, ARE HEREBY EXCLUDED. SUN AND ITS LICENSORS SHALL NOT BE
  22.  * LIABLE FOR ANY DAMAGES SUFFERED BY LICENSEE AS A RESULT OF USING, MODIFYING
  23.  * OR DISTRIBUTING THE SOFTWARE OR ITS DERIVATIVES. IN NO EVENT WILL SUN OR ITS
  24.  * LICENSORS BE LIABLE FOR ANY LOST REVENUE, PROFIT OR DATA, OR FOR DIRECT,
  25.  * INDIRECT, SPECIAL, CONSEQUENTIAL, INCIDENTAL OR PUNITIVE DAMAGES, HOWEVER
  26.  * CAUSED AND REGARDLESS OF THE THEORY OF LIABILITY, ARISING OUT OF THE USE OF
  27.  * OR INABILITY TO USE SOFTWARE, EVEN IF SUN HAS BEEN ADVISED OF THE
  28.  * POSSIBILITY OF SUCH DAMAGES.
  29.  * 
  30.  * You acknowledge that Software is not designed,licensed or intended for use in 
  31.  * the design, construction, operation or maintenance of any nuclear facility.
  32.  */
  33. import java.awt.Point;
  34. import java.awt.color.ColorSpace;
  35. import java.awt.image.BufferedImage;
  36. import java.awt.image.ColorModel;
  37. import java.awt.image.IndexColorModel;
  38. import java.awt.image.DataBuffer;
  39. import java.awt.image.WritableRaster;
  40. import java.awt.image.RenderedImage;
  41. import java.awt.image.SampleModel;
  42. import java.io.IOException;
  43. import java.io.OutputStream;
  44. //
  45. // Need these classes since we are currently using the 
  46. // Java2D JpegEncoder for our Jpeg Implementation.
  47. //
  48. import com.sun.image.codec.jpeg.JPEGQTable;
  49. import com.sun.image.codec.jpeg.JPEGDecodeParam;
  50. /**
  51.  * An ImageEncoder for the JPEG (JFIF) file format.
  52.  *
  53.  * The common cases of single band grayscale and three or four band RGB images
  54.  * are handled so as to minimize the amount of information required of the
  55.  * programmer. See the comments pertaining to the constructor and the
  56.  * <code>writeToStream()</code> method for more detailed information.
  57.  *
  58.  */
  59. public class JPEGImageEncoder extends ImageEncoderImpl {
  60.     private JPEGEncodeParam jaiEP = null;
  61.     public JPEGImageEncoder(OutputStream output,
  62.                             ImageEncodeParam param) {
  63.         super(output, param);
  64.         if (param != null) {
  65.             jaiEP = (JPEGEncodeParam)param;
  66.         }
  67.     }
  68.     //
  69.     // Go through the settable encoding parameters and see 
  70.     // if any of them have been set. If so, transfer then to the 
  71.     // com.sun.image.codec.jpeg.JPEGEncodeParam object.
  72.     //
  73.     static void modifyEncodeParam(JPEGEncodeParam jaiEP,
  74.           com.sun.image.codec.jpeg.JPEGEncodeParam j2dEP,
  75.                                                int nbands) {
  76.         int val;
  77.         int[] qTab;
  78.         for(int i=0; i<nbands; i++) {
  79.             //
  80.             // If subsampling factors were set, apply them
  81.             //
  82.             val = jaiEP.getHorizontalSubsampling(i);
  83.             j2dEP.setHorizontalSubsampling(i, val);
  84.             val = jaiEP.getVerticalSubsampling(i);
  85.             j2dEP.setVerticalSubsampling(i, val);
  86.             //
  87.             // If new Q factors were supplied, apply them
  88.             //
  89.             if (jaiEP.isQTableSet(i)) {
  90.                 qTab = jaiEP.getQTable(i);
  91.                 val = jaiEP.getQTableSlot(i);
  92.                 j2dEP.setQTableComponentMapping(i, val);
  93.                 j2dEP.setQTable(val, new JPEGQTable(qTab));
  94.             }
  95.         }
  96.         // Apply new quality, if set
  97.         if (jaiEP.isQualitySet()) {
  98.             float fval = jaiEP.getQuality();
  99.             j2dEP.setQuality(fval, true);
  100.         }
  101.         // Apply new restart interval, if set
  102.         val = jaiEP.getRestartInterval();
  103.         j2dEP.setRestartInterval(val);
  104.         // Write a tables-only abbreviated JPEG file
  105.         if (jaiEP.getWriteTablesOnly() == true) {
  106.             j2dEP.setImageInfoValid(false);
  107.             j2dEP.setTableInfoValid(true);
  108.         }
  109.         // Write an image-only abbreviated JPEG file
  110.         if (jaiEP.getWriteImageOnly() == true) {
  111.             j2dEP.setTableInfoValid(false);
  112.             j2dEP.setImageInfoValid(true);
  113.         }
  114.         // Write the JFIF (APP0) marker
  115.         if (jaiEP.getWriteJFIFHeader() == false) {
  116.             j2dEP.setMarkerData(
  117.               com.sun.image.codec.jpeg.JPEGDecodeParam.APP0_MARKER, null);
  118.         }
  119.     }
  120.     /**
  121.      * Encodes a RenderedImage and writes the output to the
  122.      * OutputStream associated with this ImageEncoder.
  123.      */
  124.     public void encode(RenderedImage im) throws IOException {
  125.         //
  126.         // Check data type and band count compatibility.
  127.         // This implementation handles only 1 and 3 band source images.
  128.         //
  129.         SampleModel sampleModel = im.getSampleModel();
  130.         ColorModel  colorModel  = im.getColorModel();
  131.         // Must be a 1 or 3 component BYTE image
  132.         int numBands  = colorModel.getNumColorComponents();
  133.         int transType = sampleModel.getTransferType();
  134.         if ((transType != DataBuffer.TYPE_BYTE) ||
  135.             ((numBands != 1) && (numBands != 3) )) {
  136.             throw new RuntimeException(JaiI18N.getString("JPEGImageEncoder0"));
  137.         }
  138.         // Must be GRAY or RGB
  139.         int cspaceType = colorModel.getColorSpace().getType();
  140.         if (cspaceType != ColorSpace.TYPE_GRAY &&
  141.             cspaceType != ColorSpace.TYPE_RGB) {
  142.             throw new Error(JaiI18N.getString("JPEGImageEncoder1"));
  143.         }
  144.         //
  145.         // Create a BufferedImage to be encoded.
  146.         // The JPEG interfaces really need a whole image.
  147.         //
  148.         WritableRaster wRas;
  149.         BufferedImage bi;
  150.         //
  151.         // Get a contiguous raster. Jpeg compression can't work
  152.         // on tiled data in most cases.
  153.         // Also need to be sure that the raster doesn't have a
  154.         // non-zero origin, since BufferedImage won't accept that.
  155.         // (Bug ID 4253990)
  156.         //
  157.         wRas = (WritableRaster)im.getData();
  158.         if (wRas.getMinX() != 0 || wRas.getMinY() != 0) {
  159.             wRas = wRas.createWritableTranslatedChild(0, 0);
  160.         }
  161.         // First create the Java2D encodeParam based on the BufferedImage
  162.         com.sun.image.codec.jpeg.JPEGEncodeParam j2dEP = null;
  163.         if (colorModel instanceof IndexColorModel) {
  164.             //
  165.             // Need to expand the indexed data to components.
  166.             // The convertToIntDiscrete method is used to perform this.
  167.             //
  168.             IndexColorModel icm = (IndexColorModel)colorModel;
  169.             bi = icm.convertToIntDiscrete(wRas, false);
  170.             j2dEP = com.sun.image.codec.jpeg.JPEGCodec.getDefaultJPEGEncodeParam(bi);
  171.         } else {
  172.             bi = new BufferedImage(colorModel, wRas, false, null);
  173.             j2dEP = com.sun.image.codec.jpeg.JPEGCodec.getDefaultJPEGEncodeParam(bi);
  174.         }
  175.         // Now modify the Java2D encodeParam based on the options set
  176.         // in the JAI encodeParam object.
  177.         if (jaiEP != null) {
  178.             modifyEncodeParam(jaiEP, j2dEP, numBands);
  179.         }
  180.         // Now create the encoder with the modified Java2D encodeParam
  181.         com.sun.image.codec.jpeg.JPEGImageEncoder encoder;
  182.         encoder = com.sun.image.codec.jpeg.JPEGCodec.createJPEGEncoder(
  183.                     output, j2dEP);
  184.         try {
  185.           // Write the image data.
  186.             encoder.encode(bi);
  187.         } catch(IOException e) {
  188.             throw new RuntimeException(e.getMessage());
  189.         }
  190.     }
  191. }