Skip to content

Commit

Permalink
Browse files Browse the repository at this point in the history
added isOrphan, deleteAttributes
git-svn-id: http://svn.osgeo.org/qgis/trunk@4941 c8812cc2-4d05-0410-92ff-de0c093fc19c
  • Loading branch information
rblazek committed Mar 2, 2006
1 parent 70f08c7 commit 8d8b5e1
Show file tree
Hide file tree
Showing 2 changed files with 157 additions and 34 deletions.
168 changes: 134 additions & 34 deletions src/providers/grass/qgsgrassprovider.cpp
Expand Up @@ -1361,6 +1361,7 @@ bool QgsGrassProvider::startEdit ( void )

return false;
}
Vect_set_category_index_update ( map->map );

// Write history
Vect_hist_command ( map->map );
Expand Down Expand Up @@ -1963,7 +1964,62 @@ int QgsGrassProvider::dbLinkField ( int link )
return ( fi->number );
}

QString *QgsGrassProvider::executeSql ( int field, const QString &sql )
{
#ifdef QGISDEBUG
std::cerr << "QgsGrassProvider::executeSql field = " << field << std::endl;
#endif

QString *error = new QString();
struct field_info *fi = Vect_get_field( mMap, field); // should work also with field = 0

// Read attributes
if ( fi == NULL ) {
#ifdef QGISDEBUG
std::cerr << "No field info -> no attributes" << std::endl;
#endif
error->setLatin1( "Cannot get field info" );
return error;
}

#ifdef QGISDEBUG
std::cerr << "Field info found -> open database" << std::endl;
#endif

QgsGrass::setMapset ( mGisdbase, mLocation, mMapset );
dbDriver *driver = db_start_driver_open_database ( fi->driver, fi->database );

if ( driver == NULL ) {
std::cerr << "Cannot open database " << fi->database << " by driver " << fi->driver << std::endl;
error->setAscii("Cannot open database");
return error;
}

#ifdef QGISDEBUG
std::cerr << "Database opened" << std::endl;
#endif

dbString dbstr;
db_init_string (&dbstr);
db_set_string (&dbstr, (char *)sql.latin1());

#ifdef QGISDEBUG
std::cerr << "SQL: " << db_get_string(&dbstr) << std::endl;
#endif

int ret = db_execute_immediate (driver, &dbstr);

if ( ret != DB_OK) {
std::cerr << "Error: " << db_get_error_msg() << std::endl;
error->setLatin1( db_get_error_msg() );
}

db_close_database_shutdown_driver ( driver );
db_free_string(&dbstr);

return error;

}

QString *QgsGrassProvider::createTable ( int field, const QString &key, const QString &columns )
{
Expand Down Expand Up @@ -2061,62 +2117,99 @@ QString *QgsGrassProvider::addColumn ( int field, const QString &column )
return error;
}

QString query;

query.sprintf("alter table %s add column %s", fi->table, column.latin1() );

delete error;
return executeSql ( field, query );
}

