106 lines
3.7 KiB
Python
106 lines
3.7 KiB
Python
import collections
|
|
from tclCommands.TclCommand import TclCommand
|
|
|
|
from shapely.ops import unary_union
|
|
|
|
import gettext
|
|
import appTranslation as fcTranslate
|
|
import builtins
|
|
|
|
fcTranslate.apply_language('strings')
|
|
if '_' not in builtins.__dict__:
|
|
_ = gettext.gettext
|
|
|
|
|
|
class TclCommandBbox(TclCommand):
|
|
"""
|
|
Tcl shell command to follow a Gerber file
|
|
"""
|
|
|
|
# array of all command aliases, to be able use old names for backward compatibility (add_poly, add_polygon)
|
|
aliases = ['bounding_box', 'bbox']
|
|
|
|
description = '%s %s' % ("--", "Creates a rectangular Geometry object that surrounds the object.")
|
|
|
|
# dictionary of types from Tcl command, needs to be ordered
|
|
arg_names = collections.OrderedDict([
|
|
('name', str)
|
|
])
|
|
|
|
# dictionary of types from Tcl command, needs to be ordered , this is for options like -optionname value
|
|
option_types = collections.OrderedDict([
|
|
('outname', str),
|
|
('margin', float),
|
|
('rounded', str)
|
|
])
|
|
|
|
# array of mandatory options for current Tcl command: required = {'name','outname'}
|
|
required = ['name']
|
|
|
|
# structured help for current command, args needs to be ordered
|
|
help = {
|
|
'main': "Creates a rectangular Geometry object that surrounds the object.",
|
|
'args': collections.OrderedDict([
|
|
('name', 'Object name for which to create bounding box. String'),
|
|
('margin', "Distance of the edges of the box to the nearest polygon."
|
|
"Float number."),
|
|
('rounded', "If the bounding box has to have rounded corners their radius is equal to the margin. "
|
|
"True (1) or False (0)."),
|
|
('outname', 'Name of the resulting Geometry object. String.')
|
|
]),
|
|
'examples': ['bbox name -outname name_bbox']
|
|
}
|
|
|
|
def execute(self, args, unnamed_args):
|
|
"""
|
|
execute current TCL shell command
|
|
|
|
:param args: array of known named arguments and options
|
|
:param unnamed_args: array of other values which were passed into command
|
|
without -somename and we do not have them in known arg_names
|
|
:return: None or exception
|
|
"""
|
|
|
|
name = args['name']
|
|
|
|
if 'outname' not in args:
|
|
args['outname'] = name + "_bbox"
|
|
|
|
obj = self.app.collection.get_by_name(name)
|
|
if obj is None:
|
|
self.raise_tcl_error("%s: %s" % (_("Object not found"), name))
|
|
|
|
if obj.kind != 'gerber' and obj.kind != 'geometry':
|
|
self.raise_tcl_error('%s %s: %s.' % (
|
|
_("Expected GerberObject or GeometryObject, got"), name, type(obj)))
|
|
|
|
if 'margin' not in args:
|
|
args['margin'] = float(self.app.defaults["gerber_bboxmargin"])
|
|
margin = args['margin']
|
|
|
|
if 'rounded' in args:
|
|
try:
|
|
par = args['rounded'].capitalize()
|
|
except AttributeError:
|
|
par = args['rounded']
|
|
rounded = bool(eval(par))
|
|
else:
|
|
rounded = bool(eval(self.app.defaults["gerber_bboxrounded"]))
|
|
|
|
del args['name']
|
|
|
|
try:
|
|
def geo_init(geo_obj, app_obj):
|
|
# assert geo_obj.kind == 'geometry'
|
|
|
|
# Bounding box with rounded corners
|
|
geo = unary_union(obj.solid_geometry)
|
|
bounding_box = geo.envelope.buffer(float(margin))
|
|
if not rounded: # Remove rounded corners
|
|
bounding_box = bounding_box.envelope
|
|
geo_obj.solid_geometry = bounding_box
|
|
|
|
self.app.app_obj.new_object("geometry", args['outname'], geo_init, plot=False)
|
|
except Exception as e:
|
|
return "Operation failed: %s" % str(e)
|