1249 lines
51 KiB
Python
1249 lines
51 KiB
Python
# ##########################################################
|
|
# FlatCAM: 2D Post-processing for Manufacturing #
|
|
# File Author: Marius Adrian Stanciu (c) #
|
|
# Date: 3/10/2019 #
|
|
# MIT Licence #
|
|
# ##########################################################
|
|
|
|
from PyQt5 import QtWidgets, QtGui, QtCore
|
|
from FlatCAMTool import FlatCAMTool
|
|
from flatcamGUI.GUIElements import FCDoubleSpinner, FCCheckBox, RadioSet, FCComboBox, OptionalInputSection
|
|
from FlatCAMObj import FlatCAMGerber
|
|
|
|
from shapely.geometry import box, MultiPolygon, Polygon, LineString, LinearRing
|
|
from shapely.ops import cascaded_union, unary_union
|
|
import shapely.affinity as affinity
|
|
|
|
from matplotlib.backend_bases import KeyEvent as mpl_key_event
|
|
|
|
from numpy import Inf
|
|
from copy import deepcopy
|
|
import math
|
|
import logging
|
|
import gettext
|
|
import FlatCAMTranslation as fcTranslate
|
|
import builtins
|
|
|
|
fcTranslate.apply_language('strings')
|
|
if '_' not in builtins.__dict__:
|
|
_ = gettext.gettext
|
|
|
|
log = logging.getLogger('base')
|
|
|
|
settings = QtCore.QSettings("Open Source", "FlatCAM")
|
|
if settings.contains("machinist"):
|
|
machinist_setting = settings.value('machinist', type=int)
|
|
else:
|
|
machinist_setting = 0
|
|
|
|
|
|
class CutOut(FlatCAMTool):
|
|
|
|
toolName = _("Cutout PCB")
|
|
|
|
def __init__(self, app):
|
|
FlatCAMTool.__init__(self, app)
|
|
|
|
self.app = app
|
|
self.canvas = app.plotcanvas
|
|
self.decimals = self.app.decimals
|
|
|
|
# Title
|
|
title_label = QtWidgets.QLabel("%s" % self.toolName)
|
|
title_label.setStyleSheet("""
|
|
QLabel
|
|
{
|
|
font-size: 16px;
|
|
font-weight: bold;
|
|
}
|
|
""")
|
|
self.layout.addWidget(title_label)
|
|
|
|
# Form Layout
|
|
grid0 = QtWidgets.QGridLayout()
|
|
grid0.setColumnStretch(0, 0)
|
|
grid0.setColumnStretch(1, 1)
|
|
self.layout.addLayout(grid0)
|
|
|
|
# Type of object to be cutout
|
|
self.type_obj_combo = QtWidgets.QComboBox()
|
|
self.type_obj_combo.addItem("Gerber")
|
|
self.type_obj_combo.addItem("Excellon")
|
|
self.type_obj_combo.addItem("Geometry")
|
|
|
|
# we get rid of item1 ("Excellon") as it is not suitable for creating film
|
|
self.type_obj_combo.view().setRowHidden(1, True)
|
|
self.type_obj_combo.setItemIcon(0, QtGui.QIcon(self.app.resource_location + "/flatcam_icon16.png"))
|
|
# self.type_obj_combo.setItemIcon(1, QtGui.QIcon(self.app.resource_location + "/drill16.png"))
|
|
self.type_obj_combo.setItemIcon(2, QtGui.QIcon(self.app.resource_location + "/geometry16.png"))
|
|
|
|
self.type_obj_combo_label = QtWidgets.QLabel('%s:' % _("Object Type"))
|
|
self.type_obj_combo_label.setToolTip(
|
|
_("Specify the type of object to be cutout.\n"
|
|
"It can be of type: Gerber or Geometry.\n"
|
|
"What is selected here will dictate the kind\n"
|
|
"of objects that will populate the 'Object' combobox.")
|
|
)
|
|
self.type_obj_combo_label.setMinimumWidth(60)
|
|
grid0.addWidget(self.type_obj_combo_label, 0, 0)
|
|
grid0.addWidget(self.type_obj_combo, 0, 1)
|
|
|
|
self.object_label = QtWidgets.QLabel('<b>%s:</b>' % _("Object to be cutout"))
|
|
self.object_label.setToolTip('%s.' % _("Object to be cutout"))
|
|
|
|
# Object to be cutout
|
|
self.obj_combo = QtWidgets.QComboBox()
|
|
self.obj_combo.setModel(self.app.collection)
|
|
self.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
|
|
self.obj_combo.setCurrentIndex(1)
|
|
|
|
grid0.addWidget(self.object_label, 1, 0, 1, 2)
|
|
grid0.addWidget(self.obj_combo, 2, 0, 1, 2)
|
|
|
|
# Object kind
|
|
self.kindlabel = QtWidgets.QLabel('%s:' % _('Object kind'))
|
|
self.kindlabel.setToolTip(
|
|
_("Choice of what kind the object we want to cutout is.<BR>"
|
|
"- <B>Single</B>: contain a single PCB Gerber outline object.<BR>"
|
|
"- <B>Panel</B>: a panel PCB Gerber object, which is made\n"
|
|
"out of many individual PCB outlines.")
|
|
)
|
|
self.obj_kind_combo = RadioSet([
|
|
{"label": _("Single"), "value": "single"},
|
|
{"label": _("Panel"), "value": "panel"},
|
|
])
|
|
grid0.addWidget(self.kindlabel, 3, 0)
|
|
grid0.addWidget(self.obj_kind_combo, 3, 1)
|
|
|
|
# Tool Diameter
|
|
self.dia = FCDoubleSpinner()
|
|
self.dia.set_precision(self.decimals)
|
|
self.dia.set_range(0.0000, 9999.9999)
|
|
|
|
self.dia_label = QtWidgets.QLabel('%s:' % _("Tool Diameter"))
|
|
self.dia_label.setToolTip(
|
|
_("Diameter of the tool used to cutout\n"
|
|
"the PCB shape out of the surrounding material.")
|
|
)
|
|
grid0.addWidget(self.dia_label, 4, 0)
|
|
grid0.addWidget(self.dia, 4, 1)
|
|
|
|
# Cut Z
|
|
cutzlabel = QtWidgets.QLabel('%s:' % _('Cut Z'))
|
|
cutzlabel.setToolTip(
|
|
_(
|
|
"Cutting depth (negative)\n"
|
|
"below the copper surface."
|
|
)
|
|
)
|
|
self.cutz_entry = FCDoubleSpinner()
|
|
self.cutz_entry.set_precision(self.decimals)
|
|
|
|
if machinist_setting == 0:
|
|
self.cutz_entry.setRange(-9999.9999, -0.00001)
|
|
else:
|
|
self.cutz_entry.setRange(-9999.9999, 9999.9999)
|
|
|
|
self.cutz_entry.setSingleStep(0.1)
|
|
|
|
grid0.addWidget(cutzlabel, 5, 0)
|
|
grid0.addWidget(self.cutz_entry, 5, 1)
|
|
|
|
# Multi-pass
|
|
self.mpass_cb = FCCheckBox('%s:' % _("Multi-Depth"))
|
|
self.mpass_cb.setToolTip(
|
|
_(
|
|
"Use multiple passes to limit\n"
|
|
"the cut depth in each pass. Will\n"
|
|
"cut multiple times until Cut Z is\n"
|
|
"reached."
|
|
)
|
|
)
|
|
|
|
self.maxdepth_entry = FCDoubleSpinner()
|
|
self.maxdepth_entry.set_precision(self.decimals)
|
|
self.maxdepth_entry.setRange(0, 9999.9999)
|
|
self.maxdepth_entry.setSingleStep(0.1)
|
|
|
|
self.maxdepth_entry.setToolTip(
|
|
_(
|
|
"Depth of each pass (positive)."
|
|
)
|
|
)
|
|
self.ois_mpass_geo = OptionalInputSection(self.mpass_cb, [self.maxdepth_entry])
|
|
|
|
grid0.addWidget(self.mpass_cb, 6, 0)
|
|
grid0.addWidget(self.maxdepth_entry, 6, 1)
|
|
|
|
# Margin
|
|
self.margin = FCDoubleSpinner()
|
|
self.margin.set_precision(self.decimals)
|
|
|
|
self.margin_label = QtWidgets.QLabel('%s:' % _("Margin"))
|
|
self.margin_label.setToolTip(
|
|
_("Margin over bounds. A positive value here\n"
|
|
"will make the cutout of the PCB further from\n"
|
|
"the actual PCB border")
|
|
)
|
|
grid0.addWidget(self.margin_label, 7, 0)
|
|
grid0.addWidget(self.margin, 7, 1)
|
|
|
|
# Gapsize
|
|
self.gapsize = FCDoubleSpinner()
|
|
self.gapsize.set_precision(self.decimals)
|
|
|
|
self.gapsize_label = QtWidgets.QLabel('%s:' % _("Gap size"))
|
|
self.gapsize_label.setToolTip(
|
|
_("The size of the bridge gaps in the cutout\n"
|
|
"used to keep the board connected to\n"
|
|
"the surrounding material (the one \n"
|
|
"from which the PCB is cutout).")
|
|
)
|
|
grid0.addWidget(self.gapsize_label, 8, 0)
|
|
grid0.addWidget(self.gapsize, 8, 1)
|
|
|
|
# How gaps wil be rendered:
|
|
# lr - left + right
|
|
# tb - top + bottom
|
|
# 4 - left + right +top + bottom
|
|
# 2lr - 2*left + 2*right
|
|
# 2tb - 2*top + 2*bottom
|
|
# 8 - 2*left + 2*right +2*top + 2*bottom
|
|
|
|
# Surrounding convex box shape
|
|
self.convex_box = FCCheckBox('%s' % _("Convex Shape"))
|
|
# self.convex_box_label = QtWidgets.QLabel('%s' % _("Convex Sh."))
|
|
self.convex_box.setToolTip(
|
|
_("Create a convex shape surrounding the entire PCB.\n"
|
|
"Used only if the source object type is Gerber.")
|
|
)
|
|
grid0.addWidget(self.convex_box, 9, 0, 1, 2)
|
|
|
|
separator_line = QtWidgets.QFrame()
|
|
separator_line.setFrameShape(QtWidgets.QFrame.HLine)
|
|
separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
|
|
grid0.addWidget(separator_line, 10, 0, 1, 2)
|
|
|
|
# Title2
|
|
title_param_label = QtWidgets.QLabel("<font size=4><b>%s</b></font>" % _('A. Automatic Bridge Gaps'))
|
|
title_param_label.setToolTip(
|
|
_("This section handle creation of automatic bridge gaps.")
|
|
)
|
|
self.layout.addWidget(title_param_label)
|
|
|
|
# Form Layout
|
|
form_layout_2 = QtWidgets.QFormLayout()
|
|
self.layout.addLayout(form_layout_2)
|
|
|
|
# Gaps
|
|
gaps_label = QtWidgets.QLabel('%s:' % _('Gaps'))
|
|
gaps_label.setToolTip(
|
|
_("Number of gaps used for the Automatic cutout.\n"
|
|
"There can be maximum 8 bridges/gaps.\n"
|
|
"The choices are:\n"
|
|
"- None - no gaps\n"
|
|
"- lr - left + right\n"
|
|
"- tb - top + bottom\n"
|
|
"- 4 - left + right +top + bottom\n"
|
|
"- 2lr - 2*left + 2*right\n"
|
|
"- 2tb - 2*top + 2*bottom\n"
|
|
"- 8 - 2*left + 2*right +2*top + 2*bottom")
|
|
)
|
|
gaps_label.setMinimumWidth(60)
|
|
|
|
self.gaps = FCComboBox()
|
|
gaps_items = ['None', 'LR', 'TB', '4', '2LR', '2TB', '8']
|
|
for it in gaps_items:
|
|
self.gaps.addItem(it)
|
|
self.gaps.setStyleSheet('background-color: rgb(255,255,255)')
|
|
form_layout_2.addRow(gaps_label, self.gaps)
|
|
|
|
# Buttons
|
|
self.ff_cutout_object_btn = QtWidgets.QPushButton(_("Generate Freeform Geometry"))
|
|
self.ff_cutout_object_btn.setToolTip(
|
|
_("Cutout the selected object.\n"
|
|
"The cutout shape can be of any shape.\n"
|
|
"Useful when the PCB has a non-rectangular shape.")
|
|
)
|
|
self.ff_cutout_object_btn.setStyleSheet("""
|
|
QPushButton
|
|
{
|
|
font-weight: bold;
|
|
}
|
|
""")
|
|
self.layout.addWidget(self.ff_cutout_object_btn)
|
|
|
|
self.rect_cutout_object_btn = QtWidgets.QPushButton(_("Generate Rectangular Geometry"))
|
|
self.rect_cutout_object_btn.setToolTip(
|
|
_("Cutout the selected object.\n"
|
|
"The resulting cutout shape is\n"
|
|
"always a rectangle shape and it will be\n"
|
|
"the bounding box of the Object.")
|
|
)
|
|
self.rect_cutout_object_btn.setStyleSheet("""
|
|
QPushButton
|
|
{
|
|
font-weight: bold;
|
|
}
|
|
""")
|
|
self.layout.addWidget(self.rect_cutout_object_btn)
|
|
|
|
separator_line = QtWidgets.QFrame()
|
|
separator_line.setFrameShape(QtWidgets.QFrame.HLine)
|
|
separator_line.setFrameShadow(QtWidgets.QFrame.Sunken)
|
|
self.layout.addWidget(separator_line)
|
|
|
|
# Title5
|
|
title_manual_label = QtWidgets.QLabel("<font size=4><b>%s</b></font>" % _('B. Manual Bridge Gaps'))
|
|
title_manual_label.setToolTip(
|
|
_("This section handle creation of manual bridge gaps.\n"
|
|
"This is done by mouse clicking on the perimeter of the\n"
|
|
"Geometry object that is used as a cutout object. ")
|
|
)
|
|
self.layout.addWidget(title_manual_label)
|
|
|
|
# Form Layout
|
|
form_layout_3 = QtWidgets.QFormLayout()
|
|
self.layout.addLayout(form_layout_3)
|
|
|
|
# Manual Geo Object
|
|
self.man_object_combo = QtWidgets.QComboBox()
|
|
self.man_object_combo.setModel(self.app.collection)
|
|
self.man_object_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
|
|
self.man_object_combo.setCurrentIndex(1)
|
|
|
|
self.man_object_label = QtWidgets.QLabel('%s:' % _("Geometry Object"))
|
|
self.man_object_label.setToolTip(
|
|
_("Geometry object used to create the manual cutout.")
|
|
)
|
|
self.man_object_label.setMinimumWidth(60)
|
|
|
|
form_layout_3.addRow(self.man_object_label)
|
|
form_layout_3.addRow(self.man_object_combo)
|
|
|
|
# form_layout_3.addRow(e_lab_0)
|
|
|
|
self.man_geo_creation_btn = QtWidgets.QPushButton(_("Generate Manual Geometry"))
|
|
self.man_geo_creation_btn.setToolTip(
|
|
_("If the object to be cutout is a Gerber\n"
|
|
"first create a Geometry that surrounds it,\n"
|
|
"to be used as the cutout, if one doesn't exist yet.\n"
|
|
"Select the source Gerber file in the top object combobox.")
|
|
)
|
|
self.man_geo_creation_btn.setStyleSheet("""
|
|
QPushButton
|
|
{
|
|
font-weight: bold;
|
|
}
|
|
""")
|
|
self.layout.addWidget(self.man_geo_creation_btn)
|
|
|
|
self.man_gaps_creation_btn = QtWidgets.QPushButton(_("Manual Add Bridge Gaps"))
|
|
self.man_gaps_creation_btn.setToolTip(
|
|
_("Use the left mouse button (LMB) click\n"
|
|
"to create a bridge gap to separate the PCB from\n"
|
|
"the surrounding material.\n"
|
|
"The LMB click has to be done on the perimeter of\n"
|
|
"the Geometry object used as a cutout geometry.")
|
|
)
|
|
self.man_gaps_creation_btn.setStyleSheet("""
|
|
QPushButton
|
|
{
|
|
font-weight: bold;
|
|
}
|
|
""")
|
|
self.layout.addWidget(self.man_gaps_creation_btn)
|
|
|
|
self.layout.addStretch()
|
|
|
|
# ## Reset Tool
|
|
self.reset_button = QtWidgets.QPushButton(_("Reset Tool"))
|
|
self.reset_button.setToolTip(
|
|
_("Will reset the tool parameters.")
|
|
)
|
|
self.reset_button.setStyleSheet("""
|
|
QPushButton
|
|
{
|
|
font-weight: bold;
|
|
}
|
|
""")
|
|
self.layout.addWidget(self.reset_button)
|
|
|
|
self.cutting_gapsize = 0.0
|
|
self.cutting_dia = 0.0
|
|
|
|
# true if we want to repeat the gap without clicking again on the button
|
|
self.repeat_gap = False
|
|
|
|
self.flat_geometry = []
|
|
|
|
# this is the Geometry object generated in this class to be used for adding manual gaps
|
|
self.man_cutout_obj = None
|
|
|
|
# if mouse is dragging set the object True
|
|
self.mouse_is_dragging = False
|
|
|
|
# event handlers references
|
|
self.kp = None
|
|
self.mm = None
|
|
self.mr = None
|
|
|
|
# hold the mouse position here
|
|
self.x_pos = None
|
|
self.y_pos = None
|
|
|
|
# Signals
|
|
self.ff_cutout_object_btn.clicked.connect(self.on_freeform_cutout)
|
|
self.rect_cutout_object_btn.clicked.connect(self.on_rectangular_cutout)
|
|
|
|
self.type_obj_combo.currentIndexChanged.connect(self.on_type_obj_index_changed)
|
|
self.man_geo_creation_btn.clicked.connect(self.on_manual_geo)
|
|
self.man_gaps_creation_btn.clicked.connect(self.on_manual_gap_click)
|
|
self.reset_button.clicked.connect(self.set_tool_ui)
|
|
|
|
def on_type_obj_index_changed(self, index):
|
|
obj_type = self.type_obj_combo.currentIndex()
|
|
self.obj_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
|
|
self.obj_combo.setCurrentIndex(0)
|
|
|
|
def run(self, toggle=True):
|
|
self.app.report_usage("ToolCutOut()")
|
|
|
|
if toggle:
|
|
# if the splitter is hidden, display it, else hide it but only if the current widget is the same
|
|
if self.app.ui.splitter.sizes()[0] == 0:
|
|
self.app.ui.splitter.setSizes([1, 1])
|
|
else:
|
|
try:
|
|
if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
|
|
# if tab is populated with the tool but it does not have the focus, focus on it
|
|
if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
|
|
# focus on Tool Tab
|
|
self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
|
|
else:
|
|
self.app.ui.splitter.setSizes([0, 1])
|
|
except AttributeError:
|
|
pass
|
|
else:
|
|
if self.app.ui.splitter.sizes()[0] == 0:
|
|
self.app.ui.splitter.setSizes([1, 1])
|
|
|
|
FlatCAMTool.run(self)
|
|
self.set_tool_ui()
|
|
|
|
self.app.ui.notebook.setTabText(2, _("Cutout Tool"))
|
|
|
|
def install(self, icon=None, separator=None, **kwargs):
|
|
FlatCAMTool.install(self, icon, separator, shortcut='ALT+X', **kwargs)
|
|
|
|
def set_tool_ui(self):
|
|
self.reset_fields()
|
|
|
|
self.dia.set_value(float(self.app.defaults["tools_cutouttooldia"]))
|
|
self.obj_kind_combo.set_value(self.app.defaults["tools_cutoutkind"])
|
|
self.margin.set_value(float(self.app.defaults["tools_cutoutmargin"]))
|
|
self.cutz_entry.set_value(float(self.app.defaults["tools_cutout_z"]))
|
|
self.mpass_cb.set_value(float(self.app.defaults["tools_cutout_mdepth"]))
|
|
self.maxdepth_entry.set_value(float(self.app.defaults["tools_cutout_depthperpass"]))
|
|
|
|
self.gapsize.set_value(float(self.app.defaults["tools_cutoutgapsize"]))
|
|
self.gaps.set_value(self.app.defaults["tools_gaps_ff"])
|
|
self.convex_box.set_value(self.app.defaults['tools_cutout_convexshape'])
|
|
|
|
def on_freeform_cutout(self):
|
|
|
|
# def subtract_rectangle(obj_, x0, y0, x1, y1):
|
|
# pts = [(x0, y0), (x1, y0), (x1, y1), (x0, y1)]
|
|
# obj_.subtract_polygon(pts)
|
|
|
|
name = self.obj_combo.currentText()
|
|
|
|
# Get source object.
|
|
try:
|
|
cutout_obj = self.app.collection.get_by_name(str(name))
|
|
except Exception as e:
|
|
log.debug("CutOut.on_freeform_cutout() --> %s" % str(e))
|
|
self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), name))
|
|
return "Could not retrieve object: %s" % name
|
|
|
|
if cutout_obj is None:
|
|
self.app.inform.emit('[ERROR_NOTCL] %s' %
|
|
_("There is no object selected for Cutout.\nSelect one and try again."))
|
|
return
|
|
|
|
dia = float(self.dia.get_value())
|
|
if 0 in {dia}:
|
|
self.app.inform.emit('[WARNING_NOTCL] %s' %
|
|
_("Tool Diameter is zero value. Change it to a positive real number."))
|
|
return "Tool Diameter is zero value. Change it to a positive real number."
|
|
|
|
try:
|
|
kind = self.obj_kind_combo.get_value()
|
|
except ValueError:
|
|
return
|
|
|
|
margin = float(self.margin.get_value())
|
|
gapsize = float(self.gapsize.get_value())
|
|
|
|
try:
|
|
gaps = self.gaps.get_value()
|
|
except TypeError:
|
|
self.app.inform.emit('[WARNING_NOTCL] %s' % _("Number of gaps value is missing. Add it and retry."))
|
|
return
|
|
|
|
if gaps not in ['None', 'LR', 'TB', '2LR', '2TB', '4', '8']:
|
|
self.app.inform.emit('[WARNING_NOTCL] %s' %
|
|
_("Gaps value can be only one of: 'None', 'lr', 'tb', '2lr', '2tb', 4 or 8. "
|
|
"Fill in a correct value and retry. "))
|
|
return
|
|
|
|
if cutout_obj.multigeo is True:
|
|
self.app.inform.emit('[ERROR] %s' % _("Cutout operation cannot be done on a multi-geo Geometry.\n"
|
|
"Optionally, this Multi-geo Geometry can be converted to "
|
|
"Single-geo Geometry,\n"
|
|
"and after that perform Cutout."))
|
|
return
|
|
|
|
convex_box = self.convex_box.get_value()
|
|
|
|
gapsize = gapsize / 2 + (dia / 2)
|
|
|
|
def geo_init(geo_obj, app_obj):
|
|
solid_geo = []
|
|
|
|
if isinstance(cutout_obj, FlatCAMGerber):
|
|
if convex_box:
|
|
object_geo = cutout_obj.solid_geometry.convex_hull
|
|
else:
|
|
object_geo = cutout_obj.solid_geometry
|
|
else:
|
|
object_geo = cutout_obj.solid_geometry
|
|
|
|
def cutout_handler(geom):
|
|
# Get min and max data for each object as we just cut rectangles across X or Y
|
|
xmin, ymin, xmax, ymax = recursive_bounds(geom)
|
|
|
|
px = 0.5 * (xmin + xmax) + margin
|
|
py = 0.5 * (ymin + ymax) + margin
|
|
lenx = (xmax - xmin) + (margin * 2)
|
|
leny = (ymax - ymin) + (margin * 2)
|
|
|
|
proc_geometry = []
|
|
if gaps == 'None':
|
|
pass
|
|
else:
|
|
if gaps == '8' or gaps == '2LR':
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
xmin - gapsize, # botleft_x
|
|
py - gapsize + leny / 4, # botleft_y
|
|
xmax + gapsize, # topright_x
|
|
py + gapsize + leny / 4) # topright_y
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
xmin - gapsize,
|
|
py - gapsize - leny / 4,
|
|
xmax + gapsize,
|
|
py + gapsize - leny / 4)
|
|
|
|
if gaps == '8' or gaps == '2TB':
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
px - gapsize + lenx / 4,
|
|
ymin - gapsize,
|
|
px + gapsize + lenx / 4,
|
|
ymax + gapsize)
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
px - gapsize - lenx / 4,
|
|
ymin - gapsize,
|
|
px + gapsize - lenx / 4,
|
|
ymax + gapsize)
|
|
|
|
if gaps == '4' or gaps == 'LR':
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
xmin - gapsize,
|
|
py - gapsize,
|
|
xmax + gapsize,
|
|
py + gapsize)
|
|
|
|
if gaps == '4' or gaps == 'TB':
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
px - gapsize,
|
|
ymin - gapsize,
|
|
px + gapsize,
|
|
ymax + gapsize)
|
|
|
|
try:
|
|
for g in geom:
|
|
proc_geometry.append(g)
|
|
except TypeError:
|
|
proc_geometry.append(geom)
|
|
|
|
return proc_geometry
|
|
|
|
if kind == 'single':
|
|
object_geo = unary_union(object_geo)
|
|
|
|
# for geo in object_geo:
|
|
if isinstance(cutout_obj, FlatCAMGerber):
|
|
if isinstance(object_geo, MultiPolygon):
|
|
x0, y0, x1, y1 = object_geo.bounds
|
|
object_geo = box(x0, y0, x1, y1)
|
|
|
|
geo_buf = object_geo.buffer(margin + abs(dia / 2))
|
|
geo = geo_buf.exterior
|
|
else:
|
|
geo = object_geo
|
|
|
|
solid_geo = cutout_handler(geom=geo)
|
|
else:
|
|
try:
|
|
__ = iter(object_geo)
|
|
except TypeError:
|
|
object_geo = [object_geo]
|
|
|
|
for geom_struct in object_geo:
|
|
if isinstance(cutout_obj, FlatCAMGerber):
|
|
geom_struct = (geom_struct.buffer(margin + abs(dia / 2))).exterior
|
|
|
|
solid_geo += cutout_handler(geom=geom_struct)
|
|
|
|
geo_obj.solid_geometry = deepcopy(solid_geo)
|
|
xmin, ymin, xmax, ymax = recursive_bounds(geo_obj.solid_geometry)
|
|
geo_obj.options['xmin'] = xmin
|
|
geo_obj.options['ymin'] = ymin
|
|
geo_obj.options['xmax'] = xmax
|
|
geo_obj.options['ymax'] = ymax
|
|
geo_obj.options['cnctooldia'] = str(dia)
|
|
geo_obj.options['cutz'] = self.cutz_entry.get_value()
|
|
geo_obj.options['multidepth'] = self.mpass_cb.get_value()
|
|
geo_obj.options['depthperpass'] = self.maxdepth_entry.get_value()
|
|
|
|
outname = cutout_obj.options["name"] + "_cutout"
|
|
self.app.new_object('geometry', outname, geo_init)
|
|
|
|
cutout_obj.plot()
|
|
self.app.inform.emit('[success] %s' % _("Any form CutOut operation finished."))
|
|
self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
|
|
self.app.should_we_save = True
|
|
|
|
def on_rectangular_cutout(self):
|
|
|
|
# def subtract_rectangle(obj_, x0, y0, x1, y1):
|
|
# pts = [(x0, y0), (x1, y0), (x1, y1), (x0, y1)]
|
|
# obj_.subtract_polygon(pts)
|
|
|
|
name = self.obj_combo.currentText()
|
|
|
|
# Get source object.
|
|
try:
|
|
cutout_obj = self.app.collection.get_by_name(str(name))
|
|
except Exception as e:
|
|
log.debug("CutOut.on_rectangular_cutout() --> %s" % str(e))
|
|
self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve object"), name))
|
|
return "Could not retrieve object: %s" % name
|
|
|
|
if cutout_obj is None:
|
|
self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Object not found"), str(name)))
|
|
|
|
dia = float(self.dia.get_value())
|
|
if 0 in {dia}:
|
|
self.app.inform.emit('[ERROR_NOTCL] %s' %
|
|
_("Tool Diameter is zero value. Change it to a positive real number."))
|
|
return "Tool Diameter is zero value. Change it to a positive real number."
|
|
|
|
try:
|
|
kind = self.obj_kind_combo.get_value()
|
|
except ValueError:
|
|
return
|
|
|
|
margin = float(self.margin.get_value())
|
|
gapsize = float(self.gapsize.get_value())
|
|
|
|
try:
|
|
gaps = self.gaps.get_value()
|
|
except TypeError:
|
|
self.app.inform.emit('[WARNING_NOTCL] %s' %
|
|
_("Number of gaps value is missing. Add it and retry."))
|
|
return
|
|
|
|
if gaps not in ['None', 'LR', 'TB', '2LR', '2TB', '4', '8']:
|
|
self.app.inform.emit('[WARNING_NOTCL] %s' % _("Gaps value can be only one of: "
|
|
"'None', 'lr', 'tb', '2lr', '2tb', 4 or 8. "
|
|
"Fill in a correct value and retry. "))
|
|
return
|
|
|
|
if cutout_obj.multigeo is True:
|
|
self.app.inform.emit('[ERROR] %s' % _("Cutout operation cannot be done on a multi-geo Geometry.\n"
|
|
"Optionally, this Multi-geo Geometry can be converted to "
|
|
"Single-geo Geometry,\n"
|
|
"and after that perform Cutout."))
|
|
return
|
|
|
|
# Get min and max data for each object as we just cut rectangles across X or Y
|
|
|
|
gapsize = gapsize / 2 + (dia / 2)
|
|
|
|
def geo_init(geo_obj, app_obj):
|
|
solid_geo = []
|
|
object_geo = cutout_obj.solid_geometry
|
|
|
|
def cutout_rect_handler(geom):
|
|
proc_geometry = []
|
|
|
|
px = 0.5 * (xmin + xmax) + margin
|
|
py = 0.5 * (ymin + ymax) + margin
|
|
lenx = (xmax - xmin) + (margin * 2)
|
|
leny = (ymax - ymin) + (margin * 2)
|
|
|
|
if gaps == 'None':
|
|
pass
|
|
else:
|
|
if gaps == '8' or gaps == '2LR':
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
xmin - gapsize, # botleft_x
|
|
py - gapsize + leny / 4, # botleft_y
|
|
xmax + gapsize, # topright_x
|
|
py + gapsize + leny / 4) # topright_y
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
xmin - gapsize,
|
|
py - gapsize - leny / 4,
|
|
xmax + gapsize,
|
|
py + gapsize - leny / 4)
|
|
|
|
if gaps == '8' or gaps == '2TB':
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
px - gapsize + lenx / 4,
|
|
ymin - gapsize,
|
|
px + gapsize + lenx / 4,
|
|
ymax + gapsize)
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
px - gapsize - lenx / 4,
|
|
ymin - gapsize,
|
|
px + gapsize - lenx / 4,
|
|
ymax + gapsize)
|
|
|
|
if gaps == '4' or gaps == 'LR':
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
xmin - gapsize,
|
|
py - gapsize,
|
|
xmax + gapsize,
|
|
py + gapsize)
|
|
|
|
if gaps == '4' or gaps == 'TB':
|
|
geom = self.subtract_poly_from_geo(geom,
|
|
px - gapsize,
|
|
ymin - gapsize,
|
|
px + gapsize,
|
|
ymax + gapsize)
|
|
try:
|
|
for g in geom:
|
|
proc_geometry.append(g)
|
|
except TypeError:
|
|
proc_geometry.append(geom)
|
|
return proc_geometry
|
|
|
|
if kind == 'single':
|
|
object_geo = unary_union(object_geo)
|
|
|
|
xmin, ymin, xmax, ymax = object_geo.bounds
|
|
geo = box(xmin, ymin, xmax, ymax)
|
|
|
|
# if Gerber create a buffer at a distance
|
|
# if Geometry then cut through the geometry
|
|
if isinstance(cutout_obj, FlatCAMGerber):
|
|
geo = geo.buffer(margin + abs(dia / 2))
|
|
|
|
solid_geo = cutout_rect_handler(geom=geo)
|
|
else:
|
|
try:
|
|
__ = iter(object_geo)
|
|
except TypeError:
|
|
object_geo = [object_geo]
|
|
|
|
for geom_struct in object_geo:
|
|
geom_struct = unary_union(geom_struct)
|
|
xmin, ymin, xmax, ymax = geom_struct.bounds
|
|
geom_struct = box(xmin, ymin, xmax, ymax)
|
|
|
|
if isinstance(cutout_obj, FlatCAMGerber):
|
|
geom_struct = geom_struct.buffer(margin + abs(dia / 2))
|
|
|
|
solid_geo += cutout_rect_handler(geom=geom_struct)
|
|
|
|
geo_obj.solid_geometry = deepcopy(solid_geo)
|
|
geo_obj.options['cnctooldia'] = str(dia)
|
|
geo_obj.options['cutz'] = self.cutz_entry.get_value()
|
|
geo_obj.options['multidepth'] = self.mpass_cb.get_value()
|
|
geo_obj.options['depthperpass'] = self.maxdepth_entry.get_value()
|
|
|
|
outname = cutout_obj.options["name"] + "_cutout"
|
|
self.app.new_object('geometry', outname, geo_init)
|
|
|
|
# cutout_obj.plot()
|
|
self.app.inform.emit('[success] %s' %
|
|
_("Any form CutOut operation finished."))
|
|
self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
|
|
self.app.should_we_save = True
|
|
|
|
def on_manual_gap_click(self):
|
|
self.app.inform.emit(_("Click on the selected geometry object perimeter to create a bridge gap ..."))
|
|
self.app.geo_editor.tool_shape.enabled = True
|
|
|
|
self.cutting_dia = float(self.dia.get_value())
|
|
if 0 in {self.cutting_dia}:
|
|
self.app.inform.emit('[ERROR_NOTCL] %s' %
|
|
_("Tool Diameter is zero value. Change it to a positive real number."))
|
|
return "Tool Diameter is zero value. Change it to a positive real number."
|
|
|
|
self.cutting_gapsize = float(self.gapsize.get_value())
|
|
|
|
name = self.man_object_combo.currentText()
|
|
# Get Geometry source object to be used as target for Manual adding Gaps
|
|
try:
|
|
self.man_cutout_obj = self.app.collection.get_by_name(str(name))
|
|
except Exception as e:
|
|
log.debug("CutOut.on_manual_cutout() --> %s" % str(e))
|
|
self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Geometry object"), name))
|
|
return "Could not retrieve object: %s" % name
|
|
|
|
if self.app.is_legacy is False:
|
|
self.app.plotcanvas.graph_event_disconnect('key_press', self.app.ui.keyPressEvent)
|
|
self.app.plotcanvas.graph_event_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
|
|
self.app.plotcanvas.graph_event_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
|
|
self.app.plotcanvas.graph_event_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
|
|
else:
|
|
self.app.plotcanvas.graph_event_disconnect(self.app.kp)
|
|
self.app.plotcanvas.graph_event_disconnect(self.app.mp)
|
|
self.app.plotcanvas.graph_event_disconnect(self.app.mr)
|
|
self.app.plotcanvas.graph_event_disconnect(self.app.mm)
|
|
|
|
self.kp = self.app.plotcanvas.graph_event_connect('key_press', self.on_key_press)
|
|
self.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.on_mouse_move)
|
|
self.mr = self.app.plotcanvas.graph_event_connect('mouse_release', self.on_mouse_click_release)
|
|
|
|
def on_manual_cutout(self, click_pos):
|
|
name = self.man_object_combo.currentText()
|
|
|
|
# Get source object.
|
|
try:
|
|
self.man_cutout_obj = self.app.collection.get_by_name(str(name))
|
|
except Exception as e:
|
|
log.debug("CutOut.on_manual_cutout() --> %s" % str(e))
|
|
self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Geometry object"), name))
|
|
return "Could not retrieve object: %s" % name
|
|
|
|
if self.man_cutout_obj is None:
|
|
self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
|
|
(_("Geometry object for manual cutout not found"), self.man_cutout_obj))
|
|
return
|
|
|
|
# use the snapped position as reference
|
|
snapped_pos = self.app.geo_editor.snap(click_pos[0], click_pos[1])
|
|
|
|
cut_poly = self.cutting_geo(pos=(snapped_pos[0], snapped_pos[1]))
|
|
self.man_cutout_obj.subtract_polygon(cut_poly)
|
|
|
|
self.man_cutout_obj.plot()
|
|
self.app.inform.emit('[success] %s' % _("Added manual Bridge Gap."))
|
|
|
|
self.app.should_we_save = True
|
|
|
|
def on_manual_geo(self):
|
|
name = self.obj_combo.currentText()
|
|
|
|
# Get source object.
|
|
try:
|
|
cutout_obj = self.app.collection.get_by_name(str(name))
|
|
except Exception as e:
|
|
log.debug("CutOut.on_manual_geo() --> %s" % str(e))
|
|
self.app.inform.emit('[ERROR_NOTCL] %s: %s' % (_("Could not retrieve Gerber object"), name))
|
|
return "Could not retrieve object: %s" % name
|
|
|
|
if cutout_obj is None:
|
|
self.app.inform.emit('[ERROR_NOTCL] %s' %
|
|
_("There is no Gerber object selected for Cutout.\n"
|
|
"Select one and try again."))
|
|
return
|
|
|
|
if not isinstance(cutout_obj, FlatCAMGerber):
|
|
self.app.inform.emit('[ERROR_NOTCL] %s' %
|
|
_("The selected object has to be of Gerber type.\n"
|
|
"Select a Gerber file and try again."))
|
|
return
|
|
|
|
dia = float(self.dia.get_value())
|
|
if 0 in {dia}:
|
|
self.app.inform.emit('[ERROR_NOTCL] %s' %
|
|
_("Tool Diameter is zero value. Change it to a positive real number."))
|
|
return "Tool Diameter is zero value. Change it to a positive real number."
|
|
|
|
try:
|
|
kind = self.obj_kind_combo.get_value()
|
|
except ValueError:
|
|
return
|
|
|
|
margin = float(self.margin.get_value())
|
|
convex_box = self.convex_box.get_value()
|
|
|
|
def geo_init(geo_obj, app_obj):
|
|
geo_union = unary_union(cutout_obj.solid_geometry)
|
|
|
|
if convex_box:
|
|
geo = geo_union.convex_hull
|
|
geo_obj.solid_geometry = geo.buffer(margin + abs(dia / 2))
|
|
elif kind == 'single':
|
|
if isinstance(geo_union, Polygon) or \
|
|
(isinstance(geo_union, list) and len(geo_union) == 1) or \
|
|
(isinstance(geo_union, MultiPolygon) and len(geo_union) == 1):
|
|
geo_obj.solid_geometry = geo_union.buffer(margin + abs(dia / 2)).exterior
|
|
elif isinstance(geo_union, MultiPolygon):
|
|
x0, y0, x1, y1 = geo_union.bounds
|
|
geo = box(x0, y0, x1, y1)
|
|
geo_obj.solid_geometry = geo.buffer(margin + abs(dia / 2))
|
|
else:
|
|
self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
|
|
(_("Geometry not supported for cutout"), type(geo_union)))
|
|
return 'fail'
|
|
else:
|
|
geo = geo_union
|
|
geo = geo.buffer(margin + abs(dia / 2))
|
|
if isinstance(geo, Polygon):
|
|
geo_obj.solid_geometry = geo.exterior
|
|
elif isinstance(geo, MultiPolygon):
|
|
solid_geo = []
|
|
for poly in geo:
|
|
solid_geo.append(poly.exterior)
|
|
geo_obj.solid_geometry = deepcopy(solid_geo)
|
|
geo_obj.options['cnctooldia'] = str(dia)
|
|
geo_obj.options['cutz'] = self.cutz_entry.get_value()
|
|
geo_obj.options['multidepth'] = self.mpass_cb.get_value()
|
|
geo_obj.options['depthperpass'] = self.maxdepth_entry.get_value()
|
|
|
|
outname = cutout_obj.options["name"] + "_cutout"
|
|
self.app.new_object('geometry', outname, geo_init)
|
|
|
|
def cutting_geo(self, pos):
|
|
offset = self.cutting_dia / 2 + self.cutting_gapsize / 2
|
|
|
|
# cutting area definition
|
|
orig_x = pos[0]
|
|
orig_y = pos[1]
|
|
xmin = orig_x - offset
|
|
ymin = orig_y - offset
|
|
xmax = orig_x + offset
|
|
ymax = orig_y + offset
|
|
|
|
cut_poly = box(xmin, ymin, xmax, ymax)
|
|
return cut_poly
|
|
|
|
# To be called after clicking on the plot.
|
|
def on_mouse_click_release(self, event):
|
|
|
|
if self.app.is_legacy is False:
|
|
event_pos = event.pos
|
|
event_is_dragging = event.is_dragging
|
|
right_button = 2
|
|
else:
|
|
event_pos = (event.xdata, event.ydata)
|
|
event_is_dragging = self.app.plotcanvas.is_dragging
|
|
right_button = 3
|
|
|
|
try:
|
|
x = float(event_pos[0])
|
|
y = float(event_pos[1])
|
|
except TypeError:
|
|
return
|
|
event_pos = (x, y)
|
|
|
|
# do paint single only for left mouse clicks
|
|
if event.button == 1:
|
|
self.app.inform.emit(_("Making manual bridge gap..."))
|
|
|
|
pos = self.app.plotcanvas.translate_coords(event_pos)
|
|
|
|
self.on_manual_cutout(click_pos=pos)
|
|
|
|
# if RMB then we exit
|
|
elif event.button == right_button and self.mouse_is_dragging is False:
|
|
if self.app.is_legacy is False:
|
|
self.app.plotcanvas.graph_event_disconnect('key_press', self.on_key_press)
|
|
self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
|
|
self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
|
|
else:
|
|
self.app.plotcanvas.graph_event_disconnect(self.kp)
|
|
self.app.plotcanvas.graph_event_disconnect(self.mm)
|
|
self.app.plotcanvas.graph_event_disconnect(self.mr)
|
|
|
|
self.app.kp = self.app.plotcanvas.graph_event_connect('key_press', self.app.ui.keyPressEvent)
|
|
self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press', self.app.on_mouse_click_over_plot)
|
|
self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
|
|
self.app.on_mouse_click_release_over_plot)
|
|
self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.app.on_mouse_move_over_plot)
|
|
|
|
# Remove any previous utility shape
|
|
self.app.geo_editor.tool_shape.clear(update=True)
|
|
self.app.geo_editor.tool_shape.enabled = False
|
|
|
|
def on_mouse_move(self, event):
|
|
|
|
self.app.on_mouse_move_over_plot(event=event)
|
|
|
|
if self.app.is_legacy is False:
|
|
event_pos = event.pos
|
|
event_is_dragging = event.is_dragging
|
|
right_button = 2
|
|
else:
|
|
event_pos = (event.xdata, event.ydata)
|
|
event_is_dragging = self.app.plotcanvas.is_dragging
|
|
right_button = 3
|
|
|
|
try:
|
|
x = float(event_pos[0])
|
|
y = float(event_pos[1])
|
|
except TypeError:
|
|
return
|
|
event_pos = (x, y)
|
|
|
|
pos = self.canvas.translate_coords(event_pos)
|
|
event.xdata, event.ydata = pos[0], pos[1]
|
|
|
|
if event_is_dragging is True:
|
|
self.mouse_is_dragging = True
|
|
else:
|
|
self.mouse_is_dragging = False
|
|
|
|
try:
|
|
x = float(event.xdata)
|
|
y = float(event.ydata)
|
|
except TypeError:
|
|
return
|
|
|
|
if self.app.grid_status() == True:
|
|
snap_x, snap_y = self.app.geo_editor.snap(x, y)
|
|
else:
|
|
snap_x, snap_y = x, y
|
|
|
|
self.x_pos, self.y_pos = snap_x, snap_y
|
|
|
|
# #################################################
|
|
# ### This section makes the cutting geo to #######
|
|
# ### rotate if it intersects the target geo ######
|
|
# #################################################
|
|
cut_geo = self.cutting_geo(pos=(snap_x, snap_y))
|
|
man_geo = self.man_cutout_obj.solid_geometry
|
|
|
|
def get_angle(geo):
|
|
line = cut_geo.intersection(geo)
|
|
|
|
try:
|
|
pt1_x = line.coords[0][0]
|
|
pt1_y = line.coords[0][1]
|
|
pt2_x = line.coords[1][0]
|
|
pt2_y = line.coords[1][1]
|
|
dx = pt1_x - pt2_x
|
|
dy = pt1_y - pt2_y
|
|
|
|
if dx == 0 or dy == 0:
|
|
angle = 0
|
|
else:
|
|
radian = math.atan(dx / dy)
|
|
angle = radian * 180 / math.pi
|
|
except Exception as e:
|
|
angle = 0
|
|
return angle
|
|
|
|
try:
|
|
rot_angle = 0
|
|
for geo_el in man_geo:
|
|
if isinstance(geo_el, Polygon):
|
|
work_geo = geo_el.exterior
|
|
if cut_geo.intersects(work_geo):
|
|
rot_angle = get_angle(geo=work_geo)
|
|
else:
|
|
rot_angle = 0
|
|
else:
|
|
rot_angle = 0
|
|
if cut_geo.intersects(geo_el):
|
|
rot_angle = get_angle(geo=geo_el)
|
|
if rot_angle != 0:
|
|
break
|
|
except TypeError:
|
|
if isinstance(man_geo, Polygon):
|
|
work_geo = man_geo.exterior
|
|
if cut_geo.intersects(work_geo):
|
|
rot_angle = get_angle(geo=work_geo)
|
|
else:
|
|
rot_angle = 0
|
|
else:
|
|
rot_angle = 0
|
|
if cut_geo.intersects(man_geo):
|
|
rot_angle = get_angle(geo=man_geo)
|
|
|
|
# rotate only if there is an angle to rotate to
|
|
if rot_angle != 0:
|
|
cut_geo = affinity.rotate(cut_geo, -rot_angle)
|
|
|
|
# Remove any previous utility shape
|
|
self.app.geo_editor.tool_shape.clear(update=True)
|
|
self.draw_utility_geometry(geo=cut_geo)
|
|
|
|
def draw_utility_geometry(self, geo):
|
|
self.app.geo_editor.tool_shape.add(
|
|
shape=geo,
|
|
color=(self.app.defaults["global_draw_color"] + '80'),
|
|
update=False,
|
|
layer=0,
|
|
tolerance=None)
|
|
self.app.geo_editor.tool_shape.redraw()
|
|
|
|
def on_key_press(self, event):
|
|
# events out of the self.app.collection view (it's about Project Tab) are of type int
|
|
if type(event) is int:
|
|
key = event
|
|
# events from the GUI are of type QKeyEvent
|
|
elif type(event) == QtGui.QKeyEvent:
|
|
key = event.key()
|
|
elif isinstance(event, mpl_key_event): # MatPlotLib key events are trickier to interpret than the rest
|
|
key = event.key
|
|
key = QtGui.QKeySequence(key)
|
|
|
|
# check for modifiers
|
|
key_string = key.toString().lower()
|
|
if '+' in key_string:
|
|
mod, __, key_text = key_string.rpartition('+')
|
|
if mod.lower() == 'ctrl':
|
|
modifiers = QtCore.Qt.ControlModifier
|
|
elif mod.lower() == 'alt':
|
|
modifiers = QtCore.Qt.AltModifier
|
|
elif mod.lower() == 'shift':
|
|
modifiers = QtCore.Qt.ShiftModifier
|
|
else:
|
|
modifiers = QtCore.Qt.NoModifier
|
|
key = QtGui.QKeySequence(key_text)
|
|
# events from Vispy are of type KeyEvent
|
|
else:
|
|
key = event.key
|
|
|
|
# Escape = Deselect All
|
|
if key == QtCore.Qt.Key_Escape or key == 'Escape':
|
|
if self.app.is_legacy is False:
|
|
self.app.plotcanvas.graph_event_disconnect('key_press', self.on_key_press)
|
|
self.app.plotcanvas.graph_event_disconnect('mouse_move', self.on_mouse_move)
|
|
self.app.plotcanvas.graph_event_disconnect('mouse_release', self.on_mouse_click_release)
|
|
else:
|
|
self.app.plotcanvas.graph_event_disconnect(self.kp)
|
|
self.app.plotcanvas.graph_event_disconnect(self.mm)
|
|
self.app.plotcanvas.graph_event_disconnect(self.mr)
|
|
|
|
self.app.kp = self.app.plotcanvas.graph_event_connect('key_press', self.app.ui.keyPressEvent)
|
|
self.app.mp = self.app.plotcanvas.graph_event_connect('mouse_press', self.app.on_mouse_click_over_plot)
|
|
self.app.mr = self.app.plotcanvas.graph_event_connect('mouse_release',
|
|
self.app.on_mouse_click_release_over_plot)
|
|
self.app.mm = self.app.plotcanvas.graph_event_connect('mouse_move', self.app.on_mouse_move_over_plot)
|
|
|
|
# Remove any previous utility shape
|
|
self.app.geo_editor.tool_shape.clear(update=True)
|
|
self.app.geo_editor.tool_shape.enabled = False
|
|
|
|
# Grid toggle
|
|
if key == QtCore.Qt.Key_G or key == 'G':
|
|
self.app.ui.grid_snap_btn.trigger()
|
|
|
|
# Jump to coords
|
|
if key == QtCore.Qt.Key_J or key == 'J':
|
|
l_x, l_y = self.app.on_jump_to()
|
|
self.app.geo_editor.tool_shape.clear(update=True)
|
|
geo = self.cutting_geo(pos=(l_x, l_y))
|
|
self.draw_utility_geometry(geo=geo)
|
|
|
|
def subtract_poly_from_geo(self, solid_geo, x0, y0, x1, y1):
|
|
"""
|
|
Subtract polygon made from points from the given object.
|
|
This only operates on the paths in the original geometry,
|
|
i.e. it converts polygons into paths.
|
|
|
|
:param x0: x coord for lower left vertice of the polygon.
|
|
:param y0: y coord for lower left vertice of the polygon.
|
|
:param x1: x coord for upper right vertice of the polygon.
|
|
:param y1: y coord for upper right vertice of the polygon.
|
|
|
|
:param solid_geo: Geometry from which to substract. If none, use the solid_geomety property of the object
|
|
:return: none
|
|
"""
|
|
points = [(x0, y0), (x1, y0), (x1, y1), (x0, y1)]
|
|
|
|
# pathonly should be allways True, otherwise polygons are not subtracted
|
|
flat_geometry = flatten(geometry=solid_geo)
|
|
|
|
log.debug("%d paths" % len(flat_geometry))
|
|
|
|
polygon = Polygon(points)
|
|
toolgeo = cascaded_union(polygon)
|
|
diffs = []
|
|
for target in flat_geometry:
|
|
if type(target) == LineString or type(target) == LinearRing:
|
|
diffs.append(target.difference(toolgeo))
|
|
else:
|
|
log.warning("Not implemented.")
|
|
|
|
return unary_union(diffs)
|
|
|
|
def reset_fields(self):
|
|
self.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
|
|
|
|
|
|
def flatten(geometry):
|
|
"""
|
|
Creates a list of non-iterable linear geometry objects.
|
|
Polygons are expanded into its exterior and interiors.
|
|
|
|
Results are placed in self.flat_geometry
|
|
|
|
:param geometry: Shapely type or list or list of list of such.
|
|
"""
|
|
flat_geo = []
|
|
try:
|
|
for geo in geometry:
|
|
if type(geo) == Polygon:
|
|
flat_geo.append(geo.exterior)
|
|
for subgeo in geo.interiors:
|
|
flat_geo.append(subgeo)
|
|
else:
|
|
flat_geo.append(geo)
|
|
except TypeError:
|
|
if type(geometry) == Polygon:
|
|
flat_geo.append(geometry.exterior)
|
|
for subgeo in geometry.interiors:
|
|
flat_geo.append(subgeo)
|
|
else:
|
|
flat_geo.append(geometry)
|
|
|
|
return flat_geo
|
|
|
|
|
|
def recursive_bounds(geometry):
|
|
"""
|
|
Returns coordinates of rectangular bounds
|
|
of geometry: (xmin, ymin, xmax, ymax).
|
|
"""
|
|
|
|
# now it can get bounds for nested lists of objects
|
|
|
|
def bounds_rec(obj):
|
|
try:
|
|
minx = Inf
|
|
miny = Inf
|
|
maxx = -Inf
|
|
maxy = -Inf
|
|
|
|
for k in obj:
|
|
minx_, miny_, maxx_, maxy_ = bounds_rec(k)
|
|
minx = min(minx, minx_)
|
|
miny = min(miny, miny_)
|
|
maxx = max(maxx, maxx_)
|
|
maxy = max(maxy, maxy_)
|
|
return minx, miny, maxx, maxy
|
|
except TypeError:
|
|
# it's a Shapely object, return it's bounds
|
|
return obj.bounds
|
|
|
|
return bounds_rec(geometry)
|