"""
FitPanel class contains fields allowing to display results when
fitting a model and one data
"""
import sys
import wx
import wx.lib.newevent
import numpy
import copy
import math
import time
import traceback
from sasmodels.weights import MODELS as POLYDISPERSITY_MODELS
from sas.sasgui.guiframe.events import StatusEvent, NewPlotEvent, \
PlotQrangeEvent
from sas.sasgui.guiframe.dataFitting import check_data_validity
from sas.sasgui.guiframe.utils import format_number, check_float
from sas.sasgui.guiframe.documentation_window import DocumentationWindow
from sas.sasgui.perspectives.fitting.basepage import BasicPage as BasicPage
from sas.sasgui.perspectives.fitting.basepage import PageInfoEvent as \
PageInfoEvent
from sas.sascalc.data_util.qsmearing import smear_selection
from .basepage import ModelTextCtrl
(Chi2UpdateEvent, EVT_CHI2_UPDATE) = wx.lib.newevent.NewEvent()
_BOX_WIDTH = 76
_DATA_BOX_WIDTH = 300
SMEAR_SIZE_H = 0.00
CUSTOM_MODEL = 'Plugin Models'
[docs]class FitPage(BasicPage):
"""
FitPanel class contains fields allowing to display results when
fitting a model and one data
:note: For Fit to be performed the user should check at least one parameter
on fit Panel window.
"""
def __init__(self, parent, color=None):
"""
Initialization of the Panel
"""
BasicPage.__init__(self, parent, color=color)
# draw sizer
self._fill_data_sizer()
self.is_2D = None
self.fit_started = False
self.weightbt_string = None
self.m_name = None
# get smear info from data
self._get_smear_info()
self._fill_model_sizer(self.sizer1)
self._get_defult_custom_smear()
self._fill_range_sizer()
self._set_smear(self.data)
self.Bind(EVT_CHI2_UPDATE, self.on_complete_chisqr)
# bind key event
self.Bind(wx.EVT_RIGHT_DOWN, self.on_right_down)
self._set_bookmark_flag(False)
self._set_save_flag(False)
self._set_preview_flag(False)
self._set_copy_flag(False)
self._set_paste_flag(False)
self.btFit.SetFocus()
self.enable_fit_button()
self.fill_data_combobox(data_list=self.data_list)
# create a default data for an empty panel
self.create_default_data()
self._manager.frame.Bind(wx.EVT_SET_FOCUS, self.on_set_focus)
[docs] def enable_fit_button(self):
"""
Enable fit button if data is valid and model is valid
"""
flag = check_data_validity(self.data) & (self.model is not None)
self.btFit.Enable(flag)
[docs] def on_set_focus(self, event):
"""
Override the basepage focus method to ensure the save flag is set
properly when focusing on the fit page.
"""
flag = check_data_validity(self.data) & (self.model is not None)
self._set_save_flag(flag)
self.parent.on_set_focus(event)
self.on_tap_focus()
def _fill_data_sizer(self):
"""
fill sizer 0 with data info
"""
self.data_box_description = wx.StaticBox(self, wx.ID_ANY,
'I(q) Data Source')
if check_data_validity(self.data):
dname_color = wx.BLUE
else:
dname_color = wx.RED
self.data_box_description.SetForegroundColour(dname_color)
boxsizer1 = wx.StaticBoxSizer(self.data_box_description, wx.VERTICAL)
# ----------------------------------------------------------
sizer_data = wx.BoxSizer(wx.HORIZONTAL)
self.dataSource = wx.ComboBox(self, wx.ID_ANY, style=wx.CB_READONLY)
wx.EVT_COMBOBOX(self.dataSource, wx.ID_ANY, self.on_select_data)
self.dataSource.SetMinSize((_DATA_BOX_WIDTH, -1))
sizer_data.Add(wx.StaticText(self, wx.ID_ANY, 'Name : '))
sizer_data.Add(self.dataSource)
sizer_data.Add((0, 5))
boxsizer1.Add(sizer_data, 0, wx.ALL, 10)
self.sizer0.Add(boxsizer1, 0, wx.EXPAND | wx.ALL, 10)
self.sizer0.Layout()
[docs] def enable_datasource(self):
"""
Enable or disable data source control depending on existing data
"""
if not self.data_list:
self.dataSource.Disable()
else:
self.dataSource.Enable()
[docs] def fill_data_combobox(self, data_list):
"""
Get a list of data and fill the corresponding combobox
"""
self.dataSource.Clear()
self.data_list = data_list
self.enable_datasource()
if len(data_list) > 0:
# find the maximum range covering all data
qmin, qmax, npts = self.compute_data_set_range(data_list)
self.qmin_data_set = qmin
self.qmax_data_set = qmax
self.npts_data_set = npts
self.qmin.SetValue(str(self.qmin_data_set))
self.qmax.SetValue(str(self.qmax_data_set))
self.qmin.SetBackgroundColour("white")
self.qmax.SetBackgroundColour("white")
self.qmin_x = self.qmin_data_set
self.qmax_x = self.qmax_data_set
self.state.qmin = self.qmin_x
self.state.qmax = self.qmax_x
is_data = False
for data in self.data_list:
if data is not None:
self.dataSource.Append(str(data.name), clientData=data)
if not is_data:
is_data = check_data_validity(data)
if is_data:
self.dataSource.SetSelection(0)
self.on_select_data(event=None)
if len(data_list) == 1:
self.dataSource.Disable()
[docs] def on_select_data(self, event=None):
"""
On_select_data
"""
if self.dataSource.GetCount() > 0:
pos = self.dataSource.GetSelection() if event is not None else 0
data = self.dataSource.GetClientData(pos)
self.set_data(data)
def _on_fit_complete(self):
"""
When fit is complete ,reset the fit button label.
"""
self.fit_started = False
self.set_fitbutton()
def _is_2D(self):
"""
Check if data_name is Data2D
:return: True or False
"""
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
return True
return False
def _fill_range_sizer(self):
"""
Fill the Fitting sizer on the fit panel which contains: the smearing
information (dq), the weighting information (dI or other), the plotting
range, access to the 2D mask editor, the compute, fit, and help
buttons, xi^2, number of points etc.
"""
is_2d_data = False
# Check if data is 2D
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
is_2d_data = True
title = "Fitting"
# smear messages & titles
smear_message_none = "No smearing is selected..."
smear_message_dqdata = "The dQ data is being used for smearing..."
smear_message_2d = \
"Higher accuracy is very time-expensive. Use it with care..."
smear_message_new_ssmear = \
"Please enter only the value of interest to customize smearing..."
smear_message_new_psmear = \
"Please enter a fixed percentage to be applied to all Q values..."
smear_message_2d_x_title = "<dQp>[1/A]:"
smear_message_2d_y_title = "<dQs>[1/A]:"
smear_message_pinhole_percent_title = "dQ[%]:"
smear_message_slit_height_title = "Slit height[1/A]:"
smear_message_slit_width_title = "Slit width[1/A]:"
self._get_smear_info()
# Sizers
box_description_range = wx.StaticBox(self, wx.ID_ANY, str(title))
box_description_range.SetForegroundColour(wx.BLUE)
boxsizer_range = wx.StaticBoxSizer(box_description_range, wx.VERTICAL)
self.sizer_set_smearer = wx.BoxSizer(wx.VERTICAL)
sizer_smearer = wx.BoxSizer(wx.HORIZONTAL)
self.sizer_new_smear = wx.BoxSizer(wx.HORIZONTAL)
self.sizer_set_masking = wx.BoxSizer(wx.HORIZONTAL)
sizer_chi2 = wx.BoxSizer(wx.VERTICAL)
smear_set_box = wx.StaticBox(self, wx.ID_ANY,
'Set Instrumental Smearing')
sizer_smearer_box = wx.StaticBoxSizer(smear_set_box, wx.HORIZONTAL)
sizer_smearer_box.SetMinSize((_DATA_BOX_WIDTH, 60))
weighting_set_box = wx.StaticBox(self, wx.ID_ANY,
'Set Weighting by Selecting dI Source')
weighting_box = wx.StaticBoxSizer(weighting_set_box, wx.HORIZONTAL)
sizer_weighting = wx.BoxSizer(wx.HORIZONTAL)
weighting_box.SetMinSize((_DATA_BOX_WIDTH, 40))
# Filling the sizer containing weighting info.
self.dI_noweight = wx.RadioButton(self, wx.ID_ANY,
'No Weighting', style=wx.RB_GROUP)
self.dI_didata = wx.RadioButton(self, wx.ID_ANY, 'Use dI Data')
self.dI_sqrdata = wx.RadioButton(self, wx.ID_ANY, 'Use |sqrt(I Data)|')
self.dI_idata = wx.RadioButton(self, wx.ID_ANY, 'Use |I Data|')
self.Bind(wx.EVT_RADIOBUTTON, self.onWeighting,
id=self.dI_noweight.GetId())
self.Bind(wx.EVT_RADIOBUTTON, self.onWeighting,
id=self.dI_didata.GetId())
self.Bind(wx.EVT_RADIOBUTTON, self.onWeighting,
id=self.dI_sqrdata.GetId())
self.Bind(wx.EVT_RADIOBUTTON, self.onWeighting,
id=self.dI_idata.GetId())
self.dI_noweight.SetValue(True)
# add 4 types of weighting to the sizer
sizer_weighting.Add(self.dI_noweight, 0, wx.LEFT, 10)
sizer_weighting.Add((14, 10))
sizer_weighting.Add(self.dI_didata)
sizer_weighting.Add((14, 10))
sizer_weighting.Add(self.dI_sqrdata)
sizer_weighting.Add((14, 10))
sizer_weighting.Add(self.dI_idata)
sizer_weighting.Add((10, 10))
self.dI_noweight.Enable(True)
self.dI_didata.Enable(False)
self.dI_sqrdata.Enable(False)
self.dI_idata.Enable(False)
weighting_box.Add(sizer_weighting)
# combobox for smear2d accuracy selection
self.smear_accuracy = wx.ComboBox(self, wx.ID_ANY,
size=(50, -1), style=wx.CB_READONLY)
self._set_accuracy_list()
self.smear_accuracy.SetValue(self.smear2d_accuracy)
self.smear_accuracy.SetSelection(0)
self.smear_accuracy.SetToolTipString(
"'Higher' uses more Gaussian points for smearing computation.")
wx.EVT_COMBOBOX(self.smear_accuracy, wx.ID_ANY,
self._on_select_accuracy)
# Fit button
self.btFit = wx.Button(self, self._ids.next(), 'Fit')
self.default_bt_colour = self.btFit.GetDefaultAttributes()
self.btFit.Bind(wx.EVT_BUTTON, self._onFit, id=self.btFit.GetId())
self.btFit.SetToolTipString("Start fitting.")
# General Help button
self.btFitHelp = wx.Button(self, wx.ID_ANY, 'Help')
self.btFitHelp.SetToolTipString("General fitting help.")
self.btFitHelp.Bind(wx.EVT_BUTTON, self._onFitHelp)
# Resolution Smearing Help button (for now use same technique as
# used for dI help to get tiniest possible button that works
# both on MAC and PC. Should completely rewrite the fitting sizer
# in future. This is minimum to get out release 3.1
# comment June 14, 2015 --- PDB
if sys.platform.count("win32") > 0:
size_q = (20, 15) # on PC
else:
size_q = (30, 20) # on MAC
self.btSmearHelp = wx.Button(self, wx.ID_ANY, '?',
style=wx.BU_EXACTFIT, size=size_q)
self.btSmearHelp.SetToolTipString("Resolution smearing help.")
self.btSmearHelp.Bind(wx.EVT_BUTTON, self._onSmearHelp)
# textcntrl for custom resolution
self.smear_pinhole_percent = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH - 25, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=
self.onPinholeSmear)
self.smear_slit_height = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH - 25, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self.onSlitSmear)
self.smear_slit_width = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH - 25, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self.onSlitSmear)
# smear
self.smear_data_left = BGTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH - 25, 20), style=0)
self.smear_data_left.SetValue(str(self.dq_l))
self.smear_data_right = BGTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH - 25, 20), style=0)
self.smear_data_right.SetValue(str(self.dq_r))
# set default values for smear
self.smear_pinhole_percent.SetValue(str(self.dx_percent))
self.smear_slit_height.SetValue(str(self.dxl))
self.smear_slit_width.SetValue(str(self.dxw))
# Filling the sizer containing instruments smearing info.
self.disable_smearer = wx.RadioButton(self, wx.ID_ANY,
'None', style=wx.RB_GROUP)
self.enable_smearer = wx.RadioButton(self, wx.ID_ANY, 'Use dQ Data')
# self.enable_smearer.SetToolTipString(
# "Click to use the loaded dQ data for smearing.")
self.pinhole_smearer = wx.RadioButton(self, wx.ID_ANY,
'Custom Pinhole Smear')
# self.pinhole_smearer.SetToolTipString
# ("Click to input custom resolution for pinhole smearing.")
self.slit_smearer = wx.RadioButton(self, wx.ID_ANY, 'Custom Slit Smear')
# self.slit_smearer.SetToolTipString
# ("Click to input custom resolution for slit smearing.")
self.Bind(wx.EVT_RADIOBUTTON, self.onSmear,
id=self.disable_smearer.GetId())
self.Bind(wx.EVT_RADIOBUTTON, self.onSmear,
id=self.enable_smearer.GetId())
self.Bind(wx.EVT_RADIOBUTTON, self.onPinholeSmear,
id=self.pinhole_smearer.GetId())
self.Bind(wx.EVT_RADIOBUTTON, self.onSlitSmear,
id=self.slit_smearer.GetId())
self.disable_smearer.SetValue(True)
sizer_smearer.Add(self.disable_smearer, 0, wx.LEFT, 10)
sizer_smearer.Add(self.enable_smearer)
sizer_smearer.Add(self.pinhole_smearer)
sizer_smearer.Add(self.slit_smearer)
sizer_smearer.Add(self.btSmearHelp)
sizer_smearer.Add((10, 10))
# StaticText for chi2, N(for fitting), Npts + Log/linear spacing
self.tcChi = BGTextCtrl(self, wx.ID_ANY, "-", size=(75, 20), style=0)
self.tcChi.SetToolTipString("Chi2/Npts(Fit)")
self.Npts_fit = BGTextCtrl(self, wx.ID_ANY, "-", size=(75, 20), style=0)
self.Npts_fit.SetToolTipString(
" Npts : number of points selected for fitting")
self.Npts_total = ModelTextCtrl(self, wx.ID_ANY, size=(_BOX_WIDTH, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onQrangeEnter)
self.Npts_total.SetValue(format_number(self.npts_x))
self.Npts_total.SetToolTipString(
" Total Npts : total number of data points")
# Update and Draw button
self.draw_button = wx.Button(self, self._ids.next(), 'Compute')
self.draw_button.Bind(wx.EVT_BUTTON,
self._onDraw, id=self.draw_button.GetId())
self.draw_button.SetToolTipString("Compute and Draw.")
self.points_sizer = wx.BoxSizer(wx.HORIZONTAL)
self.pointsbox = wx.CheckBox(self, wx.ID_ANY, 'Log?', (10, 10))
self.pointsbox.SetValue(False)
self.pointsbox.SetToolTipString("Check mark to use log spaced points")
wx.EVT_CHECKBOX(self, self.pointsbox.GetId(), self.select_log)
self.points_sizer.Add(wx.StaticText(self, wx.ID_ANY, 'Npts '))
self.points_sizer.Add(self.pointsbox)
box_description_1 = wx.StaticText(self, wx.ID_ANY, ' Chi2/Npts')
box_description_2 = wx.StaticText(self, wx.ID_ANY, 'Npts(Fit)')
# StaticText for smear
self.smear_description_none = wx.StaticText(self, wx.ID_ANY,
smear_message_none, style=wx.ALIGN_LEFT)
self.smear_description_dqdata = wx.StaticText(self, wx.ID_ANY,
smear_message_dqdata, style=wx.ALIGN_LEFT)
self.smear_description_type = wx.StaticText(self, wx.ID_ANY,
"Type:", style=wx.ALIGN_LEFT)
self.smear_description_accuracy_type = wx.StaticText(self, wx.ID_ANY,
"Accuracy:", style=wx.ALIGN_LEFT)
self.smear_description_smear_type = BGTextCtrl(self, wx.ID_ANY,
size=(57, 20), style=0)
self.smear_description_smear_type.SetValue(str(self.dq_l))
self.SetBackgroundColour(self.GetParent().GetBackgroundColour())
self.smear_description_2d = wx.StaticText(self, wx.ID_ANY,
smear_message_2d, style=wx.ALIGN_LEFT)
self.smear_message_new_s = wx.StaticText(self, wx.ID_ANY,
smear_message_new_ssmear, style=wx.ALIGN_LEFT)
self.smear_message_new_p = wx.StaticText(self, wx.ID_ANY,
smear_message_new_psmear, style=wx.ALIGN_LEFT)
self.smear_description_2d_x = wx.StaticText(self, wx.ID_ANY,
smear_message_2d_x_title, style=wx.ALIGN_LEFT)
self.smear_description_2d_x.SetToolTipString(
" dQp(parallel) in q_r direction.")
self.smear_description_2d_y = wx.StaticText(self, wx.ID_ANY,
smear_message_2d_y_title, style=wx.ALIGN_LEFT)
self.smear_description_2d_y.SetToolTipString(
" dQs(perpendicular) in q_phi direction.")
self.smear_description_pin_percent = wx.StaticText(self, wx.ID_ANY,
smear_message_pinhole_percent_title,
style=wx.ALIGN_LEFT)
self.smear_description_slit_height = wx.StaticText(self, wx.ID_ANY,
smear_message_slit_height_title, style=wx.ALIGN_LEFT)
self.smear_description_slit_width = wx.StaticText(self, wx.ID_ANY,
smear_message_slit_width_title, style=wx.ALIGN_LEFT)
# arrange sizers
self.sizer_set_smearer.Add(sizer_smearer)
self.sizer_set_smearer.Add((10, 10))
self.sizer_set_smearer.Add(self.smear_description_none,
0, wx.CENTER, 10)
self.sizer_set_smearer.Add(self.smear_description_dqdata,
0, wx.CENTER, 10)
self.sizer_set_smearer.Add(self.smear_description_2d,
0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_description_type,
0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_description_accuracy_type,
0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_accuracy)
self.sizer_new_smear.Add(self.smear_description_smear_type,
0, wx.CENTER, 10)
self.sizer_new_smear.Add((15, -1))
self.sizer_new_smear.Add(self.smear_description_2d_x, 0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_description_slit_height,
0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_slit_height, 0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_data_left, 0, wx.CENTER, 10)
self.sizer_new_smear.Add((20, -1))
self.sizer_new_smear.Add(self.smear_description_2d_y,
0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_description_pin_percent,
0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_description_slit_width,
0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_pinhole_percent, 0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_slit_width, 0, wx.CENTER, 10)
self.sizer_new_smear.Add(self.smear_data_right, 0, wx.CENTER, 10)
self.sizer_set_smearer.Add(self.smear_message_new_s, 0, wx.CENTER, 10)
self.sizer_set_smearer.Add(self.smear_message_new_p, 0, wx.CENTER, 10)
self.sizer_set_smearer.Add((5, 2))
self.sizer_set_smearer.Add(self.sizer_new_smear, 0, wx.CENTER, 10)
# add all to chi2 sizer
sizer_smearer_box.Add(self.sizer_set_smearer)
sizer_chi2.Add(sizer_smearer_box)
sizer_chi2.Add((-1, 5))
sizer_chi2.Add(weighting_box)
sizer_chi2.Add((-1, 5))
# hide all smear messages and textctrl
self._hide_all_smear_info()
# get smear_selection
self.current_smearer = smear_selection(self.data, self.model)
# Show only the relevant smear messages, etc
if self.current_smearer is None:
if not is_2d_data:
self.smear_description_none.Show(True)
self.enable_smearer.Disable()
else:
self.smear_description_none.Show(True)
self.slit_smearer.Disable()
if self.data is None:
self.slit_smearer.Disable()
self.pinhole_smearer.Disable()
self.enable_smearer.Disable()
else:
self._show_smear_sizer()
boxsizer_range.Add(self.sizer_set_masking)
# 2D data? default
is_2d_data = False
# check if it is 2D data
if self.data.__class__.__name__ == "Data2D" or self.enable2D:
is_2d_data = True
self.sizer5.Clear(True)
self.qmin = ModelTextCtrl(self, wx.ID_ANY, size=(_BOX_WIDTH, 20),
style=wx.TE_PROCESS_ENTER,
set_focus_callback=self.qrang_set_focus,
text_enter_callback=self._onQrangeEnter,
name='qmin')
self.qmin.SetValue(str(self.qmin_x))
q_tip = "Click outside of the axes\n to remove the lines."
qmin_tip = "Minimun value of Q.\n"
qmin_tip += q_tip
self.qmin.SetToolTipString(qmin_tip)
self.qmax = ModelTextCtrl(self, wx.ID_ANY, size=(_BOX_WIDTH, 20),
style=wx.TE_PROCESS_ENTER,
set_focus_callback=self.qrang_set_focus,
text_enter_callback=self._onQrangeEnter,
name='qmax')
self.qmax.SetValue(str(self.qmax_x))
qmax_tip = "Maximum value of Q.\n"
qmax_tip += q_tip
self.qmax.SetToolTipString(qmax_tip)
self.qmin.Bind(wx.EVT_MOUSE_EVENTS, self.qrange_click)
self.qmax.Bind(wx.EVT_MOUSE_EVENTS, self.qrange_click)
self.qmin.Bind(wx.EVT_KEY_DOWN, self.on_key)
self.qmax.Bind(wx.EVT_KEY_DOWN, self.on_key)
self.qmin.Bind(wx.EVT_TEXT, self.on_qrange_text)
self.qmax.Bind(wx.EVT_TEXT, self.on_qrange_text)
wx_id = self._ids.next()
self.reset_qrange = wx.Button(self, wx_id, 'Reset')
self.reset_qrange.Bind(wx.EVT_BUTTON, self.on_reset_clicked, id=wx_id)
self.reset_qrange.SetToolTipString("Reset Q range to the default")
sizer = wx.GridSizer(5, 5, 2, 6)
self.btEditMask = wx.Button(self, self._ids.next(), 'Editor')
self.btEditMask.Bind(wx.EVT_BUTTON, self._onMask,
id=self.btEditMask.GetId())
self.btEditMask.SetToolTipString("Edit Mask.")
self.EditMask_title = wx.StaticText(self, wx.ID_ANY, ' Masking(2D)')
sizer.Add(wx.StaticText(self, wx.ID_ANY, ' Q range'))
sizer.Add(wx.StaticText(self, wx.ID_ANY, ' Min[1/A]'))
sizer.Add(wx.StaticText(self, wx.ID_ANY, ' Max[1/A]'))
sizer.Add(self.EditMask_title)
sizer.Add((-1, 5))
sizer.Add(self.reset_qrange)
sizer.Add(self.qmin)
sizer.Add(self.qmax)
sizer.Add(self.btEditMask)
sizer.Add((-1, 5))
sizer.AddMany(5*[(-1, 5)])
sizer.Add(box_description_1, 0, 0)
sizer.Add(box_description_2, 0, 0)
sizer.Add(self.points_sizer, 0, 0)
sizer.Add(self.draw_button, 0, 0)
sizer.Add((-1, 5))
sizer.Add(self.tcChi, 0, 0)
sizer.Add(self.Npts_fit, 0, 0)
sizer.Add(self.Npts_total, 0, 0)
sizer.Add(self.btFit, 0, 0)
sizer.Add(self.btFitHelp, 0, 0)
boxsizer_range.Add(sizer_chi2)
boxsizer_range.Add(sizer)
if is_2d_data:
self.btEditMask.Enable()
self.EditMask_title.Enable()
else:
self.btEditMask.Disable()
self.EditMask_title.Disable()
# save state
self.save_current_state()
self.sizer5.Add(boxsizer_range, 0, wx.EXPAND | wx.ALL, 10)
self.sizer5.Layout()
def _set_sizer_dispersion(self):
"""
draw sizer with gaussian dispersity parameters
"""
self.fittable_param = []
self.fixed_param = []
self.orientation_params_disp = []
self.sizer4_4.Clear(True)
if self.model is None:
# no model is selected
return
if not self.enable_disp.GetValue():
# the user didn't select dispersity display
return
self._reset_dispersity()
# fill a sizer with the combobox to select dispersion type
model_disp = wx.StaticText(self, wx.ID_ANY, 'Function')
CHECK_STATE = False
ix = 0
iy = 0
disp = wx.StaticText(self, wx.ID_ANY, ' ')
self.sizer4_4.Add(disp, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
ix += 1
values = wx.StaticText(self, wx.ID_ANY, 'PD[ratio]')
polytext = "Polydispersity (= STD/mean); "
polytext += "the standard deviation over the mean value."
values.SetToolTipString(polytext)
self.sizer4_4.Add(values, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 2
if self.is_mac:
err_text = 'Error'
else:
err_text = ''
self.text_disp_1 = wx.StaticText(self, wx.ID_ANY, err_text)
self.sizer4_4.Add(self.text_disp_1, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 1
self.text_disp_min = wx.StaticText(self, wx.ID_ANY, 'Min')
self.sizer4_4.Add(self.text_disp_min, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 1
self.text_disp_max = wx.StaticText(self, wx.ID_ANY, 'Max')
self.sizer4_4.Add(self.text_disp_max, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 1
npts = wx.StaticText(self, wx.ID_ANY, 'Npts')
npts.SetToolTipString("Number of sampling points for the numerical\n\
integration over the distribution function.")
self.sizer4_4.Add(npts, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 1
nsigmas = wx.StaticText(self, wx.ID_ANY, 'Nsigs')
nsigmas.SetToolTipString("Number of sigmas between which the range\n\
of the distribution function will be used for weighting. \n\
The value '3' covers 99.5% for Gaussian distribution \n\
function. Note: Not recommended to change this value.")
self.sizer4_4.Add(nsigmas, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 1
self.sizer4_4.Add(model_disp, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
self.text_disp_max.Show(True)
self.text_disp_min.Show(True)
for item in self.model.dispersion.keys():
if not self.magnetic_on:
if item in self.model.magnetic_params:
continue
if item not in self.model.orientation_params:
if item not in self.disp_cb_dict:
self.disp_cb_dict[item] = None
name0 = "Distribution of " + item
name1 = item + ".width"
name2 = item + ".npts"
name3 = item + ".nsigmas"
if name1 not in self.model.details:
self.model.details[name1] = ["", None, None]
iy += 1
for p in self.model.dispersion[item].keys():
if p == "width":
ix = 0
cb = wx.CheckBox(self, wx.ID_ANY, name0, (10, 10))
cb.SetValue(CHECK_STATE)
cb.SetToolTipString("Check mark to fit")
wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
self.sizer4_4.Add(cb, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 5)
ix = 1
value = self.model.getParam(name1)
ctl1 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 1.3, 20),
style=wx.TE_PROCESS_ENTER)
ctl1.SetLabel('PD[ratio]')
poly_text = "Polydispersity (STD/mean) of %s\n" % item
poly_text += "STD: the standard deviation"
poly_text += " from the mean value."
ctl1.SetToolTipString(poly_text)
ctl1.SetValue(str(format_number(value, True)))
self.sizer4_4.Add(ctl1, (iy, ix), (1, 1), wx.EXPAND)
# text to show error sign
ix = 2
text2 = wx.StaticText(self, wx.ID_ANY, '+/-')
self.sizer4_4.Add(text2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
if not self.is_mac:
text2.Hide()
ix = 3
ctl2 = wx.TextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 1.3, 20),
style=0)
self.sizer4_4.Add(ctl2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
if not self.is_mac:
ctl2.Hide()
ix = 4
ctl3 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 2, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onparamRangeEnter)
self.sizer4_4.Add(ctl3, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix = 5
ctl4 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 2, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onparamRangeEnter)
self.sizer4_4.Add(ctl4, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ctl3.Show(True)
ctl4.Show(True)
elif p == "npts":
ix = 6
value = self.model.getParam(name2)
Tctl = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 2.2, 20),
style=wx.TE_PROCESS_ENTER)
Tctl.SetValue(str(format_number(value)))
self.sizer4_4.Add(Tctl, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
self.fixed_param.append([None, name2, Tctl, None, None,
None, None, None])
elif p == "nsigmas":
ix = 7
value = self.model.getParam(name3)
Tct2 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 2.2, 20),
style=wx.TE_PROCESS_ENTER)
Tct2.SetValue(str(format_number(value)))
self.sizer4_4.Add(Tct2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
self.fixed_param.append([None, name3, Tct2,
None, None, None,
None, None])
ix = 8
disp_box = wx.ComboBox(self, wx.ID_ANY, size=(65, -1),
style=wx.CB_READONLY, name='%s' % name1)
for key, value in POLYDISPERSITY_MODELS.iteritems():
name_disp = str(key)
disp_box.Append(name_disp, value)
disp_box.SetStringSelection("gaussian")
wx.EVT_COMBOBOX(disp_box, wx.ID_ANY, self._on_disp_func)
self.sizer4_4.Add(disp_box, (iy, ix), (1, 1), wx.EXPAND)
self.fittable_param.append([cb, name1, ctl1, text2,
ctl2, ctl3, ctl4, disp_box])
ix = 0
iy += 1
self.sizer4_4.Add((20, 20), (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
first_orient = True
for item in self.model.dispersion.keys():
if not self.magnetic_on:
if item in self.model.magnetic_params:
continue
if item in self.model.orientation_params:
if item not in self.disp_cb_dict:
self.disp_cb_dict[item] = None
name0 = "Distribution of " + item
name1 = item + ".width"
name2 = item + ".npts"
name3 = item + ".nsigmas"
if name1 not in self.model.details:
self.model.details[name1] = ["", None, None]
iy += 1
for p in self.model.dispersion[item].keys():
if p == "width":
ix = 0
cb = wx.CheckBox(self, wx.ID_ANY, name0, (10, 10))
cb.SetValue(CHECK_STATE)
cb.SetToolTipString("Check mark to fit")
wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
self.sizer4_4.Add(cb, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 5)
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
cb.Show(True)
elif cb.IsShown():
cb.Hide()
ix = 1
value = self.model.getParam(name1)
ctl1 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 1.3, 20),
style=wx.TE_PROCESS_ENTER)
poly_tip = "Absolute Sigma for %s." % item
ctl1.SetToolTipString(poly_tip)
ctl1.SetValue(str(format_number(value, True)))
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
if first_orient:
values.SetLabel('PD[ratio], Sig[deg]')
poly_text = "PD(polydispersity for lengths):\n"
poly_text += "It should be a value between"
poly_text += "0 and 1\n"
poly_text += "Sigma for angles: \n"
poly_text += "It is the STD (ratio*mean)"
poly_text += " of the distribution.\n "
values.SetToolTipString(poly_text)
first_orient = False
ctl1.Show(True)
elif ctl1.IsShown():
ctl1.Hide()
self.sizer4_4.Add(ctl1, (iy, ix), (1, 1), wx.EXPAND)
# text to show error sign
ix = 2
text2 = wx.StaticText(self, wx.ID_ANY, '+/-')
self.sizer4_4.Add(text2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
text2.Hide()
ix = 3
ctl2 = wx.TextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 1.3, 20),
style=0)
self.sizer4_4.Add(ctl2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ctl2.Hide()
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
if self.is_mac:
text2.Show(True)
ctl2.Show(True)
ix = 4
ctl3 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 2, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onparamRangeEnter)
self.sizer4_4.Add(ctl3, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ctl3.Hide()
ix = 5
ctl4 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 2, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onparamRangeEnter)
self.sizer4_4.Add(ctl4, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ctl4.Hide()
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
ctl3.Show(True)
ctl4.Show(True)
elif p == "npts":
ix = 6
value = self.model.getParam(name2)
Tctl = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 2.2, 20),
style=wx.TE_PROCESS_ENTER)
Tctl.SetValue(str(format_number(value)))
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
Tctl.Show(True)
else:
Tctl.Hide()
self.sizer4_4.Add(Tctl, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
self.fixed_param.append([None, name2, Tctl, None, None,
None, None, None])
self.orientation_params_disp.append([None, name2,
Tctl, None, None,
None, None, None])
elif p == "nsigmas":
ix = 7
value = self.model.getParam(name3)
Tct2 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 2.2, 20),
style=wx.TE_PROCESS_ENTER)
Tct2.SetValue(str(format_number(value)))
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
Tct2.Show(True)
else:
Tct2.Hide()
self.sizer4_4.Add(Tct2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
self.fixed_param.append([None, name3, Tct2,
None, None, None, None, None])
self.orientation_params_disp.append([None, name3,
Tct2, None, None, None, None, None])
ix = 8
disp_box = wx.ComboBox(self, wx.ID_ANY, size=(65, -1),
style=wx.CB_READONLY, name='%s' % name1)
for key, value in POLYDISPERSITY_MODELS.iteritems():
name_disp = str(key)
disp_box.Append(name_disp, value)
disp_box.SetStringSelection("gaussian")
wx.EVT_COMBOBOX(disp_box, wx.ID_ANY, self._on_disp_func)
self.sizer4_4.Add(disp_box, (iy, ix), (1, 1), wx.EXPAND)
self.fittable_param.append([cb, name1, ctl1, text2,
ctl2, ctl3, ctl4, disp_box])
self.orientation_params_disp.append([cb, name1, ctl1,
text2, ctl2, ctl3, ctl4, disp_box])
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
disp_box.Show(True)
else:
disp_box.Hide()
self.state.disp_cb_dict = copy.deepcopy(self.disp_cb_dict)
self.state.model = self.model.clone()
# save state into
self._copy_parameters_state(self.parameters, self.state.parameters)
self._copy_parameters_state(self.orientation_params_disp,
self.state.orientation_params_disp)
self._copy_parameters_state(self.fittable_param,
self.state.fittable_param)
self._copy_parameters_state(self.fixed_param, self.state.fixed_param)
wx.PostEvent(self.parent,
StatusEvent(status=" Selected Distribution: Gaussian"))
# Fill the list of fittable parameters
self.get_all_checked_params()
self.Layout()
def _onDraw(self, event):
"""
Update and Draw the model
"""
if self.model is None:
msg = "Please select a Model first..."
wx.MessageBox(msg, 'Info')
return
"""
if not self.data.is_data:
self.npts_x = self.Npts_total.GetValue()
self.Npts_fit.SetValue(self.npts_x)
self.create_default_data()
"""
flag, is_modified = self._update_paramv_on_fit()
wx.CallAfter(self._onparamEnter_helper, is_modified)
if not flag:
msg = "The parameters are invalid"
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
return
def _onFit(self, event):
"""
Allow to fit
"""
if event is not None:
event.Skip()
if self.fit_started:
self._StopFit()
self.fit_started = False
wx.CallAfter(self.set_fitbutton)
return
if self.data is None:
msg = "Please get Data first..."
wx.MessageBox(msg, 'Info')
wx.PostEvent(self._manager.parent,
StatusEvent(status="Fit: %s" % msg))
return
if self.model is None:
msg = "Please select a Model first..."
wx.MessageBox(msg, 'Info')
wx.PostEvent(self._manager.parent,
StatusEvent(status="Fit: %s" % msg, type="stop"))
return
if len(self.param_toFit) <= 0:
msg = "Select at least one parameter to fit"
wx.MessageBox(msg, 'Info')
wx.PostEvent(self._manager.parent,
StatusEvent(status=msg, type="stop"))
return
flag = self._update_paramv_on_fit()
if self.batch_on and not self._is_2D():
if not self._validate_Npts_1D():
return
if not flag:
msg = "Fitting range or parameters are invalid"
wx.PostEvent(self._manager.parent,
StatusEvent(status=msg, type="stop"))
return
self.select_param()
# Remove or do not allow fitting on the Q=0 point, especially
# when y(q=0)=None at x[0].
self.qmin_x = float(self.qmin.GetValue())
self.qmax_x = float(self.qmax.GetValue())
self._manager._reset_schedule_problem(value=0, uid=self.uid)
self._manager.schedule_for_fit(uid=self.uid, value=1)
self._manager.set_fit_range(uid=self.uid, qmin=self.qmin_x,
qmax=self.qmax_x)
# single fit
# self._manager.onFit(uid=self.uid)
self.fit_started = self._manager.onFit(uid=self.uid)
wx.CallAfter(self.set_fitbutton)
def _onFitHelp(self, event):
"""
Bring up the Full Fitting Documentation whenever the HELP button is
clicked.
Calls DocumentationWindow with the path of the location within the
documentation tree (after /doc/ ....". Note that when using old
versions of Wx (before 2.9) and thus not the release version of
installers, the help comes up at the top level of the file as
web browser does not pass anything past the # to the browser when it is
running "file:///...."
:param evt: Triggers on clicking the help button
"""
_TreeLocation = "user/sasgui/perspectives/fitting/fitting_help.html"
_doc_viewer = DocumentationWindow(self, wx.ID_ANY, _TreeLocation, "",
"General Fitting Help")
def _onSmearHelp(self, event):
"""
Bring up the instrumental resolution smearing Documentation whenever
the ? button in the smearing box is clicked.
Calls DocumentationWindow with the path of the location within the
documentation tree (after /doc/ ....". Note that when using old
versions of Wx (before 2.9) and thus not the release version of
installers, the help comes up at the top level of the file as
web browser does not pass anything past the # to the browser when it is
running "file:///...."
:param evt: Triggers on clicking the help button
"""
_TreeLocation = "user/sasgui/perspectives/fitting/sm_help.html"
_doc_viewer = DocumentationWindow(self, wx.ID_ANY, _TreeLocation, "",
"Instrumental Resolution Smearing \
Help")
[docs] def set_fitbutton(self):
"""
Set fit button label depending on the fit_started[bool]
"""
# Skip this feature if we are not on Windows
# NOTE: the is_mac data member actually means "is no Windows".
if self.is_mac:
return
if self.fit_started:
label = "Stop"
color = "red"
else:
label = "Fit"
color = "black"
# self.btFit.Enable(False)
self.btFit.SetLabel(label)
self.btFit.SetForegroundColour(color)
self.btFit.Enable(True)
[docs] def get_weight_flag(self):
"""
Get flag corresponding to a given weighting dI data.
"""
button_list = [self.dI_noweight,
self.dI_didata,
self.dI_sqrdata,
self.dI_idata]
flag = 1
for item in button_list:
if item.GetValue():
if button_list.index(item) == 0:
flag = 0 # dy = numpy.ones_like(dy_data)
elif button_list.index(item) == 1:
flag = 1 # dy = dy_data
elif button_list.index(item) == 2:
flag = 2 # dy = numpy.sqrt(numpy.abs(data))
elif button_list.index(item) == 3:
flag = 3 # dy = numpy.abs(data)
break
return flag
def _StopFit(self, event=None):
"""
Stop fit
"""
if event is not None:
event.Skip()
self._manager.stop_fit(self.uid)
self._manager._reset_schedule_problem(value=0)
self._on_fit_complete()
[docs] def rename_model(self):
"""
find a short name for model
"""
if self.model is not None:
self.model.name = "M" + str(self.index_model)
def _on_select_model(self, event=None):
"""
call back for model selection
"""
self.Show(False)
copy_flag = False
is_poly_enabled = None
if event is not None:
if (event.GetEventObject() == self.formfactorbox
and self.structurebox.GetLabel() != 'None')\
or event.GetEventObject() == self.structurebox\
or event.GetEventObject() == self.multifactorbox:
copy_flag = self.get_copy_params()
is_poly_enabled = self.enable_disp.GetValue()
self._on_select_model_helper()
self.set_model_param_sizer(self.model)
if self.model is None:
self._set_bookmark_flag(False)
self._keep.Enable(False)
self._set_save_flag(False)
self.enable_disp.SetValue(False)
self.disable_disp.SetValue(True)
# TODO: should not have an untrapped exception when displaying disperser
# TODO: do we need to create the disperser panel on every model change?
# Note: if we fix this, then remove ID_DISPERSER_HELP from basepage
try:
self.set_dispers_sizer()
except:
pass
self.state.enable_disp = self.enable_disp.GetValue()
self.state.disable_disp = self.disable_disp.GetValue()
self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
self.state.slit_smearer = self.slit_smearer.GetValue()
self.state.structurecombobox = self.structurebox.GetValue()
self.state.formfactorcombobox = self.formfactorbox.GetValue()
self.state.categorycombobox = self.categorybox.GetValue()
self.enable_fit_button()
if self.model is not None:
self.m_name = self.model.name
self.state.m_name = self.m_name
self.rename_model()
self._set_copy_flag(True)
self._set_paste_flag(True)
if self.data is not None:
is_data = check_data_validity(self.data)
if is_data:
self._set_bookmark_flag(not self.batch_on)
self._keep.Enable(not self.batch_on)
self._set_save_flag(True)
#Setting smearing for cases with and without data.
self._set_smear(self.data)
# more disables for 2D
self._set_smear_buttons()
try:
# update smearer sizer
#This call for smearing set up caused double evaluation of
#I(q) and double compilation as results
#self.onSmear(None)
temp_smear = None
if not self.disable_smearer.GetValue():
# Set the smearer environments
temp_smear = self.current_smearer
except:
raise
# error occured on chisqr computation
# pass
# event to post model to fit to fitting plugins
(ModelEventbox, EVT_MODEL_BOX) = wx.lib.newevent.NewEvent()
# set smearing value whether or not data contain the smearing info
evt = ModelEventbox(model=self.model,
smearer=temp_smear,
enable_smearer=not self.disable_smearer.GetValue(),
qmin=float(self.qmin_x),
uid=self.uid,
caption=self.window_caption,
qmax=float(self.qmax_x))
self._manager._on_model_panel(evt=evt)
self.mbox_description.SetLabel("Model [ %s ]" %
str(self.model.name))
self.mbox_description.SetForegroundColour(wx.BLUE)
self.state.model = self.model.clone()
self.state.model.name = self.model.name
if event is not None:
# post state to fit panel
new_event = PageInfoEvent(page=self)
wx.PostEvent(self.parent, new_event)
# update list of plugins if new plugin is available
mod_cat = self.categorybox.GetStringSelection()
if mod_cat == CUSTOM_MODEL:
temp_id = self.model.id
temp = self.parent.update_model_list()
for v in self.parent.model_dictionary.values():
if v.id == temp_id:
self.model = v()
break
if temp:
self.model_list_box = temp
current_val = self.formfactorbox.GetLabel()
pos = self.formfactorbox.GetSelection()
self._show_combox_helper()
self.formfactorbox.SetSelection(pos)
self.formfactorbox.SetValue(current_val)
# when select a model only from guictr/button
if is_poly_enabled is not None:
self.enable_disp.SetValue(is_poly_enabled)
self.disable_disp.SetValue(not is_poly_enabled)
self._set_dipers_Param(event=None)
self.state.enable_disp = self.enable_disp.GetValue()
self.state.disable_disp = self.disable_disp.GetValue()
# Keep the previous param values
if copy_flag:
self.get_paste_params(copy_flag)
wx.CallAfter(self._onDraw, None)
else:
self._draw_model()
if self.batch_on:
self.slit_smearer.Enable(False)
self.pinhole_smearer.Enable(False)
self.btEditMask.Disable()
self.EditMask_title.Disable()
self.Show(True)
self.SetupScrolling()
def _onparamEnter(self, event):
"""
when enter value on panel redraw model according to changed
"""
if self.model is None:
msg = "Please select a Model first..."
wx.MessageBox(msg, 'Info')
return
# default flag
flag = False
self.fitrange = True
# get event object
tcrtl = event.GetEventObject()
# Clear msg if previously shown.
msg = ""
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
if check_float(tcrtl):
flag = self._onparamEnter_helper()
self.show_npts2fit()
if self.fitrange:
temp_smearer = None
if not self.disable_smearer.GetValue():
temp_smearer = self.current_smearer
# set smearing value whether or not data contain the
# smearing info
if self.slit_smearer.GetValue():
flag1 = self.update_slit_smear()
flag = flag or flag1
elif self.pinhole_smearer.GetValue():
flag1 = self.update_pinhole_smear()
flag = flag or flag1
elif self.data.__class__.__name__ != "Data2D" and \
not self.enable2D:
enable_smearer = not self.disable_smearer.GetValue()
self._manager.set_smearer(smearer=temp_smearer,
fid=self.data.id,
uid=self.uid,
qmin=float(self.qmin_x),
qmax=float(self.qmax_x),
enable_smearer=enable_smearer)
if flag:
# self.compute_chisqr(smearer= temp_smearer)
# new state posted
if self.state_change:
# self._undo.Enable(True)
event = PageInfoEvent(page=self)
wx.PostEvent(self.parent, event)
self.state_change = False
else:
# invalid fit range: do nothing here:
# msg already displayed in validate
return
else:
self.save_current_state()
msg = "Cannot Plot: Must enter a number!!! "
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
self.save_current_state()
return
def _onparamRangeEnter(self, event):
"""
Check validity of value enter in the parameters range field
"""
tcrtl = event.GetEventObject()
# Clear msg if previously shown.
msg = ""
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
# Flag to register when a parameter has changed.
is_modified = False
if tcrtl.GetValue().lstrip().rstrip() != "":
try:
tcrtl.SetBackgroundColour(wx.WHITE)
self._check_value_enter(self.fittable_param)
self._check_value_enter(self.parameters)
except:
tcrtl.SetBackgroundColour("pink")
msg = "Model Error:wrong value entered : %s" % sys.exc_value
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
return
else:
tcrtl.SetBackgroundColour(wx.WHITE)
# self._undo.Enable(True)
self.save_current_state()
event = PageInfoEvent(page=self)
wx.PostEvent(self.parent, event)
self.state_change = False
[docs] def qrang_set_focus(self, event=None):
"""
ON Qrange focus
"""
if event is not None:
event.Skip()
# tcrtl = event.GetEventObject()
self._validate_qrange(self.qmin, self.qmax)
[docs] def qrange_click(self, event):
"""
On Qrange textctrl click, make the qrange lines in the plot
"""
if event is not None:
event.Skip()
if self.data.__class__.__name__ == "Data2D":
return
is_click = event.LeftDown()
if is_click:
d_id = self.data.id
d_group_id = self.data.group_id
act_ctrl = event.GetEventObject()
wx.PostEvent(self._manager.parent,
PlotQrangeEvent(ctrl=[self.qmin, self.qmax], id=d_id,
group_id=d_group_id, leftdown=is_click,
active=act_ctrl))
[docs] def on_qrange_text(self, event):
"""
#On q range value updated. DO not combine with qrange_click().
"""
if event is not None:
event.Skip()
if self.data.__class__.__name__ == "Data2D":
return
act_ctrl = event.GetEventObject()
d_id = self.data.id
d_group_id = self.data.group_id
wx.PostEvent(self._manager.parent,
PlotQrangeEvent(ctrl=[self.qmin, self.qmax], id=d_id,
group_id=d_group_id, leftdown=False,
active=act_ctrl))
self._validate_qrange(self.qmin, self.qmax)
[docs] def on_key(self, event):
"""
On Key down
"""
event.Skip()
if self.data.__class__.__name__ == "Data2D":
return
ctrl = event.GetEventObject()
try:
x_data = float(ctrl.GetValue())
except:
return
key = event.GetKeyCode()
length = len(self.data.x)
indx = (numpy.abs(self.data.x - x_data)).argmin()
# return array.flat[idx]
if key == wx.WXK_PAGEUP or key == wx.WXK_NUMPAD_PAGEUP:
indx += 1
if indx >= length:
indx = length - 1
elif key == wx.WXK_PAGEDOWN or key == wx.WXK_NUMPAD_PAGEDOWN:
indx -= 1
if indx < 0:
indx = 0
else:
return
ctrl.SetValue(str(self.data.x[indx]))
self._validate_qrange(self.qmin, self.qmax)
def _onQrangeEnter(self, event):
"""
Check validity of value enter in the Q range field
"""
tcrtl = event.GetEventObject()
# Clear msg if previously shown.
msg = ""
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
# For theory mode
if not self.data.is_data:
self.npts_x = self.Npts_total.GetValue()
self.Npts_fit.SetValue(self.npts_x)
self.create_default_data()
# Flag to register when a parameter has changed.
if tcrtl.GetValue().lstrip().rstrip() != "":
try:
tcrtl.SetBackgroundColour(wx.WHITE)
# If qmin and qmax have been modified, update qmin and qmax
if self._validate_qrange(self.qmin, self.qmax):
tempmin = float(self.qmin.GetValue())
if tempmin != self.qmin_x:
self.qmin_x = tempmin
tempmax = float(self.qmax.GetValue())
if tempmax != self.qmax_x:
self.qmax_x = tempmax
else:
tcrtl.SetBackgroundColour("pink")
msg = "Model Error:wrong value entered : %s" % sys.exc_value
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
return
except:
tcrtl.SetBackgroundColour("pink")
msg = "Model Error:wrong value entered : %s" % sys.exc_value
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
return
# Check if # of points for theory model are valid(>0).
# check for 2d
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
# set mask
radius = numpy.sqrt(self.data.qx_data * self.data.qx_data +
self.data.qy_data * self.data.qy_data)
index_data = ((self.qmin_x <= radius) & (radius <= self.qmax_x))
index_data = (index_data) & (self.data.mask)
index_data = (index_data) & (numpy.isfinite(self.data.data))
if len(index_data[index_data]) < 10:
msg = "Cannot Plot :No or too little npts in"
msg += " that data range!!! "
wx.PostEvent(self._manager.parent,
StatusEvent(status=msg))
return
else:
# self.data.mask = index_data
# self.Npts_fit.SetValue(str(len(self.data.mask)))
self.show_npts2fit()
else:
index_data = ((self.qmin_x <= self.data.x) &
(self.data.x <= self.qmax_x))
self.Npts_fit.SetValue(str(len(self.data.x[index_data])))
self.npts_x = self.Npts_total.GetValue()
self.create_default_data()
self._save_plotting_range()
else:
tcrtl.SetBackgroundColour("pink")
msg = "Model Error:wrong value entered!!!"
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
self._draw_model()
self.save_current_state()
event = PageInfoEvent(page=self)
wx.PostEvent(self.parent, event)
self.state_change = False
return
def _clear_Err_on_Fit(self):
"""
hide the error text control shown
after fitting
"""
if self.is_mac:
return
if hasattr(self, "text2_3"):
self.text2_3.Hide()
if len(self.parameters) > 0:
for item in self.parameters:
if item[0].IsShown():
# Skip the angle parameters if 1D data
if self.data.__class__.__name__ != "Data2D" and \
not self.enable2D:
if item in self.orientation_params:
continue
if item in self.param_toFit:
continue
# hide statictext +/-
if len(item) < 4:
continue
if item[3] is not None and item[3].IsShown():
item[3].Hide()
# hide textcrtl for error after fit
if item[4] is not None and item[4].IsShown():
item[4].Hide()
if len(self.fittable_param) > 0:
for item in self.fittable_param:
if item[0].IsShown():
# Skip the angle parameters if 1D data
if self.data.__class__.__name__ != "Data2D" and \
not self.enable2D:
if item in self.orientation_params:
continue
if item in self.param_toFit:
continue
if len(item) < 4:
continue
# hide statictext +/-
if item[3] is not None and item[3].IsShown():
item[3].Hide()
# hide textcrtl for error after fit
if item[4] is not None and item[4].IsShown():
item[4].Hide()
return
def _get_defult_custom_smear(self):
"""
Get the defult values for custum smearing.
"""
# get the default values
if self.dxl is None:
self.dxl = 0.0
if self.dxw is None:
self.dxw = ""
if self.dx_percent is None:
self.dx_percent = SMEAR_SIZE_H
def _get_smear_info(self):
"""
Get the smear info from data.
:return: self.smear_type, self.dq_l and self.dq_r,
respectively the type of the smear, dq_min and
dq_max for pinhole smear data
while dxl and dxw for slit smear
"""
# default
self.smear_type = None
self.dq_l = None
self.dq_r = None
data = self.data
if self.data is None:
return
elif self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
if data.dqx_data is None or data.dqy_data is None:
return
elif self.current_smearer is not None \
and data.dqx_data.any() != 0 \
and data.dqx_data.any() != 0:
self.smear_type = "Pinhole2d"
self.dq_l = format_number(numpy.average(data.dqx_data))
self.dq_r = format_number(numpy.average(data.dqy_data))
return
else:
return
# check if it is pinhole smear and get min max if it is.
if data.dx is not None and numpy.any(data.dx):
self.smear_type = "Pinhole"
self.dq_l = data.dx[0]
self.dq_r = data.dx[-1]
# check if it is slit smear and get min max if it is.
elif data.dxl is not None or data.dxw is not None:
self.smear_type = "Slit"
if data.dxl is not None and numpy.all(data.dxl, 0):
self.dq_l = data.dxl[0]
if data.dxw is not None and numpy.all(data.dxw, 0):
self.dq_r = data.dxw[0]
# return self.smear_type,self.dq_l,self.dq_r
def _show_smear_sizer(self):
"""
Show only the sizers depending on smear selection
"""
# smear disabled
if self.disable_smearer.GetValue():
self.smear_description_none.Show(True)
# 2Dsmear
elif self._is_2D():
self.smear_description_accuracy_type.Show(True)
self.smear_accuracy.Show(True)
self.smear_description_accuracy_type.Show(True)
self.smear_description_2d.Show(True)
self.smear_description_2d_x.Show(True)
self.smear_description_2d_y.Show(True)
if self.pinhole_smearer.GetValue():
self.smear_pinhole_percent.Show(True)
# smear from data
elif self.enable_smearer.GetValue():
self.smear_description_dqdata.Show(True)
if self.smear_type is not None:
self.smear_description_smear_type.Show(True)
if self.smear_type == 'Slit':
self.smear_description_slit_height.Show(True)
self.smear_description_slit_width.Show(True)
elif self.smear_type == 'Pinhole':
self.smear_description_pin_percent.Show(True)
self.smear_description_smear_type.Show(True)
self.smear_description_type.Show(True)
self.smear_data_left.Show(True)
self.smear_data_right.Show(True)
# custom pinhole smear
elif self.pinhole_smearer.GetValue():
if self.smear_type == 'Pinhole':
self.smear_message_new_p.Show(True)
self.smear_description_pin_percent.Show(True)
self.smear_pinhole_percent.Show(True)
# custom slit smear
elif self.slit_smearer.GetValue():
self.smear_message_new_s.Show(True)
self.smear_description_slit_height.Show(True)
self.smear_slit_height.Show(True)
self.smear_description_slit_width.Show(True)
self.smear_slit_width.Show(True)
def _hide_all_smear_info(self):
"""
Hide all smearing messages in the set_smearer sizer
"""
self.smear_description_none.Hide()
self.smear_description_dqdata.Hide()
self.smear_description_type.Hide()
self.smear_description_smear_type.Hide()
self.smear_description_accuracy_type.Hide()
self.smear_description_2d_x.Hide()
self.smear_description_2d_y.Hide()
self.smear_description_2d.Hide()
self.smear_accuracy.Hide()
self.smear_data_left.Hide()
self.smear_data_right.Hide()
self.smear_description_pin_percent.Hide()
self.smear_pinhole_percent.Hide()
self.smear_description_slit_height.Hide()
self.smear_slit_height.Hide()
self.smear_description_slit_width.Hide()
self.smear_slit_width.Hide()
self.smear_message_new_p.Hide()
self.smear_message_new_s.Hide()
def _set_accuracy_list(self):
"""
Set the list of an accuracy in 2D custum smear:
Xhigh, High, Med, or Low
"""
# list of accuracy choices
list = ['Low', 'Med', 'High', 'Xhigh']
for idx in range(len(list)):
self.smear_accuracy.Append(list[idx], idx)
def _set_fun_box_list(self, fun_box):
"""
Set the list of func for multifunctional models
"""
# Check if it is multi_functional model
if self.model.__class__ not in self.model_list_box["Multi-Functions"] \
and not self.temp_multi_functional:
return None
# Get the func name list
list = self.model.fun_list
if len(list) == 0:
return None
# build function (combo)box
ind = 0
while(ind < len(list)):
for key, val in list.iteritems():
if (val == ind):
fun_box.Append(key, val)
break
ind += 1
def _on_select_accuracy(self, event):
"""
Select an accuracy in 2D custom smear: Xhigh, High, Med, or Low
"""
# event.Skip()
# Check if the accuracy is same as before
# self.smear2d_accuracy = event.GetEventObject().GetValue()
self.smear2d_accuracy = self.smear_accuracy.GetValue()
if self.pinhole_smearer.GetValue():
self.onPinholeSmear(event=None)
else:
self.onSmear(event=None)
if self.current_smearer is not None:
self.current_smearer.set_accuracy(accuracy=\
self.smear2d_accuracy)
event.Skip()
def _on_fun_box(self, event):
"""
Select an func: Erf,Rparabola,LParabola
"""
fun_val = None
fun_box = event.GetEventObject()
name = fun_box.Name
value = fun_box.GetValue()
if value in self.model.fun_list:
fun_val = self.model.fun_list[value]
self.model.setParam(name, fun_val)
# save state
self._copy_parameters_state(self.str_parameters,
self.state.str_parameters)
# update params
self._update_paramv_on_fit()
# draw
self._draw_model()
self.Refresh()
# get ready for new event
event.Skip()
def _onMask(self, event):
"""
Build a panel to allow to edit Mask
"""
from sas.sasgui.guiframe.local_perspectives.plotting.masking import \
MaskPanel as MaskDialog
self.panel = MaskDialog(base=self, data=self.data, id=wx.NewId())
self.panel.ShowModal()
def _draw_masked_model(self, event):
"""
Draw model image w/mask
"""
is_valid_qrange = self._update_paramv_on_fit()
if is_valid_qrange and self.model is not None:
self.panel.MakeModal(False)
event.Skip()
# try re draw the model plot if it exists
self._draw_model()
self.show_npts2fit()
elif self.model is None:
self.panel.MakeModal(False)
event.Skip()
self.show_npts2fit()
msg = "No model is found on updating MASK in the model plot... "
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
else:
event.Skip()
msg = ' Please consider your Q range, too.'
self.panel.ShowMessage(msg)
def _set_smear(self, data):
"""
Set_smear
"""
if data is None:
return
self.current_smearer = smear_selection(data, self.model)
flag = self.disable_smearer.GetValue()
if self.current_smearer is None:
self.enable_smearer.Disable()
else:
self.enable_smearer.Enable()
if not flag:
self.onSmear(None)
def _mac_sleep(self, sec=0.2):
"""
Give sleep to MAC
"""
if self.is_mac:
time.sleep(sec)
[docs] def get_view_mode(self):
"""
return True if the panel allow 2D or False if 1D
"""
return self.enable2D
[docs] def compute_data_set_range(self, data_list):
"""
find the range that include all data in the set
return the minimum and the maximum values
"""
if data_list is not None and data_list != []:
for data in data_list:
qmin, qmax, npts = self.compute_data_range(data)
self.qmin_data_set = min(self.qmin_data_set, qmin)
self.qmax_data_set = max(self.qmax_data_set, qmax)
self.npts_data_set += npts
return self.qmin_data_set, self.qmax_data_set, self.npts_data_set
[docs] def compute_data_range(self, data):
"""
compute the minimum and the maximum range of the data
return the npts contains in data
:param data:
"""
qmin, qmax, npts = None, None, None
if data is not None:
if not hasattr(data, "data"):
try:
qmin = min(data.x)
# Maximum value of data
qmax = max(data.x)
npts = len(data.x)
except:
msg = "Unable to find min/max/length of \n data named %s" %\
data.filename
wx.PostEvent(self._manager.parent, StatusEvent(status=msg,
info="error"))
raise ValueError, msg
else:
qmin = 0
try:
x = max(math.fabs(data.xmin), math.fabs(data.xmax))
y = max(math.fabs(data.ymin), math.fabs(data.ymax))
except:
msg = "Unable to find min/max of \n data named %s" % \
data.filename
wx.PostEvent(self._manager.parent, StatusEvent(status=msg,
info="error"))
raise ValueError, msg
# Maximum value of data
qmax = math.sqrt(x * x + y * y)
npts = len(data.data)
return qmin, qmax, npts
[docs] def set_data(self, data):
"""
reset the current data
"""
id = None
flag = False
is_data = False
npts = 0
try:
old_id = self.data.id
old_group_id = self.data.group_id
except:
old_id = id
old_group_id = id
if self.data is not None:
is_data = check_data_validity(self.data)
if not is_data and data is not None:
flag = True
if data is not None:
if is_data:
self.graph_id = self.data.group_id
flag = (data.id != self.data.id)
self.data = data
if check_data_validity(data):
self.graph_id = data.group_id
self.data.group_id = self.graph_id
if self.data is None:
data_name = ""
self._set_bookmark_flag(False)
self._keep.Enable(False)
self._set_save_flag(False)
else:
if self.model is not None:
self._set_bookmark_flag(not self.batch_on)
self._keep.Enable(not self.batch_on)
if self.data.is_data:
self._set_save_flag(True)
self._set_preview_flag(True)
# more disables for 2D
di_flag = False
dq_flag = False
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
self.slit_smearer.Disable()
self.pinhole_smearer.Enable(True)
self.default_mask = copy.deepcopy(self.data.mask)
if self.data.err_data is not None \
and numpy.any(self.data.err_data):
di_flag = True
if self.data.dqx_data is not None \
and numpy.any(self.data.dqx_data):
dq_flag = True
else:
self.slit_smearer.Enable(True)
self.pinhole_smearer.Enable(True)
if self.data.dy is not None and numpy.any(self.data.dy):
di_flag = True
if self.data.dx is not None and numpy.any(self.data.dx):
dq_flag = True
elif self.data.dxl is not None and numpy.any(self.data.dxl):
dq_flag = True
if dq_flag:
self.enable_smearer.Enable(True)
self.enable_smearer.SetValue(True)
self.disable_smearer.SetValue(False)
else:
self.enable_smearer.Disable()
self.disable_smearer.Enable(True)
self.disable_smearer.SetValue(True)
if di_flag:
self.dI_didata.Enable(True)
self.dI_didata.SetValue(True)
self.weightbt_string = self.dI_didata.GetLabelText()
else:
self.dI_didata.Enable(False)
self.dI_noweight.SetValue(True)
self.weightbt_string = self.dI_noweight.GetLabelText()
# Enable weighting radio buttons
self.dI_noweight.Enable(True)
self.dI_sqrdata.Enable(True)
self.dI_idata.Enable(True)
self.formfactorbox.Enable()
self.structurebox.Enable()
data_name = self.data.name
_, _, npts = self.compute_data_range(self.data)
# set maximum range for x in linear scale
if not hasattr(self.data, "data"): # Display only for 1D data fit
self.btEditMask.Disable()
self.EditMask_title.Disable()
else:
self.btEditMask.Enable()
self.EditMask_title.Enable()
self.Npts_total.SetValue(str(npts))
# default:number of data points selected to fit
self.Npts_fit.SetValue(str(npts))
self.Npts_total.SetEditable(False)
self.Npts_total.SetBackgroundColour(
self.GetParent().GetBackgroundColour())
self.Npts_total.Bind(wx.EVT_MOUSE_EVENTS, self._npts_click)
self.pointsbox.Disable()
self.dataSource.SetValue(data_name)
self.state.data = data
self.enable_fit_button()
# send graph_id to page_finder
self._manager.set_graph_id(uid=self.uid, graph_id=self.graph_id)
# focus the page
if check_data_validity(data):
self.data_box_description.SetForegroundColour(wx.BLUE)
if self.batch_on:
self.slit_smearer.Enable(False)
self.pinhole_smearer.Enable(False)
self.btEditMask.Disable()
self.EditMask_title.Disable()
self.on_smear_helper()
self.on_set_focus(None)
self.Refresh()
# update model plot with new data information
if flag:
if self.data.__class__.__name__ == "Data2D":
self.enable2D = True
self.model_view.SetLabel("2D Mode")
else:
self.enable2D = False
self.model_view.SetLabel("1D Mode")
self.model_view.Disable()
# replace data plot on combo box selection
# by removing the previous selected data
try:
wx.PostEvent(self._manager.parent,
NewPlotEvent(action="delete",
group_id=old_group_id, id=old_id))
except:
pass
# plot the current selected data
wx.PostEvent(self._manager.parent,
NewPlotEvent(action="check", plot=self.data,
title=str(self.data.title)))
self._draw_model()
def _npts_click(self, event):
"""
Prevent further handling of the mouse event on Npts_total
by not calling Skip().
"""
pass
[docs] def reset_page(self, state, first=False):
"""
reset the state
"""
try:
self.reset_page_helper(state)
self.select_param()
# Save state_fit
self.save_current_state_fit()
except:
self._show_combox_helper()
msg = "Error: This model state has missing or outdated "
msg += "information.\n"
msg += traceback.format_exc()
wx.PostEvent(self._manager.parent,
StatusEvent(status=msg, info="error"))
self._lay_out()
self.Refresh()
[docs] def get_range(self):
"""
return the fitting range
"""
return float(self.qmin_x), float(self.qmax_x)
[docs] def get_npts2fit(self):
"""
return numbers of data points within qrange
:Note: This is to normalize chisq by Npts of fit
"""
if self.data is None:
return
npts2fit = 0
qmin, qmax = self.get_range()
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
radius = numpy.sqrt(self.data.qx_data * self.data.qx_data +
self.data.qy_data * self.data.qy_data)
index_data = (self.qmin_x <= radius) & (radius <= self.qmax_x)
index_data = (index_data) & (self.data.mask)
index_data = (index_data) & (numpy.isfinite(self.data.data))
npts2fit = len(self.data.data[index_data])
else:
for qx in self.data.x:
if qmax >= qx >= qmin:
npts2fit += 1
return npts2fit
[docs] def show_npts2fit(self):
"""
setValue Npts for fitting
"""
self.Npts_fit.SetValue(str(self.get_npts2fit()))
[docs] def get_chi2(self):
"""
return the current chi2
"""
return self.tcChi.GetValue()
[docs] def onsetValues(self, chisqr, p_name, out, cov):
"""
Build the panel from the fit result
:param chisqr: Value of the goodness of fit metric
:param p_name: the name of parameters
:param out: list of parameter with the best value found during fitting
:param cov: Covariance matrix
"""
# make sure stop button to fit button all the time
self._on_fit_complete()
if out is None or not numpy.isfinite(chisqr):
raise ValueError, "Fit error occured..."
is_modified = False
has_error = False
dispersity = ''
# Hide textctrl boxes of errors.
self._clear_Err_on_Fit()
# Check if chi2 is finite
if chisqr is not None and numpy.isfinite(chisqr):
# format chi2
chi2 = format_number(chisqr, True)
self.tcChi.SetValue(chi2)
self.tcChi.Refresh()
else:
self.tcChi.SetValue("-")
# Hide error title
if self.text2_3.IsShown() and not self.is_mac:
self.text2_3.Hide()
try:
if self.enable_disp.GetValue():
if hasattr(self, "text_disp_1"):
if self.text_disp_1 is not None and not self.is_mac:
self.text_disp_1.Hide()
except:
dispersity = None
pass
i = 0
# Set the panel when fit result are list
for item in self.param_toFit:
if len(item) > 5 and item is not None:
if item[0].IsShown():
# reset error value to initial state
if not self.is_mac:
item[3].Hide()
item[4].Hide()
for ind in range(len(out)):
if item[1] == p_name[ind]:
break
if len(out) > 0 and out[ind] is not None:
val_out = format_number(out[ind], True)
item[2].SetValue(val_out)
if(cov is not None and len(cov) == len(out)):
try:
if dispersity is not None:
if self.enable_disp.GetValue():
if hasattr(self, "text_disp_1"):
if self.text_disp_1 is not None:
if not self.text_disp_1.IsShown()\
and not self.is_mac:
self.text_disp_1.Show(True)
except:
pass
if cov[ind] is not None:
if numpy.isfinite(float(cov[ind])):
val_err = format_number(cov[ind], True)
item[4].SetForegroundColour(wx.BLACK)
else:
val_err = 'NaN'
item[4].SetForegroundColour(wx.RED)
if not self.is_mac:
item[3].Show(True)
item[4].Show(True)
item[4].SetValue(val_err)
has_error = True
i += 1
else:
raise ValueError, "onsetValues: Invalid parameters..."
# Show error title when any errors displayed
if has_error:
if not self.text2_3.IsShown():
self.text2_3.Show(True)
# save current state
self.save_current_state()
if not self.is_mac:
self.Layout()
self.Refresh()
self._mac_sleep(0.1)
# plot model ( when drawing, do not update chisqr value again)
self._draw_model(update_chisqr=False, source='fit')
[docs] def onWeighting(self, event):
"""
On Weighting radio button event, sets the weightbt_string
"""
self.weightbt_string = event.GetEventObject().GetLabelText()
self._set_weight()
def _set_weight(self, is_2D=None):
"""
Set weight in fit problem
"""
# compute weight for the current data
flag_weight = self.get_weight_flag()
if is_2D is None:
is_2D = self._is_2D()
self._manager.set_fit_weight(uid=self.uid,
flag=flag_weight,
is2d=is_2D,
fid=None)
[docs] def onPinholeSmear(self, event):
"""
Create a custom pinhole smear object that will change the way residuals
are compute when fitting
:Note: accuracy is given by strings'High','Med', 'Low' FOR 2d,
None for 1D
"""
# Need update param values
self._update_paramv_on_fit()
if event is not None:
tcrtl = event.GetEventObject()
# event case of radio button
if tcrtl.GetValue():
self.dx_percent = 0.0
is_new_pinhole = True
else:
is_new_pinhole = self._is_changed_pinhole()
else:
is_new_pinhole = True
# if any value is changed
if is_new_pinhole:
self._set_pinhole_smear()
# hide all silt sizer
self._hide_all_smear_info()
# show relevant slit sizers
self._show_smear_sizer()
self.sizer_set_smearer.Layout()
# we need FitInside here not just self.Layout to ensure all the sizers
# end up with the necessasary space to in the scroll panel. In
# particular the compute and fit buttons end up on top of each other
# PDB Nov 28 2015.
self.FitInside()
if event is not None:
event.Skip()
# self._undo.Enable(True)
self.save_current_state()
event = PageInfoEvent(page=self)
wx.PostEvent(self.parent, event)
def _is_changed_pinhole(self):
"""
check if any of pinhole smear is changed
:return: True or False
"""
# get the values
pin_percent = self.smear_pinhole_percent.GetValue()
# Check changes in slit heigth
try:
dx_percent = float(pin_percent)
except:
return True
if self.dx_percent != dx_percent:
return True
return False
def _set_pinhole_smear(self):
"""
Set custom pinhole smear
:return: msg
"""
# copy data
data = copy.deepcopy(self.data)
if self._is_2D():
self.smear_type = 'Pinhole2d'
len_data = len(data.data)
data.dqx_data = numpy.zeros(len_data)
data.dqy_data = numpy.zeros(len_data)
else:
self.smear_type = 'Pinhole'
len_data = len(data.x)
data.dx = numpy.zeros(len_data)
data.dxl = None
data.dxw = None
msg = None
get_pin_percent = self.smear_pinhole_percent
if not check_float(get_pin_percent):
get_pin_percent.SetBackgroundColour("pink")
msg = "Model Error:wrong value entered!!!"
else:
if len_data < 2:
len_data = 2
self.dx_percent = float(get_pin_percent.GetValue())
if self.dx_percent < 0:
get_pin_percent.SetBackgroundColour("pink")
msg = "Model Error:This value can not be negative!!!"
elif self.dx_percent is not None:
percent = self.dx_percent/100
if self._is_2D():
data.dqx_data[data.dqx_data == 0] = percent * data.qx_data
data.dqy_data[data.dqy_data == 0] = percent * data.qy_data
else:
data.dx = percent * data.x
self.current_smearer = smear_selection(data, self.model)
# 2D need to set accuracy
if self._is_2D():
self.current_smearer.set_accuracy(
accuracy=self.smear2d_accuracy)
if msg is not None:
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
else:
get_pin_percent.SetBackgroundColour("white")
# set smearing value whether or not the data contain the smearing info
enable_smearer = not self.disable_smearer.GetValue()
self._manager.set_smearer(smearer=self.current_smearer,
fid=self.data.id,
qmin=float(self.qmin_x),
qmax=float(self.qmax_x),
enable_smearer=enable_smearer,
uid=self.uid)
return msg
[docs] def update_pinhole_smear(self):
"""
called by kill_focus on pinhole TextCntrl
to update the changes
:return: False when wrong value was entered
"""
# msg default
msg = None
# check if any value is changed
if self._is_changed_pinhole():
msg = self._set_pinhole_smear()
wx.CallAfter(self.save_current_state)
if msg is not None:
return False
else:
return True
[docs] def onSlitSmear(self, event):
"""
Create a custom slit smear object that will change the way residuals
are compute when fitting
"""
# Need update param values
self._update_paramv_on_fit()
# msg default
msg = None
# for event given
if event is not None:
tcrtl = event.GetEventObject()
# event case of radio button
if tcrtl.GetValue():
self.dxl = 0.0
self.dxw = 0.0
is_new_slit = True
else:
is_new_slit = self._is_changed_slit()
else:
is_new_slit = True
# if any value is changed
if is_new_slit:
msg = self._set_slit_smear()
# hide all silt sizer
self._hide_all_smear_info()
# show relevant slit sizers
self._show_smear_sizer()
self.sizer_set_smearer.Layout()
# we need FitInside here not just self.Layout to ensure all the sizers
# end up with the necessasary space to in the scroll panel. In
# particular the compute and fit buttons end up on top of each other
# PDB Nov 28 2015.
self.FitInside()
if event is not None:
event.Skip()
self.save_current_state()
event = PageInfoEvent(page=self)
wx.PostEvent(self.parent, event)
if msg is not None:
wx.PostEvent(self._manager.parent, StatusEvent(status=msg))
def _is_changed_slit(self):
"""
check if any of slit lengths is changed
:return: True or False
"""
# get the values
width = self.smear_slit_width.GetValue()
height = self.smear_slit_height.GetValue()
# check and change the box bg color if it was pink
# but it should be white now
# because this is the case that _set_slit_smear() will not handle
if height.lstrip().rstrip() == "":
self.smear_slit_height.SetBackgroundColour(wx.WHITE)
if width.lstrip().rstrip() == "":
self.smear_slit_width.SetBackgroundColour(wx.WHITE)
# Check changes in slit width
if width == "":
dxw = 0.0
else:
try:
dxw = float(width)
except:
return True
if self.dxw != dxw:
return True
# Check changes in slit heigth
if height == "":
dxl = 0.0
else:
try:
dxl = float(height)
except:
return True
if self.dxl != dxl:
return True
return False
def _set_slit_smear(self):
"""
Set custom slit smear
:return: message to inform the user about the validity
of the values entered for slit smear
"""
if self.data.__class__.__name__ == "Data2D" or self.enable2D:
return
# make sure once more if it is smearer
data = copy.deepcopy(self.data)
data_len = len(data.x)
data.dx = None
data.dxl = None
data.dxw = None
msg = None
try:
self.dxl = float(self.smear_slit_height.GetValue())
data.dxl = self.dxl * numpy.ones(data_len)
self.smear_slit_height.SetBackgroundColour(wx.WHITE)
except:
self.dxl = None
data.dxl = numpy.zeros(data_len)
if self.smear_slit_height.GetValue().lstrip().rstrip() != "":
self.smear_slit_height.SetBackgroundColour("pink")
msg = "Wrong value entered... "
else:
self.smear_slit_height.SetBackgroundColour(wx.WHITE)
try:
self.dxw = float(self.smear_slit_width.GetValue())
self.smear_slit_width.SetBackgroundColour(wx.WHITE)
data.dxw = self.dxw * numpy.ones(data_len)
except:
self.dxw = None
data.dxw = numpy.zeros(data_len)
if self.smear_slit_width.GetValue().lstrip().rstrip() != "":
self.smear_slit_width.SetBackgroundColour("pink")
msg = "Wrong Fit value entered... "
else:
self.smear_slit_width.SetBackgroundColour(wx.WHITE)
self.current_smearer = smear_selection(data, self.model)
# set smearing value whether or not the data contain the smearing info
enable_smearer = not self.disable_smearer.GetValue()
self._manager.set_smearer(smearer=self.current_smearer,
fid=self.data.id,
qmin=float(self.qmin_x),
qmax=float(self.qmax_x),
enable_smearer=enable_smearer,
uid=self.uid)
return msg
[docs] def update_slit_smear(self):
"""
called by kill_focus on pinhole TextCntrl
to update the changes
:return: False when wrong value was entered
"""
# msg default
msg = None
# check if any value is changed
if self._is_changed_slit():
msg = self._set_slit_smear()
# self._undo.Enable(True)
self.save_current_state()
if msg is not None:
return False
else:
return True
[docs] def onSmear(self, event):
"""
Create a smear object that will change the way residuals
are computed when fitting
"""
if event is not None:
event.Skip()
if self.data is None:
return
# Need update param values
self._update_paramv_on_fit()
if self.model is not None:
if self.data.is_data:
self._manager.page_finder[self.uid].add_data(data=self.data)
temp_smearer = self.on_smear_helper()
self.sizer_set_smearer.Layout()
# we need FitInside here not just self.Layout to ensure all the sizers
# end up with the necessasary space to in the scroll panel. In
# particular the compute and fit buttons end up on top of each other
# PDB Nov 28 2015.
self.FitInside()
self._set_weight()
# set smearing value whether or not the data contain the smearing info
enable_smearer = not self.disable_smearer.GetValue()
wx.CallAfter(self._manager.set_smearer, uid=self.uid,
smearer=temp_smearer,
fid=self.data.id,
qmin=float(self.qmin_x),
qmax=float(self.qmax_x),
enable_smearer=enable_smearer)
self.state.enable_smearer = self.enable_smearer.GetValue()
self.state.disable_smearer = self.disable_smearer.GetValue()
self.state.pinhole_smearer = self.pinhole_smearer.GetValue()
self.state.slit_smearer = self.slit_smearer.GetValue()
[docs] def on_smear_helper(self, update=False):
"""
Help for onSmear
:param update: force or not to update
"""
self._get_smear_info()
# renew smear sizer
if self.smear_type is not None:
self.smear_description_smear_type.SetValue(str(self.smear_type))
self.smear_data_left.SetValue(str(self.dq_l))
self.smear_data_right.SetValue(str(self.dq_r))
self._hide_all_smear_info()
data = copy.deepcopy(self.data)
# make sure once more if it is smearer
temp_smearer = smear_selection(data, self.model)
if self.current_smearer != temp_smearer or update:
self.current_smearer = temp_smearer
if self.enable_smearer.GetValue():
if self.current_smearer is None:
wx.PostEvent(self._manager.parent,
StatusEvent(status="Data contains no smearing information"))
else:
wx.PostEvent(self._manager.parent,
StatusEvent(status="Data contains smearing information"))
self.smear_data_left.Show(True)
self.smear_data_right.Show(True)
temp_smearer = self.current_smearer
elif self.disable_smearer.GetValue():
self.smear_description_none.Show(True)
elif self.pinhole_smearer.GetValue():
self.onPinholeSmear(None)
elif self.slit_smearer.GetValue():
self.onSlitSmear(None)
self._show_smear_sizer()
return temp_smearer
[docs] def on_complete_chisqr(self, event):
"""
Display result chisqr on the panel
:event: activated by fitting/ complete after draw
"""
try:
if event is None:
output = "-"
elif not numpy.isfinite(event.output):
output = "-"
else:
output = event.output
self.tcChi.SetValue(str(format_number(output, True)))
self.state.tcChi = self.tcChi.GetValue()
except:
pass
[docs] def get_all_checked_params(self):
"""
Found all parameters current check and add them to list of parameters
to fit
"""
self.param_toFit = []
for item in self.parameters:
if item[0].GetValue() and item not in self.param_toFit:
if item[0].IsShown():
self.param_toFit.append(item)
for item in self.fittable_param:
if item[0].GetValue() and item not in self.param_toFit:
if item[0].IsShown():
self.param_toFit.append(item)
self.save_current_state_fit()
event = PageInfoEvent(page=self)
wx.PostEvent(self.parent, event)
param2fit = []
for item in self.param_toFit:
if item[0] and item[0].IsShown():
param2fit.append(item[1])
self._manager.set_param2fit(self.uid, param2fit)
[docs] def select_param(self, event=None):
"""
Select TextCtrl checked for fitting purpose and stores them
in self.param_toFit=[] list
"""
self.param_toFit = []
for item in self.parameters:
# Skip t ifhe angle parameters if 1D data
if self.data.__class__.__name__ != "Data2D" and\
not self.enable2D:
if item in self.orientation_params:
continue
# Select parameters to fit for list of primary parameters
if item[0].GetValue() and item[0].IsShown():
if not (item in self.param_toFit):
self.param_toFit.append(item)
else:
#remove parameters from the fitting list
if item in self.param_toFit:
self.param_toFit.remove(item)
# Select parameters to fit for list of fittable parameters
# with dispersion
for item in self.fittable_param:
# Skip t ifhe angle parameters if 1D data
if self.data.__class__.__name__ != "Data2D" and\
not self.enable2D:
if item in self.orientation_params:
continue
if item[0].GetValue() and item[0].IsShown():
if not (item in self.param_toFit):
self.param_toFit.append(item)
else:
# remove parameters from the fitting list
if item in self.param_toFit:
self.param_toFit.remove(item)
# Calculate num. of angle parameters
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
len_orient_para = 0
else:
len_orient_para = len(self.orientation_params) # assume even len
# Total num. of angle parameters
if len(self.fittable_param) > 0:
len_orient_para *= 2
self.save_current_state_fit()
if event is not None:
# post state to fit panel
event = PageInfoEvent(page=self)
wx.PostEvent(self.parent, event)
param2fit = []
for item in self.param_toFit:
if item[0] and item[0].IsShown():
param2fit.append(item[1])
self._manager.set_param2fit(self.uid, param2fit)
[docs] def set_model_param_sizer(self, model):
"""
Build the panel from the model content
:param model: the model selected in combo box for fitting purpose
"""
self.sizer3.Clear(True)
self.parameters = []
self.str_parameters = []
self.param_toFit = []
self.fittable_param = []
self.fixed_param = []
self.orientation_params = []
self.orientation_params_disp = []
if model is None:
self.sizer3.Layout()
self.SetupScrolling()
return
box_description = wx.StaticBox(self, wx.ID_ANY, str("Model Parameters"))
boxsizer1 = wx.StaticBoxSizer(box_description, wx.VERTICAL)
sizer = wx.GridBagSizer(5, 5)
# save the current model
self.model = model
keys = self.model.getParamList()
# list of dispersion parameters
self.disp_list = self.model.getDispParamList()
def custom_compare(a, b):
"""
Custom compare to order, first by alphabets then second by number.
"""
# number at the last digit
a_last = a[len(a) - 1]
b_last = b[len(b) - 1]
# default
num_a = None
num_b = None
# split the names
a2 = a.lower().split('_')
b2 = b.lower().split('_')
# check length of a2, b2
len_a2 = len(a2)
len_b2 = len(b2)
# check if it contains a int number(<10)
try:
num_a = int(a_last)
except:
pass
try:
num_b = int(b_last)
except:
pass
# Put 'scale' near the top; happens
# when numbered param name exists
if a == 'scale':
return -1
# both have a number
if num_a is not None and num_b is not None:
if num_a > num_b:
return -1
# same number
elif num_a == num_b:
# different last names
if a2[len_a2 - 1] != b2[len_b2 - 1] and num_a != 0:
return -cmp(a2[len_a2 - 1], b2[len_b2 - 1])
else:
return cmp(a, b)
else:
return 1
# one of them has a number
elif num_a is not None:
return 1
elif num_b is not None:
return -1
# no numbers
else:
return cmp(a.lower(), b.lower())
# keys obtained now from ordered dict, so commenting alphabetical
# ordering keys.sort(custom_compare)
iy = 0
ix = 0
sizer.Add(wx.StaticText(self, wx.ID_ANY, 'Parameter'),
(iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 1
self.text2_2 = wx.StaticText(self, wx.ID_ANY, 'Value')
sizer.Add(self.text2_2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 2
self.text2_3 = wx.StaticText(self, wx.ID_ANY, 'Error')
sizer.Add(self.text2_3, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
if not self.is_mac:
self.text2_3.Hide()
ix += 1
self.text2_min = wx.StaticText(self, wx.ID_ANY, 'Min')
sizer.Add(self.text2_min, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
# self.text2_min.Hide()
ix += 1
self.text2_max = wx.StaticText(self, wx.ID_ANY, 'Max')
sizer.Add(self.text2_max, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
# self.text2_max.Hide()
ix += 1
self.text2_4 = wx.StaticText(self, wx.ID_ANY, '[Units]')
sizer.Add(self.text2_4, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
self.text2_4.Hide()
CHECK_STATE = False
for item in keys:
if item not in self.disp_list and not item in \
self.model.orientation_params:
# prepare a spot to store errors
if item not in self.model.details:
self.model.details[item] = ["", None, None]
iy += 1
ix = 0
if (self.model.__class__ in
self.model_list_box["Multi-Functions"] or
self.temp_multi_functional)\
and (item in self.model.non_fittable):
non_fittable_name = wx.StaticText(self, wx.ID_ANY, item)
sizer.Add(non_fittable_name, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 21)
# add parameter value
ix += 1
value = self.model.getParam(item)
if len(self.model.fun_list) > 0:
# num = item.split('_')[1][5:7]
fun_box = wx.ComboBox(self, wx.ID_ANY, size=(100, -1),
style=wx.CB_READONLY, name='%s' % item)
self._set_fun_box_list(fun_box)
fun_box.SetSelection(0)
# self.fun_box.SetToolTipString("A function
# describing the interface")
wx.EVT_COMBOBOX(fun_box, wx.ID_ANY, self._on_fun_box)
else:
fun_box = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH, 20),
style=wx.TE_PROCESS_ENTER, name='%s' % item)
fun_box.SetToolTipString(
"Hit 'Enter' after typing to update the plot.")
fun_box.SetValue(format_number(value, True))
sizer.Add(fun_box, (iy, ix), (1, 1), wx.EXPAND)
self.str_parameters.append([None, item, fun_box,
None, None, None,
None, None])
else:
# add parameters name with checkbox for selecting to fit
cb = wx.CheckBox(self, wx.ID_ANY, item)
cb.SetValue(CHECK_STATE)
cb.SetToolTipString(" Check mark to fit.")
# cb.SetValue(True)
wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
sizer.Add(cb, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 5)
# add parameter value
ix += 1
value = self.model.getParam(item)
ctl1 = ModelTextCtrl(self, wx.ID_ANY, size=(_BOX_WIDTH, 20),
style=wx.TE_PROCESS_ENTER)
ctl1.SetToolTipString(
"Hit 'Enter' after typing to update the plot.")
ctl1.SetValue(format_number(value, True))
sizer.Add(ctl1, (iy, ix), (1, 1), wx.EXPAND)
# text to show error sign
ix += 1
text2 = wx.StaticText(self, wx.ID_ANY, '+/-')
sizer.Add(text2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
if not self.is_mac:
text2.Hide()
ix += 1
ctl2 = wx.TextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 1.2, 20), style=0)
sizer.Add(ctl2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
if not self.is_mac:
ctl2.Hide()
ix += 1
ctl3 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 1.9, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onparamRangeEnter)
min_bound = self.model.details[item][1]
if min_bound is not None:
ctl3.SetValue(format_number(min_bound, True))
sizer.Add(ctl3, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ix += 1
ctl4 = ModelTextCtrl(self, wx.ID_ANY,
size=(_BOX_WIDTH / 1.9, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onparamRangeEnter)
max_bound = self.model.details[item][2]
if max_bound is not None:
ctl4.SetValue(format_number(max_bound, True))
sizer.Add(ctl4, (iy, ix), (1, 1),
wx.EXPAND | wx.FIXED_MINSIZE, 0)
ix += 1
# Units
if item in self.model.details:
units = wx.StaticText(self, wx.ID_ANY,
self.model.details[item][0], style=wx.ALIGN_LEFT)
else:
units = wx.StaticText(self, wx.ID_ANY, "",
style=wx.ALIGN_LEFT)
sizer.Add(units, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
self.parameters.append([cb, item, ctl1,
text2, ctl2, ctl3, ctl4, units])
iy += 1
sizer.Add((10, 10), (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
# type can be either Guassian or Array
if len(self.model.dispersion.values()) > 0:
type = self.model.dispersion.values()[0]["type"]
else:
type = "Gaussian"
iy += 1
ix = 0
# Add tile for orientational angle
for item in keys:
if item in self.model.orientation_params:
orient_angle = wx.StaticText(self, wx.ID_ANY, '[For 2D only]:')
mag_on_button = wx.Button(self, wx.ID_ANY, "Magnetic ON")
mag_on_button.SetToolTipString("Turn Pol Beam/Mag scatt on/off")
mag_on_button.Bind(wx.EVT_BUTTON, self._on_mag_on)
mag_angle_help_button = wx.Button(self, wx.ID_ANY,
"Magnetic angles?")
mag_angle_help_button.SetToolTipString("see angle definitions")
mag_help_button = wx.Button(self, wx.ID_ANY, "Mag HELP")
mag_help_button.SetToolTipString("Help on pol beam/mag fitting")
mag_help_button.Bind(wx.EVT_BUTTON, self._on_mag_help)
mag_angle_help_button.Bind(wx.EVT_BUTTON,
self._on_mag_angle_help)
sizer.Add(orient_angle, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
iy += 1
sizer.Add(mag_on_button, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
ix += 1
sizer.Add(mag_angle_help_button, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
sizer.Add(mag_help_button, (iy, ix + 1), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
# handle the magnetic buttons
# clean this up so that assume mag is off then turn
# all buttons on IF mag has mag and has 2D
if not self._has_magnetic:
mag_on_button.Show(False)
elif not self.data.__class__.__name__ == "Data2D":
mag_on_button.Show(False)
else:
mag_on_button.Show(True)
mag_help_button.Show(False)
mag_angle_help_button.Show(False)
if mag_on_button.IsShown():
if self.magnetic_on:
mag_on_button.SetLabel("Magnetic OFF")
mag_help_button.Show(True)
mag_angle_help_button.Show(True)
else:
mag_on_button.SetLabel("Magnetic ON")
mag_help_button.Show(False)
mag_angle_help_button.Show(False)
if not self.data.__class__.__name__ == "Data2D" and \
not self.enable2D:
orient_angle.Hide()
else:
orient_angle.Show(True)
break
# For Gaussian only
if type.lower() != "array":
for item in self.model.orientation_params:
if not self.magnetic_on:
if item in self.model.magnetic_params:
continue
if item not in self.disp_list:
# prepare a spot to store min max
if item not in self.model.details:
self.model.details[item] = ["", None, None]
iy += 1
ix = 0
# add parameters name with checkbox for selecting to fit
cb = wx.CheckBox(self, wx.ID_ANY, item)
cb.SetValue(CHECK_STATE)
cb.SetToolTipString("Check mark to fit")
wx.EVT_CHECKBOX(self, cb.GetId(), self.select_param)
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
cb.Show(True)
else:
cb.Hide()
sizer.Add(cb, (iy, ix), (1, 1),
wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 5)
# add parameter value
ix += 1
value = self.model.getParam(item)
ctl1 = ModelTextCtrl(self, -1, size=(_BOX_WIDTH, 20),
style=wx.TE_PROCESS_ENTER)
ctl1.SetToolTipString(
"Hit 'Enter' after typing to update the plot.")
ctl1.SetValue(format_number(value, True))
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
ctl1.Show(True)
else:
ctl1.Hide()
sizer.Add(ctl1, (iy, ix), (1, 1), wx.EXPAND)
# text to show error sign
ix += 1
text2 = wx.StaticText(self, -1, '+/-')
sizer.Add(text2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
text2.Hide()
ix += 1
ctl2 = wx.TextCtrl(self, -1,
size=(_BOX_WIDTH / 1.2, 20), style=0)
sizer.Add(ctl2, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ctl2.Hide()
ix += 1
ctl3 = ModelTextCtrl(self, -1,
size=(_BOX_WIDTH / 1.8, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onparamRangeEnter)
sizer.Add(ctl3, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ctl3.Hide()
ix += 1
ctl4 = ModelTextCtrl(self, -1,
size=(_BOX_WIDTH / 1.8, 20),
style=wx.TE_PROCESS_ENTER,
text_enter_callback=self._onparamRangeEnter)
sizer.Add(ctl4, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
ctl4.Hide()
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
if self.is_mac:
text2.Show(True)
ctl2.Show(True)
ctl3.Show(True)
ctl4.Show(True)
ix += 1
# Units
if item in self.model.details:
units = wx.StaticText(self, -1,
self.model.details[item][0],
style=wx.ALIGN_LEFT)
else:
units = wx.StaticText(self, -1, "",
style=wx.ALIGN_LEFT)
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
units.Show(True)
else:
units.Hide()
sizer.Add(units, (iy, ix), (1, 1),
wx.EXPAND | wx.ADJUST_MINSIZE, 0)
self.parameters.append([cb, item, ctl1,
text2, ctl2, ctl3, ctl4, units])
self.orientation_params.append([cb, item, ctl1,
text2, ctl2, ctl3, ctl4, units])
iy += 1
box_description.SetForegroundColour(wx.BLUE)
# Display units text on panel
for item in keys:
if item in self.model.details:
self.text2_4.Show()
# Fill the list of fittable parameters
self.get_all_checked_params()
self.save_current_state_fit()
boxsizer1.Add(sizer)
self.sizer3.Add(boxsizer1, 0, wx.EXPAND | wx.ALL, 10)
self.sizer3.Layout()
self.Layout()
[docs] def on_right_down(self, event):
"""
Get key stroke event
"""
if self.data is None:
return
# Figuring out key combo: Cmd for copy, Alt for paste
if event.AltDown() and event.ShiftDown():
flag = True
elif event.AltDown() or event.ShiftDown():
flag = False
else:
return
# make event free
event.Skip()
# messages depending on the flag
if not flag:
infor = 'warning'
# inform msg to wx
wx.PostEvent(self._manager.parent,
StatusEvent(status=msg, info=infor))
def _onModel2D(self, event):
"""
toggle view of model from 1D to 2D or 2D from 1D
"""
if self.model_view.GetLabelText() == "Show 2D":
self.model_view.SetLabel("Show 1D")
self.enable2D = True
else:
self.model_view.SetLabel("Show 2D")
self.enable2D = False
self.Show(False)
self.create_default_data()
self._manager.store_data(self.uid, data_list=[self.data])
self.set_model_param_sizer(self.model)
self._set_sizer_dispersion()
self._set_weight(is_2D=self.enable2D)
self._set_smear_buttons()
self.Show(True)
self.SetupScrolling()
self._draw_model()
self.state.enable2D = copy.deepcopy(self.enable2D)
def _set_smear_buttons(self):
"""
Set semarer radio buttons
"""
# more disables for 2D
if self.data.__class__.__name__ == "Data2D" or \
self.enable2D:
self.slit_smearer.Disable()
self.pinhole_smearer.Enable(True)
self.default_mask = copy.deepcopy(self.data.mask)
else:
self.slit_smearer.Enable(True)
self.pinhole_smearer.Enable(True)
[docs]class BGTextCtrl(wx.TextCtrl):
"""
Text control used to display outputs.
No editing allowed. The background is
grayed out. User can't select text.
"""
def __init__(self, *args, **kwds):
wx.TextCtrl.__init__(self, *args, **kwds)
self.SetEditable(False)
self.SetBackgroundColour(self.GetParent().parent.GetBackgroundColour())
# Bind to mouse event to avoid text highlighting
# The event will be skipped once the call-back
# is called.
self.Bind(wx.EVT_MOUSE_EVENTS, self._click)
def _click(self, event):
"""
Prevent further handling of the mouse event
by not calling Skip().
"""
pass