Metacat

/**

 *  '$RCSfile$'

 *  Copyright: 2000 Regents of the University of California and the

 *             National Center for Ecological Analysis and Synthesis

 *    Authors: John Harris

 *    Release: @release@

 *

 *   '$Author: harris $'

 *     '$Date: 2005-10-04 10:52:04 -0700 (Tue, 04 Oct 2005) $'

 * '$Revision: 2639 $'

 *

 * 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

 */

#include "shapefil.h"

#include <stdio.h>

#include <stdlib.h>

#include <string.h>

#include <math.h>

#include "iostream.h"

#include <fstream>

#include <string>

#include <vector>

using namespace std;

/************************************************************************

 * Method to write the points array to the shape file and the associated

 * dbf file using the shapelib-1.2.ave_x libraries

 *

 * @param nSHPType -- the type of shapefile (eg. poly vs point )

 *

 * @param pszFilename -- the name of the shape file, sans the extension

 *

 * @param min_pntx -- this is the float value of the min x-location

 * @param max_pntx -- this is the float value of the max x-location

 *

 * @param min_pnty -- this is the float value of the min y-location

 * @param max_pnty -- this is the float value of the max y-location

 *

 * @param pointz -- this is the float value of the optional p-value

 * (assuming points)

 *

 * @param the number of lines in the coordinates arrays

 ************************************************************************/

static void WritePointsArray ( int nSHPType,

  const char * pszFilename,

  float min_pntx[40000][100],

  float max_pntx[40000][100],

  float min_pnty[40000][100],

  float max_pnty[40000][100],

  float pointz[40000][100],

  vector<string> urlvec,

  vector<string> docidvec,

  int number_data_lines ) {

       SHPHandle hSHPHandle;

       SHPObject * psShape;

       double x, y, z, m, j, dx[number_data_lines], dy[number_data_lines], dz[number_data_lines];

       int ii, iRecord, i;

       char buf[20];

       char intString;

       //for the dbf file

       DBFHandle hDBF;

       //create the dbf file

       hDBF = DBFCreate( pszFilename );

       if ( hDBF == NULL )

       {

         printf( "DBFCreate(%s) failed.\n" );

         exit( 2 );

       }

       //add to the dbf file attributes for ID, pval(elevation), url

       if ( DBFAddField( hDBF, "ID", FTInteger, 10, 0 ) == -1 )

       {

         printf( "DBFAddField(%s,FTInteger,%d,0) failed.\n" );

       }

       if ( DBFAddField( hDBF, "area_coef", FTDouble, 10, 4 ) == -1 )

       {

         printf( "DBFAddField(%s,FTDouble,%d,0) failed.\n" );

       }

       if ( DBFAddField( hDBF, "url", FTString, 220, 0 ) == -1 )

       {

         printf( "DBFAddField(%s,FTString,%d,0) failed.\n" );

       }

       if ( DBFAddField( hDBF, "docid", FTString, 220, 0 ) == -1 )

       {

         printf( "DBFAddField(%s,FTString,%d,0) failed.\n" );

       }

      //cout<<" DBF Record Count --> "<<iRecord<<endl;

      int id_field = DBFGetFieldIndex( hDBF, "ID" );

      int url_field = DBFGetFieldIndex( hDBF, "url" );

      int docid_field = DBFGetFieldIndex( hDBF, "docid" );

      int area_field = DBFGetFieldIndex( hDBF, "area_coef" );

      hSHPHandle = SHPCreate( pszFilename, nSHPType );

       ii = 0;

       for ( ii = 0; ii < number_data_lines; ii++ )

       {

         double area_coef = sqrt( pow((max_pntx[2][ii]-min_pntx[2][ii]),2) + pow((max_pnty[2][ii]-min_pnty[2][ii]),2)  );

         double ave_x = ( (min_pntx[2][ii]+max_pntx[2][ii])/2 );

         double ave_y = ( (min_pnty[2][ii]+max_pnty[2][ii])/2 );

         double ave_z = ( pointz[2][ii] );

         if (ave_x < 180 && ave_x > -180 && ave_y < 90 && ave_y >-90 ) {

         psShape = SHPCreateObject( SHPT_POINT, ii, 0, NULL, NULL, 1, &ave_x, &ave_y, &ave_z, NULL );

         SHPWriteObject( hSHPHandle, -1, psShape );

         SHPDestroyObject( psShape );

         //do the update to the dbf file too

         iRecord = DBFGetRecordCount( hDBF );

        DBFWriteStringAttribute(hDBF, iRecord, url_field, urlvec[ii].c_str() );

        DBFWriteStringAttribute(hDBF, iRecord, docid_field, docidvec[ii].c_str() );

        DBFWriteIntegerAttribute(hDBF, iRecord, id_field, int(ii) );

        DBFWriteDoubleAttribute(hDBF, iRecord, area_field, area_coef );

        }

       }

       //recompute the extents

       //SHPComputeExtents( psShape );

       //cout << "xmin: " << psShape->dfXMin << " xmax: " << psShape->dfXMax << endl;

       //cout << "ymin: " << psShape->dfYMin << " ymax: " << psShape->dfYMax << endl;

       DBFClose( hDBF );

       SHPClose( hSHPHandle );

}

