com.datastax.dse.driver.internal.core.data.geometry.DefaultGeometry Maven / Gradle / Ivy

/*
 * Copyright DataStax, Inc.
 *
 * This software can be used solely with DataStax Enterprise. Please consult the license at
 * http://www.datastax.com/terms/datastax-dse-driver-license-terms
 */
package com.datastax.dse.driver.internal.core.data.geometry;

import com.datastax.dse.driver.api.core.data.geometry.Geometry;
import com.datastax.dse.driver.api.core.data.geometry.Point;
import com.datastax.oss.driver.shaded.guava.common.base.Preconditions;
import com.datastax.oss.driver.shaded.guava.common.collect.ImmutableList;
import com.datastax.dse.driver.shaded.esri.core.geometry.GeometryException;
import com.datastax.dse.driver.shaded.esri.core.geometry.SpatialReference;
import com.datastax.dse.driver.shaded.esri.core.geometry.ogc.OGCGeometry;
import com.datastax.dse.driver.shaded.esri.core.geometry.ogc.OGCLineString;
import edu.umd.cs.findbugs.annotations.NonNull;
import java.io.InvalidObjectException;
import java.io.ObjectInputStream;
import java.io.Serializable;
import java.nio.ByteBuffer;
import net.jcip.annotations.Immutable;

@Immutable
public abstract class DefaultGeometry implements Geometry, Serializable {

  private static final long serialVersionUID = 1L;

  /**
   * Default spatial reference for Well Known Text / Well Known Binary.
   *
   * 4326 is the EPSG identifier of the World Geodetic System (WGS) in
   * its later revision, WGS 84.
   */
  public static final SpatialReference SPATIAL_REFERENCE_4326 = SpatialReference.create(4326);

  @NonNull
  public static  T fromOgcWellKnownText(
      @NonNull String source, @NonNull Class klass) {
    OGCGeometry geometry;
    try {
      geometry = OGCGeometry.fromText(source);
    } catch (IllegalArgumentException e) {
      throw new IllegalArgumentException(e.getMessage());
    }
    validateType(geometry, klass);
    return klass.cast(geometry);
  }

  @NonNull
  public static  T fromOgcWellKnownBinary(
      @NonNull ByteBuffer source, @NonNull Class klass) {
    OGCGeometry geometry;
    try {
      geometry = OGCGeometry.fromBinary(source);
    } catch (IllegalArgumentException e) {
      throw new IllegalArgumentException(e.getMessage());
    }
    validateType(geometry, klass);
    return klass.cast(geometry);
  }

  @NonNull
  public static  T fromOgcGeoJson(
      @NonNull String source, @NonNull Class klass) {
    OGCGeometry geometry;
    try {
      geometry = OGCGeometry.fromGeoJson(source);
    } catch (Exception e) {
      throw new IllegalArgumentException(e.getMessage());
    }
    validateType(geometry, klass);
    return klass.cast(geometry);
  }

  private static void validateType(OGCGeometry geometry, Class klass) {
    if (!geometry.getClass().equals(klass)) {
      throw new IllegalArgumentException(
          String.format(
              "%s is not of type %s", geometry.getClass().getSimpleName(), klass.getSimpleName()));
    }
  }

  private final OGCGeometry ogcGeometry;

  protected DefaultGeometry(@NonNull OGCGeometry ogcGeometry) {
    this.ogcGeometry = ogcGeometry;
    Preconditions.checkNotNull(ogcGeometry);
    validateOgcGeometry(ogcGeometry);
  }

  private static void validateOgcGeometry(OGCGeometry geometry) {
    try {
      if (geometry.is3D()) {
        throw new IllegalArgumentException(String.format("'%s' is not 2D", geometry.asText()));
      }
      if (!geometry.isSimple()) {
        throw new IllegalArgumentException(
            String.format(
                "'%s' is not simple. Points and edges cannot self-intersect.", geometry.asText()));
      }
    } catch (GeometryException e) {
      throw new IllegalArgumentException("Invalid geometry" + e.getMessage());
    }
  }

  @NonNull
  public static ImmutableList getPoints(@NonNull OGCLineString lineString) {
    ImmutableList.Builder builder = ImmutableList.builder();
    for (int i = 0; i < lineString.numPoints(); i++) {
      builder.add(new DefaultPoint(lineString.pointN(i)));
    }
    return builder.build();
  }

  protected static com.datastax.dse.driver.shaded.esri.core.geometry.Point toEsri(Point p) {
    return new com.datastax.dse.driver.shaded.esri.core.geometry.Point(p.X(), p.Y());
  }

  @NonNull
  public OGCGeometry getOgcGeometry() {
    return ogcGeometry;
  }

  @NonNull
  public com.datastax.dse.driver.shaded.esri.core.geometry.Geometry getEsriGeometry() {
    return ogcGeometry.getEsriGeometry();
  }

  @NonNull
  @Override
  public String asWellKnownText() {
    return ogcGeometry.asText();
  }

  @NonNull
  @Override
  public ByteBuffer asWellKnownBinary() {
    return WkbUtil.asLittleEndianBinary(ogcGeometry);
  }

  @NonNull
  @Override
  public String asGeoJson() {
    return ogcGeometry.asGeoJson();
  }

  @Override
  public boolean contains(@NonNull Geometry other) {
    Preconditions.checkNotNull(other);
    if (other instanceof DefaultGeometry) {
      DefaultGeometry defautlOther = (DefaultGeometry) other;
      return getOgcGeometry().contains(defautlOther.getOgcGeometry());
    }
    return false;
  }

  @Override
  public boolean equals(Object o) {
    if (this == o) {
      return true;
    }
    if (o == null || getClass() != o.getClass()) {
      return false;
    }
    DefaultGeometry that = (DefaultGeometry) o;
    return this.getOgcGeometry().equals(that.getOgcGeometry());
  }

  @Override
  public int hashCode() {
    // OGCGeometry subclasses do not overwrite Object.hashCode()
    // while com.datastax.dse.driver.shaded.esri.core.geometry.Geometry subclasses usually do,
    // so use these instead; this is consistent with equals
    // because OGCGeometry.equals() actually compare between
    // com.datastax.dse.driver.shaded.esri.core.geometry.Geometry objects
    return getEsriGeometry().hashCode();
  }

  // Should never be called since we serialize a proxy (see subclasses)
  private void readObject(ObjectInputStream stream) throws InvalidObjectException {
    throw new InvalidObjectException("Proxy required");
  }

  @Override
  public String toString() {
    return asWellKnownText();
  }
}