e8a15156db
- upgraded the FCButton widget (and made it used everywhere instead of the QPushButton) so it can have the color and font weight properties settable
261 lines
10 KiB
Python
261 lines
10 KiB
Python
|
|
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.ui.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), size=16, bold=True)
|
|
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"), bold=True)
|
|
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.layout.addWidget(self.simplification_btn)
|
|
|
|
GLay.set_common_column_size([grid0, coords_grid, par_grid], 0)
|
|
self.layout.addStretch(1)
|