############################################################ # FlatCAM: 2D Post-processing for Manufacturing # # http://flatcam.org # # File Author: Marius Adrian Stanciu (c) # # Date: 3/10/2019 # # MIT Licence # ############################################################ from FlatCAMTool import FlatCAMTool from FlatCAMObj import * from flatcamGUI.VisPyVisuals import * from math import sqrt import gettext import FlatCAMTranslation as fcTranslate fcTranslate.apply_language('strings') import builtins if '_' not in builtins.__dict__: _ = gettext.gettext class Measurement(FlatCAMTool): toolName = _("Measurement") def __init__(self, app): FlatCAMTool.__init__(self, app) self.app = app self.canvas = self.app.plotcanvas self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().lower() ## Title title_label = QtWidgets.QLabel("%s
" % self.toolName) self.layout.addWidget(title_label) ## Form Layout form_layout = QtWidgets.QFormLayout() self.layout.addLayout(form_layout) self.units_label = QtWidgets.QLabel(_("Units:")) self.units_label.setToolTip(_("Those are the units in which the distance is measured.")) self.units_value = QtWidgets.QLabel("%s" % str({'mm': "METRIC (mm)", 'in': "INCH (in)"}[self.units])) self.units_value.setDisabled(True) self.start_label = QtWidgets.QLabel("%s %s:" % (_('Start'), _('Coords'))) self.start_label.setToolTip(_("This is measuring Start point coordinates.")) self.stop_label = QtWidgets.QLabel("%s %s:" % (_('Stop'), _('Coords'))) self.stop_label.setToolTip(_("This is the measuring Stop point coordinates.")) self.distance_x_label = QtWidgets.QLabel("Dx:") self.distance_x_label.setToolTip(_("This is the distance measured over the X axis.")) self.distance_y_label = QtWidgets.QLabel("Dy:") self.distance_y_label.setToolTip(_("This is the distance measured over the Y axis.")) self.total_distance_label = QtWidgets.QLabel("%s:" % _('DISTANCE')) self.total_distance_label.setToolTip(_("This is the point to point Euclidian distance.")) self.start_entry = FCEntry() self.start_entry.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter) self.start_entry.setToolTip(_("This is measuring Start point coordinates.")) self.stop_entry = FCEntry() self.stop_entry.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter) self.stop_entry.setToolTip(_("This is the measuring Stop point coordinates.")) self.distance_x_entry = FCEntry() self.distance_x_entry.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter) self.distance_x_entry.setToolTip(_("This is the distance measured over the X axis.")) self.distance_y_entry = FCEntry() self.distance_y_entry.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter) self.distance_y_entry.setToolTip(_("This is the distance measured over the Y axis.")) self.total_distance_entry = FCEntry() self.total_distance_entry.setAlignment(QtCore.Qt.AlignRight | QtCore.Qt.AlignVCenter) self.total_distance_entry.setToolTip(_("This is the point to point Euclidian distance.")) self.measure_btn = QtWidgets.QPushButton(_("Measure")) # self.measure_btn.setFixedWidth(70) self.layout.addWidget(self.measure_btn) form_layout.addRow(self.units_label, self.units_value) form_layout.addRow(self.start_label, self.start_entry) form_layout.addRow(self.stop_label, self.stop_entry) form_layout.addRow(self.distance_x_label, self.distance_x_entry) form_layout.addRow(self.distance_y_label, self.distance_y_entry) form_layout.addRow(self.total_distance_label, self.total_distance_entry) # initial view of the layout self.start_entry.set_value('(0, 0)') self.stop_entry.set_value('(0, 0)') self.distance_x_entry.set_value('0') self.distance_y_entry.set_value('0') self.total_distance_entry.set_value('0') self.layout.addStretch() self.clicked_meas = 0 self.point1 = None self.point2 = None # the default state is disabled for the Move command # self.setVisible(False) self.active = 0 self.original_call_source = 'app' # VisPy visuals self.sel_shapes = ShapeCollection(parent=self.app.plotcanvas.vispy_canvas.view.scene, layers=1) self.measure_btn.clicked.connect(lambda: self.on_measure(activate=True)) def run(self, toggle=False): self.app.report_usage("ToolMeasurement()") if self.app.tool_tab_locked is True: return self.app.ui.notebook.setTabText(2, _("Meas. Tool")) # if the splitter is hidden, display it if self.app.ui.splitter.sizes()[0] == 0: self.app.ui.splitter.setSizes([1, 1]) self.on_measure(activate=True) def install(self, icon=None, separator=None, **kwargs): FlatCAMTool.install(self, icon, separator, shortcut='CTRL+M', **kwargs) def set_tool_ui(self): # Remove anything else in the GUI self.app.ui.tool_scroll_area.takeWidget() # Put ourself in the GUI self.app.ui.tool_scroll_area.setWidget(self) # Switch notebook to tool page self.app.ui.notebook.setCurrentWidget(self.app.ui.tool_tab) self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().lower() self.app.command_active = "Measurement" # initial view of the layout self.start_entry.set_value('(0, 0)') self.stop_entry.set_value('(0, 0)') self.distance_x_entry.set_value('0') self.distance_y_entry.set_value('0') self.total_distance_entry.set_value('0') def activate_measure_tool(self): # we disconnect the mouse/key handlers from wherever the measurement tool was called self.canvas.vis_disconnect('mouse_move') self.canvas.vis_disconnect('mouse_press') self.canvas.vis_disconnect('mouse_release') # we can safely connect the app mouse events to the measurement tool self.canvas.vis_connect('mouse_move', self.on_mouse_move_meas) self.canvas.vis_connect('mouse_release', self.on_mouse_click_release) self.set_tool_ui() def deactivate_measure_tool(self): # disconnect the mouse/key events from functions of measurement tool self.canvas.vis_disconnect('mouse_move', self.on_mouse_move_meas) self.canvas.vis_disconnect('mouse_release', self.on_mouse_click_release) if self.app.call_source == 'app': self.canvas.vis_connect('mouse_move', self.app.on_mouse_move_over_plot) self.canvas.vis_connect('mouse_press', self.app.on_mouse_click_over_plot) self.canvas.vis_connect('mouse_release', self.app.on_mouse_click_release_over_plot) elif self.app.call_source == 'geo_editor': self.canvas.vis_connect('mouse_move', self.app.geo_editor.on_canvas_move) self.canvas.vis_connect('mouse_press', self.app.geo_editor.on_canvas_click) self.canvas.vis_connect('mouse_release', self.app.geo_editor.on_geo_click_release) elif self.app.call_source == 'exc_editor': self.canvas.vis_connect('mouse_move', self.app.exc_editor.on_canvas_move) self.canvas.vis_connect('mouse_press', self.app.exc_editor.on_canvas_click) self.canvas.vis_connect('mouse_release', self.app.exc_editor.on_exc_click_release) elif self.app.call_source == 'grb_editor': self.canvas.vis_connect('mouse_move', self.app.grb_editor.on_canvas_move) self.canvas.vis_connect('mouse_press', self.app.grb_editor.on_canvas_click) self.canvas.vis_connect('mouse_release', self.app.grb_editor.on_grb_click_release) self.app.ui.notebook.setTabText(2, _("Tools")) self.app.ui.notebook.setCurrentWidget(self.app.ui.project_tab) def on_measure(self, signal=None, activate=None): log.debug("Measurement.on_measure()") if activate is True: # ENABLE the Measuring TOOL self.clicked_meas = 0 self.original_call_source = copy(self.app.call_source) self.app.call_source = 'measurement' self.app.inform.emit(_("MEASURING: Click on the Start point ...")) self.units = self.app.ui.general_defaults_form.general_app_group.units_radio.get_value().lower() self.activate_measure_tool() log.debug("Measurement Tool --> tool initialized") else: # DISABLE the Measuring TOOL self.deactivate_measure_tool() self.app.call_source = copy(self.original_call_source) self.app.command_active = None # delete the measuring line self.delete_shape() log.debug("Measurement Tool --> exit tool") def on_mouse_click_release(self, event): # mouse click releases will be accepted only if the left button is clicked # this is necessary because right mouse click or middle mouse click # are used for panning on the canvas if event.button == 1: pos_canvas = self.canvas.vispy_canvas.translate_coords(event.pos) if self.clicked_meas == 0: self.clicked_meas = 1 # if GRID is active we need to get the snapped positions if self.app.grid_status() == True: pos = self.app.geo_editor.snap(pos_canvas[0], pos_canvas[1]) else: pos = pos_canvas[0], pos_canvas[1] self.point1 = pos self.start_entry.set_value("(%.4f, %.4f)" % pos) self.app.inform.emit(_("MEASURING: Click on the Destination point ...")) else: # delete the selection bounding box self.delete_shape() # if GRID is active we need to get the snapped positions if self.app.grid_status() is True: pos = self.app.geo_editor.snap(pos_canvas[0], pos_canvas[1]) else: pos = pos_canvas[0], pos_canvas[1] dx = pos[0] - self.point1[0] dy = pos[1] - self.point1[1] d = sqrt(dx ** 2 + dy ** 2) self.stop_entry.set_value("(%.4f, %.4f)" % pos) self.app.inform.emit(_("MEASURING: Result D(x) = {d_x} | D(y) = {d_y} | Distance = {d_z}").format( d_x='%4f' % abs(dx), d_y='%4f' % abs(dy), d_z='%4f' % abs(d))) self.distance_x_entry.set_value('%.4f' % abs(dx)) self.distance_y_entry.set_value('%.4f' % abs(dy)) self.total_distance_entry.set_value('%.4f' % abs(d)) # TODO: I don't understand why I have to do it twice ... but without it the mouse handlers are # TODO: are left disconnected ... self.on_measure(activate=False) self.deactivate() def on_mouse_move_meas(self, event): pos_canvas = self.canvas.vispy_canvas.translate_coords(event.pos) # if GRID is active we need to get the snapped positions if self.app.grid_status() == True: pos = self.app.geo_editor.snap(pos_canvas[0], pos_canvas[1]) self.app.app_cursor.enabled = True # Update cursor self.app.app_cursor.set_data(np.asarray([(pos[0], pos[1])]), symbol='++', edge_color='black', size=20) else: pos = pos_canvas self.app.app_enabled = False self.point2 = (pos[0], pos[1]) # update utility geometry if self.clicked_meas == 1: # first delete old shape self.delete_shape() # second draw the new shape of the utility geometry self.meas_line = LineString([self.point2, self.point1]) self.sel_shapes.add(self.meas_line, color='black', update=True, layer=0, tolerance=None) def delete_shape(self): self.sel_shapes.clear() self.sel_shapes.redraw() def set_meas_units(self, units): self.meas.units_label.setText("[" + self.app.options["units"].lower() + "]") # end of file