123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976977978979980981982983984985986987988989990991992993994995996997998999100010011002100310041005100610071008100910101011101210131014101510161017101810191020102110221023102410251026102710281029103010311032103310341035103610371038103910401041104210431044104510461047104810491050105110521053105410551056105710581059106010611062106310641065106610671068106910701071107210731074107510761077107810791080108110821083108410851086108710881089109010911092109310941095109610971098109911001101110211031104110511061107110811091110111111121113111411151116111711181119112011211122112311241125112611271128112911301131113211331134113511361137113811391140114111421143114411451146114711481149115011511152115311541155115611571158115911601161116211631164116511661167116811691170117111721173117411751176117711781179118011811182118311841185118611871188118911901191119211931194119511961197119811991200120112021203120412051206120712081209121012111212121312141215121612171218121912201221122212231224122512261227122812291230123112321233123412351236123712381239124012411242124312441245124612471248124912501251125212531254125512561257125812591260126112621263126412651266126712681269127012711272127312741275127612771278127912801281128212831284128512861287128812891290129112921293129412951296129712981299130013011302130313041305130613071308130913101311131213131314131513161317131813191320132113221323132413251326132713281329133013311332133313341335133613371338133913401341134213431344134513461347134813491350135113521353135413551356135713581359136013611362136313641365136613671368136913701371137213731374137513761377137813791380138113821383138413851386138713881389139013911392139313941395139613971398139914001401140214031404140514061407140814091410141114121413141414151416141714181419142014211422142314241425142614271428142914301431143214331434143514361437143814391440144114421443144414451446144714481449145014511452145314541455145614571458145914601461146214631464146514661467146814691470147114721473147414751476147714781479148014811482148314841485148614871488148914901491149214931494149514961497149814991500150115021503150415051506150715081509151015111512151315141515151615171518151915201521152215231524152515261527152815291530153115321533153415351536153715381539154015411542154315441545154615471548154915501551155215531554155515561557155815591560156115621563156415651566156715681569157015711572157315741575157615771578157915801581158215831584158515861587158815891590159115921593159415951596159715981599160016011602160316041605160616071608160916101611161216131614161516161617161816191620162116221623162416251626162716281629163016311632163316341635163616371638163916401641164216431644164516461647164816491650165116521653165416551656 |
- /*!
- \file lib/vector/Vlib/open_pg.c
- \brief Vector library - Open PostGIS layer as vector map layer
- Higher level functions for reading/writing/manipulating vectors.
- (C) 2011-2012 by the GRASS Development Team
- This program is free software under the GNU General Public License
- (>=v2). Read the file COPYING that comes with GRASS for details.
- \author Martin Landa <landa.martin gmail.com>
- */
- #include <string.h>
- #include <stdlib.h>
- #include <grass/vector.h>
- #include <grass/dbmi.h>
- #include <grass/glocale.h>
- #ifdef HAVE_POSTGRES
- #include "pg_local_proto.h"
- #define TOPOGEOM_COLUMN "topo"
- struct edge_data {
- int id;
- int start_node;
- int end_node;
- int left_face;
- int right_face;
- char *wkb_geom;
- };
- static char *get_key_column(struct Format_info_pg *);
- static SF_FeatureType ftype_from_string(const char *);
- static int drop_table(struct Format_info_pg *);
- static int check_schema(const struct Format_info_pg *);
- static int create_table(struct Format_info_pg *, const struct field_info *);
- static int create_topo_schema(struct Format_info_pg *, int);
- static void connect_db(struct Format_info_pg *);
- static int check_topo(struct Format_info_pg *, struct Plus_head *);
- static int parse_bbox(const char *, struct bound_box *);
- static int num_of_records(const struct Format_info_pg *, const char *);
- static struct P_node *read_p_node(struct Plus_head *, int, int,
- const char *, struct Format_info_pg *);
- static struct P_line *read_p_line(struct Plus_head *, int,
- const struct edge_data *,
- struct Format_info_cache *);
- static int load_plus_head(struct Format_info_pg *, struct Plus_head *);
- static void notice_processor(void *, const char *);
- static char *get_sftype(SF_FeatureType);
- #endif
- /*!
- \brief Open vector map - PostGIS feature table (level 1 - without topology)
- \todo Check database instead of geometry_columns
- \param[in,out] Map pointer to Map_info structure
- \param update TRUE for write mode, otherwise read-only
- \return 0 success
- \return -1 error
- */
- int V1_open_old_pg(struct Map_info *Map, int update)
- {
- #ifdef HAVE_POSTGRES
- int found;
- char stmt[DB_SQL_MAX];
- PGresult *res;
- struct Format_info_pg *pg_info;
- G_debug(2, "V1_open_old_pg(): update = %d", update);
-
- pg_info = &(Map->fInfo.pg);
- if (!pg_info->conninfo) {
- G_warning(_("Connection string not defined"));
- return -1;
- }
- if (!pg_info->table_name) {
- G_warning(_("PostGIS feature table not defined"));
- return -1;
- }
- G_debug(1, "V1_open_old_pg(): conninfo='%s' table='%s'",
- pg_info->conninfo, pg_info->table_name);
- /* connect database */
- if (!pg_info->conn)
- connect_db(pg_info);
-
- /* get DB name */
- pg_info->db_name = G_store(PQdb(pg_info->conn));
- if (!pg_info->db_name) {
- G_warning(_("Unable to get database name"));
- return -1;
- }
- /* get fid and geometry column */
- sprintf(stmt, "SELECT f_geometry_column, coord_dimension, srid, type "
- "FROM geometry_columns WHERE f_table_schema = '%s' AND "
- "f_table_name = '%s'", pg_info->schema_name, pg_info->table_name);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
- G_fatal_error("%s\n%s", _("No feature tables found in database."),
- PQresultErrorMessage(res));
- found = PQntuples(res) > 0 ? TRUE : FALSE;
- if (found) {
- /* geometry column */
- pg_info->geom_column = G_store(PQgetvalue(res, 0, 0));
- G_debug(3, "\t-> table = %s column = %s", pg_info->table_name,
- pg_info->geom_column);
- /* fid column */
- pg_info->fid_column = get_key_column(pg_info);
- /* coordinates dimension */
- pg_info->coor_dim = atoi(PQgetvalue(res, 0, 1));
- /* SRS ID */
- pg_info->srid = atoi(PQgetvalue(res, 0, 2));
- /* feature type */
- pg_info->feature_type = ftype_from_string(PQgetvalue(res, 0, 3));
- }
- PQclear(res);
- /* no feature in cache */
- pg_info->cache.fid = -1;
- if (!found) {
- G_warning(_("Feature table <%s> not found in 'geometry_columns'"),
- pg_info->table_name);
- return -1;
- }
- /* check for topo schema */
- check_topo(pg_info, &(Map->plus));
-
- return 0;
- #else
- G_fatal_error(_("GRASS is not compiled with PostgreSQL support"));
- return -1;
- #endif
- }
- /*!
- \brief Open vector map - PostGIS feature table (level 2 - feature index)
- \param[in,out] Map pointer to Map_info structure
- \return 0 success
- \return -1 error
- */
- int V2_open_old_pg(struct Map_info *Map)
- {
- #ifdef HAVE_POSTGRES
- struct Format_info_pg *pg_info;
-
- G_debug(3, "V2_open_old_pg(): name = %s mapset = %s", Map->name,
- Map->mapset);
- pg_info = &(Map->fInfo.pg);
-
- if (pg_info->toposchema_name)
- /* no fidx file needed for PostGIS topology access */
- return 0;
-
- if (Vect_open_fidx(Map, &(pg_info->offset)) != 0) {
- G_warning(_("Unable to open feature index file for vector map <%s>"),
- Vect_get_full_name(Map));
- G_zero(&(pg_info->offset), sizeof(struct Format_info_offset));
- }
- return 0;
- #else
- G_fatal_error(_("GRASS is not compiled with PostgreSQL support"));
- return -1;
- #endif
- }
- /*!
- \brief Prepare PostGIS database for creating new feature table
- (level 1)
- \todo To implement
- \param[out] Map pointer to Map_info structure
- \param name name of PostGIS feature table to create
- \param with_z WITH_Z for 3D vector data otherwise WITHOUT_Z
- \return 0 success
- \return -1 error
- */
- int V1_open_new_pg(struct Map_info *Map, const char *name, int with_z)
- {
- #ifdef HAVE_POSTGRES
- char stmt[DB_SQL_MAX];
- struct Format_info_pg *pg_info;
- PGresult *res;
- G_debug(2, "V1_open_new_pg(): name = %s with_z = %d", name, with_z);
- pg_info = &(Map->fInfo.pg);
- if (!pg_info->conninfo) {
- G_warning(_("Connection string not defined"));
- return -1;
- }
- if (!pg_info->table_name) {
- G_warning(_("PostGIS feature table not defined"));
- return -1;
- }
- G_debug(1, "V1_open_new_pg(): conninfo='%s' table='%s'",
- pg_info->conninfo, pg_info->table_name);
- /* connect database */
- connect_db(pg_info);
-
- /* get DB name */
- pg_info->db_name = G_store(PQdb(pg_info->conn));
- if (!pg_info->db_name) {
- G_warning(_("Unable to get database name"));
- return -1;
- }
- /* if schema not defined, use 'public' */
- if (!pg_info->schema_name) {
- pg_info->schema_name = G_store("public");
- }
- /* if fid_column not defined, use 'ogc_fid' */
- if (!pg_info->fid_column) {
- pg_info->fid_column = G_store(FID_COLUMN);
- }
- /* if geom_column not defined, use 'wkb_geometry' */
- if (!pg_info->geom_column) {
- pg_info->geom_column = G_store(GEOMETRY_COLUMN);
- }
- /* check if feature table already exists */
- sprintf(stmt, "SELECT * FROM pg_tables "
- "WHERE schemaname = '%s' AND tablename = '%s'",
- pg_info->schema_name, pg_info->table_name);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK)
- G_fatal_error("%s\n%s", _("No feature tables found in database."),
- PQresultErrorMessage(res));
- if (PQntuples(res) > 0) {
- /* table found */
- if (G_get_overwrite()) {
- G_warning(_("PostGIS layer <%s.%s> already exists and will be overwritten"),
- pg_info->schema_name, pg_info->table_name);
- if (drop_table(pg_info) == -1) {
- G_warning(_("Unable to delete PostGIS layer <%s>"),
- pg_info->table_name);
- return -1;
- }
- }
- else {
- G_fatal_error(_("PostGIS layer <%s.%s> already exists in database '%s'"),
- pg_info->schema_name, pg_info->table_name,
- pg_info->db_name);
- return -1;
- }
- }
- /* no feature in cache */
- pg_info->cache.fid = -1;
- /* unknown feature type */
- pg_info->feature_type = SF_UNKNOWN;
- PQclear(res);
- return 0;
- #else
- G_fatal_error(_("GRASS is not compiled with PostgreSQL support"));
- return -1;
- #endif
- }
- /*!
- \brief Create new PostGIS layer in given database (level 2)
- V1_open_new_pg() is required to be called before this function.
- List of currently supported types:
- - GV_POINT (SF_POINT)
- - GV_LINE (SF_LINESTRING)
- - GV_BOUNDARY (SF_POLYGON)
- \param[in,out] Map pointer to Map_info structure
- \param type feature type (GV_POINT, GV_LINE, ...)
- \return 0 success
- \return -1 error
- */
- int V2_open_new_pg(struct Map_info *Map, int type)
- {
- #ifdef HAVE_POSTGRES
- int ndblinks;
- struct Format_info_pg *pg_info;
- struct field_info *Fi;
- struct Key_Value *projinfo, *projunits;
- Fi = NULL;
- pg_info = &(Map->fInfo.pg);
- if (!pg_info->conninfo) {
- G_warning(_("Connection string not defined"));
- return -1;
- }
- if (!pg_info->table_name) {
- G_warning(_("PostGIS feature table not defined"));
- return -1;
- }
- G_debug(1, "V2_open_new_pg(): conninfo='%s' table='%s' -> type = %d",
- pg_info->conninfo, pg_info->table_name, type);
- /* get spatial reference */
- projinfo = G_get_projinfo();
- projunits = G_get_projunits();
- pg_info->srid = 0; /* TODO */
- // Ogr_spatial_ref = GPJ_grass_to_osr(projinfo, projunits);
- G_free_key_value(projinfo);
- G_free_key_value(projunits);
- /* determine geometry type */
- switch (type) {
- case GV_POINT:
- pg_info->feature_type = SF_POINT;
- break;
- case GV_LINE:
- pg_info->feature_type = SF_LINESTRING;
- break;
- case GV_BOUNDARY:
- pg_info->feature_type = SF_POLYGON;
- break;
- default:
- G_warning(_("Unsupported geometry type (%d)"), type);
- return -1;
- }
- /* coordinate dimension */
- pg_info->coor_dim = Vect_is_3d(Map) ? 3 : 2;
- /* create new PostGIS table */
- ndblinks = Vect_get_num_dblinks(Map);
- if (ndblinks > 0) {
- Fi = Vect_get_dblink(Map, 0);
- if (Fi) {
- if (ndblinks > 1)
- G_warning(_("More layers defined, using driver <%s> and "
- "database <%s>"), Fi->driver, Fi->database);
- }
- else {
- G_warning(_("Database connection not defined. "
- "Unable to write attributes."));
- }
- }
- /* create new feature table */
- if (create_table(pg_info, Fi) == -1) {
- G_warning(_("Unable to create new PostGIS feature table"));
- return -1;
- }
-
- /* create new topology schema (if PostGIS topology support is enabled) */
- if(pg_info->toposchema_name) {
- if (create_topo_schema(pg_info, Vect_is_3d(Map)) == -1) {
- G_warning(_("Unable to create new PostGIS topology schema"));
- return -1;
- }
- }
-
- if (Fi)
- G_free(Fi);
- return 0;
- #else
- G_fatal_error(_("GRASS is not compiled with PostgreSQL support"));
- return -1;
- #endif
- }
- /*!
- \brief Read full-topology for PostGIS links
-
- Note: Only 2D topological primitives are currently supported
-
- \param[in,out] Map pointer to Map_info structure
- \param head_only TRUE to read only header
-
- \return 0 on success
- \return 1 topology layer does not exist
- \return -1 on error
- */
- int Vect_open_topo_pg(struct Map_info *Map, int head_only)
- {
- #ifdef HAVE_POSTGRES
- struct Plus_head *plus;
- struct Format_info_pg *pg_info;
-
- Map->open = VECT_OPEN_CODE; /* needed by load_plus */
- plus = &(Map->plus);
- pg_info = &(Map->fInfo.pg);
-
- /* check for topo schema */
- if (check_topo(pg_info, plus) != 0)
- return 1;
-
- /* free and init plus structure */
- dig_init_plus(plus);
-
- return load_plus(Map, head_only);
- #else
- G_fatal_error(_("GRASS is not compiled with PostgreSQL support"));
- return -1;
- #endif
- }
- #ifdef HAVE_POSTGRES
- /*!
- \brief Get key column for feature table
- \param pg_info pointer to Format_info_pg
-
- \return string buffer with key column name
- \return NULL on missing key column
- */
- char *get_key_column(struct Format_info_pg *pg_info)
- {
- char *key_column;
- char stmt[DB_SQL_MAX];
- PGresult *res;
- sprintf(stmt,
- "SELECT kcu.column_name "
- "FROM INFORMATION_SCHEMA.TABLES t "
- "LEFT JOIN INFORMATION_SCHEMA.TABLE_CONSTRAINTS tc "
- "ON tc.table_catalog = t.table_catalog "
- "AND tc.table_schema = t.table_schema "
- "AND tc.table_name = t.table_name "
- "AND tc.constraint_type = 'PRIMARY KEY' "
- "LEFT JOIN INFORMATION_SCHEMA.KEY_COLUMN_USAGE kcu "
- "ON kcu.table_catalog = tc.table_catalog "
- "AND kcu.table_schema = tc.table_schema "
- "AND kcu.table_name = tc.table_name "
- "AND kcu.constraint_name = tc.constraint_name "
- "WHERE t.table_schema = '%s' AND t.table_name = '%s'",
- pg_info->schema_name, pg_info->table_name);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK ||
- PQntuples(res) != 1 || strlen(PQgetvalue(res, 0, 0)) < 1) {
- G_warning(_("No key column detected."));
- if (res)
- PQclear(res);
- return NULL;
- }
- key_column = G_store(PQgetvalue(res, 0, 0));
- PQclear(res);
- return key_column;
- }
- /*!
- \brief Get simple feature type from string
- \param type string
- \return SF type
- */
- SF_FeatureType ftype_from_string(const char *type)
- {
- SF_FeatureType sf_type;
-
- if (G_strcasecmp(type, "POINT") == 0)
- return SF_POINT;
- else if (G_strcasecmp(type, "LINESTRING") == 0)
- return SF_LINESTRING;
- else if (G_strcasecmp(type, "POLYGON") == 0)
- return SF_POLYGON;
- else if (G_strcasecmp(type, "MULTIPOINT") == 0)
- return SF_MULTIPOINT;
- else if (G_strcasecmp(type, "MULTILINESTRING") == 0)
- return SF_MULTILINESTRING;
- else if (G_strcasecmp(type, "MULTIPOLYGON") == 0)
- return SF_MULTIPOLYGON;
- else if (G_strcasecmp(type, "GEOMETRYCOLLECTION") == 0)
- return SF_GEOMETRYCOLLECTION;
- else
- return SF_UNKNOWN;
-
- G_debug(3, "ftype_from_string(): type='%s' -> %d", type, sf_type);
-
- return sf_type;
- }
- /*!
- \brief Drop feature table and topology schema if exists
- \param pg_info pointer to Format_info_pg
- \return -1 on error
- \return 0 on success
- */
- int drop_table(struct Format_info_pg *pg_info)
- {
- int i;
- char stmt[DB_SQL_MAX];
- char *topo_schema;
-
- PGresult *result, *result_drop;
-
- /* drop topology schema(s) related to the feature table */
- sprintf(stmt, "SELECT t.name FROM topology.layer AS l JOIN "
- "topology.topology AS t ON l.topology_id = t.id "
- "WHERE l.table_name = '%s'", pg_info->table_name);
- G_debug(2, "SQL: %s", stmt);
-
- result = PQexec(pg_info->conn, stmt);
- if (!result || PQresultStatus(result) != PGRES_TUPLES_OK) {
- G_warning(_("Execution failed: %s"), PQerrorMessage(pg_info->conn));
- PQclear(result);
- return -1;
- }
- for (i = 0; i < PQntuples(result); i++) {
- topo_schema = PQgetvalue(result, i, 0);
- sprintf(stmt, "SELECT topology.DropTopology('%s')",
- topo_schema);
- G_debug(2, "SQL: %s", stmt);
-
- result_drop = PQexec(pg_info->conn, stmt);
- if (!result_drop || PQresultStatus(result_drop) != PGRES_TUPLES_OK)
- G_warning(_("Execution failed: %s"), PQerrorMessage(pg_info->conn));
-
- G_verbose_message(_("PostGIS topology schema <%s> dropped"),
- topo_schema);
- PQclear(result_drop);
- }
- PQclear(result);
-
- /* drop feature table */
- sprintf(stmt, "DROP TABLE \"%s\".\"%s\"",
- pg_info->schema_name, pg_info->table_name);
- G_debug(2, "SQL: %s", stmt);
-
- if (execute(pg_info->conn, stmt) == -1) {
- return -1;
- }
- return 0;
- }
- /*!
- \brief Creates new schema for feature table if not exists
- \param pg_info pointer to Format_info_pg
- \return -1 on error
- \return 0 on success
- */
- int check_schema(const struct Format_info_pg *pg_info)
- {
- int i, found, nschema;
- char stmt[DB_SQL_MAX];
- PGresult *result;
- /* add geometry column */
- sprintf(stmt, "SELECT nspname FROM pg_namespace");
- G_debug(2, "SQL: %s", stmt);
- result = PQexec(pg_info->conn, stmt);
- if (!result || PQresultStatus(result) != PGRES_TUPLES_OK) {
- PQclear(result);
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
- found = FALSE;
- nschema = PQntuples(result);
- for (i = 0; i < nschema && !found; i++) {
- if (strcmp(pg_info->schema_name, PQgetvalue(result, i, 0)) == 0)
- found = TRUE;
- }
- PQclear(result);
- if (!found) {
- sprintf(stmt, "CREATE SCHEMA %s", pg_info->schema_name);
- if (execute(pg_info->conn, stmt) == -1) {
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
- G_warning(_("Schema <%s> doesn't exist, created"),
- pg_info->schema_name);
- }
- return 0;
- }
- /*!
- \brief Create new feature table
- \param pg_info pointer to Format_info_pg
- \param Fi pointer to field_info
- \return -1 on error
- \return 0 on success
- */
- int create_table(struct Format_info_pg *pg_info, const struct field_info *Fi)
- {
- int spatial_index, primary_key;
- char stmt[DB_SQL_MAX];
- char *geom_type, *def_file;
-
- PGresult *result;
- def_file = getenv("GRASS_VECTOR_PGFILE");
-
- /* by default create spatial index & add primary key */
- spatial_index = primary_key = TRUE;
- if (G_find_file2("", def_file ? def_file : "PG", G_mapset())) {
- FILE *fp;
- const char *p;
- struct Key_Value *key_val;
- fp = G_fopen_old("", def_file ? def_file : "PG", G_mapset());
- if (!fp) {
- G_fatal_error(_("Unable to open PG file"));
- }
- key_val = G_fread_key_value(fp);
- fclose(fp);
- /* disable spatial index ? */
- p = G_find_key_value("spatial_index", key_val);
- if (p && G_strcasecmp(p, "off") == 0)
- spatial_index = FALSE;
-
- /* disable primary key ? */
- p = G_find_key_value("primary_key", key_val);
- if (p && G_strcasecmp(p, "off") == 0)
- primary_key = FALSE;
- /* PostGIS topology enabled ? */
- p = G_find_key_value("topology", key_val);
- if (p && G_strcasecmp(p, "on") == 0) {
- /* define topology name
- this should be configurable by the user
- */
- G_asprintf(&(pg_info->toposchema_name), "topo_%s",
- pg_info->table_name);
- }
- }
- /* create schema if not exists */
- if (G_strcasecmp(pg_info->schema_name, "public") != 0) {
- if (check_schema(pg_info) != 0)
- return -1;
- }
- /* prepare CREATE TABLE statement */
- sprintf(stmt, "CREATE TABLE \"%s\".\"%s\" (%s SERIAL",
- pg_info->schema_name, pg_info->table_name, pg_info->fid_column);
- if (Fi) {
- /* append attributes */
- int col, ncols, sqltype, length, ctype;
- char stmt_col[DB_SQL_MAX];
- const char *colname;
- dbString dbstmt;
- dbHandle handle;
- dbDriver *driver;
- dbCursor cursor;
- dbTable *table;
- dbColumn *column;
- db_init_string(&dbstmt);
- db_init_handle(&handle);
- pg_info->dbdriver = driver = db_start_driver(Fi->driver);
- if (!driver) {
- G_warning(_("Unable to start driver <%s>"), Fi->driver);
- return -1;
- }
- db_set_handle(&handle, Fi->database, NULL);
- if (db_open_database(driver, &handle) != DB_OK) {
- G_warning(_("Unable to open database <%s> by driver <%s>"),
- Fi->database, Fi->driver);
- db_close_database_shutdown_driver(driver);
- pg_info->dbdriver = NULL;
- return -1;
- }
- /* describe table */
- db_set_string(&dbstmt, "select * from ");
- db_append_string(&dbstmt, Fi->table);
- db_append_string(&dbstmt, " where 0 = 1");
- if (db_open_select_cursor(driver, &dbstmt,
- &cursor, DB_SEQUENTIAL) != DB_OK) {
- G_warning(_("Unable to open select cursor: '%s'"),
- db_get_string(&dbstmt));
- db_close_database_shutdown_driver(driver);
- pg_info->dbdriver = NULL;
- return -1;
- }
- table = db_get_cursor_table(&cursor);
- ncols = db_get_table_number_of_columns(table);
- G_debug(3,
- "copying attributes: driver = %s database = %s table = %s cols = %d",
- Fi->driver, Fi->database, Fi->table, ncols);
- for (col = 0; col < ncols; col++) {
- column = db_get_table_column(table, col);
- colname = db_get_column_name(column);
- sqltype = db_get_column_sqltype(column);
- ctype = db_sqltype_to_Ctype(sqltype);
- length = db_get_column_length(column);
- G_debug(3, "\tcolumn = %d name = %s type = %d length = %d",
- col, colname, sqltype, length);
- if (strcmp(pg_info->fid_column, colname) == 0) {
- /* skip fid column if exists */
- G_debug(3, "\t%s skipped", pg_info->fid_column);
- continue;
- }
- /* append column */
- sprintf(stmt_col, ",%s %s", colname, db_sqltype_name(sqltype));
- strcat(stmt, stmt_col);
- if (ctype == DB_C_TYPE_STRING) {
- /* length only for string columns */
- sprintf(stmt_col, "(%d)", length);
- strcat(stmt, stmt_col);
- }
- }
- db_free_string(&dbstmt);
- }
- strcat(stmt, ")"); /* close CREATE TABLE statement */
- /* begin transaction (create table) */
- if (execute(pg_info->conn, "BEGIN") == -1) {
- return -1;
- }
- /* create table */
- G_debug(2, "SQL: %s", stmt);
- if (execute(pg_info->conn, stmt) == -1) {
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
- /* add primary key ? */
- if (primary_key) {
- sprintf(stmt, "ALTER TABLE \"%s\".\"%s\" ADD PRIMARY KEY (%s)",
- pg_info->schema_name, pg_info->table_name,
- pg_info->fid_column);
- G_debug(2, "SQL: %s", stmt);
- if (execute(pg_info->conn, stmt) == -1) {
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
- }
- /* determine geometry type (string) */
- switch (pg_info->feature_type) {
- case (SF_POINT):
- geom_type = "POINT";
- break;
- case (SF_LINESTRING):
- geom_type = "LINESTRING";
- break;
- case (SF_POLYGON):
- geom_type = "POLYGON";
- break;
- default:
- G_warning(_("Unsupported feature type %d"), pg_info->feature_type);
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
-
- /* add geometry column */
- sprintf(stmt, "SELECT AddGeometryColumn('%s', '%s', "
- "'%s', %d, '%s', %d)",
- pg_info->schema_name, pg_info->table_name,
- pg_info->geom_column, pg_info->srid,
- geom_type, pg_info->coor_dim);
- G_debug(2, "SQL: %s", stmt);
- result = PQexec(pg_info->conn, stmt);
-
- if (!result || PQresultStatus(result) != PGRES_TUPLES_OK) {
- PQclear(result);
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
-
- /* create index ? */
- if (spatial_index) {
- G_verbose_message(_("Building spatial index on <%s>..."),
- pg_info->geom_column);
- sprintf(stmt,
- "CREATE INDEX %s_%s_idx ON \"%s\".\"%s\" USING GIST (%s)",
- pg_info->table_name, pg_info->geom_column,
- pg_info->schema_name, pg_info->table_name,
- pg_info->geom_column);
- G_debug(2, "SQL: %s", stmt);
-
- if (execute(pg_info->conn, stmt) == -1) {
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
- }
- /* close transaction (create table) */
- if (execute(pg_info->conn, "COMMIT") == -1) {
- return -1;
- }
- return 0;
- }
- /*!
- \brief Create new PostGIS topology schema
- - create topology schema
- - add topology column to the feature table
-
- \param pg_info pointer to Format_info_pg
- \return 0 on success
- \return 1 topology disable, nothing to do
- \return -1 on failure
- */
- int create_topo_schema(struct Format_info_pg *pg_info, int with_z)
- {
- double tolerance;
- char stmt[DB_SQL_MAX];
- char *def_file;
-
- PGresult *result;
-
- def_file = getenv("GRASS_VECTOR_PGFILE");
-
- /* read default values from PG file*/
- tolerance = 0.;
- if (G_find_file2("", def_file ? def_file : "PG", G_mapset())) {
- FILE *fp;
- const char *p;
- struct Key_Value *key_val;
- fp = G_fopen_old("", def_file ? def_file : "PG", G_mapset());
- if (!fp) {
- G_fatal_error(_("Unable to open PG file"));
- }
- key_val = G_fread_key_value(fp);
- fclose(fp);
- /* tolerance */
- p = G_find_key_value("tolerance", key_val);
- if (p)
- tolerance = atof(p);
- /* topogeom column */
- p = G_find_key_value("topogeom_column", key_val);
- if (p)
- pg_info->topogeom_column = G_store(p);
- else
- pg_info->topogeom_column = G_store(TOPOGEOM_COLUMN);
- }
- /* begin transaction (create topo schema) */
- if (execute(pg_info->conn, "BEGIN") == -1) {
- return -1;
- }
- /* create topology schema */
- G_verbose_message(_("Creating topology schema <%s>..."),
- pg_info->toposchema_name);
- sprintf(stmt, "SELECT topology.createtopology('%s', "
- "find_srid('%s', '%s', '%s'), %f, '%s')",
- pg_info->toposchema_name, pg_info->schema_name,
- pg_info->table_name, pg_info->geom_column, tolerance,
- with_z == WITH_Z ? "t" : "f");
- G_debug(2, "SQL: %s", stmt);
- result = PQexec(pg_info->conn, stmt);
- if (!result || PQresultStatus(result) != PGRES_TUPLES_OK) {
- G_warning(_("Execution failed: %s"), PQerrorMessage(pg_info->conn));
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
-
- /* add topo column to the feature table */
- G_verbose_message(_("Adding new topology column <%s>..."),
- pg_info->topogeom_column);
- sprintf(stmt, "SELECT topology.AddTopoGeometryColumn('%s', '%s', '%s', "
- "'%s', '%s')", pg_info->toposchema_name, pg_info->schema_name,
- pg_info->table_name, pg_info->topogeom_column,
- get_sftype(pg_info->feature_type));
- G_debug(2, "SQL: %s", stmt);
- result = PQexec(pg_info->conn, stmt);
- if (!result || PQresultStatus(result) != PGRES_TUPLES_OK) {
- G_warning(_("Execution failed: %s"), PQerrorMessage(pg_info->conn));
- execute(pg_info->conn, "ROLLBACK");
- return -1;
- }
- /* close transaction (create topo schema) */
- if (execute(pg_info->conn, "COMMIT") == -1) {
- return -1;
- }
- return 0;
- }
- /*!
- \brief Establish PG connection (pg_info->conninfo)
- \param pg_info pointer to Format_info_pg
- */
- void connect_db(struct Format_info_pg *pg_info)
- {
- pg_info->conn = PQconnectdb(pg_info->conninfo);
- G_debug(2, " PQconnectdb(): %s", pg_info->conninfo);
- if (PQstatus(pg_info->conn) == CONNECTION_BAD)
- G_fatal_error("%s\n%s",
- _("Connection ton PostgreSQL database failed."),
- PQerrorMessage(pg_info->conn));
-
- /* print notice messages only on verbose level */
- PQsetNoticeProcessor(pg_info->conn, notice_processor, NULL);
- }
- /*!
- \brief Check for topology schema (pg_info->toposchema_name)
- \param pg_info pointer to Format_info_pg
- \return 0 schema exists
- \return 1 schema doesn't exists
- */
- int check_topo(struct Format_info_pg *pg_info, struct Plus_head *plus)
- {
- char stmt[DB_SQL_MAX];
-
- PGresult *res;
-
- /* connect database */
- if (!pg_info->conn)
- connect_db(pg_info);
-
- if (pg_info->toposchema_name)
- return 0;
-
- /* check if topology layer/schema exists */
- sprintf(stmt,
- "SELECT t.name,t.hasz,l.feature_column FROM topology.layer "
- "AS l JOIN topology.topology AS t ON l.topology_id = t.id "
- "WHERE schema_name = '%s' AND table_name = '%s'",
- pg_info->schema_name, pg_info->table_name);
- G_debug(2, "SQL: %s", stmt);
-
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK ||
- PQntuples(res) != 1) {
- G_debug(1, "Topology layers for '%s.%s' not found (%s)",
- pg_info->schema_name, pg_info->table_name,
- PQerrorMessage(pg_info->conn));
- if (res)
- PQclear(res);
- return 1;
- }
- pg_info->toposchema_name = G_store(PQgetvalue(res, 0, 0));
- pg_info->topogeom_column = G_store(PQgetvalue(res, 0, 2));
- G_debug(1, "PostGIS topology detected: schema = %s column = %s",
- pg_info->toposchema_name, pg_info->topogeom_column);
-
- /* check for 3D */
- if (strcmp(PQgetvalue(res, 0, 1), "t") == 0)
- plus->with_z = WITH_Z;
- PQclear(res);
-
- return 0;
- }
- /*!
- \brief Parse BBOX string
-
- \param value string buffer
- \param[out] bbox pointer to output bound_box struct
- \return 0 on success
- \return -1 on error
- */
- int parse_bbox(const char *value, struct bound_box *bbox)
- {
- unsigned int i;
- size_t length, prefix_length;
- char **tokens, **tokens_coord, *coord;
-
- if (strlen(value) < 1) {
- G_warning(_("Empty bounding box"));
- return -1;
- }
-
- prefix_length = strlen("box3d(");
- if (G_strncasecmp(value, "box3d(", prefix_length) != 0)
- return -1;
-
- /* strip off "bbox3d(...)" */
- length = strlen(value);
- coord = G_malloc(length - prefix_length);
- for (i = prefix_length; i < length; i++)
- coord[i-prefix_length] = value[i];
- coord[length-prefix_length-1] = '\0';
-
- tokens = G_tokenize(coord, ",");
- G_free(coord);
-
- if (G_number_of_tokens(tokens) != 2) {
- G_free_tokens(tokens);
- return -1;
- }
-
- /* parse bbox LL corner */
- tokens_coord = G_tokenize(tokens[0], " ");
- if (G_number_of_tokens(tokens_coord) != 3) {
- G_free_tokens(tokens);
- G_free_tokens(tokens_coord);
- }
- bbox->W = atof(tokens_coord[0]);
- bbox->S = atof(tokens_coord[1]);
- bbox->B = atof(tokens_coord[2]);
-
- G_free_tokens(tokens_coord);
-
- /* parse bbox UR corner */
- tokens_coord = G_tokenize(tokens[1], " ");
- if (G_number_of_tokens(tokens_coord) != 3) {
- G_free_tokens(tokens);
- G_free_tokens(tokens_coord);
- }
- bbox->E = atof(tokens_coord[0]);
- bbox->N = atof(tokens_coord[1]);
- bbox->T = atof(tokens_coord[2]);
-
- G_free_tokens(tokens_coord);
- G_free_tokens(tokens);
-
- return 0;
- }
- /*!
- \brief Get number of records for given SQL statement
- \param stmt string buffer with SQL statement
-
- \return number of returned records
- \return -1 on error
- */
- int num_of_records(const struct Format_info_pg *pg_info,
- const char *stmt)
- {
- int result;
- PGresult *res;
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK ||
- PQntuples(res) != 1) {
- G_warning(_("Unable to get number of records for:\n%s"), stmt);
- if (res)
- PQclear(res);
- return -1;
- }
- result = atoi(PQgetvalue(res, 0, 0));
- PQclear(res);
-
- return result;
- }
- /*!
- \brief Read P_node structure
-
- See dig_Rd_P_node() for reference.
-
- \param plus pointer to Plus_head structure
- \param n index (starts at 1)
- \param id node id (table "node")
- \param wkb_data geometry data (wkb)
- \param pg_info pointer to Format_info_pg sttucture
- \return pointer to new P_node struct
- \return NULL on error
- */
- struct P_node *read_p_node(struct Plus_head *plus, int n,
- int id, const char *wkb_data,
- struct Format_info_pg *pg_info)
- {
- int i, cnt;
- char stmt[DB_SQL_MAX];
-
- struct P_node *node;
- struct line_pnts *points;
-
- PGresult *res;
-
- /* get lines connected to the node */
- sprintf(stmt,
- "SELECT edge_id,'s' as node,"
- "ST_Azimuth(ST_StartPoint(geom), ST_PointN(geom, 2)) AS angle"
- " FROM \"%s\".edge WHERE start_node = %d UNION ALL "
- "SELECT edge_id,'e' as node,"
- "ST_Azimuth(ST_EndPoint(geom), ST_PointN(geom, ST_NumPoints(geom) - 1)) AS angle"
- " FROM \"%s\".edge WHERE end_node = %d"
- " ORDER BY angle DESC",
- pg_info->toposchema_name, id,
- pg_info->toposchema_name, id);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK) {
- G_warning(_("Unable to read node %d"), id);
- if (res)
- PQclear(res);
- return NULL;
- }
- cnt = PQntuples(res);
-
- if (cnt == 0) { /* dead ??? */
- plus->Node[n] = NULL;
- return NULL;
- }
- node = dig_alloc_node();
- node->n_lines = cnt;
- G_debug(4, "read_p_node(): id = %d, n_lines = %d", id, cnt);
-
- if (dig_node_alloc_line(node, node->n_lines) == -1)
- return NULL;
- /* lines / angles */
- for (i = 0; i < node->n_lines; i++) {
- node->lines[i] = atoi(PQgetvalue(res, i, 0));
- if (strcmp(PQgetvalue(res, i, 1), "s") != 0) {
- /* end node */
- node->lines[i] *= -1;
- }
- node->angles[i] = M_PI / 2 - atof(PQgetvalue(res, i, 2));
- /* angles range <-PI; PI> */
- if (node->angles[i] > M_PI)
- node->angles[i] = node->angles[i] - 2 * M_PI;
- if (node->angles[i] < -1.0 * M_PI)
- node->angles[i] = node->angles[i] + 2 * M_PI;
- G_debug(5, "\tline = %d angle = %f", node->lines[i],
- node->angles[i]);
- }
- PQclear(res);
-
- /* get node coordinates */
- if (SF_POINT != cache_feature(wkb_data, FALSE, FALSE,
- &(pg_info->cache), NULL))
- G_warning(_("Node %d: unexpected feature type %d"),
- n, pg_info->cache.sf_type);
-
- points = pg_info->cache.lines[0];
- node->x = points->x[0];
- node->y = points->y[0];
- if (plus->with_z)
- node->z = points->z[0];
- else
- node->z = 0.0;
-
- /* update spatial index */
- dig_spidx_add_node(plus, n, node->x, node->y, node->z);
-
- plus->Node[n] = node;
-
- return node;
- }
- /*!
- \brief Read P_line structure
-
- See dig_Rd_P_line() for reference.
-
- Supported feature types:
- - GV_POINT
- - GV_LINE
- - GV_BOUNDARY
-
- \param plus pointer to Plus_head structure
- \param n index (starts at 1)
- \param data edge data (id, start/end node, left/right face, ...)
- \param pg_info pointer to Format_info_pg sttucture
- \return pointer to P_line struct
- \return NULL on error
- */
- struct P_line *read_p_line(struct Plus_head *plus, int n,
- const struct edge_data *data,
- struct Format_info_cache *cache)
- {
- int tp, itype;
- struct P_line *line;
-
- struct line_pnts *points;
- struct bound_box box;
-
- if (data->start_node == 0 && data->end_node == 0) {
- if (data->left_face == 0)
- tp = GV_POINT;
- else
- tp = GV_CENTROID;
- }
- else if (data->left_face == 0 && data->right_face == 0) {
- tp = GV_LINE;
- }
- else {
- tp = GV_BOUNDARY;
- }
-
- if (tp == 0) { /* dead ??? */
- plus->Line[n] = NULL;
- return NULL;
- }
- line = dig_alloc_line();
-
- /* type & offset ( = id) */
- line->type = tp;
- line->offset = data->id;
- G_debug(4, "read_p_line(): id/offset = %d type = %d", data->id, line->type);
-
- /* topo */
- if (line->type == GV_POINT) {
- line->topo = NULL;
- }
- else {
- line->topo = dig_alloc_topo(line->type);
- /* lines */
- if (line->type == GV_LINE) {
- struct P_topo_l *topo = (struct P_topo_l *)line->topo;
-
- topo->N1 = data->start_node;
- topo->N2 = data->end_node;
- }
- /* boundaries */
- else if (line->type == GV_BOUNDARY) {
- struct P_topo_b *topo = (struct P_topo_b *)line->topo;
-
- topo->N1 = data->start_node;
- topo->N2 = data->end_node;
- /* skip left/right area - will be detected when building
- areas/isles */
- topo->left = topo->right = 0;
- }
- /* centroids */
- else if (line->type == GV_CENTROID) {
- struct P_topo_c *topo = (struct P_topo_c *)line->topo;
-
- topo->area = data->left_face;
- }
- /* TODO: faces | kernels */
- }
- /* update spatial index */
- cache_feature(data->wkb_geom, FALSE, FALSE, cache, NULL);
- itype = cache->lines_types[0];
- if ((line->type & GV_POINTS && itype != GV_POINT) ||
- (line->type & GV_LINES && itype != GV_LINE))
- G_warning(_("Line %d: unexpected feature type"), n);
-
- points = cache->lines[0];
- dig_line_box(points, &box);
- dig_spidx_add_line(plus, n, &box);
-
- plus->Line[n] = line;
-
- return line;
- }
- /*!
- \brief Read topo (from PostGIS topology schema) header info only
- \param[in,out] plus pointer to Plus_head struct
- \return 0 on success
- \return -1 on error
- */
- int load_plus_head(struct Format_info_pg *pg_info, struct Plus_head *plus)
- {
- char stmt[DB_SQL_MAX];
-
- PGresult *res;
-
- plus->off_t_size = -1;
-
- /* get map bounding box */
- sprintf(stmt,
- "SELECT ST_3DExtent(%s) FROM \"%s\".\"%s\"",
- pg_info->topogeom_column, pg_info->schema_name, pg_info->table_name);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK ||
- PQntuples(res) != 1) {
- G_warning(_("Unable to get map bounding box from topology"));
- if (res)
- PQclear(res);
- return -1;
- }
- if (parse_bbox(PQgetvalue(res, 0, 0), &(plus->box)) != 0) {
- G_warning(_("Unable to parse map bounding box:\n%s"),
- PQgetvalue(res, 0, 0));
- return -1;
- }
- PQclear(res);
-
- /* number of topological primitives */
- /* nodes
- note: isolated nodes are registered in GRASS Topology model */
- sprintf(stmt,
- "SELECT COUNT(DISTINCT node) FROM (SELECT start_node AS node "
- "FROM \"%s\".edge GROUP BY start_node UNION ALL SELECT end_node "
- "AS node FROM \"%s\".edge GROUP BY end_node) AS foo",
- pg_info->toposchema_name, pg_info->toposchema_name);
- plus->n_nodes = num_of_records(pg_info, stmt);
- G_debug(3, "Vect_open_topo_pg(): n_nodes=%d", plus->n_nodes);
- /* lines (edges in PostGIS Topology model) */
- sprintf(stmt,
- "SELECT COUNT(*) FROM \"%s\".edge",
- pg_info->toposchema_name);
- /* + isolated nodes as points
- + centroids */
- plus->n_lines = num_of_records(pg_info, stmt);
- /* areas (faces in PostGIS Topology model)
- sprintf(stmt,
- "SELECT COUNT(*) FROM \"%s\".face WHERE mbr IS NOT NULL",
- pg_info->toposchema_name);
- plus->n_areas = num_of_records(pg_info, stmt);
- G_debug(3, "Vect_open_topo_pg(): n_areas=%d", plus->n_areas);
- */
- /* TODO: n_isles | n_volumes | n_holes */
-
- /* number of features group by type */
- /* points */
- sprintf(stmt,
- "SELECT COUNT(*) FROM \"%s\".node WHERE node_id NOT IN "
- "(SELECT node FROM (SELECT start_node AS node FROM \"%s\".edge "
- "GROUP BY start_node UNION ALL SELECT end_node AS node FROM "
- "\"%s\".edge GROUP BY end_node) AS foo)",
- pg_info->toposchema_name, pg_info->toposchema_name,
- pg_info->toposchema_name);
- plus->n_plines = num_of_records(pg_info, stmt);
- G_debug(3, "Vect_open_topo_pg(): n_plines=%d", plus->n_plines);
- /* lines */
- sprintf(stmt,
- "SELECT COUNT(*) FROM \"%s\".edge WHERE "
- "left_face = 0 AND right_face = 0",
- pg_info->toposchema_name);
- plus->n_llines = num_of_records(pg_info, stmt);
- G_debug(3, "Vect_open_topo_pg(): n_llines=%d", plus->n_llines);
- /* boundaries */
- sprintf(stmt,
- "SELECT COUNT(*) FROM \"%s\".edge WHERE "
- "left_face != 0 OR right_face != 0",
- pg_info->toposchema_name);
- plus->n_blines = num_of_records(pg_info, stmt);
- G_debug(3, "Vect_open_topo_pg(): n_blines=%d", plus->n_blines);
- /* centroids */
- sprintf(stmt,
- "SELECT COUNT(*) FROM \"%s\".face WHERE mbr IS NOT NULL",
- pg_info->toposchema_name);
- plus->n_clines = num_of_records(pg_info, stmt);
- G_debug(3, "Vect_open_topo_pg(): n_clines=%d", plus->n_clines);
- /* TODO: nflines | n_klines */
- /* lines - register isolated nodes as points and centroids */
- plus->n_lines += plus->n_plines + plus->n_clines;
- G_debug(3, "Vect_open_topo_pg(): n_lines=%d", plus->n_lines);
- return 0;
- }
- /*!
- \brief Read topo info (from PostGIS topology schema)
- \param pg_info pointer to Format_info_pg
- \param[in,out] plus pointer to Plus_head struct
- \param head_only TRUE to read only header info
-
- \return 0 on success
- \return -1 on error
- */
- int load_plus(struct Map_info *Map, int head_only)
- {
- int i, id, ntuples;
- char stmt[DB_SQL_MAX];
- struct edge_data line_data;
-
- struct Format_info_pg *pg_info;
- struct Plus_head *plus;
- struct P_line *line;
- struct P_area *area;
-
- PGresult *res;
-
- pg_info = &(Map->fInfo.pg);
- plus = &(Map->plus);
- if (load_plus_head(pg_info, plus) != 0)
- return -1;
- if (head_only)
- return 0;
-
- /* read nodes (GRASS Topo)
- note: standalone nodes are ignored
- */
- sprintf(stmt,
- "SELECT node_id,geom FROM \"%s\".node WHERE node_id IN "
- "(SELECT node FROM (SELECT start_node AS node FROM \"%s\".edge "
- "GROUP BY start_node UNION ALL SELECT end_node AS node FROM "
- "\"%s\".edge GROUP BY end_node) AS foo)",
- pg_info->toposchema_name, pg_info->toposchema_name,
- pg_info->toposchema_name);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK ||
- PQntuples(res) != plus->n_nodes) {
- G_warning(_("Unable to read nodes"));
- if (res)
- PQclear(res);
- return -1;
- }
- G_debug(3, "load_plus(): n_nodes = %d", plus->n_nodes);
- dig_alloc_nodes(plus, plus->n_nodes);
- for (i = 0; i < plus->n_nodes; i++) {
- id = atoi(PQgetvalue(res, i, 0));
- read_p_node(plus, i + 1, /* node index starts at 1 */
- id, (const char *) PQgetvalue(res, i, 1), pg_info);
- }
- PQclear(res);
- /* read lines (GRASS Topo)
- - standalone nodes -> points
- - edges -> lines/boundaries
- */
- G_debug(3, "load_plus(): n_lines = %d", plus->n_lines);
- dig_alloc_lines(plus, plus->n_lines);
-
- /* read PostGIS Topo standalone nodes
- -> points
- */
- sprintf(stmt,
- "SELECT node_id,geom FROM \"%s\".node WHERE node_id NOT IN "
- "(SELECT node FROM (SELECT start_node AS node FROM \"%s\".edge "
- "GROUP BY start_node UNION ALL SELECT end_node AS node FROM "
- "\"%s\".edge GROUP BY end_node) AS foo)",
- pg_info->toposchema_name, pg_info->toposchema_name,
- pg_info->toposchema_name);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK ||
- PQntuples(res) > plus->n_plines) {
- G_warning(_("Unable to read lines"));
- if (res)
- PQclear(res);
- return -1;
- }
-
- ntuples = PQntuples(res); /* plus->n_plines */
- G_zero(&line_data, sizeof(struct edge_data));
- for (i = 0; i < ntuples; i++) {
- /* process standalone nodes (PostGIS Topo) */
- line_data.id = atoi(PQgetvalue(res, i, 0));
- line_data.wkb_geom = (char *) PQgetvalue(res, i, 1);
- read_p_line(plus, i + 1, &line_data, &(pg_info->cache));
- }
- PQclear(res);
-
- /* read PostGIS Topo edges
- -> lines
- -> boundaries
- */
- sprintf(stmt,
- "SELECT edge_id,start_node,end_node,left_face,right_face,geom "
- "FROM \"%s\".edge",
- pg_info->toposchema_name);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK ||
- PQntuples(res) > plus->n_lines) {
- G_warning(_("Unable to read lines"));
- if (res)
- PQclear(res);
- return -1;
- }
- ntuples = PQntuples(res);
- for (i = 0; i < ntuples; i++) {
- /* process edges (PostGIS Topo) */
- line_data.id = atoi(PQgetvalue(res, i, 0));
- line_data.start_node = atoi(PQgetvalue(res, i, 1));
- line_data.end_node = atoi(PQgetvalue(res, i, 2));
- line_data.left_face = atoi(PQgetvalue(res, i, 3));
- line_data.right_face = atoi(PQgetvalue(res, i, 4));
- line_data.wkb_geom = (char *) PQgetvalue(res, i, 5);
-
- id = plus->n_plines + i + 1; /* points already registered */
- line = read_p_line(plus, id, &line_data, &(pg_info->cache));
- if (line_data.left_face != 0 || line_data.right_face != 0) {
- /* boundary detected -> build area/isle on left and right*/
-
- int s, side;
- for (s = 0; s < 2; s++) {
- if (s == 0)
- side = GV_LEFT;
- else
- side = GV_RIGHT;
-
- G_debug(3, "Build area for line = %d, side = %d",
- id, side);
- Vect_build_line_area(Map, id, side);
- }
- }
- if (line->type == GV_BOUNDARY) {
- struct P_topo_b *topo;
- if (line_data.left_face == 0)
- line_data.left_face = -1;
- if (line_data.right_face == 0)
- line_data.right_face = -1;
-
- /* check topo - left / right areas */
- topo = (struct P_topo_b *)line->topo;
- if (topo->left != line_data.left_face)
- G_warning(_("Left area detected as %d (should be %d"),
- topo->left, line_data.left_face);
- if (topo->right != line_data.right_face)
- G_warning(_("Right area detected as %d (should be %d"),
- topo->right, line_data.right_face);
- }
- }
- PQclear(res);
-
- /* attach centroids */
- if (plus->n_areas > 0) {
- sprintf(stmt,
- "SELECT ST_PointOnSurface(geom) AS geom FROM "
- "ST_GetFaceGeometry('%s',"
- "(SELECT face_id FROM \"%s\".face WHERE face_id > 0)) "
- "AS geom",
- pg_info->toposchema_name, pg_info->toposchema_name);
- G_debug(2, "SQL: %s", stmt);
- res = PQexec(pg_info->conn, stmt);
- if (!res || PQresultStatus(res) != PGRES_TUPLES_OK ||
- PQntuples(res) > plus->n_areas) {
- G_warning(_("Unable to attach centroids"));
- if (res)
- PQclear(res);
- return -1;
- }
-
- G_zero(&line_data, sizeof(struct edge_data));
- for (i = 1; i <= plus->n_areas; i++) {
- area = plus->Area[i];
-
- id = plus->n_lines - plus->n_clines + i;
- line_data.id = line_data.left_face = i;
- line_data.wkb_geom = (char *)PQgetvalue(res, 0, 0);
-
- read_p_line(plus, id, &line_data, &(pg_info->cache));
- area->centroid = line_data.id;
- }
- }
- return 0;
- }
- /*
- \brief PostgreSQL notice processor
- Print out NOTICE message only on verbose level
- */
- void notice_processor(void *arg, const char *message)
- {
- if (G_verbose() > G_verbose_std()) {
- fprintf(stderr, "%s", message);
- }
- }
- /*!
- \brief Get simple feature type as a string
- Used for AddTopoGeometryColumn().
- Valid types:
- - SF_POINT
- - SF_LINESTRING
- - SF_POLYGON
- \return string with feature type
- \return empty string
- */
- char *get_sftype(SF_FeatureType sftype)
- {
- if (sftype == SF_POINT)
- return "POINT";
- else if (sftype == SF_LINESTRING)
- return "LINE";
- else if (sftype == SF_POLYGON)
- return "POLYGON";
- else
- G_warning(_("Unsupported feature type %d"), sftype);
-
- return "";
- }
- #endif
|