# -*- coding: utf-8 -*-
# /***************************************************************************
# Irmt
# A QGIS plugin
# OpenQuake Integrated Risk Modelling Toolkit
# -------------------
# begin : 2013-10-24
# copyright : (C) 2014 by GEM Foundation
# email : devops@openquake.org
# ***************************************************************************/
#
# OpenQuake is free software: you can redistribute it and/or modify it
# under the terms of the GNU Affero General Public License as published
# by the Free Software Foundation, either version 3 of the License, or
# (at your option) any later version.
#
# OpenQuake is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU Affero General Public License
# along with OpenQuake. If not, see <http://www.gnu.org/licenses/>.
from qgis.core import QgsFeature, QgsGeometry, QgsPoint
from svir.dialogs.load_output_as_layer_dialog import LoadOutputAsLayerDialog
from svir.calculations.calculate_utils import add_numeric_attribute
from svir.utilities.utils import (LayerEditingManager,
log_msg,
WaitCursorManager,
extract_npz,
)
from svir.utilities.shared import DEBUG
[docs]class LoadUhsAsLayerDialog(LoadOutputAsLayerDialog):
"""
Modal dialog to load uniform hazard spectra from an oq-engine output,
as layer
"""
def __init__(self, iface, viewer_dock, session, hostname, calc_id,
output_type='uhs', path=None, mode=None):
assert output_type == 'uhs'
LoadOutputAsLayerDialog.__init__(
self, iface, viewer_dock, session, hostname, calc_id,
output_type, path, mode)
# FIXME: add layout only for output types that load from file
self.remove_file_hlayout()
self.setWindowTitle(
'Load uniform hazard spectra as layer')
self.create_num_sites_indicator()
self.create_load_selected_only_ckb()
self.create_poe_selector()
self.npz_file = extract_npz(
session, hostname, calc_id, output_type,
message_bar=iface.messageBar(), params=None)
self.populate_out_dep_widgets()
self.adjustSize()
self.set_ok_button()
[docs] def populate_dataset(self):
self.rlzs_or_stats = [
key for key in sorted(self.npz_file['all'].dtype.names)
if key not in ('lon', 'lat')]
self.dataset = self.npz_file['all']
self.poes = self.dataset[self.rlzs_or_stats[0]].dtype.names
self.poe_cbx.clear()
self.poe_cbx.setEnabled(True)
self.poe_cbx.addItems(self.poes)
self.set_ok_button()
[docs] def show_num_sites(self):
self.num_sites_lbl.setText(
self.num_sites_msg % self.dataset.shape)
[docs] def on_rlz_or_stat_changed(self):
self.dataset = self.npz_file['all'][self.rlz_or_stat_cbx.currentText()]
self.poes = self.dataset.dtype.names
self.poe_cbx.clear()
self.poe_cbx.setEnabled(True)
self.poe_cbx.addItems(self.poes)
self.set_ok_button()
[docs] def build_layer_name(self, **kwargs):
poe = kwargs['poe']
investigation_time = self.get_investigation_time()
layer_name = "uhs_poe-%s_%sy" % (poe, investigation_time)
return layer_name
[docs] def get_field_names(self, **kwargs):
poe = kwargs['poe']
field_names = []
for rlz_or_stat in self.rlzs_or_stats:
field_names.extend([
"%s_%s" % (rlz_or_stat, imt)
for imt in self.dataset[rlz_or_stat][poe].dtype.names])
return field_names
[docs] def add_field_to_layer(self, field_name):
# NOTE: add_numeric_attribute uses LayerEditingManager
added_field_name = add_numeric_attribute(
field_name, self.layer)
return added_field_name
[docs] def read_npz_into_layer(self, field_names, **kwargs):
poe = kwargs['poe']
with LayerEditingManager(self.layer, 'Reading npz', DEBUG):
lons = self.npz_file['all']['lon']
lats = self.npz_file['all']['lat']
feats = []
for row_idx, row in enumerate(self.dataset):
# add a feature
feat = QgsFeature(self.layer.pendingFields())
for field_name_idx, field_name in enumerate(field_names):
rlz_or_stat, imt = field_name.split('_')
iml = row[rlz_or_stat][poe][imt]
feat.setAttribute(field_name, float(iml))
feat.setGeometry(QgsGeometry.fromPoint(
QgsPoint(lons[row_idx], lats[row_idx])))
feats.append(feat)
added_ok = self.layer.addFeatures(feats, makeSelected=False)
if not added_ok:
msg = 'There was a problem adding features to the layer.'
log_msg(msg, level='C', message_bar=self.iface.messageBar())
[docs] def load_from_npz(self):
for poe in self.poes:
if (self.load_selected_only_ckb.isChecked()
and poe != self.poe_cbx.currentText()):
continue
with WaitCursorManager(
'Creating layer for poe "%s"...'
% poe, self.iface.messageBar()):
self.build_layer(poe=poe)
self.style_curves()