1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
#include <stdio.h>
#include <stdlib.h>
#include <unistd.h>
#include <string.h>
#include "gis.h"
#include "dbmi.h"
#include "Vect.h"
#include "glocale.h"
/* * convert dig_cats to database table
* return: number of records inserted
* -1 error */
int attributes (char *in, struct Map_info *Out )
{
int i, cat, count;
char *mapset;
struct Categories Cats;
char buf[1024];
dbString sql, lab;
dbDriver *driver;
dbHandle handle;
struct field_info *fi;
int len, clen;
db_init_string (&sql);
db_init_string (&lab);
/* Find vector mapset */
if ( NULL == (mapset = G_find_file ("dig", in, "") )) {
G_warning ( _("Input vector was not found.")) ; /* Should not happen */
return 0;
}
/* Find dig_cats if exists */
if ( NULL == G_find_file ("dig_cats", in, mapset) ) {
G_message ( _("No category labels (dig_cats) found, no table created.\n")) ;
return 0;
}
if (G_read_vector_cats ( in, mapset, &Cats) == -1) {
G_warning ( _("Cannot open dig_cats file."));
return -1;
}
fi = Vect_default_field_info ( Out, 1, NULL, GV_1TABLE );
Vect_map_add_dblink ( Out, 1, NULL, fi->table, "cat", fi->database, fi->driver);
/* Get maximum column length */
clen = 0;
for ( i = 0; i < Cats.ncats; i++) {
len = strlen ( Cats.labels[i] );
if ( len > clen ) clen = len;
}
clen += 10;
/* Create new table */
sprintf ( buf, "create table %s ( cat integer, label varchar(%d) )", fi->table, clen );
db_append_string ( &sql, buf );
G_debug ( 1, db_get_string ( &sql ) );
driver = db_start_driver( fi->driver );
if (driver == NULL) G_fatal_error ( "Cannot open driver %s", fi->driver );
db_init_handle (&handle);
db_set_handle (&handle, Vect_subst_var(fi->database, Out), NULL);
if (db_open_database(driver, &handle) != DB_OK) {
db_shutdown_driver(driver);
G_fatal_error ( _("Cannot open database %s"), fi->database );
}
if (db_execute_immediate (driver, &sql) != DB_OK ) {
db_close_database(driver);
db_shutdown_driver(driver);
G_fatal_error ( _("Cannot create table: %s"), db_get_string ( &sql ) );
}
if ( db_create_index2(driver, fi->table, "cat" ) != DB_OK )
G_warning ( "Cannot create index" );
if (db_grant_on_table (driver, fi->table, DB_PRIV_SELECT, DB_GROUP|DB_PUBLIC ) != DB_OK )
G_fatal_error ( "Cannot grant privileges on table %s", fi->table );
G_debug ( 1, "ncats = %d", Cats.ncats );
count = 0;
for ( i = 0; i < Cats.ncats; i++) {
cat = (int) Cats.q.table[i].dLow; /* cats are in dLow/High not in cLow/High !!! */
G_debug ( 3, "%d cat = %d label = %s", i, cat, Cats.labels[i] );
db_set_string ( &lab, Cats.labels[i]);
db_double_quote_string ( &lab );
sprintf (buf, "insert into %s values ( %d, '%s')", fi->table, cat, db_get_string(&lab) );
db_set_string ( &sql, buf);
G_debug ( 3, db_get_string ( &sql ) );
if (db_execute_immediate (driver, &sql) != DB_OK ) {
db_close_database(driver);
db_shutdown_driver(driver);
G_fatal_error ( _("Cannot insert into table: %s"), db_get_string ( &sql ) );
}
count++;
}
db_close_database(driver);
db_shutdown_driver(driver);
return count;
}
|
