Metacat

/**

 *  '$RCSfile$'

 *  Copyright: 2003 Regents of the University of California.

 *

 * Author: Matthew Perry 

 * '$Date: 2006-10-25 18:29:04 -0700 (Wed, 25 Oct 2006) $'

 * '$Revision: 3068 $'

 *

 * This program is free software; you can redistribute it and/or modify

 * it under the terms of the GNU General Public License as published by

 * the Free Software Foundation; either version 2 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 General Public License for more details.

 *

 * You should have received a copy of the GNU General Public License

 * along with this program; if not, write to the Free Software

 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA

 */

package edu.ucsb.nceas.metacat.spatial;

import java.io.File;

import edu.ucsb.nceas.metacat.DBConnection;

import edu.ucsb.nceas.metacat.MetaCatUtil;

import com.vividsolutions.jts.geom.Coordinate;

import com.vividsolutions.jts.geom.Point;

import com.vividsolutions.jts.geom.Polygon;

import com.vividsolutions.jts.geom.MultiPolygon;

import com.vividsolutions.jts.geom.MultiPoint;

import com.vividsolutions.jts.geom.GeometryFactory;

import com.vividsolutions.jts.geom.PrecisionModel;

import org.geotools.feature.AttributeType;

import org.geotools.feature.AttributeTypeFactory;

import org.geotools.feature.Feature;

import org.geotools.feature.FeatureType;

import org.geotools.feature.FeatureTypeFactory;

import org.geotools.feature.SchemaException;

import java.sql.ResultSet;

import java.sql.PreparedStatement;

import java.util.Vector;

import org.apache.log4j.Logger;

/**

 * Class representing the spatial portions of an xml document

 * as a geotools Feature.

 */

public class SpatialDocument {

  private DBConnection dbconn;

  private static Logger log =

      Logger.getLogger(SpatialDocument.class.getName());

  private SpatialFeatureSchema featureSchema = new SpatialFeatureSchema();

  Vector west = new Vector();

  Vector south = new Vector();

  Vector east = new Vector();

  Vector north = new Vector();

  String title = null;

  String docid = null;

  /** 

   * Constructor that queries the db 

   *

   * @param docid The document id to be represented spatially

   * @param dbconn The database connection shared from the refering method.

   * @todo Rewrite the title query to use xml_path_index

   */

  public SpatialDocument( String docid , DBConnection dbconn ) {

    this.docid = docid;

    PreparedStatement pstmt = null;

    ResultSet rs = null;

    this.dbconn = dbconn;

    /*

     * Get the bounding coordinates

     */

    String query = "SELECT path, nodedatanumerical, parentnodeid FROM xml_path_index"

    + " WHERE docid = '" + docid.trim() + "'"

    + " AND docid IN (SELECT distinct docid FROM xml_access WHERE docid = '" + docid.trim() 

    + "' AND principal_name = 'public' AND perm_type = 'allow')"

    + " AND (path = '" + MetaCatUtil.getOption("westBoundingCoordinatePath") + "'"

    + "  OR path = '" + MetaCatUtil.getOption("southBoundingCoordinatePath") + "'"

    + "  OR path = '" + MetaCatUtil.getOption("eastBoundingCoordinatePath") + "'"

    + "  OR path = '" + MetaCatUtil.getOption("northBoundingCoordinatePath") + "'"

    + " ) ORDER BY parentnodeid;";

    try {

      pstmt = dbconn.prepareStatement(query);

      pstmt.execute();

      rs = pstmt.getResultSet();

      while (rs.next()) {

        if ( rs.getString(1).equals( MetaCatUtil.getOption("westBoundingCoordinatePath") ) )

            this.west.add( rs.getFloat(2) );

        else if ( rs.getString(1).equals( MetaCatUtil.getOption("southBoundingCoordinatePath") ) )

            this.south.add( rs.getFloat(2) );

        else if ( rs.getString(1).equals( MetaCatUtil.getOption("eastBoundingCoordinatePath") ) )

            this.east.add( rs.getFloat(2) );

        else if ( rs.getString(1).equals( MetaCatUtil.getOption("northBoundingCoordinatePath") ) )

            this.north.add( rs.getFloat(2) );

        else

            log.error("** An xml path not related to your bounding coordinates was returned by this query \n" + query + "\n");

      }

      rs.close();

      pstmt.close();

    }

    catch(Exception e) {

      log.error(" ---- Could not get bounding coordinates for " + docid + );

      e.printStackTrace();

    }      

    /*

     * Get the title

     */

    String docTitlePath = MetaCatUtil.getOption("docTitle");

    query = "select nodedata from xml_path_index where path = '" 

          + docTitlePath.trim() + "' and docid = '" + docid.trim() + "'";

    try {

      pstmt = dbconn.prepareStatement(query);

      pstmt.execute();

      rs = pstmt.getResultSet();

      if (rs.next())

        this.title = rs.getString(1);

        //log.warn(" \n\n****** TITLE query is " + query + " \n\n");

      rs.close();

      pstmt.close();

    }

    catch(Exception e) {

      log.error(" **** Error getting docids from getTitle for docid = "+docid);

      e.printStackTrace();

      log.error(" query ============== \n " + query);

      this.title = docid;

    }

    /* try {

        dbconn.close();

    } catch( java.sql.SQLException e ) {

        log.error("java.sql.SQLException");

    }

    */

  }

