ToolNonCopperClear.py 41 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798799800801802803804805806807808809810811812813814815816817818819820821822823824825826827828829830831832833834835836837838839840841842843844845846847848849850851852853854855856857858859860861862863864865866867868869870871872873874875876877878879880881882883884885886887888889890891892893894895896897898899900901902903904905906907908909910911912913914915916917918919920921922923924925926927928929930931932933934935936937938939940941942943944945946947948949950951952953954955956957958959960961962963964965966967968969970971972973974975976
  1. # ########################################################## ##
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # http://flatcam.org #
  4. # File Modified by: Marius Adrian Stanciu (c) #
  5. # Date: 3/10/2019 #
  6. # MIT Licence #
  7. # ########################################################## ##
  8. from FlatCAMTool import FlatCAMTool
  9. from copy import copy,deepcopy
  10. from ObjectCollection import *
  11. import time
  12. import gettext
  13. import FlatCAMTranslation as fcTranslate
  14. from shapely.geometry import base
  15. fcTranslate.apply_language('strings')
  16. import builtins
  17. if '_' not in builtins.__dict__:
  18. _ = gettext.gettext
  19. class NonCopperClear(FlatCAMTool, Gerber):
  20. toolName = _("Non-Copper Clearing")
  21. def __init__(self, app):
  22. self.app = app
  23. FlatCAMTool.__init__(self, app)
  24. Gerber.__init__(self, steps_per_circle=self.app.defaults["gerber_circle_steps"])
  25. self.tools_frame = QtWidgets.QFrame()
  26. self.tools_frame.setContentsMargins(0, 0, 0, 0)
  27. self.layout.addWidget(self.tools_frame)
  28. self.tools_box = QtWidgets.QVBoxLayout()
  29. self.tools_box.setContentsMargins(0, 0, 0, 0)
  30. self.tools_frame.setLayout(self.tools_box)
  31. # ## Title
  32. title_label = QtWidgets.QLabel("%s" % self.toolName)
  33. title_label.setStyleSheet("""
  34. QLabel
  35. {
  36. font-size: 16px;
  37. font-weight: bold;
  38. }
  39. """)
  40. self.tools_box.addWidget(title_label)
  41. # ## Form Layout
  42. form_layout = QtWidgets.QFormLayout()
  43. self.tools_box.addLayout(form_layout)
  44. # ## Object
  45. self.object_combo = QtWidgets.QComboBox()
  46. self.object_combo.setModel(self.app.collection)
  47. self.object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  48. self.object_combo.setCurrentIndex(1)
  49. self.object_label = QtWidgets.QLabel("Gerber:")
  50. self.object_label.setToolTip(
  51. _("Gerber object to be cleared of excess copper. ")
  52. )
  53. e_lab_0 = QtWidgets.QLabel('')
  54. form_layout.addRow(self.object_label, self.object_combo)
  55. form_layout.addRow(e_lab_0)
  56. #### Tools ## ##
  57. self.tools_table_label = QtWidgets.QLabel('<b>%s</b>' % _('Tools Table'))
  58. self.tools_table_label.setToolTip(
  59. _("Tools pool from which the algorithm\n"
  60. "will pick the ones used for copper clearing.")
  61. )
  62. self.tools_box.addWidget(self.tools_table_label)
  63. self.tools_table = FCTable()
  64. self.tools_box.addWidget(self.tools_table)
  65. self.tools_table.setColumnCount(4)
  66. self.tools_table.setHorizontalHeaderLabels(['#', _('Diameter'), 'TT', ''])
  67. self.tools_table.setColumnHidden(3, True)
  68. self.tools_table.setSortingEnabled(False)
  69. # self.tools_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
  70. self.tools_table.horizontalHeaderItem(0).setToolTip(
  71. _("This is the Tool Number.\n"
  72. "Non copper clearing will start with the tool with the biggest \n"
  73. "diameter, continuing until there are no more tools.\n"
  74. "Only tools that create NCC clearing geometry will still be present\n"
  75. "in the resulting geometry. This is because with some tools\n"
  76. "this function will not be able to create painting geometry.")
  77. )
  78. self.tools_table.horizontalHeaderItem(1).setToolTip(
  79. _("Tool Diameter. It's value (in current FlatCAM units) \n"
  80. "is the cut width into the material."))
  81. self.tools_table.horizontalHeaderItem(2).setToolTip(
  82. _("The Tool Type (TT) can be:<BR>"
  83. "- <B>Circular</B> with 1 ... 4 teeth -> it is informative only. Being circular, <BR>"
  84. "the cut width in material is exactly the tool diameter.<BR>"
  85. "- <B>Ball</B> -> informative only and make reference to the Ball type endmill.<BR>"
  86. "- <B>V-Shape</B> -> it will disable de Z-Cut parameter in the resulting geometry UI form "
  87. "and enable two additional UI form fields in the resulting geometry: V-Tip Dia and "
  88. "V-Tip Angle. Adjusting those two values will adjust the Z-Cut parameter such "
  89. "as the cut width into material will be equal with the value in the Tool Diameter "
  90. "column of this table.<BR>"
  91. "Choosing the <B>V-Shape</B> Tool Type automatically will select the Operation Type "
  92. "in the resulting geometry as Isolation."))
  93. self.empty_label = QtWidgets.QLabel('')
  94. self.tools_box.addWidget(self.empty_label)
  95. #### Add a new Tool ## ##
  96. hlay = QtWidgets.QHBoxLayout()
  97. self.tools_box.addLayout(hlay)
  98. self.addtool_entry_lbl = QtWidgets.QLabel('<b>%s:</b>' % _('Tool Dia'))
  99. self.addtool_entry_lbl.setToolTip(
  100. _("Diameter for the new tool to add in the Tool Table")
  101. )
  102. self.addtool_entry = FCEntry2()
  103. # hlay.addWidget(self.addtool_label)
  104. # hlay.addStretch()
  105. hlay.addWidget(self.addtool_entry_lbl)
  106. hlay.addWidget(self.addtool_entry)
  107. grid2 = QtWidgets.QGridLayout()
  108. self.tools_box.addLayout(grid2)
  109. self.addtool_btn = QtWidgets.QPushButton(_('Add'))
  110. self.addtool_btn.setToolTip(
  111. _("Add a new tool to the Tool Table\n"
  112. "with the diameter specified above.")
  113. )
  114. # self.copytool_btn = QtWidgets.QPushButton('Copy')
  115. # self.copytool_btn.setToolTip(
  116. # "Copy a selection of tools in the Tool Table\n"
  117. # "by first selecting a row in the Tool Table."
  118. # )
  119. self.deltool_btn = QtWidgets.QPushButton(_('Delete'))
  120. self.deltool_btn.setToolTip(
  121. _("Delete a selection of tools in the Tool Table\n"
  122. "by first selecting a row(s) in the Tool Table.")
  123. )
  124. grid2.addWidget(self.addtool_btn, 0, 0)
  125. # grid2.addWidget(self.copytool_btn, 0, 1)
  126. grid2.addWidget(self.deltool_btn, 0,2)
  127. self.empty_label_0 = QtWidgets.QLabel('')
  128. self.tools_box.addWidget(self.empty_label_0)
  129. grid3 = QtWidgets.QGridLayout()
  130. self.tools_box.addLayout(grid3)
  131. e_lab_1 = QtWidgets.QLabel('')
  132. grid3.addWidget(e_lab_1, 0, 0)
  133. nccoverlabel = QtWidgets.QLabel(_('Overlap Rate:'))
  134. nccoverlabel.setToolTip(
  135. _("How much (fraction) of the tool width to overlap each tool pass.\n"
  136. "Example:\n"
  137. "A value here of 0.25 means 25% from the tool diameter found above.\n\n"
  138. "Adjust the value starting with lower values\n"
  139. "and increasing it if areas that should be cleared are still \n"
  140. "not cleared.\n"
  141. "Lower values = faster processing, faster execution on PCB.\n"
  142. "Higher values = slow processing and slow execution on CNC\n"
  143. "due of too many paths.")
  144. )
  145. grid3.addWidget(nccoverlabel, 1, 0)
  146. self.ncc_overlap_entry = FCEntry()
  147. grid3.addWidget(self.ncc_overlap_entry, 1, 1)
  148. nccmarginlabel = QtWidgets.QLabel(_('Margin:'))
  149. nccmarginlabel.setToolTip(
  150. _("Bounding box margin.")
  151. )
  152. grid3.addWidget(nccmarginlabel, 2, 0)
  153. self.ncc_margin_entry = FCEntry()
  154. grid3.addWidget(self.ncc_margin_entry, 2, 1)
  155. # Method
  156. methodlabel = QtWidgets.QLabel(_('Method:'))
  157. methodlabel.setToolTip(
  158. _("Algorithm for non-copper clearing:<BR>"
  159. "<B>Standard</B>: Fixed step inwards.<BR>"
  160. "<B>Seed-based</B>: Outwards from seed.<BR>"
  161. "<B>Line-based</B>: Parallel lines.")
  162. )
  163. grid3.addWidget(methodlabel, 3, 0)
  164. self.ncc_method_radio = RadioSet([
  165. {"label": _("Standard"), "value": "standard"},
  166. {"label": _("Seed-based"), "value": "seed"},
  167. {"label": _("Straight lines"), "value": "lines"}
  168. ], orientation='vertical', stretch=False)
  169. grid3.addWidget(self.ncc_method_radio, 3, 1)
  170. # Connect lines
  171. pathconnectlabel = QtWidgets.QLabel(_("Connect:"))
  172. pathconnectlabel.setToolTip(
  173. _("Draw lines between resulting\n"
  174. "segments to minimize tool lifts.")
  175. )
  176. grid3.addWidget(pathconnectlabel, 4, 0)
  177. self.ncc_connect_cb = FCCheckBox()
  178. grid3.addWidget(self.ncc_connect_cb, 4, 1)
  179. contourlabel = QtWidgets.QLabel(_("Contour:"))
  180. contourlabel.setToolTip(
  181. _("Cut around the perimeter of the polygon\n"
  182. "to trim rough edges.")
  183. )
  184. grid3.addWidget(contourlabel, 5, 0)
  185. self.ncc_contour_cb = FCCheckBox()
  186. grid3.addWidget(self.ncc_contour_cb, 5, 1)
  187. restlabel = QtWidgets.QLabel(_("Rest M.:"))
  188. restlabel.setToolTip(
  189. _("If checked, use 'rest machining'.\n"
  190. "Basically it will clear copper outside PCB features,\n"
  191. "using the biggest tool and continue with the next tools,\n"
  192. "from bigger to smaller, to clear areas of copper that\n"
  193. "could not be cleared by previous tool, until there is\n"
  194. "no more copper to clear or there are no more tools.\n"
  195. "If not checked, use the standard algorithm.")
  196. )
  197. grid3.addWidget(restlabel, 6, 0)
  198. self.ncc_rest_cb = FCCheckBox()
  199. grid3.addWidget(self.ncc_rest_cb, 6, 1)
  200. self.generate_ncc_button = QtWidgets.QPushButton(_('Generate Geometry'))
  201. self.generate_ncc_button.setToolTip(
  202. _("Create the Geometry Object\n"
  203. "for non-copper routing.")
  204. )
  205. self.tools_box.addWidget(self.generate_ncc_button)
  206. self.units = ''
  207. self.ncc_tools = {}
  208. self.tooluid = 0
  209. # store here the default data for Geometry Data
  210. self.default_data = {}
  211. self.obj_name = ""
  212. self.ncc_obj = None
  213. self.tools_box.addStretch()
  214. self.addtool_btn.clicked.connect(self.on_tool_add)
  215. self.addtool_entry.returnPressed.connect(self.on_tool_add)
  216. self.deltool_btn.clicked.connect(self.on_tool_delete)
  217. self.generate_ncc_button.clicked.connect(self.on_ncc)
  218. def install(self, icon=None, separator=None, **kwargs):
  219. FlatCAMTool.install(self, icon, separator, shortcut='ALT+N', **kwargs)
  220. def run(self, toggle=True):
  221. self.app.report_usage("ToolNonCopperClear()")
  222. if toggle:
  223. # if the splitter is hidden, display it, else hide it but only if the current widget is the same
  224. if self.app.ui.splitter.sizes()[0] == 0:
  225. self.app.ui.splitter.setSizes([1, 1])
  226. else:
  227. try:
  228. if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
  229. self.app.ui.splitter.setSizes([0, 1])
  230. except AttributeError:
  231. pass
  232. else:
  233. if self.app.ui.splitter.sizes()[0] == 0:
  234. self.app.ui.splitter.setSizes([1, 1])
  235. FlatCAMTool.run(self)
  236. self.set_tool_ui()
  237. self.build_ui()
  238. self.app.ui.notebook.setTabText(2, _("NCC Tool"))
  239. def set_tool_ui(self):
  240. self.tools_frame.show()
  241. self.ncc_overlap_entry.set_value(self.app.defaults["tools_nccoverlap"])
  242. self.ncc_margin_entry.set_value(self.app.defaults["tools_nccmargin"])
  243. self.ncc_method_radio.set_value(self.app.defaults["tools_nccmethod"])
  244. self.ncc_connect_cb.set_value(self.app.defaults["tools_nccconnect"])
  245. self.ncc_contour_cb.set_value(self.app.defaults["tools_ncccontour"])
  246. self.ncc_rest_cb.set_value(self.app.defaults["tools_nccrest"])
  247. self.tools_table.setupContextMenu()
  248. self.tools_table.addContextMenu(
  249. "Add", lambda: self.on_tool_add(dia=None, muted=None), icon=QtGui.QIcon("share/plus16.png"))
  250. self.tools_table.addContextMenu(
  251. "Delete", lambda:
  252. self.on_tool_delete(rows_to_delete=None, all=None), icon=QtGui.QIcon("share/delete32.png"))
  253. # init the working variables
  254. self.default_data.clear()
  255. self.default_data.update({
  256. "name": '_ncc',
  257. "plot": self.app.defaults["geometry_plot"],
  258. "cutz": self.app.defaults["geometry_cutz"],
  259. "vtipdia": 0.1,
  260. "vtipangle": 30,
  261. "travelz": self.app.defaults["geometry_travelz"],
  262. "feedrate": self.app.defaults["geometry_feedrate"],
  263. "feedrate_z": self.app.defaults["geometry_feedrate_z"],
  264. "feedrate_rapid": self.app.defaults["geometry_feedrate_rapid"],
  265. "dwell": self.app.defaults["geometry_dwell"],
  266. "dwelltime": self.app.defaults["geometry_dwelltime"],
  267. "multidepth": self.app.defaults["geometry_multidepth"],
  268. "ppname_g": self.app.defaults["geometry_ppname_g"],
  269. "depthperpass": self.app.defaults["geometry_depthperpass"],
  270. "extracut": self.app.defaults["geometry_extracut"],
  271. "toolchange": self.app.defaults["geometry_toolchange"],
  272. "toolchangez": self.app.defaults["geometry_toolchangez"],
  273. "endz": self.app.defaults["geometry_endz"],
  274. "spindlespeed": self.app.defaults["geometry_spindlespeed"],
  275. "toolchangexy": self.app.defaults["geometry_toolchangexy"],
  276. "startz": self.app.defaults["geometry_startz"],
  277. "tooldia": self.app.defaults["tools_painttooldia"],
  278. "paintmargin": self.app.defaults["tools_paintmargin"],
  279. "paintmethod": self.app.defaults["tools_paintmethod"],
  280. "selectmethod": self.app.defaults["tools_selectmethod"],
  281. "pathconnect": self.app.defaults["tools_pathconnect"],
  282. "paintcontour": self.app.defaults["tools_paintcontour"],
  283. "paintoverlap": self.app.defaults["tools_paintoverlap"],
  284. "nccoverlap": self.app.defaults["tools_nccoverlap"],
  285. "nccmargin": self.app.defaults["tools_nccmargin"],
  286. "nccmethod": self.app.defaults["tools_nccmethod"],
  287. "nccconnect": self.app.defaults["tools_nccconnect"],
  288. "ncccontour": self.app.defaults["tools_ncccontour"],
  289. "nccrest": self.app.defaults["tools_nccrest"]
  290. })
  291. try:
  292. dias = [float(eval(dia)) for dia in self.app.defaults["tools_ncctools"].split(",")]
  293. except:
  294. log.error("At least one tool diameter needed. Verify in Edit -> Preferences -> TOOLS -> NCC Tools.")
  295. return
  296. self.tooluid = 0
  297. self.ncc_tools.clear()
  298. for tool_dia in dias:
  299. self.tooluid += 1
  300. self.ncc_tools.update({
  301. int(self.tooluid): {
  302. 'tooldia': float('%.4f' % tool_dia),
  303. 'offset': 'Path',
  304. 'offset_value': 0.0,
  305. 'type': 'Iso',
  306. 'tool_type': 'V',
  307. 'data': dict(self.default_data),
  308. 'solid_geometry': []
  309. }
  310. })
  311. self.obj_name = ""
  312. self.ncc_obj = None
  313. self.tool_type_item_options = ["C1", "C2", "C3", "C4", "B", "V"]
  314. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  315. def build_ui(self):
  316. self.ui_disconnect()
  317. # updated units
  318. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  319. if self.units == "IN":
  320. self.addtool_entry.set_value(0.039)
  321. else:
  322. self.addtool_entry.set_value(1)
  323. sorted_tools = []
  324. for k, v in self.ncc_tools.items():
  325. sorted_tools.append(float('%.4f' % float(v['tooldia'])))
  326. sorted_tools.sort()
  327. n = len(sorted_tools)
  328. self.tools_table.setRowCount(n)
  329. tool_id = 0
  330. for tool_sorted in sorted_tools:
  331. for tooluid_key, tooluid_value in self.ncc_tools.items():
  332. if float('%.4f' % tooluid_value['tooldia']) == tool_sorted:
  333. tool_id += 1
  334. id = QtWidgets.QTableWidgetItem('%d' % int(tool_id))
  335. id.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  336. row_no = tool_id - 1
  337. self.tools_table.setItem(row_no, 0, id) # Tool name/id
  338. # Make sure that the drill diameter when in MM is with no more than 2 decimals
  339. # There are no drill bits in MM with more than 3 decimals diameter
  340. # For INCH the decimals should be no more than 3. There are no drills under 10mils
  341. if self.units == 'MM':
  342. dia = QtWidgets.QTableWidgetItem('%.2f' % tooluid_value['tooldia'])
  343. else:
  344. dia = QtWidgets.QTableWidgetItem('%.3f' % tooluid_value['tooldia'])
  345. dia.setFlags(QtCore.Qt.ItemIsEnabled)
  346. tool_type_item = QtWidgets.QComboBox()
  347. for item in self.tool_type_item_options:
  348. tool_type_item.addItem(item)
  349. tool_type_item.setStyleSheet('background-color: rgb(255,255,255)')
  350. idx = tool_type_item.findText(tooluid_value['tool_type'])
  351. tool_type_item.setCurrentIndex(idx)
  352. tool_uid_item = QtWidgets.QTableWidgetItem(str(int(tooluid_key)))
  353. self.tools_table.setItem(row_no, 1, dia) # Diameter
  354. self.tools_table.setCellWidget(row_no, 2, tool_type_item)
  355. # ## REMEMBER: THIS COLUMN IS HIDDEN IN OBJECTUI.PY # ##
  356. self.tools_table.setItem(row_no, 3, tool_uid_item) # Tool unique ID
  357. # make the diameter column editable
  358. for row in range(tool_id):
  359. self.tools_table.item(row, 1).setFlags(
  360. QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  361. # all the tools are selected by default
  362. self.tools_table.selectColumn(0)
  363. #
  364. self.tools_table.resizeColumnsToContents()
  365. self.tools_table.resizeRowsToContents()
  366. vertical_header = self.tools_table.verticalHeader()
  367. vertical_header.hide()
  368. self.tools_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  369. horizontal_header = self.tools_table.horizontalHeader()
  370. horizontal_header.setMinimumSectionSize(10)
  371. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  372. horizontal_header.resizeSection(0, 20)
  373. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
  374. # self.tools_table.setSortingEnabled(True)
  375. # sort by tool diameter
  376. # self.tools_table.sortItems(1)
  377. self.tools_table.setMinimumHeight(self.tools_table.getHeight())
  378. self.tools_table.setMaximumHeight(self.tools_table.getHeight())
  379. self.ui_connect()
  380. def ui_connect(self):
  381. self.tools_table.itemChanged.connect(self.on_tool_edit)
  382. def ui_disconnect(self):
  383. try:
  384. # if connected, disconnect the signal from the slot on item_changed as it creates issues
  385. self.tools_table.itemChanged.disconnect(self.on_tool_edit)
  386. except:
  387. pass
  388. def on_tool_add(self, dia=None, muted=None):
  389. self.ui_disconnect()
  390. if dia:
  391. tool_dia = dia
  392. else:
  393. try:
  394. tool_dia = float(self.addtool_entry.get_value())
  395. except ValueError:
  396. # try to convert comma to decimal point. if it's still not working error message and return
  397. try:
  398. tool_dia = float(self.addtool_entry.get_value().replace(',', '.'))
  399. except ValueError:
  400. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  401. "use a number."))
  402. return
  403. if tool_dia is None:
  404. self.build_ui()
  405. self.app.inform.emit(_("[WARNING_NOTCL] Please enter a tool diameter to add, in Float format."))
  406. return
  407. if tool_dia == 0:
  408. self.app.inform.emit(_("[WARNING_NOTCL] Please enter a tool diameter with non-zero value, in Float format."))
  409. return
  410. # construct a list of all 'tooluid' in the self.tools
  411. tool_uid_list = []
  412. for tooluid_key in self.ncc_tools:
  413. tool_uid_item = int(tooluid_key)
  414. tool_uid_list.append(tool_uid_item)
  415. # find maximum from the temp_uid, add 1 and this is the new 'tooluid'
  416. if not tool_uid_list:
  417. max_uid = 0
  418. else:
  419. max_uid = max(tool_uid_list)
  420. self.tooluid = int(max_uid + 1)
  421. tool_dias = []
  422. for k, v in self.ncc_tools.items():
  423. for tool_v in v.keys():
  424. if tool_v == 'tooldia':
  425. tool_dias.append(float('%.4f' % v[tool_v]))
  426. if float('%.4f' % tool_dia) in tool_dias:
  427. if muted is None:
  428. self.app.inform.emit(_("[WARNING_NOTCL] Adding tool cancelled. Tool already in Tool Table."))
  429. self.tools_table.itemChanged.connect(self.on_tool_edit)
  430. return
  431. else:
  432. if muted is None:
  433. self.app.inform.emit(_("[success] New tool added to Tool Table."))
  434. self.ncc_tools.update({
  435. int(self.tooluid): {
  436. 'tooldia': float('%.4f' % tool_dia),
  437. 'offset': 'Path',
  438. 'offset_value': 0.0,
  439. 'type': 'Iso',
  440. 'tool_type': 'V',
  441. 'data': dict(self.default_data),
  442. 'solid_geometry': []
  443. }
  444. })
  445. self.build_ui()
  446. def on_tool_edit(self):
  447. self.ui_disconnect()
  448. tool_dias = []
  449. for k, v in self.ncc_tools.items():
  450. for tool_v in v.keys():
  451. if tool_v == 'tooldia':
  452. tool_dias.append(float('%.4f' % v[tool_v]))
  453. for row in range(self.tools_table.rowCount()):
  454. try:
  455. new_tool_dia = float(self.tools_table.item(row, 1).text())
  456. except ValueError:
  457. # try to convert comma to decimal point. if it's still not working error message and return
  458. try:
  459. new_tool_dia = float(self.tools_table.item(row, 1).text().replace(',', '.'))
  460. except ValueError:
  461. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  462. "use a number."))
  463. return
  464. tooluid = int(self.tools_table.item(row, 3).text())
  465. # identify the tool that was edited and get it's tooluid
  466. if new_tool_dia not in tool_dias:
  467. self.ncc_tools[tooluid]['tooldia'] = new_tool_dia
  468. self.app.inform.emit(_("[success] Tool from Tool Table was edited."))
  469. self.build_ui()
  470. return
  471. else:
  472. # identify the old tool_dia and restore the text in tool table
  473. for k, v in self.ncc_tools.items():
  474. if k == tooluid:
  475. old_tool_dia = v['tooldia']
  476. break
  477. restore_dia_item = self.tools_table.item(row, 1)
  478. restore_dia_item.setText(str(old_tool_dia))
  479. self.app.inform.emit(_("[WARNING_NOTCL] Edit cancelled. New diameter value is already in the Tool Table."))
  480. self.build_ui()
  481. def on_tool_delete(self, rows_to_delete=None, all=None):
  482. self.ui_disconnect()
  483. deleted_tools_list = []
  484. if all:
  485. self.paint_tools.clear()
  486. self.build_ui()
  487. return
  488. if rows_to_delete:
  489. try:
  490. for row in rows_to_delete:
  491. tooluid_del = int(self.tools_table.item(row, 3).text())
  492. deleted_tools_list.append(tooluid_del)
  493. except TypeError:
  494. deleted_tools_list.append(rows_to_delete)
  495. for t in deleted_tools_list:
  496. self.ncc_tools.pop(t, None)
  497. self.build_ui()
  498. return
  499. try:
  500. if self.tools_table.selectedItems():
  501. for row_sel in self.tools_table.selectedItems():
  502. row = row_sel.row()
  503. if row < 0:
  504. continue
  505. tooluid_del = int(self.tools_table.item(row, 3).text())
  506. deleted_tools_list.append(tooluid_del)
  507. for t in deleted_tools_list:
  508. self.ncc_tools.pop(t, None)
  509. except AttributeError:
  510. self.app.inform.emit(_("[WARNING_NOTCL] Delete failed. Select a tool to delete."))
  511. return
  512. except Exception as e:
  513. log.debug(str(e))
  514. self.app.inform.emit(_("[success] Tool(s) deleted from Tool Table."))
  515. self.build_ui()
  516. def on_ncc(self):
  517. try:
  518. over = float(self.ncc_overlap_entry.get_value())
  519. except ValueError:
  520. # try to convert comma to decimal point. if it's still not working error message and return
  521. try:
  522. over = float(self.ncc_overlap_entry.get_value().replace(',', '.'))
  523. except ValueError:
  524. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  525. "use a number."))
  526. return
  527. over = over if over else self.app.defaults["tools_nccoverlap"]
  528. if over >= 1 or over < 0:
  529. self.app.inform.emit(_("[ERROR_NOTCL] Overlap value must be between "
  530. "0 (inclusive) and 1 (exclusive), "))
  531. return
  532. try:
  533. margin = float(self.ncc_margin_entry.get_value())
  534. except ValueError:
  535. # try to convert comma to decimal point. if it's still not working error message and return
  536. try:
  537. margin = float(self.ncc_margin_entry.get_value().replace(',', '.'))
  538. except ValueError:
  539. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  540. "use a number."))
  541. return
  542. margin = margin if margin else self.app.defaults["tools_nccmargin"]
  543. connect = self.ncc_connect_cb.get_value()
  544. connect = connect if connect else self.app.defaults["tools_nccconnect"]
  545. contour = self.ncc_contour_cb.get_value()
  546. contour = contour if contour else self.app.defaults["tools_ncccontour"]
  547. clearing_method = self.ncc_rest_cb.get_value()
  548. clearing_method = clearing_method if clearing_method else self.app.defaults["tools_nccrest"]
  549. pol_method = self.ncc_method_radio.get_value()
  550. pol_method = pol_method if pol_method else self.app.defaults["tools_nccmethod"]
  551. self.obj_name = self.object_combo.currentText()
  552. # Get source object.
  553. try:
  554. self.ncc_obj = self.app.collection.get_by_name(self.obj_name)
  555. except:
  556. self.app.inform.emit(_("[ERROR_NOTCL] Could not retrieve object: %s") % self.obj_name)
  557. return "Could not retrieve object: %s" % self.obj_name
  558. # Prepare non-copper polygons
  559. try:
  560. bounding_box = self.ncc_obj.solid_geometry.envelope.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  561. except AttributeError:
  562. self.app.inform.emit(_("[ERROR_NOTCL] No Gerber file available."))
  563. return
  564. # calculate the empty area by subtracting the solid_geometry from the object bounding box geometry
  565. empty = self.ncc_obj.get_empty_area(bounding_box)
  566. if type(empty) is Polygon:
  567. empty = MultiPolygon([empty])
  568. # clear non copper using standard algorithm
  569. if clearing_method == False:
  570. self.clear_non_copper(
  571. empty=empty,
  572. over=over,
  573. pol_method=pol_method,
  574. connect=connect,
  575. contour=contour
  576. )
  577. # clear non copper using rest machining algorithm
  578. else:
  579. self.clear_non_copper_rest(
  580. empty=empty,
  581. over=over,
  582. pol_method=pol_method,
  583. connect=connect,
  584. contour=contour
  585. )
  586. def clear_non_copper(self, empty, over, pol_method, outname=None, connect=True, contour=True):
  587. name = outname if outname else self.obj_name + "_ncc"
  588. # Sort tools in descending order
  589. sorted_tools = []
  590. for k, v in self.ncc_tools.items():
  591. sorted_tools.append(float('%.4f' % float(v['tooldia'])))
  592. sorted_tools.sort(reverse=True)
  593. # Do job in background
  594. proc = self.app.proc_container.new(_("Clearing Non-Copper areas."))
  595. def initialize(geo_obj, app_obj):
  596. assert isinstance(geo_obj, FlatCAMGeometry), \
  597. "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
  598. cleared_geo = []
  599. # Already cleared area
  600. cleared = MultiPolygon()
  601. # flag for polygons not cleared
  602. app_obj.poly_not_cleared = False
  603. # Generate area for each tool
  604. offset = sum(sorted_tools)
  605. current_uid = int(1)
  606. for tool in sorted_tools:
  607. self.app.inform.emit(_('[success] Non-Copper Clearing with ToolDia = %s started.') % str(tool))
  608. cleared_geo[:] = []
  609. # Get remaining tools offset
  610. offset -= (tool - 1e-12)
  611. # Area to clear
  612. area = empty.buffer(-offset)
  613. try:
  614. area = area.difference(cleared)
  615. except:
  616. continue
  617. # Transform area to MultiPolygon
  618. if type(area) is Polygon:
  619. area = MultiPolygon([area])
  620. if area.geoms:
  621. if len(area.geoms) > 0:
  622. for p in area.geoms:
  623. try:
  624. if pol_method == 'standard':
  625. cp = self.clear_polygon(p, tool, self.app.defaults["gerber_circle_steps"],
  626. overlap=over, contour=contour, connect=connect)
  627. elif pol_method == 'seed':
  628. cp = self.clear_polygon2(p, tool, self.app.defaults["gerber_circle_steps"],
  629. overlap=over, contour=contour, connect=connect)
  630. else:
  631. cp = self.clear_polygon3(p, tool, self.app.defaults["gerber_circle_steps"],
  632. overlap=over, contour=contour, connect=connect)
  633. if cp:
  634. cleared_geo += list(cp.get_objects())
  635. except:
  636. log.warning("Polygon can not be cleared.")
  637. app_obj.poly_not_cleared = True
  638. continue
  639. # check if there is a geometry at all in the cleared geometry
  640. if cleared_geo:
  641. # Overall cleared area
  642. cleared = empty.buffer(-offset * (1 + over)).buffer(-tool / 1.999999).buffer(
  643. tool / 1.999999)
  644. # clean-up cleared geo
  645. cleared = cleared.buffer(0)
  646. # find the tooluid associated with the current tool_dia so we know where to add the tool
  647. # solid_geometry
  648. for k, v in self.ncc_tools.items():
  649. if float('%.4f' % v['tooldia']) == float('%.4f' % tool):
  650. current_uid = int(k)
  651. # add the solid_geometry to the current too in self.paint_tools dictionary
  652. # and then reset the temporary list that stored that solid_geometry
  653. v['solid_geometry'] = deepcopy(cleared_geo)
  654. v['data']['name'] = name
  655. break
  656. geo_obj.tools[current_uid] = dict(self.ncc_tools[current_uid])
  657. else:
  658. log.debug("There are no geometries in the cleared polygon.")
  659. geo_obj.options["cnctooldia"] = tool
  660. geo_obj.multigeo = True
  661. def job_thread(app_obj):
  662. try:
  663. app_obj.new_object("geometry", name, initialize)
  664. except Exception as e:
  665. proc.done()
  666. self.app.inform.emit(_('[ERROR_NOTCL] NCCTool.clear_non_copper() --> %s') % str(e))
  667. return
  668. proc.done()
  669. if app_obj.poly_not_cleared is False:
  670. self.app.inform.emit(_('[success] NCC Tool finished.'))
  671. else:
  672. self.app.inform.emit(_('[WARNING_NOTCL] NCC Tool finished but some PCB features could not be cleared. '
  673. 'Check the result.'))
  674. # reset the variable for next use
  675. app_obj.poly_not_cleared = False
  676. # focus on Selected Tab
  677. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  678. self.tools_frame.hide()
  679. self.app.ui.notebook.setTabText(2, _("Tools"))
  680. # Promise object with the new name
  681. self.app.collection.promise(name)
  682. # Background
  683. self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
  684. # clear copper with 'rest-machining' algorithm
  685. def clear_non_copper_rest(self, empty, over, pol_method, outname=None, connect=True, contour=True):
  686. name = outname if outname is not None else self.obj_name + "_ncc_rm"
  687. # Sort tools in descending order
  688. sorted_tools = []
  689. for k, v in self.ncc_tools.items():
  690. sorted_tools.append(float('%.4f' % float(v['tooldia'])))
  691. sorted_tools.sort(reverse=True)
  692. # Do job in background
  693. proc = self.app.proc_container.new(_("Clearing Non-Copper areas."))
  694. def initialize_rm(geo_obj, app_obj):
  695. assert isinstance(geo_obj, FlatCAMGeometry), \
  696. "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
  697. cleared_geo = []
  698. cleared_by_last_tool = []
  699. rest_geo = []
  700. current_uid = 1
  701. # repurposed flag for final object, geo_obj. True if it has any solid_geometry, False if not.
  702. app_obj.poly_not_cleared = True
  703. area = empty.buffer(0)
  704. # Generate area for each tool
  705. while sorted_tools:
  706. tool = sorted_tools.pop(0)
  707. self.app.inform.emit(_('[success] Non-Copper Rest Clearing with ToolDia = %s started.') % str(tool))
  708. tool_used = tool - 1e-12
  709. cleared_geo[:] = []
  710. # Area to clear
  711. for poly in cleared_by_last_tool:
  712. try:
  713. area = area.difference(poly)
  714. except:
  715. pass
  716. cleared_by_last_tool[:] = []
  717. # Transform area to MultiPolygon
  718. if type(area) is Polygon:
  719. area = MultiPolygon([area])
  720. # add the rest that was not able to be cleared previously; area is a MultyPolygon
  721. # and rest_geo it's a list
  722. allparts = [p.buffer(0) for p in area.geoms]
  723. allparts += deepcopy(rest_geo)
  724. rest_geo[:] = []
  725. area = MultiPolygon(deepcopy(allparts))
  726. allparts[:] = []
  727. if area.geoms:
  728. if len(area.geoms) > 0:
  729. for p in area.geoms:
  730. try:
  731. if pol_method == 'standard':
  732. cp = self.clear_polygon(p, tool_used, self.app.defaults["gerber_circle_steps"],
  733. overlap=over, contour=contour, connect=connect)
  734. elif pol_method == 'seed':
  735. cp = self.clear_polygon2(p, tool_used,
  736. self.app.defaults["gerber_circle_steps"],
  737. overlap=over, contour=contour, connect=connect)
  738. else:
  739. cp = self.clear_polygon3(p, tool_used,
  740. self.app.defaults["gerber_circle_steps"],
  741. overlap=over, contour=contour, connect=connect)
  742. cleared_geo.append(list(cp.get_objects()))
  743. except:
  744. log.warning("Polygon can't be cleared.")
  745. # this polygon should be added to a list and then try clear it with a smaller tool
  746. rest_geo.append(p)
  747. # check if there is a geometry at all in the cleared geometry
  748. if cleared_geo:
  749. # Overall cleared area
  750. cleared_area = list(self.flatten_list(cleared_geo))
  751. # cleared = MultiPolygon([p.buffer(tool_used / 2).buffer(-tool_used / 2)
  752. # for p in cleared_area])
  753. # here we store the poly's already processed in the original geometry by the current tool
  754. # into cleared_by_last_tool list
  755. # this will be sustracted from the original geometry_to_be_cleared and make data for
  756. # the next tool
  757. buffer_value = tool_used / 2
  758. for p in cleared_area:
  759. poly = p.buffer(buffer_value)
  760. cleared_by_last_tool.append(poly)
  761. # find the tooluid associated with the current tool_dia so we know
  762. # where to add the tool solid_geometry
  763. for k, v in self.ncc_tools.items():
  764. if float('%.4f' % v['tooldia']) == float('%.4f' % tool):
  765. current_uid = int(k)
  766. # add the solid_geometry to the current too in self.paint_tools dictionary
  767. # and then reset the temporary list that stored that solid_geometry
  768. v['solid_geometry'] = deepcopy(cleared_area)
  769. v['data']['name'] = name
  770. cleared_area[:] = []
  771. break
  772. geo_obj.tools[current_uid] = dict(self.ncc_tools[current_uid])
  773. else:
  774. log.debug("There are no geometries in the cleared polygon.")
  775. geo_obj.multigeo = True
  776. geo_obj.options["cnctooldia"] = tool
  777. # check to see if geo_obj.tools is empty
  778. # it will be updated only if there is a solid_geometry for tools
  779. if geo_obj.tools:
  780. return
  781. else:
  782. # I will use this variable for this purpose although it was meant for something else
  783. # signal that we have no geo in the object therefore don't create it
  784. app_obj.poly_not_cleared = False
  785. return "fail"
  786. def job_thread(app_obj):
  787. try:
  788. app_obj.new_object("geometry", name, initialize_rm)
  789. except Exception as e:
  790. proc.done()
  791. self.app.inform.emit(_('[ERROR_NOTCL] NCCTool.clear_non_copper_rest() --> %s') % str(e))
  792. return
  793. if app_obj.poly_not_cleared is True:
  794. self.app.inform.emit('[success] NCC Tool finished.')
  795. # focus on Selected Tab
  796. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  797. else:
  798. self.app.inform.emit(_('[ERROR_NOTCL] NCC Tool finished but could not clear the object '
  799. 'with current settings.'))
  800. # focus on Project Tab
  801. self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  802. proc.done()
  803. # reset the variable for next use
  804. app_obj.poly_not_cleared = False
  805. self.tools_frame.hide()
  806. self.app.ui.notebook.setTabText(2, "Tools")
  807. # Promise object with the new name
  808. self.app.collection.promise(name)
  809. # Background
  810. self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
  811. def reset_fields(self):
  812. self.object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))