GeometryFunctionMethods.java example

Explorer
teiid-designer-master
/*
 * JBoss, Home of Professional Open Source.
 * See the COPYRIGHT.txt file distributed with this work for information
 * regarding copyright ownership.  Some portions may be licensed
 * to Red Hat, Inc. under one or more contributor license agreements.
 *
 * 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.
 *
 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
 * 02110-1301 USA.
 */

package org.teiid.query.function;

import java.sql.Blob;
import java.sql.SQLException;

import org.teiid.core.types.BlobType;
import org.teiid.core.types.ClobType;
import org.teiid.core.types.GeometryType;
import org.teiid.designer.annotation.Updated;
import org.teiid.designer.runtime.version.spi.TeiidServerVersion.Version;
import org.teiid.metadata.FunctionMethod.PushDown;
import org.teiid.query.function.metadata.FunctionCategoryConstants;
import org.teiid.query.util.CommandContext;
import org.teiid.translator.SourceSystemFunctions;


public class GeometryFunctionMethods {

    		@TeiidFunction(name=SourceSystemFunctions.ST_ASTEXT, 
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static ClobType asText(GeometryType geometry) 
     throws Exception {
return GeometryUtils.geometryToClob(geometry, false);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_ASEWKT,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static ClobType asEwkt(GeometryType geometry)
     throws Exception {
 return GeometryUtils.geometryToClob(geometry, true);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_ASBINARY,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static BlobType asBlob(GeometryType geometry) {
 Blob b = geometry.getReference();
 return new BlobType(b);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_ASEWKB,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static BlobType asEwkb(final GeometryType geometry) {
	return GeometryUtils.geometryToEwkb(geometry);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_ASGEOJSON,
            category=FunctionCategoryConstants.GEOMETRY,
 		   pushdown=PushDown.CAN_PUSHDOWN,
            nullOnNull=true)
public static ClobType asGeoJson(GeometryType geometry) 
     throws Exception {
 return GeometryUtils.geometryToGeoJson(geometry);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_ASGML,
            category=FunctionCategoryConstants.GEOMETRY,
 		   pushdown=PushDown.CAN_PUSHDOWN,
            nullOnNull=true)
public static ClobType asGml(CommandContext context, GeometryType geometry) 
     throws Exception {
 return GeometryUtils.geometryToGml(context, geometry, true);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_ASKML,
            category=FunctionCategoryConstants.GEOMETRY,
            pushdown=PushDown.CAN_PUSHDOWN,
            nullOnNull=true)
public static ClobType asKml(CommandContext context, GeometryType geometry) 
     throws Exception {
 return GeometryUtils.geometryToGml(context, geometry, false);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMTEXT,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true)
public static GeometryType geomFromText(ClobType wkt) 
     throws Exception {
	return GeometryUtils.geometryFromClob(wkt);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMTEXT,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static GeometryType geomFromText(ClobType wkt, int srid) 
     throws Exception {
	return GeometryUtils.geometryFromClob(wkt, srid, false);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMWKB,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            alias="ST_GEOMFROMBINARY")
public static GeometryType geoFromBlob(BlobType wkb) 
     throws Exception {
	return GeometryUtils.geometryFromBlob(wkb);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMWKB,
            category=FunctionCategoryConstants.GEOMETRY,
            pushdown=PushDown.CAN_PUSHDOWN,
 		   nullOnNull=true,
            alias="ST_GEOMFROMBINARY")
public static GeometryType geoFromBlob(BlobType wkb, int srid) 
     throws Exception {
	return GeometryUtils.geometryFromBlob(wkb, srid);
}
 
			@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMGEOJSON,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true)
public static GeometryType geomFromGeoJson(ClobType clob) 
     throws Exception {
 return GeometryUtils.geometryFromGeoJson(clob);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMGEOJSON,
            category=FunctionCategoryConstants.GEOMETRY,
 		   pushdown=PushDown.CAN_PUSHDOWN,
            nullOnNull=true)
public static GeometryType geomFromGeoJson(ClobType clob, int srid) 
     throws Exception {
 return GeometryUtils.geometryFromGeoJson(clob, srid);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMGML,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true)
public static GeometryType geomFromGml(ClobType gml) 
     throws Exception {
 return GeometryUtils.geometryFromGml(gml, null);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMGML,
            category=FunctionCategoryConstants.GEOMETRY,
 		   pushdown=PushDown.CAN_PUSHDOWN,
            nullOnNull=true)
public static GeometryType geomFromGml(ClobType gml, int srid) 
     throws Exception {
 return GeometryUtils.geometryFromGml(gml, srid);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_INTERSECTS,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean intersects(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.intersects(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_CONTAINS,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean contains(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.contains(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_CROSSES,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean crosses(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.crosses(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_DISJOINT,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean disjoint(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.disjoint(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_DISTANCE,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static Double distance(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.distance(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_OVERLAPS,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean overlaps(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.overlaps(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_TOUCHES,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean touches(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.touches(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_SRID,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static int getSrid(GeometryType geom1) {
 return geom1.getSrid();
}

			@TeiidFunction(name=SourceSystemFunctions.ST_SETSRID,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static GeometryType setSrid(GeometryType geom1, int srid) {
	GeometryType gt = new GeometryType();
	gt.setReference(geom1.getReference());
	gt.setSrid(srid);
 return gt;
}

			@TeiidFunction(name=SourceSystemFunctions.ST_EQUALS,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean equals(GeometryType geom1, GeometryType geom2) throws Exception {
 return GeometryUtils.equals(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_TRANSFORM,
            category=FunctionCategoryConstants.GEOMETRY,
            nullOnNull=true,
            pushdown=PushDown.CAN_PUSHDOWN)
public static GeometryType transform(CommandContext context,
                                  GeometryType geom, 
                                  int srid) 
     throws Exception {        
 return GeometryTransformUtils.transform(context, geom, srid);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_ENVELOPE,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static GeometryType envelope(GeometryType geom) throws Exception {
	return GeometryUtils.envelope(geom);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_WITHIN,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean within(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.within(geom1, geom2);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_DWITHIN,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static Boolean dwithin(GeometryType geom1, GeometryType geom2, double distance) throws Exception {
	return GeometryUtils.dwithin(geom1, geom2, distance);
}

/*
* PostGIS compatibility
*/

			@TeiidFunction(name=SourceSystemFunctions.ST_SIMPLIFY,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static GeometryType simplify(GeometryType geom, 
                               double tolerance)
                             		  throws Exception {        
	return GeometryUtils.simplify(geom, tolerance);
}

			@TeiidFunction(name=SourceSystemFunctions.ST_FORCE_2D,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static GeometryType force2D(GeometryType geom) {
	return geom; //higher dimensions not supported
}

			@TeiidFunction(name=SourceSystemFunctions.ST_HASARC,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static boolean hasArc(@SuppressWarnings("unused") GeometryType geom) {
	return false; //curved not supported
}

			@TeiidFunction(name=SourceSystemFunctions.ST_ASEWKB,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static boolean boundingBoxIntersects(GeometryType geom1, GeometryType geom2) throws Exception {
	return GeometryUtils.boundingBoxIntersects(geom1, geom2);
}

				@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMEWKT,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static GeometryType geomFromEwkt(ClobType ewkt) 
  throws Exception {
	return GeometryUtils.geometryFromClob(ewkt, null, true);
}

				@TeiidFunction(name=SourceSystemFunctions.ST_GEOMFROMEWKB,
     category=FunctionCategoryConstants.GEOMETRY,
     nullOnNull=true,
     pushdown=PushDown.CAN_PUSHDOWN)
public static GeometryType geomFromEwkb(BlobType ewkb) 
  throws Exception, SQLException {
	return GeometryUtils.geometryFromEwkb(ewkb.getBinaryStream(), null);
}
}