  /** 

   * Returns a geotools (multi)polygon feature with geometry plus attributes

   * ready to be inserted into our spatial dataset cache

   */

  public Feature getPolygonFeature() {

      // Get polygon feature type

      FeatureType polyType = featureSchema.getPolygonFeatureType();

      MultiPolygon theGeom = getPolygonGeometry();

      if (theGeom == null)

          return null;

      // Populate the feature schema

      try {

          Feature polyFeature = polyType.create(new Object[]{ 

              theGeom,

              this.docid,

              getUrl(this.docid), 

              this.title });

          return polyFeature; 

      } catch (org.geotools.feature.IllegalAttributeException e) {

          log.error("!!!!!!! org.geotools.feature.IllegalAttributeException");

          return null;

      }

  }

  /**

   * Returns a geotools (multi)point feature with geometry plus attributes

   * ready to be inserted into our spatial dataset cache

   *

   */

  public Feature getPointFeature() {

      // Get polygon feature type

      FeatureType pointType = featureSchema.getPointFeatureType();

      MultiPoint theGeom = getPointGeometry();

      if (theGeom == null)

          return null;

      // Populate the feature schema

      try {

          Feature pointFeature = pointType.create(new Object[]{ 

              theGeom,

              this.docid,

              getUrl(this.docid), 

              this.title });

          return pointFeature;

      } catch (org.geotools.feature.IllegalAttributeException e) {

          log.error("!!!!!!! org.geotools.feature.IllegalAttributeException");

          return null;

      }

  }

  /**

   * Given a valid docid, return an appropriate URL

   * for viewing the metadata document

   *

   * @param docid The document id for which to construct the access url.

   */

  private String getUrl( String docid ) {

     String docUrl = MetaCatUtil.getOption("metacatUrl")  

                    + "?action=read"

                    + "&docid=" + docid 

                    + "&qformat=" + MetaCatUtil.getOption("default-style");

     return docUrl;

  }

  /**

   * Returns a mutlipolygon geometry representing the geographic coverage(s) of the document

   *

   */

