int main(int argc, char **argv) { dbDriver *driver; dbHandle handle; dbString *names; int i, count; int system_tables; parse_command_line(argc, argv); driver = db_start_driver(parms.driver); if (driver == NULL) G_fatal_error(_("Unable to start driver <%s>"), parms.driver); db_init_handle(&handle); db_set_handle(&handle, parms.database, NULL); if (db_open_database(driver, &handle) != DB_OK) G_fatal_error(_("Unable to open database <%s>"), parms.database); system_tables = parms.s; if (db_list_tables(driver, &names, &count, system_tables) != DB_OK) exit(ERROR); for (i = 0; i < count; i++) fprintf(stdout, "%s\n", db_get_string(&names[i])); db_close_database(driver); db_shutdown_driver(driver); exit(EXIT_SUCCESS); }
int main(int argc, char **argv) { dbString stmt; dbDriver *driver; dbHandle handle; int ret; FILE *fd; int error; error = 0; parse_command_line(argc, argv); if (strcmp(parms.input, "-")) { fd = fopen(parms.input, "r"); if (fd == NULL) { G_fatal_error(_("Unable to open file <%s> for reading.\n" "Details: %s"), parms.input, strerror(errno)); } } else { fd = stdin; } driver = db_start_driver(parms.driver); if (driver == NULL) { G_fatal_error(_("Unable to start driver <%s>"), parms.driver); } db_init_handle(&handle); db_set_handle(&handle, parms.database, parms.schema); if (db_open_database(driver, &handle) != DB_OK) G_fatal_error(_("Unable to open database <%s>"), parms.database); while (get_stmt(fd, &stmt)) { if (!stmt_is_empty(&stmt)) { G_debug(3, "sql: %s", db_get_string(&stmt)); ret = db_execute_immediate(driver, &stmt); if (ret != DB_OK) { if (parms.i) { /* ignore SQL errors */ G_warning(_("Error while executing: '%s'"), db_get_string(&stmt)); error++; } else G_fatal_error(_("Error while executing: '%s'"), db_get_string(&stmt)); } } } db_close_database(driver); db_shutdown_driver(driver); exit(error ? EXIT_FAILURE : EXIT_SUCCESS); }
/*! \brief Fetches list of DB column names and types of vector map attribute table \param Map vector map \param field layer number \return list of column(s) types on success \retutn NULL on error */ const char *Vect_get_column_names_types(struct Map_info *Map, int field) { int num_dblinks, ncols, col; struct field_info *fi; dbDriver *driver = NULL; dbHandle handle; dbString table_name; dbTable *table; char buf[2000], temp_buf[2000]; num_dblinks = Vect_get_num_dblinks(Map); if (num_dblinks <= 0) return (NULL); G_debug(3, "Displaying column types for database connection of layer %d:", field); if ((fi = Vect_get_field(Map, field)) == NULL) return (NULL); driver = db_start_driver(fi->driver); if (driver == NULL) return (NULL); db_init_handle(&handle); db_set_handle(&handle, fi->database, NULL); if (db_open_database(driver, &handle) != DB_OK) return (NULL); db_init_string(&table_name); db_set_string(&table_name, fi->table); if (db_describe_table(driver, &table_name, &table) != DB_OK) return (NULL); ncols = db_get_table_number_of_columns(table); sprintf(buf, " "); for (col = 0; col < ncols; col++) { if (col == 0) sprintf(buf, "%s(%s)", db_get_column_name(db_get_table_column(table, col)), db_sqltype_name(db_get_column_sqltype (db_get_table_column(table, col)))); else { sprintf(temp_buf, ",%s(%s)", db_get_column_name(db_get_table_column(table, col)), db_sqltype_name(db_get_column_sqltype (db_get_table_column(table, col)))); strcat(buf, temp_buf); } } G_debug(3, "%s", buf); db_close_database(driver); db_shutdown_driver(driver); return G_store(G_chop(buf)); }
void print_columns(const struct Map_info *Map, const char *input_opt, const char *field_opt) { int num_dblinks, col, ncols; struct field_info *fi; dbDriver *driver = NULL; dbHandle handle; dbString table_name; dbTable *table; num_dblinks = Vect_get_num_dblinks(Map); if (num_dblinks <= 0) { G_fatal_error(_("Database connection for map <%s> is not defined in DB file"), input_opt); } G_message(_("Displaying column types/names for database connection of layer <%s>:"), field_opt); if ((fi = Vect_get_field2(Map, field_opt)) == NULL) G_fatal_error(_("Database connection not defined for layer <%s>"), field_opt); driver = db_start_driver(fi->driver); if (driver == NULL) G_fatal_error(_("Unable to open driver <%s>"), fi->driver); db_init_handle(&handle); db_set_handle(&handle, fi->database, NULL); if (db_open_database(driver, &handle) != DB_OK) G_fatal_error(_("Unable to open database <%s> by driver <%s>"), fi->database, fi->driver); db_init_string(&table_name); db_set_string(&table_name, fi->table); if (db_describe_table(driver, &table_name, &table) != DB_OK) G_fatal_error(_("Unable to describe table <%s>"), fi->table); ncols = db_get_table_number_of_columns(table); for (col = 0; col < ncols; col++) fprintf(stdout, "%s|%s\n", db_sqltype_name(db_get_column_sqltype (db_get_table_column (table, col))), db_get_column_name(db_get_table_column (table, col))); db_close_database(driver); db_shutdown_driver(driver); }
int main(int argc, char **argv) { dbDriver *driver; dbHandle handle; int stat; parse_command_line(argc, argv); driver = db_start_driver(parms.driver); if (driver == NULL) G_fatal_error(_("Unable to start driver <%s>"), parms.driver); db_init_handle(&handle); db_set_handle(&handle, parms.database, NULL); stat = db_delete_database(driver, &handle); db_shutdown_driver(driver); exit(stat == DB_OK ? EXIT_SUCCESS : EXIT_FAILURE); }
int main(int argc, char *argv[]) { int i, j, precision, field, type, nlines; int do_attr = 0, attr_cols[8], attr_size = 0, db_open = 0, cnt = 0; double width, radius; struct Option *in_opt, *out_opt, *prec_opt, *type_opt, *attr_opt, *field_opt; struct GModule *module; struct Map_info In; struct bound_box box; /* vector */ struct line_pnts *Points; struct line_cats *Cats; /* attribs */ dbDriver *Driver = NULL; dbHandle handle; dbTable *Table; dbString dbstring; struct field_info *Fi; /* init */ G_gisinit(argv[0]); /* parse command-line */ module = G_define_module(); module->description = _("Exports a vector map to SVG file."); G_add_keyword(_("vector")); G_add_keyword(_("export")); in_opt = G_define_standard_option(G_OPT_V_INPUT); field_opt = G_define_standard_option(G_OPT_V_FIELD_ALL); out_opt = G_define_standard_option(G_OPT_F_OUTPUT); out_opt->description = _("Name for SVG output file"); type_opt = G_define_option(); type_opt->key = "type"; type_opt->type = TYPE_STRING; type_opt->required = YES; type_opt->multiple = NO; type_opt->answer = "poly"; type_opt->options = "poly,line,point"; type_opt->label = _("Output type"); type_opt->description = _("Defines which feature-type will be extracted"); prec_opt = G_define_option(); prec_opt->key = "precision"; prec_opt->type = TYPE_INTEGER; prec_opt->required = NO; prec_opt->answer = "6"; prec_opt->multiple = NO; prec_opt->description = _("Coordinate precision"); attr_opt = G_define_standard_option(G_OPT_DB_COLUMNS); attr_opt->key = "attribute"; attr_opt->required = NO; attr_opt->multiple = YES; attr_opt->description = _("Attribute(s) to include in output SVG"); if (G_parser(argc, argv)) exit(EXIT_FAILURE); Points = Vect_new_line_struct(); Cats = Vect_new_cats_struct(); if (type_opt->answer[0] == 'l') { type = TYPE_LINE; } else { if (type_opt->answer[2] == 'l') type = TYPE_POLY; else type = TYPE_POINT; } /* override coordinate precision if any */ precision = atof(prec_opt->answer); if (precision < 0) { G_fatal_error(_("Precision must not be negative")); } if (precision > 15) { G_fatal_error(_("Precision must not be higher than 15")); } /* open input vector */ Vect_set_open_level(2); if (Vect_open_old2(&In, in_opt->answer, "", field_opt->answer) < 0) G_fatal_error(_("Unable to open vector map <%s>"), in_opt->answer); /* parse field number */ field = Vect_get_field_number(&In, field_opt->answer); /* open db-driver to attribs */ db_init_string(&dbstring); /* check for requested field */ Fi = Vect_get_field(&In, field); if (Fi != NULL) { Driver = db_start_driver(Fi->driver); if (Driver == NULL) { G_fatal_error(_("Unable to start driver <%s>"), Fi->driver); } /* open db */ db_init_handle(&handle); db_set_handle(&handle, Fi->database, NULL); if (db_open_database(Driver, &handle) != DB_OK) { G_fatal_error(_("Unable to open database <%s> by driver <%s>"), Fi->database, Fi->driver); } db_set_string(&dbstring, Fi->table); if (db_describe_table(Driver, &dbstring, &Table) != DB_OK) { G_fatal_error(_("Unable to describe table <%s>"), Fi->table); } /* define column-indices for columns to extract */ dbColumn *Column; for (i = 0; i < db_get_table_number_of_columns(Table); i++) { Column = db_get_table_column(Table, i); if (attr_opt->answer != NULL) { for (j = 0; attr_opt->answers[j] != NULL; j++) { if (G_strcasecmp(attr_opt->answers[j], db_get_column_name(Column)) == 0) { attr_cols[attr_size] = i; attr_size += 1; break; } } } } do_attr = 1; db_open = 1; } /* parse bounding box and define default stroke-width, radius */ Vect_get_map_box(&In, &box); if ((box.E - box.W) >= (box.N - box.S)) { radius = (box.E - box.W) * RADIUS_SCALE; width = (box.E - box.W) * WIDTH_SCALE; } else { radius = (box.N - box.S) * RADIUS_SCALE; width = (box.N - box.S) * WIDTH_SCALE; } /* open output SVG-file and print SVG-header with viewBox and Namenspaces */ if ((fpsvg = fopen(out_opt->answer, "w")) == NULL) { G_fatal_error(_("Unable to create SVG file <%s>"), out_opt->answer); } fprintf(fpsvg, "<svg xmlns=\"%s\" xmlns:xlink=\"%s\" xmlns:gg=\"%s\" ", SVG_NS, XLINK_NS, GRASS_NS); fprintf(fpsvg, "viewBox=\"%.*f %.*f %.*f %.*f\">\n", precision, box.W, precision, box.N * -1, precision, box.E - box.W, precision, box.N - box.S); fprintf(fpsvg, "<title>v.out.svg %s %s</title>\n", in_opt->answer, out_opt->answer); nlines = Vect_get_num_lines(&In); /* extract areas if any or requested */ if (type == TYPE_POLY) { if (Vect_get_num_areas(&In) == 0) { G_warning(_("No areas found, skipping %s"), "type=poly"); } else { int nareas; nareas = Vect_get_num_areas(&In); /* extract area as paths */ fprintf(fpsvg, " <g id=\"%s\" fill=\"#CCC\" stroke=\"#000\" stroke-width=\"%.*f\" >\n", G_Areas, precision, width); for (i = 1; i <= nareas; i++) { G_percent(i, nareas, 5); /* skip areas without centroid */ if (Vect_get_area_centroid(&In, i) == 0) { G_warning(_("Skipping area %d without centroid"), i); continue; } /* extract attribs, parse area */ Vect_get_area_cats(&In, i, Cats); fprintf(fpsvg, " <path "); if (Cats->n_cats > 0) { mk_attribs(Cats->cat[0], Fi, Driver, Table, attr_cols, attr_size, do_attr); } fprintf(fpsvg, "d=\""); Vect_get_area_points(&In, i, Points); mk_path(Points, precision); /* append islands if any within current path */ for (j = 0; j < Vect_get_area_num_isles(&In, i); j++) { Vect_get_isle_points(&In, Vect_get_area_isle(&In, i, j), Points); mk_path(Points, precision); } fprintf(fpsvg, "\" />\n"); cnt += 1; } fprintf(fpsvg, " </g>\n"); G_message(_("%d areas extracted"), cnt); } } /* extract points if requested */ if (type == TYPE_POINT) { if (Vect_get_num_primitives(&In, GV_POINTS) == 0) { G_warning(_("No points found, skipping %s"), "type=point"); } else { /* extract points as circles */ fprintf(fpsvg, " <g id=\"%s\" fill=\"#FC0\" stroke=\"#000\" " "stroke-width=\"%.*f\" >\n", G_Points, precision, width); for (i = 1; i <= nlines; i++) { G_percent(i, nlines, 5); if (!(Vect_read_line(&In, Points, Cats, i) & GV_POINTS)) continue; if (field != -1 && !Vect_cat_get(Cats, field, NULL)) continue; for (j = 0; j < Points->n_points; j++) { fprintf(fpsvg, " <circle "); if (Cats->n_cats > 0) { mk_attribs(Cats->cat[j], Fi, Driver, Table, attr_cols, attr_size, do_attr); } fprintf(fpsvg, "cx=\"%.*f\" cy=\"%.*f\" r=\"%.*f\" />\n", precision, Points->x[j], precision, Points->y[j] * -1, precision, radius); cnt += 1; } } fprintf(fpsvg, " </g>\n"); G_message(_("%d points extracted"), cnt); } } /* extract lines if requested */ if (type == TYPE_LINE) { if (Vect_get_num_primitives(&In, GV_LINES) == 0) { G_warning(_("No lines found, skipping %s"), "type=line"); } else { /* extract lines as paths */ fprintf(fpsvg, " <g id=\"%s\" fill=\"none\" stroke=\"#000\" " "stroke-width=\"%.*f\" >\n", G_Lines, precision, width); for (i = 1; i <= nlines; i++) { G_percent(i, nlines, 5); if (!(Vect_read_line(&In, Points, Cats, i) & GV_LINES)) continue; if (field != -1 && !Vect_cat_get(Cats, field, NULL)) continue; fprintf(fpsvg, " <path "); if (Cats->n_cats > 0) { mk_attribs(Cats->cat[0], Fi, Driver, Table, attr_cols, attr_size, do_attr); } fprintf(fpsvg, "d=\""); mk_path(Points, precision); fprintf(fpsvg, "\" />\n"); cnt += 1; } fprintf(fpsvg, " </g>\n"); G_message(_("%d lines extracted"), cnt); } } /* finish code */ fprintf(fpsvg, "</svg>\n"); if (db_open == 1) { /* close database handle */ db_close_database(Driver); db_shutdown_driver(Driver); } /* close SVG-file */ fclose(fpsvg); exit(EXIT_SUCCESS); }
int db__copy_table(const char *from_drvname, const char *from_dbname, const char *from_tblname, const char *to_drvname, const char *to_dbname, const char *to_tblname, const char *where, const char *select, const char *selcol, int *ivals, int nvals) { int col, ncols, sqltype, ctype, more, selcol_found; char buf[1000]; int *ivalues; dbHandle from_handle, to_handle; dbString tblname, sql; dbString value_string; dbString *tblnames; dbTable *table, *out_table; dbCursor cursor; dbColumn *column; dbValue *value; const char *colname; dbDriver *from_driver, *to_driver; int count, i; G_debug(3, "db_copy_table():\n from driver = %s, db = %s, table = %s\n" " to driver = %s, db = %s, table = %s, where = %s, select = %s", from_drvname, from_dbname, from_tblname, to_drvname, to_dbname, to_tblname, where, select); db_init_handle(&from_handle); db_init_handle(&to_handle); db_init_string(&tblname); db_init_string(&sql); db_init_string(&value_string); /* Make a copy of input values and sort it */ if (ivals) { ivalues = (int *)G_malloc(nvals * sizeof(int)); memcpy(ivalues, ivals, nvals * sizeof(int)); qsort((void *)ivalues, nvals, sizeof(int), cmp); } /* Open input driver and database */ from_driver = db_start_driver(from_drvname); if (from_driver == NULL) { G_warning(_("Unable to start driver <%s>"), from_drvname); return DB_FAILED; } db_set_handle(&from_handle, from_dbname, NULL); if (db_open_database(from_driver, &from_handle) != DB_OK) { G_warning(_("Unable to open database <%s> by driver <%s>"), from_dbname, from_drvname); db_close_database_shutdown_driver(from_driver); return DB_FAILED; } /* Open output driver and database */ if (strcmp(from_drvname, to_drvname) == 0 && strcmp(from_dbname, to_dbname) == 0) { G_debug(3, "Use the same driver"); to_driver = from_driver; } else { to_driver = db_start_driver(to_drvname); if (to_driver == NULL) { G_warning(_("Unable to start driver <%s>"), to_drvname); db_close_database_shutdown_driver(from_driver); return DB_FAILED; } db_set_handle(&to_handle, to_dbname, NULL); if (db_open_database(to_driver, &to_handle) != DB_OK) { G_warning(_("Unable to open database <%s> by driver <%s>"), to_dbname, to_drvname); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) { db_close_database_shutdown_driver(from_driver); } return DB_FAILED; } } db_begin_transaction(to_driver); /* Because in SQLite3 an opened cursor is no more valid if 'schema' is modified (create table), we have to open cursor twice */ /* test if the table exists */ if (db_list_tables(to_driver, &tblnames, &count, 0) != DB_OK) { G_warning(_("Unable to get list tables in database <%s>"), to_dbname); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) db_close_database_shutdown_driver(from_driver); return DB_FAILED; } for (i = 0; i < count; i++) { const char *tblname = db_get_string(&tblnames[i]); if (strcmp(to_tblname, tblname) == 0) { G_warning(_("Table <%s> already exists in database <%s>"), to_tblname, to_dbname); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) db_close_database_shutdown_driver(from_driver); return DB_FAILED; } } /* Create new table */ /* Open cursor for data structure */ if (select) { db_set_string(&sql, select); /* TODO!: cannot use this because it will not work if a query * ends with 'group by' for example */ /* tmp = strdup ( select ); G_tolcase ( tmp ); if ( !strstr( tmp,"where") ) { db_append_string ( &sql, " where 0 = 1"); } else { db_append_string ( &sql, " and 0 = 1"); } free (tmp); */ } else { db_set_string(&sql, "select * from "); db_append_string(&sql, from_tblname); db_append_string(&sql, " where 0 = 1"); /* to get no data */ } G_debug(3, db_get_string(&sql)); if (db_open_select_cursor(from_driver, &sql, &cursor, DB_SEQUENTIAL) != DB_OK) { G_warning(_("Unable to open select cursor: '%s'"), db_get_string(&sql)); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) { db_close_database_shutdown_driver(from_driver); } return DB_FAILED; } G_debug(3, "Select cursor opened"); table = db_get_cursor_table(&cursor); ncols = db_get_table_number_of_columns(table); G_debug(3, "ncols = %d", ncols); out_table = db_alloc_table(ncols); db_set_table_name(out_table, to_tblname); selcol_found = 0; for (col = 0; col < ncols; col++) { dbColumn *out_column; 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); G_debug(3, "%s (%s)", colname, db_sqltype_name(sqltype)); out_column = db_get_table_column(out_table, col); if (selcol && G_strcasecmp(colname, selcol) == 0) { if (ctype != DB_C_TYPE_INT) G_fatal_error(_("Column <%s> is not integer"), colname); selcol_found = 1; } db_set_column_name(out_column, db_get_column_name(column)); db_set_column_description(out_column, db_get_column_description(column)); db_set_column_sqltype(out_column, db_get_column_sqltype(column)); db_set_column_length(out_column, db_get_column_length(column)); db_set_column_precision(out_column, db_get_column_precision(column)); db_set_column_scale(out_column, db_get_column_scale(column)); } db_close_cursor(&cursor); if (selcol && !selcol_found) G_fatal_error(_("Column <%s> not found"), selcol); if (db_create_table(to_driver, out_table) != DB_OK) { G_warning(_("Unable to create table <%s>"), to_tblname); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) { db_close_database_shutdown_driver(from_driver); } return DB_FAILED; } /* Open cursor with data */ if (select) { db_set_string(&sql, select); } else { db_set_string(&sql, "select * from "); db_append_string(&sql, from_tblname); if (where) { db_append_string(&sql, " where "); db_append_string(&sql, where); } } G_debug(3, db_get_string(&sql)); if (db_open_select_cursor(from_driver, &sql, &cursor, DB_SEQUENTIAL) != DB_OK) { G_warning(_("Unable to open select cursor: '%s'"), db_get_string(&sql)); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) { db_close_database_shutdown_driver(from_driver); } return DB_FAILED; } G_debug(3, "Select cursor opened"); table = db_get_cursor_table(&cursor); ncols = db_get_table_number_of_columns(table); G_debug(3, "ncols = %d", ncols); /* Copy all rows */ while (1) { int select; if (db_fetch(&cursor, DB_NEXT, &more) != DB_OK) { G_warning(_("Unable to fetch data from table <%s>"), from_tblname); db_close_cursor(&cursor); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) { db_close_database_shutdown_driver(from_driver); } return DB_FAILED; } if (!more) break; sprintf(buf, "insert into %s values ( ", to_tblname); db_set_string(&sql, buf); select = 1; 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); value = db_get_column_value(column); if (selcol && G_strcasecmp(colname, selcol) == 0) { if (db_test_value_isnull(value)) continue; if (!bsearch(&(value->i), ivalues, nvals, sizeof(int), cmp)) { select = 0; break; } } if (col > 0) db_append_string(&sql, ", "); db_convert_value_to_string(value, sqltype, &value_string); switch (ctype) { case DB_C_TYPE_STRING: case DB_C_TYPE_DATETIME: if (db_test_value_isnull(value)) { db_append_string(&sql, "null"); } else { db_double_quote_string(&value_string); db_append_string(&sql, "'"); db_append_string(&sql, db_get_string(&value_string)); db_append_string(&sql, "'"); } break; case DB_C_TYPE_INT: case DB_C_TYPE_DOUBLE: if (db_test_value_isnull(value)) { db_append_string(&sql, "null"); } else { db_append_string(&sql, db_get_string(&value_string)); } break; default: G_warning(_("Unknown column type (column <%s>)"), colname); db_close_cursor(&cursor); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) { db_close_database_shutdown_driver(from_driver); } return DB_FAILED; } } if (!select) continue; db_append_string(&sql, ")"); G_debug(3, db_get_string(&sql)); if (db_execute_immediate(to_driver, &sql) != DB_OK) { G_warning("Unable to insert new record: '%s'", db_get_string(&sql)); db_close_cursor(&cursor); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) { db_close_database_shutdown_driver(from_driver); } return DB_FAILED; } } if (selcol) G_free(ivalues); G_debug(3, "Table copy OK"); db_close_cursor(&cursor); db_commit_transaction(to_driver); db_close_database_shutdown_driver(to_driver); if (from_driver != to_driver) { db_close_database_shutdown_driver(from_driver); } return DB_OK; }
/** \brief Select features according to SQL where statement \param[in] Map vector map \param[in] layer layer number \param[in] type feature type \param[in] where 'where' statement \param[in,out] List list of selected features \return number of selected lines */ int sel_by_where(struct Map_info *Map, int layer, int type, char *where, struct ilist *List) { struct cat_list *cat_list; struct ilist *List_tmp; struct field_info *Fi; dbDriver *driver; dbHandle handle; int *cats, ncats; if (first_selection) { List_tmp = List; first_selection = 0; } else { List_tmp = Vect_new_list(); } cat_list = Vect_new_cat_list(); if (layer < 1) { G_fatal_error(_("Layer must be > 0 for 'where'")); } Fi = Vect_get_field(Map, layer); if (!Fi) { G_fatal_error(_("Database connection not defined for layer %d"), layer); } driver = db_start_driver(Fi->driver); if (!driver) G_fatal_error(_("Unable to start driver <%s>"), Fi->driver); db_init_handle(&handle); db_set_handle(&handle, Fi->database, NULL); if (db_open_database(driver, &handle) != DB_OK) G_fatal_error(_("Unable to open database <%s> by driver <%s>"), Fi->database, Fi->driver); ncats = db_select_int(driver, Fi->table, Fi->key, where, &cats); db_close_database(driver); db_shutdown_driver(driver); Vect_array_to_cat_list(cats, ncats, cat_list); /* free array of cats */ if (ncats >= 0) G_free(cats); sel_by_cat(Map, cat_list, layer, type, NULL, List_tmp); G_debug(1, " %d lines selected (by where)", List_tmp->n_values); /* merge lists (only duplicate items) */ if (List_tmp != List) { merge_lists(List, List_tmp); Vect_destroy_list(List_tmp); } Vect_destroy_cat_list(cat_list); return List->n_values; }
dbDriver *create_table(OGRLayerH hLayer, const struct field_info *Fi) { int col, ncols; int sqltype, ogrtype, length; const char *colname; dbDriver *driver; dbHandle handle; dbCursor cursor; dbTable *table; dbColumn *column; dbString sql; OGRFieldDefnH hFieldDefn; OGRFeatureDefnH hFeatureDefn; db_init_string(&sql); db_init_handle(&handle); driver = db_start_driver(Fi->driver); if (!driver) { G_warning(_("Unable to start driver <%s>"), Fi->driver); return NULL; } 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); return NULL; } /* to get no data */ db_set_string(&sql, "select * from "); db_append_string(&sql, Fi->table); db_append_string(&sql, " where 0 = 1"); if (db_open_select_cursor(driver, &sql, &cursor, DB_SEQUENTIAL) != DB_OK) { G_warning(_("Unable to open select cursor: '%s'"), db_get_string(&sql)); db_close_database_shutdown_driver(driver); return NULL; } table = db_get_cursor_table(&cursor); ncols = db_get_table_number_of_columns(table); hFeatureDefn = OGR_L_GetLayerDefn(hLayer); 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); ogrtype = sqltype_to_ogrtype(sqltype); length = db_get_column_length(column); if (strcmp(OGR_L_GetFIDColumn(hLayer), colname) == 0 || OGR_FD_GetFieldIndex(hFeatureDefn, colname) > -1) { /* field already exists */ continue; } hFieldDefn = OGR_Fld_Create(colname, ogrtype); /* GDAL 1.9.0 (r22968) uses VARCHAR instead of CHAR */ if (ogrtype == OFTString && length > 0) OGR_Fld_SetWidth(hFieldDefn, length); if (OGR_L_CreateField(hLayer, hFieldDefn, TRUE) != OGRERR_NONE) { G_warning(_("Creating field <%s> failed"), colname); db_close_database_shutdown_driver(driver); return NULL; } OGR_Fld_Destroy(hFieldDefn); } return driver; }
int main(int argc, char **argv) { dbString stmt; dbDriver *driver; dbHandle handle; int stat; FILE *fd; parse_command_line(argc, argv); if (parms.table) { if (!db_table_exists(parms.driver, parms.database, parms.table)) { G_warning(_("Table <%s> not found in database <%s> using driver <%s>"), parms.table, parms.database, parms.driver); exit(EXIT_FAILURE); } } /* read from file or stdin ? */ if (parms.input && strcmp(parms.input, "-") != 0) { fd = fopen(parms.input, "r"); if (fd == NULL) { G_fatal_error(_("Unable to open file <%s>: %s"), parms.input, strerror(errno)); } } else fd = stdin; /* open DB connection */ db_init_string(&stmt); driver = db_start_driver(parms.driver); if (driver == NULL) { G_fatal_error(_("Unable to start driver <%s>"), parms.driver); } db_init_handle(&handle); db_set_handle(&handle, parms.database, NULL); if (db_open_database(driver, &handle) != DB_OK) G_fatal_error(_("Unable to open database <%s>"), parms.database); db_set_error_handler_driver(driver); /* check for sql, table, and input */ if (parms.sql) { /* parms.sql */ db_set_string(&stmt, parms.sql); stat = sel(driver, &stmt); } else if (parms.table) { /* parms.table */ db_set_string(&stmt, "SELECT * FROM "); db_append_string(&stmt, parms.table); stat = sel(driver, &stmt); } else { /* -> parms.input */ stat = DB_OK; while (stat == DB_OK && get_stmt(fd, &stmt)) { if (!stmt_is_empty(&stmt)) stat = sel(driver, &stmt); } } if (parms.test_only) G_verbose_message(_("Test %s."), stat ? _("failed") : _("succeeded")); db_close_database(driver); db_shutdown_driver(driver); exit(stat == DB_OK ? EXIT_SUCCESS : EXIT_FAILURE); }
/*! \brief Write data to GRASS ASCII vector format Prints message if some features without category are skipped. \param[out] ascii pointer to the output ASCII file \param[out] att att file (< version 5 only) \param Map pointer to Map_info structure \param ver version number 4 or 5 \param format format GV_ASCII_FORMAT_POINT or GV_ASCII_FORMAT_STD \param dp number of significant digits \param fs field separator \param region_flag check region \param type feature type filter \param field field number \param Clist list of categories to filter features or NULL \param where SQL select where statement to filter features or NULL \param column_names array of columns to be included to the output or NULL "*" as the first item in the array indicates all columns \param header TRUE to print also header \return number of written features \return -1 on error */ int Vect_write_ascii(FILE *ascii, FILE *att, struct Map_info *Map, int ver, int format, int dp, char *fs, int region_flag, int type, int field, const struct cat_list *Clist, const char* where, const char **column_names, int header) { int ltype, ctype, i, cat, line, left, right, found; double *xptr, *yptr, *zptr, x, y; static struct line_pnts *Points; struct line_cats *Cats, *ACats; char *xstring, *ystring, *zstring; size_t xsize, ysize, zsize; struct Cell_head window; struct ilist *fcats; int count, n_skipped; /* where || columns */ struct field_info *Fi; dbDriver *driver; dbValue value; dbHandle handle; int *cats, ncats, more; dbTable *Table; dbString dbstring; dbColumn *Column; dbValue *Value; char *buf; size_t bufsize; dbCursor cursor; /* columns */ char **columns; int *coltypes; char *all_columns; Fi = NULL; driver = NULL; columns = NULL; coltypes = NULL; all_columns = NULL; G_zero(&value, sizeof(dbValue)); db_init_string(&dbstring); xstring = NULL; ystring = NULL; zstring = NULL; xsize = 0; ysize = 0; zsize = 0; buf = NULL; bufsize = 0; /* get the region */ G_get_window(&window); count = ncats = 0; xstring = ystring = zstring = NULL; cats = NULL; if (field > 0 && (where || column_names)) { Fi = Vect_get_field(Map, field); if (!Fi) { G_fatal_error(_("Database connection not defined for layer %d"), field); } driver = db_start_driver(Fi->driver); if (!driver) G_fatal_error(_("Unable to start driver <%s>"), Fi->driver); db_init_handle(&handle); db_set_handle(&handle, Fi->database, NULL); if (db_open_database(driver, &handle) != DB_OK) G_fatal_error(_("Unable to open database <%s> by driver <%s>"), Fi->database, Fi->driver); /* select cats (sorted array) */ ncats = db_select_int(driver, Fi->table, Fi->key, where, &cats); G_debug(3, "%d categories selected from table <%s>", ncats, Fi->table); if (!column_names) { db_close_database(driver); db_shutdown_driver(driver); } else { int icol, ncols; const char *col_name; int len_all = 0; db_set_string(&dbstring, Fi->table); if (db_describe_table(driver, &dbstring, &Table) != DB_OK) { G_warning(_("Unable to describe table <%s>"), Fi->table); return -1; } ncols = db_get_table_number_of_columns(Table); columns = (char **) G_malloc((ncols + 1) * sizeof(char *)); if (column_names[0] && strcmp(column_names[0], "*") == 0) { /* all columns */ icol = 0; for (i = 0; i < ncols; i++) { col_name = db_get_column_name(db_get_table_column(Table, i)); /* key column skipped */ if (strcmp(Fi->key, col_name) != 0) columns[icol++] = G_store(col_name); } columns[icol] = NULL; } else { int j; icol = 0; i = 0; while (column_names[i]) { /* key column skipped */ if (strcmp(Fi->key, column_names[i]) != 0) { found = 0; for (j = 0; j < ncols; j++) { col_name = db_get_column_name(db_get_table_column(Table, j)); if (strcmp(col_name, column_names[i]) == 0) { columns[icol++] = G_store(col_name); found = 1; break; } } if (!found) { G_warning(_("Column <%s> does not exist"), column_names[i]); G_important_message(_("Available columns:")); for (j = 0; j < ncols; j++) { col_name = db_get_column_name(db_get_table_column(Table, j)); G_important_message("%s", col_name); } G_warning(_("Export cancelled")); db_close_database(driver); db_shutdown_driver(driver); return -1; } } i++; } columns[icol] = NULL; } db_zero_string(&dbstring); db_free_table(Table); Table = NULL; if (columns[0]) { /* selected columns only */ i = 0; while (columns[i]) len_all += strlen(columns[i++]); coltypes = G_malloc(i * sizeof(int)); all_columns = G_malloc(len_all + i + 2); i = 0; strcpy(all_columns, columns[0]); while (columns[i]) { /* get column types */ coltypes[i] = db_column_Ctype(driver, Fi->table, columns[i]); if (coltypes[i] < 0) { db_close_database(driver); db_shutdown_driver(driver); G_warning(_("Unknown type of column <%s>, export cancelled"), columns[i]); return -1; } if (i > 0) { strcat(all_columns, ","); strcat(all_columns, columns[i]); } i++; } } else { /* no column or only key column selected */ G_free(columns); columns = NULL; db_close_database(driver); db_shutdown_driver(driver); } } } if (format == GV_ASCII_FORMAT_POINT && header) { /* print header */ if (Map->head.with_z) fprintf(ascii, "east%snorth%sheight%scat", fs, fs, fs); else fprintf(ascii, "east%snorth%scat", fs, fs); if (columns) { for (i = 0; columns[i]; i++) { if (db_select_value (driver, Fi->table, Fi->key, cat, columns[i], &value) < 0) G_fatal_error(_("Unable to select record from table <%s> (key %s, column %s)"), Fi->table, Fi->key, columns[i]); if (columns[i]) fprintf(ascii, "%s%s", fs, columns[i]); else fprintf(ascii, "%s", columns[i]); /* can not happen */ } } fprintf(ascii, "%s", HOST_NEWLINE); } Points = Vect_new_line_struct(); Cats = Vect_new_cats_struct(); ACats = Vect_new_cats_struct(); fcats = Vect_new_list(); /* by default, read_next_line will NOT read Dead lines */ /* but we can override that (in Level I only) by specifying */ /* the type -1, which means match all line types */ Vect_rewind(Map); count = n_skipped = line = 0; while (TRUE) { ltype = Vect_read_next_line(Map, Points, Cats); if (ltype == -1 ) { /* failure */ if (columns) { db_close_database(driver); db_shutdown_driver(driver); free_col_arrays(coltypes, all_columns, column_names && strcmp(column_names[0], "*") == 0 ? columns : NULL); } return -1; } if (ltype == -2) { /* EOF */ if (columns) { db_close_database(driver); db_shutdown_driver(driver); free_col_arrays(coltypes, all_columns, column_names && strcmp(column_names[0], "*") == 0 ? columns : NULL); } break; } line++; if (!(ltype & type)) continue; if (format == GV_ASCII_FORMAT_POINT && !(ltype & GV_POINTS)) continue; found = get_cat(Cats, Clist, cats, ncats, field, &cat); if (!found && field > 0 && ltype == GV_BOUNDARY && type & GV_AREA && Vect_level(Map) > 1) { Vect_get_line_areas(Map, line, &left, &right); if (left < 0) left = Vect_get_isle_area(Map, abs(left)); if (left > 0) { Vect_get_area_cats(Map, left, ACats); found = get_cat(ACats, Clist, cats, ncats, field, &cat); } if (right < 0) right = Vect_get_isle_area(Map, abs(right)); if (!found && right > 0) { Vect_get_area_cats(Map, right, ACats); found = get_cat(ACats, Clist, cats, ncats, field, &cat); } } if (!found) { if (Cats->n_cats < 1) n_skipped++; continue; } if (ver < 5) { Vect_cat_get(Cats, 1, &cat); } switch (ltype) { case GV_BOUNDARY: if (ver == 5) ctype = 'B'; else ctype = 'A'; break; case GV_CENTROID: if (ver < 5) { if (att != NULL) { if (cat > 0) { G_rasprintf(&xstring, &xsize, "%.*f", dp, Points->x[0]); G_trim_decimal(xstring); G_rasprintf(&ystring, &ysize, "%.*f", dp, Points->y[0]); G_trim_decimal(ystring); fprintf(att, "A %s %s %d%s", xstring, ystring, cat, HOST_NEWLINE); } } continue; } ctype = 'C'; break; case GV_LINE: ctype = 'L'; break; case GV_POINT: ctype = 'P'; break; case GV_FACE: ctype = 'F'; break; case GV_KERNEL: ctype = 'K'; break; default: ctype = 'X'; G_warning(_("Unknown feature type %d"), (int)ltype); break; } if (format == GV_ASCII_FORMAT_POINT) { if (region_flag) { if ((window.east < Points->x[0]) || (window.west > Points->x[0])) continue; } G_rasprintf(&xstring, &xsize, "%.*f", dp, Points->x[0]); G_trim_decimal(xstring); if (region_flag) { if ((window.north < Points->y[0]) || (window.south > Points->y[0])) continue; } G_rasprintf(&ystring, &ysize, "%.*f", dp, Points->y[0]); G_trim_decimal(ystring); Vect_field_cat_get(Cats, field, fcats); if (Map->head.with_z && ver == 5) { if (region_flag) { if ((window.top < Points->z[0]) || (window.bottom > Points->z[0])) continue; } G_rasprintf(&zstring, &zsize, "%.*f", dp, Points->z[0]); G_trim_decimal(zstring); fprintf(ascii, "%s%s%s%s%s", xstring, fs, ystring, fs, zstring); } else { fprintf(ascii, "%s%s%s", xstring, fs, ystring); } if (fcats->n_values > 0 && cat > -1) { if (fcats->n_values > 1) { G_warning(_("Feature has more categories. Only one category (%d) " "is exported."), cat); } fprintf(ascii, "%s%d", fs, cat); /* print attributes */ if (columns) { G_rasprintf(&buf, &bufsize, "SELECT %s FROM %s WHERE %s = %d", all_columns, Fi->table, Fi->key, cat); G_debug(2, "SQL: %s", buf); db_set_string(&dbstring, buf); if (db_open_select_cursor (driver, &dbstring, &cursor, DB_SEQUENTIAL) != DB_OK) { db_close_database(driver); db_shutdown_driver(driver); G_fatal_error(_("Cannot select attributes for cat = %d"), cat); } if (db_fetch(&cursor, DB_NEXT, &more) != DB_OK) { db_close_database(driver); db_shutdown_driver(driver); G_fatal_error(_("Unable to fetch data from table")); } Table = db_get_cursor_table(&cursor); for (i = 0; columns[i]; i++) { Column = db_get_table_column(Table, i); Value = db_get_column_value(Column); if (db_test_value_isnull(Value)) { fprintf(ascii, "%s", fs); } else { switch(coltypes[i]) { case DB_C_TYPE_INT: { fprintf(ascii, "%s%d", fs, db_get_value_int(Value)); break; } case DB_C_TYPE_DOUBLE: { fprintf(ascii, "%s%.*f", fs, dp, db_get_value_double(Value)); break; } case DB_C_TYPE_STRING: { fprintf(ascii, "%s%s", fs, db_get_value_string(Value)); break; } case DB_C_TYPE_DATETIME: { break; } case -1: G_fatal_error(_("Column <%s> not found in table <%s>"), columns[i], Fi->table); default: G_fatal_error(_("Column <%s>: unsupported data type"), columns[i]); } } } db_close_cursor(&cursor); } } fprintf(ascii, "%s", HOST_NEWLINE); } else if (format == GV_ASCII_FORMAT_STD) { /* FORMAT_STANDARD */ if (ver == 5 && Cats->n_cats > 0) fprintf(ascii, "%c %d %d%s", ctype, Points->n_points, Cats->n_cats, HOST_NEWLINE); else fprintf(ascii, "%c %d%s", ctype, Points->n_points, HOST_NEWLINE); xptr = Points->x; yptr = Points->y; zptr = Points->z; while (Points->n_points--) { G_rasprintf(&xstring, &xsize, "%.*f", dp, *xptr++); G_trim_decimal(xstring); G_rasprintf(&ystring, &ysize, "%.*f", dp, *yptr++); G_trim_decimal(ystring); if (ver == 5) { if (Map->head.with_z) { G_rasprintf(&zstring, &zsize, "%.*f", dp, *zptr++); G_trim_decimal(zstring); fprintf(ascii, " %-12s %-12s %-12s%s", xstring, ystring, zstring, HOST_NEWLINE); } else { fprintf(ascii, " %-12s %-12s%s", xstring, ystring, HOST_NEWLINE); } } /*Version 4 */ else { fprintf(ascii, " %-12s %-12s%s", ystring, xstring, HOST_NEWLINE); } } if (ver == 5) { for (i = 0; i < Cats->n_cats; i++) { fprintf(ascii, " %-5d %-10d%s", Cats->field[i], Cats->cat[i], HOST_NEWLINE); } } else { if (cat > -1) { if (ltype == GV_POINT) { G_rasprintf(&xstring, &xsize, "%.*f", dp, Points->x[0]); G_trim_decimal(xstring); G_rasprintf(&ystring, &ysize, "%.*f", dp, Points->y[0]); G_trim_decimal(ystring); fprintf(att, "P %s %s %d%s", xstring, ystring, cat, HOST_NEWLINE); } else { x = (Points->x[1] + Points->x[0]) / 2; y = (Points->y[1] + Points->y[0]) / 2; G_rasprintf(&xstring, &xsize, "%.*f", dp, x); G_trim_decimal(xstring); G_rasprintf(&ystring, &ysize, "%.*f", dp, y); G_trim_decimal(ystring); fprintf(att, "L %s %s %d%s", xstring, ystring, cat, HOST_NEWLINE); } } } } else if (format == GV_ASCII_FORMAT_WKT) { if (ltype & (GV_BOUNDARY | GV_CENTROID | GV_FACE | GV_KERNEL)) continue; /* Well-Known Text */ Vect_sfa_line_astext(Points, ltype, Vect_is_3d(Map), dp, ascii); count++; } else { G_fatal_error(_("Unknown format")); } count++; } if (format == GV_ASCII_FORMAT_WKT) { /* process areas - topology required */ int i, area, nareas, isle, nisles; if (Vect_level(Map) < 2) { G_warning(_("Topology not available, unable to process areas")); nareas = 0; } else { nareas = Vect_get_num_areas(Map); } for (area = 1; area <= nareas; area++) { if (!Vect_area_alive(Map, area)) /* skip dead areas */ continue; if (Vect_get_area_cat(Map, area, field) < 0) continue; /* get boundary -> linearring */ if (Vect_get_area_points(Map, area, Points) < 0) { G_warning(_("Unable to get boundary of area id %d"), area); continue; } fprintf(ascii, "POLYGON("); /* write outter ring */ Vect_sfa_line_astext(Points, GV_BOUNDARY, 0, dp, ascii); /* boundary is always 2D */ /* get isles (holes) -> inner rings */ nisles = Vect_get_area_num_isles(Map, area); for (i = 0; i < nisles; i++) { /* get isle boundary -> linearring */ isle = Vect_get_area_isle(Map, area, i); if (Vect_get_isle_points(Map, isle, Points) < 0) { G_warning(_("Unable to get boundary of isle id %d (area id %d)"), isle, area); continue; } fprintf(ascii, ", "); /* write inner ring */ Vect_sfa_line_astext(Points, GV_BOUNDARY, 0, dp, ascii); /* boundary is always 2D */ } fprintf(ascii, ")%s", HOST_NEWLINE); count++; } } if (n_skipped > 0) G_important_message(_("%d features without category skipped. To export also " "features without category use '%s=-1'."), n_skipped, "layer"); Vect_destroy_line_struct(Points); Vect_destroy_cats_struct(Cats); Vect_destroy_cats_struct(ACats); return count; }
int main(int argc, char **argv) { int lfield, pfield, n_points, n_outside, n_found, n_no_record, n_many_records; int line, type, nlines; double thresh, multip; struct Option *lines_opt, *points_opt; struct Option *lfield_opt, *pfield_opt; struct Option *driver_opt, *database_opt, *table_opt, *thresh_opt; struct GModule *module; const char *mapset; struct Map_info LMap, PMap; struct line_cats *LCats, *PCats; struct line_pnts *LPoints, *PPoints; dbDriver *rsdriver; dbHandle rshandle; dbString rsstmt; G_gisinit(argv[0]); module = G_define_module(); G_add_keyword(_("vector")); G_add_keyword(_("Linear Reference System")); G_add_keyword(_("networking")); module->description = _("Finds line id and real km+offset for given points in vector map " "using linear reference system."); lines_opt = G_define_standard_option(G_OPT_V_INPUT); lines_opt->key = "lines"; lines_opt->description = _("Input vector map containing lines"); points_opt = G_define_standard_option(G_OPT_V_INPUT); points_opt->key = "points"; points_opt->description = _("Input vector map containing points"); lfield_opt = G_define_standard_option(G_OPT_V_FIELD); lfield_opt->key = "llayer"; lfield_opt->answer = "1"; lfield_opt->description = _("Line layer"); pfield_opt = G_define_standard_option(G_OPT_V_FIELD); pfield_opt->key = "player"; pfield_opt->answer = "1"; pfield_opt->description = _("Point layer"); driver_opt = G_define_option(); driver_opt->key = "rsdriver"; driver_opt->type = TYPE_STRING; driver_opt->required = NO; driver_opt->description = _("Driver name for reference system table"); driver_opt->options = db_list_drivers(); driver_opt->answer = db_get_default_driver_name(); database_opt = G_define_option(); database_opt->key = "rsdatabase"; database_opt->type = TYPE_STRING; database_opt->required = NO; database_opt->description = _("Database name for reference system table"); database_opt->answer = db_get_default_database_name(); table_opt = G_define_option(); table_opt->key = "rstable"; table_opt->type = TYPE_STRING; table_opt->required = YES; table_opt->description = _("Name of the reference system table"); thresh_opt = G_define_option(); thresh_opt->key = "threshold"; thresh_opt->type = TYPE_DOUBLE; thresh_opt->required = NO; thresh_opt->answer = "1000"; thresh_opt->description = _("Maximum distance to nearest line"); if (G_parser(argc, argv)) exit(EXIT_FAILURE); LCats = Vect_new_cats_struct(); PCats = Vect_new_cats_struct(); LPoints = Vect_new_line_struct(); PPoints = Vect_new_line_struct(); lfield = atoi(lfield_opt->answer); pfield = atoi(pfield_opt->answer); multip = 1000; /* Number of map units per MP unit */ thresh = atof(thresh_opt->answer); /* Open input lines */ mapset = G_find_vector2(lines_opt->answer, NULL); if (mapset == NULL) G_fatal_error(_("Vector map <%s> not found"), lines_opt->answer); Vect_set_open_level(2); if (Vect_open_old(&LMap, lines_opt->answer, mapset) < 0) G_fatal_error(_("Unable to open vector map <%s>"), lines_opt->answer); /* Open input points */ mapset = G_find_vector2(points_opt->answer, NULL); if (mapset == NULL) G_fatal_error(_("Vector map <%s> not found"), points_opt->answer); Vect_set_open_level(2); if (Vect_open_old(&PMap, points_opt->answer, mapset) < 0) G_fatal_error(_("Unable to open vector map <%s>"), points_opt->answer); db_init_handle(&rshandle); db_init_string(&rsstmt); rsdriver = db_start_driver(driver_opt->answer); db_set_handle(&rshandle, database_opt->answer, NULL); if (db_open_database(rsdriver, &rshandle) != DB_OK) G_fatal_error(_("Unable to open database for reference table")); n_points = n_outside = n_found = n_no_record = n_many_records = 0; nlines = Vect_get_num_lines(&PMap); G_debug(2, "nlines = %d", nlines); G_message("pcat|lid|mpost|offset"); for (line = 1; line <= nlines; line++) { int nearest, pcat, lcat, lid, ret; double along, mpost, offset; G_debug(3, "point = %d", line); type = Vect_read_line(&PMap, PPoints, PCats, line); if (type != GV_POINT) continue; Vect_cat_get(PCats, pfield, &pcat); if (pcat < 0) continue; n_points++; nearest = Vect_find_line(&LMap, PPoints->x[0], PPoints->y[0], 0.0, GV_LINE, thresh, 0, 0); fprintf(stdout, "%d", pcat); if (nearest <= 0) { fprintf(stdout, "|-|- # outside threshold\n"); n_outside++; continue; } /* Read nearest line */ Vect_read_line(&LMap, LPoints, LCats, nearest); Vect_cat_get(LCats, lfield, &lcat); Vect_line_distance(LPoints, PPoints->x[0], PPoints->y[0], 0.0, 0, NULL, NULL, NULL, NULL, NULL, &along); G_debug(3, " nearest = %d lcat = %d along = %f", nearest, lcat, along); if (lcat >= 0) { ret = LR_get_milepost(rsdriver, table_opt->answer, "lcat", "lid", "start_map", "end_map", "start_mp", "start_off", "end_mp", "end_off", lcat, along, multip, &lid, &mpost, &offset); } else { ret = 0; } if (ret == 0) { n_no_record++; fprintf(stdout, "|-|- # no record\n"); continue; } if (ret == 2) { n_many_records++; fprintf(stdout, "|-|- # too many records\n"); continue; } G_debug(3, " lid = %d mpost = %f offset = %f", lid, mpost, offset); fprintf(stdout, "|%d|%f+%f\n", lid, mpost, offset); n_found++; } db_close_database(rsdriver); /* Free, close ... */ Vect_close(&LMap); Vect_close(&PMap); G_message(_n("[%d] point read from input", "[%d] points read from input", n_points), n_points); G_message(_n("[%d] position found", "[%d] positions found", n_found), n_found); if (n_outside) G_message(_n("[%d] point outside threshold", "[%d] points outside threshold", n_outside), n_outside); if (n_no_record) G_message(_n("[%d] point - no record found", "[%d] points - no record found", n_no_record), n_no_record); if (n_many_records) G_message(_n("[%d] point - too many records found", "[%d] points - too many records found", n_many_records), n_many_records); exit(EXIT_SUCCESS); }
int main(int argc, char **argv) { char *mapset; int ret, level; int i, stat = 0, type, display; int chcat = 0; int r, g, b; int has_color, has_fcolor; struct color_rgb color, fcolor; double size; int default_width; double width_scale; int verbose = FALSE; double minreg, maxreg, reg; char map_name[128]; struct GModule *module; struct Option *map_opt; struct Option *color_opt, *fcolor_opt, *rgbcol_opt, *zcol_opt; struct Option *type_opt, *display_opt; struct Option *icon_opt, *size_opt, *sizecolumn_opt, *rotcolumn_opt; struct Option *where_opt; struct Option *field_opt, *cat_opt, *lfield_opt; struct Option *lcolor_opt, *bgcolor_opt, *bcolor_opt; struct Option *lsize_opt, *font_opt, *xref_opt, *yref_opt; struct Option *attrcol_opt, *maxreg_opt, *minreg_opt; struct Option *width_opt, *wcolumn_opt, *wscale_opt; struct Option *render_opt; struct Flag *verbose_flag; /* please remove before GRASS 7 released */ struct Flag *id_flag, *table_acolors_flag, *cats_acolors_flag, *x_flag, *zcol_flag; struct cat_list *Clist; int *cats, ncat; LATTR lattr; struct Map_info Map; struct field_info *fi; dbDriver *driver; dbHandle handle; struct Cell_head window; BOUND_BOX box; double overlap; /* Initialize the GIS calls */ G_gisinit(argv[0]); module = G_define_module(); module->keywords = _("display, vector"); module->description = _("Displays user-specified vector map " "in the active graphics frame."); map_opt = G_define_standard_option(G_OPT_V_MAP); display_opt = G_define_option(); display_opt->key = "display"; display_opt->type = TYPE_STRING; display_opt->required = YES; display_opt->multiple = YES; display_opt->answer = "shape"; display_opt->options = "shape,cat,topo,dir,attr,zcoor"; display_opt->description = _("Display"); display_opt->descriptions = _("shape;Display geometry of features;" "cat;Display category numbers of features;" "topo;Display topology information (nodes, edges);" "dir;Display direction of linear features;" "attr;Display selected attribute based on 'attrcol';" "zcoor;Display z-coordinate of features (only for 3D vector maps)"); /* Query */ type_opt = G_define_standard_option(G_OPT_V_TYPE); type_opt->answer = "point,line,boundary,centroid,area,face"; type_opt->options = "point,line,boundary,centroid,area,face"; type_opt->guisection = _("Selection"); field_opt = G_define_standard_option(G_OPT_V_FIELD); field_opt->label = _("Layer number (if -1, all layers are displayed)"); field_opt->gisprompt = "old_layer,layer,layer_all"; field_opt->guisection = _("Selection"); cat_opt = G_define_standard_option(G_OPT_V_CATS); cat_opt->guisection = _("Selection"); where_opt = G_define_standard_option(G_OPT_WHERE); where_opt->guisection = _("Selection"); /* Colors */ color_opt = G_define_option(); color_opt->key = "color"; color_opt->type = TYPE_STRING; color_opt->answer = DEFAULT_FG_COLOR; color_opt->label = _("Feature color"); color_opt->guisection = _("Colors"); color_opt->gisprompt = "old_color,color,color_none"; color_opt->description = _("Either a standard GRASS color, R:G:B triplet, or \"none\""); fcolor_opt = G_define_option(); fcolor_opt->key = "fcolor"; fcolor_opt->type = TYPE_STRING; fcolor_opt->answer = "200:200:200"; fcolor_opt->label = _("Area fill color"); fcolor_opt->guisection = _("Colors"); fcolor_opt->gisprompt = "old_color,color,color_none"; fcolor_opt->description = _("Either a standard GRASS color, R:G:B triplet, or \"none\""); rgbcol_opt = G_define_standard_option(G_OPT_COLUMN); rgbcol_opt->key = "rgb_column"; rgbcol_opt->guisection = _("Colors"); rgbcol_opt->description = _("Name of color definition column (for use with -a flag)"); rgbcol_opt->answer = "GRASSRGB"; zcol_opt = G_define_option(); zcol_opt->key = "zcolor"; zcol_opt->key_desc = "style"; zcol_opt->type = TYPE_STRING; zcol_opt->required = NO; zcol_opt->description = _("Type of color table (for use with -z flag)"); zcol_opt->answer = "terrain"; zcol_opt->guisection = _("Colors"); /* Lines */ width_opt = G_define_option(); width_opt->key = "width"; width_opt->type = TYPE_INTEGER; width_opt->answer = "0"; width_opt->guisection = _("Lines"); width_opt->description = _("Line width"); wcolumn_opt = G_define_standard_option(G_OPT_COLUMN); wcolumn_opt->key = "wcolumn"; wcolumn_opt->guisection = _("Lines"); wcolumn_opt->description = _("Name of column for line widths (these values will be scaled by wscale)"); wscale_opt = G_define_option(); wscale_opt->key = "wscale"; wscale_opt->type = TYPE_DOUBLE; wscale_opt->answer = "1"; wscale_opt->guisection = _("Lines"); wscale_opt->description = _("Scale factor for wcolumn"); /* Symbols */ icon_opt = G_define_option(); icon_opt->key = "icon"; icon_opt->type = TYPE_STRING; icon_opt->required = NO; icon_opt->multiple = NO; icon_opt->guisection = _("Symbols"); icon_opt->answer = "basic/x"; /* This could also use ->gisprompt = "old,symbol,symbol" instead of ->options */ icon_opt->options = icon_files(); icon_opt->description = _("Point and centroid symbol"); size_opt = G_define_option(); size_opt->key = "size"; size_opt->type = TYPE_DOUBLE; size_opt->answer = "5"; size_opt->guisection = _("Symbols"); size_opt->label = _("Symbol size"); size_opt->description = _("When used with the size_column option this becomes the scale factor"); sizecolumn_opt = G_define_standard_option(G_OPT_COLUMN); sizecolumn_opt->key = "size_column"; sizecolumn_opt->guisection = _("Symbols"); sizecolumn_opt->description = _("Name of numeric column containing symbol size"); rotcolumn_opt = G_define_standard_option(G_OPT_COLUMN); rotcolumn_opt->key = "rot_column"; rotcolumn_opt->guisection = _("Symbols"); rotcolumn_opt->label = _("Name of numeric column containing symbol rotation angle"); rotcolumn_opt->description = _("Measured in degrees CCW from east"); /* Labels */ lfield_opt = G_define_standard_option(G_OPT_V_FIELD); lfield_opt->key = "llayer"; lfield_opt->guisection = _("Labels"); lfield_opt->description = _("Layer number for labels (default: the given layer number)"); attrcol_opt = G_define_standard_option(G_OPT_COLUMN); attrcol_opt->key = "attrcol"; attrcol_opt->multiple = NO; /* or fix attr.c, around line 102 */ attrcol_opt->guisection = _("Labels"); attrcol_opt->description = _("Name of column to be displayed"); lcolor_opt = G_define_option(); lcolor_opt->key = "lcolor"; lcolor_opt->type = TYPE_STRING; lcolor_opt->answer = "red"; lcolor_opt->label = _("Label color"); lcolor_opt->guisection = _("Labels"); lcolor_opt->gisprompt = "old_color,color,color"; lcolor_opt->description = _("Either a standard color name or R:G:B triplet"); bgcolor_opt = G_define_option(); bgcolor_opt->key = "bgcolor"; bgcolor_opt->type = TYPE_STRING; bgcolor_opt->answer = "none"; bgcolor_opt->guisection = _("Labels"); bgcolor_opt->label = _("Label background color"); bgcolor_opt->gisprompt = "old_color,color,color_none"; bgcolor_opt->description = _("Either a standard GRASS color, R:G:B triplet, or \"none\""); bcolor_opt = G_define_option(); bcolor_opt->key = "bcolor"; bcolor_opt->type = TYPE_STRING; bcolor_opt->answer = "none"; bcolor_opt->guisection = _("Labels"); bcolor_opt->label = _("Label border color"); bcolor_opt->gisprompt = "old_color,color,color_none"; bcolor_opt->description = _("Either a standard GRASS color, R:G:B triplet, or \"none\""); lsize_opt = G_define_option(); lsize_opt->key = "lsize"; lsize_opt->type = TYPE_INTEGER; lsize_opt->answer = "8"; lsize_opt->guisection = _("Labels"); lsize_opt->description = _("Label size (pixels)"); font_opt = G_define_option(); font_opt->key = "font"; font_opt->type = TYPE_STRING; font_opt->guisection = _("Labels"); font_opt->description = _("Font name"); xref_opt = G_define_option(); xref_opt->key = "xref"; xref_opt->type = TYPE_STRING; xref_opt->guisection = _("Labels"); xref_opt->answer = "left"; xref_opt->options = "left,center,right"; xref_opt->description = _("Label horizontal justification"); yref_opt = G_define_option(); yref_opt->key = "yref"; yref_opt->type = TYPE_STRING; yref_opt->guisection = _("Labels"); yref_opt->answer = "center"; yref_opt->options = "top,center,bottom"; yref_opt->description = _("Label vertical justification"); minreg_opt = G_define_option(); minreg_opt->key = "minreg"; minreg_opt->type = TYPE_DOUBLE; minreg_opt->required = NO; minreg_opt->description = _("Minimum region size (average from height and width) " "when map is displayed"); maxreg_opt = G_define_option(); maxreg_opt->key = "maxreg"; maxreg_opt->type = TYPE_DOUBLE; maxreg_opt->required = NO; maxreg_opt->description = _("Maximum region size (average from height and width) " "when map is displayed"); render_opt = G_define_option(); render_opt->key = "render"; render_opt->type = TYPE_STRING; render_opt->required = NO; render_opt->multiple = NO; render_opt->answer = "c"; render_opt->options = "g,r,d,c,l"; render_opt->description = _("Rendering method for filled polygons"); render_opt->descriptions = _("g;use the libgis render functions (features: clipping);" "r;use the raster graphics library functions (features: polylines);" "d;use the display library basic functions (features: polylines);" "c;use the display library clipping functions (features: clipping);" "l;use the display library culling functions (features: culling, polylines)"); /* please remove before GRASS 7 released */ verbose_flag = G_define_flag(); verbose_flag->key = 'v'; verbose_flag->description = _("Run verbosely"); /* Colors */ table_acolors_flag = G_define_flag(); table_acolors_flag->key = 'a'; table_acolors_flag->guisection = _("Colors"); table_acolors_flag->description = _("Get colors from map table column (of form RRR:GGG:BBB)"); cats_acolors_flag = G_define_flag(); cats_acolors_flag->key = 'c'; cats_acolors_flag->guisection = _("Colors"); cats_acolors_flag->description = _("Random colors according to category number " "(or layer number if 'layer=-1' is given)"); /* Query */ id_flag = G_define_flag(); id_flag->key = 'i'; id_flag->guisection = _("Selection"); id_flag->description = _("Use values from 'cats' option as feature id"); x_flag = G_define_flag(); x_flag->key = 'x'; x_flag->description = _("Don't add to list of vectors and commands in monitor " "(it won't be drawn if the monitor is refreshed)"); zcol_flag = G_define_flag(); zcol_flag->key = 'z'; zcol_flag->description = _("Colorize polygons according to z height"); zcol_flag->guisection = _("Colors"); /* Check command line */ if (G_parser(argc, argv)) exit(EXIT_FAILURE); if (G_strcasecmp(render_opt->answer, "g") == 0) render = RENDER_GPP; else if (G_strcasecmp(render_opt->answer, "r") == 0) render = RENDER_RPA; else if (G_strcasecmp(render_opt->answer, "d") == 0) render = RENDER_DP; else if (G_strcasecmp(render_opt->answer, "c") == 0) render = RENDER_DPC; else if (G_strcasecmp(render_opt->answer, "l") == 0) render = RENDER_DPL; else render = RENDER_GPP; /* please remove -v flag before GRASS 7 released */ if (verbose_flag->answer) { G_putenv("GRASS_VERBOSE", "3"); G_warning(_("The '-v' flag is superseded and will be removed " "in future. Please use '--verbose' instead.")); } /* but keep this */ if (G_verbose() > G_verbose_std()) verbose = TRUE; G_get_set_window(&window); if (R_open_driver() != 0) G_fatal_error(_("No graphics device selected")); /* Read map options */ /* Check min/max region */ reg = ((window.east - window.west) + (window.north - window.south)) / 2; if (minreg_opt->answer) { minreg = atof(minreg_opt->answer); if (reg < minreg) { G_message(_("Region size is lower than minreg, nothing displayed.")); D_add_to_list(G_recreate_command()); exit(EXIT_SUCCESS); } } if (maxreg_opt->answer) { maxreg = atof(maxreg_opt->answer); if (reg > maxreg) { G_message(_("Region size is greater than maxreg, nothing displayed.")); D_add_to_list(G_recreate_command()); exit(EXIT_SUCCESS); } } G_strcpy(map_name, map_opt->answer); default_width = atoi(width_opt->answer); if (default_width < 0) default_width = 0; width_scale = atof(wscale_opt->answer); if (table_acolors_flag->answer && cats_acolors_flag->answer) { cats_acolors_flag->answer = '\0'; G_warning(_("The '-c' and '-a' flags cannot be used together, " "the '-c' flag will be ignored!")); } color = G_standard_color_rgb(WHITE); ret = G_str_to_color(color_opt->answer, &r, &g, &b); if (ret == 1) { has_color = 1; color.r = r; color.g = g; color.b = b; } else if (ret == 2) { /* none */ has_color = 0; } else if (ret == 0) { /* error */ G_fatal_error(_("Unknown color: [%s]"), color_opt->answer); } fcolor = G_standard_color_rgb(WHITE); ret = G_str_to_color(fcolor_opt->answer, &r, &g, &b); if (ret == 1) { has_fcolor = 1; fcolor.r = r; fcolor.g = g; fcolor.b = b; } else if (ret == 2) { /* none */ has_fcolor = 0; } else if (ret == 0) { /* error */ G_fatal_error(_("Unknown color: '%s'"), fcolor_opt->answer); } size = atof(size_opt->answer); /* Make sure map is available */ mapset = G_find_vector2(map_name, ""); if (mapset == NULL) G_fatal_error(_("Vector map <%s> not found"), map_name); /* if where_opt was specified select categories from db * otherwise parse cat_opt */ Clist = Vect_new_cat_list(); Clist->field = atoi(field_opt->answer); /* open vector */ level = Vect_open_old(&Map, map_name, mapset); if (where_opt->answer) { if (Clist->field < 1) G_fatal_error(_("'layer' must be > 0 for 'where'.")); chcat = 1; if ((fi = Vect_get_field(&Map, Clist->field)) == NULL) G_fatal_error(_("Database connection not defined")); if (fi != NULL) { driver = db_start_driver(fi->driver); if (driver == NULL) G_fatal_error(_("Unable to start driver <%s>"), fi->driver); db_init_handle(&handle); db_set_handle(&handle, fi->database, NULL); if (db_open_database(driver, &handle) != DB_OK) G_fatal_error(_("Unable to open database <%s>"), fi->database); ncat = db_select_int(driver, fi->table, fi->key, where_opt->answer, &cats); db_close_database(driver); db_shutdown_driver(driver); Vect_array_to_cat_list(cats, ncat, Clist); } } else if (cat_opt->answer) { if (Clist->field < 1) G_fatal_error(_("'layer' must be > 0 for 'cats'.")); chcat = 1; ret = Vect_str_to_cat_list(cat_opt->answer, Clist); if (ret > 0) G_warning(_("%d errors in cat option"), ret); } type = Vect_option_to_types(type_opt); i = 0; display = 0; while (display_opt->answers[i]) { switch (display_opt->answers[i][0]) { case 's': display |= DISP_SHAPE; break; case 'c': display |= DISP_CAT; break; case 't': display |= DISP_TOPO; break; case 'd': display |= DISP_DIR; break; case 'a': display |= DISP_ATTR; break; case 'z': display |= DISP_ZCOOR; break; } i++; } /* Read label options */ if (lfield_opt->answer != NULL) lattr.field = atoi(lfield_opt->answer); else lattr.field = Clist->field; lattr.color.R = lattr.color.G = lattr.color.B = 255; if (G_str_to_color(lcolor_opt->answer, &r, &g, &b)) { lattr.color.R = r; lattr.color.G = g; lattr.color.B = b; } lattr.has_bgcolor = 0; if (G_str_to_color(bgcolor_opt->answer, &r, &g, &b) == 1) { lattr.has_bgcolor = 1; lattr.bgcolor.R = r; lattr.bgcolor.G = g; lattr.bgcolor.B = b; } lattr.has_bcolor = 0; if (G_str_to_color(bcolor_opt->answer, &r, &g, &b) == 1) { lattr.has_bcolor = 1; lattr.bcolor.R = r; lattr.bcolor.G = g; lattr.bcolor.B = b; } lattr.size = atoi(lsize_opt->answer); lattr.font = font_opt->answer; switch (xref_opt->answer[0]) { case 'l': lattr.xref = LLEFT; break; case 'c': lattr.xref = LCENTER; break; case 'r': lattr.xref = LRIGHT; break; } switch (yref_opt->answer[0]) { case 't': lattr.yref = LTOP; break; case 'c': lattr.yref = LCENTER; break; case 'b': lattr.yref = LBOTTOM; break; } D_setup(0); G_setup_plot(D_get_d_north(), D_get_d_south(), D_get_d_west(), D_get_d_east(), D_move_abs, D_cont_abs); if (verbose) G_message(_("Plotting ...")); if (level >= 2) Vect_get_map_box(&Map, &box); if (level >= 2 && (window.north < box.S || window.south > box.N || window.east < box.W || window.west > G_adjust_easting(box.E, &window))) { G_message(_("The bounding box of the map is outside the current region, " "nothing drawn.")); stat = 0; } else { overlap = G_window_percentage_overlap(&window, box.N, box.S, box.E, box.W); G_debug(1, "overlap = %f \n", overlap); if (overlap < 1) Vect_set_constraint_region(&Map, window.north, window.south, window.east, window.west, PORT_DOUBLE_MAX, -PORT_DOUBLE_MAX); /* default line width */ if (!wcolumn_opt->answer) D_line_width(default_width); if (type & GV_AREA) { if (level >= 2) { if (display & DISP_SHAPE) { stat = darea(&Map, Clist, has_color ? &color : NULL, has_fcolor ? &fcolor : NULL, chcat, (int)id_flag->answer, table_acolors_flag->answer, cats_acolors_flag->answer, &window, rgbcol_opt->answer, default_width, wcolumn_opt->answer, width_scale, zcol_flag->answer, zcol_opt->answer); } if (wcolumn_opt->answer) D_line_width(default_width); } else G_warning(_("Unable to display areas, topology not available")); } if (display & DISP_SHAPE) { if (id_flag->answer && level < 2) { G_warning(_("Unable to display lines by id, topology not available")); } else { stat = plot1(&Map, type, Clist, has_color ? &color : NULL, has_fcolor ? &fcolor : NULL, chcat, icon_opt->answer, size, sizecolumn_opt->answer, rotcolumn_opt->answer, (int)id_flag->answer, table_acolors_flag->answer, cats_acolors_flag->answer, rgbcol_opt->answer, default_width, wcolumn_opt->answer, width_scale, zcol_flag->answer, zcol_opt->answer); if (wcolumn_opt->answer) D_line_width(default_width); } } if (has_color) { R_RGB_color(color.r, color.g, color.b); if (display & DISP_DIR) stat = dir(&Map, type, Clist, chcat, size); } /* reset line width: Do we need to get line width from display * driver (not implemented)? It will help restore previous line * width (not just 0) determined by another module (e.g., * d.linewidth). */ if (!wcolumn_opt->answer) R_line_width(0); if (display & DISP_CAT) stat = label(&Map, type, Clist, &lattr, chcat); if (display & DISP_ATTR) stat = attr(&Map, type, attrcol_opt->answer, Clist, &lattr, chcat); if (display & DISP_ZCOOR) stat = zcoor(&Map, type, &lattr); if (display & DISP_TOPO) { if (level >= 2) stat = topo(&Map, type, &lattr); else G_warning(_("Unable to display topology, not available")); } } if (!x_flag->answer) { D_add_to_list(G_recreate_command()); D_set_dig_name(G_fully_qualified_name(map_name, mapset)); D_add_to_dig_list(G_fully_qualified_name(map_name, mapset)); } R_close_driver(); if (verbose) G_done_msg(" "); Vect_close(&Map); Vect_destroy_cat_list(Clist); exit(stat); }