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: 2006-02-06 12:21:50 -0800 (Mon, 06 Feb 2006) $'

 * '$Revision: 2907 $'

 *

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

}