1 /*

     2  * Copyright (c) 1997, 2013, Oracle and/or its affiliates. All rights reserved.

     3  * DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.

     4  *

     5  * This code is free software; you can redistribute it and/or modify it

     6  * under the terms of the GNU General Public License version 2 only, as

     7  * published by the Free Software Foundation.  Oracle designates this

     8  * particular file as subject to the "Classpath" exception as provided

     9  * by Oracle in the LICENSE file that accompanied this code.

    10  *

    11  * This code is distributed in the hope that it will be useful, but WITHOUT

    12  * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or

    13  * FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License

    14  * version 2 for more details (a copy is included in the LICENSE file that

    15  * accompanied this code).

    16  *

    17  * You should have received a copy of the GNU General Public License version

    18  * 2 along with this work; if not, write to the Free Software Foundation,

    19  * Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.

    20  *

    21  * Please contact Oracle, 500 Oracle Parkway, Redwood Shores, CA 94065 USA

    22  * or visit www.oracle.com if you need additional information or have any

    23  * questions.

    24  */

    25 

    26 package com.sun.xml.internal.messaging.saaj.soap;

    27 

    28 import java.awt.*;

    29 import java.awt.datatransfer.DataFlavor;

    30 import java.awt.image.BufferedImage;

    31 import java.io.*;

    32 

    33 import javax.activation.*;

    34 

    35 //import com.sun.image.codec.jpeg.*;

    36 import javax.imageio.ImageIO;

    37 

    38 /**

    39  * JAF data handler for Jpeg content

    40  *

    41  * @author Ana Lindstrom-Tamer

    42  */

    43 

    44 public class JpegDataContentHandler

    45     extends Component

    46     implements DataContentHandler {

    47     public static final String STR_SRC = "java.awt.Image";

    48 

    49     /**

    50      * return the DataFlavors for this <code>DataContentHandler</code>

    51      * @return The DataFlavors.

    52      */

    53     public DataFlavor[] getTransferDataFlavors() { // throws Exception;

    54         DataFlavor flavors[] = new DataFlavor[1];

    55 

    56         try {

    57             flavors[0] =

    58                 new ActivationDataFlavor(

    59                     Class.forName(STR_SRC),

    60                     "image/jpeg",

    61                     "JPEG");

    62         } catch (Exception e) {

    63             System.out.println(e);

    64         }

    65 

    66         return flavors;

    67     }

    68 

    69     /**

    70      * return the Transfer Data of type DataFlavor from InputStream

    71      * @param df The DataFlavor.

    72      * @param ins The InputStream corresponding to the data.

    73      * @return The constructed Object.

    74      */

    75     public Object getTransferData(DataFlavor df, DataSource ds) {

    76 

    77         // this is sort of hacky, but will work for the

    78         // sake of testing...

    79         if (df.getMimeType().startsWith("image/jpeg")) {

    80             if (df.getRepresentationClass().getName().equals(STR_SRC)) {

    81                 InputStream inputStream = null;

    82                 BufferedImage jpegLoadImage = null;

    83 

    84                 try {

    85                     inputStream = ds.getInputStream();

    86                     jpegLoadImage = ImageIO.read(inputStream);

    87 

    88                 } catch (Exception e) {

    89                     System.out.println(e);

    90                 }

    91 

    92                 return jpegLoadImage;

    93             }

    94         }

    95         return null;

    96     }

    97 

    98     /**

    99      *

   100      */

   101     public Object getContent(DataSource ds) { // throws Exception;

   102         InputStream inputStream = null;

   103         BufferedImage jpegLoadImage = null;

   104 

   105         try {

   106             inputStream = ds.getInputStream();

   107             jpegLoadImage = ImageIO.read(inputStream);

   108 

   109         } catch (Exception e) {

   110         }

   111 

   112         return (Image) jpegLoadImage;

   113     }

   114 

   115     /**

   116      * construct an object from a byte stream

   117      * (similar semantically to previous method, we are deciding

   118      *  which one to support)

   119      */

   120     public void writeTo(Object obj, String mimeType, OutputStream os)

   121         throws IOException {

   122         if (!mimeType.equals("image/jpeg"))

   123             throw new IOException(

   124                 "Invalid content type \""

   125                     + mimeType

   126                     + "\" for ImageContentHandler");

   127 

   128         if (obj == null) {

   129             throw new IOException("Null object for ImageContentHandler");

   130         }

   131 

   132         try {

   133             BufferedImage bufImage = null;

   134             if (obj instanceof BufferedImage) {

   135                 bufImage = (BufferedImage) obj;

   136 

   137             } else {

   138                 Image img = (Image) obj;

   139                 MediaTracker tracker = new MediaTracker(this);

   140                 tracker.addImage(img, 0);

   141                 tracker.waitForAll();

   142                 if (tracker.isErrorAny()) {

   143                         throw new IOException("Error while loading image");

   144                 }

   145                 bufImage =

   146                     new BufferedImage(

   147                         img.getWidth(null),

   148                         img.getHeight(null),

   149                         BufferedImage.TYPE_INT_RGB);

   150 

   151                 Graphics g = bufImage.createGraphics();

   152                 g.drawImage(img, 0, 0, null);

   153             }

   154             ImageIO.write(bufImage, "jpeg", os);

   155 

   156         } catch (Exception ex) {

   157             throw new IOException(

   158                 "Unable to run the JPEG Encoder on a stream "

   159                     + ex.getMessage());

   160         }

   161     }

   162 }