262 lines
9.9 KiB
Python
262 lines
9.9 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(list, 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('[WARNING_NOTCL] %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.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('<b>%s</b>:' % _("Coordinates"))
|
|
coords_lbl.setToolTip(
|
|
_("The coordinates of the selected geometry element.")
|
|
)
|
|
grid0.addWidget(coords_lbl, 22, 0, 1, 3)
|
|
|
|
self.geo_coords_entry = FCTextEdit()
|
|
self.geo_coords_entry.setPlaceholderText(
|
|
_("The coordinates of the selected geometry element.")
|
|
)
|
|
grid0.addWidget(self.geo_coords_entry, 24, 0, 1, 3)
|
|
|
|
# Vertex Points Number
|
|
vertex_lbl = FCLabel('<b>%s</b>:' % _("Vertex Points"))
|
|
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)
|
|
|
|
grid0.addWidget(vertex_lbl, 26, 0)
|
|
grid0.addWidget(self.geo_vertex_entry, 26, 1, 1, 2)
|
|
|
|
separator_line = QtWidgets.QFrame()
|
|
separator_line.setFrameShape(QtWidgets.QFrame.Shape.HLine)
|
|
separator_line.setFrameShadow(QtWidgets.QFrame.Shadow.Sunken)
|
|
grid0.addWidget(separator_line, 28, 0, 1, 3)
|
|
|
|
# Simplification Title
|
|
simplif_lbl = FCLabel('<b>%s</b>:' % _("Simplification"))
|
|
simplif_lbl.setToolTip(
|
|
_("Simplify a geometry by reducing its vertex points number.")
|
|
)
|
|
grid0.addWidget(simplif_lbl, 30, 0, 1, 3)
|
|
|
|
# Simplification Tolerance
|
|
simplification_tol_lbl = FCLabel('<b>%s</b>:' % _("Tolerance"))
|
|
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)
|
|
|
|
grid0.addWidget(simplification_tol_lbl, 32, 0)
|
|
grid0.addWidget(self.geo_tol_entry, 32, 1, 1, 2)
|
|
|
|
# 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;
|
|
}
|
|
""")
|
|
|
|
grid0.addWidget(self.simplification_btn, 34, 0, 1, 3)
|
|
|
|
self.layout.addStretch(1)
|