| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972 |
- from FlatCAMTool import FlatCAMTool
- from ObjectCollection import *
- from FlatCAMApp import *
- from shapely.geometry import box
- import gettext
- import FlatCAMTranslation as fcTranslate
- import builtins
- fcTranslate.apply_language('strings')
- if '_' not in builtins.__dict__:
- _ = gettext.gettext
- class CutOut(FlatCAMTool):
- toolName = _("Cutout PCB")
- gapFinished = pyqtSignal()
- def __init__(self, app):
- FlatCAMTool.__init__(self, app)
- self.app = app
- self.canvas = app.plotcanvas
- # 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
- form_layout = QtWidgets.QFormLayout()
- self.layout.addLayout(form_layout)
- # 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("share/flatcam_icon16.png"))
- # self.type_obj_combo.setItemIcon(1, QtGui.QIcon("share/drill16.png"))
- self.type_obj_combo.setItemIcon(2, QtGui.QIcon("share/geometry16.png"))
- self.type_obj_combo_label = QtWidgets.QLabel(_("Obj 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.setFixedWidth(60)
- form_layout.addRow(self.type_obj_combo_label, self.type_obj_combo)
- # 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)
- self.object_label = QtWidgets.QLabel(_("Object:"))
- self.object_label.setToolTip(
- _("Object to be cutout. ")
- )
- form_layout.addRow(self.object_label, self.obj_combo)
- # Tool Diameter
- self.dia = FCEntry()
- self.dia_label = QtWidgets.QLabel(_("Tool Dia:"))
- self.dia_label.setToolTip(
- _("Diameter of the tool used to cutout\n"
- "the PCB shape out of the surrounding material.")
- )
- form_layout.addRow(self.dia_label, self.dia)
- # Margin
- self.margin = FCEntry()
- self.margin_label = QtWidgets.QLabel(_("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")
- )
- form_layout.addRow(self.margin_label, self.margin)
- # Gapsize
- self.gapsize = FCEntry()
- self.gapsize_label = QtWidgets.QLabel(_("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).")
- )
- form_layout.addRow(self.gapsize_label, self.gapsize)
- # 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()
- self.convex_box_label = QtWidgets.QLabel(_("Convex Sh.:"))
- self.convex_box_label.setToolTip(
- _("Create a convex shape surrounding the entire PCB.\n"
- "Used only if the source object type is Gerber.")
- )
- form_layout.addRow(self.convex_box_label, self.convex_box)
- # 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(_('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"
- "- 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.setFixedWidth(60)
- self.gaps = FCComboBox()
- gaps_items = ['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
- hlay = QtWidgets.QHBoxLayout()
- self.layout.addLayout(hlay)
- title_ff_label = QtWidgets.QLabel("<b>%s</b>" % _('FreeForm:'))
- title_ff_label.setToolTip(
- _("The cutout shape can be of ny shape.\n"
- "Useful when the PCB has a non-rectangular shape.")
- )
- hlay.addWidget(title_ff_label)
- hlay.addStretch()
- self.ff_cutout_object_btn = QtWidgets.QPushButton(_("Generate Geo"))
- 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.")
- )
- hlay.addWidget(self.ff_cutout_object_btn)
- hlay2 = QtWidgets.QHBoxLayout()
- self.layout.addLayout(hlay2)
- title_rct_label = QtWidgets.QLabel("<b>%s</b>" % _('Rectangular:'))
- title_rct_label.setToolTip(
- _("The resulting cutout shape is\n"
- "always a rectangle shape and it will be\n"
- "the bounding box of the Object.")
- )
- hlay2.addWidget(title_rct_label)
- hlay2.addStretch()
- self.rect_cutout_object_btn = QtWidgets.QPushButton(_("Generate Geo"))
- 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.")
- )
- hlay2.addWidget(self.rect_cutout_object_btn)
- # 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(_("Geo Obj:"))
- self.man_object_label.setToolTip(
- _("Geometry object used to create the manual cutout.")
- )
- self.man_object_label.setFixedWidth(60)
- # e_lab_0 = QtWidgets.QLabel('')
- form_layout_3.addRow(self.man_object_label, self.man_object_combo)
- # form_layout_3.addRow(e_lab_0)
- hlay3 = QtWidgets.QHBoxLayout()
- self.layout.addLayout(hlay3)
- self.man_geo_label = QtWidgets.QLabel(_("Manual Geo:"))
- self.man_geo_label.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.")
- )
- hlay3.addWidget(self.man_geo_label)
- hlay3.addStretch()
- self.man_geo_creation_btn = QtWidgets.QPushButton(_("Generate Geo"))
- 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.")
- )
- hlay3.addWidget(self.man_geo_creation_btn)
- hlay4 = QtWidgets.QHBoxLayout()
- self.layout.addLayout(hlay4)
- self.man_bridge_gaps_label = QtWidgets.QLabel(_("Manual Add Bridge Gaps:"))
- self.man_bridge_gaps_label.setToolTip(
- _("Use the left mouse button (LMB) click\n"
- "to create a bridge gap to separate the PCB from\n"
- "the surrounding material.")
- )
- hlay4.addWidget(self.man_bridge_gaps_label)
- hlay4.addStretch()
- self.man_gaps_creation_btn = QtWidgets.QPushButton(_("Generate Gap"))
- 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.")
- )
- hlay4.addWidget(self.man_gaps_creation_btn)
- self.layout.addStretch()
- 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 = []
- # 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)
- 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:
- 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+U', **kwargs)
- def set_tool_ui(self):
- self.reset_fields()
- self.dia.set_value(float(self.app.defaults["tools_cutouttooldia"]))
- self.margin.set_value(float(self.app.defaults["tools_cutoutmargin"]))
- 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'])
- self.gapFinished.connect(self.on_gap_finished)
- 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:
- self.app.inform.emit(_("[ERROR_NOTCL] Could not retrieve object: %s") % name)
- return "Could not retrieve object: %s" % name
- if cutout_obj is None:
- self.app.inform.emit(_("[ERROR_NOTCL] There is no object selected for Cutout.\nSelect one and try again."))
- return
- try:
- dia = float(self.dia.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- dia = float(self.dia.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Tool diameter value is missing or wrong format. "
- "Add it and retry."))
- return
- if 0 in {dia}:
- self.app.inform.emit(_("[WARNING_NOTCL] 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:
- margin = float(self.margin.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- margin = float(self.margin.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Margin value is missing or wrong format. "
- "Add it and retry."))
- return
- try:
- gapsize = float(self.gapsize.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- gapsize = float(self.gapsize.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Gap size value is missing or wrong format. "
- "Add it and retry."))
- return
- try:
- gaps = self.gaps.get_value()
- except TypeError:
- self.app.inform.emit(_("[WARNING_NOTCL] Number of gaps value is missing. Add it and retry."))
- return
- if gaps not in ['LR', 'TB', '2LR', '2TB', '4', '8']:
- self.app.inform.emit(_("[WARNING_NOTCL] Gaps value can be only one of: '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]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
- # try:
- # __ = iter(object_geo)
- # except TypeError:
- # object_geo = [object_geo]
- object_geo = unary_union(object_geo)
- # for geo in object_geo:
- if isinstance(cutout_obj, FlatCAMGerber):
- geo = object_geo.buffer(margin + abs(dia / 2))
- geo = geo.exterior
- else:
- geo = object_geo
- # Get min and max data for each object as we just cut rectangles across X or Y
- xmin, ymin, xmax, ymax = recursive_bounds(geo)
- px = 0.5 * (xmin + xmax) + margin
- py = 0.5 * (ymin + ymax) + margin
- lenx = (xmax - xmin) + (margin * 2)
- leny = (ymax - ymin) + (margin * 2)
- if gaps == '8' or gaps == '2LR':
- geo = self.subtract_poly_from_geo(geo,
- xmin - gapsize, # botleft_x
- py - gapsize + leny / 4, # botleft_y
- xmax + gapsize, # topright_x
- py + gapsize + leny / 4) # topright_y
- geo = self.subtract_poly_from_geo(geo,
- xmin - gapsize,
- py - gapsize - leny / 4,
- xmax + gapsize,
- py + gapsize - leny / 4)
- if gaps == '8' or gaps == '2TB':
- geo = self.subtract_poly_from_geo(geo,
- px - gapsize + lenx / 4,
- ymin - gapsize,
- px + gapsize + lenx / 4,
- ymax + gapsize)
- geo = self.subtract_poly_from_geo(geo,
- px - gapsize - lenx / 4,
- ymin - gapsize,
- px + gapsize - lenx / 4,
- ymax + gapsize)
- if gaps == '4' or gaps == 'LR':
- geo = self.subtract_poly_from_geo(geo,
- xmin - gapsize,
- py - gapsize,
- xmax + gapsize,
- py + gapsize)
- if gaps == '4' or gaps == 'TB':
- geo = self.subtract_poly_from_geo(geo,
- px - gapsize,
- ymin - gapsize,
- px + gapsize,
- ymax + gapsize)
- try:
- for g in geo:
- solid_geo.append(g)
- except TypeError:
- solid_geo.append(geo)
- 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)
- outname = cutout_obj.options["name"] + "_cutout"
- self.app.new_object('geometry', outname, geo_init)
- cutout_obj.plot()
- self.app.inform.emit(_("[success] 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:
- self.app.inform.emit(_("[ERROR_NOTCL] Could not retrieve object: %s") % name)
- return "Could not retrieve object: %s" % name
- if cutout_obj is None:
- self.app.inform.emit(_("[ERROR_NOTCL] Object not found: %s") % cutout_obj)
- try:
- dia = float(self.dia.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- dia = float(self.dia.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Tool diameter value is missing or wrong format. "
- "Add it and retry."))
- return
- if 0 in {dia}:
- self.app.inform.emit(_("[ERROR_NOTCL] 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:
- margin = float(self.margin.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- margin = float(self.margin.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Margin value is missing or wrong format. "
- "Add it and retry."))
- return
- try:
- gapsize = float(self.gapsize.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- gapsize = float(self.gapsize.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Gap size value is missing or wrong format. "
- "Add it and retry."))
- return
- try:
- gaps = self.gaps.get_value()
- except TypeError:
- self.app.inform.emit(_("[WARNING_NOTCL] Number of gaps value is missing. Add it and retry."))
- return
- if gaps not in ['LR', 'TB', '2LR', '2TB', '4', '8']:
- self.app.inform.emit(_("[WARNING_NOTCL] Gaps value can be only one of: '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]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
- try:
- __ = iter(object_geo)
- except TypeError:
- object_geo = [object_geo]
- 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))
- px = 0.5 * (xmin + xmax) + margin
- py = 0.5 * (ymin + ymax) + margin
- lenx = (xmax - xmin) + (margin * 2)
- leny = (ymax - ymin) + (margin * 2)
- if gaps == '8' or gaps == '2LR':
- geo = self.subtract_poly_from_geo(geo,
- xmin - gapsize, # botleft_x
- py - gapsize + leny / 4, # botleft_y
- xmax + gapsize, # topright_x
- py + gapsize + leny / 4) # topright_y
- geo = self.subtract_poly_from_geo(geo,
- xmin - gapsize,
- py - gapsize - leny / 4,
- xmax + gapsize,
- py + gapsize - leny / 4)
- if gaps == '8' or gaps == '2TB':
- geo = self.subtract_poly_from_geo(geo,
- px - gapsize + lenx / 4,
- ymin - gapsize,
- px + gapsize + lenx / 4,
- ymax + gapsize)
- geo = self.subtract_poly_from_geo(geo,
- px - gapsize - lenx / 4,
- ymin - gapsize,
- px + gapsize - lenx / 4,
- ymax + gapsize)
- if gaps == '4' or gaps == 'LR':
- geo = self.subtract_poly_from_geo(geo,
- xmin - gapsize,
- py - gapsize,
- xmax + gapsize,
- py + gapsize)
- if gaps == '4' or gaps == 'TB':
- geo = self.subtract_poly_from_geo(geo,
- px - gapsize,
- ymin - gapsize,
- px + gapsize,
- ymax + gapsize)
- try:
- for g in geo:
- solid_geo.append(g)
- except TypeError:
- solid_geo.append(geo)
- geo_obj.solid_geometry = deepcopy(solid_geo)
- geo_obj.options['cnctooldia'] = str(dia)
- outname = cutout_obj.options["name"] + "_cutout"
- self.app.new_object('geometry', outname, geo_init)
- # cutout_obj.plot()
- self.app.inform.emit(_("[success] 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
- try:
- self.cutting_dia = float(self.dia.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- self.cutting_dia = float(self.dia.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Tool diameter value is missing or wrong format. "
- "Add it and retry."))
- return
- if 0 in {self.cutting_dia}:
- self.app.inform.emit(_("[ERROR_NOTCL] 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:
- self.cutting_gapsize = float(self.gapsize.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- self.cutting_gapsize = float(self.gapsize.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Gap size value is missing or wrong format. "
- "Add it and retry."))
- return
- self.app.plotcanvas.vis_disconnect('key_press', self.app.ui.keyPressEvent)
- self.app.plotcanvas.vis_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
- self.app.plotcanvas.vis_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
- self.app.plotcanvas.vis_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
- self.app.plotcanvas.vis_connect('key_press', self.on_key_press)
- self.app.plotcanvas.vis_connect('mouse_move', self.on_mouse_move)
- self.app.plotcanvas.vis_connect('mouse_release', self.doit)
- # To be called after clicking on the plot.
- def doit(self, event):
- # do paint single only for left mouse clicks
- if event.button == 1:
- self.app.inform.emit(_("Making manual bridge gap..."))
- pos = self.app.plotcanvas.vispy_canvas.translate_coords(event.pos)
- self.on_manual_cutout(click_pos=pos)
- self.app.plotcanvas.vis_disconnect('key_press', self.on_key_press)
- self.app.plotcanvas.vis_disconnect('mouse_move', self.on_mouse_move)
- self.app.plotcanvas.vis_disconnect('mouse_release', self.doit)
- self.app.plotcanvas.vis_connect('key_press', self.app.ui.keyPressEvent)
- self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
- self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
- self.app.plotcanvas.vis_connect('mouse_move', self.app.on_mouse_move_over_plot)
- self.app.geo_editor.tool_shape.clear(update=True)
- self.app.geo_editor.tool_shape.enabled = False
- self.gapFinished.emit()
- def on_manual_cutout(self, click_pos):
- name = self.man_object_combo.currentText()
- # Get source object.
- try:
- cutout_obj = self.app.collection.get_by_name(str(name))
- except:
- self.app.inform.emit(_("[ERROR_NOTCL] Could not retrieve Geometry object: %s") % name)
- return "Could not retrieve object: %s" % name
- if cutout_obj is None:
- self.app.inform.emit(_("[ERROR_NOTCL] Geometry object for manual cutout not found: %s") % 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]))
- cutout_obj.subtract_polygon(cut_poly)
- cutout_obj.plot()
- self.app.inform.emit(_("[success] Added manual Bridge Gap."))
- self.app.should_we_save = True
- def on_gap_finished(self):
- # if CTRL key modifier is pressed then repeat the bridge gap cut
- key_modifier = QtWidgets.QApplication.keyboardModifiers()
- if key_modifier == Qt.ControlModifier:
- self.on_manual_gap_click()
- 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:
- self.app.inform.emit(_("[ERROR_NOTCL] Could not retrieve Gerber object: %s") % name)
- return "Could not retrieve object: %s" % name
- if cutout_obj is None:
- self.app.inform.emit(_("[ERROR_NOTCL] 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] The selected object has to be of Gerber type.\n"
- "Select a Gerber file and try again."))
- return
- try:
- dia = float(self.dia.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- dia = float(self.dia.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Tool diameter value is missing or wrong format. "
- "Add it and retry."))
- return
- if 0 in {dia}:
- self.app.inform.emit(_("[ERROR_NOTCL] 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:
- margin = float(self.margin.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- margin = float(self.margin.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit(_("[WARNING_NOTCL] Margin value is missing or wrong format. "
- "Add it and retry."))
- return
- 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))
- 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)
- 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
- def on_mouse_move(self, event):
- self.app.on_mouse_move_over_plot(event=event)
- pos = self.canvas.vispy_canvas.translate_coords(event.pos)
- event.xdata, event.ydata = pos[0], pos[1]
- try:
- x = float(event.xdata)
- y = float(event.ydata)
- except TypeError:
- return
- snap_x, snap_y = self.app.geo_editor.snap(x, y)
- geo = self.cutting_geo(pos=(snap_x, snap_y))
- # Remove any previous utility shape
- self.app.geo_editor.tool_shape.clear(update=True)
- self.draw_utility_geometry(geo=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()
- # events from Vispy are of type KeyEvent
- else:
- key = event.key
- # Escape = Deselect All
- if key == QtCore.Qt.Key_Escape or key == 'Escape':
- self.app.plotcanvas.vis_disconnect('key_press', self.on_key_press)
- self.app.plotcanvas.vis_disconnect('mouse_move', self.on_mouse_move)
- self.app.plotcanvas.vis_disconnect('mouse_release', self.doit)
- self.app.plotcanvas.vis_connect('key_press', self.app.ui.keyPressEvent)
- self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
- self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
- self.app.plotcanvas.vis_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 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)
|