01dcfb69a6
- renamed the other current files that holds 'Tools' to start with Tool like Tcl commands start with TclCommand
198 lines
7.4 KiB
Python
198 lines
7.4 KiB
Python
from PyQt4 import QtGui
|
|
from GUIElements import RadioSet, EvalEntry, LengthEntry
|
|
from FlatCAMTool import FlatCAMTool
|
|
#from FlatCAMObj import FlatCAMGerber, FlatCAMExcellon
|
|
from FlatCAMObj import *
|
|
from shapely.geometry import Point
|
|
from shapely import affinity
|
|
|
|
|
|
class DblSidedTool(FlatCAMTool):
|
|
|
|
toolName = "Double-Sided PCB Tool"
|
|
|
|
def __init__(self, app):
|
|
FlatCAMTool.__init__(self, app)
|
|
|
|
## Title
|
|
title_label = QtGui.QLabel("<font size=4><b>%s</b></font>" % self.toolName)
|
|
self.layout.addWidget(title_label)
|
|
|
|
## Form Layout
|
|
form_layout = QtGui.QFormLayout()
|
|
self.layout.addLayout(form_layout)
|
|
|
|
## Layer to mirror
|
|
self.object_combo = QtGui.QComboBox()
|
|
self.object_combo.setModel(self.app.collection.model)
|
|
self.botlay_label = QtGui.QLabel("Bottom Layer:")
|
|
self.botlay_label.setToolTip(
|
|
"Layer to be mirrorer."
|
|
)
|
|
# form_layout.addRow("Bottom Layer:", self.object_combo)
|
|
form_layout.addRow(self.botlay_label, self.object_combo)
|
|
|
|
## Axis
|
|
self.mirror_axis = RadioSet([{'label': 'X', 'value': 'X'},
|
|
{'label': 'Y', 'value': 'Y'}])
|
|
self.mirax_label = QtGui.QLabel("Mirror Axis:")
|
|
self.mirax_label.setToolTip(
|
|
"Mirror vertically (X) or horizontally (Y)."
|
|
)
|
|
# form_layout.addRow("Mirror Axis:", self.mirror_axis)
|
|
form_layout.addRow(self.mirax_label, self.mirror_axis)
|
|
|
|
## Axis Location
|
|
self.axis_location = RadioSet([{'label': 'Point', 'value': 'point'},
|
|
{'label': 'Box', 'value': 'box'}])
|
|
self.axloc_label = QtGui.QLabel("Axis Location:")
|
|
self.axloc_label.setToolTip(
|
|
"The axis should pass through a <b>point</b> or cut "
|
|
"a specified <b>box</b> (in a Geometry object) in "
|
|
"the middle."
|
|
)
|
|
# form_layout.addRow("Axis Location:", self.axis_location)
|
|
form_layout.addRow(self.axloc_label, self.axis_location)
|
|
|
|
## Point/Box
|
|
self.point_box_container = QtGui.QVBoxLayout()
|
|
self.pb_label = QtGui.QLabel("Point/Box:")
|
|
self.pb_label.setToolTip(
|
|
"Specify the point (x, y) through which the mirror axis "
|
|
"passes or the Geometry object containing a rectangle "
|
|
"that the mirror axis cuts in half."
|
|
)
|
|
# form_layout.addRow("Point/Box:", self.point_box_container)
|
|
form_layout.addRow(self.pb_label, self.point_box_container)
|
|
|
|
self.point = EvalEntry()
|
|
self.point_box_container.addWidget(self.point)
|
|
self.box_combo = QtGui.QComboBox()
|
|
self.box_combo.setModel(self.app.collection.model)
|
|
self.point_box_container.addWidget(self.box_combo)
|
|
self.box_combo.hide()
|
|
|
|
## Alignment holes
|
|
self.alignment_holes = EvalEntry()
|
|
self.ah_label = QtGui.QLabel("Alignment Holes:")
|
|
self.ah_label.setToolTip(
|
|
"Alignment holes (x1, y1), (x2, y2), ... "
|
|
"on one side of the mirror axis."
|
|
)
|
|
form_layout.addRow(self.ah_label, self.alignment_holes)
|
|
|
|
## Drill diameter for alignment holes
|
|
self.drill_dia = LengthEntry()
|
|
self.dd_label = QtGui.QLabel("Drill diam.:")
|
|
self.dd_label.setToolTip(
|
|
"Diameter of the drill for the "
|
|
"alignment holes."
|
|
)
|
|
form_layout.addRow(self.dd_label, self.drill_dia)
|
|
|
|
## Buttons
|
|
hlay = QtGui.QHBoxLayout()
|
|
self.layout.addLayout(hlay)
|
|
hlay.addStretch()
|
|
self.create_alignment_hole_button = QtGui.QPushButton("Create Alignment Drill")
|
|
self.create_alignment_hole_button.setToolTip(
|
|
"Creates an Excellon Object containing the "
|
|
"specified alignment holes and their mirror "
|
|
"images."
|
|
)
|
|
self.mirror_object_button = QtGui.QPushButton("Mirror Object")
|
|
self.mirror_object_button.setToolTip(
|
|
"Mirrors (flips) the specified object around "
|
|
"the specified axis. Does not create a new "
|
|
"object, but modifies it."
|
|
)
|
|
hlay.addWidget(self.create_alignment_hole_button)
|
|
hlay.addWidget(self.mirror_object_button)
|
|
|
|
self.layout.addStretch()
|
|
|
|
## Signals
|
|
self.create_alignment_hole_button.clicked.connect(self.on_create_alignment_holes)
|
|
self.mirror_object_button.clicked.connect(self.on_mirror)
|
|
|
|
self.axis_location.group_toggle_fn = self.on_toggle_pointbox
|
|
|
|
## Initialize form
|
|
self.mirror_axis.set_value('X')
|
|
self.axis_location.set_value('point')
|
|
|
|
def on_create_alignment_holes(self):
|
|
axis = self.mirror_axis.get_value()
|
|
mode = self.axis_location.get_value()
|
|
|
|
if mode == "point":
|
|
px, py = self.point.get_value()
|
|
else:
|
|
selection_index = self.box_combo.currentIndex()
|
|
bb_obj = self.app.collection.object_list[selection_index] # TODO: Direct access??
|
|
xmin, ymin, xmax, ymax = bb_obj.bounds()
|
|
px = 0.5 * (xmin + xmax)
|
|
py = 0.5 * (ymin + ymax)
|
|
|
|
xscale, yscale = {"X": (1.0, -1.0), "Y": (-1.0, 1.0)}[axis]
|
|
|
|
dia = self.drill_dia.get_value()
|
|
tools = {"1": {"C": dia}}
|
|
|
|
# holes = self.alignment_holes.get_value()
|
|
holes = eval('[{}]'.format(self.alignment_holes.text()))
|
|
drills = []
|
|
|
|
for hole in holes:
|
|
point = Point(hole)
|
|
point_mirror = affinity.scale(point, xscale, yscale, origin=(px, py))
|
|
drills.append({"point": point, "tool": "1"})
|
|
drills.append({"point": point_mirror, "tool": "1"})
|
|
|
|
def obj_init(obj_inst, app_inst):
|
|
obj_inst.tools = tools
|
|
obj_inst.drills = drills
|
|
obj_inst.create_geometry()
|
|
|
|
self.app.new_object("excellon", "Alignment Drills", obj_init)
|
|
|
|
def on_mirror(self):
|
|
selection_index = self.object_combo.currentIndex()
|
|
fcobj = self.app.collection.object_list[selection_index]
|
|
|
|
# For now, lets limit to Gerbers and Excellons.
|
|
# assert isinstance(gerb, FlatCAMGerber)
|
|
if not isinstance(fcobj, FlatCAMGerber) and \
|
|
not isinstance(fcobj, FlatCAMExcellon) and \
|
|
not isinstance(fcobj, FlatCAMGeometry):
|
|
self.app.inform.emit("ERROR: Only Gerber, Excellon and Geometry objects can be mirrored.")
|
|
return
|
|
|
|
axis = self.mirror_axis.get_value()
|
|
mode = self.axis_location.get_value()
|
|
|
|
if mode == "point":
|
|
px, py = self.point.get_value()
|
|
else:
|
|
selection_index = self.box_combo.currentIndex()
|
|
bb_obj = self.app.collection.object_list[selection_index] # TODO: Direct access??
|
|
xmin, ymin, xmax, ymax = bb_obj.bounds()
|
|
px = 0.5 * (xmin + xmax)
|
|
py = 0.5 * (ymin + ymax)
|
|
|
|
# Ensure that the selected object will display when it is mirrored.
|
|
# If an object's plot setting is False it will still be available in
|
|
# the combo box. If the plot is not enforced to True then the user
|
|
# gets no feedback of the operation.
|
|
fcobj.options["plot"] = True;
|
|
|
|
fcobj.mirror(axis, [px, py])
|
|
fcobj.plot()
|
|
|
|
def on_toggle_pointbox(self):
|
|
if self.axis_location.get_value() == "point":
|
|
self.point.show()
|
|
self.box_combo.hide()
|
|
else:
|
|
self.point.hide()
|
|
self.box_combo.show() |