hooke/ui/gui/dialog/points.py

   1 # Copyright (C) 2010 W. Trevor King <wking@drexel.edu>
   2 #
   3 # This file is part of Hooke.
   4 #
   5 # Hooke is free software: you can redistribute it and/or modify it
   6 # under the terms of the GNU Lesser General Public License as
   7 # published by the Free Software Foundation, either version 3 of the
   8 # License, or (at your option) any later version.
   9 #
  10 # Hooke is distributed in the hope that it will be useful, but WITHOUT
  11 # ANY WARRANTY; without even the implied warranty of MERCHANTABILITY
  12 # or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU Lesser General
  13 # Public License for more details.
  14 #
  15 # You should have received a copy of the GNU Lesser General Public
  16 # License along with Hooke.  If not, see
  17 # <http://www.gnu.org/licenses/>.
  18
  19 import wx
  20
  21
  22 def measure_N_points(hooke_frame, N, message='', block=0):
  23     '''
  24     General helper function for N-points measurements
  25     By default, measurements are done on the retraction
  26     '''
  27     if message:
  28         dialog = wx.MessageDialog(None, message, 'Info', wx.OK)
  29         dialog.ShowModal()
  30
  31     figure = self.GetActiveFigure()
  32
  33     xvector = self.displayed_plot.curves[block].x
  34     yvector = self.displayed_plot.curves[block].y
  35
  36     clicked_points = figure.ginput(N, timeout=-1, show_clicks=True)
  37
  38     points = []
  39     for clicked_point in clicked_points:
  40         point = lib.clickedpoint.ClickedPoint()
  41         point.absolute_coords = clicked_point[0], clicked_point[1]
  42         point.dest = 0
  43         #TODO: make this optional?
  44         #so far, the clicked point is taken, not the corresponding data point
  45         point.find_graph_coords(xvector, yvector)
  46         point.is_line_edge = True
  47         point.is_marker = True
  48         points.append(point)
  49     return points