DBLineSegment2D.java

/*
 * Copyright 2015 gregory.graham.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *      http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package nz.co.gregs.dbvolution.datatypes.spatial2D;

import java.sql.ResultSet;
import java.sql.SQLException;
import nz.co.gregs.dbvolution.datatypes.QueryableDatatype;
import java.util.logging.Level;
import java.util.logging.Logger;
import nz.co.gregs.dbvolution.columns.LineSegment2DColumn;
import nz.co.gregs.dbvolution.databases.definitions.DBDefinition;
import nz.co.gregs.dbvolution.exceptions.IncorrectGeometryReturnedForDatatype;
import nz.co.gregs.dbvolution.exceptions.IncorrectRowProviderInstanceSuppliedException;
import nz.co.gregs.dbvolution.exceptions.ParsingSpatialValueException;
import nz.co.gregs.dbvolution.expressions.spatial2D.LineSegment2DExpression;
import nz.co.gregs.dbvolution.expressions.StringExpression;
import nz.co.gregs.dbvolution.query.RowDefinition;
import nz.co.gregs.dbvolution.results.LineSegment2DResult;
import com.vividsolutions.jts.geom.*;
import com.vividsolutions.jts.io.ParseException;
import java.util.Comparator;
import nz.co.gregs.dbvolution.utility.comparators.ComparableComparator;

/**
 * Represents datatypes and columns that are composed of a 2 points connected as
 * a line.
 *
 * <p>
 * Use this type if the database column stores a series of 2-dimensional (that
 * is X and Y points) that are contiguous and open.
 *
 * <p>
 * Alternatives to a DBLineSegment2D are a series of points constituting a line
 * {@link DBLine2D}, infinite lines (TODO), closed paths (TODO), and closed
 * paths defining a solid {@link DBPolygon2D}.
 *
 * <p>
 * Common datatypes covered by this type include LINESTRING.
 *
 * <p>
 * Spatial types are not automatically generated during schema extraction so you
 * may need to change some DBString fields.
 *
 * <p style="color: #F90;">Support DBvolution at
 * <a href="http://patreon.com/dbvolution" target=new>Patreon</a></p>
 *
 * @author gregorygraham
 */
public class DBLineSegment2D extends QueryableDatatype<LineSegment> implements LineSegment2DResult {

	private static final long serialVersionUID = 1L;

	/**
	 * Default constructor.
	 *
	 * Use this method to create the DBLine2D used in your DBRow subclass.
	 *
	 */
	public DBLineSegment2D() {
	}

	/**
	 * Create a DBLine2D with the value set to the {@link LineString} provided.
	 *
	 * <p>
	 * This is a convenient way to assign a constant value in an expression or
	 * DBRow subclass.
	 *
	 * @param lineSegment the value to apply
	 */
	public DBLineSegment2D(LineSegment lineSegment) {
		super(lineSegment);
	}

	/**
	 * Create a DBLine2D using the expression supplied.
	 *
	 * <p>
	 * Useful for defining expression columns in DBRow subclass that acquire their
	 * value from a transformation of data at query time.
	 *
	 * @param columnExpression the column expression to use to fill this field
	 */
	public DBLineSegment2D(nz.co.gregs.dbvolution.expressions.spatial2D.LineSegment2DExpression columnExpression) {
		super(columnExpression);
	}

	public DBLineSegment2D(Double point1x, Double point1y, Double point2x, Double point2y) {
		super(new LineSegment(point1x, point1y, point2x, point2y));

	}

	public DBLineSegment2D(Point point1, Point point2) {
		this(new LineSegment(point1.getCoordinate(), point2.getCoordinate()));
	}

	public DBLineSegment2D(Coordinate coord1, Coordinate coord2) {
		this(new LineSegment(coord1, coord2));
	}

	/**
	 * Set the value of this DBLine2D to the value provided.
	 *
	 * <p>
	 * Use this method to define the value of a field/column before inserting the
	 * DBRow subclass into the database.
	 *
	 * @param line the value to apply
	 */
	@Override
	public void setValue(LineSegment line) {
		setLiteralValue(line);
	}

