import wx
import sys
from copy import deepcopy
from sas.dataloader.loader import Loader
from sas.guiframe.utils import check_float
_BOX_WIDTH = 60
if sys.platform.count("win32")>0:
_STATICBOX_WIDTH = 450
PANEL_WIDTH = 500
PANEL_HEIGHT = 290
FONT_VARIANT = 0
else:
_STATICBOX_WIDTH = 480
PANEL_WIDTH = 530
PANEL_HEIGHT = 320
FONT_VARIANT = 1
class ApertureDialog(wx.Dialog):
[docs] def __init__(self, parent=None, manager=None, aperture=None, *args, **kwds):
"""
Dialog allows to enter values for aperture
"""
kwds['size'] = (PANEL_WIDTH, PANEL_HEIGHT)
kwds['title'] = "Aperture Editor"
wx.Dialog.__init__(self, parent=parent, *args, **kwds)
self.parent = parent
self.manager = manager
self._aperture = aperture
self._reset_aperture = deepcopy(aperture)
self._notes = ""
#self_description = "Edit aperture"
#Attributes for panel
self.aperture_name_tcl = None
self.main_sizer = None
self.box_aperture = None
self.boxsizer_aperture = None
self.name_sizer = None
self.name_sizer = None
self.size_name_tcl = None
self.type_sizer = None
self.distance_sizer = None
self.size_name_sizer = None
self.aperture_size_unit_tcl = None
self.aperture_size_sizer = None
self.button_sizer = None
self.aperture_name_tcl = None
self.type_tcl = None
self.distance_tcl = None
self.distance_unit_tcl = None
self.x_aperture_size_tcl = None
self.y_aperture_size_tcl = None
self.z_aperture_size_tcl = None
self.bt_apply = None
self.bt_cancel = None
self.bt_close = None
self._do_layout()
self.set_values()
def _define_structure(self):
"""
define initial sizer
"""
self.main_sizer = wx.BoxSizer(wx.VERTICAL)
self.box_aperture = wx.StaticBox(self, -1, str("Aperture"))
self.boxsizer_aperture = wx.StaticBoxSizer(self.box_aperture,
wx.VERTICAL)
self.name_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.type_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.distance_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.size_name_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.aperture_size_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
def _layout_name(self):
"""
Do the layout for aperture name related widgets
"""
#Aperture name [string]
aperture_name_txt = wx.StaticText(self, -1, 'Aperture Name : ')
self.aperture_name_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH*5, 20),
style=0)
self.name_sizer.AddMany([(aperture_name_txt, 0, wx.LEFT|wx.RIGHT, 10),
(self.aperture_name_tcl, 0, wx.EXPAND)])
def _layout_type(self):
"""
Do the layout for aperture type related widgets
"""
#Aperture type [string]
type_txt = wx.StaticText(self, -1, 'Type: ')
self.type_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
self.type_sizer.AddMany([(type_txt, 0, wx.LEFT|wx.RIGHT, 10),
(self.type_tcl, 0, wx.LEFT, 20)])
def _layout_distance(self):
"""
Do the layout for aperture distance related widgets
"""
#Aperture distance [float]
distance_txt = wx.StaticText(self, -1, 'Distance:')
self.distance_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20),
style=0)
distance_unit_txt = wx.StaticText(self, -1, 'Unit: ')
self.distance_unit_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH, 20),
style=0)
self.distance_sizer.AddMany([(distance_txt, 0, wx.LEFT|wx.RIGHT, 10),
(self.distance_tcl, 0, wx.LEFT, 10),
(distance_unit_txt, 0, wx.LEFT|wx.RIGHT, 10),
(self.distance_unit_tcl, 0, wx.EXPAND)])
def _layout_size_name(self):
"""
Do the layout for size name related widgets
"""
# Size name [string]
size_name_txt = wx.StaticText(self, -1, 'Size Name : ')
self.size_name_tcl = wx.TextCtrl(self, -1, size=(_BOX_WIDTH*5, 20),
style=0)
self.size_name_sizer.AddMany([(size_name_txt, 0, wx.LEFT|wx.RIGHT, 10),
(self.size_name_tcl, 0, wx.EXPAND)])
def _layout_size(self):
"""
Do the layout for aperture size related widgets
"""
#Aperture size [Vector]
aperture_size_txt = wx.StaticText(self, -1, 'Size:')
x_aperture_size_txt = wx.StaticText(self, -1, 'x = ')
self.x_aperture_size_tcl = wx.TextCtrl(self, -1,
size=(_BOX_WIDTH,20), style=0)
y_aperture_size_txt = wx.StaticText(self, -1, 'y = ')
self.y_aperture_size_tcl = wx.TextCtrl(self, -1,
size=(_BOX_WIDTH,20), style=0)
z_aperture_size_txt = wx.StaticText(self, -1, 'z = ')
self.z_aperture_size_tcl = wx.TextCtrl(self, -1,
size=(_BOX_WIDTH,20), style=0)
aperture_size_unit_txt = wx.StaticText(self, -1, 'Unit: ')
self.aperture_size_unit_tcl = wx.TextCtrl(self, -1,
size=(_BOX_WIDTH,20), style=0)
self.aperture_size_sizer.AddMany([(aperture_size_txt,
0, wx.LEFT|wx.RIGHT, 10),
(x_aperture_size_txt, 0, wx.LEFT, 17),
(self.x_aperture_size_tcl, 0, wx.RIGHT, 10),
(y_aperture_size_txt, 0, wx.EXPAND),
(self.y_aperture_size_tcl, 0, wx.RIGHT, 10),
(z_aperture_size_txt, 0, wx.EXPAND),
(self.z_aperture_size_tcl, 0, wx.RIGHT, 10),
(aperture_size_unit_txt, 0, wx.EXPAND),
(self.aperture_size_unit_tcl, 0, wx.RIGHT, 10)])
def _layout_button(self):
"""
Do the layout for the button widgets
"""
self.bt_apply = wx.Button(self, -1,'Apply')
self.bt_apply.Bind(wx.EVT_BUTTON, self.on_click_apply)
self.bt_apply.SetToolTipString("Apply current changes to aperture.")
self.bt_cancel = wx.Button(self, -1,'Cancel')
self.bt_cancel.SetToolTipString("Cancel current changes.")
self.bt_cancel.Bind(wx.EVT_BUTTON, self.on_click_cancel)
self.bt_close = wx.Button(self, wx.ID_CANCEL,'Close')
self.bt_close.SetToolTipString("Close window.")
self.button_sizer.AddMany([(self.bt_apply, 0, wx.LEFT, 200),
(self.bt_cancel, 0, wx.LEFT, 10),
(self.bt_close, 0, wx.LEFT, 10)])
def _do_layout(self ):#, data=None):
"""
Draw the current panel
"""
self._define_structure()
self._layout_name()
self._layout_type()
self._layout_distance()
self._layout_size_name()
self._layout_size()
self._layout_button()
self.boxsizer_aperture.AddMany([(self.name_sizer, 0,
wx.EXPAND|wx.TOP|wx.BOTTOM, 5),
(self.type_sizer, 0,
wx.EXPAND|wx.TOP|wx.BOTTOM, 5),
(self.distance_sizer, 0,
wx.EXPAND|wx.TOP|wx.BOTTOM, 5),
(self.size_name_sizer, 0,
wx.EXPAND|wx.TOP|wx.BOTTOM, 5),
(self.aperture_size_sizer, 0,
wx.EXPAND|wx.TOP|wx.BOTTOM, 5)])
self.main_sizer.AddMany([(self.boxsizer_aperture, 0, wx.ALL, 10),
(self.button_sizer, 0,
wx.EXPAND|wx.TOP|wx.BOTTOM, 5)])
self.SetSizer(self.main_sizer)
self.SetAutoLayout(True)
def set_manager(self, manager):
[docs] """
Set manager of this window
"""
self.manager = manager
def reset_aperture(self):
[docs] """
put the default value of the detector back to the current aperture
"""
self._aperture.name = self._reset_aperture.name
self._aperture.type = self._reset_aperture.type
self._aperture.size_name = self._reset_aperture.size_name
self._aperture.size.x = self._reset_aperture.size.x
self._aperture.size.y = self._reset_aperture.size.y
self._aperture.size.z = self._reset_aperture.size.z
self._aperture.size_unit = self._reset_aperture.size_unit
self._aperture.distance = self._reset_aperture.distance
self._aperture.distance_unit = self._reset_aperture.distance_unit
def set_values(self):
[docs] """
take the aperture values of the current data and display them
through the panel
"""
aperture = self._aperture
#Name
self.aperture_name_tcl.SetValue(str(aperture.name))
#Type
self.type_tcl.SetValue(str(aperture.type))
#distance
self.distance_tcl.SetValue(str(aperture.distance))
#distance unit
self.distance_unit_tcl.SetValue(str(aperture.distance_unit))
#Size name
self.size_name_tcl.SetValue(str(aperture.size_name))
#Aperture size as a vector
x, y, z = aperture.size.x, aperture.size.y, aperture.size.z
self.x_aperture_size_tcl.SetValue(str(x))
self.y_aperture_size_tcl.SetValue(str(y))
self.z_aperture_size_tcl.SetValue(str(z))
self.aperture_size_unit_tcl.SetValue(str(aperture.size_unit))
def get_aperture(self):
[docs] """
return the current aperture
"""
return self._aperture
def get_notes(self):
[docs] """
return notes
"""
return self._notes
def on_change_name(self):
[docs] """
Change name
"""
#Change the name of the aperture
name = self.aperture_name_tcl.GetValue().lstrip().rstrip()
if name == "":
name = str(None)
if self._aperture.name != name:
self._notes += "Change sample 's "
self._notes += "name from %s to %s \n"% (self._aperture.name, name)
self._aperture.name = name
def on_change_type(self):
[docs] """
Change aperture type
"""
#Change type
type = self.type_tcl.GetValue().lstrip().rstrip()
self._aperture.type = type
self._notes += " Change type from"
self._notes += " %s to %s \n"% (self._aperture.type, type)
def on_change_distance(self):
[docs] """
Change distance of the aperture
"""
#Change distance
distance = self.distance_tcl.GetValue().lstrip().rstrip()
if distance == "" or distance == str(None):
distance = None
self._aperture.distance = distance
else:
if check_float(self.distance_tcl):
if self._aperture.distance != float(distance):
self._notes += "Change distance from "
self._notes += "%s to %s \n"% (self._aperture.distance,
distance)
self._aperture.distance = float(distance)
else:
self._notes += "Error: Expected a float for distance "
self._notes += "won't changes distance from "
self._notes += "%s to %s"% (self._aperture.distance, distance)
#change the distance unit
unit = self.distance_unit_tcl.GetValue().lstrip().rstrip()
if self._aperture.distance_unit != unit:
self._notes += " Change distance 's unit from "
self._notes += "%s to %s"% (self._aperture.distance_unit, unit)
def on_change_size_name(self):
[docs] """
Change the size's name
"""
#Change size name
size_name = self.size_name_tcl.GetValue().lstrip().rstrip()
self._aperture.size_name = size_name
self._notes += " Change size name from"
self._notes += " %s to %s \n"% (self._aperture.size_name, size_name)
def on_change_size(self):
[docs] """
Change aperture size
"""
#Change x coordinate
x_aperture_size = self.x_aperture_size_tcl.GetValue().lstrip().rstrip()
if x_aperture_size == "" or x_aperture_size == str(None):
x_aperture_size = None
else:
if check_float(self.x_aperture_size_tcl):
if self._aperture.size.x != float(x_aperture_size) :
self._notes += "Change x of aperture size from "
self._notes += "%s to %s \n"% (self._aperture.size.x,
x_aperture_size)
self._aperture.aperture_size.x = float(x_aperture_size)
else:
self._notes += "Error: Expected a"
self._notes += " float for the aperture size 's x "
self._notes += "won't changes x aperture size from "
self._notes += "%s to %s"% (self._aperture.size.x,
x_aperture_size)
#Change y coordinate
y_aperture_size = self.y_aperture_size_tcl.GetValue().lstrip().rstrip()
if y_aperture_size == "" or y_aperture_size == str(None):
y_aperture_size = None
self._aperture.size.y = y_aperture_size
else:
if check_float(self.y_aperture_size_tcl):
if self._aperture.size.y != float(y_aperture_size):
self._notes += "Change y of aperture size from "
self._notes += "%s to %s \n"% (self._aperture.size.y,
y_aperture_size)
self._aperture.size.y = float(y_aperture_size)
else:
self._notes += "Error: Expected a float for the"
self._notes += " aperture size's y "
self._notes += "won't changes y aperture size from "
self._notes += "%s to %s"% (self._aperture.size.y,
y_aperture_size)
#Change z coordinate
z_aperture_size = self.z_aperture_size_tcl.GetValue().lstrip().rstrip()
if z_aperture_size == "" or z_aperture_size == str(None):
z_aperture_size = None
self._aperture.size.z = z_aperture_size
else:
if check_float(self.z_aperture_size_tcl):
if self._aperture.size.z != float(z_aperture_size):
self._notes += "Change z of aperture size from "
self._notes += "%s to %s \n"% (self._aperture.size.z,
z_aperture_size)
self._aperture.size.z = float(z_aperture_size)
else:
self._notes += "Error: Expected a float for the offset 's x "
self._notes += "won't changes z aperture size from "
self._notes += "%s to %s"% (self._aperture.size.z,
z_aperture_size)
#change the aperture center unit
unit = self.aperture_size_unit_tcl.GetValue().lstrip().rstrip()
if self._aperture.size_unit != unit:
self._notes += " Change aperture size's unit from "
self._notes += "%s to %s"% (self._aperture.size_unit, unit)
self._aperture.size_unit = unit
def on_click_apply(self, event):
[docs] """
Apply user values to the aperture
"""
self.on_change_name()
self.on_change_type()
self.on_change_distance()
self.on_change_size_name()
self.on_change_size()
self.set_values()
if self.manager is not None:
self.manager.set_aperture(self._aperture)
if event is not None:
event.Skip()
def on_click_cancel(self, event):
[docs] """
reset the current aperture to its initial values
"""
self.reset_aperture()
self.set_values()
if self.manager is not None:
self.manager.set_aperture(self._aperture)
if event is not None:
event.Skip()
if __name__ == "__main__":
app = wx.App()
# Instantiate a loader
loader = Loader()
# Load data
from sas.dataloader.data_info import Aperture
ap = Aperture()
dlg = ApertureDialog(aperture=ap)
dlg.ShowModal()
app.MainLoop()