Metacat

/**

 *  '$RCSfile$'

 *  Copyright: 2003 Regents of the University of California.

 *

 * Author: Matthew Perry 

 * '$Date: 2006-08-31 16:54:32 -0700 (Thu, 31 Aug 2006) $'

 * '$Revision: 3035 $'

 *

 * 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;

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 **/

  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 (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(" ---- Error getting bounding coordinates for " + docid);

      e.printStackTrace();

    }      

    // Get the Title

    query = "select docid, nodedata, nodeid "

          + "from xml_nodes "

          + "where parentnodeid = "

          + "  ( select  nodeid "

          + "    from  xml_nodes "

          + "    where docid like '" + docid.trim() +"' "

          + "    and nodename like 'title%' "

          + "    and parentnodeid = "

          + "      ( select nodeid "

          + "        from xml_nodes "

          + "        where docid like '" + docid.trim() + "' "

          + "        and nodename = 'dataset' "

          + "        limit 1) "

          + "    limit 1)" ;

    try {

      pstmt = dbconn.prepareStatement(query);

      pstmt.execute();

      rs = pstmt.getResultSet();

      if (rs.next())

        this.title = rs.getString(2);

      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 jts (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 jts (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

   */

  private String getUrl( String docid ) {

     String docUrl = MetaCatUtil.getOption("metacatUrl")  

                    + "?action=read"

                    + "&docid=" + docid 

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

     return docUrl;

  }

  /**

   * returns the title of the docid

   */

  private String getTitle(String docid) {

    String title = null;

    PreparedStatement pstmt = null;

    ResultSet rs = null;

    String query = "select docid, nodedata, nodeid "

                 + "from xml_nodes "

                 + "where parentnodeid = "

                 + "  ( select  nodeid "

                 + "    from  xml_nodes "

                 + "    where docid like '" + docid.trim() +"' "

                 + "    and nodename like 'title%' "

                 + "    and parentnodeid = "

                 + "      ( select nodeid "

                 + "        from xml_nodes "

                 + "        where docid like '" + docid.trim() + "' "

                 + "        and nodename = 'dataset' "

                 + "        limit 1) "

                 + "    limit 1)" ;

    /*

     * String query = "select docid, nodedata, nodeid from xml_nodes where "

     *   + "nodeid =(select  nodeid from  xml_nodes where docid  like '"

     *   + docid.trim() + "' and nodename like 'title%');";

     */

    try {

      dbconn = new DBConnection();

      pstmt = dbconn.prepareStatement(query);

      pstmt.execute();

      rs = pstmt.getResultSet();

      if (rs.next())

        title = rs.getString(2);

      rs.close();

      pstmt.close();

      dbconn.close();

    }

    catch(Exception e) {

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

      e.printStackTrace();

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

      title = docid;

    }

    return title;

  }

  /*

   * Returns polygon geometry

   */

  private MultiPolygon getPolygonGeometry() {

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

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

    Vector polygons = new Vector();

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

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

            Coordinate[] linestringCoordinates = new Coordinate[5];

            // Check if it's actually a valid polygon

            if ( (Float)west.elementAt(i) == 0.0 &&

                 (Float)east.elementAt(i) == 0.0 && 

                 (Float)north.elementAt(i) == 0.0 &&

                 (Float)south.elementAt(i) == 0.0) {

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

                continue;

            } else if( ((Float)west.elementAt(i)).compareTo( (Float)east.elementAt(i) ) == 0 && 

                       ((Float)north.elementAt(i)).compareTo( (Float)south.elementAt(i) ) == 0 ) {

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

                continue;

            }

            linestringCoordinates[0] = new Coordinate( (Float)west.elementAt(i), (Float)south.elementAt(i) );

            linestringCoordinates[1] = new Coordinate( (Float)west.elementAt(i), (Float)north.elementAt(i));

            linestringCoordinates[2] = new Coordinate( (Float)east.elementAt(i), (Float)north.elementAt(i));

            linestringCoordinates[3] = new Coordinate( (Float)east.elementAt(i), (Float)south.elementAt(i));

            linestringCoordinates[4] = new Coordinate( (Float)west.elementAt(i), (Float)south.elementAt(i));

            polygons.add( geomFac.createPolygon( geomFac.createLinearRing(linestringCoordinates), null) );

        }

    } 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 point geometry

   */

  private MultiPoint getPointGeometry() {

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

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

    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++) {

            // Check if it's actually a valid point

            if ( (Float)west.elementAt(i) == (float)0.0 &&

                 (Float)east.elementAt(i) == (float)0.0 && 

                 (Float)north.elementAt(i) == (float)0.0 &&

                 (Float)south.elementAt(i) == (float)0.0 ) {

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

                continue;

            }

            Double xCenter = ( (Float)west.elementAt(i) + (Float)east.elementAt(i) ) / (Double) 2.0;

            Double yCenter = ( (Float)south.elementAt(i) + (Float)north.elementAt(i) ) / (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;

    } 

  }

}