Application: gvSIG desktop

package org.gvsig.expressionevaluator.impl.function.image;

import java.io.File;

import java.io.IOException;

import java.io.OutputStream;

import java.util.Objects;

import java.util.logging.Level;

import java.util.logging.Logger;

import org.apache.commons.lang3.Range;

import org.gvsig.expressionevaluator.ExpressionRuntimeException;

import org.gvsig.expressionevaluator.Interpreter;

import org.gvsig.expressionevaluator.impl.function.programming.FileFunction;

import org.gvsig.expressionevaluator.spi.AbstractFunction;

public class ImageWriteFunction extends AbstractFunction {

    public ImageWriteFunction() {

        super(

                "Image", "IMAGEWRITE", Range.is(2),

                "Read an image from a data source and return it as a 'Java BufferedImage' object.",

                "IMAGEWRITE({{simpleimage}}, target, format)",

                new String[]{

                    "Image to write",

                    "Value to save the image in",

                    "Format name, PNG, JPEG,...",

                },

                "SimpleImage",

                false

        );

    }

    @Override

    public boolean allowConstantFolding() {

        return true;

    }

    @Override

    public Object call(Interpreter interpreter, Object[] args) {

        String formatName = this.getStr(args, 2);

        Object arg0 = this.getObject(args, 0);

        SimpleImage image = ImageReadFunction.createSimpleImage(arg0);

        Object target = this.getObject(args, 1);

        try {

            if( target instanceof OutputStream ) {

                image.save((OutputStream) target, formatName);

            } else {

              File f = FileFunction.toFile(target);

              image.save(f, formatName);

            }

        } catch (IOException ex) {

          throw new ExpressionRuntimeException("Can't write image to '"+Objects.toString(target)+" as format '"+formatName+".", ex);

        }

        return image;

    }

}