class NafiDockWidget(QtWidgets.QDockWidget, Ui_NafiDockWidgetBase):
closingPlugin = pyqtSignal()
def __init__(self, parent=None):
"""Constructor."""
super(NafiDockWidget, self).__init__(parent)
self.setupUi(self)
# set up QTreeView
self.treeView.setHeaderHidden(True)
self.treeView.setSortingEnabled(True)
self.treeView.setFocusPolicy(Qt.NoFocus)
self.treeView.pressed.connect(self.treeViewPressed)
# set up search signal
self.lineEdit.textChanged.connect(self.searchTextChanged)
self.searchText = ""
# set up clear search
self.clearSearchButton.clicked.connect(self.clearSearch)
# set up About … dialog
self.aboutButton.clicked.connect(self.showAboutDialog)
# set up Download NAFI data … link button
self.dataButton.clicked.connect(
lambda: webbrowser.open(getNafiDataUrl()))
# set up base model
self.treeViewModel = NafiTreeViewModel(getNafiUrl())
# set up proxy model for filtering
self.proxyModel = QSortFilterProxyModel(self.treeView)
self.proxyModel.setSourceModel(self.treeViewModel)
self.proxyModel.setRecursiveFilteringEnabled(True)
self.treeView.setModel(self.proxyModel)
self.reader = NafiCapabilitiesReader()
self.reader.capabilitiesDownloaded.connect(
lambda xml: self.initModel(xml))
# restore the view from source whenever this dock widget is made visible again
self.visibilityChanged.connect(
lambda visible: visible and self.loadNafiWms())
# initialise proxied tree view model from WMS contents
# self.loadNafiWms()
def loadNafiWms(self):
"""Load the NAFI WMS and additional layers."""
self.wmsUrl = getNafiUrl()
self.reader.downloadCapabilities(self.wmsUrl)
def initModel(self, wmsXml):
"""Initialise a QStandardItemModel from the NAFI WMS."""
googSat = GoogleXyzItem()
googHyb = GoogleXyzItem("y")
googStr = GoogleXyzItem("m")
# ibraWms = IbraWmsItem()
ozTopoWmts = OzTopoWmtsItem()
self.treeViewModel.loadWms(
self.wmsUrl,
wmsXml,
additionalItems=[googSat, googHyb, googStr, ozTopoWmts])
# set default sort and expansion
self.proxyModel.sort(0, Qt.AscendingOrder)
self.expandTopLevel()
def expandTopLevel(self):
# expand the top level items
for row in range(self.proxyModel.rowCount()):
self.treeView.expand(self.proxyModel.index(row, 0))
def treeViewPressed(self, index):
"""Load a NAFI WMS layer given an index in the tree view."""
assert isinstance(
index, QModelIndex), "Supplied parameter is not a QModelIndex"
realIndex = self.proxyModel.mapToSource(index)
modelNode = self.treeViewModel.itemFromIndex(realIndex)
# if we've got a layer and not a layer group, add to map
if modelNode is not None:
if isinstance(
modelNode,
(GoogleXyzItem, IbraWmsItem, OzTopoWmtsItem, WmsItem)):
modelNode.addLayer()
def searchTextChanged(self, text):
"""Process a change in the search filter text."""
# user adding characters and has exceeded 3 or more, or is removing characters
if len(text) >= 3 or len(self.searchText) > len(text):
regex = QRegExp(text, Qt.CaseInsensitive, QRegExp.RegExp)
self.proxyModel.setFilterRegExp(regex)
self.treeView.expandAll()
# update last search text state
self.searchText = text
def clearSearch(self):
"""Clear search data."""
self.lineEdit.setText(None)
self.treeView.collapseAll()
def sizeHint(self):
return QtCore.QSize(150, 400)
def showAboutDialog(self):
"""Show an About … dialog."""
aboutDialog = NafiAboutDialog()
aboutDialog.exec_()
def closeEvent(self, event):
"""Handle plug-in close."""
self.closingPlugin.emit()
event.accept()
def refresh_model(self, source_model, db_connector=None, silent=False):
filtered_source_model = QSortFilterProxyModel()
filtered_source_model.setSourceModel(source_model)
filtered_source_model.setFilterRole(int(SourceModel.Roles.TYPE))
self.print_info.emit(self.tr("Refresh available models:"))
self.clear()
previously_checked_models = self._checked_models
self._checked_models = {}
# models from db
db_modelnames = self._db_modelnames(db_connector)
# models from the repos
filtered_source_model.setFilterFixedString("model")
for r in range(0, filtered_source_model.rowCount()):
filtered_source_model_index = filtered_source_model.index(
r, SourceModel.Columns.SOURCE)
modelname = filtered_source_model_index.data(
int(SourceModel.Roles.NAME))
if modelname:
enabled = modelname not in db_modelnames
self.add_source(
modelname,
filtered_source_model_index.data(
int(SourceModel.Roles.TYPE)),
filtered_source_model_index.data(
int(SourceModel.Roles.PATH)),
filtered_source_model_index.data(
int(SourceModel.Roles.ORIGIN_INFO)),
previously_checked_models.get(
(
modelname,
filtered_source_model_index.data(
int(SourceModel.Roles.PATH)),
),
Qt.Checked,
) if enabled and modelname not in self.checked_models()
else Qt.Unchecked,
enabled,
)
if not silent:
self.print_info.emit(
self.tr("- Append (repository) model {}{}").format(
modelname,
" (inactive because it already exists in the database)"
if not enabled else "",
))
# models from the files
filtered_source_model.setFilterFixedString("ili")
for r in range(0, filtered_source_model.rowCount()):
filtered_source_model_index = filtered_source_model.index(
r, SourceModel.Columns.SOURCE)
ili_file_path = filtered_source_model_index.data(
int(SourceModel.Roles.PATH))
self.ilicache = IliCache(None, ili_file_path)
models = self.ilicache.process_ili_file(ili_file_path)
for model in models:
if model["name"]:
enabled = model["name"] not in db_modelnames
self.add_source(
model["name"],
filtered_source_model_index.data(
int(SourceModel.Roles.TYPE)),
filtered_source_model_index.data(
int(SourceModel.Roles.PATH)),
filtered_source_model_index.data(
int(SourceModel.Roles.ORIGIN_INFO)),
previously_checked_models.get(
(
model["name"],
filtered_source_model_index.data(
int(SourceModel.Roles.PATH)),
),
Qt.Checked if model is models[-1] and enabled
and model["name"] not in self.checked_models() else
Qt.Unchecked,
),
enabled,
)
if not silent:
self.print_info.emit(
self.tr("- Append (file) model {}{} from {}").
format(
model["name"],
" (inactive because it already exists in the database)"
if not enabled else "",
ili_file_path,
))
# models from the transfer files
filtered_source_model.setFilterRegExp("|".join(TransferExtensions))
for r in range(0, filtered_source_model.rowCount()):
filtered_source_model_index = filtered_source_model.index(
r, SourceModel.Columns.SOURCE)
xtf_file_path = filtered_source_model_index.data(
int(SourceModel.Roles.PATH))
models = self._transfer_file_models(xtf_file_path)
for model in models:
if model["name"]:
enabled = model["name"] not in db_modelnames
self.add_source(
model["name"],
filtered_source_model_index.data(
int(SourceModel.Roles.TYPE)),
filtered_source_model_index.data(
int(SourceModel.Roles.PATH)),
filtered_source_model_index.data(
int(SourceModel.Roles.ORIGIN_INFO)),
previously_checked_models.get(
(
model["name"],
filtered_source_model_index.data(
int(SourceModel.Roles.PATH)),
),
Qt.Checked if enabled
and model["name"] not in self.checked_models() else
Qt.Unchecked,
),
enabled,
)
if not silent:
self.print_info.emit(
self.tr("- Append (xtf file) model {}{} from {}").
format(
model["name"],
" (inactive because it already exists in the database)"
if not enabled else "",
xtf_file_path,
))
return self.rowCount()