Application: gvSIG desktop » gvSIG data provider for PostgreSQL

package org.gvsig.postgresql.dal.expressionbuilderformatter;

import java.text.MessageFormat;

import java.util.List;

import org.apache.commons.lang3.StringUtils;

import org.gvsig.expressionevaluator.ExpressionBuilder;

import org.gvsig.expressionevaluator.ExpressionBuilder.Function;

import org.gvsig.expressionevaluator.ExpressionBuilder.Value;

import org.gvsig.expressionevaluator.Formatter;

import org.gvsig.fmap.dal.SQLBuilder;

import org.gvsig.fmap.dal.feature.FeatureType;

import static org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase.PROP_FEATURE_TYPE;

/**

 *

 * @author jjdelcerro

 */

public class ST_GeomFromWKB implements Formatter<Value> {

    private final SQLBuilder sqlbuilder;

    private final Formatter<Value> formatter;

    public ST_GeomFromWKB(SQLBuilder sqlbuilder, Formatter<Value> formatter) {

        this.sqlbuilder = sqlbuilder;

        this.formatter = formatter;

    }

    @Override

    public boolean canApply(ExpressionBuilder.Value value) {

        if (value instanceof ExpressionBuilder.Function) {

            return StringUtils.equalsIgnoreCase(ExpressionBuilder.FUNCTION_ST_GEOMFROMWKB, ((Function) value).name());

        }

        return false;

    }

    @Override

    public String format(Value function) {

        String r;

        String wkb;

        String srid;

        List<Value> parameters = ((Function) function).parameters();

        switch(parameters.size()) {

            case 1:

                wkb = parameters.get(0).toString(formatter);

                r = MessageFormat.format("ST_GeomFromWKB({0})", wkb);

                break;

            case 2:

                wkb = parameters.get(0).toString(formatter);

                srid = parameters.get(1).toString(formatter);

                if( StringUtils.equals("0", srid) ) {

                    r = MessageFormat.format("ST_GeomFromWKB({0})", wkb);

                } else {

                    FeatureType ftype = (FeatureType) function.getProperty(PROP_FEATURE_TYPE);

                    if( ftype == null || ftype.getDefaultSRS()!=null ) {

                        r = MessageFormat.format("ST_GeomFromWKB({0},{1})", wkb, srid);

                    } else {

                        r = MessageFormat.format("ST_GeomFromWKB({0})", wkb);

                    }

                }

                break;

            default:

                throw new IllegalArgumentException("Wrong number of arguments ("+parameters.size()+") in function ST_GeomFromWKB; requiered 1 or 2 arguments.");

        }

        return r;

    }

}