| 123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742 |
- from FlatCAMTool import FlatCAMTool
- from copy import copy,deepcopy
- from ObjectCollection import *
- from FlatCAMApp import *
- from PyQt5 import QtGui, QtCore, QtWidgets
- from GUIElements import IntEntry, RadioSet, LengthEntry
- from FlatCAMObj import FlatCAMGeometry, FlatCAMExcellon, FlatCAMGerber
- class ToolSolderPaste(FlatCAMTool):
- toolName = "Solder Paste Tool"
- def __init__(self, app):
- FlatCAMTool.__init__(self, app)
- ## 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
- obj_form_layout = QtWidgets.QFormLayout()
- self.layout.addLayout(obj_form_layout)
- ## Gerber Object to be used for solderpaste dispensing
- 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("Gerber: ")
- self.object_label.setToolTip(
- "Gerber Solder paste object. "
- )
- obj_form_layout.addRow(self.object_label, self.obj_combo)
- #### Tools ####
- self.tools_table_label = QtWidgets.QLabel('<b>Tools Table</b>')
- self.tools_table_label.setToolTip(
- "Tools pool from which the algorithm\n"
- "will pick the ones used for dispensing solder paste."
- )
- self.layout.addWidget(self.tools_table_label)
- self.tools_table = FCTable()
- self.layout.addWidget(self.tools_table)
- self.tools_table.setColumnCount(3)
- self.tools_table.setHorizontalHeaderLabels(['#', 'Diameter', ''])
- self.tools_table.setColumnHidden(2, True)
- self.tools_table.setSortingEnabled(False)
- # self.tools_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
- self.tools_table.horizontalHeaderItem(0).setToolTip(
- "This is the Tool Number.\n"
- "The solder dispensing will start with the tool with the biggest \n"
- "diameter, continuing until there are no more Nozzle tools.\n"
- "If there are no longer tools but there are still pads not covered\n "
- "with solder paste, the app will issue a warning message box."
- )
- self.tools_table.horizontalHeaderItem(1).setToolTip(
- "Nozzle tool Diameter. It's value (in current FlatCAM units)\n"
- "is the width of the solder paste dispensed.")
- self.empty_label = QtWidgets.QLabel('')
- self.layout.addWidget(self.empty_label)
- #### Add a new Tool ####
- hlay_tools = QtWidgets.QHBoxLayout()
- self.layout.addLayout(hlay_tools)
- self.addtool_entry_lbl = QtWidgets.QLabel('<b>Nozzle Dia:</b>')
- self.addtool_entry_lbl.setToolTip(
- "Diameter for the new Nozzle tool to add in the Tool Table"
- )
- self.addtool_entry = FCEntry()
- # hlay.addWidget(self.addtool_label)
- # hlay.addStretch()
- hlay_tools.addWidget(self.addtool_entry_lbl)
- hlay_tools.addWidget(self.addtool_entry)
- grid0 = QtWidgets.QGridLayout()
- self.layout.addLayout(grid0)
- self.addtool_btn = QtWidgets.QPushButton('Add')
- self.addtool_btn.setToolTip(
- "Add a new nozzle tool to the Tool Table\n"
- "with the diameter specified above."
- )
- self.deltool_btn = QtWidgets.QPushButton('Delete')
- self.deltool_btn.setToolTip(
- "Delete a selection of tools in the Tool Table\n"
- "by first selecting a row(s) in the Tool Table."
- )
- self.soldergeo_btn = QtWidgets.QPushButton("Generate Geo")
- self.soldergeo_btn.setToolTip(
- "Generate solder paste dispensing geometry."
- )
- grid0.addWidget(self.addtool_btn, 0, 0)
- # grid2.addWidget(self.copytool_btn, 0, 1)
- grid0.addWidget(self.deltool_btn, 0, 2)
- grid0.addWidget(self.soldergeo_btn, 2, 2)
- ## Form Layout
- geo_form_layout = QtWidgets.QFormLayout()
- self.layout.addLayout(geo_form_layout)
- ## Gerber Object to be used for solderpaste dispensing
- self.geo_obj_combo = QtWidgets.QComboBox()
- self.geo_obj_combo.setModel(self.app.collection)
- self.geo_obj_combo.setRootModelIndex(self.app.collection.index(2, 0, QtCore.QModelIndex()))
- self.geo_obj_combo.setCurrentIndex(1)
- self.geo_object_label = QtWidgets.QLabel("Geometry:")
- self.geo_object_label.setToolTip(
- "Geometry Solder paste object.\n"
- "In order to enable the GCode generation section,\n"
- "the name of the object has to end in:\n"
- "'_solderpaste' as a protection."
- )
- geo_form_layout.addRow(self.geo_object_label, self.geo_obj_combo)
- self.gcode_frame = QtWidgets.QFrame()
- self.gcode_frame.setContentsMargins(0, 0, 0, 0)
- self.layout.addWidget(self.gcode_frame)
- self.gcode_box = QtWidgets.QVBoxLayout()
- self.gcode_box.setContentsMargins(0, 0, 0, 0)
- self.gcode_frame.setLayout(self.gcode_box)
- ## Form Layout
- form_layout = QtWidgets.QFormLayout()
- self.gcode_box.addLayout(form_layout)
- # Z dispense start
- self.z_start_entry = FCEntry()
- self.z_start_label = QtWidgets.QLabel("Z Dispense Start:")
- self.z_start_label.setToolTip(
- "The size of the 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.z_start_label, self.z_start_entry)
- # Z dispense
- self.z_dispense_entry = FCEntry()
- self.z_dispense_label = QtWidgets.QLabel("Z Dispense:")
- self.z_dispense_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.z_dispense_label, self.z_dispense_entry)
- # Z dispense stop
- self.z_stop_entry = FCEntry()
- self.z_stop_label = QtWidgets.QLabel("Z Dispense Stop:")
- self.z_stop_label.setToolTip(
- "The size of the 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.z_stop_label, self.z_stop_entry)
- # Z travel
- self.z_travel_entry = FCEntry()
- self.z_travel_label = QtWidgets.QLabel("Z Travel:")
- self.z_travel_label.setToolTip(
- "The size of the 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.z_travel_label, self.z_travel_entry)
- # Feedrate X-Y
- self.frxy_entry = FCEntry()
- self.frxy_label = QtWidgets.QLabel("Feedrate X-Y:")
- self.frxy_label.setToolTip(
- "The size of the 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.frxy_label, self.frxy_entry)
- # Feedrate Z
- self.frz_entry = FCEntry()
- self.frz_label = QtWidgets.QLabel("Feedrate Z:")
- self.frz_label.setToolTip(
- "The size of the 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.frz_label, self.frz_entry)
- # Spindle Speed Forward
- self.speedfwd_entry = FCEntry()
- self.speedfwd_label = QtWidgets.QLabel("Spindle Speed FWD:")
- self.speedfwd_label.setToolTip(
- "The size of the 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.speedfwd_label, self.speedfwd_entry)
- # Dwell Forward
- self.dwellfwd_entry = FCEntry()
- self.dwellfwd_label = QtWidgets.QLabel("Dwell FWD:")
- self.dwellfwd_label.setToolTip(
- "The size of the 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.dwellfwd_label, self.dwellfwd_entry)
- # Spindle Speed Reverse
- self.speedrev_entry = FCEntry()
- self.speedrev_label = QtWidgets.QLabel("Spindle Speed REV:")
- self.speedrev_label.setToolTip(
- "The size of the 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.speedrev_label, self.speedrev_entry)
- # Dwell Reverse
- self.dwellrev_entry = FCEntry()
- self.dwellrev_label = QtWidgets.QLabel("Dwell REV:")
- self.dwellrev_label.setToolTip(
- "The size of the 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.dwellrev_label, self.dwellrev_entry)
- # Postprocessors
- pp_label = QtWidgets.QLabel('PostProcessors:')
- pp_label.setToolTip(
- "Files that control the GCoe generation."
- )
- self.pp_combo = FCComboBox()
- pp_items = [1, 2, 3, 4, 5]
- for it in pp_items:
- self.pp_combo.addItem(str(it))
- self.pp_combo.setStyleSheet('background-color: rgb(255,255,255)')
- form_layout.addRow(pp_label, self.pp_combo)
- ## Buttons
- hlay = QtWidgets.QHBoxLayout()
- self.gcode_box.addLayout(hlay)
- hlay.addStretch()
- self.solder_gcode = QtWidgets.QPushButton("Generate GCode")
- self.solder_gcode.setToolTip(
- "Generate GCode to dispense Solder Paste\n"
- "on PCB pads."
- )
- hlay.addWidget(self.solder_gcode)
- self.layout.addStretch()
- self.gcode_frame.setDisabled(True)
- self.tools = {}
- self.tooluid = 0
- ## Signals
- self.addtool_btn.clicked.connect(self.on_tool_add)
- self.deltool_btn.clicked.connect(self.on_tool_delete)
- self.soldergeo_btn.clicked.connect(self.on_create_geo)
- self.solder_gcode.clicked.connect(self.on_create_gcode)
- self.geo_obj_combo.currentIndexChanged.connect(self.on_geo_select)
- def run(self):
- self.app.report_usage("ToolSolderPaste()")
- FlatCAMTool.run(self)
- self.set_tool_ui()
- # if the splitter us hidden, display it
- if self.app.ui.splitter.sizes()[0] == 0:
- self.app.ui.splitter.setSizes([1, 1])
- self.build_ui()
- self.app.ui.notebook.setTabText(2, "SolderPaste Tool")
- def install(self, icon=None, separator=None, **kwargs):
- FlatCAMTool.install(self, icon, separator, shortcut='ALT+K', **kwargs)
- def set_tool_ui(self):
- # self.ncc_overlap_entry.set_value(self.app.defaults["tools_nccoverlap"])
- # self.ncc_margin_entry.set_value(self.app.defaults["tools_nccmargin"])
- # self.ncc_method_radio.set_value(self.app.defaults["tools_nccmethod"])
- # self.ncc_connect_cb.set_value(self.app.defaults["tools_nccconnect"])
- # self.ncc_contour_cb.set_value(self.app.defaults["tools_ncccontour"])
- # self.ncc_rest_cb.set_value(self.app.defaults["tools_nccrest"])
- self.tools_table.setupContextMenu()
- self.tools_table.addContextMenu(
- "Add", lambda: self.on_tool_add(dia=None, muted=None), icon=QtGui.QIcon("share/plus16.png"))
- self.tools_table.addContextMenu(
- "Delete", lambda:
- self.on_tool_delete(rows_to_delete=None, all=None), icon=QtGui.QIcon("share/delete32.png"))
- try:
- dias = [float(eval(dia)) for dia in self.app.defaults["tools_solderpaste_tools"].split(",")]
- except:
- log.error("At least one Nozzle tool diameter needed. "
- "Verify in Edit -> Preferences -> TOOLS -> Solder Paste Tools.")
- return
- self.tooluid = 0
- self.tools.clear()
- for tool_dia in dias:
- self.tooluid += 1
- self.tools.update({
- int(self.tooluid): {
- 'tooldia': float('%.4f' % tool_dia),
- 'solid_geometry': []
- }
- })
- self.name = ""
- self.obj = None
- self.units = self.app.general_options_form.general_app_group.units_radio.get_value().upper()
- self.reset_fields()
- def build_ui(self):
- self.ui_disconnect()
- # updated units
- self.units = self.app.general_options_form.general_app_group.units_radio.get_value().upper()
- if self.units == "IN":
- self.addtool_entry.set_value(0.039)
- else:
- self.addtool_entry.set_value(1)
- sorted_tools = []
- for k, v in self.tools.items():
- sorted_tools.append(float('%.4f' % float(v['tooldia'])))
- sorted_tools.sort(reverse=True)
- n = len(sorted_tools)
- self.tools_table.setRowCount(n)
- tool_id = 0
- for tool_sorted in sorted_tools:
- for tooluid_key, tooluid_value in self.tools.items():
- if float('%.4f' % tooluid_value['tooldia']) == tool_sorted:
- tool_id += 1
- id = QtWidgets.QTableWidgetItem('%d' % int(tool_id))
- id.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
- row_no = tool_id - 1
- self.tools_table.setItem(row_no, 0, id) # Tool name/id
- # Make sure that the drill diameter when in MM is with no more than 2 decimals
- # There are no drill bits in MM with more than 3 decimals diameter
- # For INCH the decimals should be no more than 3. There are no drills under 10mils
- if self.units == 'MM':
- dia = QtWidgets.QTableWidgetItem('%.2f' % tooluid_value['tooldia'])
- else:
- dia = QtWidgets.QTableWidgetItem('%.3f' % tooluid_value['tooldia'])
- dia.setFlags(QtCore.Qt.ItemIsEnabled)
- tool_uid_item = QtWidgets.QTableWidgetItem(str(int(tooluid_key)))
- self.tools_table.setItem(row_no, 1, dia) # Diameter
- self.tools_table.setItem(row_no, 2, tool_uid_item) # Tool unique ID
- # make the diameter column editable
- for row in range(tool_id):
- self.tools_table.item(row, 1).setFlags(
- QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
- # all the tools are selected by default
- self.tools_table.selectColumn(0)
- #
- self.tools_table.resizeColumnsToContents()
- self.tools_table.resizeRowsToContents()
- vertical_header = self.tools_table.verticalHeader()
- vertical_header.hide()
- self.tools_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
- horizontal_header = self.tools_table.horizontalHeader()
- horizontal_header.setMinimumSectionSize(10)
- horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
- horizontal_header.resizeSection(0, 20)
- horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
- # self.tools_table.setSortingEnabled(True)
- # sort by tool diameter
- # self.tools_table.sortItems(1)
- self.tools_table.setMinimumHeight(self.tools_table.getHeight())
- self.tools_table.setMaximumHeight(self.tools_table.getHeight())
- self.ui_connect()
- def ui_connect(self):
- self.tools_table.itemChanged.connect(self.on_tool_edit)
- def ui_disconnect(self):
- try:
- # if connected, disconnect the signal from the slot on item_changed as it creates issues
- self.tools_table.itemChanged.disconnect(self.on_tool_edit)
- except:
- pass
- def on_tool_add(self, dia=None, muted=None):
- self.ui_disconnect()
- if dia:
- tool_dia = dia
- else:
- try:
- tool_dia = float(self.addtool_entry.get_value())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- tool_dia = float(self.addtool_entry.get_value().replace(',', '.'))
- except ValueError:
- self.app.inform.emit("[ERROR_NOTCL]Wrong value format entered, "
- "use a number.")
- return
- if tool_dia is None:
- self.build_ui()
- self.app.inform.emit("[WARNING_NOTCL] Please enter a tool diameter to add, in Float format.")
- return
- if tool_dia == 0:
- self.app.inform.emit("[WARNING_NOTCL] Please enter a tool diameter with non-zero value, in Float format.")
- return
- # construct a list of all 'tooluid' in the self.tools
- tool_uid_list = []
- for tooluid_key in self.tools:
- tool_uid_item = int(tooluid_key)
- tool_uid_list.append(tool_uid_item)
- # find maximum from the temp_uid, add 1 and this is the new 'tooluid'
- if not tool_uid_list:
- max_uid = 0
- else:
- max_uid = max(tool_uid_list)
- self.tooluid = int(max_uid + 1)
- tool_dias = []
- for k, v in self.tools.items():
- for tool_v in v.keys():
- if tool_v == 'tooldia':
- tool_dias.append(float('%.4f' % v[tool_v]))
- if float('%.4f' % tool_dia) in tool_dias:
- if muted is None:
- self.app.inform.emit("[WARNING_NOTCL]Adding Nozzle tool cancelled. Tool already in Tool Table.")
- self.tools_table.itemChanged.connect(self.on_tool_edit)
- return
- else:
- if muted is None:
- self.app.inform.emit("[success] New Nozzle tool added to Tool Table.")
- self.tools.update({
- int(self.tooluid): {
- 'tooldia': float('%.4f' % tool_dia),
- 'solid_geometry': []
- }
- })
- self.build_ui()
- def on_tool_edit(self):
- self.ui_disconnect()
- tool_dias = []
- for k, v in self.tools.items():
- for tool_v in v.keys():
- if tool_v == 'tooldia':
- tool_dias.append(float('%.4f' % v[tool_v]))
- for row in range(self.tools_table.rowCount()):
- try:
- new_tool_dia = float(self.tools_table.item(row, 1).text())
- except ValueError:
- # try to convert comma to decimal point. if it's still not working error message and return
- try:
- new_tool_dia = float(self.tools_table.item(row, 1).text().replace(',', '.'))
- except ValueError:
- self.app.inform.emit("[ERROR_NOTCL]Wrong value format entered, "
- "use a number.")
- return
- tooluid = int(self.tools_table.item(row, 2).text())
- # identify the tool that was edited and get it's tooluid
- if new_tool_dia not in tool_dias:
- self.tools[tooluid]['tooldia'] = new_tool_dia
- self.app.inform.emit("[success] Nozzle tool from Tool Table was edited.")
- self.build_ui()
- return
- else:
- # identify the old tool_dia and restore the text in tool table
- for k, v in self.tools.items():
- if k == tooluid:
- old_tool_dia = v['tooldia']
- break
- restore_dia_item = self.tools_table.item(row, 1)
- restore_dia_item.setText(str(old_tool_dia))
- self.app.inform.emit("[WARNING_NOTCL] Edit cancelled. New diameter value is already in the Tool Table.")
- self.build_ui()
- def on_tool_delete(self, rows_to_delete=None, all=None):
- self.ui_disconnect()
- deleted_tools_list = []
- if all:
- self.tools.clear()
- self.build_ui()
- return
- if rows_to_delete:
- try:
- for row in rows_to_delete:
- tooluid_del = int(self.tools_table.item(row, 2).text())
- deleted_tools_list.append(tooluid_del)
- except TypeError:
- deleted_tools_list.append(rows_to_delete)
- for t in deleted_tools_list:
- self.tools.pop(t, None)
- self.build_ui()
- return
- try:
- if self.tools_table.selectedItems():
- for row_sel in self.tools_table.selectedItems():
- row = row_sel.row()
- if row < 0:
- continue
- tooluid_del = int(self.tools_table.item(row, 2).text())
- deleted_tools_list.append(tooluid_del)
- for t in deleted_tools_list:
- self.tools.pop(t, None)
- except AttributeError:
- self.app.inform.emit("[WARNING_NOTCL]Delete failed. Select a Nozzle tool to delete.")
- return
- except Exception as e:
- log.debug(str(e))
- self.app.inform.emit("[success] Nozzle tool(s) deleted from Tool Table.")
- self.build_ui()
- def on_geo_select(self):
- if self.geo_obj_combo.currentText().rpartition('_')[2] == 'solderpaste':
- self.gcode_frame.setDisabled(False)
- else:
- self.gcode_frame.setDisabled(True)
- @staticmethod
- def distance(pt1, pt2):
- return sqrt((pt1[0] - pt2[0]) ** 2 + (pt1[1] - pt2[1]) ** 2)
- def on_create_geo(self):
- proc = self.app.proc_container.new("Creating Solder Paste dispensing geometry.")
- name = self.obj_combo.currentText()
- obj = self.app.collection.get_by_name(name)
- if type(obj.solid_geometry) is not list:
- obj.solid_geometry = [obj.solid_geometry]
- # Sort tools in descending order
- sorted_tools = []
- for k, v in self.tools.items():
- sorted_tools.append(float('%.4f' % float(v['tooldia'])))
- sorted_tools.sort(reverse=True)
- def geo_init(geo_obj, app_obj):
- geo_obj.solid_geometry = []
- geo_obj.tools = {}
- geo_obj.multigeo = True
- geo_obj.multitool = True
- geo_obj.tools = {}
- def solder_line(p, offset):
- xmin, ymin, xmax, ymax = p.bounds
- min = [xmin, ymin]
- max = [xmax, ymax]
- min_r = [xmin, ymax]
- max_r = [xmax, ymin]
- diagonal_1 = LineString([min, max])
- diagonal_2 = LineString([min_r, max_r])
- round_diag_1 = round(diagonal_1.intersection(p).length, 4)
- round_diag_2 = round(diagonal_2.intersection(p).length, 4)
- if round_diag_1 == round_diag_2:
- l = distance((xmin, ymin), (xmax, ymin))
- h = distance((xmin, ymin), (xmin, ymax))
- if offset >= l /2 or offset >= h / 2:
- return "fail"
- if l > h:
- h_half = h / 2
- start = [xmin, (ymin + h_half)]
- stop = [(xmin + l), (ymin + h_half)]
- else:
- l_half = l / 2
- start = [(xmin + l_half), ymin]
- stop = [(xmin + l_half), (ymin + h)]
- geo = LineString([start, stop])
- elif round_diag_1 > round_diag_2:
- geo = diagonal_1.intersection(p)
- else:
- geo = diagonal_2.intersection(p)
- offseted_poly = p.buffer(-offset)
- geo = geo.intersection(offseted_poly)
- return geo
- work_geo = obj.solid_geometry
- rest_geo = []
- tooluid = 1
- for tool in sorted_tools:
- offset = tool / 2
- for uid, v in self.tools.items():
- if float('%.4f' % float(v['tooldia'])) == tool:
- tooluid = int(uid)
- break
- for g in work_geo:
- if type(g) == MultiPolygon:
- for poly in g:
- geom = solder_line(poly, offset=offset)
- if geom != 'fail':
- try:
- geo_obj.tools[tooluid]['solid_geometry'].append(geom)
- except KeyError:
- geo_obj.tools[tooluid] = {}
- geo_obj.tools[tooluid]['solid_geometry'] = []
- geo_obj.tools[tooluid]['solid_geometry'].append(geom)
- geo_obj.tools[tooluid]['tooldia'] = tool
- geo_obj.tools[tooluid]['offset'] = 'Path'
- geo_obj.tools[tooluid]['offset_value'] = 0.0
- geo_obj.tools[tooluid]['type'] = ' '
- geo_obj.tools[tooluid]['tool_type'] = ' '
- geo_obj.tools[tooluid]['data'] = {}
- else:
- rest_geo.append(poly)
- elif type(g) == Polygon:
- geom = solder_line(g, offset=offset)
- if geom != 'fail':
- try:
- geo_obj.tools[tooluid]['solid_geometry'].append(geom)
- except KeyError:
- geo_obj.tools[tooluid] = {}
- geo_obj.tools[tooluid]['solid_geometry'] = []
- geo_obj.tools[tooluid]['solid_geometry'].append(geom)
- geo_obj.tools[tooluid]['tooldia'] = tool
- geo_obj.tools[tooluid]['offset'] = 'Path'
- geo_obj.tools[tooluid]['offset_value'] = 0.0
- geo_obj.tools[tooluid]['type'] = ' '
- geo_obj.tools[tooluid]['tool_type'] = ' '
- geo_obj.tools[tooluid]['data'] = {}
- else:
- rest_geo.append(g)
- work_geo = rest_geo
- if not work_geo:
- app_obj.inform.emit("[success] Solder Paste geometry generated successfully...")
- return
- # if we still have geometry not processed at the end of the tools then we failed
- # some or all the pads are not covered with solder paste
- if rest_geo:
- app_obj.inform.emit("[WARNING_NOTCL] Some or all pads have no solder "
- "due of inadequate nozzle diameters...")
- return 'fail'
- def job_thread(app_obj):
- try:
- app_obj.new_object("geometry", name + "_solderpaste", geo_init)
- except Exception as e:
- proc.done()
- traceback.print_stack()
- return
- proc.done()
- self.app.inform.emit("Generating Solder Paste dispensing geometry...")
- # Promise object with the new name
- self.app.collection.promise(name)
- # Background
- self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
- # self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
- def on_create_gcode(self):
- name = self.obj_combo.currentText()
- def geo_init(geo_obj, app_obj):
- pass
- # self.app.new_object("geometry", name + "_cutout", geo_init)
- # self.app.inform.emit("[success] Rectangular CutOut operation finished.")
- # self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
- def reset_fields(self):
- self.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
|