ToolPanelize.py 37 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544545546547548549550551552553554555556557558559560561562563564565566567568569570571572573574575576577578579580581582583584585586587588589590591592593594595596597598599600601602603604605606607608609610611612613614615616617618619620621622623624625626627628629630631632633634635636637638639640641642643644645646647648649650651652653654655656657658659660661662663664665666667668669670671672673674675676677678679680681682683684685686687688689690691692693694695696697698699700701702703704705706707708709710711712713714715716717718719720721722723724725726727728729730731732733734735736737738739740741742743744745746747748749750751752753754755756757758759760761762763764765766767768769770771772773774775776777778779780781782783784785786787788789790791792793794795796797798
  1. # ##########################################################
  2. # FlatCAM: 2D Post-processing for Manufacturing #
  3. # File Author: Marius Adrian Stanciu (c) #
  4. # Date: 3/10/2019 #
  5. # MIT Licence #
  6. # ##########################################################
  7. from PyQt5 import QtWidgets, QtGui, QtCore
  8. from FlatCAMTool import FlatCAMTool
  9. from flatcamGUI.GUIElements import FCSpinner, FCDoubleSpinner, RadioSet, FCCheckBox, OptionalInputSection
  10. from FlatCAMObj import FlatCAMGeometry, FlatCAMGerber, FlatCAMExcellon
  11. import FlatCAMApp
  12. from copy import deepcopy
  13. # from ObjectCollection import *
  14. import numpy as np
  15. import shapely.affinity as affinity
  16. import gettext
  17. import FlatCAMTranslation as fcTranslate
  18. import builtins
  19. import logging
  20. fcTranslate.apply_language('strings')
  21. if '_' not in builtins.__dict__:
  22. _ = gettext.gettext
  23. log = logging.getLogger('base')
  24. class Panelize(FlatCAMTool):
  25. toolName = _("Panelize PCB")
  26. def __init__(self, app):
  27. super(Panelize, self).__init__(self)
  28. self.app = app
  29. # ## Title
  30. title_label = QtWidgets.QLabel("%s" % self.toolName)
  31. title_label.setStyleSheet("""
  32. QLabel
  33. {
  34. font-size: 16px;
  35. font-weight: bold;
  36. }
  37. """)
  38. self.layout.addWidget(title_label)
  39. # Form Layout
  40. form_layout_0 = QtWidgets.QFormLayout()
  41. self.layout.addLayout(form_layout_0)
  42. # Type of object to be panelized
  43. self.type_obj_combo = QtWidgets.QComboBox()
  44. self.type_obj_combo.addItem("Gerber")
  45. self.type_obj_combo.addItem("Excellon")
  46. self.type_obj_combo.addItem("Geometry")
  47. self.type_obj_combo.setItemIcon(0, QtGui.QIcon("share/flatcam_icon16.png"))
  48. self.type_obj_combo.setItemIcon(1, QtGui.QIcon("share/drill16.png"))
  49. self.type_obj_combo.setItemIcon(2, QtGui.QIcon("share/geometry16.png"))
  50. self.type_obj_combo_label = QtWidgets.QLabel('%s:' % _("Object Type"))
  51. self.type_obj_combo_label.setToolTip(
  52. _("Specify the type of object to be panelized\n"
  53. "It can be of type: Gerber, Excellon or Geometry.\n"
  54. "The selection here decide the type of objects that will be\n"
  55. "in the Object combobox.")
  56. )
  57. form_layout_0.addRow(self.type_obj_combo_label, self.type_obj_combo)
  58. # Object to be panelized
  59. self.object_combo = QtWidgets.QComboBox()
  60. self.object_combo.setModel(self.app.collection)
  61. self.object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  62. self.object_combo.setCurrentIndex(1)
  63. self.object_label = QtWidgets.QLabel('%s:' % _("Object"))
  64. self.object_label.setToolTip(
  65. _("Object to be panelized. This means that it will\n"
  66. "be duplicated in an array of rows and columns.")
  67. )
  68. form_layout_0.addRow(self.object_label, self.object_combo)
  69. form_layout_0.addRow(QtWidgets.QLabel(""))
  70. # Form Layout
  71. form_layout = QtWidgets.QFormLayout()
  72. self.layout.addLayout(form_layout)
  73. # Type of box Panel object
  74. self.reference_radio = RadioSet([{'label': _('Object'), 'value': 'object'},
  75. {'label': _('Bounding Box'), 'value': 'bbox'}])
  76. self.box_label = QtWidgets.QLabel("<b>%s:</b>" % _("Penelization Reference"))
  77. self.box_label.setToolTip(
  78. _("Choose the reference for panelization:\n"
  79. "- Object = the bounding box of a different object\n"
  80. "- Bounding Box = the bounding box of the object to be panelized\n"
  81. "\n"
  82. "The reference is useful when doing panelization for more than one\n"
  83. "object. The spacings (really offsets) will be applied in reference\n"
  84. "to this reference object therefore maintaining the panelized\n"
  85. "objects in sync.")
  86. )
  87. form_layout.addRow(self.box_label)
  88. form_layout.addRow(self.reference_radio)
  89. # Type of Box Object to be used as an envelope for panelization
  90. self.type_box_combo = QtWidgets.QComboBox()
  91. self.type_box_combo.addItem("Gerber")
  92. self.type_box_combo.addItem("Excellon")
  93. self.type_box_combo.addItem("Geometry")
  94. # we get rid of item1 ("Excellon") as it is not suitable for use as a "box" for panelizing
  95. self.type_box_combo.view().setRowHidden(1, True)
  96. self.type_box_combo.setItemIcon(0, QtGui.QIcon("share/flatcam_icon16.png"))
  97. self.type_box_combo.setItemIcon(2, QtGui.QIcon("share/geometry16.png"))
  98. self.type_box_combo_label = QtWidgets.QLabel('%s:' % _("Box Type"))
  99. self.type_box_combo_label.setToolTip(
  100. _("Specify the type of object to be used as an container for\n"
  101. "panelization. It can be: Gerber or Geometry type.\n"
  102. "The selection here decide the type of objects that will be\n"
  103. "in the Box Object combobox.")
  104. )
  105. form_layout.addRow(self.type_box_combo_label, self.type_box_combo)
  106. # Box
  107. self.box_combo = QtWidgets.QComboBox()
  108. self.box_combo.setModel(self.app.collection)
  109. self.box_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  110. self.box_combo.setCurrentIndex(1)
  111. self.box_combo_label = QtWidgets.QLabel('%s:' % _("Box Object"))
  112. self.box_combo_label.setToolTip(
  113. _("The actual object that is used a container for the\n "
  114. "selected object that is to be panelized.")
  115. )
  116. form_layout.addRow(self.box_combo_label, self.box_combo)
  117. form_layout.addRow(QtWidgets.QLabel(""))
  118. panel_data_label = QtWidgets.QLabel("<b>%s:</b>" % _("Panel Data"))
  119. panel_data_label.setToolTip(
  120. _("This informations will shape the resulting panel.\n"
  121. "The number of rows and columns will set how many\n"
  122. "duplicates of the original geometry will be generated.\n"
  123. "\n"
  124. "The spacings will set the distance between any two\n"
  125. "elements of the panel array.")
  126. )
  127. form_layout.addRow(panel_data_label)
  128. # Spacing Columns
  129. self.spacing_columns = FCDoubleSpinner()
  130. self.spacing_columns.set_range(0, 9999)
  131. self.spacing_columns.set_precision(4)
  132. self.spacing_columns_label = QtWidgets.QLabel('%s:' % _("Spacing cols"))
  133. self.spacing_columns_label.setToolTip(
  134. _("Spacing between columns of the desired panel.\n"
  135. "In current units.")
  136. )
  137. form_layout.addRow(self.spacing_columns_label, self.spacing_columns)
  138. # Spacing Rows
  139. self.spacing_rows = FCDoubleSpinner()
  140. self.spacing_rows.set_range(0, 9999)
  141. self.spacing_rows.set_precision(4)
  142. self.spacing_rows_label = QtWidgets.QLabel('%s:' % _("Spacing rows"))
  143. self.spacing_rows_label.setToolTip(
  144. _("Spacing between rows of the desired panel.\n"
  145. "In current units.")
  146. )
  147. form_layout.addRow(self.spacing_rows_label, self.spacing_rows)
  148. # Columns
  149. self.columns = FCSpinner()
  150. self.columns.set_range(0, 9999)
  151. self.columns_label = QtWidgets.QLabel('%s:' % _("Columns"))
  152. self.columns_label.setToolTip(
  153. _("Number of columns of the desired panel")
  154. )
  155. form_layout.addRow(self.columns_label, self.columns)
  156. # Rows
  157. self.rows = FCSpinner()
  158. self.rows.set_range(0, 9999)
  159. self.rows_label = QtWidgets.QLabel('%s:' % _("Rows"))
  160. self.rows_label.setToolTip(
  161. _("Number of rows of the desired panel")
  162. )
  163. form_layout.addRow(self.rows_label, self.rows)
  164. form_layout.addRow(QtWidgets.QLabel(""))
  165. # Type of resulting Panel object
  166. self.panel_type_radio = RadioSet([{'label': _('Gerber'), 'value': 'gerber'},
  167. {'label': _('Geo'), 'value': 'geometry'}])
  168. self.panel_type_label = QtWidgets.QLabel("<b>%s:</b>" % _("Panel Type"))
  169. self.panel_type_label.setToolTip(
  170. _("Choose the type of object for the panel object:\n"
  171. "- Geometry\n"
  172. "- Gerber")
  173. )
  174. form_layout.addRow(self.panel_type_label)
  175. form_layout.addRow(self.panel_type_radio)
  176. # Constrains
  177. self.constrain_cb = FCCheckBox('%s:' % _("Constrain panel within"))
  178. self.constrain_cb.setToolTip(
  179. _("Area define by DX and DY within to constrain the panel.\n"
  180. "DX and DY values are in current units.\n"
  181. "Regardless of how many columns and rows are desired,\n"
  182. "the final panel will have as many columns and rows as\n"
  183. "they fit completely within selected area.")
  184. )
  185. form_layout.addRow(self.constrain_cb)
  186. self.x_width_entry = FCDoubleSpinner()
  187. self.x_width_entry.set_precision(4)
  188. self.x_width_entry.set_range(0, 9999)
  189. self.x_width_lbl = QtWidgets.QLabel('%s:' % _("Width (DX)"))
  190. self.x_width_lbl.setToolTip(
  191. _("The width (DX) within which the panel must fit.\n"
  192. "In current units.")
  193. )
  194. form_layout.addRow(self.x_width_lbl, self.x_width_entry)
  195. self.y_height_entry = FCDoubleSpinner()
  196. self.y_height_entry.set_range(0, 9999)
  197. self.y_height_entry.set_precision(4)
  198. self.y_height_lbl = QtWidgets.QLabel('%s:' % _("Height (DY)"))
  199. self.y_height_lbl.setToolTip(
  200. _("The height (DY)within which the panel must fit.\n"
  201. "In current units.")
  202. )
  203. form_layout.addRow(self.y_height_lbl, self.y_height_entry)
  204. self.constrain_sel = OptionalInputSection(
  205. self.constrain_cb, [self.x_width_lbl, self.x_width_entry, self.y_height_lbl, self.y_height_entry])
  206. # Buttons
  207. hlay_2 = QtWidgets.QHBoxLayout()
  208. self.layout.addLayout(hlay_2)
  209. hlay_2.addStretch()
  210. self.panelize_object_button = QtWidgets.QPushButton(_("Panelize Object"))
  211. self.panelize_object_button.setToolTip(
  212. _("Panelize the specified object around the specified box.\n"
  213. "In other words it creates multiple copies of the source object,\n"
  214. "arranged in a 2D array of rows and columns.")
  215. )
  216. hlay_2.addWidget(self.panelize_object_button)
  217. self.layout.addStretch()
  218. # Signals
  219. self.reference_radio.activated_custom.connect(self.on_reference_radio_changed)
  220. self.panelize_object_button.clicked.connect(self.on_panelize)
  221. self.type_obj_combo.currentIndexChanged.connect(self.on_type_obj_index_changed)
  222. self.type_box_combo.currentIndexChanged.connect(self.on_type_box_index_changed)
  223. # list to hold the temporary objects
  224. self.objs = []
  225. # final name for the panel object
  226. self.outname = ""
  227. # flag to signal the constrain was activated
  228. self.constrain_flag = False
  229. def run(self, toggle=True):
  230. self.app.report_usage("ToolPanelize()")
  231. if toggle:
  232. # if the splitter is hidden, display it, else hide it but only if the current widget is the same
  233. if self.app.ui.splitter.sizes()[0] == 0:
  234. self.app.ui.splitter.setSizes([1, 1])
  235. else:
  236. try:
  237. if self.app.ui.tool_scroll_area.widget().objectName() == self.toolName:
  238. # if tab is populated with the tool but it does not have the focus, focus on it
  239. if not self.app.ui.notebook.currentWidget() is self.app.ui.tool_tab:
  240. # focus on Tool Tab
  241. self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab)
  242. else:
  243. self.app.ui.splitter.setSizes([0, 1])
  244. except AttributeError:
  245. pass
  246. else:
  247. if self.app.ui.splitter.sizes()[0] == 0:
  248. self.app.ui.splitter.setSizes([1, 1])
  249. FlatCAMTool.run(self)
  250. self.set_tool_ui()
  251. self.app.ui.notebook.setTabText(2, _("Panel. Tool"))
  252. def install(self, icon=None, separator=None, **kwargs):
  253. FlatCAMTool.install(self, icon, separator, shortcut='ALT+Z', **kwargs)
  254. def set_tool_ui(self):
  255. self.reset_fields()
  256. self.reference_radio.set_value('bbox')
  257. sp_c = self.app.defaults["tools_panelize_spacing_columns"] if \
  258. self.app.defaults["tools_panelize_spacing_columns"] else 0.0
  259. self.spacing_columns.set_value(float(sp_c))
  260. sp_r = self.app.defaults["tools_panelize_spacing_rows"] if \
  261. self.app.defaults["tools_panelize_spacing_rows"] else 0.0
  262. self.spacing_rows.set_value(float(sp_r))
  263. rr = self.app.defaults["tools_panelize_rows"] if \
  264. self.app.defaults["tools_panelize_rows"] else 0.0
  265. self.rows.set_value(int(rr))
  266. cc = self.app.defaults["tools_panelize_columns"] if \
  267. self.app.defaults["tools_panelize_columns"] else 0.0
  268. self.columns.set_value(int(cc))
  269. c_cb = self.app.defaults["tools_panelize_constrain"] if \
  270. self.app.defaults["tools_panelize_constrain"] else False
  271. self.constrain_cb.set_value(c_cb)
  272. x_w = self.app.defaults["tools_panelize_constrainx"] if \
  273. self.app.defaults["tools_panelize_constrainx"] else 0.0
  274. self.x_width_entry.set_value(float(x_w))
  275. y_w = self.app.defaults["tools_panelize_constrainy"] if \
  276. self.app.defaults["tools_panelize_constrainy"] else 0.0
  277. self.y_height_entry.set_value(float(y_w))
  278. panel_type = self.app.defaults["tools_panelize_panel_type"] if \
  279. self.app.defaults["tools_panelize_panel_type"] else 'gerber'
  280. self.panel_type_radio.set_value(panel_type)
  281. def on_type_obj_index_changed(self):
  282. obj_type = self.type_obj_combo.currentIndex()
  283. self.object_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  284. self.object_combo.setCurrentIndex(0)
  285. # hide the panel type for Excellons, the panel can be only of type Geometry
  286. if self.type_obj_combo.currentText() != 'Excellon':
  287. self.panel_type_label.setDisabled(False)
  288. self.panel_type_radio.setDisabled(False)
  289. else:
  290. self.panel_type_label.setDisabled(True)
  291. self.panel_type_radio.setDisabled(True)
  292. self.panel_type_radio.set_value('geometry')
  293. def on_type_box_index_changed(self):
  294. obj_type = self.type_box_combo.currentIndex()
  295. self.box_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
  296. self.box_combo.setCurrentIndex(0)
  297. def on_reference_radio_changed(self, current_val):
  298. if current_val == 'object':
  299. self.type_box_combo.setDisabled(False)
  300. self.type_box_combo_label.setDisabled(False)
  301. self.box_combo.setDisabled(False)
  302. self.box_combo_label.setDisabled(False)
  303. else:
  304. self.type_box_combo.setDisabled(True)
  305. self.type_box_combo_label.setDisabled(True)
  306. self.box_combo.setDisabled(True)
  307. self.box_combo_label.setDisabled(True)
  308. def on_panelize(self):
  309. name = self.object_combo.currentText()
  310. # Get source object.
  311. try:
  312. panel_obj = self.app.collection.get_by_name(str(name))
  313. except Exception as e:
  314. log.debug("Panelize.on_panelize() --> %s" % str(e))
  315. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  316. (_("Could not retrieve object"), name))
  317. return "Could not retrieve object: %s" % name
  318. if panel_obj is None:
  319. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  320. (_("Object not found"), panel_obj))
  321. return "Object not found: %s" % panel_obj
  322. boxname = self.box_combo.currentText()
  323. try:
  324. box = self.app.collection.get_by_name(boxname)
  325. except Exception as e:
  326. log.debug("Panelize.on_panelize() --> %s" % str(e))
  327. self.app.inform.emit('[ERROR_NOTCL] %s: %s' %
  328. (_("Could not retrieve object"), boxname))
  329. return "Could not retrieve object: %s" % boxname
  330. if box is None:
  331. self.app.inform.emit('[WARNING_NOTCL]%s: %s' %
  332. (_("No object Box. Using instead"), panel_obj))
  333. self.reference_radio.set_value('bbox')
  334. if self.reference_radio.get_value() == 'bbox':
  335. box = panel_obj
  336. self.outname = name + '_panelized'
  337. spacing_columns = float(self.spacing_columns.get_value())
  338. spacing_columns = spacing_columns if spacing_columns is not None else 0
  339. spacing_rows = float(self.spacing_rows.get_value())
  340. spacing_rows = spacing_rows if spacing_rows is not None else 0
  341. rows = int(self.rows.get_value())
  342. rows = rows if rows is not None else 1
  343. columns = int(self.columns.get_value())
  344. columns = columns if columns is not None else 1
  345. constrain_dx = float(self.x_width_entry.get_value())
  346. constrain_dy = float(self.y_height_entry.get_value())
  347. panel_type = str(self.panel_type_radio.get_value())
  348. if 0 in {columns, rows}:
  349. self.app.inform.emit('[ERROR_NOTCL] %s' %
  350. _("Columns or Rows are zero value. Change them to a positive integer."))
  351. return "Columns or Rows are zero value. Change them to a positive integer."
  352. xmin, ymin, xmax, ymax = box.bounds()
  353. lenghtx = xmax - xmin + spacing_columns
  354. lenghty = ymax - ymin + spacing_rows
  355. # check if constrain within an area is desired
  356. if self.constrain_cb.isChecked():
  357. panel_lengthx = ((xmax - xmin) * columns) + (spacing_columns * (columns - 1))
  358. panel_lengthy = ((ymax - ymin) * rows) + (spacing_rows * (rows - 1))
  359. # adjust the number of columns and/or rows so the panel will fit within the panel constraint area
  360. if (panel_lengthx > constrain_dx) or (panel_lengthy > constrain_dy):
  361. self.constrain_flag = True
  362. while panel_lengthx > constrain_dx:
  363. columns -= 1
  364. panel_lengthx = ((xmax - xmin) * columns) + (spacing_columns * (columns - 1))
  365. while panel_lengthy > constrain_dy:
  366. rows -= 1
  367. panel_lengthy = ((ymax - ymin) * rows) + (spacing_rows * (rows - 1))
  368. if isinstance(panel_obj, FlatCAMExcellon) or isinstance(panel_obj, FlatCAMGeometry):
  369. # make a copy of the panelized Excellon or Geometry tools
  370. copied_tools = dict()
  371. for tt, tt_val in list(panel_obj.tools.items()):
  372. copied_tools[tt] = deepcopy(tt_val)
  373. if isinstance(panel_obj, FlatCAMGerber):
  374. # make a copy of the panelized Gerber apertures
  375. copied_apertures = dict()
  376. for tt, tt_val in list(panel_obj.apertures.items()):
  377. copied_apertures[tt] = deepcopy(tt_val)
  378. def panelize_2():
  379. if panel_obj is not None:
  380. self.app.inform.emit(_("Generating panel ... "))
  381. self.app.progress.emit(0)
  382. def job_init_excellon(obj_fin, app_obj):
  383. currenty = 0.0
  384. self.app.progress.emit(10)
  385. obj_fin.tools = copied_tools
  386. obj_fin.drills = []
  387. obj_fin.slots = []
  388. obj_fin.solid_geometry = []
  389. for option in panel_obj.options:
  390. if option is not 'name':
  391. try:
  392. obj_fin.options[option] = panel_obj.options[option]
  393. except KeyError:
  394. log.warning("Failed to copy option. %s" % str(option))
  395. geo_len_drills = len(panel_obj.drills) if panel_obj.drills else 0
  396. geo_len_slots = len(panel_obj.slots) if panel_obj.slots else 0
  397. element = 0
  398. for row in range(rows):
  399. currentx = 0.0
  400. for col in range(columns):
  401. element += 1
  402. old_disp_number = 0
  403. if panel_obj.drills:
  404. drill_nr = 0
  405. for tool_dict in panel_obj.drills:
  406. if self.app.abort_flag:
  407. # graceful abort requested by the user
  408. raise FlatCAMApp.GracefulException
  409. point_offseted = affinity.translate(tool_dict['point'], currentx, currenty)
  410. obj_fin.drills.append(
  411. {
  412. "point": point_offseted,
  413. "tool": tool_dict['tool']
  414. }
  415. )
  416. drill_nr += 1
  417. disp_number = int(np.interp(drill_nr, [0, geo_len_drills], [0, 100]))
  418. if old_disp_number < disp_number <= 100:
  419. self.app.proc_container.update_view_text(' %s: %d D:%d%%' %
  420. (_("Copy"),
  421. int(element),
  422. disp_number))
  423. old_disp_number = disp_number
  424. if panel_obj.slots:
  425. slot_nr = 0
  426. for tool_dict in panel_obj.slots:
  427. if self.app.abort_flag:
  428. # graceful abort requested by the user
  429. raise FlatCAMApp.GracefulException
  430. start_offseted = affinity.translate(tool_dict['start'], currentx, currenty)
  431. stop_offseted = affinity.translate(tool_dict['stop'], currentx, currenty)
  432. obj_fin.slots.append(
  433. {
  434. "start": start_offseted,
  435. "stop": stop_offseted,
  436. "tool": tool_dict['tool']
  437. }
  438. )
  439. slot_nr += 1
  440. disp_number = int(np.interp(slot_nr, [0, geo_len_slots], [0, 100]))
  441. if old_disp_number < disp_number <= 100:
  442. self.app.proc_container.update_view_text(' %s: %d S:%d%%' %
  443. (_("Copy"),
  444. int(element),
  445. disp_number))
  446. old_disp_number = disp_number
  447. currentx += lenghtx
  448. currenty += lenghty
  449. obj_fin.create_geometry()
  450. obj_fin.zeros = panel_obj.zeros
  451. obj_fin.units = panel_obj.units
  452. self.app.proc_container.update_view_text('')
  453. def job_init_geometry(obj_fin, app_obj):
  454. currentx = 0.0
  455. currenty = 0.0
  456. def translate_recursion(geom):
  457. if type(geom) == list:
  458. geoms = list()
  459. for local_geom in geom:
  460. res_geo = translate_recursion(local_geom)
  461. try:
  462. geoms += res_geo
  463. except TypeError:
  464. geoms.append(res_geo)
  465. return geoms
  466. else:
  467. return affinity.translate(geom, xoff=currentx, yoff=currenty)
  468. obj_fin.solid_geometry = []
  469. # create the initial structure on which to create the panel
  470. if isinstance(panel_obj, FlatCAMGeometry):
  471. obj_fin.multigeo = panel_obj.multigeo
  472. obj_fin.tools = copied_tools
  473. if panel_obj.multigeo is True:
  474. for tool in panel_obj.tools:
  475. obj_fin.tools[tool]['solid_geometry'][:] = []
  476. elif isinstance(panel_obj, FlatCAMGerber):
  477. obj_fin.apertures = copied_apertures
  478. for ap in obj_fin.apertures:
  479. obj_fin.apertures[ap]['geometry'] = list()
  480. # find the number of polygons in the source solid_geometry
  481. geo_len = 0
  482. if isinstance(panel_obj, FlatCAMGeometry):
  483. if panel_obj.multigeo is True:
  484. for tool in panel_obj.tools:
  485. try:
  486. for pol in panel_obj.tools[tool]['solid_geometry']:
  487. geo_len += 1
  488. except TypeError:
  489. geo_len = 1
  490. else:
  491. try:
  492. for pol in panel_obj.solid_geometry:
  493. geo_len += 1
  494. except TypeError:
  495. geo_len = 1
  496. elif isinstance(panel_obj, FlatCAMGerber):
  497. for ap in panel_obj.apertures:
  498. for elem in panel_obj.apertures[ap]['geometry']:
  499. geo_len += 1
  500. self.app.progress.emit(0)
  501. element = 0
  502. for row in range(rows):
  503. currentx = 0.0
  504. for col in range(columns):
  505. element += 1
  506. old_disp_number = 0
  507. if isinstance(panel_obj, FlatCAMGeometry):
  508. if panel_obj.multigeo is True:
  509. for tool in panel_obj.tools:
  510. if self.app.abort_flag:
  511. # graceful abort requested by the user
  512. raise FlatCAMApp.GracefulException
  513. # geo = translate_recursion(panel_obj.tools[tool]['solid_geometry'])
  514. # if isinstance(geo, list):
  515. # obj_fin.tools[tool]['solid_geometry'] += geo
  516. # else:
  517. # obj_fin.tools[tool]['solid_geometry'].append(geo)
  518. # calculate the number of polygons
  519. geo_len = len(panel_obj.tools[tool]['solid_geometry'])
  520. pol_nr = 0
  521. for geo_el in panel_obj.tools[tool]['solid_geometry']:
  522. trans_geo = translate_recursion(geo_el)
  523. obj_fin.tools[tool]['solid_geometry'].append(trans_geo)
  524. pol_nr += 1
  525. disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
  526. if old_disp_number < disp_number <= 100:
  527. self.app.proc_container.update_view_text(' %s: %d %d%%' %
  528. (_("Copy"),
  529. int(element),
  530. disp_number))
  531. old_disp_number = disp_number
  532. else:
  533. # geo = translate_recursion(panel_obj.solid_geometry)
  534. # if isinstance(geo, list):
  535. # obj_fin.solid_geometry += geo
  536. # else:
  537. # obj_fin.solid_geometry.append(geo)
  538. if self.app.abort_flag:
  539. # graceful abort requested by the user
  540. raise FlatCAMApp.GracefulException
  541. try:
  542. # calculate the number of polygons
  543. geo_len = len(panel_obj.solid_geometry)
  544. except TypeError:
  545. geo_len = 1
  546. pol_nr = 0
  547. try:
  548. for geo_el in panel_obj.solid_geometry:
  549. if self.app.abort_flag:
  550. # graceful abort requested by the user
  551. raise FlatCAMApp.GracefulException
  552. trans_geo = translate_recursion(geo_el)
  553. obj_fin.solid_geometry.append(trans_geo)
  554. pol_nr += 1
  555. disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
  556. if old_disp_number < disp_number <= 100:
  557. self.app.proc_container.update_view_text(' %s: %d %d%%' %
  558. (_("Copy"),
  559. int(element),
  560. disp_number))
  561. old_disp_number = disp_number
  562. except TypeError:
  563. trans_geo = translate_recursion(panel_obj.solid_geometry)
  564. obj_fin.solid_geometry.append(trans_geo)
  565. else:
  566. # geo = translate_recursion(panel_obj.solid_geometry)
  567. # if isinstance(geo, list):
  568. # obj_fin.solid_geometry += geo
  569. # else:
  570. # obj_fin.solid_geometry.append(geo)
  571. if self.app.abort_flag:
  572. # graceful abort requested by the user
  573. raise FlatCAMApp.GracefulException
  574. try:
  575. for geo_el in panel_obj.solid_geometry:
  576. if self.app.abort_flag:
  577. # graceful abort requested by the user
  578. raise FlatCAMApp.GracefulException
  579. trans_geo = translate_recursion(geo_el)
  580. obj_fin.solid_geometry.append(trans_geo)
  581. except TypeError:
  582. trans_geo = translate_recursion(panel_obj.solid_geometry)
  583. obj_fin.solid_geometry.append(trans_geo)
  584. for apid in panel_obj.apertures:
  585. if self.app.abort_flag:
  586. # graceful abort requested by the user
  587. raise FlatCAMApp.GracefulException
  588. try:
  589. # calculate the number of polygons
  590. geo_len = len(panel_obj.apertures[apid]['geometry'])
  591. except TypeError:
  592. geo_len = 1
  593. pol_nr = 0
  594. for el in panel_obj.apertures[apid]['geometry']:
  595. if self.app.abort_flag:
  596. # graceful abort requested by the user
  597. raise FlatCAMApp.GracefulException
  598. new_el = dict()
  599. if 'solid' in el:
  600. geo_aper = translate_recursion(el['solid'])
  601. new_el['solid'] = geo_aper
  602. if 'clear' in el:
  603. geo_aper = translate_recursion(el['clear'])
  604. new_el['clear'] = geo_aper
  605. if 'follow' in el:
  606. geo_aper = translate_recursion(el['follow'])
  607. new_el['follow'] = geo_aper
  608. obj_fin.apertures[apid]['geometry'].append(deepcopy(new_el))
  609. pol_nr += 1
  610. disp_number = int(np.interp(pol_nr, [0, geo_len], [0, 100]))
  611. if old_disp_number < disp_number <= 100:
  612. self.app.proc_container.update_view_text(' %s: %d %d%%' %
  613. (_("Copy"),
  614. int(element),
  615. disp_number))
  616. old_disp_number = disp_number
  617. currentx += lenghtx
  618. currenty += lenghty
  619. if panel_type == 'gerber':
  620. self.app.inform.emit('%s' %
  621. _("Generating panel ... Adding the Gerber code."))
  622. obj_fin.source_file = self.app.export_gerber(obj_name=self.outname, filename=None,
  623. local_use=obj_fin, use_thread=False)
  624. # app_obj.log.debug("Found %s geometries. Creating a panel geometry cascaded union ..." %
  625. # len(obj_fin.solid_geometry))
  626. # obj_fin.solid_geometry = cascaded_union(obj_fin.solid_geometry)
  627. # app_obj.log.debug("Finished creating a cascaded union for the panel.")
  628. self.app.proc_container.update_view_text('')
  629. self.app.inform.emit('%s: %d' %
  630. (_("Generating panel... Spawning copies"), (int(rows * columns))))
  631. if isinstance(panel_obj, FlatCAMExcellon):
  632. self.app.progress.emit(50)
  633. self.app.new_object("excellon", self.outname, job_init_excellon, plot=True, autoselected=True)
  634. else:
  635. self.app.progress.emit(50)
  636. self.app.new_object(panel_type, self.outname, job_init_geometry,
  637. plot=True, autoselected=True)
  638. if self.constrain_flag is False:
  639. self.app.inform.emit('[success] %s' % _("Panel done..."))
  640. else:
  641. self.constrain_flag = False
  642. self.app.inform.emit(_("{text} Too big for the constrain area. "
  643. "Final panel has {col} columns and {row} rows").format(
  644. text='[WARNING] ', col=columns, row=rows))
  645. proc = self.app.proc_container.new(_("Working..."))
  646. def job_thread(app_obj):
  647. try:
  648. panelize_2()
  649. self.app.inform.emit('[success] %s' % _("Panel created successfully."))
  650. except Exception as ee:
  651. proc.done()
  652. log.debug(str(ee))
  653. return
  654. proc.done()
  655. self.app.collection.promise(self.outname)
  656. self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
  657. def reset_fields(self):
  658. self.object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
  659. self.box_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))