QString *QgsGrassProvider::insertAttributes ( int field, int cat )
{
#ifdef QGISDEBUG
std::cerr << "Field info found -> open database" << std::endl;
std::cerr << "QgsGrassProvider::insertAttributes() field = " << field << " cat = " << cat << std::endl;
#endif
QgsGrass::setMapset ( mGisdbase, mLocation, mMapset );
dbDriver *driver = db_start_driver_open_database ( fi->driver, fi->database );

if ( driver == NULL ) {
std::cerr << "Cannot open database " << fi->database << " by driver " << fi->driver << std::endl;
error->setAscii("Cannot open database");
QString *error = new QString();
struct field_info *fi = Vect_get_field( mMap, field); // should work also with field = 0

// Read attributes
if ( fi == NULL ) {
#ifdef QGISDEBUG
std::cerr << "No field info -> no attributes" << std::endl;
#endif
error->setLatin1( "Cannot get field info" );
return error;
}

#ifdef QGISDEBUG
std::cerr << "Database opened -> add column" << std::endl;
#endif

dbString dbstr;
db_init_string (&dbstr);
QString query;

query.sprintf("alter table %s add column %s", fi->table, column.latin1() );
db_set_string (&dbstr, (char *)query.latin1());

query.sprintf("insert into %s ( %s ) values ( %d )", fi->table, fi->key, cat );

delete error;
return executeSql ( field, query );
}

QString *QgsGrassProvider::deleteAttributes ( int field, int cat )
{
#ifdef QGISDEBUG
std::cerr << "SQL: " << db_get_string(&dbstr) << std::endl;
std::cerr << "QgsGrassProvider::deleteAttributes() field = " << field << " cat = " << cat << std::endl;
#endif

int ret = db_execute_immediate (driver, &dbstr);
QString *error = new QString();
struct field_info *fi = Vect_get_field( mMap, field); // should work also with field = 0

if ( ret != DB_OK) {
std::cerr << "Error: " << db_get_error_msg() << std::endl;
error->setLatin1( db_get_error_msg() );
// Read attributes
if ( fi == NULL ) {
#ifdef QGISDEBUG
std::cerr << "No field info -> no attributes" << std::endl;
#endif
error->setLatin1( "Cannot get field info" );
return error;
}

db_close_database_shutdown_driver ( driver );
db_free_string(&dbstr);
QString query;

query.sprintf("delete from %s where %s = %d", fi->table, fi->key, cat );

return error;
delete error;
return executeSql ( field, query );
}

QString *QgsGrassProvider::insertAttributes ( int field, int cat )
QString *QgsGrassProvider::isOrphan ( int field, int cat, int *orphan)
{
#ifdef QGISDEBUG
std::cerr << "QgsGrassProvider::insertAttributes() field = " << field << " cat = " << cat << std::endl;
std::cerr << "QgsGrassProvider::isOrphan() field = " << field << " cat = " << cat << std::endl;
#endif

QString *error = new QString();

// Check first if another line with such cat exists
int fieldIndex = Vect_cidx_get_field_index ( mMap, field );
if ( fieldIndex >= 0 )
{
int t, id;
int ret = Vect_cidx_find_next ( mMap, fieldIndex, cat,
GV_POINTS|GV_LINES, 0, &t, &id );

if ( ret >= 0 ) {
// Category exists
*orphan = false;
return error;
}
}

// Check if attribute exists
struct field_info *fi = Vect_get_field( mMap, field); // should work also with field = 0

// Read attributes
if ( fi == NULL ) {
#ifdef QGISDEBUG
std::cerr << "No field info -> no attributes" << std::endl;
#endif
error->setLatin1( "Cannot get field info" );
return error;
*orphan = false;
return error;
}

#ifdef QGISDEBUG
Expand All @@ -2132,26 +2225,33 @@ QString *QgsGrassProvider::insertAttributes ( int field, int cat )
}

#ifdef QGISDEBUG
std::cerr << "Database opened -> insert new record" << std::endl;
std::cerr << "Database opened -> select record" << std::endl;
#endif

dbString dbstr;
db_init_string (&dbstr);
QString query;

query.sprintf("insert into %s ( %s ) values ( %d )", fi->table, fi->key, cat );
query.sprintf("select %s from %s where %s = %d", fi->key, fi->table, fi->key, cat );
db_set_string (&dbstr, (char *)query.latin1());

#ifdef QGISDEBUG
std::cerr << "SQL: " << db_get_string(&dbstr) << std::endl;
#endif

int ret = db_execute_immediate (driver, &dbstr);

if ( ret != DB_OK) {
std::cerr << "Error: " << db_get_error_msg() << std::endl;
error->setLatin1( db_get_error_msg() );
dbCursor cursor;
if ( db_open_select_cursor(driver, &dbstr, &cursor, DB_SCROLL) != DB_OK )
{
db_close_database_shutdown_driver ( driver );
error->setAscii("Cannot query database: " + query );
return error;
}
int nRecords = db_get_num_rows ( &cursor );
#ifdef QGISDEBUG
std::cerr << "Number of records: " << nRecords << std::endl;
#endif

if ( nRecords > 0 ) { *orphan = true; }

db_close_database_shutdown_driver ( driver );
db_free_string(&dbstr);
Expand Down
23 changes: 23 additions & 0 deletions src/providers/grass/qgsgrassprovider.h
Expand Up @@ -385,6 +385,13 @@ class QgsGrassProvider : public QgsVectorDataProvider
*/
int dbLinkField ( int link );

/** Execute SQL statement
* @param field
* @param sql
* @return empty string or error message
*/
QString *executeSql ( int field, const QString &sql );

/** Update attributes
* @param field
* @param cat
Expand All @@ -400,6 +407,22 @@ class QgsGrassProvider : public QgsVectorDataProvider
*/
QString *insertAttributes ( int field, int cat );

/** Delete attributes from the table
* @param field
* @param cat
* @return empty string or error message
*/
QString *deleteAttributes ( int field, int cat );

/** Check if a database row exists and it is orphan (no more lines with
* that category)
* @param field
* @param cat
* @param orphan set to true if a record exits and it is orphan
* @return empty string or error message
*/
QString *isOrphan ( int field, int cat, int *orphan);

/** Create table and link vector to this table
* @param field
* @param columns SQL definition for columns, e.g. cat integer, label varchar(10)
Expand Down

0 comments on commit 8d8b5e1

Please sign in to comment.