Project

General

Profile

metacat / src / spatial / cpp / metacat_shapefile.cpp @ 2639

1
/**
2
 *  '$RCSfile$'
3
 *  Copyright: 2000 Regents of the University of California and the
4
 *             National Center for Ecological Analysis and Synthesis
5
 *    Authors: John Harris
6
 *    Release: @release@
7
 *
8
 *   '$Author: harris $'
9
 *     '$Date: 2005-10-04 10:52:04 -0700 (Tue, 04 Oct 2005) $'
10
 * '$Revision: 2639 $'
11
 *
12
 * This program is free software; you can redistribute it and/or modify
13
 * it under the terms of the GNU General Public License as published by
14
 * the Free Software Foundation; either version 2 of the License, or
15
 * (at your option) any later version.
16
 *
17
 * This program is distributed in the hope that it will be useful,
18
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
19
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
20
 * GNU General Public License for more details.
21
 *
22
 * You should have received a copy of the GNU General Public License
23
 * along with this program; if not, write to the Free Software
24
 * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
25
 */
26

    
27
#include "shapefil.h"
28
#include <stdio.h>
29
#include <stdlib.h>
30
#include <string.h>
31
#include <math.h>
32
#include "iostream.h"
33
#include <fstream>
34
#include <string>
35
#include <vector>
36

    
37
using namespace std;
38

    
39

    
40
/************************************************************************
41
 * Method to write the points array to the shape file and the associated
42
 * dbf file using the shapelib-1.2.ave_x libraries
43
 *
44
 * @param nSHPType -- the type of shapefile (eg. poly vs point )
45
 *
46
 * @param pszFilename -- the name of the shape file, sans the extension
47
 *
48
 * @param min_pntx -- this is the float value of the min x-location
49
 * @param max_pntx -- this is the float value of the max x-location
50
 *
51
 * @param min_pnty -- this is the float value of the min y-location
52
 * @param max_pnty -- this is the float value of the max y-location
53
 *
54
 * @param pointz -- this is the float value of the optional p-value
55
 * (assuming points)
56
 *
57
 * @param the number of lines in the coordinates arrays
58
 ************************************************************************/
59
static void WritePointsArray ( int nSHPType,
60
  const char * pszFilename,
61
  float min_pntx[40000][100],
62
  float max_pntx[40000][100],
63
  float min_pnty[40000][100],
64
  float max_pnty[40000][100],
65
  float pointz[40000][100],
66
  vector<string> urlvec,
67
  vector<string> docidvec,
68
  int number_data_lines ) {
69

    
70
       SHPHandle hSHPHandle;
71
       SHPObject * psShape;
72
       double x, y, z, m, j, dx[number_data_lines], dy[number_data_lines], dz[number_data_lines];
73
       int ii, iRecord, i;
74
       char buf[20];
75
       char intString;
76
       //for the dbf file
77
       DBFHandle hDBF;
78

    
79
       //create the dbf file
80
       hDBF = DBFCreate( pszFilename );
81
       if ( hDBF == NULL )
82
       {
83
         printf( "DBFCreate(%s) failed.\n" );
84
         exit( 2 );
85
       }
86

    
87
       //add to the dbf file attributes for ID, pval(elevation), url
88
       if ( DBFAddField( hDBF, "ID", FTInteger, 10, 0 ) == -1 )
89
       {
90
         printf( "DBFAddField(%s,FTInteger,%d,0) failed.\n" );
91
       }
92
       if ( DBFAddField( hDBF, "area_coef", FTDouble, 10, 4 ) == -1 )
93
       {
94
         printf( "DBFAddField(%s,FTDouble,%d,0) failed.\n" );
95
       }
96
       if ( DBFAddField( hDBF, "url", FTString, 220, 0 ) == -1 )
97
       {
98
         printf( "DBFAddField(%s,FTString,%d,0) failed.\n" );
99
       }
100
       if ( DBFAddField( hDBF, "docid", FTString, 220, 0 ) == -1 )
101
       {
102
         printf( "DBFAddField(%s,FTString,%d,0) failed.\n" );
103
       }
104

    
105

    
106
      //cout<<" DBF Record Count --> "<<iRecord<<endl;
107
      int id_field = DBFGetFieldIndex( hDBF, "ID" );
108
      int url_field = DBFGetFieldIndex( hDBF, "url" );
109
      int docid_field = DBFGetFieldIndex( hDBF, "docid" );
110
      int area_field = DBFGetFieldIndex( hDBF, "area_coef" );
111

    
112
      hSHPHandle = SHPCreate( pszFilename, nSHPType );
113

    
114
       ii = 0;
115
       for ( ii = 0; ii < number_data_lines; ii++ )
116
       {
117
         double area_coef = sqrt( pow((max_pntx[2][ii]-min_pntx[2][ii]),2) + pow((max_pnty[2][ii]-min_pnty[2][ii]),2)  );
118
         double ave_x = ( (min_pntx[2][ii]+max_pntx[2][ii])/2 );
119
         double ave_y = ( (min_pnty[2][ii]+max_pnty[2][ii])/2 );
120
         double ave_z = ( pointz[2][ii] );
121

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

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

    
126
         SHPWriteObject( hSHPHandle, -1, psShape );
127
         SHPDestroyObject( psShape );
128
         //do the update to the dbf file too
129
         iRecord = DBFGetRecordCount( hDBF );
130

    
131
        DBFWriteStringAttribute(hDBF, iRecord, url_field, urlvec[ii].c_str() );
132
        DBFWriteStringAttribute(hDBF, iRecord, docid_field, docidvec[ii].c_str() );
133
        DBFWriteIntegerAttribute(hDBF, iRecord, id_field, int(ii) );
134
        DBFWriteDoubleAttribute(hDBF, iRecord, area_field, area_coef );
135
        }
136
       }
