Files
flatcam-wsl/tclCommands/TclCommandSetOrigin.py

123 lines
4.6 KiB
Python

# ##########################################################
# FlatCAM: 2D Post-processing for Manufacturing #
# File Author: Marius Adrian Stanciu (c) #
# Date: 8/17/2019 #
# MIT Licence #
# ##########################################################
from tclCommands.TclCommand import TclCommand
import collections
from camlib import get_bounds
import logging
import gettext
import appTranslation as fcTranslate
import builtins
fcTranslate.apply_language('strings')
if '_' not in builtins.__dict__:
_ = gettext.gettext
log = logging.getLogger('base')
class TclCommandSetOrigin(TclCommand):
"""
Tcl shell command to set the origin to zero or to a specified location for all loaded objects in FlatCAM.
example:
"""
# List of all command aliases, to be able use old names for backward compatibility (add_poly, add_polygon)
aliases = ['set_origin', 'origin']
description = '%s %s' % ("--", "Set the origin at the specified x,y location.")
# Dictionary of types from Tcl command, needs to be ordered
arg_names = collections.OrderedDict([
('loc', str)
])
# Dictionary of types from Tcl command, needs to be ordered , this is for options like -optionname value
option_types = collections.OrderedDict([
('auto', str),
('obj_name', str)
])
# array of mandatory options for current Tcl command: required = {'name','outname'}
required = []
# structured help for current command, args needs to be ordered
help = {
'main': "Will set the origin at the specified x,y location.\n"
"If it is called without arguments it will set origin at (0, 0)",
'args': collections.OrderedDict([
('loc', 'Location to offset all the selected objects. NO SPACES ALLOWED in X and Y pair.\n'
'Use like this: 2,3'),
('auto', 'If set to True it will set the origin to the minimum x, y of the object selection bounding box.'
'-auto=True is not correct but -auto 1 or -auto True is correct. True (1) or False (0).'),
('obj_name', 'If used and followed by an object name that is already loaded, the origin for that object\n'
'will be set at: "loc" - if used else the command will not be applied')
]),
'examples': ['set_origin 3,2', 'set_origin -auto 1', 'origin', 'set_origin 10,10 -obj_name "my_top_gerber"']
}
def execute(self, args, unnamed_args):
"""
:param args:
:param unnamed_args:
:return:
"""
loc = []
if 'auto' in args:
if bool(args['auto']) is True:
objs = self.app.collection.get_list()
minx, miny, __, ___ = get_bounds(objs)
loc.append(0 - minx)
loc.append(0 - miny)
else:
loc = [0, 0]
elif 'loc' in args:
try:
location = [float(eval(coord)) for coord in str(args['loc']).split(",") if coord != '']
except AttributeError as e:
self.app.log.debug("TclCommandSetOrigin.execute --> %s" % str(e))
location = (0, 0)
if len(location) != 2:
self.raise_tcl_error('%s: %s' % (
_("Expected a pair of (x, y) coordinates. Got"), str(len(location))))
return 'fail'
if 'obj_name' not in args:
objs = self.app.collection.get_list()
minx, miny, __, ___ = get_bounds(objs)
loc.append(location[0] - minx)
loc.append(location[1] - miny)
else:
object_name = args['obj_name']
obj = self.app.collection.get_by_name(object_name)
minx, miny, __, ___ = get_bounds([obj])
loc.append(location[0] - minx)
loc.append(location[1] - miny)
obj.offset((loc[0], loc[1]))
msg = '[success] Tcl %s %s %s: %s' % (
_("Done."), _("Offsetting..."), object_name, '{0:.4f}, {0:.4f}'.format(loc[0], loc[1]))
self.app.inform_shell.emit(msg)
return
else:
loc = [0, 0]
self.app.on_set_zero_click(event=None, location=loc, noplot=True, use_thread=False)
msg = '[success] Tcl %s: %s' % (_('Origin set by offsetting all loaded objects with '),
'{0:.4f}, {0:.4f}'.format(loc[0], loc[1]))
self.app.inform_shell.emit(msg)