Project

General

Profile

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: 2006-02-23 11:16:08 -0800 (Thu, 23 Feb 2006) $'
10
 * '$Revision: 2915 $'
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
// 
60
// 2/19/06 R Reeves modified array declarations for WritePointsArray funciton.
61
//
62
const int number_data_lines = 100;
63
static void WritePointsArray ( int nSHPType,
64
                               const char * pszFilename,
65
                               float min_pntx[][100],
66
                               float max_pntx[][100],
67
                               float min_pnty[][100],
68
                               float max_pnty[][100],
69
                               float pointz[][100],
70
                               vector<string> urlvec,
71
                               vector<string> docidvec,
72
                               int number_data_lines_x
73
							  )
74
{  
75
       SHPHandle hSHPHandle;
76
       SHPObject * psShape;
77
       double x, y, z, m, j, dx[number_data_lines], dy[number_data_lines], dz[number_data_lines];
78
       int ii, iRecord, i;
79
       char buf[20];
80
       char intString;
81
       //for the dbf file
82
       DBFHandle hDBF;
83

    
84
       //create the dbf file
85
       hDBF = DBFCreate( pszFilename );
86
       if ( hDBF == NULL )
87
       {
88
         printf( "DBFCreate(%s) failed.\n" );
89
         exit( 2 );
90
       }
91

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

    
110

    
111
      //cout<<" DBF Record Count --> "<<iRecord<<endl;
112
      int id_field = DBFGetFieldIndex( hDBF, "ID" );
113
      int url_field = DBFGetFieldIndex( hDBF, "url" );
114
      int docid_field = DBFGetFieldIndex( hDBF, "docid" );
115
      int area_field = DBFGetFieldIndex( hDBF, "area_coef" );
116

    
117
      hSHPHandle = SHPCreate( pszFilename, nSHPType );
118

    
119
       ii = 0;
120
       for ( ii = 0; ii < number_data_lines; ii++ )
121
       {
122
         double area_coef = sqrt( pow((max_pntx[2][ii]-min_pntx[2][ii]),2) + pow((max_pnty[2][ii]-min_pnty[2][ii]),2)  );
123
         double ave_x = ( (min_pntx[2][ii]+max_pntx[2][ii])/2 );
124
         double ave_y = ( (min_pnty[2][ii]+max_pnty[2][ii])/2 );
125
         double ave_z = ( pointz[2][ii] );
126

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

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

    
131
         SHPWriteObject( hSHPHandle, -1, psShape );
132
         SHPDestroyObject( psShape );
133
         //do the update to the dbf file too
134
         iRecord = DBFGetRecordCount( hDBF );
135

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

    
150

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

    
162
  cout<<"          shapefile -- {output} is a ESRI shapefile\n\n";
163

    
164
  cout<<"  see:\n";
165
  cout<<"  example: metacat2shape metacat_export.txt test.shp\n\n";
166

    
167
  cout<<"  ##><><><><><><><><><><><><><><><><><><><><><><><><>##\n";
168
  cout<<"  ##           John Harris                           ##\n";
169
  cout<<"  ##           harris@nceas.ucsb.edu                 ##\n";
170
  cout<<"  ##           Copyright 1999-2003                   ##\n";
171
  cout<<"  ##><><><><><><><><><><><><><><><><><><><><><><><><>##\n";
172

    
173
  cout << endl;
174

    
175
}
176

    
177

    
178

    
179
int main( int argc, char * * argv )
180
{
181
  //get the usage info if the inputs are not right
182
  if ( argc != 3 )
183
  {
184
    getUsage();
185
    exit( 1 );
186
  }
187

    
188
  //these arrays are for the input x, y, z data
189
  float min_pntx[1000] [100];
190
  float min_pnty[1000] [100];
191
  float max_pntx[1000] [100];
192
  float max_pnty[1000] [100];
193

    
194
  float pointz[1000] [100];
195
  vector<string> docidvec;
196
  vector<string> urlvec;
197

    
198
  FILE * infile;
199
  char oneline[200];
200
  char * value;
201

    
202
  int cnt = 0;
203
  int number_lines = 0;
204
  char buf_docid[100];
205
  char buf_xmin[100];
206
  char buf_xmax[100];
207
  char buf_ymin[100];
208
  char buf_ymax[100];
209
  char buf_url[100];
210

    
211
  SHPHandle hSHP;
212
  int nShapeType, nEntities, i, iPart;
213
  const char * pszPlus;
214

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

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

    
238
    /**** cout<<" docid -> " << bufa
239
      <<"\n cnt -> " << cnt
240
      <<"\n x -> " << buf
241
      <<"\n y -> " << buf1
242
      <<"\n p -> " << buf2
243
      <<endl; *******/
244

    
245
    min_pntx[2] [cnt] = atof(buf_xmin);
246
    max_pntx[2] [cnt] = atof(buf_xmax);
247
    min_pnty[2] [cnt] = atof(buf_ymin);
248
    max_pnty[2] [cnt] = atof(buf_ymax);
249

    
250
    urlvec.push_back(buf_url);
251
    docidvec.push_back(buf_docid);
252

    
253
    cnt++;
254
    number_lines = cnt;
255
  }
256
  fclose( infile );
257

    
258
  cout<<"creating a new shape file, with "<< number_lines <<" records "<<endl;
259
  WritePointsArray( SHPT_POINT,
260
                    out_shape,
261
                    min_pntx,
262
                    max_pntx,
263
                    min_pnty,
264
                    max_pnty,
265
                    pointz,
266
                    urlvec,
267
                    docidvec,
268
                    number_lines );
269

    
270
#ifdef USE_DBMALLOC
271
  malloc_dump( 2 );
272
#endif
273
  exit( 0 );
274
}
(1-1/2)