from appTool import * fcTranslate.apply_language('strings') if '_' not in builtins.__dict__: _ = gettext.gettext class SimplificationTool(AppTool): """ Do a shape simplification for the selected geometry. """ update_ui = pyqtSignal(object, int) def __init__(self, app, draw_app): AppTool.__init__(self, app) self.draw_app = draw_app self.decimals = app.decimals self.app = self.draw_app.app self.ui = SimplificationEditorUI(layout=self.layout, simp_class=self) self.plugin_name = self.ui.pluginName self.connect_signals_at_init() self.set_tool_ui() def connect_signals_at_init(self): # Signals self.ui.simplification_btn.clicked.connect(self.on_simplification_click) self.update_ui.connect(self.on_update_ui) def run(self): self.app.defaults.report_usage("Geo Editor SimplificationTool()") AppTool.run(self) # if the splitter us hidden, display it if self.app.ui.splitter.sizes()[0] == 0: self.app.ui.splitter.setSizes([1, 1]) # if the Tool Tab is hidden display it, else hide it but only if the objectName is the same found_idx = None for idx in range(self.app.ui.notebook.count()): if self.app.ui.notebook.widget(idx).objectName() == "plugin_tab": found_idx = idx break # show the Tab if not found_idx: try: self.app.ui.notebook.addTab(self.app.ui.plugin_tab, _("Plugin")) except RuntimeError: self.app.ui.plugin_tab = QtWidgets.QWidget() self.app.ui.plugin_tab.setObjectName("plugin_tab") self.app.ui.plugin_tab_layout = QtWidgets.QVBoxLayout(self.app.ui.plugin_tab) self.app.ui.plugin_tab_layout.setContentsMargins(2, 2, 2, 2) self.app.ui.plugin_scroll_area = VerticalScrollArea() self.app.ui.plugin_tab_layout.addWidget(self.app.ui.plugin_scroll_area) self.app.ui.notebook.addTab(self.app.ui.plugin_tab, _("Plugin")) # focus on Tool Tab self.app.ui.notebook.setCurrentWidget(self.app.ui.plugin_tab) # self.app.ui.notebook.callback_on_close = self.on_tab_close self.app.ui.notebook.setTabText(2, self.plugin_name) def set_tool_ui(self): # Init appGUI self.ui.geo_tol_entry.set_value(0.01 if self.draw_app.units == 'MM' else 0.0004) selected_shapes_geos = [] selected_tree_items = self.draw_app.tw.selectedItems() for sel in selected_tree_items: for obj_shape in self.draw_app.storage.get_objects(): try: if id(obj_shape) == int(sel.text(0)): selected_shapes_geos.append(obj_shape.geo) except ValueError: pass if selected_shapes_geos: # those are displayed by triggering the signal self.update_ui self.calculate_coords_vertex(selected_shapes_geos[-1]) def on_tab_close(self): self.draw_app.select_tool("select") self.app.ui.notebook.callback_on_close = lambda: None def on_simplification_click(self): self.app.log.debug("FCSimplification.on_simplification_click()") selected_shapes_geos = [] tol = self.ui.geo_tol_entry.get_value() def task_job(self): with self.app.proc_container.new('%s...' % _("Simplify")): selected_shapes = self.draw_app.get_selected() self.draw_app.interdict_selection = True for obj_shape in selected_shapes: selected_shapes_geos.append(obj_shape.geo.simplify(tolerance=tol)) if not selected_shapes: self.app.inform.emit('%s' % _("Failed.")) return for shape in selected_shapes: self.draw_app.delete_shape(shape=shape) for geo in selected_shapes_geos: self.draw_app.add_shape(geo, build_ui=False) self.draw_app.selected = [] last_sel_geo = selected_shapes_geos[-1] self.calculate_coords_vertex(last_sel_geo) self.app.inform.emit('%s' % _("Done.")) self.draw_app.plot_all() self.draw_app.interdict_selection = False self.draw_app.build_ui_sig.emit() self.app.worker_task.emit({'fcn': task_job, 'params': [self]}) def calculate_coords_vertex(self, last_sel_geo): if last_sel_geo: if last_sel_geo.geom_type == 'MultiLineString': coords = '' vertex_nr = 0 for idx, line in enumerate(last_sel_geo.geoms): line_coords = list(line.coords) vertex_nr += len(line_coords) coords += 'Line %s\n' % str(idx) coords += str(line_coords) + '\n' elif last_sel_geo.geom_type == 'MultiPolygon': coords = '' vertex_nr = 0 for idx, poly in enumerate(last_sel_geo.geoms): poly_coords = list(poly.exterior.coords) + [list(i.coords) for i in poly.interiors] vertex_nr += len(poly_coords) coords += 'Polygon %s\n' % str(idx) coords += str(poly_coords) + '\n' elif last_sel_geo.geom_type in ['LinearRing', 'LineString']: coords = list(last_sel_geo.coords) vertex_nr = len(coords) elif last_sel_geo.geom_type == 'Polygon': coords = list(last_sel_geo.exterior.coords) vertex_nr = len(coords) else: coords = 'None' vertex_nr = 0 self.update_ui.emit(coords, vertex_nr) def on_update_ui(self, coords, vertex_nr): self.ui.geo_coords_entry.set_value(str(coords)) self.ui.geo_vertex_entry.set_value(vertex_nr) def hide_tool(self): self.ui.simp_frame.hide() self.app.ui.notebook.setCurrentWidget(self.app.ui.properties_tab) class SimplificationEditorUI: pluginName = _("Simplification") def __init__(self, layout, simp_class): self.simp_class = simp_class self.app = self.simp_class.app self.decimals = self.app.decimals self.layout = layout # Title title_label = FCLabel("%s" % ('Editor ' + self.pluginName)) title_label.setStyleSheet(""" QLabel { font-size: 16px; font-weight: bold; } """) self.layout.addWidget(title_label) # this way I can hide/show the frame self.simp_frame = QtWidgets.QFrame() self.simp_frame.setContentsMargins(0, 0, 0, 0) self.layout.addWidget(self.simp_frame) self.simp_tools_box = QtWidgets.QVBoxLayout() self.simp_tools_box.setContentsMargins(0, 0, 0, 0) self.simp_frame.setLayout(self.simp_tools_box) # Grid Layout grid0 = GLay(v_spacing=5, h_spacing=3) self.simp_tools_box.addLayout(grid0) # Coordinates coords_lbl = FCLabel('%s' % _("Coordinates"), bold=True, color='red') coords_lbl.setToolTip( _("The coordinates of the selected geometry element.") ) grid0.addWidget(coords_lbl, 0, 0, 1, 2) # ############################################################################################################# # Coordinates Frame # ############################################################################################################# coors_frame = FCFrame() grid0.addWidget(coors_frame, 2, 0, 1, 2) coords_grid = GLay(v_spacing=5, h_spacing=3) coors_frame.setLayout(coords_grid) self.geo_coords_entry = FCTextEdit() self.geo_coords_entry.setPlaceholderText( _("The coordinates of the selected geometry element.") ) coords_grid.addWidget(self.geo_coords_entry, 0, 0, 1, 2) # Vertex Points Number vertex_lbl = FCLabel('%s:' % _("Vertex Points"), bold=False) vertex_lbl.setToolTip( _("The number of vertex points in the selected geometry element.") ) self.geo_vertex_entry = FCEntry(decimals=self.decimals) self.geo_vertex_entry.setReadOnly(True) coords_grid.addWidget(vertex_lbl, 2, 0) coords_grid.addWidget(self.geo_vertex_entry, 2, 1) # Simplification Title par_lbl = FCLabel('%s' % _("Parameters"), bold=True, color='blue') grid0.addWidget(par_lbl, 4, 0, 1, 2) # ############################################################################################################# # Parameters Frame # ############################################################################################################# par_frame = FCFrame() grid0.addWidget(par_frame, 6, 0, 1, 2) par_grid = GLay(v_spacing=5, h_spacing=3) par_frame.setLayout(par_grid) # Simplification Tolerance simplification_tol_lbl = FCLabel('%s' % _("Tolerance"), bold=True) simplification_tol_lbl.setToolTip( _("All points in the simplified object will be\n" "within the tolerance distance of the original geometry.") ) self.geo_tol_entry = FCDoubleSpinner() self.geo_tol_entry.set_precision(self.decimals) self.geo_tol_entry.setSingleStep(10 ** -self.decimals) self.geo_tol_entry.set_range(0.0000, 10000.0000) par_grid.addWidget(simplification_tol_lbl, 0, 0) par_grid.addWidget(self.geo_tol_entry, 0, 1) # Simplification button self.simplification_btn = FCButton(_("Simplify")) self.simplification_btn.setIcon(QtGui.QIcon(self.app.resource_location + '/simplify32.png')) self.simplification_btn.setToolTip( _("Simplify a geometry element by reducing its vertex points number.") ) self.simplification_btn.setStyleSheet(""" QPushButton { font-weight: bold; } """) self.layout.addWidget(self.simplification_btn) GLay.set_common_column_size([grid0, coords_grid, par_grid], 0) self.layout.addStretch(1)