Source code for svir.dialogs.load_ruptures_as_layer_dialog

# -*- 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/>.

import os
import tempfile
from collections import OrderedDict
from svir.utilities.utils import (import_layer_from_csv,
                                  get_params_from_comment_line,
                                  count_heading_commented_lines,
                                  log_msg,
                                  )
from svir.dialogs.load_output_as_layer_dialog import LoadOutputAsLayerDialog


[docs]class LoadRupturesAsLayerDialog(LoadOutputAsLayerDialog): """ Dialog to load ruptures from an oq-engine output, as layer """ def __init__(self, iface, viewer_dock, session, hostname, calc_id, output_type='ruptures', path=None, mode=None, engine_version=None): assert output_type == 'ruptures' LoadOutputAsLayerDialog.__init__( self, iface, viewer_dock, session, hostname, calc_id, output_type=output_type, path=path, mode=mode, engine_version=engine_version) self.style_by_items = OrderedDict([ ('Tectonic region type', 'trt'), ('Magnitude', 'mag'), ]) self.create_file_hlayout() self.create_file_size_indicator() self.create_save_as_shp_ckb() self.create_style_by_selector() self.populate_out_dep_widgets() self.setWindowTitle('Load ruptures from CSV, as layer') self.adjustSize() self.set_ok_button() self.file_browser_tbn.setEnabled(True) if self.path: self.path_le.setText(self.path) self.show()
[docs] def set_ok_button(self): self.ok_button.setEnabled(bool(self.path))
[docs] def populate_out_dep_widgets(self): self.show_file_size() self.populate_style_by_cbx()
[docs] def populate_style_by_cbx(self): self.style_by_cbx.clear() for item in self.style_by_items: self.style_by_cbx.addItem(item, self.style_by_items[item])
[docs] def load_from_csv(self): if self.mode == 'testing': dest_shp = tempfile.mkstemp(suffix='.shp')[1] else: dest_shp = None # the destination file will be selected via GUI csv_path = self.path_le.text() # extract the investigation_time from the heading commented line with open(csv_path, 'r', newline='') as f: comment_line = f.readline() try: params_dict = get_params_from_comment_line(comment_line) except LookupError as exc: log_msg(str(exc), level='C', message_bar=self.iface.messageBar(), exception=exc) return try: investigation_time = params_dict['investigation_time'] except KeyError as exc: log_msg('Investigation time not found', level='C', message_bar=self.iface.messageBar(), exception=exc) return # extract the name of the csv file and remove the extension layer_name = os.path.splitext(os.path.basename(csv_path))[0] layer_name += '_%sy' % investigation_time n_lines_to_skip = count_heading_commented_lines(csv_path) try: self.layer = import_layer_from_csv( self, csv_path, layer_name, self.iface, wkt_field='boundary', delimiter='\\t', lines_to_skip_count=n_lines_to_skip, save_as_shp=self.save_as_shp_ckb.isChecked(), dest_shp=dest_shp) except RuntimeError as exc: log_msg(str(exc), level='C', message_bar=self.iface.messageBar(), exception=exc) return self.layer.setCustomProperty('investigation_time', investigation_time) style_by = self.style_by_cbx.itemData(self.style_by_cbx.currentIndex()) if style_by == 'mag': self.style_maps(self.layer, style_by, self.iface, self.output_type) else: # 'trt' self.style_categorized(layer=self.layer, style_by=style_by) log_msg('Layer %s was loaded successfully' % layer_name, level='S', message_bar=self.iface.messageBar())