	/**
	 * Set the value of this DBLine2D to the value provided.
	 *
	 * <p>
	 * The series of points will combined into a line for you.
	 *
	 * <p>
	 * Use this method to define the value of a field/column before inserting the
	 * DBRow subclass into the database.
	 *
	 * @param point1 the starting point of the line segment
	 * @param point2 the end point of the line segment
	 */
	public void setValue(Point point1, Point point2) {
		LineSegment line = new LineSegment(point1.getCoordinate(), point2.getCoordinate());
		setLiteralValue(line);
	}

	/**
	 * Set the value of this DBLine2D to the value provided.
	 *
	 * <p>
	 * The series of coordinates will combined into a line for you.
	 *
	 * <p>
	 * Use this method to define the value of a field/column before inserting the
	 * DBRow subclass into the database.
	 *
	 * @param coord1 the starting point of the line segment
	 * @param coord2 the end point of the line segment
	 */
	public void setValue(Coordinate coord1, Coordinate coord2) {
		LineSegment line = new LineSegment(coord1, coord2);
		setLiteralValue(line);
	}

	@Override
	public LineSegment getValue() {
		if (!isDefined() || isNull()) {
			return null;
		} else {
			return getLiteralValue();
		}
	}

	/**
	 * Transform the value of the DBLine2D into a
	 * {@link LineString JTS LineString}
	 *
	 * @return the value of this object if defined and not NULL, NULL otherwise.
	 */
	public LineSegment jtsLineSegmentValue() {
		final LineSegment value = getValue();
		value.normalize();
		return value;
	}

	@Override
	public void setFromResultSet(DBDefinition defn, ResultSet resultSet, String resultSetColumnName) throws SQLException {
		removeConstraints();
		if (resultSet == null || resultSetColumnName == null) {
			this.setToNull(defn);
		} else {
			LineSegment dbValue;
			try {
				dbValue = getFromResultSet(defn, resultSet, resultSetColumnName);
				if (resultSet.wasNull()) {
					dbValue = null;
				} else if (defn.requiresReversingLineStringsFromDatabase()) {
					dbValue.reverse();
				}
			} catch (SQLException ex) {
				// Probably means the column wasn't selected.
				dbValue = null;
			}
			if (dbValue == null) {
				this.setToNull(defn);
			} else {
				this.setLiteralValue(dbValue);
			}
		}
		setUnchanged();
		setDefined(true);
//		propertyWrapperDefn = null;
	}

	@Override
	public String getSQLDatatype() {
		return " LINESTRING ";
	}

	@Override
	protected String formatValueForSQLStatement(DBDefinition db) {
		LineSegment lineString = getValue();
		if (lineString == null) {
			return db.getNull();
		} else {
			String str = db.transformLineSegmentIntoDatabaseLineSegment2DFormat(lineString);
			return str;
		}
	}

	@Override
	protected LineSegment getFromResultSet(DBDefinition database, ResultSet resultSet, String fullColumnName) throws SQLException, IncorrectGeometryReturnedForDatatype {

		LineSegment lineSegment = null;
		String string = resultSet.getString(fullColumnName);
		if (string == null) {
			return null;
		} else {
			try {
				if (string.equals("GEOMETRYCOLLECTION()")) {
					lineSegment = new LineSegment();
				} else {
				lineSegment = database.transformDatabaseLineSegment2DValueToJTSLineSegment(string);}
			} catch (ParseException ex) {
				Logger.getLogger(DBLineSegment2D.class.getName()).log(Level.SEVERE, null, ex);
				throw new ParsingSpatialValueException(fullColumnName, string, ex);
			}
			return lineSegment;
		}
	}

	@Override
	public boolean isAggregator() {
		return false;
	}

	@Override
	public boolean getIncludesNull() {
		return false;
	}

	@Override
	public StringExpression stringResult() {
		return LineSegment2DExpression.value(this).stringResult();
	}

	@Override
	protected void setValueFromStandardStringEncoding(String encodedValue) {
		throw new UnsupportedOperationException("DBLineSegment2D does not support setValueFromStandardStringEncoding(String) yet."); //To change body of generated methods, choose Tools | Templates.
	}

	@Override
	public LineSegment2DColumn getColumn(RowDefinition row) throws IncorrectRowProviderInstanceSuppliedException {
		return new LineSegment2DColumn(row, this);
	}


	@Override
	@SuppressWarnings("unchecked")
	public Comparator<LineSegment> getComparator() {
		return ComparableComparator.forClass(LineSegment.class);
	}
}