ToolNonCopperClear.py 61 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379
  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. from shapely.geometry import base
  13. import gettext
  14. import FlatCAMTranslation as fcTranslate
  15. import builtins
  16. fcTranslate.apply_language('strings')
  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. # ################################################
  45. # ##### Type of object to be copper cleaned ######
  46. # ################################################
  47. self.type_obj_combo = QtWidgets.QComboBox()
  48. self.type_obj_combo.addItem("Gerber")
  49. self.type_obj_combo.addItem("Excellon")
  50. self.type_obj_combo.addItem("Geometry")
  51. # we get rid of item1 ("Excellon") as it is not suitable
  52. self.type_obj_combo.view().setRowHidden(1, True)
  53. self.type_obj_combo.setItemIcon(0, QtGui.QIcon("share/flatcam_icon16.png"))
  54. self.type_obj_combo.setItemIcon(2, QtGui.QIcon("share/geometry16.png"))
  55. self.type_obj_combo_label = QtWidgets.QLabel('%s:' % _("Obj Type"))
  56. self.type_obj_combo_label.setToolTip(
  57. _("Specify the type of object to be cleared of excess copper.\n"
  58. "It can be of type: Gerber or Geometry.\n"
  59. "What is selected here will dictate the kind\n"
  60. "of objects that will populate the 'Object' combobox.")
  61. )
  62. self.type_obj_combo_label.setMinimumWidth(60)
  63. form_layout.addRow(self.type_obj_combo_label, self.type_obj_combo)
  64. # ################################################
  65. # ##### The object to be copper cleaned ##########
  66. # ################################################
  67. self.obj_combo = QtWidgets.QComboBox()
  68. self.obj_combo.setModel(self.app.collection)
  69. self.obj_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  70. self.obj_combo.setCurrentIndex(1)
  71. self.object_label = QtWidgets.QLabel('%s:' % _("Object"))
  72. self.object_label.setToolTip(_("Object to be cleared of excess copper."))
  73. form_layout.addRow(self.object_label, self.obj_combo)
  74. e_lab_0 = QtWidgets.QLabel('')
  75. form_layout.addRow(e_lab_0)
  76. # ### Tools ## ##
  77. self.tools_table_label = QtWidgets.QLabel('<b>%s</b>' % _('Tools Table'))
  78. self.tools_table_label.setToolTip(
  79. _("Tools pool from which the algorithm\n"
  80. "will pick the ones used for copper clearing.")
  81. )
  82. self.tools_box.addWidget(self.tools_table_label)
  83. self.tools_table = FCTable()
  84. self.tools_box.addWidget(self.tools_table)
  85. self.tools_table.setColumnCount(4)
  86. self.tools_table.setHorizontalHeaderLabels(['#', _('Diameter'), _('TT'), ''])
  87. self.tools_table.setColumnHidden(3, True)
  88. self.tools_table.setSortingEnabled(False)
  89. # self.tools_table.setSelectionBehavior(QtWidgets.QAbstractItemView.SelectRows)
  90. self.tools_table.horizontalHeaderItem(0).setToolTip(
  91. _("This is the Tool Number.\n"
  92. "Non copper clearing will start with the tool with the biggest \n"
  93. "diameter, continuing until there are no more tools.\n"
  94. "Only tools that create NCC clearing geometry will still be present\n"
  95. "in the resulting geometry. This is because with some tools\n"
  96. "this function will not be able to create painting geometry.")
  97. )
  98. self.tools_table.horizontalHeaderItem(1).setToolTip(
  99. _("Tool Diameter. It's value (in current FlatCAM units) \n"
  100. "is the cut width into the material."))
  101. self.tools_table.horizontalHeaderItem(2).setToolTip(
  102. _("The Tool Type (TT) can be:<BR>"
  103. "- <B>Circular</B> with 1 ... 4 teeth -> it is informative only. Being circular, <BR>"
  104. "the cut width in material is exactly the tool diameter.<BR>"
  105. "- <B>Ball</B> -> informative only and make reference to the Ball type endmill.<BR>"
  106. "- <B>V-Shape</B> -> it will disable de Z-Cut parameter in the resulting geometry UI form "
  107. "and enable two additional UI form fields in the resulting geometry: V-Tip Dia and "
  108. "V-Tip Angle. Adjusting those two values will adjust the Z-Cut parameter such "
  109. "as the cut width into material will be equal with the value in the Tool Diameter "
  110. "column of this table.<BR>"
  111. "Choosing the <B>V-Shape</B> Tool Type automatically will select the Operation Type "
  112. "in the resulting geometry as Isolation."))
  113. self.ncc_order_label = QtWidgets.QLabel('<b>%s:</b>' % _('Tool order'))
  114. self.ncc_order_label.setToolTip(_("This set the way that the tools in the tools table are used.\n"
  115. "'No' --> means that the used order is the one in the tool table\n"
  116. "'Forward' --> means that the tools will be ordered from small to big\n"
  117. "'Reverse' --> menas that the tools will ordered from big to small\n\n"
  118. "WARNING: using rest machining will automatically set the order\n"
  119. "in reverse and disable this control."))
  120. self.ncc_order_radio = RadioSet([{'label': _('No'), 'value': 'no'},
  121. {'label': _('Forward'), 'value': 'fwd'},
  122. {'label': _('Reverse'), 'value': 'rev'}])
  123. self.ncc_order_radio.setToolTip(_("This set the way that the tools in the tools table are used.\n"
  124. "'No' --> means that the used order is the one in the tool table\n"
  125. "'Forward' --> means that the tools will be ordered from small to big\n"
  126. "'Reverse' --> menas that the tools will ordered from big to small\n\n"
  127. "WARNING: using rest machining will automatically set the order\n"
  128. "in reverse and disable this control."))
  129. form = QtWidgets.QFormLayout()
  130. self.tools_box.addLayout(form)
  131. form.addRow(QtWidgets.QLabel(''), QtWidgets.QLabel(''))
  132. form.addRow(self.ncc_order_label, self.ncc_order_radio)
  133. # ### Add a new Tool ####
  134. self.addtool_entry_lbl = QtWidgets.QLabel('<b>%s:</b>' % _('Tool Dia'))
  135. self.addtool_entry_lbl.setToolTip(
  136. _("Diameter for the new tool to add in the Tool Table")
  137. )
  138. self.addtool_entry = FCEntry2()
  139. form.addRow(self.addtool_entry_lbl, self.addtool_entry)
  140. grid2 = QtWidgets.QGridLayout()
  141. self.tools_box.addLayout(grid2)
  142. self.addtool_btn = QtWidgets.QPushButton(_('Add'))
  143. self.addtool_btn.setToolTip(
  144. _("Add a new tool to the Tool Table\n"
  145. "with the diameter specified above.")
  146. )
  147. # self.copytool_btn = QtWidgets.QPushButton('Copy')
  148. # self.copytool_btn.setToolTip(
  149. # "Copy a selection of tools in the Tool Table\n"
  150. # "by first selecting a row in the Tool Table."
  151. # )
  152. self.deltool_btn = QtWidgets.QPushButton(_('Delete'))
  153. self.deltool_btn.setToolTip(
  154. _("Delete a selection of tools in the Tool Table\n"
  155. "by first selecting a row(s) in the Tool Table.")
  156. )
  157. grid2.addWidget(self.addtool_btn, 0, 0)
  158. # grid2.addWidget(self.copytool_btn, 0, 1)
  159. grid2.addWidget(self.deltool_btn, 0, 2)
  160. self.empty_label_0 = QtWidgets.QLabel('')
  161. self.tools_box.addWidget(self.empty_label_0)
  162. grid3 = QtWidgets.QGridLayout()
  163. self.tools_box.addLayout(grid3)
  164. e_lab_1 = QtWidgets.QLabel('<b>%s:</b>' % _("Parameters"))
  165. grid3.addWidget(e_lab_1, 0, 0)
  166. nccoverlabel = QtWidgets.QLabel(_('Overlap Rate:'))
  167. nccoverlabel.setToolTip(
  168. _("How much (fraction) of the tool width to overlap each tool pass.\n"
  169. "Example:\n"
  170. "A value here of 0.25 means 25% from the tool diameter found above.\n\n"
  171. "Adjust the value starting with lower values\n"
  172. "and increasing it if areas that should be cleared are still \n"
  173. "not cleared.\n"
  174. "Lower values = faster processing, faster execution on PCB.\n"
  175. "Higher values = slow processing and slow execution on CNC\n"
  176. "due of too many paths.")
  177. )
  178. grid3.addWidget(nccoverlabel, 1, 0)
  179. self.ncc_overlap_entry = FCEntry()
  180. grid3.addWidget(self.ncc_overlap_entry, 1, 1)
  181. nccmarginlabel = QtWidgets.QLabel('%s:' % _('Margin'))
  182. nccmarginlabel.setToolTip(
  183. _("Bounding box margin.")
  184. )
  185. grid3.addWidget(nccmarginlabel, 2, 0)
  186. self.ncc_margin_entry = FCEntry()
  187. grid3.addWidget(self.ncc_margin_entry, 2, 1)
  188. # Method
  189. methodlabel = QtWidgets.QLabel('%s:' % _('Method'))
  190. methodlabel.setToolTip(
  191. _("Algorithm for non-copper clearing:<BR>"
  192. "<B>Standard</B>: Fixed step inwards.<BR>"
  193. "<B>Seed-based</B>: Outwards from seed.<BR>"
  194. "<B>Line-based</B>: Parallel lines.")
  195. )
  196. grid3.addWidget(methodlabel, 3, 0)
  197. self.ncc_method_radio = RadioSet([
  198. {"label": _("Standard"), "value": "standard"},
  199. {"label": _("Seed-based"), "value": "seed"},
  200. {"label": _("Straight lines"), "value": "lines"}
  201. ], orientation='vertical', stretch=False)
  202. grid3.addWidget(self.ncc_method_radio, 3, 1)
  203. # Connect lines
  204. pathconnectlabel = QtWidgets.QLabel('%s:' % _("Connect"))
  205. pathconnectlabel.setToolTip(
  206. _("Draw lines between resulting\n"
  207. "segments to minimize tool lifts.")
  208. )
  209. grid3.addWidget(pathconnectlabel, 4, 0)
  210. self.ncc_connect_cb = FCCheckBox()
  211. grid3.addWidget(self.ncc_connect_cb, 4, 1)
  212. contourlabel = QtWidgets.QLabel('%s:' % _("Contour"))
  213. contourlabel.setToolTip(
  214. _("Cut around the perimeter of the polygon\n"
  215. "to trim rough edges.")
  216. )
  217. grid3.addWidget(contourlabel, 5, 0)
  218. self.ncc_contour_cb = FCCheckBox()
  219. grid3.addWidget(self.ncc_contour_cb, 5, 1)
  220. restlabel = QtWidgets.QLabel('%s:' % _("Rest M."))
  221. restlabel.setToolTip(
  222. _("If checked, use 'rest machining'.\n"
  223. "Basically it will clear copper outside PCB features,\n"
  224. "using the biggest tool and continue with the next tools,\n"
  225. "from bigger to smaller, to clear areas of copper that\n"
  226. "could not be cleared by previous tool, until there is\n"
  227. "no more copper to clear or there are no more tools.\n"
  228. "If not checked, use the standard algorithm.")
  229. )
  230. grid3.addWidget(restlabel, 6, 0)
  231. self.ncc_rest_cb = FCCheckBox()
  232. grid3.addWidget(self.ncc_rest_cb, 6, 1)
  233. # ## NCC Offset choice
  234. self.ncc_offset_choice_label = QtWidgets.QLabel('%s:' % _("Offset"))
  235. self.ncc_offset_choice_label.setToolTip(
  236. _("If used, it will add an offset to the copper features.\n"
  237. "The copper clearing will finish to a distance\n"
  238. "from the copper features.\n"
  239. "The value can be between 0 and 10 FlatCAM units.")
  240. )
  241. grid3.addWidget(self.ncc_offset_choice_label, 7, 0)
  242. self.ncc_choice_offset_cb = FCCheckBox()
  243. grid3.addWidget(self.ncc_choice_offset_cb, 7, 1)
  244. # ## NCC Offset value
  245. self.ncc_offset_label = QtWidgets.QLabel('%s:' % _("Offset value"))
  246. self.ncc_offset_label.setToolTip(
  247. _("If used, it will add an offset to the copper features.\n"
  248. "The copper clearing will finish to a distance\n"
  249. "from the copper features.\n"
  250. "The value can be between 0 and 10 FlatCAM units.")
  251. )
  252. grid3.addWidget(self.ncc_offset_label, 8, 0)
  253. self.ncc_offset_spinner = FCDoubleSpinner()
  254. self.ncc_offset_spinner.set_range(0.00, 10.00)
  255. self.ncc_offset_spinner.set_precision(4)
  256. self.ncc_offset_spinner.setWrapping(True)
  257. units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  258. if units == 'MM':
  259. self.ncc_offset_spinner.setSingleStep(0.1)
  260. else:
  261. self.ncc_offset_spinner.setSingleStep(0.01)
  262. grid3.addWidget(self.ncc_offset_spinner, 8, 1)
  263. self.ncc_offset_label.hide()
  264. self.ncc_offset_spinner.hide()
  265. # ## Reference
  266. self.reference_radio = RadioSet([
  267. {'label': _('Itself'), 'value': 'itself'},
  268. {"label": _("Area Selection"), "value": "area"},
  269. {'label': _("Reference Object"), 'value': 'box'}
  270. ], orientation='vertical', stretch=False)
  271. self.reference_label = QtWidgets.QLabel(_("Reference:"))
  272. self.reference_label.setToolTip(
  273. _("- 'Itself' - the non copper clearing extent\n"
  274. "is based on the object that is copper cleared.\n "
  275. "- 'Area Selection' - left mouse click to start selection of the area to be painted.\n"
  276. "Keeping a modifier key pressed (CTRL or SHIFT) will allow to add multiple areas.\n"
  277. "- 'Reference Object' - will do non copper clearing within the area\n"
  278. "specified by another object.")
  279. )
  280. grid3.addWidget(self.reference_label, 9, 0)
  281. grid3.addWidget(self.reference_radio, 9, 1)
  282. form1 = QtWidgets.QFormLayout()
  283. self.tools_box.addLayout(form1)
  284. self.box_combo_type_label = QtWidgets.QLabel('%s:' % _("Ref. Type"))
  285. self.box_combo_type_label.setToolTip(
  286. _("The type of FlatCAM object to be used as non copper clearing reference.\n"
  287. "It can be Gerber, Excellon or Geometry.")
  288. )
  289. self.box_combo_type = QtWidgets.QComboBox()
  290. self.box_combo_type.addItem(_("Gerber Reference Box Object"))
  291. self.box_combo_type.addItem(_("Excellon Reference Box Object"))
  292. self.box_combo_type.addItem(_("Geometry Reference Box Object"))
  293. form1.addRow(self.box_combo_type_label, self.box_combo_type)
  294. self.box_combo_label = QtWidgets.QLabel('%s:' % _("Ref. Object"))
  295. self.box_combo_label.setToolTip(
  296. _("The FlatCAM object to be used as non copper clearing reference.")
  297. )
  298. self.box_combo = QtWidgets.QComboBox()
  299. self.box_combo.setModel(self.app.collection)
  300. self.box_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  301. self.box_combo.setCurrentIndex(1)
  302. form1.addRow(self.box_combo_label, self.box_combo)
  303. self.box_combo.hide()
  304. self.box_combo_label.hide()
  305. self.box_combo_type.hide()
  306. self.box_combo_type_label.hide()
  307. self.generate_ncc_button = QtWidgets.QPushButton(_('Generate Geometry'))
  308. self.generate_ncc_button.setToolTip(
  309. _("Create the Geometry Object\n"
  310. "for non-copper routing.")
  311. )
  312. self.tools_box.addWidget(self.generate_ncc_button)
  313. self.tools_box.addStretch()
  314. self.units = ''
  315. self.ncc_tools = {}
  316. self.tooluid = 0
  317. # store here the default data for Geometry Data
  318. self.default_data = {}
  319. self.obj_name = ""
  320. self.ncc_obj = None
  321. self.sel_rect = []
  322. self.bound_obj_name = ""
  323. self.bound_obj = None
  324. self.first_click = False
  325. self.cursor_pos = None
  326. self.mouse_is_dragging = False
  327. self.addtool_btn.clicked.connect(self.on_tool_add)
  328. self.addtool_entry.returnPressed.connect(self.on_tool_add)
  329. self.deltool_btn.clicked.connect(self.on_tool_delete)
  330. self.generate_ncc_button.clicked.connect(self.on_ncc_click)
  331. self.box_combo_type.currentIndexChanged.connect(self.on_combo_box_type)
  332. self.reference_radio.group_toggle_fn = self.on_toggle_reference
  333. self.ncc_choice_offset_cb.stateChanged.connect(self.on_offset_choice)
  334. self.ncc_rest_cb.stateChanged.connect(self.on_rest_machining_check)
  335. self.ncc_order_radio.activated_custom[str].connect(self.on_order_changed)
  336. self.type_obj_combo.currentIndexChanged.connect(self.on_type_obj_index_changed)
  337. def on_type_obj_index_changed(self, index):
  338. obj_type = self.type_obj_combo.currentIndex()
  339. self.obj_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  340. self.obj_combo.setCurrentIndex(0)
  341. def install(self, icon=None, separator=None, **kwargs):
  342. FlatCAMTool.install(self, icon, separator, shortcut='ALT+N', **kwargs)
  343. def run(self, toggle=True):
  344. self.app.report_usage("ToolNonCopperClear()")
  345. if toggle:
  346. # if the splitter is hidden, display it, else hide it but only if the current widget is the same
  347. if self.app.ui.splitter.sizes()[0] == 0:
  348. self.app.ui.splitter.setSizes([1, 1])
  349. else:
  350. try:
  351. if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
  352. # if tab is populated with the tool but it does not have the focus, focus on it
  353. if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
  354. # focus on Tool Tab
  355. self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
  356. else:
  357. self.app.ui.splitter.setSizes([0, 1])
  358. except AttributeError:
  359. pass
  360. else:
  361. if self.app.ui.splitter.sizes()[0] == 0:
  362. self.app.ui.splitter.setSizes([1, 1])
  363. FlatCAMTool.run(self)
  364. self.set_tool_ui()
  365. # reset those objects on a new run
  366. self.ncc_obj = None
  367. self.bound_obj = None
  368. self.obj_name = ''
  369. self.bound_obj_name = ''
  370. self.build_ui()
  371. self.app.ui.notebook.setTabText(2, _("NCC Tool"))
  372. def set_tool_ui(self):
  373. self.tools_frame.show()
  374. self.ncc_order_radio.set_value(self.app.defaults["tools_nccorder"])
  375. self.ncc_overlap_entry.set_value(self.app.defaults["tools_nccoverlap"])
  376. self.ncc_margin_entry.set_value(self.app.defaults["tools_nccmargin"])
  377. self.ncc_method_radio.set_value(self.app.defaults["tools_nccmethod"])
  378. self.ncc_connect_cb.set_value(self.app.defaults["tools_nccconnect"])
  379. self.ncc_contour_cb.set_value(self.app.defaults["tools_ncccontour"])
  380. self.ncc_rest_cb.set_value(self.app.defaults["tools_nccrest"])
  381. self.reference_radio.set_value(self.app.defaults["tools_nccref"])
  382. self.tools_table.setupContextMenu()
  383. self.tools_table.addContextMenu(
  384. "Add", lambda: self.on_tool_add(dia=None, muted=None), icon=QtGui.QIcon("share/plus16.png"))
  385. self.tools_table.addContextMenu(
  386. "Delete", lambda:
  387. self.on_tool_delete(rows_to_delete=None, all=None), icon=QtGui.QIcon("share/delete32.png"))
  388. # init the working variables
  389. self.default_data.clear()
  390. self.default_data.update({
  391. "name": '_ncc',
  392. "plot": self.app.defaults["geometry_plot"],
  393. "cutz": self.app.defaults["geometry_cutz"],
  394. "vtipdia": 0.1,
  395. "vtipangle": 30,
  396. "travelz": self.app.defaults["geometry_travelz"],
  397. "feedrate": self.app.defaults["geometry_feedrate"],
  398. "feedrate_z": self.app.defaults["geometry_feedrate_z"],
  399. "feedrate_rapid": self.app.defaults["geometry_feedrate_rapid"],
  400. "dwell": self.app.defaults["geometry_dwell"],
  401. "dwelltime": self.app.defaults["geometry_dwelltime"],
  402. "multidepth": self.app.defaults["geometry_multidepth"],
  403. "ppname_g": self.app.defaults["geometry_ppname_g"],
  404. "depthperpass": self.app.defaults["geometry_depthperpass"],
  405. "extracut": self.app.defaults["geometry_extracut"],
  406. "toolchange": self.app.defaults["geometry_toolchange"],
  407. "toolchangez": self.app.defaults["geometry_toolchangez"],
  408. "endz": self.app.defaults["geometry_endz"],
  409. "spindlespeed": self.app.defaults["geometry_spindlespeed"],
  410. "toolchangexy": self.app.defaults["geometry_toolchangexy"],
  411. "startz": self.app.defaults["geometry_startz"],
  412. "tooldia": self.app.defaults["tools_painttooldia"],
  413. "paintmargin": self.app.defaults["tools_paintmargin"],
  414. "paintmethod": self.app.defaults["tools_paintmethod"],
  415. "selectmethod": self.app.defaults["tools_selectmethod"],
  416. "pathconnect": self.app.defaults["tools_pathconnect"],
  417. "paintcontour": self.app.defaults["tools_paintcontour"],
  418. "paintoverlap": self.app.defaults["tools_paintoverlap"],
  419. "nccoverlap": self.app.defaults["tools_nccoverlap"],
  420. "nccmargin": self.app.defaults["tools_nccmargin"],
  421. "nccmethod": self.app.defaults["tools_nccmethod"],
  422. "nccconnect": self.app.defaults["tools_nccconnect"],
  423. "ncccontour": self.app.defaults["tools_ncccontour"],
  424. "nccrest": self.app.defaults["tools_nccrest"]
  425. })
  426. try:
  427. dias = [float(eval(dia)) for dia in self.app.defaults["tools_ncctools"].split(",") if dia != '']
  428. except Exception as e:
  429. log.error("At least one tool diameter needed. "
  430. "Verify in Edit -> Preferences -> TOOLS -> NCC Tools. %s" % str(e))
  431. return
  432. self.tooluid = 0
  433. self.ncc_tools.clear()
  434. for tool_dia in dias:
  435. self.tooluid += 1
  436. self.ncc_tools.update({
  437. int(self.tooluid): {
  438. 'tooldia': float('%.4f' % tool_dia),
  439. 'offset': 'Path',
  440. 'offset_value': 0.0,
  441. 'type': 'Iso',
  442. 'tool_type': 'V',
  443. 'data': dict(self.default_data),
  444. 'solid_geometry': []
  445. }
  446. })
  447. self.obj_name = ""
  448. self.ncc_obj = None
  449. self.bound_obj_name = ""
  450. self.bound_obj = None
  451. self.tool_type_item_options = ["C1", "C2", "C3", "C4", "B", "V"]
  452. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  453. def build_ui(self):
  454. self.ui_disconnect()
  455. # updated units
  456. self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().upper()
  457. if self.units == "IN":
  458. self.addtool_entry.set_value(0.039)
  459. else:
  460. self.addtool_entry.set_value(1)
  461. sorted_tools = []
  462. for k, v in self.ncc_tools.items():
  463. sorted_tools.append(float('%.4f' % float(v['tooldia'])))
  464. order = self.ncc_order_radio.get_value()
  465. if order == 'fwd':
  466. sorted_tools.sort(reverse=False)
  467. elif order == 'rev':
  468. sorted_tools.sort(reverse=True)
  469. else:
  470. pass
  471. n = len(sorted_tools)
  472. self.tools_table.setRowCount(n)
  473. tool_id = 0
  474. for tool_sorted in sorted_tools:
  475. for tooluid_key, tooluid_value in self.ncc_tools.items():
  476. if float('%.4f' % tooluid_value['tooldia']) == tool_sorted:
  477. tool_id += 1
  478. id_ = QtWidgets.QTableWidgetItem('%d' % int(tool_id))
  479. id_.setFlags(QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  480. row_no = tool_id - 1
  481. self.tools_table.setItem(row_no, 0, id_) # Tool name/id
  482. # Make sure that the drill diameter when in MM is with no more than 2 decimals
  483. # There are no drill bits in MM with more than 3 decimals diameter
  484. # For INCH the decimals should be no more than 3. There are no drills under 10mils
  485. if self.units == 'MM':
  486. dia = QtWidgets.QTableWidgetItem('%.2f' % tooluid_value['tooldia'])
  487. else:
  488. dia = QtWidgets.QTableWidgetItem('%.4f' % tooluid_value['tooldia'])
  489. dia.setFlags(QtCore.Qt.ItemIsEnabled)
  490. tool_type_item = QtWidgets.QComboBox()
  491. for item in self.tool_type_item_options:
  492. tool_type_item.addItem(item)
  493. tool_type_item.setStyleSheet('background-color: rgb(255,255,255)')
  494. idx = tool_type_item.findText(tooluid_value['tool_type'])
  495. tool_type_item.setCurrentIndex(idx)
  496. tool_uid_item = QtWidgets.QTableWidgetItem(str(int(tooluid_key)))
  497. self.tools_table.setItem(row_no, 1, dia) # Diameter
  498. self.tools_table.setCellWidget(row_no, 2, tool_type_item)
  499. # ## REMEMBER: THIS COLUMN IS HIDDEN IN OBJECTUI.PY # ##
  500. self.tools_table.setItem(row_no, 3, tool_uid_item) # Tool unique ID
  501. # make the diameter column editable
  502. for row in range(tool_id):
  503. self.tools_table.item(row, 1).setFlags(
  504. QtCore.Qt.ItemIsEditable | QtCore.Qt.ItemIsSelectable | QtCore.Qt.ItemIsEnabled)
  505. # all the tools are selected by default
  506. self.tools_table.selectColumn(0)
  507. #
  508. self.tools_table.resizeColumnsToContents()
  509. self.tools_table.resizeRowsToContents()
  510. vertical_header = self.tools_table.verticalHeader()
  511. vertical_header.hide()
  512. self.tools_table.setVerticalScrollBarPolicy(QtCore.Qt.ScrollBarAlwaysOff)
  513. horizontal_header = self.tools_table.horizontalHeader()
  514. horizontal_header.setMinimumSectionSize(10)
  515. horizontal_header.setSectionResizeMode(0, QtWidgets.QHeaderView.Fixed)
  516. horizontal_header.resizeSection(0, 20)
  517. horizontal_header.setSectionResizeMode(1, QtWidgets.QHeaderView.Stretch)
  518. # self.tools_table.setSortingEnabled(True)
  519. # sort by tool diameter
  520. # self.tools_table.sortItems(1)
  521. self.tools_table.setMinimumHeight(self.tools_table.getHeight())
  522. self.tools_table.setMaximumHeight(self.tools_table.getHeight())
  523. self.ui_connect()
  524. def ui_connect(self):
  525. self.tools_table.itemChanged.connect(self.on_tool_edit)
  526. def ui_disconnect(self):
  527. try:
  528. # if connected, disconnect the signal from the slot on item_changed as it creates issues
  529. self.tools_table.itemChanged.disconnect(self.on_tool_edit)
  530. except (TypeError, AttributeError):
  531. pass
  532. def on_combo_box_type(self):
  533. obj_type = self.box_combo_type.currentIndex()
  534. self.box_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  535. self.box_combo.setCurrentIndex(0)
  536. def on_toggle_reference(self):
  537. if self.reference_radio.get_value() == "itself" or self.reference_radio.get_value() == "area":
  538. self.box_combo.hide()
  539. self.box_combo_label.hide()
  540. self.box_combo_type.hide()
  541. self.box_combo_type_label.hide()
  542. else:
  543. self.box_combo.show()
  544. self.box_combo_label.show()
  545. self.box_combo_type.show()
  546. self.box_combo_type_label.show()
  547. def on_offset_choice(self, state):
  548. if state:
  549. self.ncc_offset_label.show()
  550. self.ncc_offset_spinner.show()
  551. else:
  552. self.ncc_offset_label.hide()
  553. self.ncc_offset_spinner.hide()
  554. def on_order_changed(self, order):
  555. if order != 'no':
  556. self.build_ui()
  557. def on_rest_machining_check(self, state):
  558. if state:
  559. self.ncc_order_radio.set_value('rev')
  560. self.ncc_order_label.setDisabled(True)
  561. self.ncc_order_radio.setDisabled(True)
  562. else:
  563. self.ncc_order_label.setDisabled(False)
  564. self.ncc_order_radio.setDisabled(False)
  565. def on_tool_add(self, dia=None, muted=None):
  566. self.ui_disconnect()
  567. if dia:
  568. tool_dia = dia
  569. else:
  570. try:
  571. tool_dia = float(self.addtool_entry.get_value())
  572. except ValueError:
  573. # try to convert comma to decimal point. if it's still not working error message and return
  574. try:
  575. tool_dia = float(self.addtool_entry.get_value().replace(',', '.'))
  576. except ValueError:
  577. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  578. "use a number."))
  579. return
  580. if tool_dia is None:
  581. self.build_ui()
  582. self.app.inform.emit(_("[WARNING_NOTCL] Please enter a tool diameter to add, in Float format."))
  583. return
  584. if tool_dia == 0:
  585. self.app.inform.emit(_("[WARNING_NOTCL] Please enter a tool diameter with non-zero value, "
  586. "in Float format."))
  587. return
  588. # construct a list of all 'tooluid' in the self.tools
  589. tool_uid_list = []
  590. for tooluid_key in self.ncc_tools:
  591. tool_uid_item = int(tooluid_key)
  592. tool_uid_list.append(tool_uid_item)
  593. # find maximum from the temp_uid, add 1 and this is the new 'tooluid'
  594. if not tool_uid_list:
  595. max_uid = 0
  596. else:
  597. max_uid = max(tool_uid_list)
  598. self.tooluid = int(max_uid + 1)
  599. tool_dias = []
  600. for k, v in self.ncc_tools.items():
  601. for tool_v in v.keys():
  602. if tool_v == 'tooldia':
  603. tool_dias.append(float('%.4f' % v[tool_v]))
  604. if float('%.4f' % tool_dia) in tool_dias:
  605. if muted is None:
  606. self.app.inform.emit(_("[WARNING_NOTCL] Adding tool cancelled. Tool already in Tool Table."))
  607. self.tools_table.itemChanged.connect(self.on_tool_edit)
  608. return
  609. else:
  610. if muted is None:
  611. self.app.inform.emit(_("[success] New tool added to Tool Table."))
  612. self.ncc_tools.update({
  613. int(self.tooluid): {
  614. 'tooldia': float('%.4f' % tool_dia),
  615. 'offset': 'Path',
  616. 'offset_value': 0.0,
  617. 'type': 'Iso',
  618. 'tool_type': 'V',
  619. 'data': dict(self.default_data),
  620. 'solid_geometry': []
  621. }
  622. })
  623. self.build_ui()
  624. def on_tool_edit(self):
  625. self.ui_disconnect()
  626. tool_dias = []
  627. for k, v in self.ncc_tools.items():
  628. for tool_v in v.keys():
  629. if tool_v == 'tooldia':
  630. tool_dias.append(float('%.4f' % v[tool_v]))
  631. for row in range(self.tools_table.rowCount()):
  632. try:
  633. new_tool_dia = float(self.tools_table.item(row, 1).text())
  634. except ValueError:
  635. # try to convert comma to decimal point. if it's still not working error message and return
  636. try:
  637. new_tool_dia = float(self.tools_table.item(row, 1).text().replace(',', '.'))
  638. except ValueError:
  639. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  640. "use a number."))
  641. return
  642. tooluid = int(self.tools_table.item(row, 3).text())
  643. # identify the tool that was edited and get it's tooluid
  644. if new_tool_dia not in tool_dias:
  645. self.ncc_tools[tooluid]['tooldia'] = new_tool_dia
  646. self.app.inform.emit(_("[success] Tool from Tool Table was edited."))
  647. self.build_ui()
  648. return
  649. else:
  650. # identify the old tool_dia and restore the text in tool table
  651. for k, v in self.ncc_tools.items():
  652. if k == tooluid:
  653. old_tool_dia = v['tooldia']
  654. break
  655. restore_dia_item = self.tools_table.item(row, 1)
  656. restore_dia_item.setText(str(old_tool_dia))
  657. self.app.inform.emit(_("[WARNING_NOTCL] Edit cancelled. "
  658. "New diameter value is already in the Tool Table."))
  659. self.build_ui()
  660. def on_tool_delete(self, rows_to_delete=None, all=None):
  661. self.ui_disconnect()
  662. deleted_tools_list = []
  663. if all:
  664. self.paint_tools.clear()
  665. self.build_ui()
  666. return
  667. if rows_to_delete:
  668. try:
  669. for row in rows_to_delete:
  670. tooluid_del = int(self.tools_table.item(row, 3).text())
  671. deleted_tools_list.append(tooluid_del)
  672. except TypeError:
  673. deleted_tools_list.append(rows_to_delete)
  674. for t in deleted_tools_list:
  675. self.ncc_tools.pop(t, None)
  676. self.build_ui()
  677. return
  678. try:
  679. if self.tools_table.selectedItems():
  680. for row_sel in self.tools_table.selectedItems():
  681. row = row_sel.row()
  682. if row < 0:
  683. continue
  684. tooluid_del = int(self.tools_table.item(row, 3).text())
  685. deleted_tools_list.append(tooluid_del)
  686. for t in deleted_tools_list:
  687. self.ncc_tools.pop(t, None)
  688. except AttributeError:
  689. self.app.inform.emit(_("[WARNING_NOTCL] Delete failed. Select a tool to delete."))
  690. return
  691. except Exception as e:
  692. log.debug(str(e))
  693. self.app.inform.emit(_("[success] Tool(s) deleted from Tool Table."))
  694. self.build_ui()
  695. def on_ncc_click(self):
  696. self.bound_obj = None
  697. self.ncc_obj = None
  698. ref_choice = self.reference_radio.get_value()
  699. if ref_choice == 'itself':
  700. self.bound_obj_name = self.object_combo.currentText()
  701. # Get source object.
  702. try:
  703. self.bound_obj = self.app.collection.get_by_name(self.bound_obj_name)
  704. except Exception as e:
  705. self.app.inform.emit(_("[ERROR_NOTCL] Could not retrieve object: %s") % self.obj_name)
  706. return "Could not retrieve object: %s" % self.obj_name
  707. self.on_ncc()
  708. elif ref_choice == 'box':
  709. self.bound_obj_name = self.box_combo.currentText()
  710. # Get source object.
  711. try:
  712. self.bound_obj = self.app.collection.get_by_name(self.bound_obj_name)
  713. except Exception as e:
  714. self.app.inform.emit(_("[ERROR_NOTCL] Could not retrieve object: %s") % self.bound_obj_name)
  715. return "Could not retrieve object: %s. Error: %s" % (self.bound_obj_name, str(e))
  716. self.on_ncc()
  717. else:
  718. self.app.inform.emit(_("[WARNING_NOTCL] Click the start point of the area."))
  719. # use the first tool in the tool table; get the diameter
  720. tooldia = float('%.4f' % float(self.tools_table.item(0, 1).text()))
  721. # To be called after clicking on the plot.
  722. def on_mouse_release(event):
  723. # do paint single only for left mouse clicks
  724. if event.button == 1:
  725. if self.first_click is False:
  726. self.first_click = True
  727. self.app.inform.emit(_("[WARNING_NOTCL] Click the end point of the paint area."))
  728. self.cursor_pos = self.app.plotcanvas.translate_coords(event.pos)
  729. if self.app.grid_status() == True:
  730. self.cursor_pos = self.app.geo_editor.snap(self.cursor_pos[0], self.cursor_pos[1])
  731. else:
  732. self.app.inform.emit(_("Zone added. Right click to finish."))
  733. self.app.delete_selection_shape()
  734. curr_pos = self.app.plotcanvas.translate_coords(event.pos)
  735. if self.app.grid_status() == True:
  736. curr_pos = self.app.geo_editor.snap(curr_pos[0], curr_pos[1])
  737. x0, y0 = self.cursor_pos[0], self.cursor_pos[1]
  738. x1, y1 = curr_pos[0], curr_pos[1]
  739. pt1 = (x0, y0)
  740. pt2 = (x1, y0)
  741. pt3 = (x1, y1)
  742. pt4 = (x0, y1)
  743. self.sel_rect.append(Polygon([pt1, pt2, pt3, pt4]))
  744. modifiers = QtWidgets.QApplication.keyboardModifiers()
  745. if modifiers == QtCore.Qt.ShiftModifier:
  746. mod_key = 'Shift'
  747. elif modifiers == QtCore.Qt.ControlModifier:
  748. mod_key = 'Control'
  749. else:
  750. mod_key = None
  751. if mod_key == self.app.defaults["global_mselect_key"]:
  752. self.first_click = False
  753. return
  754. self.app.plotcanvas.vis_disconnect('mouse_release', on_mouse_release)
  755. self.app.plotcanvas.vis_disconnect('mouse_move', on_mouse_move)
  756. self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
  757. self.app.plotcanvas.vis_connect('mouse_move', self.app.on_mouse_move_over_plot)
  758. self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
  759. self.on_ncc()
  760. elif event.button == 2 and self.first_click is False and self.mouse_is_dragging is False:
  761. self.first_click = False
  762. self.app.plotcanvas.vis_disconnect('mouse_release', on_mouse_release)
  763. self.app.plotcanvas.vis_disconnect('mouse_move', on_mouse_move)
  764. self.app.plotcanvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot)
  765. self.app.plotcanvas.vis_connect('mouse_move', self.app.on_mouse_move_over_plot)
  766. self.app.plotcanvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot)
  767. self.on_ncc()
  768. # called on mouse move
  769. def on_mouse_move(event):
  770. curr_pos = self.app.plotcanvas.translate_coords(event.pos)
  771. self.app.app_cursor.enabled = False
  772. if event.button == 2:
  773. if event.is_dragging is True:
  774. self.mouse_is_dragging = True
  775. else:
  776. self.mouse_is_dragging = False
  777. if self.app.grid_status() == True:
  778. self.app.app_cursor.enabled = True
  779. # Update cursor
  780. curr_pos = self.app.geo_editor.snap(curr_pos[0], curr_pos[1])
  781. self.app.app_cursor.set_data(np.asarray([(curr_pos[0], curr_pos[1])]),
  782. symbol='++', edge_color='black', size=20)
  783. if self.first_click:
  784. self.app.delete_selection_shape()
  785. self.app.draw_moving_selection_shape(old_coords=(self.cursor_pos[0], self.cursor_pos[1]),
  786. coords=(curr_pos[0], curr_pos[1]),
  787. face_alpha=0.0)
  788. self.app.plotcanvas.vis_disconnect('mouse_press', self.app.on_mouse_click_over_plot)
  789. self.app.plotcanvas.vis_disconnect('mouse_move', self.app.on_mouse_move_over_plot)
  790. self.app.plotcanvas.vis_disconnect('mouse_release', self.app.on_mouse_click_release_over_plot)
  791. self.app.plotcanvas.vis_connect('mouse_release', on_mouse_release)
  792. self.app.plotcanvas.vis_connect('mouse_move', on_mouse_move)
  793. def on_ncc(self):
  794. try:
  795. over = float(self.ncc_overlap_entry.get_value())
  796. except ValueError:
  797. # try to convert comma to decimal point. if it's still not working error message and return
  798. try:
  799. over = float(self.ncc_overlap_entry.get_value().replace(',', '.'))
  800. except ValueError:
  801. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  802. "use a number."))
  803. return
  804. over = over if over else self.app.defaults["tools_nccoverlap"]
  805. if over >= 1 or over < 0:
  806. self.app.inform.emit(_("[ERROR_NOTCL] Overlap value must be between "
  807. "0 (inclusive) and 1 (exclusive), "))
  808. return
  809. try:
  810. margin = float(self.ncc_margin_entry.get_value())
  811. except ValueError:
  812. # try to convert comma to decimal point. if it's still not working error message and return
  813. try:
  814. margin = float(self.ncc_margin_entry.get_value().replace(',', '.'))
  815. except ValueError:
  816. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  817. "use a number."))
  818. return
  819. margin = margin if margin is not None else float(self.app.defaults["tools_nccmargin"])
  820. try:
  821. ncc_offset_value = float(self.ncc_offset_spinner.get_value())
  822. except ValueError:
  823. self.app.inform.emit(_("[ERROR_NOTCL] Wrong value format entered, "
  824. "use a number."))
  825. return
  826. ncc_offset_value = ncc_offset_value if ncc_offset_value is not None \
  827. else float(self.app.defaults["tools_ncc_offset_value"])
  828. connect = self.ncc_connect_cb.get_value()
  829. connect = connect if connect else self.app.defaults["tools_nccconnect"]
  830. contour = self.ncc_contour_cb.get_value()
  831. contour = contour if contour else self.app.defaults["tools_ncccontour"]
  832. clearing_method = self.ncc_rest_cb.get_value()
  833. clearing_method = clearing_method if clearing_method else self.app.defaults["tools_nccrest"]
  834. pol_method = self.ncc_method_radio.get_value()
  835. pol_method = pol_method if pol_method else self.app.defaults["tools_nccmethod"]
  836. self.obj_name = self.obj_combo.currentText()
  837. # Get source object.
  838. try:
  839. self.ncc_obj = self.app.collection.get_by_name(self.obj_name)
  840. except Exception as e:
  841. self.app.inform.emit(_("[ERROR_NOTCL] Could not retrieve object: %s") % self.obj_name)
  842. return "Could not retrieve object: %s" % self.obj_name
  843. # Prepare non-copper polygons
  844. if self.reference_radio.get_value() == 'area':
  845. geo_n = self.sel_rect
  846. geo_buff_list = []
  847. for poly in geo_n:
  848. geo_buff_list.append(poly.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre))
  849. bounding_box = cascaded_union(geo_buff_list)
  850. else:
  851. geo_n = self.bound_obj.solid_geometry
  852. try:
  853. if isinstance(geo_n, MultiPolygon):
  854. env_obj = geo_n.convex_hull
  855. elif (isinstance(geo_n, MultiPolygon) and len(geo_n) == 1) or \
  856. (isinstance(geo_n, list) and len(geo_n) == 1) and isinstance(geo_n[0], Polygon):
  857. env_obj = cascaded_union(geo_n)
  858. else:
  859. env_obj = cascaded_union(geo_n)
  860. env_obj = env_obj.convex_hull
  861. bounding_box = env_obj.buffer(distance=margin, join_style=base.JOIN_STYLE.mitre)
  862. except Exception as e:
  863. log.debug("NonCopperClear.on_ncc() --> %s" % str(e))
  864. self.app.inform.emit(_("[ERROR_NOTCL] No object available."))
  865. return
  866. # calculate the empty area by subtracting the solid_geometry from the object bounding box geometry
  867. if isinstance(self.ncc_obj, FlatCAMGerber):
  868. if self.ncc_choice_offset_cb.isChecked():
  869. self.app.inform.emit(_("[WARNING_NOTCL] Buffering ..."))
  870. offseted_geo = self.ncc_obj.solid_geometry.buffer(distance=ncc_offset_value)
  871. self.app.inform.emit(_("[success] Buffering finished ..."))
  872. empty = self.get_ncc_empty_area(target=offseted_geo, boundary=bounding_box)
  873. else:
  874. empty = self.get_ncc_empty_area(target=self.ncc_obj.solid_geometry, boundary=bounding_box)
  875. elif isinstance(self.ncc_obj, FlatCAMGeometry):
  876. sol_geo = cascaded_union(self.ncc_obj.solid_geometry)
  877. if self.ncc_choice_offset_cb.isChecked():
  878. self.app.inform.emit(_("[WARNING_NOTCL] Buffering ..."))
  879. offseted_geo = sol_geo.buffer(distance=ncc_offset_value)
  880. self.app.inform.emit(_("[success] Buffering finished ..."))
  881. empty = self.get_ncc_empty_area(target=offseted_geo, boundary=bounding_box)
  882. else:
  883. empty = self.get_ncc_empty_area(target=sol_geo, boundary=bounding_box)
  884. else:
  885. self.inform.emit(_('[ERROR_NOTCL] The selected object is not suitable for copper clearing.'))
  886. return
  887. if type(empty) is Polygon:
  888. empty = MultiPolygon([empty])
  889. if empty.is_empty:
  890. self.app.inform.emit(_("[ERROR_NOTCL] Could not get the extent of the area to be non copper cleared."))
  891. return
  892. # clear non copper using standard algorithm
  893. if clearing_method is False:
  894. self.clear_non_copper(
  895. empty=empty,
  896. over=over,
  897. pol_method=pol_method,
  898. connect=connect,
  899. contour=contour
  900. )
  901. # clear non copper using rest machining algorithm
  902. else:
  903. self.clear_non_copper_rest(
  904. empty=empty,
  905. over=over,
  906. pol_method=pol_method,
  907. connect=connect,
  908. contour=contour
  909. )
  910. def clear_non_copper(self, empty, over, pol_method, outname=None, connect=True, contour=True):
  911. name = outname if outname else self.obj_name + "_ncc"
  912. # Sort tools in descending order
  913. sorted_tools = []
  914. for k, v in self.ncc_tools.items():
  915. sorted_tools.append(float('%.4f' % float(v['tooldia'])))
  916. order = self.ncc_order_radio.get_value()
  917. if order == 'fwd':
  918. sorted_tools.sort(reverse=False)
  919. elif order == 'rev':
  920. sorted_tools.sort(reverse=True)
  921. else:
  922. pass
  923. # Do job in background
  924. proc = self.app.proc_container.new(_("Clearing Non-Copper areas."))
  925. def initialize(geo_obj, app_obj):
  926. assert isinstance(geo_obj, FlatCAMGeometry), \
  927. "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
  928. cleared_geo = []
  929. # Already cleared area
  930. cleared = MultiPolygon()
  931. # flag for polygons not cleared
  932. app_obj.poly_not_cleared = False
  933. # Generate area for each tool
  934. offset = sum(sorted_tools)
  935. current_uid = int(1)
  936. tool = eval(self.app.defaults["tools_ncctools"])[0]
  937. for tool in sorted_tools:
  938. self.app.inform.emit(_('[success] Non-Copper Clearing with ToolDia = %s started.') % str(tool))
  939. cleared_geo[:] = []
  940. # Get remaining tools offset
  941. offset -= (tool - 1e-12)
  942. # Area to clear
  943. area = empty.buffer(-offset)
  944. try:
  945. area = area.difference(cleared)
  946. except Exception as e:
  947. continue
  948. # Transform area to MultiPolygon
  949. if type(area) is Polygon:
  950. area = MultiPolygon([area])
  951. if area.geoms:
  952. if len(area.geoms) > 0:
  953. for p in area.geoms:
  954. try:
  955. if pol_method == 'standard':
  956. cp = self.clear_polygon(p, tool, self.app.defaults["gerber_circle_steps"],
  957. overlap=over, contour=contour, connect=connect)
  958. elif pol_method == 'seed':
  959. cp = self.clear_polygon2(p, tool, self.app.defaults["gerber_circle_steps"],
  960. overlap=over, contour=contour, connect=connect)
  961. else:
  962. cp = self.clear_polygon3(p, tool, self.app.defaults["gerber_circle_steps"],
  963. overlap=over, contour=contour, connect=connect)
  964. if cp:
  965. cleared_geo += list(cp.get_objects())
  966. except Exception as e:
  967. log.warning("Polygon can not be cleared. %s" % str(e))
  968. app_obj.poly_not_cleared = True
  969. continue
  970. # check if there is a geometry at all in the cleared geometry
  971. if cleared_geo:
  972. # Overall cleared area
  973. cleared = empty.buffer(-offset * (1 + over)).buffer(-tool / 1.999999).buffer(
  974. tool / 1.999999)
  975. # clean-up cleared geo
  976. cleared = cleared.buffer(0)
  977. # find the tooluid associated with the current tool_dia so we know where to add the tool
  978. # solid_geometry
  979. for k, v in self.ncc_tools.items():
  980. if float('%.4f' % v['tooldia']) == float('%.4f' % tool):
  981. current_uid = int(k)
  982. # add the solid_geometry to the current too in self.paint_tools dictionary
  983. # and then reset the temporary list that stored that solid_geometry
  984. v['solid_geometry'] = deepcopy(cleared_geo)
  985. v['data']['name'] = name
  986. break
  987. geo_obj.tools[current_uid] = dict(self.ncc_tools[current_uid])
  988. else:
  989. log.debug("There are no geometries in the cleared polygon.")
  990. geo_obj.options["cnctooldia"] = str(tool)
  991. geo_obj.multigeo = True
  992. def job_thread(app_obj):
  993. try:
  994. app_obj.new_object("geometry", name, initialize)
  995. except Exception as e:
  996. proc.done()
  997. self.app.inform.emit(_('[ERROR_NOTCL] NCCTool.clear_non_copper() --> %s') % str(e))
  998. return
  999. proc.done()
  1000. if app_obj.poly_not_cleared is False:
  1001. self.app.inform.emit(_('[success] NCC Tool finished.'))
  1002. else:
  1003. self.app.inform.emit(_('[WARNING_NOTCL] NCC Tool finished but some PCB features could not be cleared. '
  1004. 'Check the result.'))
  1005. # reset the variable for next use
  1006. app_obj.poly_not_cleared = False
  1007. # focus on Selected Tab
  1008. self.app.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  1009. # Promise object with the new name
  1010. self.app.collection.promise(name)
  1011. # Background
  1012. self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
  1013. # clear copper with 'rest-machining' algorithm
  1014. def clear_non_copper_rest(self, empty, over, pol_method, outname=None, connect=True, contour=True):
  1015. name = outname if outname is not None else self.obj_name + "_ncc_rm"
  1016. # Sort tools in descending order
  1017. sorted_tools = []
  1018. for k, v in self.ncc_tools.items():
  1019. sorted_tools.append(float('%.4f' % float(v['tooldia'])))
  1020. sorted_tools.sort(reverse=True)
  1021. # Do job in background
  1022. proc = self.app.proc_container.new(_("Clearing Non-Copper areas."))
  1023. def initialize_rm(geo_obj, app_obj):
  1024. assert isinstance(geo_obj, FlatCAMGeometry), \
  1025. "Initializer expected a FlatCAMGeometry, got %s" % type(geo_obj)
  1026. cleared_geo = []
  1027. cleared_by_last_tool = []
  1028. rest_geo = []
  1029. current_uid = 1
  1030. tool = eval(self.app.defaults["tools_ncctools"])[0]
  1031. # repurposed flag for final object, geo_obj. True if it has any solid_geometry, False if not.
  1032. app_obj.poly_not_cleared = True
  1033. area = empty.buffer(0)
  1034. # Generate area for each tool
  1035. while sorted_tools:
  1036. tool = sorted_tools.pop(0)
  1037. self.app.inform.emit(_('[success] Non-Copper Rest Clearing with ToolDia = %s started.') % str(tool))
  1038. tool_used = tool - 1e-12
  1039. cleared_geo[:] = []
  1040. # Area to clear
  1041. for poly in cleared_by_last_tool:
  1042. try:
  1043. area = area.difference(poly)
  1044. except Exception as e:
  1045. pass
  1046. cleared_by_last_tool[:] = []
  1047. # Transform area to MultiPolygon
  1048. if type(area) is Polygon:
  1049. area = MultiPolygon([area])
  1050. # add the rest that was not able to be cleared previously; area is a MultyPolygon
  1051. # and rest_geo it's a list
  1052. allparts = [p.buffer(0) for p in area.geoms]
  1053. allparts += deepcopy(rest_geo)
  1054. rest_geo[:] = []
  1055. area = MultiPolygon(deepcopy(allparts))
  1056. allparts[:] = []
  1057. if area.geoms:
  1058. if len(area.geoms) > 0:
  1059. for p in area.geoms:
  1060. try:
  1061. if pol_method == 'standard':
  1062. cp = self.clear_polygon(p, tool_used, self.app.defaults["gerber_circle_steps"],
  1063. overlap=over, contour=contour, connect=connect)
  1064. elif pol_method == 'seed':
  1065. cp = self.clear_polygon2(p, tool_used,
  1066. self.app.defaults["gerber_circle_steps"],
  1067. overlap=over, contour=contour, connect=connect)
  1068. else:
  1069. cp = self.clear_polygon3(p, tool_used,
  1070. self.app.defaults["gerber_circle_steps"],
  1071. overlap=over, contour=contour, connect=connect)
  1072. cleared_geo.append(list(cp.get_objects()))
  1073. except:
  1074. log.warning("Polygon can't be cleared.")
  1075. # this polygon should be added to a list and then try clear it with a smaller tool
  1076. rest_geo.append(p)
  1077. # check if there is a geometry at all in the cleared geometry
  1078. if cleared_geo:
  1079. # Overall cleared area
  1080. cleared_area = list(self.flatten_list(cleared_geo))
  1081. # cleared = MultiPolygon([p.buffer(tool_used / 2).buffer(-tool_used / 2)
  1082. # for p in cleared_area])
  1083. # here we store the poly's already processed in the original geometry by the current tool
  1084. # into cleared_by_last_tool list
  1085. # this will be sustracted from the original geometry_to_be_cleared and make data for
  1086. # the next tool
  1087. buffer_value = tool_used / 2
  1088. for p in cleared_area:
  1089. poly = p.buffer(buffer_value)
  1090. cleared_by_last_tool.append(poly)
  1091. # find the tooluid associated with the current tool_dia so we know
  1092. # where to add the tool solid_geometry
  1093. for k, v in self.ncc_tools.items():
  1094. if float('%.4f' % v['tooldia']) == float('%.4f' % tool):
  1095. current_uid = int(k)
  1096. # add the solid_geometry to the current too in self.paint_tools dictionary
  1097. # and then reset the temporary list that stored that solid_geometry
  1098. v['solid_geometry'] = deepcopy(cleared_area)
  1099. v['data']['name'] = name
  1100. cleared_area[:] = []
  1101. break
  1102. geo_obj.tools[current_uid] = dict(self.ncc_tools[current_uid])
  1103. else:
  1104. log.debug("There are no geometries in the cleared polygon.")
  1105. geo_obj.multigeo = True
  1106. geo_obj.options["cnctooldia"] = str(tool)
  1107. # check to see if geo_obj.tools is empty
  1108. # it will be updated only if there is a solid_geometry for tools
  1109. if geo_obj.tools:
  1110. return
  1111. else:
  1112. # I will use this variable for this purpose although it was meant for something else
  1113. # signal that we have no geo in the object therefore don't create it
  1114. app_obj.poly_not_cleared = False
  1115. return "fail"
  1116. def job_thread(app_obj):
  1117. try:
  1118. app_obj.new_object("geometry", name, initialize_rm)
  1119. except Exception as e:
  1120. proc.done()
  1121. app_obj.inform.emit(_('[ERROR_NOTCL] NCCTool.clear_non_copper_rest() --> %s') % str(e))
  1122. return
  1123. if app_obj.poly_not_cleared is True:
  1124. app_obj.inform.emit('[success] NCC Tool finished.')
  1125. # focus on Selected Tab
  1126. app_obj.ui.notebook.setCurrentWidget(self.app.ui.selected_tab)
  1127. else:
  1128. app_obj.inform.emit(_('[ERROR_NOTCL] NCC Tool finished but could not clear the object '
  1129. 'with current settings.'))
  1130. # focus on Project Tab
  1131. app_obj.ui.notebook.setCurrentWidget(self.app.ui.project_tab)
  1132. proc.done()
  1133. # reset the variable for next use
  1134. app_obj.poly_not_cleared = False
  1135. # Promise object with the new name
  1136. self.app.collection.promise(name)
  1137. # Background
  1138. self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
  1139. @staticmethod
  1140. def get_ncc_empty_area(target, boundary=None):
  1141. """
  1142. Returns the complement of target geometry within
  1143. the given boundary polygon. If not specified, it defaults to
  1144. the rectangular bounding box of target geometry.
  1145. """
  1146. if boundary is None:
  1147. boundary = target.envelope
  1148. return boundary.difference(target)
  1149. def reset_fields(self):
  1150. self.object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  1151. def reset_usage(self):
  1152. self.obj_name = ""
  1153. self.ncc_obj = None
  1154. self.bound_obj = None
  1155. self.first_click = False
  1156. self.cursor_pos = None
  1157. self.mouse_is_dragging = False
  1158. self.sel_rect = []