370 lines
15 KiB
Python
370 lines
15 KiB
Python
|
from FlatCAMTool import FlatCAMTool
|
||
|
|
from copy import copy, deepcopy
|
||
|
|
from ObjectCollection import *
|
||
|
|
import time
|
||
|
|
|
||
|
|
|
||
|
|
class Panelize(FlatCAMTool):
|
||
|
|
|
||
|
|
toolName = "Panelize PCB Tool"
|
||
|
|
|
||
|
|
def __init__(self, app):
|
||
|
|
super(Panelize, self).__init__(self)
|
||
|
|
self.app = app
|
||
|
|
|
||
|
|
## Title
|
||
|
|
title_label = QtWidgets.QLabel("<font size=4><b>%s</b></font>" % self.toolName)
|
||
|
|
self.layout.addWidget(title_label)
|
||
|
|
|
||
|
|
## Form Layout
|
||
|
|
form_layout = QtWidgets.QFormLayout()
|
||
|
|
self.layout.addLayout(form_layout)
|
||
|
|
|
||
|
|
## Type of object to be panelized
|
||
|
|
self.type_obj_combo = QtWidgets.QComboBox()
|
||
|
|
self.type_obj_combo.addItem("Gerber")
|
||
|
|
self.type_obj_combo.addItem("Excellon")
|
||
|
|
self.type_obj_combo.addItem("Geometry")
|
||
|
|
|
||
|
|
self.type_obj_combo.setItemIcon(0, QtGui.QIcon("share/flatcam_icon16.png"))
|
||
|
|
self.type_obj_combo.setItemIcon(1, QtGui.QIcon("share/drill16.png"))
|
||
|
|
self.type_obj_combo.setItemIcon(2, QtGui.QIcon("share/geometry16.png"))
|
||
|
|
|
||
|
|
self.type_obj_combo_label = QtWidgets.QLabel("Object Type:")
|
||
|
|
self.type_obj_combo_label.setToolTip(
|
||
|
|
"Specify the type of object to be panelized\n"
|
||
|
|
"It can be of type: Gerber, Excellon or Geometry.\n"
|
||
|
|
"The selection here decide the type of objects that will be\n"
|
||
|
|
"in the Object combobox."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.type_obj_combo_label, self.type_obj_combo)
|
||
|
|
|
||
|
|
## Object to be panelized
|
||
|
|
self.object_combo = QtWidgets.QComboBox()
|
||
|
|
self.object_combo.setModel(self.app.collection)
|
||
|
|
self.object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
|
||
|
|
self.object_combo.setCurrentIndex(1)
|
||
|
|
self.object_label = QtWidgets.QLabel("Object:")
|
||
|
|
self.object_label.setToolTip(
|
||
|
|
"Object to be panelized. This means that it will\n"
|
||
|
|
"be duplicated in an array of rows and columns."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.object_label, self.object_combo)
|
||
|
|
|
||
|
|
## Type of Box Object to be used as an envelope for panelization
|
||
|
|
self.type_box_combo = QtWidgets.QComboBox()
|
||
|
|
self.type_box_combo.addItem("Gerber")
|
||
|
|
self.type_box_combo.addItem("Excellon")
|
||
|
|
self.type_box_combo.addItem("Geometry")
|
||
|
|
|
||
|
|
# we get rid of item1 ("Excellon") as it is not suitable for use as a "box" for panelizing
|
||
|
|
self.type_box_combo.view().setRowHidden(1, True)
|
||
|
|
self.type_box_combo.setItemIcon(0, QtGui.QIcon("share/flatcam_icon16.png"))
|
||
|
|
self.type_box_combo.setItemIcon(2, QtGui.QIcon("share/geometry16.png"))
|
||
|
|
|
||
|
|
self.type_box_combo_label = QtWidgets.QLabel("Box Type:")
|
||
|
|
self.type_box_combo_label.setToolTip(
|
||
|
|
"Specify the type of object to be used as an container for\n"
|
||
|
|
"panelization. It can be: Gerber or Geometry type.\n"
|
||
|
|
"The selection here decide the type of objects that will be\n"
|
||
|
|
"in the Box Object combobox."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.type_box_combo_label, self.type_box_combo)
|
||
|
|
|
||
|
|
## Box
|
||
|
|
self.box_combo = QtWidgets.QComboBox()
|
||
|
|
self.box_combo.setModel(self.app.collection)
|
||
|
|
self.box_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
|
||
|
|
self.box_combo.setCurrentIndex(1)
|
||
|
|
self.box_combo_label = QtWidgets.QLabel("Box Object:")
|
||
|
|
self.box_combo_label.setToolTip(
|
||
|
|
"The actual object that is used a container for the\n "
|
||
|
|
"selected object that is to be panelized."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.box_combo_label, self.box_combo)
|
||
|
|
|
||
|
|
## Spacing Columns
|
||
|
|
self.spacing_columns = FloatEntry()
|
||
|
|
self.spacing_columns.set_value(0.0)
|
||
|
|
self.spacing_columns_label = QtWidgets.QLabel("Spacing cols:")
|
||
|
|
self.spacing_columns_label.setToolTip(
|
||
|
|
"Spacing between columns of the desired panel.\n"
|
||
|
|
"In current units."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.spacing_columns_label, self.spacing_columns)
|
||
|
|
|
||
|
|
## Spacing Rows
|
||
|
|
self.spacing_rows = FloatEntry()
|
||
|
|
self.spacing_rows.set_value(0.0)
|
||
|
|
self.spacing_rows_label = QtWidgets.QLabel("Spacing rows:")
|
||
|
|
self.spacing_rows_label.setToolTip(
|
||
|
|
"Spacing between rows of the desired panel.\n"
|
||
|
|
"In current units."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.spacing_rows_label, self.spacing_rows)
|
||
|
|
|
||
|
|
## Columns
|
||
|
|
self.columns = IntEntry()
|
||
|
|
self.columns.set_value(1)
|
||
|
|
self.columns_label = QtWidgets.QLabel("Columns:")
|
||
|
|
self.columns_label.setToolTip(
|
||
|
|
"Number of columns of the desired panel"
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.columns_label, self.columns)
|
||
|
|
|
||
|
|
## Rows
|
||
|
|
self.rows = IntEntry()
|
||
|
|
self.rows.set_value(1)
|
||
|
|
self.rows_label = QtWidgets.QLabel("Rows:")
|
||
|
|
self.rows_label.setToolTip(
|
||
|
|
"Number of rows of the desired panel"
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.rows_label, self.rows)
|
||
|
|
|
||
|
|
## Constrains
|
||
|
|
self.constrain_cb = FCCheckBox("Constrain panel within:")
|
||
|
|
self.constrain_cb.setToolTip(
|
||
|
|
"Area define by DX and DY within to constrain the panel.\n"
|
||
|
|
"DX and DY values are in current units.\n"
|
||
|
|
"Regardless of how many columns and rows are desired,\n"
|
||
|
|
"the final panel will have as many columns and rows as\n"
|
||
|
|
"they fit completely within selected area."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.constrain_cb)
|
||
|
|
|
||
|
|
self.x_width_entry = FloatEntry()
|
||
|
|
self.x_width_entry.set_value(0.0)
|
||
|
|
self.x_width_lbl = QtWidgets.QLabel("Width (DX):")
|
||
|
|
self.x_width_lbl.setToolTip(
|
||
|
|
"The width (DX) within which the panel must fit.\n"
|
||
|
|
"In current units."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.x_width_lbl, self.x_width_entry)
|
||
|
|
|
||
|
|
self.y_height_entry = FloatEntry()
|
||
|
|
self.y_height_entry.set_value(0.0)
|
||
|
|
self.y_height_lbl = QtWidgets.QLabel("Height (DY):")
|
||
|
|
self.y_height_lbl.setToolTip(
|
||
|
|
"The height (DY)within which the panel must fit.\n"
|
||
|
|
"In current units."
|
||
|
|
)
|
||
|
|
form_layout.addRow(self.y_height_lbl, self.y_height_entry)
|
||
|
|
|
||
|
|
self.constrain_sel = OptionalInputSection(
|
||
|
|
self.constrain_cb, [self.x_width_lbl, self.x_width_entry, self.y_height_lbl, self.y_height_entry])
|
||
|
|
|
||
|
|
|
||
|
|
## Buttons
|
||
|
|
hlay_2 = QtWidgets.QHBoxLayout()
|
||
|
|
self.layout.addLayout(hlay_2)
|
||
|
|
|
||
|
|
hlay_2.addStretch()
|
||
|
|
self.panelize_object_button = QtWidgets.QPushButton("Panelize Object")
|
||
|
|
self.panelize_object_button.setToolTip(
|
||
|
|
"Panelize the specified object around the specified box.\n"
|
||
|
|
"In other words it creates multiple copies of the source object,\n"
|
||
|
|
"arranged in a 2D array of rows and columns."
|
||
|
|
)
|
||
|
|
hlay_2.addWidget(self.panelize_object_button)
|
||
|
|
|
||
|
|
self.layout.addStretch()
|
||
|
|
|
||
|
|
## Signals
|
||
|
|
self.panelize_object_button.clicked.connect(self.on_panelize)
|
||
|
|
self.type_obj_combo.currentIndexChanged.connect(self.on_type_obj_index_changed)
|
||
|
|
self.type_box_combo.currentIndexChanged.connect(self.on_type_box_index_changed)
|
||
|
|
|
||
|
|
# list to hold the temporary objects
|
||
|
|
self.objs = []
|
||
|
|
|
||
|
|
# final name for the panel object
|
||
|
|
self.outname = ""
|
||
|
|
|
||
|
|
# flag to signal the constrain was activated
|
||
|
|
self.constrain_flag = False
|
||
|
|
|
||
|
|
def on_type_obj_index_changed(self):
|
||
|
|
obj_type = self.type_obj_combo.currentIndex()
|
||
|
|
self.object_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
|
||
|
|
self.object_combo.setCurrentIndex(0)
|
||
|
|
|
||
|
|
def on_type_box_index_changed(self):
|
||
|
|
obj_type = self.type_box_combo.currentIndex()
|
||
|
|
self.box_combo.setRootModelIndex(self.app.collection.index(obj_type, 0, QtCore.QModelIndex()))
|
||
|
|
self.box_combo.setCurrentIndex(0)
|
||
|
|
|
||
|
|
def run(self):
|
||
|
|
FlatCAMTool.run(self)
|
||
|
|
self.app.ui.notebook.setTabText(2, "Panel. Tool")
|
||
|
|
|
||
|
|
def on_panelize(self):
|
||
|
|
name = self.object_combo.currentText()
|
||
|
|
|
||
|
|
# Get source object.
|
||
|
|
try:
|
||
|
|
obj = self.app.collection.get_by_name(str(name))
|
||
|
|
except:
|
||
|
|
self.app.inform.emit("[error_notcl]Could not retrieve object: %s" % name)
|
||
|
|
return "Could not retrieve object: %s" % name
|
||
|
|
|
||
|
|
panel_obj = obj
|
||
|
|
|
||
|
|
if panel_obj is None:
|
||
|
|
self.app.inform.emit("[error_notcl]Object not found: %s" % panel_obj)
|
||
|
|
return "Object not found: %s" % panel_obj
|
||
|
|
|
||
|
|
boxname = self.box_combo.currentText()
|
||
|
|
|
||
|
|
try:
|
||
|
|
box = self.app.collection.get_by_name(boxname)
|
||
|
|
except:
|
||
|
|
self.app.inform.emit("[error_notcl]Could not retrieve object: %s" % boxname)
|
||
|
|
return "Could not retrieve object: %s" % boxname
|
||
|
|
|
||
|
|
if box is None:
|
||
|
|
self.app.inform.emit("[warning]No object Box. Using instead %s" % panel_obj)
|
||
|
|
box = panel_obj
|
||
|
|
|
||
|
|
self.outname = name + '_panelized'
|
||
|
|
|
||
|
|
spacing_columns = self.spacing_columns.get_value()
|
||
|
|
spacing_columns = spacing_columns if spacing_columns is not None else 0
|
||
|
|
|
||
|
|
spacing_rows = self.spacing_rows.get_value()
|
||
|
|
spacing_rows = spacing_rows if spacing_rows is not None else 0
|
||
|
|
|
||
|
|
rows = self.rows.get_value()
|
||
|
|
rows = rows if rows is not None else 1
|
||
|
|
|
||
|
|
columns = self.columns.get_value()
|
||
|
|
columns = columns if columns is not None else 1
|
||
|
|
|
||
|
|
constrain_dx = self.x_width_entry.get_value()
|
||
|
|
constrain_dy = self.y_height_entry.get_value()
|
||
|
|
|
||
|
|
if 0 in {columns, rows}:
|
||
|
|
self.app.inform.emit("[error_notcl]Columns or Rows are zero value. Change them to a positive integer.")
|
||
|
|
return "Columns or Rows are zero value. Change them to a positive integer."
|
||
|
|
|
||
|
|
xmin, ymin, xmax, ymax = box.bounds()
|
||
|
|
lenghtx = xmax - xmin + spacing_columns
|
||
|
|
lenghty = ymax - ymin + spacing_rows
|
||
|
|
|
||
|
|
# check if constrain within an area is desired
|
||
|
|
if self.constrain_cb.isChecked():
|
||
|
|
panel_lengthx = ((xmax - xmin) * columns) + (spacing_columns * (columns - 1))
|
||
|
|
panel_lengthy = ((ymax - ymin) * rows) + (spacing_rows * (rows - 1))
|
||
|
|
|
||
|
|
# adjust the number of columns and/or rows so the panel will fit within the panel constraint area
|
||
|
|
if (panel_lengthx > constrain_dx) or (panel_lengthy > constrain_dy):
|
||
|
|
self.constrain_flag = True
|
||
|
|
|
||
|
|
while panel_lengthx > constrain_dx:
|
||
|
|
columns -= 1
|
||
|
|
panel_lengthx = ((xmax - xmin) * columns) + (spacing_columns * (columns - 1))
|
||
|
|
while panel_lengthy > constrain_dy:
|
||
|
|
rows -= 1
|
||
|
|
panel_lengthy = ((ymax - ymin) * rows) + (spacing_rows * (rows - 1))
|
||
|
|
|
||
|
|
def clean_temp():
|
||
|
|
# deselect all to avoid delete selected object when run delete from shell
|
||
|
|
self.app.collection.set_all_inactive()
|
||
|
|
|
||
|
|
for del_obj in self.objs:
|
||
|
|
self.app.collection.set_active(del_obj.options['name'])
|
||
|
|
self.app.on_delete()
|
||
|
|
|
||
|
|
self.objs[:] = []
|
||
|
|
|
||
|
|
def panelize():
|
||
|
|
if panel_obj is not None:
|
||
|
|
self.app.inform.emit("Generating panel ... Please wait.")
|
||
|
|
|
||
|
|
self.app.progress.emit(10)
|
||
|
|
|
||
|
|
if isinstance(panel_obj, FlatCAMExcellon):
|
||
|
|
currenty = 0.0
|
||
|
|
self.app.progress.emit(0)
|
||
|
|
|
||
|
|
def initialize_local_excellon(obj_init, app):
|
||
|
|
obj_init.tools = panel_obj.tools
|
||
|
|
# drills are offset, so they need to be deep copied
|
||
|
|
obj_init.drills = deepcopy(panel_obj.drills)
|
||
|
|
obj_init.offset([float(currentx), float(currenty)])
|
||
|
|
obj_init.create_geometry()
|
||
|
|
self.objs.append(obj_init)
|
||
|
|
|
||
|
|
self.app.progress.emit(0)
|
||
|
|
for row in range(rows):
|
||
|
|
currentx = 0.0
|
||
|
|
for col in range(columns):
|
||
|
|
local_outname = self.outname + ".tmp." + str(col) + "." + str(row)
|
||
|
|
self.app.new_object("excellon", local_outname, initialize_local_excellon, plot=False,
|
||
|
|
autoselected=False)
|
||
|
|
currentx += lenghtx
|
||
|
|
currenty += lenghty
|
||
|
|
else:
|
||
|
|
currenty = 0
|
||
|
|
self.app.progress.emit(0)
|
||
|
|
|
||
|
|
def initialize_local_geometry(obj_init, app):
|
||
|
|
obj_init.solid_geometry = panel_obj.solid_geometry
|
||
|
|
obj_init.offset([float(currentx), float(currenty)]),
|
||
|
|
self.objs.append(obj_init)
|
||
|
|
|
||
|
|
self.app.progress.emit(0)
|
||
|
|
for row in range(rows):
|
||
|
|
currentx = 0
|
||
|
|
|
||
|
|
for col in range(columns):
|
||
|
|
local_outname = self.outname + ".tmp." + str(col) + "." + str(row)
|
||
|
|
self.app.new_object("geometry", local_outname, initialize_local_geometry, plot=False,
|
||
|
|
autoselected=False)
|
||
|
|
currentx += lenghtx
|
||
|
|
currenty += lenghty
|
||
|
|
|
||
|
|
def job_init_geometry(obj_fin, app_obj):
|
||
|
|
FlatCAMGeometry.merge(self.objs, obj_fin)
|
||
|
|
|
||
|
|
def job_init_excellon(obj_fin, app_obj):
|
||
|
|
# merge expects tools to exist in the target object
|
||
|
|
obj_fin.tools = panel_obj.tools.copy()
|
||
|
|
FlatCAMExcellon.merge(self.objs, obj_fin)
|
||
|
|
|
||
|
|
if isinstance(panel_obj, FlatCAMExcellon):
|
||
|
|
self.app.progress.emit(50)
|
||
|
|
self.app.new_object("excellon", self.outname, job_init_excellon, plot=True, autoselected=True)
|
||
|
|
else:
|
||
|
|
self.app.progress.emit(50)
|
||
|
|
self.app.new_object("geometry", self.outname, job_init_geometry, plot=True, autoselected=True)
|
||
|
|
|
||
|
|
else:
|
||
|
|
self.app.inform.emit("[error_notcl] Obj is None")
|
||
|
|
return "ERROR: Obj is None"
|
||
|
|
|
||
|
|
panelize()
|
||
|
|
clean_temp()
|
||
|
|
if self.constrain_flag is False:
|
||
|
|
self.app.inform.emit("[success]Panel done...")
|
||
|
|
else:
|
||
|
|
self.constrain_flag = False
|
||
|
|
self.app.inform.emit("[warning] Too big for the constrain area. Final panel has %s columns and %s rows" %
|
||
|
|
(columns, rows))
|
||
|
|
|
||
|
|
# proc = self.app.proc_container.new("Generating panel ... Please wait.")
|
||
|
|
#
|
||
|
|
# def job_thread(app_obj):
|
||
|
|
# try:
|
||
|
|
# panelize()
|
||
|
|
# except Exception as e:
|
||
|
|
# proc.done()
|
||
|
|
# raise e
|
||
|
|
# proc.done()
|
||
|
|
#
|
||
|
|
# self.app.collection.promise(self.outname)
|
||
|
|
# self.app.worker_task.emit({'fcn': job_thread, 'params': [self.app]})
|
||
|
|
|
||
|
|
def reset_fields(self):
|
||
|
|
self.object_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
|
||
|
|
self.box_combo.setRootModelIndex(self.app.collection.index(0, 0, QtCore.QModelIndex()))
|