38 #include <QPushButton>
40 #include <QMessageBox>
41 #include <QFileDialog>
45 #include <ogr_srs_api.h>
46 #include <gdal_version.h>
47 #include <cpl_error.h>
48 #include <cpl_string.h>
50 #define DEFAULT_OGR_FID_COLUMN_TITLE "fid" // default value from OGR
53 : QDialog( parent, fl )
56 setObjectName( QStringLiteral(
"QgsNewGeoPackageLayerDialog" ) );
59 connect( mAddAttributeButton, &QToolButton::clicked,
this, &QgsNewGeoPackageLayerDialog::mAddAttributeButton_clicked );
60 connect( mRemoveAttributeButton, &QToolButton::clicked,
this, &QgsNewGeoPackageLayerDialog::mRemoveAttributeButton_clicked );
61 connect( mFieldTypeBox,
static_cast<void ( QComboBox::* )(
int )
>( &QComboBox::currentIndexChanged ),
this, &QgsNewGeoPackageLayerDialog::mFieldTypeBox_currentIndexChanged );
62 connect( mGeometryTypeBox,
static_cast<void ( QComboBox::* )(
int )
>( &QComboBox::currentIndexChanged ),
this, &QgsNewGeoPackageLayerDialog::mGeometryTypeBox_currentIndexChanged );
63 connect( mTableNameEdit, &QLineEdit::textChanged,
this, &QgsNewGeoPackageLayerDialog::mTableNameEdit_textChanged );
64 connect( mTableNameEdit, &QLineEdit::textEdited,
this, &QgsNewGeoPackageLayerDialog::mTableNameEdit_textEdited );
65 connect( mLayerIdentifierEdit, &QLineEdit::textEdited,
this, &QgsNewGeoPackageLayerDialog::mLayerIdentifierEdit_textEdited );
66 connect( buttonBox, &QDialogButtonBox::accepted,
this, &QgsNewGeoPackageLayerDialog::buttonBox_accepted );
67 connect( buttonBox, &QDialogButtonBox::rejected,
this, &QgsNewGeoPackageLayerDialog::buttonBox_rejected );
68 connect( buttonBox, &QDialogButtonBox::helpRequested,
this, &QgsNewGeoPackageLayerDialog::showHelp );
73 const auto addGeomItem = [
this]( OGRwkbGeometryType ogrGeomType )
79 addGeomItem( wkbNone );
80 addGeomItem( wkbPoint );
81 addGeomItem( wkbLineString );
82 addGeomItem( wkbPolygon );
83 addGeomItem( wkbMultiPoint );
84 addGeomItem( wkbMultiLineString );
85 addGeomItem( wkbMultiPolygon );
89 addGeomItem( wkbCircularString );
91 addGeomItem( wkbCompoundCurve );
92 addGeomItem( wkbCurvePolygon );
93 addGeomItem( wkbMultiCurve );
94 addGeomItem( wkbMultiSurface );
95 mGeometryTypeBox->setCurrentIndex( -1 );
97 mGeometryWithZCheckBox->setEnabled(
false );
98 mGeometryWithMCheckBox->setEnabled(
false );
99 mGeometryColumnEdit->setEnabled(
false );
100 mGeometryColumnEdit->setText( QStringLiteral(
"geometry" ) );
102 mCheckBoxCreateSpatialIndex->setEnabled(
false );
103 mCrsSelector->setEnabled(
false );
104 mCrsSelector->setShowAccuracyWarnings(
true );
116 mOkButton = buttonBox->button( QDialogButtonBox::Ok );
117 mOkButton->setEnabled(
false );
119 connect( mFieldNameEdit, &QLineEdit::textChanged,
this, &QgsNewGeoPackageLayerDialog::fieldNameChanged );
120 connect( mAttributeView, &QTreeWidget::itemSelectionChanged,
this, &QgsNewGeoPackageLayerDialog::selectionChanged );
121 connect( mTableNameEdit, &QLineEdit::textChanged,
this, &QgsNewGeoPackageLayerDialog::checkOk );
122 connect( mGeometryTypeBox,
static_cast<void( QComboBox::* )(
int )
>( &QComboBox::currentIndexChanged ),
this, &QgsNewGeoPackageLayerDialog::checkOk );
124 mAddAttributeButton->setEnabled(
false );
125 mRemoveAttributeButton->setEnabled(
false );
127 mCheckBoxCreateSpatialIndex->setChecked(
true );
131 mDatabase->setFilter( tr(
"GeoPackage" ) +
" (*.gpkg)" );
132 mDatabase->setDialogTitle( tr(
"Select Existing or Create a New GeoPackage Database Fileā¦" ) );
133 mDatabase->setDefaultRoot( settings.
value( QStringLiteral(
"UI/lastVectorFileFilterDir" ), QDir::homePath() ).toString() );
134 mDatabase->setConfirmOverwrite(
false );
138 const QFileInfo tmplFileInfo( filePath );
139 settings.
setValue( QStringLiteral(
"UI/lastVectorFileFilterDir" ), tmplFileInfo.absolutePath() );
140 if ( !filePath.isEmpty() && !mTableNameEdited )
142 const QFileInfo fileInfo( filePath );
143 mTableNameEdit->setText( fileInfo.baseName() );
150 QCompleter *completer =
new QCompleter(
this );
151 completer->setModel( ogrProviderModel );
153 completer->setCompletionMode( QCompleter::PopupCompletion );
154 completer->setFilterMode( Qt::MatchContains );
155 mDatabase->lineEdit()->setCompleter( completer );
160 mCrsSelector->setCrs(
crs );
165 mDatabase->setReadOnly(
true );
168 void QgsNewGeoPackageLayerDialog::mFieldTypeBox_currentIndexChanged(
int )
170 const QString myType = mFieldTypeBox->currentData( Qt::UserRole ).toString();
171 mFieldLengthEdit->setEnabled( myType == QLatin1String(
"text" ) );
172 if ( myType != QLatin1String(
"text" ) )
173 mFieldLengthEdit->clear();
177 void QgsNewGeoPackageLayerDialog::mGeometryTypeBox_currentIndexChanged(
int )
179 const OGRwkbGeometryType geomType =
static_cast<OGRwkbGeometryType
>
180 ( mGeometryTypeBox->currentData( Qt::UserRole ).toInt() );
181 const bool isSpatial = geomType != wkbNone;
182 mGeometryWithZCheckBox->setEnabled( isSpatial );
183 mGeometryWithMCheckBox->setEnabled( isSpatial );
184 mGeometryColumnEdit->setEnabled( isSpatial );
185 mCheckBoxCreateSpatialIndex->setEnabled( isSpatial );
186 mCrsSelector->setEnabled( isSpatial );
189 void QgsNewGeoPackageLayerDialog::mTableNameEdit_textChanged(
const QString &text )
191 mTableNameEdited = !text.isEmpty();
192 if ( !text.isEmpty() && !mLayerIdentifierEdited )
194 mLayerIdentifierEdit->setText( text );
198 void QgsNewGeoPackageLayerDialog::mTableNameEdit_textEdited(
const QString &text )
201 mTableNameEdited = !text.isEmpty();
204 void QgsNewGeoPackageLayerDialog::mLayerIdentifierEdit_textEdited(
const QString &text )
207 mLayerIdentifierEdited = !text.isEmpty();
210 void QgsNewGeoPackageLayerDialog::checkOk()
212 const bool ok = !mDatabase->filePath().isEmpty() &&
213 !mTableNameEdit->text().isEmpty() &&
214 mGeometryTypeBox->currentIndex() != -1;
216 mOkButton->setEnabled( ok );
219 void QgsNewGeoPackageLayerDialog::mAddAttributeButton_clicked()
221 if ( !mFieldNameEdit->text().isEmpty() )
223 const QString myName = mFieldNameEdit->text();
224 const QString featureId = mFeatureIdColumnEdit->text().isEmpty() ? QStringLiteral(
DEFAULT_OGR_FID_COLUMN_TITLE ) : mFeatureIdColumnEdit->text();
225 if ( myName.compare( featureId, Qt::CaseInsensitive ) == 0 )
227 QMessageBox::critical(
this, tr(
"Add Field" ), tr(
"The field cannot have the same name as the feature identifier." ) );
232 const QString myType = mFieldTypeBox->currentData( Qt::UserRole ).toString();
233 const QString length = mFieldLengthEdit->text();
234 mAttributeView->addTopLevelItem(
new QTreeWidgetItem( QStringList() << myName << myType << length ) );
238 mFieldNameEdit->clear();
242 void QgsNewGeoPackageLayerDialog::mRemoveAttributeButton_clicked()
244 delete mAttributeView->currentItem();
249 void QgsNewGeoPackageLayerDialog::fieldNameChanged(
const QString &name )
251 mAddAttributeButton->setDisabled( name.isEmpty() || ! mAttributeView->findItems( name, Qt::MatchExactly ).isEmpty() );
254 void QgsNewGeoPackageLayerDialog::selectionChanged()
256 mRemoveAttributeButton->setDisabled( mAttributeView->selectedItems().isEmpty() );
259 void QgsNewGeoPackageLayerDialog::buttonBox_accepted()
265 void QgsNewGeoPackageLayerDialog::buttonBox_rejected()
270 bool QgsNewGeoPackageLayerDialog::apply()
272 QString fileName( mDatabase->filePath() );
273 if ( !fileName.endsWith( QLatin1String(
".gpkg" ), Qt::CaseInsensitive ) )
274 fileName += QLatin1String(
".gpkg" );
276 bool createNewDb =
false;
278 if ( QFile::exists( fileName ) )
280 bool overwrite =
false;
287 msgBox.setIcon( QMessageBox::Question );
288 msgBox.setWindowTitle( tr(
"New GeoPackage Layer" ) );
289 msgBox.setText( tr(
"The File already exists. Do you want to overwrite the existing file with a new database or add a new layer to it?" ) );
290 QPushButton *overwriteButton = msgBox.addButton( tr(
"Overwrite" ), QMessageBox::ActionRole );
291 QPushButton *addNewLayerButton = msgBox.addButton( tr(
"Add New Layer" ), QMessageBox::ActionRole );
292 msgBox.setStandardButtons( QMessageBox::Cancel );
293 msgBox.setDefaultButton( addNewLayerButton );
295 if ( property(
"hideDialogs" ).toBool() )
297 overwrite = property(
"question_existing_db_answer_overwrite" ).toBool();
299 cancel = !property(
"question_existing_db_answer_add_new_layer" ).toBool();
303 const int ret = msgBox.exec();
304 if ( ret == QMessageBox::Cancel )
306 if ( msgBox.clickedButton() == overwriteButton )
327 QFile( fileName ).remove();
336 OGRSFDriverH hGpkgDriver = OGRGetDriverByName(
"GPKG" );
339 if ( !property(
"hideDialogs" ).toBool() )
340 QMessageBox::critical(
this, tr(
"New GeoPackage Layer" ),
341 tr(
"Layer creation failed. GeoPackage driver not found." ) );
348 hDS.reset( OGR_Dr_CreateDataSource( hGpkgDriver, fileName.toUtf8().constData(),
nullptr ) );
351 const QString msg( tr(
"Creation of database failed (OGR error: %1)" ).arg( QString::fromUtf8( CPLGetLastErrorMsg() ) ) );
352 if ( !property(
"hideDialogs" ).toBool() )
353 QMessageBox::critical(
this, tr(
"New GeoPackage Layer" ), msg );
359 OGRSFDriverH hDriver =
nullptr;
360 hDS.reset( OGROpen( fileName.toUtf8().constData(),
true, &hDriver ) );
363 const QString msg( tr(
"Opening of database failed (OGR error: %1)" ).arg( QString::fromUtf8( CPLGetLastErrorMsg() ) ) );
364 if ( !property(
"hideDialogs" ).toBool() )
365 QMessageBox::critical(
this, tr(
"New GeoPackage Layer" ), msg );
368 if ( hDriver != hGpkgDriver )
370 const QString msg( tr(
"Opening of file succeeded, but this is not a GeoPackage database." ) );
371 if ( !property(
"hideDialogs" ).toBool() )
372 QMessageBox::critical(
this, tr(
"New GeoPackage Layer" ), msg );
377 const QString tableName( mTableNameEdit->text() );
379 bool overwriteTable =
false;
380 if ( OGR_DS_GetLayerByName( hDS.get(), tableName.toUtf8().constData() ) )
382 if ( property(
"hideDialogs" ).toBool() )
384 overwriteTable = property(
"question_existing_layer_answer_overwrite" ).toBool();
386 else if ( QMessageBox::question(
this, tr(
"New GeoPackage Layer" ),
387 tr(
"A table with the same name already exists. Do you want to overwrite it?" ),
388 QMessageBox::Yes | QMessageBox::No, QMessageBox::No ) == QMessageBox::Yes )
390 overwriteTable =
true;
393 if ( !overwriteTable )
399 const QString layerIdentifier( mLayerIdentifierEdit->text() );
400 const QString layerDescription( mLayerDescriptionEdit->text() );
402 OGRwkbGeometryType wkbType =
static_cast<OGRwkbGeometryType
>
403 ( mGeometryTypeBox->currentData( Qt::UserRole ).toInt() );
406 if ( mGeometryWithZCheckBox->isChecked() )
407 wkbType = OGR_GT_SetZ( wkbType );
409 if ( mGeometryWithMCheckBox->isChecked() )
410 wkbType = OGR_GT_SetM( wkbType );
415 if ( wkbType != wkbNone && srs.
isValid() )
421 char **options =
nullptr;
423 if ( overwriteTable )
424 options = CSLSetNameValue( options,
"OVERWRITE",
"YES" );
425 if ( !layerIdentifier.isEmpty() )
426 options = CSLSetNameValue( options,
"IDENTIFIER", layerIdentifier.toUtf8().constData() );
427 if ( !layerDescription.isEmpty() )
428 options = CSLSetNameValue( options,
"DESCRIPTION", layerDescription.toUtf8().constData() );
430 const QString featureId( mFeatureIdColumnEdit->text() );
431 if ( !featureId.isEmpty() )
432 options = CSLSetNameValue( options,
"FID", featureId.toUtf8().constData() );
434 const QString geometryColumn( mGeometryColumnEdit->text() );
435 if ( wkbType != wkbNone && !geometryColumn.isEmpty() )
436 options = CSLSetNameValue( options,
"GEOMETRY_COLUMN", geometryColumn.toUtf8().constData() );
438 if ( wkbType != wkbNone )
439 options = CSLSetNameValue( options,
"SPATIAL_INDEX", mCheckBoxCreateSpatialIndex->isChecked() ?
"YES" :
"NO" );
441 OGRLayerH hLayer = OGR_DS_CreateLayer( hDS.get(), tableName.toUtf8().constData(), hSRS, wkbType, options );
442 CSLDestroy( options );
447 const QString msg( tr(
"Creation of layer failed (OGR error: %1)" ).arg( QString::fromUtf8( CPLGetLastErrorMsg() ) ) );
448 if ( !property(
"hideDialogs" ).toBool() )
449 QMessageBox::critical(
this, tr(
"New GeoPackage Layer" ), msg );
453 QTreeWidgetItemIterator it( mAttributeView );
456 const QString fieldName( ( *it )->text( 0 ) );
457 const QString fieldType( ( *it )->text( 1 ) );
458 const QString fieldWidth( ( *it )->text( 2 ) );
460 OGRFieldType ogrType( OFTString );
461 OGRFieldSubType ogrSubType = OFSTNone;
462 if ( fieldType == QLatin1String(
"text" ) )
464 else if ( fieldType == QLatin1String(
"integer" ) )
465 ogrType = OFTInteger;
466 else if ( fieldType == QLatin1String(
"integer64" ) )
467 ogrType = OFTInteger64;
468 else if ( fieldType == QLatin1String(
"real" ) )
470 else if ( fieldType == QLatin1String(
"date" ) )
472 else if ( fieldType == QLatin1String(
"datetime" ) )
473 ogrType = OFTDateTime;
474 else if ( fieldType == QLatin1String(
"bool" ) )
476 ogrType = OFTInteger;
477 ogrSubType = OFSTBoolean;
479 else if ( fieldType == QLatin1String(
"binary" ) )
481 else if ( fieldType == QLatin1String(
"json" ) )
484 ogrSubType = OFSTJSON;
487 const int ogrWidth = fieldWidth.toInt();
490 if ( ogrSubType != OFSTNone )
491 OGR_Fld_SetSubType( fld.get(), ogrSubType );
493 if ( ogrType != OFTBinary )
494 OGR_Fld_SetWidth( fld.get(), ogrWidth );
496 if ( OGR_L_CreateField( hLayer, fld.get(),
true ) != OGRERR_NONE )
498 if ( !property(
"hideDialogs" ).toBool() )
500 QMessageBox::critical(
this, tr(
"New GeoPackage Layer" ),
501 tr(
"Creation of field %1 failed (OGR error: %2)" )
502 .arg( fieldName, QString::fromUtf8( CPLGetLastErrorMsg() ) ) );
513 OGR_L_ResetReading( hLayer );
514 if ( CPLGetLastErrorType() != CE_None )
516 const QString msg( tr(
"Creation of layer failed (OGR error: %1)" ).arg( QString::fromUtf8( CPLGetLastErrorMsg() ) ) );
517 if ( !property(
"hideDialogs" ).toBool() )
518 QMessageBox::critical(
this, tr(
"New GeoPackage Layer" ), msg );
523 const QString uri( QStringLiteral(
"%1|layername=%2" ).arg( fileName, tableName ) );
524 const QString userVisiblelayerName( layerIdentifier.isEmpty() ? tableName : layerIdentifier );
526 std::unique_ptr< QgsVectorLayer > layer = std::make_unique< QgsVectorLayer >( uri, userVisiblelayerName, QStringLiteral(
"ogr" ), layerOptions );
527 if ( layer->isValid() )
532 QList<QgsMapLayer *> myList;
533 myList << layer.release();
546 if ( !property(
"hideDialogs" ).toBool() )
547 QMessageBox::critical(
this, tr(
"New GeoPackage Layer" ), tr(
"%1 is an invalid layer and cannot be loaded." ).arg( tableName ) );
555 mBehavior = behavior;
560 mAddToProject = addToProject;
563 void QgsNewGeoPackageLayerDialog::showHelp()
565 QgsHelp::openHelp( QStringLiteral(
"managing_data_source/create_layers.html#creating-a-new-geopackage-layer" ) );