  private MultiPolygon getPolygonGeometry() {

    PrecisionModel precModel = new PrecisionModel(); // default: Floating point

    GeometryFactory geomFac = new GeometryFactory( precModel, featureSchema.srid );

    Vector polygons = new Vector();

    Float w;

    Float s;

    Float e;

    Float n;

    if ( west.size() == south.size() && south.size() == east.size() && east.size() == north.size() ) {

        for (int i = 0; i < west.size(); i++) {

            w = (Float)west.elementAt(i);

            s = (Float)south.elementAt(i);

            e = (Float)east.elementAt(i);

            n = (Float)north.elementAt(i);

            // Check if it's actually a valid polygon

            if (  w == 0.0 && s == 0.0 && e == 0.0 && n == 0.0) {

                log.warn("        Invalid or empty coodinates ... skipping");

                continue;

            } else if( w.compareTo( e ) == 0 && n.compareTo( s ) == 0 ) {

                log.warn("        Point coordinates only.. skipping polygon generation");

                continue;

            }

            // Handle the case of crossing the dateline and poles

            // dateline crossing is valid 

            // polar crossing is not ( so we swap north and south )

            // Assumes all coordinates are confined to -180 -90 180 90

            Float dl = new Float("180.0");

            Float _dl = new Float("-180.0");

            if ( w > e && s > n ) {

                log.info( "Crosses both the dateline and the poles .. split into 2 polygons, swap n & s" );

                polygons.add( createPolygonFromBbox( geomFac,   w,   n, dl, s ) );

                polygons.add( createPolygonFromBbox( geomFac, _dl,   n,  e, s ) );

            } else if ( w > e ) {

                log.info( "Crosses the dateline .. split into 2 polygons" );

                polygons.add( createPolygonFromBbox( geomFac,   w, s, dl, n ) );

                polygons.add( createPolygonFromBbox( geomFac, _dl, s,  e, n ) );

            } else if ( s > n ) {

                log.info( "Crosses the poles .. swap north and south" );

                polygons.add( createPolygonFromBbox( geomFac, w, n, e, s ) );

            } else {

                // Just a standard polygon that fits nicely onto our flat earth

                polygons.add( createPolygonFromBbox( geomFac, w, s, e, n ) );    

            }

        }

    } else {

       log.error(" *** Something went wrong.. your east,west,north and south bounding arrays are different sizes!");

    }

    if( polygons.size() > 0 ) {

       Polygon[] polyArray = geomFac.toPolygonArray( polygons );

       MultiPolygon multiPolyGeom= geomFac.createMultiPolygon( polyArray );

       return multiPolyGeom; 

    } else {

       return null;

    } 

  }

  /**

   * Returns a polygon given the four bounding box coordinates

   */

  private Polygon createPolygonFromBbox( GeometryFactory geomFac, Float w, Float s, Float e, Float n ) {

        Coordinate[] linestringCoordinates = new Coordinate[5];

        linestringCoordinates[0] = new Coordinate( w, s );

        linestringCoordinates[1] = new Coordinate( w, n );

        linestringCoordinates[2] = new Coordinate( e, n );

        linestringCoordinates[3] = new Coordinate( e, s );

        linestringCoordinates[4] = new Coordinate( w, s );

        return geomFac.createPolygon( geomFac.createLinearRing(linestringCoordinates), null);

  }

  /**

   * Returns a multipoint geometry represnting the geographic coverage(s) of the document

   *

   * @todo Handle the case of crossing the dateline and poles

   */

  private MultiPoint getPointGeometry() {

    PrecisionModel precModel = new PrecisionModel(); // default: Floating point

    GeometryFactory geomFac = new GeometryFactory( precModel, featureSchema.srid );

    Float w;

    Float s;

    Float e;

    Float n;

    PreparedStatement pstmt = null;

    ResultSet rs = null;

    Vector points = new Vector();

    if ( west.size() == south.size() && south.size() == east.size() && east.size() == north.size() ) {

        for (int i = 0; i < west.size(); i++) {

            w = (Float)west.elementAt(i);

            s = (Float)south.elementAt(i);

            e = (Float)east.elementAt(i);

            n = (Float)north.elementAt(i);

            // Check if it's actually a valid point

            if (  w == 0.0 && s == 0.0 && e == 0.0 && n == 0.0) {

                 log.warn("        Invalid or empty coodinates ... skipping");

                 continue;

            }

            Double xCenter;

            Double yCenter;

            // Handle the case of crossing the dateline and poles

            // Assumes all coordinates are confined to -180 -90 180 90

            if ( w > e ) {

                log.info( "Crosses the dateline .. " );

                xCenter = ((float)360.0 - w + e)/ (Double)2.0 + w;

                if( xCenter > 180 )

                    xCenter = xCenter - 360;

                yCenter = ( s + n ) / (Double) 2.0;

            } else {

                // Just a standard point that can be calculated by the average coordinates

                xCenter = ( w + e ) / (Double) 2.0;

                yCenter = ( s + n ) / (Double) 2.0;

            }

            points.add( geomFac.createPoint( new Coordinate( xCenter, yCenter)) );

        }

    } else {

       log.error(" *** Something went wrong.. your east,west,north and south bounding vectors are different sizes!");

    }

    if( points.size() > 0 ) {

       Point[] pointArray = geomFac.toPointArray( points );

       MultiPoint multiPointGeom= geomFac.createMultiPoint( pointArray );

       return multiPointGeom; 

    } else {

       return null;

    } 

  }

}