/** * Copyright (C) 2009-2014 FoundationDB, LLC * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License as published by * the Free Software Foundation, either version 3 of the License, or * (at your option) any later version. * * This program 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 Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see <http://www.gnu.org/licenses/>. */ package com.foundationdb.server.types.common.funcs; import com.foundationdb.server.types.LazyList; import com.foundationdb.server.types.TClass; import com.foundationdb.server.types.TExecutionContext; import com.foundationdb.server.types.TOverloadResult; import com.foundationdb.server.types.texpressions.TInputSetBuilder; import com.foundationdb.server.types.texpressions.TScalarBase; import com.foundationdb.server.types.value.ValueSource; import com.foundationdb.server.types.value.ValueTarget; import com.vividsolutions.jts.geom.Geometry; /** Get the type string for a geometry object. Thin wrapper around {@link Geometry#getGeometryType}. */ public class GeoTypeString extends TScalarBase { private final TClass stringType; private final TClass geometryType; public GeoTypeString(TClass stringType, TClass geometryType) { this.stringType = stringType; this.geometryType = geometryType; } @Override protected void buildInputSets(TInputSetBuilder builder) { builder.covers(geometryType, 0); } @Override protected void doEvaluate(TExecutionContext context, LazyList<? extends ValueSource> inputs, ValueTarget output) { Geometry geometry = (Geometry)inputs.get(0).getObject(); String typeString = geometry.getGeometryType(); output.putString(typeString, null); } @Override public String displayName() { return "GEO_TYPE_STRING"; } @Override public TOverloadResult resultType() { return TOverloadResult.fixed(stringType, 32); } }