137
       //recompute the extents
138
       //SHPComputeExtents( psShape );
139
       //cout << "xmin: " << psShape->dfXMin << " xmax: " << psShape->dfXMax << endl;
140
       //cout << "ymin: " << psShape->dfYMin << " ymax: " << psShape->dfYMax << endl;
141
       DBFClose( hDBF );
142
       SHPClose( hSHPHandle );
143
}
144

    
145

    
146
/** getUsage -- function that prints to stdout the usage for this driver */
147
static void getUsage()
148
{
149
  std::cout<<"usage: metacat2shape metacat_export.txt test.shp\n\n";
150
  std::cout<<"  where:\n\n";
151
  std::cout<<"    input -- is an input ASCII file, that contains the following cols:\n";
152
  std::cout<<"          -> docid -- the document id\n";
153
  std::cout<<"          -> x -- the xlocation\n";
154
  std::cout<<"          -> y -- the y location\n";
155
  std::cout<<"          -> url -- the url to the document\n";
156

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

    
159
  std::cout<<"  see:\n";
160
  std::cout<<"  example: metacat2shape metacat_export.txt test.shp\n\n";
161

    
162
  std::cout<<"  ##><><><><><><><><><><><><><><><><><><><><><><><><>##\n";
163
  std::cout<<"  ##           John Harris                           ##\n";
164
  std::cout<<"  ##           harris@nceas.ucsb.edu                 ##\n";
165
  std::cout<<"  ##           Copyright 1999-2003                   ##\n";
166
  std::cout<<"  ##><><><><><><><><><><><><><><><><><><><><><><><><>##\n";
167

    
168
  std::cout << std::endl;
169

    
170
}
171

    
172

    
173

    
174
int main( int argc, char * * argv )
175
{
176
  //get the usage info if the inputs are not right
177
  if ( argc != 3 )
178
  {
179
    getUsage();
180
    exit( 1 );
181
  }
182

    
183
  //these arrays are for the input x, y, z data
184
  float min_pntx[1000] [100];
185
  float min_pnty[1000] [100];
186
  float max_pntx[1000] [100];
187
  float max_pnty[1000] [100];
188

    
189
  float pointz[1000] [100];
190
  vector<string> docidvec;
191
  vector<string> urlvec;
192

    
193
  FILE * infile;
194
  char oneline[200];
195
  char * value;
196

    
197
  int cnt = 0;
198
  int number_lines = 0;
199
  char buf_docid[100];
200
  char buf_xmin[100];
201
  char buf_xmax[100];
202
  char buf_ymin[100];
203
  char buf_ymax[100];
204
  char buf_url[100];
205

    
206
  SHPHandle hSHP;
207
  int nShapeType, nEntities, i, iPart;
208
  const char * pszPlus;
209

    
210
  //test the reading of the file
211
  if ( ( infile = fopen( argv[1], "r" ) ) == ( FILE * ) NULL )
212
  {
213
    fprintf( stderr, "Can't read the file: %s.\n", argv[1] );
214
    exit( 2 );
215
  }
216

    
217
  /* read the data into an array -- first the grid nodes
218
   *
219
   * the data should have the following columns (from left to right)
220
   *
221
   * nrs.239.1
222
   * -121.558333333333
223
   * -121.558333333333
224
   * 36.2083333333333
225
   * 36.2083333333333
226
   * http://metacat.nceas.ucsb.edu/knb/servlet/metacat?action=read&docid=nrs.239.1&qformat=knb
227
   */
228
  infile = fopen( argv[1], "r" );
229
  char const * out_shape = argv[2];
230
  while ( fgets( oneline, 199, infile ) ) {
231
    sscanf( oneline, "%s %s %s %s %s %s %s", buf_docid, buf_xmin, buf_xmax, buf_ymin, buf_ymax, buf_url);
232

    
233
    /**** cout<<" docid -> " << bufa
234
      <<"\n cnt -> " << cnt
235
      <<"\n x -> " << buf
236
      <<"\n y -> " << buf1
237
      <<"\n p -> " << buf2
238
      <<endl; *******/
239

    
240
    min_pntx[2] [cnt] = atof(buf_xmin);
241
    max_pntx[2] [cnt] = atof(buf_xmax);
242
    min_pnty[2] [cnt] = atof(buf_ymin);
243
    max_pnty[2] [cnt] = atof(buf_ymax);
244

    
245
    urlvec.push_back(buf_url);
246
    docidvec.push_back(buf_docid);
247

    
248
    cnt++;
249
    number_lines = cnt;
250
  }
251
  fclose( infile );
252

    
253
  cout<<"creating a new shape file, with "<< number_lines <<" records "<<endl;
254
  WritePointsArray( SHPT_POINT,
255
                    out_shape,
256
                    min_pntx,
257
                    max_pntx,
258
                    min_pnty,
259
                    max_pnty,
260
                    pointz,
261
                    urlvec,
262
                    docidvec,
263
                    number_lines );
264

    
265
#ifdef USE_DBMALLOC
266
  malloc_dump( 2 );
267
#endif
268
  exit( 0 );
269
}