class Dock(QtGui.QDockWidget):
def __init__(self, iface, parent=None):
QtGui.QWidget.__init__(self, parent)
self.iface = iface
self.canvas = self.iface.mapCanvas()
self.extentSelector = ExtentSelector(self.canvas)
self.extentSelector.tool.rectangleCreated.connect(self.polygonCreated)
self.setupUi()
def get_credentials(self):
qs = QtCore.QSettings()
hostname = qs.value('download_exposure/hostname', '')
username = qs.value('download_exposure/username', '')
password = qs.value('download_exposure/password', '')
if not (hostname and username and password):
PlatformSettingsDialog(self.iface).exec_()
hostname = qs.value('download_exposure/hostname', '')
username = qs.value('download_exposure/username', '')
password = qs.value('download_exposure/password', '')
return hostname, username, password
def setupUi(self):
self.resize(200, 200)
self.setWindowTitle('Exposure')
self.drawBtn = QtGui.QPushButton(self)
self.drawBtn.setObjectName('drawBtn')
self.drawBtn.setText('Draw')
self.downloadBtn = QtGui.QPushButton(self)
self.downloadBtn.setObjectName('downloadBtn')
self.downloadBtn.setText('Download')
self.downloadBtn.setEnabled(False)
self.clearBtn = QtGui.QPushButton(self)
self.clearBtn.setObjectName('clearBtn')
self.clearBtn.setText('Clear')
self.contents = QtGui.QGroupBox(self)
self.layout = QtGui.QVBoxLayout(self.contents)
self.layout.addWidget(self.drawBtn)
self.layout.addWidget(self.downloadBtn)
self.layout.addWidget(self.clearBtn)
self.setWidget(self.contents)
QtCore.QMetaObject.connectSlotsByName(self)
def enableBusyCursor(self):
"""Set the hourglass enabled and stop listening for layer changes."""
QtGui.qApp.setOverrideCursor(QtGui.QCursor(QtCore.Qt.WaitCursor))
def disableBusyCursor(self):
"""Disable the hourglass cursor and listen for layer changes."""
QtGui.qApp.restoreOverrideCursor()
@QtCore.pyqtSlot()
def on_downloadBtn_clicked(self):
hostname, username, password = self.get_credentials()
self.show_exposure(hostname, username, password)
self.extentSelector.stop()
self.downloadBtn.setEnabled(False)
@QtCore.pyqtSlot()
def on_drawBtn_clicked(self):
self.get_credentials() # opens PlatformSettingsDialog if no password
self.extentSelector.start()
self.extentSelector.getExtent()
@QtCore.pyqtSlot()
def on_clearBtn_clicked(self):
self.extentSelector.stop()
self.downloadBtn.setEnabled(False)
def polygonCreated(self):
if self.selectedExtent() is not None:
self.downloadBtn.setEnabled(True)
def selectedExtent(self):
return self.extentSelector.getExtent()
def show_exposure(self, hostname, username, password):
selected_extent = self.selectedExtent()
if selected_extent is None:
QtGui.QMessageBox.warning(
self, 'Exposure Download Error', 'No region selected')
return
self.enableBusyCursor()
try:
crs = self.iface.mapCanvas().mapRenderer().destinationCrs()
xform = QgsCoordinateTransform(
crs, QgsCoordinateReferenceSystem(4326))
extent = xform.transform(selected_extent)
lon_min, lon_max = extent.xMinimum(), extent.xMaximum()
lat_min, lat_max = extent.yMinimum(), extent.yMaximum()
# download data
ed = ExposureDownloader(hostname)
ed.login(username, password)
try:
fname, msg = ed.download(lat_min, lon_min, lat_max, lon_max)
except ExposureDownloadError as e:
QtGui.QMessageBox.warning(
self, 'Exposure Download Error', str(e))
return
QgsMessageLog.logMessage(msg, 'GEM Exposure Downloader')
# don't remove the file, otherwise there will concurrency problems
uri = 'file://%s?delimiter=%s&xField=%s&yField=%s&crs=epsg:4326&' \
'skipLines=25&trimFields=yes' % (fname, ',', 'lon', 'lat')
vlayer = QgsVectorLayer(uri, 'exposure_export', 'delimitedtext')
QgsMapLayerRegistry.instance().addMapLayer(vlayer)
finally:
self.disableBusyCursor()
class Dock(QtGui.QDockWidget, Ui_Dock):
def __init__(self, iface, gemcatalogue, parent=None):
QtGui.QDockWidget.__init__(self, parent)
self.iface = iface
self.gemcatalogue = gemcatalogue
self.setupUi(self)
self.add_range_sliders()
self.canvas = self.iface.mapCanvas()
self.extentSelector = ExtentSelector(self.canvas)
self.agenciesWidget = ListMultiSelectWidget(
title='Select one or more agencies')
self.magnitudesWidget = ListMultiSelectWidget(
title='Select one or more magnitude scales')
self.verticalLayout.insertWidget(1, self.agenciesWidget)
self.verticalLayout.insertWidget(2, self.magnitudesWidget)
self.extentSelector.tool.rectangleCreated.connect(self.polygonCreated)
def closeEvent(self, event):
self.emit(QtCore.SIGNAL("closed()"), self)
return QtGui.QDockWidget.closeEvent(self, event)
def enableBusyCursor(self):
"""Set the hourglass enabled and stop listening for layer changes."""
QtGui.qApp.setOverrideCursor(QtGui.QCursor(QtCore.Qt.WaitCursor))
def disableBusyCursor(self):
"""Disable the hourglass cursor and listen for layer changes."""
QtGui.qApp.restoreOverrideCursor()
def add_range_sliders(self):
self.mag_range.setOrientation(QtCore.Qt.Horizontal)
self.mag_range.setMinimum(1)
self.mag_range.setMaximum(10)
self.mag_range.setLowValue(5)
self.mag_range.setHighValue(8)
def update_selectDbComboBox(self, db_sel):
if db_sel is not None and db_sel != '':
if self.selectDbComboBox.count() == 0:
self.selectDbComboBox.addItem(db_sel)
else:
item_index = self.selectDbComboBox.findText(db_sel)
self.selectDbComboBox.blockSignals(True)
# Elem not in list.
if item_index != -1:
self.selectDbComboBox.removeItem(item_index)
self.selectDbComboBox.insertItem(0, db_sel)
self.selectDbComboBox.blockSignals(False)
self.selectDbComboBox.setCurrentIndex(0)
def set_agencies(self, agencies):
self.agenciesWidget.set_selected_items(agencies)
def set_magnitude_scales(self, magnitude_scales):
self.magnitudesWidget.set_selected_items(magnitude_scales)
def set_dates(self, dates):
min_date, max_date = dates
self.minDateDe.setDateTimeRange(
QtCore.QDateTime(min_date), QtCore.QDateTime(max_date))
self.maxDateDe.setDateTimeRange(
QtCore.QDateTime(min_date), QtCore.QDateTime(max_date))
self.minDateDe.setDateTime(QtCore.QDateTime(min_date))
self.maxDateDe.setDateTime(QtCore.QDateTime(max_date))
@QtCore.pyqtSlot()
def on_addDbBtn_clicked(self):
db_sel = unicode(QtGui.QFileDialog.getOpenFileName(
self.iface.mainWindow(), 'Choose db',
QtCore.QDir.homePath(),
"Catalogue db file (*.db);;All files (*.*)"))
self.update_selectDbComboBox(db_sel)
@QtCore.pyqtSlot()
def on_filterBtn_clicked(self):
agencies_selected = self.agenciesWidget.get_selected_items()
mscales_selected = self.magnitudesWidget.get_selected_items()
mvalues_selected = Range(self.mag_range.lowValue(),
self.mag_range.highValue())
dvalues_selected = (self.minDateDe.dateTime().toPyDateTime(),
self.maxDateDe.dateTime().toPyDateTime())
selected_extent = self.selectedExtent()
self.gemcatalogue.update_map(agencies_selected, mscales_selected,
mvalues_selected, dvalues_selected,
selected_extent)
@QtCore.pyqtSlot()
def on_downloadBtn_clicked(self):
qs = QtCore.QSettings()
hostname = qs.value('gemcatalogue/hostname', '')
username = qs.value('gemcatalogue/username', '')
password = qs.value('gemcatalogue/password', '')
if not (hostname and username and password):
dialog = PlatformSettingsDialog(self.iface)
if dialog.exec_():
self.gemcatalogue.show_exposure(hostname, username, password)
else:
self.gemcatalogue.show_exposure(hostname, username, password)
@QtCore.pyqtSlot(str)
def on_selectDbComboBox_currentIndexChanged(self, selectedDb):
self.gemcatalogue.update_catalogue_db(selectedDb)
@QtCore.pyqtSlot()
def on_drawBtn_clicked(self):
self.extentSelector.start()
self.extentSelector.getExtent()
@QtCore.pyqtSlot()
def on_clearBtn_clicked(self):
self.extentSelector.stop()
self.downloadBtn.setEnabled(False)
def polygonCreated(self):
self.downloadBtn.setEnabled(True)
def selectedExtent(self):
return self.extentSelector.getExtent()