/** getUsage -- function that prints to stdout the usage for this driver */

static void getUsage()

{

  std::cout<<"usage: metacat2shape metacat_export.txt test.shp\n\n";

  std::cout<<"  where:\n\n";

  std::cout<<"    input -- is an input ASCII file, that contains the following cols:\n";

  std::cout<<"          -> docid -- the document id\n";

  std::cout<<"          -> x -- the xlocation\n";

  std::cout<<"          -> y -- the y location\n";

  std::cout<<"          -> url -- the url to the document\n";

  std::cout<<"          shapefile -- {output} is a ESRI shapefile\n\n";

  std::cout<<"  see:\n";

  std::cout<<"  example: metacat2shape metacat_export.txt test.shp\n\n";

  std::cout<<"  ##><><><><><><><><><><><><><><><><><><><><><><><><>##\n";

  std::cout<<"  ##           John Harris                           ##\n";

  std::cout<<"  ##           harris@nceas.ucsb.edu                 ##\n";

  std::cout<<"  ##           Copyright 1999-2003                   ##\n";

  std::cout<<"  ##><><><><><><><><><><><><><><><><><><><><><><><><>##\n";

  std::cout << std::endl;

}

int main( int argc, char * * argv )

{

  //get the usage info if the inputs are not right

  if ( argc != 3 )

  {

    getUsage();

    exit( 1 );

  }

  //these arrays are for the input x, y, z data

  float min_pntx[1000] [100];

  float min_pnty[1000] [100];

  float max_pntx[1000] [100];

  float max_pnty[1000] [100];

  float pointz[1000] [100];

  vector<string> docidvec;

  vector<string> urlvec;

  FILE * infile;

  char oneline[200];

  char * value;

  int cnt = 0;

  int number_lines = 0;

  char buf_docid[100];

  char buf_xmin[100];

  char buf_xmax[100];

  char buf_ymin[100];

  char buf_ymax[100];

  char buf_url[100];

  SHPHandle hSHP;

  int nShapeType, nEntities, i, iPart;

  const char * pszPlus;

  //test the reading of the file

  if ( ( infile = fopen( argv[1], "r" ) ) == ( FILE * ) NULL )

  {

    fprintf( stderr, "Can't read the file: %s.\n", argv[1] );

    exit( 2 );

  }

  /* read the data into an array -- first the grid nodes

   *

   * the data should have the following columns (from left to right)

   *

   * nrs.239.1

   * -121.558333333333

   * -121.558333333333

   * 36.2083333333333

   * 36.2083333333333

   * http://metacat.nceas.ucsb.edu/knb/servlet/metacat?action=read&docid=nrs.239.1&qformat=knb

   */

  infile = fopen( argv[1], "r" );

  char const * out_shape = argv[2];

  while ( fgets( oneline, 199, infile ) ) {

    sscanf( oneline, "%s %s %s %s %s %s %s", buf_docid, buf_xmin, buf_xmax, buf_ymin, buf_ymax, buf_url);

    /**** cout<<" docid -> " << bufa

      <<"\n cnt -> " << cnt

      <<"\n x -> " << buf

      <<"\n y -> " << buf1

      <<"\n p -> " << buf2

      <<endl; *******/

    min_pntx[2] [cnt] = atof(buf_xmin);

    max_pntx[2] [cnt] = atof(buf_xmax);

    min_pnty[2] [cnt] = atof(buf_ymin);

    max_pnty[2] [cnt] = atof(buf_ymax);

    urlvec.push_back(buf_url);

    docidvec.push_back(buf_docid);

    cnt++;

    number_lines = cnt;

  }

  fclose( infile );

  cout<<"creating a new shape file, with "<< number_lines <<" records "<<endl;

  WritePointsArray( SHPT_POINT,

                    out_shape,

                    min_pntx,

                    max_pntx,

                    min_pnty,

                    max_pnty,

                    pointz,

                    urlvec,

                    docidvec,

                    number_lines );

#ifdef USE_DBMALLOC

  malloc_dump( 2 );

#endif

  exit( 0 );

}