/*
 *    Geotoolkit - An Open Source Java GIS Toolkit
 *    http://www.geotoolkit.org
 *
 *    (C) 2012, Geomatys
 *
 *    This library is free software; you can redistribute it and/or
 *    modify it under the terms of the GNU Lesser General Public
 *    License as published by the Free Software Foundation; either
 *    version 2.1 of the License, or (at your option) any later version.
 *
 *    This library 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
 *    Lesser General Public License for more details.
 */
package org.geotoolkit.wps.converters.inputs.complex;

import java.awt.image.BufferedImage;
import java.awt.image.RenderedImage;
import java.io.ByteArrayInputStream;
import java.io.IOException;
import java.io.InputStream;
import java.util.List;
import java.util.Map;
import javax.imageio.ImageIO;
import net.iharder.Base64;
import org.apache.sis.util.UnconvertibleObjectException;
import org.geotoolkit.wps.io.WPSEncoding;
import org.geotoolkit.wps.xml.ComplexDataType;

/**
 * Convert an base64 encoded coverage into a RenderedImage.
 *
 * @author Quentin Boileau (Geomatys)
 */
public class ComplexToRendredImageConverter extends AbstractComplexInputConverter<RenderedImage> {

    private static ComplexToRendredImageConverter INSTANCE;

    private ComplexToRendredImageConverter(){
    }

    public static synchronized ComplexToRendredImageConverter getInstance(){
        if(INSTANCE == null){
            INSTANCE = new ComplexToRendredImageConverter();
        }
        return INSTANCE;
    }

    @Override
    public Class<RenderedImage> getTargetClass() {
        return RenderedImage.class;
    }

    @Override
    public RenderedImage convert(ComplexDataType source, Map<String, Object> params) throws UnconvertibleObjectException {

        try {
            if (params.get(ENCODING).equals(WPSEncoding.BASE64.getValue())) {
                final List<Object> data = source.getContent();
                if (data.size() != 1) {
                    throw new UnconvertibleObjectException("Only one object in Complex content.");
                }
                final String encodedImage = (String) data.get(0);
                final byte[] byteData = Base64.decode(encodedImage.trim());
                if (byteData != null && byteData.length > 0) {
                    final InputStream is = new ByteArrayInputStream(byteData);
                    if (is != null) {
                        final BufferedImage outImg = ImageIO.read(is);
                        return outImg;
                    }
                }
                throw new UnconvertibleObjectException("Error during base64 decoding.");
            } else {
                throw new UnconvertibleObjectException("Encoding should be in \"base64\"");
            }
        } catch (IOException ex) {
            throw new UnconvertibleObjectException(ex.getMessage(), ex);
        }
    }
}