/*

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

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

 *

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

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

 * published by the Free Software Foundation.  Oracle designates this

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

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

 *

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

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

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

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

 * accompanied this code).

 *

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

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

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

 *

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

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

 * questions.

 */

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

import java.awt.*;

import java.awt.datatransfer.DataFlavor;

import java.awt.image.BufferedImage;

import java.io.*;

import java.util.Arrays;

import java.util.Iterator;

import java.util.logging.Level;

import java.util.logging.Logger;

import javax.activation.*;

import javax.imageio.ImageIO;

import javax.imageio.ImageWriter;

import javax.imageio.stream.ImageOutputStream;

import com.sun.xml.internal.messaging.saaj.util.LogDomainConstants;

public class ImageDataContentHandler extends Component

    implements DataContentHandler {

    protected static final Logger log =

        Logger.getLogger(LogDomainConstants.SOAP_DOMAIN,

                         "com.sun.xml.internal.messaging.saaj.soap.LocalStrings");

    private DataFlavor[] flavor;

    public ImageDataContentHandler() {

        String[] mimeTypes = ImageIO.getReaderMIMETypes();

        flavor = new DataFlavor[mimeTypes.length];

        for(int i=0; i < mimeTypes.length; i++) {

            flavor[i] = new ActivationDataFlavor(

                java.awt.Image.class, mimeTypes[i], "Image");

        }

    }

    /**

     * Returns an array of DataFlavor objects indicating the flavors the

     * data can be provided in. The array should be ordered according to

     * preference for providing the data (from most richly descriptive to

     * least descriptive).

     *

     * @return The DataFlavors.

     */

    public DataFlavor[] getTransferDataFlavors() {

        return (DataFlavor[]) Arrays.copyOf(flavor, flavor.length);

    }

    /**

     * Returns an object which represents the data to be transferred.

     * The class of the object returned is defined by the representation class

     * of the flavor.

     *

     * @param df The DataFlavor representing the requested type.

     * @param ds The DataSource representing the data to be converted.

     * @return The constructed Object.

     */

    public Object getTransferData(DataFlavor df, DataSource ds)

        throws IOException {

        for (int i=0; i < flavor.length; i++) {

            if (flavor[i].equals(df)) {

                return getContent(ds);

            }

        }

        return null;

    }

    /**

     * Return an object representing the data in its most preferred form.

     * Generally this will be the form described by the first DataFlavor

     * returned by the <code>getTransferDataFlavors</code> method.

     *

     * @param ds The DataSource representing the data to be converted.

     * @return The constructed Object.

     */

    public Object getContent(DataSource ds) throws IOException {

        return ImageIO.read(new BufferedInputStream(ds.getInputStream()));

    }

    /**

     * Convert the object to a byte stream of the specified MIME type

     * and write it to the output stream.

     *

     * @param obj   The object to be converted.

     * @param mimeType  The requested MIME type of the resulting byte stream.

     * @param os    The output stream into which to write the converted

     *          byte stream.

     */

    public void writeTo(Object obj, String type, OutputStream os)

        throws IOException {

        try {

            BufferedImage bufImage = null;

            if (obj instanceof BufferedImage) {

                bufImage = (BufferedImage)obj;

            } else if (obj instanceof Image) {

                bufImage = render((Image)obj);

            } else {

                log.log(Level.SEVERE,

                    "SAAJ0520.soap.invalid.obj.type",

                    new String[] { obj.getClass().toString() });

                throw new IOException(

                    "ImageDataContentHandler requires Image object, "

                    + "was given object of type "

                    + obj.getClass().toString());

            }

            ImageWriter writer = null;

            Iterator i = ImageIO.getImageWritersByMIMEType(type);

            if (i.hasNext()) {

                writer = (ImageWriter)i.next();

            }

            if (writer != null) {

                ImageOutputStream stream = null;

                stream = ImageIO.createImageOutputStream(os);

                writer.setOutput(stream);

                writer.write(bufImage);

                writer.dispose();

                stream.close();

            } else {

                log.log(Level.SEVERE, "SAAJ0526.soap.unsupported.mime.type",

                    new String[] { type });

                throw new IOException("Unsupported mime type:"+ type);

            }

        } catch (Exception e) {

            log.severe("SAAJ0525.soap.cannot.encode.img");

            throw new IOException("Unable to encode the image to a stream "

                + e.getMessage());

        }

    }

    private BufferedImage render(Image img) throws InterruptedException {

        MediaTracker tracker = new MediaTracker(this);

        tracker.addImage(img, 0);

        tracker.waitForAll();

        BufferedImage bufImage = new BufferedImage(img.getWidth(null),

            img.getHeight(null), BufferedImage.TYPE_INT_RGB);

        Graphics g = bufImage.createGraphics();

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

        g.dispose();

        return bufImage;

    }

}