VladPolskiy/osm2pgsql
1
0
Fork 0
mirror of https://github.com/osm2pgsql-dev/osm2pgsql.git synced 2025-08-19 16:28:16 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
2ce8ba56de2cee40d79532fc28bf1c01c6cf5ec7
osm2pgsql/osm2pgsql.c

827 lines
27 KiB
C
Raw Blame History

/*
#-----------------------------------------------------------------------------
# osm2pgsql - converts planet.osm file into PostgreSQL
# compatible output suitable to be rendered by mapnik
# Use: osm2pgsql planet.osm.bz2
#-----------------------------------------------------------------------------
# Original Python implementation by Artem Pavlenko
# Re-implementation by Jon Burgess, Copyright 2006
#
# 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., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
#-----------------------------------------------------------------------------
*/
#define _GNU_SOURCE
#include <stdio.h>
#include <unistd.h>
#include <stdlib.h>
#include <string.h>
#include <assert.h>
#include <getopt.h>
#include <libgen.h>
#include <libpq-fe.h>
#include <libxml/xmlstring.h>
#include <libxml/xmlreader.h>
#include "osmtypes.h"
#include "build_geometry.h"
#include "keyvals.h"
#include "middle-pgsql.h"
#include "middle-ram.h"
#include "output-pgsql.h"
#include "output-gazetteer.h"
#include "output-null.h"
#include "sanitizer.h"
#include "reprojection.h"
#include "text-tree.h"
#include "input.h"
#include "sprompt.h"
typedef enum { FILETYPE_NONE, FILETYPE_OSM, FILETYPE_OSMCHANGE, FILETYPE_PLANETDIFF } filetypes_t;
typedef enum { ACTION_NONE, ACTION_CREATE, ACTION_MODIFY, ACTION_DELETE } actions_t;
static int count_node, max_node;
static int count_way, max_way;
static int count_rel, max_rel;
struct output_t *out;
/* Since {node,way} elements are not nested we can guarantee the
values in an end tag must match those of the corresponding
start tag and can therefore be cached.
*/
static double node_lon, node_lat;
static struct keyval tags;
static int *nds;
static int nd_count, nd_max;
static struct member *members;
static int member_count, member_max;
static int osm_id;
#define INIT_MAX_MEMBERS 64
#define INIT_MAX_NODES 4096
int verbose;
static void realloc_nodes();
static void realloc_members();
// Bounding box to filter imported data
const char *bbox = NULL;
static double minlon, minlat, maxlon, maxlat;
static int extra_attributes;
static void printStatus(void)
{
fprintf(stderr, "\rProcessing: Node(%dk) Way(%dk) Relation(%dk)",
count_node/1000, count_way/1000, count_rel/1000);
}
static int parse_bbox(void)
{
int n;
if (!bbox)
return 0;
n = sscanf(bbox, "%lf,%lf,%lf,%lf", &minlon, &minlat, &maxlon, &maxlat);
if (n != 4) {
fprintf(stderr, "Bounding box must be specified like: minlon,minlat,maxlon,maxlat\n");
return 1;
}
if (maxlon <= minlon) {
fprintf(stderr, "Bounding box failed due to maxlon <= minlon\n");
return 1;
}
if (maxlat <= minlat) {
fprintf(stderr, "Bounding box failed due to maxlat <= minlat\n");
return 1;
}
printf("Applying Bounding box: %f,%f to %f,%f\n", minlon,minlat,maxlon,maxlat);
return 0;
}
static int node_wanted(double lat, double lon)
{
if (!bbox)
return 1;
if (lat < minlat || lat > maxlat)
return 0;
if (lon < minlon || lon > maxlon)
return 0;
return 1;
}
static filetypes_t filetype = FILETYPE_NONE;
static actions_t action = ACTION_NONE;
/* Parses the action="foo" tags in JOSM change files. Obvisouly not useful from osmChange files */
static actions_t ParseAction( xmlTextReaderPtr reader )
{
if( filetype == FILETYPE_OSMCHANGE || filetype == FILETYPE_PLANETDIFF )
return action;
actions_t new_action = ACTION_NONE;
xmlChar *action = xmlTextReaderGetAttribute( reader, BAD_CAST "action" );
if( action == NULL )
new_action = ACTION_CREATE;
else if( strcmp((char *)action, "modify") == 0 )
new_action = ACTION_MODIFY;
else if( strcmp((char *)action, "delete") == 0 )
new_action = ACTION_DELETE;
else
{
fprintf( stderr, "Unknown value for action: %s\n", (char*)action );
exit_nicely();
}
return new_action;
}
void StartElement(xmlTextReaderPtr reader, const xmlChar *name)
{
xmlChar *xid, *xlat, *xlon, *xk, *xv, *xrole, *xtype;
char *k;
if (filetype == FILETYPE_NONE)
{
if (xmlStrEqual(name, BAD_CAST "osm"))
{
filetype = FILETYPE_OSM;
action = ACTION_CREATE;
}
else if (xmlStrEqual(name, BAD_CAST "osmChange"))
{
filetype = FILETYPE_OSMCHANGE;
action = ACTION_NONE;
}
else if (xmlStrEqual(name, BAD_CAST "planetdiff"))
{
filetype = FILETYPE_PLANETDIFF;
action = ACTION_NONE;
}
else
{
fprintf( stderr, "Unknown XML document type: %s\n", name );
exit_nicely();
}
return;
}
if (xmlStrEqual(name, BAD_CAST "node")) {
xid = xmlTextReaderGetAttribute(reader, BAD_CAST "id");
xlon = xmlTextReaderGetAttribute(reader, BAD_CAST "lon");
xlat = xmlTextReaderGetAttribute(reader, BAD_CAST "lat");
assert(xid); assert(xlon); assert(xlat);
osm_id = strtol((char *)xid, NULL, 10);
node_lon = strtod((char *)xlon, NULL);
node_lat = strtod((char *)xlat, NULL);
action = ParseAction( reader );
if (osm_id > max_node)
max_node = osm_id;
count_node++;
if (count_node%10000 == 0)
printStatus();
xmlFree(xid);
xmlFree(xlon);
xmlFree(xlat);
} else if (xmlStrEqual(name, BAD_CAST "tag")) {
xk = xmlTextReaderGetAttribute(reader, BAD_CAST "k");
assert(xk);
/* 'created_by' and 'source' are common and not interesting to mapnik renderer */
if (strcmp((char *)xk, "created_by") && strcmp((char *)xk, "source")) {
char *p;
xv = xmlTextReaderGetAttribute(reader, BAD_CAST "v");
assert(xv);
k = (char *)xmlStrdup(xk);
while ((p = strchr(k, ' ')))
*p = '_';
addItem(&tags, k, (char *)xv, 0);
xmlFree(k);
xmlFree(xv);
}
xmlFree(xk);
} else if (xmlStrEqual(name, BAD_CAST "way")) {
xid = xmlTextReaderGetAttribute(reader, BAD_CAST "id");
assert(xid);
osm_id = strtol((char *)xid, NULL, 10);
action = ParseAction( reader );
if (osm_id > max_way)
max_way = osm_id;
count_way++;
if (count_way%1000 == 0)
printStatus();
nd_count = 0;
xmlFree(xid);
} else if (xmlStrEqual(name, BAD_CAST "nd")) {
xid = xmlTextReaderGetAttribute(reader, BAD_CAST "ref");
assert(xid);
nds[nd_count++] = strtol( (char *)xid, NULL, 10 );
if( nd_count >= nd_max )
realloc_nodes();
xmlFree(xid);
} else if (xmlStrEqual(name, BAD_CAST "relation")) {
xid = xmlTextReaderGetAttribute(reader, BAD_CAST "id");
assert(xid);
osm_id = strtol((char *)xid, NULL, 10);
action = ParseAction( reader );
if (osm_id > max_rel)
max_rel = osm_id;
count_rel++;
if (count_rel%1000 == 0)
printStatus();
member_count = 0;
xmlFree(xid);
} else if (xmlStrEqual(name, BAD_CAST "member")) {
xrole = xmlTextReaderGetAttribute(reader, BAD_CAST "role");
assert(xrole);
xtype = xmlTextReaderGetAttribute(reader, BAD_CAST "type");
assert(xtype);
xid = xmlTextReaderGetAttribute(reader, BAD_CAST "ref");
assert(xid);
members[member_count].id = strtol( (char *)xid, NULL, 0 );
members[member_count].role = strdup( (char *)xrole );
/* Currently we are only interested in 'way' members since these form polygons with holes */
if (xmlStrEqual(xtype, BAD_CAST "way"))
members[member_count].type = OSMTYPE_WAY;
if (xmlStrEqual(xtype, BAD_CAST "node"))
members[member_count].type = OSMTYPE_NODE;
if (xmlStrEqual(xtype, BAD_CAST "relation"))
members[member_count].type = OSMTYPE_RELATION;
member_count++;
if( member_count >= member_max )
realloc_members();
xmlFree(xid);
xmlFree(xrole);
xmlFree(xtype);
} else if (xmlStrEqual(name, BAD_CAST "add") ||
xmlStrEqual(name, BAD_CAST "create")) {
action = ACTION_CREATE;
action = ACTION_MODIFY; // Turns all creates into modifies, makes it resiliant against inconsistant snapshots.
} else if (xmlStrEqual(name, BAD_CAST "modify")) {
action = ACTION_MODIFY;
} else if (xmlStrEqual(name, BAD_CAST "delete")) {
action = ACTION_DELETE;
} else if (xmlStrEqual(name, BAD_CAST "bound")) {
/* ignore */
} else if (xmlStrEqual(name, BAD_CAST "bounds")) {
/* ignore */
} else if (xmlStrEqual(name, BAD_CAST "changeset")) {
/* ignore */
} else {
fprintf(stderr, "%s: Unknown element name: %s\n", __FUNCTION__, name);
}
// Collect extra attribute information and add as tags
if (extra_attributes && (xmlStrEqual(name, BAD_CAST "node") ||
xmlStrEqual(name, BAD_CAST "way") ||
xmlStrEqual(name, BAD_CAST "relation")))
{
xmlChar *xtmp;
xtmp = xmlTextReaderGetAttribute(reader, BAD_CAST "user");
if (xtmp) {
addItem(&tags, "osm_user", (char *)xtmp, 0);
xmlFree(xtmp);
}
xtmp = xmlTextReaderGetAttribute(reader, BAD_CAST "uid");
if (xtmp) {
addItem(&tags, "osm_uid", (char *)xtmp, 0);
xmlFree(xtmp);
}
xtmp = xmlTextReaderGetAttribute(reader, BAD_CAST "version");
if (xtmp) {
addItem(&tags, "osm_version", (char *)xtmp, 0);
xmlFree(xtmp);
}
xtmp = xmlTextReaderGetAttribute(reader, BAD_CAST "timestamp");
if (xtmp) {
addItem(&tags, "osm_timestamp", (char *)xtmp, 0);
xmlFree(xtmp);
}
}
}
static void resetMembers()
{
int i;
for(i=0; i<member_count; i++ )
free( members[i].role );
}
void EndElement(const xmlChar *name)
{
if (xmlStrEqual(name, BAD_CAST "node")) {
if (node_wanted(node_lat, node_lon)) {
reproject(&node_lat, &node_lon);
if( action == ACTION_CREATE )
out->node_add(osm_id, node_lat, node_lon, &tags);
else if( action == ACTION_MODIFY )
out->node_modify(osm_id, node_lat, node_lon, &tags);
else if( action == ACTION_DELETE )
out->node_delete(osm_id);
else
{
fprintf( stderr, "Don't know action for node %d\n", osm_id );
exit_nicely();
}
}
resetList(&tags);
} else if (xmlStrEqual(name, BAD_CAST "way")) {
if( action == ACTION_CREATE )
out->way_add(osm_id, nds, nd_count, &tags );
else if( action == ACTION_MODIFY )
out->way_modify(osm_id, nds, nd_count, &tags );
else if( action == ACTION_DELETE )
out->way_delete(osm_id);
else
{
fprintf( stderr, "Don't know action for way %d\n", osm_id );
exit_nicely();
}
resetList(&tags);
} else if (xmlStrEqual(name, BAD_CAST "relation")) {
if( action == ACTION_CREATE )
out->relation_add(osm_id, members, member_count, &tags);
else if( action == ACTION_MODIFY )
out->relation_modify(osm_id, members, member_count, &tags);
else if( action == ACTION_DELETE )
out->relation_delete(osm_id);
else
{
fprintf( stderr, "Don't know action for relation %d\n", osm_id );
exit_nicely();
}
resetList(&tags);
resetMembers();
} else if (xmlStrEqual(name, BAD_CAST "tag")) {
/* ignore */
} else if (xmlStrEqual(name, BAD_CAST "nd")) {
/* ignore */
} else if (xmlStrEqual(name, BAD_CAST "member")) {
/* ignore */
} else if (xmlStrEqual(name, BAD_CAST "osm")) {
printStatus();
filetype = FILETYPE_NONE;
} else if (xmlStrEqual(name, BAD_CAST "osmChange")) {
printStatus();
filetype = FILETYPE_NONE;
} else if (xmlStrEqual(name, BAD_CAST "planetdiff")) {
printStatus();
filetype = FILETYPE_NONE;
} else if (xmlStrEqual(name, BAD_CAST "bound")) {
/* ignore */
} else if (xmlStrEqual(name, BAD_CAST "bounds")) {
/* ignore */
} else if (xmlStrEqual(name, BAD_CAST "changeset")) {
/* ignore */
resetList(&tags); /* We may have accumulated some tags even if we ignored the changeset */
} else if (xmlStrEqual(name, BAD_CAST "add")) {
action = ACTION_NONE;
} else if (xmlStrEqual(name, BAD_CAST "create")) {
action = ACTION_NONE;
} else if (xmlStrEqual(name, BAD_CAST "modify")) {
action = ACTION_NONE;
} else if (xmlStrEqual(name, BAD_CAST "delete")) {
action = ACTION_NONE;
} else {
fprintf(stderr, "%s: Unknown element name: %s\n", __FUNCTION__, name);
}
}
static void processNode(xmlTextReaderPtr reader) {
xmlChar *name;
name = xmlTextReaderName(reader);
if (name == NULL)
name = xmlStrdup(BAD_CAST "--");
switch(xmlTextReaderNodeType(reader)) {
case XML_READER_TYPE_ELEMENT:
StartElement(reader, name);
if (xmlTextReaderIsEmptyElement(reader))
EndElement(name); /* No end_element for self closing tags! */
break;
case XML_READER_TYPE_END_ELEMENT:
EndElement(name);
break;
case XML_READER_TYPE_SIGNIFICANT_WHITESPACE:
/* Ignore */
break;
default:
fprintf(stderr, "Unknown node type %d\n", xmlTextReaderNodeType(reader));
break;
}
xmlFree(name);
}
static int streamFile(char *filename, int sanitize) {
xmlTextReaderPtr reader;
int ret = 0;
if (sanitize)
reader = sanitizerOpen(filename);
else
reader = inputUTF8(filename);
if (reader != NULL) {
ret = xmlTextReaderRead(reader);
while (ret == 1) {
processNode(reader);
ret = xmlTextReaderRead(reader);
}
if (ret != 0) {
fprintf(stderr, "%s : failed to parse\n", filename);
return ret;
}
xmlFreeTextReader(reader);
} else {
fprintf(stderr, "Unable to open %s\n", filename);
return 1;
}
return 0;
}
void exit_nicely(void)
{
fprintf(stderr, "Error occurred, cleaning up\n");
out->cleanup();
exit(1);
}
static void short_usage(char *arg0)
{
const char *name = basename(arg0);
fprintf(stderr, "Usage error. For further information see:\n");
fprintf(stderr, "\t%s -h|--help\n", name);
}
static void long_usage(char *arg0)
{
int i;
const char *name = basename(arg0);
fprintf(stderr, "Usage:\n");
fprintf(stderr, "\t%s [options] planet.osm\n", name);
fprintf(stderr, "\t%s [options] planet.osm.{gz,bz2}\n", name);
fprintf(stderr, "\t%s [options] file1.osm file2.osm file3.osm\n", name);
fprintf(stderr, "\nThis will import the data from the OSM file(s) into a PostgreSQL database\n");
fprintf(stderr, "suitable for use by the Mapnik renderer\n");
fprintf(stderr, "\nOptions:\n");
fprintf(stderr, " -a|--append\t\tAdd the OSM file into the database without removing\n");
fprintf(stderr, " \t\texisting data.\n");
fprintf(stderr, " -b|--bbox\t\tApply a bounding box filter on the imported data\n");
fprintf(stderr, " \t\tMust be specified as: minlon,minlat,maxlon,maxlat\n");
fprintf(stderr, " \t\te.g. --bbox -0.5,51.25,0.5,51.75\n");
fprintf(stderr, " -c|--create\t\tRemove existing data from the database. This is the \n");
fprintf(stderr, " \t\tdefault if --append is not specified.\n");
fprintf(stderr, " -d|--database\tThe name of the PostgreSQL database to connect\n");
fprintf(stderr, " \tto (default: gis).\n");
fprintf(stderr, " -l|--latlong\t\tStore data in degrees of latitude & longitude.\n");
fprintf(stderr, " -m|--merc\t\tStore data in proper spherical mercator (default)\n");
fprintf(stderr, " -M|--oldmerc\t\tStore data in the legacy OSM mercator format\n");
fprintf(stderr, " -E|--proj num\tUse projection EPSG:num\n");
fprintf(stderr, " -u|--utf8-sanitize\tRepair bad UTF8 input data (present in planet\n");
fprintf(stderr, " \tdumps prior to August 2007). Adds about 10%% overhead.\n");
fprintf(stderr, " -p|--prefix\t\tPrefix for table names (default planet_osm)\n");
fprintf(stderr, " -s|--slim\t\tStore temporary data in the database. This greatly\n");
fprintf(stderr, " \t\treduces the RAM usage but is much slower.\n");
fprintf(stderr, " -S|--style\t\tLocation of the style file. Defaults to /usr/share/osm2pgsql/default.style\n");
fprintf(stderr, " -C|--cache\t\tOnly for slim mode: Use upto this many MB for caching nodes\n");
fprintf(stderr, " \t\tDefault is 800\n");
fprintf(stderr, " -U|--username\tPostgresql user name.\n");
fprintf(stderr, " -W|--password\tForce password prompt.\n");
fprintf(stderr, " -H|--host\t\tDatabase server hostname or socket location.\n");
fprintf(stderr, " -P|--port\t\tDatabase server port.\n");
fprintf(stderr, " -e|--expire-tiles [min_zoom-]max_zoom\tCreate a tile expiry list.\n");
fprintf(stderr, " -o|--expire-output filename\tOutput filename for expired tiles list.\n");
fprintf(stderr, " -O|--output\t\tOutput backend.\n");
fprintf(stderr, " \t\tpgsql - Output to a PostGIS database. (default)\n");
fprintf(stderr, " \t\tgazetteer - Output to a PostGIS database suitable for gazetteer\n");
fprintf(stderr, " \t\tnull - No output. Useful for testing.\n");
fprintf(stderr, " -x|--extra-attributes\n");
fprintf(stderr, " \t\tInclude attributes for each object in the database.\n");
fprintf(stderr, " \t\tThis includes the username, userid, timestamp and version.\n");
fprintf(stderr, " \t\tNote: this option also requires additional entries in your style file.\n");
fprintf(stderr, " -G|--multi-geometry\t\tGenerate multi-geometry features in postgresql tables.\n");
fprintf(stderr, " -h|--help\t\tHelp information.\n");
fprintf(stderr, " -v|--verbose\t\tVerbose output.\n");
fprintf(stderr, "\n");
if(!verbose)
{
fprintf(stderr, "Add -v to display supported projections.\n");
fprintf(stderr, "Use -E to access any espg projections (usually in /usr/share/proj/epsg)\n" );
}
else
{
fprintf(stderr, "Supported projections:\n" );
for(i=0; i<PROJ_COUNT; i++ )
{
fprintf( stderr, "%-20s(%2s) SRS:%6d %s\n",
Projection_Info[i].descr, Projection_Info[i].option, Projection_Info[i].srs, Projection_Info[i].proj4text);
}
}
}
const char *build_conninfo(const char *db, const char *username, const char *password, const char *host, const char *port)
{
static char conninfo[1024];
conninfo[0]='\0';
strcat(conninfo, "dbname='");
strcat(conninfo, db);
strcat(conninfo, "'");
if (username) {
strcat(conninfo, " user='");
strcat(conninfo, username);
strcat(conninfo, "'");
}
if (password) {
strcat(conninfo, " password='");
strcat(conninfo, password);
strcat(conninfo, "'");
}
if (host) {
strcat(conninfo, " host='");
strcat(conninfo, host);
strcat(conninfo, "'");
}
if (port) {
strcat(conninfo, " port='");
strcat(conninfo, port);
strcat(conninfo, "'");
}
return conninfo;
}
static void realloc_nodes()
{
if( nd_max == 0 )
nd_max = INIT_MAX_NODES;
else
nd_max <<= 1;
nds = realloc( nds, nd_max * sizeof( nds[0] ) );
if( !nds )
{
fprintf( stderr, "Failed to expand node list to %d\n", nd_max );
exit_nicely();
}
}
static void realloc_members()
{
if( member_max == 0 )
member_max = INIT_MAX_NODES;
else
member_max <<= 1;
members = realloc( members, member_max * sizeof( members[0] ) );
if( !members )
{
fprintf( stderr, "Failed to expand member list to %d\n", member_max );
exit_nicely();
}
}
int main(int argc, char *argv[])
{
int append=0;
int create=0;
int slim=0;
int sanitize=0;
int long_usage_bool=0;
int pass_prompt=0;
int projection = PROJ_SPHERE_MERC;
int expire_tiles_zoom = -1;
int expire_tiles_zoom_min = -1;
int enable_multi = 0;
const char *expire_tiles_filename = "dirty_tiles";
const char *db = "gis";
const char *username=NULL;
const char *host=NULL;
const char *password=NULL;
const char *port = "5432";
const char *conninfo = NULL;
const char *prefix = "planet_osm";
const char *style = "/usr/share/osm2pgsql/default.style";
const char *temparg;
const char *output_backend = "pgsql";
int cache = 800;
struct output_options options;
PGconn *sql_conn;
fprintf(stderr, "osm2pgsql SVN version %s\n\n", VERSION);
while (1) {
int c, option_index = 0;
static struct option long_options[] = {
{"append", 0, 0, 'a'},
{"bbox", 1, 0, 'b'},
{"create", 0, 0, 'c'},
{"database", 1, 0, 'd'},
{"latlong", 0, 0, 'l'},
{"verbose", 0, 0, 'v'},
{"slim", 0, 0, 's'},
{"prefix", 1, 0, 'p'},
{"proj", 1, 0, 'E'},
{"merc", 0, 0, 'm'},
{"oldmerc", 0, 0, 'M'},
{"utf8-sanitize", 0, 0, 'u'},
{"cache", 1, 0, 'C'},
{"username", 1, 0, 'U'},
{"password", 0, 0, 'W'},
{"host", 1, 0, 'H'},
{"port", 1, 0, 'P'},
{"help", 0, 0, 'h'},
{"style", 1, 0, 'S'},
{"expire-tiles", 1, 0, 'e'},
{"expire-output", 1, 0, 'o'},
{"output", 1, 0, 'O'},
{"extra-attributes", 0, 0, 'x'},
{"multi-geometry", 0, 0, 'G'},
{0, 0, 0, 0}
};
c = getopt_long (argc, argv, "ab:cd:hlmMp:suvU:WH:P:E:C:S:e:o:O:xG", long_options, &option_index);
if (c == -1)
break;
switch (c) {
case 'a': append=1; break;
case 'b': bbox=optarg; break;
case 'c': create=1; break;
case 'v': verbose=1; break;
case 's': slim=1; break;
case 'u': sanitize=1; break;
case 'l': projection=PROJ_LATLONG; break;
case 'm': projection=PROJ_SPHERE_MERC; break;
case 'M': projection=PROJ_MERC; break;
case 'E': projection=-atoi(optarg); break;
case 'p': prefix=optarg; break;
case 'd': db=optarg; break;
case 'C': cache = atoi(optarg); break;
case 'U': username=optarg; break;
case 'W': pass_prompt=1; break;
case 'H': host=optarg; break;
case 'P': port=optarg; break;
case 'S': style=optarg; break;
case 'e':
expire_tiles_zoom_min = atoi(optarg);
temparg = strchr(optarg, '-');
if (temparg) expire_tiles_zoom = atoi(temparg + 1);
if (expire_tiles_zoom < expire_tiles_zoom_min) expire_tiles_zoom = expire_tiles_zoom_min;
break;
case 'o': expire_tiles_filename=optarg; break;
case 'O': output_backend = optarg; break;
case 'x': extra_attributes=1; break;
case 'G': enable_multi=1; break;
case 'h': long_usage_bool=1; break;
case '?':
default:
short_usage(argv[0]);
exit(EXIT_FAILURE);
}
}
if (long_usage_bool) {
long_usage(argv[0]);
exit(EXIT_FAILURE);
}
if (argc == optind) { // No non-switch arguments
short_usage(argv[0]);
exit(EXIT_FAILURE);
}
if (append && create) {
fprintf(stderr, "Error: --append and --create options can not be used at the same time!\n");
exit(EXIT_FAILURE);
}
if( cache < 0 ) cache = 0;
if (pass_prompt)
password = simple_prompt("Password:", 100, 0);
conninfo = build_conninfo(db, username, password, host, port);
sql_conn = PQconnectdb(conninfo);
if (PQstatus(sql_conn) != CONNECTION_OK) {
fprintf(stderr, "Connection to database failed: %s\n", PQerrorMessage(sql_conn));
exit(EXIT_FAILURE);
}
PQfinish(sql_conn);
text_init();
initList(&tags);
count_node = max_node = 0;
count_way = max_way = 0;
count_rel = max_rel = 0;
LIBXML_TEST_VERSION
project_init(projection);
fprintf(stderr, "Using projection SRS %d (%s)\n",
project_getprojinfo()->srs, project_getprojinfo()->descr );
if (parse_bbox())
return 1;
options.conninfo = conninfo;
options.prefix = prefix;
options.append = append;
options.slim = slim;
options.projection = project_getprojinfo()->srs;
options.scale = (projection==PROJ_LATLONG)?10000000:100;
options.mid = slim ? &mid_pgsql : &mid_ram;
options.cache = cache;
options.style = style;
options.expire_tiles_zoom = expire_tiles_zoom;
options.expire_tiles_zoom_min = expire_tiles_zoom_min;
options.expire_tiles_filename = expire_tiles_filename;
options.enable_multi = enable_multi;
if (strcmp("pgsql", output_backend) == 0) {
out = &out_pgsql;
} else if (strcmp("gazetteer", output_backend) == 0) {
out = &out_gazetteer;
} else if (strcmp("null", output_backend) == 0) {
out = &out_null;
} else {
fprintf(stderr, "Output backend `%s' not recognised. Should be one of [pgsql, gazetteer, null].\n", output_backend);
exit(EXIT_FAILURE);
}
out->start(&options);
realloc_nodes();
realloc_members();
while (optind < argc) {
fprintf(stderr, "\nReading in file: %s\n", argv[optind]);
if (streamFile(argv[optind], sanitize) != 0)
exit_nicely();
optind++;
}
xmlCleanupParser();
xmlMemoryDump();
if (count_node || count_way || count_rel) {
fprintf(stderr, "\n");
fprintf(stderr, "Node stats: total(%d), max(%d)\n", count_node, max_node);
fprintf(stderr, "Way stats: total(%d), max(%d)\n", count_way, max_way);
fprintf(stderr, "Relation stats: total(%d), max(%d)\n", count_rel, max_rel);
}
out->stop();
free(nds);
free(members);
project_exit();
text_exit();
fprintf(stderr, "\n");
return 0;
}
Reference in New Issue View Git Blame Copy Permalink