source: sasview/invariantview/perspectives/invariant/invariant_panel.py @ 2e27cf85

ESS_GUIESS_GUI_DocsESS_GUI_batch_fittingESS_GUI_bumps_abstractionESS_GUI_iss1116ESS_GUI_iss879ESS_GUI_iss959ESS_GUI_openclESS_GUI_orderingESS_GUI_sync_sascalccostrafo411magnetic_scattrelease-4.1.1release-4.1.2release-4.2.2release_4.0.1ticket-1009ticket-1094-headlessticket-1242-2d-resolutionticket-1243ticket-1249ticket885unittest-saveload
Last change on this file since 2e27cf85 was 353f467, checked in by Jae Cho <jhjcho@…>, 15 years ago

fixed plotting bug

  • Property mode set to 100644
File size: 46.8 KB
Line 
1"""
2    This module provide GUI for the neutron scattering length density calculator
3    @author: Gervaise B. Alina
4"""
5
6import wx
7
8import sys
9from wx.lib.scrolledpanel import ScrolledPanel
10from sans.invariant import invariant
11from sans.guiframe.utils import format_number, check_float
12from sans.guicomm.events import NewPlotEvent, StatusEvent
13from invariant_details import InvariantDetailsPanel, InvariantContainer
14from invariant_widgets import OutputTextCtrl, InvTextCtrl
15
16# The minimum q-value to be used when extrapolating
17Q_MINIMUM  = 1e-5
18# The maximum q-value to be used when extrapolating
19Q_MAXIMUM  = 10
20# the maximum value to plot the theory data
21Q_MAXIMUM_PLOT = 2
22# the number of points to consider during fit
23NPTS = 10
24#Default value for background
25BACKGROUND = 0.0
26#default value for the scale
27SCALE = 1.0
28#default value of the contrast
29CONTRAST = 1.0
30#default value of the power used for power law
31POWER = 4.0
32#Invariant panel size
33_BOX_WIDTH = 76
34
35if sys.platform.count("win32")>0:
36    _STATICBOX_WIDTH = 450
37    PANEL_WIDTH = 500 
38    PANEL_HEIGHT = 700
39    FONT_VARIANT = 0
40else:
41    _STATICBOX_WIDTH = 480
42    PANEL_WIDTH = 530
43    PANEL_HEIGHT = 700
44    FONT_VARIANT = 1
45
46
47class InvariantPanel(ScrolledPanel):
48    """
49        Provides the Invariant GUI.
50    """
51    ## Internal nickname for the window, used by the AUI manager
52    window_name = "Invariant"
53    ## Name to appear on the window title bar
54    window_caption = "Invariant"
55    ## Flag to tell the AUI manager to put this panel in the center pane
56    CENTER_PANE = True
57    def __init__(self, parent, data=None, manager=None,*args, **kwds):
58        kwds["size"]= (PANEL_WIDTH, PANEL_HEIGHT)
59        kwds["style"]= wx.FULL_REPAINT_ON_RESIZE
60        ScrolledPanel.__init__(self, parent=parent, *args, **kwds)
61        #Font size
62        self.SetWindowVariant(variant=FONT_VARIANT)
63        #Object that receive status event
64        self.parent = parent
65        #plug-in using this panel
66        self._manager = manager
67        #Data uses for computation
68        self._data = data
69        self._scale = SCALE
70        self._background = BACKGROUND
71        #container of invariant value
72        self.inv_container = None
73        #Draw the panel
74        self._do_layout()
75        self.reset_panel()
76        if self.parent is not None:
77            msg = ""
78            wx.PostEvent(self.parent,StatusEvent(status=msg, info="info"))
79        self.SetupScrolling()
80       
81    def err_check_on_data(self):
82        """
83            Check if data is valid for further computation
84        """
85        flag = False
86        self.hint_msg_txt.SetLabel('')
87        #edit the panel
88        if self._data is not None:
89            if len(self._data.x[self._data.x==0]) > 0:
90                flag = True
91                msg = "Invariant: one of your q-values is zero. Delete that entry before proceeding"
92                self.hint_msg_txt.SetLabel(msg)
93                wx.PostEvent(self.parent, StatusEvent(status=msg,
94                                                      info="warning",
95                                                      type="stop")) 
96        return flag
97   
98    def set_data(self, data):
99        """
100            Set the data
101        """
102        self._data = data
103        #edit the panel
104        if self._data is not None:
105            self.err_check_on_data()
106            data_name = self._data.name
107            data_qmin = min (self._data.x)
108            data_qmax = max (self._data.x)
109            self.data_name_tcl.SetValue(str(data_name))
110            self.data_min_tcl.SetLabel(str(data_qmin))
111            self.data_max_tcl.SetLabel(str(data_qmax))
112            self.reset_panel()
113            self.compute_invariant(event=None)
114             
115    def set_message(self):
116        """
117            Display warning message if available
118        """
119        if self.inv_container is not None:
120            if self.inv_container.existing_warning:
121                msg = "Warning! Computations on invariant require your "
122                msg += "attention.\n Please click on Details button."
123                self.hint_msg_txt.SetForegroundColour("red")
124   
125                wx.PostEvent(self.parent,StatusEvent(status=msg,info="warning"))
126            else:
127                msg = "For more information, click on Details button."
128                self.hint_msg_txt.SetForegroundColour("black")
129                wx.PostEvent(self.parent,StatusEvent(status=msg,info="info"))
130            self.hint_msg_txt.SetLabel(msg)
131        self.data_name_boxsizer.Layout()
132       
133    def set_manager(self, manager):
134        """
135            set value for the manager
136        """
137        self._manager = manager
138   
139    def get_background(self):
140        """
141            @return the background textcrtl value as a float
142        """
143        background = self.background_tcl.GetValue().lstrip().rstrip()
144        if background == "":
145            raise ValueError, "Need a background"
146        if check_float(self.background_tcl):
147            return float(background)
148        else:
149            raise ValueError, "Receive invalid value for background : %s"%(background)
150   
151    def get_scale(self):
152        """
153            @return the scale textcrtl value as a float
154        """
155        scale = self.scale_tcl.GetValue().lstrip().rstrip()
156        if scale == "":
157            raise ValueError, "Need a background"
158        if check_float(self.scale_tcl):
159            if float(scale)<= 0.0:
160                self.scale_tcl.SetBackgroundColour("pink")
161                self.scale_tcl.Refresh()
162                raise ValueError, "Receive invalid value for scale: %s"%(scale)
163            return float(scale)
164        else:
165            raise ValueError, "Receive invalid value for scale : %s"%(scale)
166       
167    def get_contrast(self):
168        """
169            @return the contrast textcrtl value as a float
170        """
171        par_str = self.contrast_tcl.GetValue().strip()
172        contrast = None
173        if par_str !="" and check_float(self.contrast_tcl):
174            contrast = float(par_str)
175        return contrast
176   
177    def get_extrapolation_type(self, low_q, high_q):
178        """
179        """
180        extrapolation = None
181        if low_q  and not high_q:
182            extrapolation = "low"
183        elif not low_q  and high_q:
184            extrapolation = "high"
185        elif low_q and high_q:
186            extrapolation = "both"
187        return extrapolation
188           
189    def get_porod_const(self):
190        """
191            @return the porod constant textcrtl value as a float
192        """
193        par_str = self.porod_constant_tcl.GetValue().strip()
194        porod_const = None
195        if par_str !="" and check_float(self.porod_constant_tcl):
196            porod_const = float(par_str)
197        return porod_const
198   
199    def get_volume(self, inv, contrast, extrapolation):
200        """
201        """
202        if contrast is not None:
203            try:
204                v, dv = inv.get_volume_fraction_with_error(contrast=contrast, 
205                                                           extrapolation=extrapolation)
206                self.volume_tcl.SetValue(format_number(v))
207                self.volume_err_tcl.SetValue(format_number(dv))
208            except:
209                self.volume_tcl.SetValue(format_number(None))
210                self.volume_err_tcl.SetValue(format_number(None))
211                msg= "Error occurred computing volume fraction: %s"%sys.exc_value
212                wx.PostEvent(self.parent, StatusEvent(status=msg,
213                                                      info="error",
214                                                      type="stop"))
215               
216    def get_surface(self, inv, contrast, porod_const, extrapolation):
217        """
218        """
219        if contrast is not None and porod_const is not None:
220            try:
221                s, ds = inv.get_surface_with_error(contrast=contrast,
222                                        porod_const=porod_const,
223                                        extrapolation=extrapolation)
224                self.surface_tcl.SetValue(format_number(s))
225                self.surface_err_tcl.SetValue(format_number(ds))
226            except:
227                self.surface_tcl.SetValue(format_number(None))
228                self.surface_err_tcl.SetValue(format_number(None))
229                msg = "Error occurred computing specific surface: %s"%sys.exc_value
230                wx.PostEvent(self.parent, StatusEvent(status=msg, info="error",
231                                                       type="stop"))
232               
233    def get_total_qstar(self, inv, extrapolation):
234        """
235        """
236        try:
237            qstar_total, qstar_total_err = inv.get_qstar_with_error(extrapolation)
238           
239            self.invariant_total_tcl.SetValue(format_number(qstar_total))
240            self.invariant_total_err_tcl.SetValue(format_number(qstar_total_err))
241            self.inv_container.qstar_total = qstar_total
242            self.inv_container.qstar_total_err = qstar_total_err
243         
244        except:
245            self.inv_container.qstar_total = "Error"
246            self.inv_container.qstar_total_err = "Error"
247            self.invariant_total_tcl.SetValue(format_number(None))
248            self.invariant_total_err_tcl.SetValue(format_number(None))
249            msg= "Error occurred computing invariant using extrapolation: %s"%sys.exc_value
250            wx.PostEvent(self.parent, StatusEvent(status= msg, type="stop")) 
251           
252    def get_low_qstar(self, inv, npts_low, low_q=False):
253        """
254        """
255        if low_q:
256            try: 
257                qstar_low, qstar_low_err = inv.get_qstar_low()
258                self.inv_container.qstar_low = qstar_low
259                self.inv_container.qstar_low_err = qstar_low_err
260                extrapolated_data = inv.get_extra_data_low(npts_in=npts_low) 
261                power_low = inv.get_extrapolation_power(range='low') 
262                if self.power_law_low.GetValue():
263                    self.power_low_tcl.SetValue(format_number(power_low))
264                self._manager.plot_theory(data=extrapolated_data,
265                                           name="Low-Q extrapolation")
266            except:
267                self.inv_container.qstar_low = "ERROR"
268                self.inv_container.qstar_low_err = "ERROR"
269                self._manager.plot_theory(name="Low-Q extrapolation")
270                msg= "Error occurred computing low-Q invariant: %s"%sys.exc_value
271                wx.PostEvent(self.parent, StatusEvent(status= msg, type="stop"))
272        else:
273            self._manager.plot_theory(name="Low-Q extrapolation")
274           
275    def get_high_qstar(self, inv, high_q=False):
276        """
277        """
278        if high_q:
279            try: 
280                qstar_high, qstar_high_err = inv.get_qstar_high()
281                self.inv_container.qstar_high = qstar_high
282                self.inv_container.qstar_high_err = qstar_high_err
283                power_high = inv.get_extrapolation_power(range='high') 
284                self.power_high_tcl.SetValue(format_number(power_high))
285                high_out_data = inv.get_extra_data_high(q_end=Q_MAXIMUM_PLOT)
286                self._manager.plot_theory(data=high_out_data,
287                                           name="High-Q extrapolation")
288            except:
289                self.inv_container.qstar_high = "ERROR"
290                self.inv_container.qstar_high_err = "ERROR"
291                self._manager.plot_theory(name="High-Q extrapolation")
292                msg= "Error occurred computing high-Q invariant: %s"%sys.exc_value
293                wx.PostEvent(self.parent, StatusEvent(status= msg, type="stop"))
294        else:
295            self._manager.plot_theory(name="High-Q extrapolation")
296           
297    def get_qstar(self, inv):
298        """
299        """
300        qstar, qstar_err = inv.get_qstar_with_error()
301        self.inv_container.qstar = qstar
302        self.inv_container.qstar_err = qstar_err
303             
304    def set_extrapolation_low(self, inv, low_q=False):
305        """
306            @return float value necessary to compute invariant a low q
307        """
308        #get funtion
309        if self.guinier.GetValue():
310            function_low = "guinier"
311        # get the function
312        power_low = None #2.0/3.0
313        if self.power_law_low.GetValue():
314            function_low = "power_law"
315            if self.fit_enable_low.GetValue():
316                #set value of power_low to none to allow fitting
317                power_low = None
318            else:
319                power_low = self.power_low_tcl.GetValue().lstrip().rstrip()
320                if check_float(self.power_low_tcl):
321                    power_low = float(power_low)
322                else:
323                    if low_q :
324                        #Raise error only when qstar at low q is requested
325                        msg = "Expect float for power at low q , got %s"%(power_low)
326                        raise ValueError, msg
327       
328        #Get the number of points to extrapolated
329        npts_low = self.npts_low_tcl.GetValue().lstrip().rstrip()   
330        if check_float(self.npts_low_tcl):
331            npts_low = float(npts_low)
332        else:
333            if low_q:
334                msg = "Expect float for number of points at low q , got %s"%(npts_low)
335                raise ValueError, msg
336        #Set the invariant calculator
337        inv.set_extrapolation(range="low", npts=npts_low,
338                                   function=function_low, power=power_low)   
339        return inv, npts_low 
340       
341    def set_extrapolation_high(self, inv, high_q=False):
342        """
343            @return float value necessary to compute invariant a high q
344        """
345        power_high = None
346        #if self.power_law_high.GetValue():
347        function_high = "power_law"
348        if self.fit_enable_high.GetValue():
349            #set value of power_high to none to allow fitting
350            power_high = None
351        else:
352            power_high = self.power_high_tcl.GetValue().lstrip().rstrip()
353            if check_float(self.power_high_tcl):
354                power_high = float(power_high)
355            else:
356                if high_q :
357                    #Raise error only when qstar at high q is requested
358                    msg = "Expect float for power at high q , got %s"%(power_high)
359                    raise ValueError, msg
360                         
361        npts_high = self.npts_high_tcl.GetValue().lstrip().rstrip()   
362        if check_float(self.npts_high_tcl):
363            npts_high = float(npts_high)
364        else:
365            if high_q:
366                msg = "Expect float for number of points at high q , got %s"%(npts_high)
367                raise ValueError, msg
368        inv.set_extrapolation(range="high", npts=npts_high,
369                                   function=function_high, power=power_high)
370        return inv, npts_high
371   
372    def display_details(self, event):
373        """
374            open another panel for more details on invariant calculation
375        """
376        panel = InvariantDetailsPanel(parent=self, 
377                                           qstar_container=self.inv_container)
378        panel.ShowModal()
379        panel.Destroy()
380        self.button_calculate.SetFocus()
381       
382    def compute_invariant(self, event=None):
383        """
384            compute invariant
385        """
386        msg= ""
387        wx.PostEvent(self.parent, StatusEvent(status=msg))
388        if self._data is None or self.err_check_on_data():
389            return
390   
391        #clear outputs textctrl
392        self._reset_output()
393        try:
394            background = self.get_background()
395            scale = self.get_scale()
396        except:
397            msg= "Invariant Error: %s"%(sys.exc_value)
398            wx.PostEvent(self.parent, StatusEvent(status= msg, type="stop"))
399            return
400       
401        low_q = self.enable_low_cbox.GetValue()
402        high_q = self.enable_high_cbox.GetValue() 
403        #set invariant calculator
404        inv = invariant.InvariantCalculator(data=self._data,
405                                            background=background,
406                                            scale=scale)
407        try:
408            inv, npts_low = self.set_extrapolation_low(inv=inv, low_q=low_q)
409            inv, npts_high = self.set_extrapolation_high(inv=inv, high_q=high_q)
410        except:
411            msg = "Error occurred computing invariant: %s"%sys.exc_value
412            wx.PostEvent(self.parent, StatusEvent(status=msg,
413                                                 info="warning",type="stop"))
414            return
415        #check the type of extrapolation
416        extrapolation = self.get_extrapolation_type(low_q=low_q, high_q=high_q)
417       
418        #Compute invariant
419        bkg_changed = False
420        scale_changed = False
421        try:
422            self.get_qstar(inv=inv)
423            #if scale_changed or bkg_changed:
424            #self._manager.plot_data(data=inv.get_data())
425           
426        except:
427            msg= "Error occurred computing invariant: %s"%sys.exc_value
428            wx.PostEvent(self.parent, StatusEvent(status=msg, 
429                                                  info="warning",type="stop"))
430            return
431       
432        #Compute qstar extrapolated to low q range
433        self.get_low_qstar(inv=inv, npts_low=npts_low, low_q=low_q)
434        #Compute qstar extrapolated to high q range
435        self.get_high_qstar(inv=inv, high_q=high_q)
436        #Compute qstar extrapolated to total q range and set value to txtcrtl
437        self.get_total_qstar(inv=inv, extrapolation=extrapolation)
438        # Parse additional parameters
439        porod_const = self.get_porod_const()       
440        contrast = self.get_contrast()
441        try:
442            #Compute volume and set value to txtcrtl
443            self.get_volume(inv=inv, contrast=contrast, extrapolation=extrapolation)
444            #compute surface and set value to txtcrtl
445        except:
446            msg = "Error occurred computing invariant: %s"%sys.exc_value
447            wx.PostEvent(self.parent, StatusEvent(status=msg,
448                                                  info="warning",type="stop"))
449        try:
450            self.get_surface(inv=inv, contrast=contrast, porod_const=porod_const, 
451                                    extrapolation=extrapolation)
452        except:
453            msg = "Error occurred computing invariant: %s"%sys.exc_value
454            wx.PostEvent(self.parent, StatusEvent(status=msg,
455                                                  info="warning",type="stop"))
456           
457        #compute percentage of each invariant
458        self.inv_container.compute_percentage()
459        #display a message
460        self.set_message()
461        #enable the button_ok for more details
462        self.button_details.Enable()
463        self.button_details.SetFocus()
464       
465    def reset_panel(self):
466        """
467            set the panel at its initial state.
468        """
469        self.background_tcl.SetValue(str(BACKGROUND))
470        self.scale_tcl.SetValue(str(SCALE)) 
471        self.contrast_tcl.SetValue(str(CONTRAST))
472        self.porod_constant_tcl.SetValue('') 
473        self.npts_low_tcl.SetValue(str(NPTS))
474        self.enable_low_cbox.SetValue(False)
475        self.fix_enable_low.SetValue(True)
476        self.power_low_tcl.SetValue(str(POWER))
477        self.guinier.SetValue(True)
478        self.power_low_tcl.Disable()
479        self.enable_high_cbox.SetValue(False)
480        self.fix_enable_high.SetValue(True)
481        self.power_high_tcl.SetValue(str(POWER))
482        self.npts_high_tcl.SetValue(str(NPTS))
483        self.button_details.Disable()
484        #Change the state of txtcrtl to enable/disable
485        self._enable_low_q_section()
486        #Change the state of txtcrtl to enable/disable
487        self._enable_high_q_section()
488        self._reset_output()
489        self.button_calculate.SetFocus()
490       
491    def _reset_output(self):
492        """
493            clear outputs textcrtl
494        """
495        self.invariant_total_tcl.Clear()
496        self.invariant_total_err_tcl.Clear()
497        self.volume_tcl.Clear()
498        self.volume_err_tcl.Clear()
499        self.surface_tcl.Clear()
500        self.surface_err_tcl.Clear()
501        #prepare a new container to put result of invariant
502        self.inv_container = InvariantContainer()
503       
504    def _define_structure(self):
505        """
506            Define main sizers needed for this panel
507        """
508        ## Box sizers must be defined first before defining buttons/textctrls (MAC).
509        self.main_sizer = wx.BoxSizer(wx.VERTICAL)
510        #Sizer related to outputs
511        outputs_box = wx.StaticBox(self, -1, "Outputs")
512        self.outputs_sizer = wx.StaticBoxSizer(outputs_box, wx.VERTICAL)
513        self.outputs_sizer.SetMinSize((PANEL_WIDTH,-1))
514        #Sizer related to data
515        data_name_box = wx.StaticBox(self, -1, "I(q) Data Source")
516        self.data_name_boxsizer = wx.StaticBoxSizer(data_name_box, wx.VERTICAL)
517        self.data_name_boxsizer.SetMinSize((PANEL_WIDTH,-1))
518        self.hint_msg_sizer = wx.BoxSizer(wx.HORIZONTAL)
519        self.data_name_sizer = wx.BoxSizer(wx.HORIZONTAL)
520        self.data_range_sizer = wx.BoxSizer(wx.HORIZONTAL)
521        #Sizer related to background and scale
522        self.bkg_scale_sizer = wx.BoxSizer(wx.HORIZONTAL) 
523        #Sizer related to contrast and porod constant
524        self.contrast_porod_sizer = wx.BoxSizer(wx.HORIZONTAL) 
525        #Sizer related to inputs
526        inputs_box = wx.StaticBox(self, -1, "Customized Inputs")
527        self.inputs_sizer = wx.StaticBoxSizer(inputs_box, wx.VERTICAL)
528        #Sizer related to extrapolation
529        extrapolation_box = wx.StaticBox(self, -1, "Extrapolation")
530        self.extrapolation_sizer = wx.StaticBoxSizer(extrapolation_box,
531                                                        wx.VERTICAL)
532        self.extrapolation_sizer.SetMinSize((PANEL_WIDTH,-1))
533        self.extrapolation_range_sizer = wx.BoxSizer(wx.HORIZONTAL)
534        self.extrapolation_low_high_sizer = wx.BoxSizer(wx.HORIZONTAL)
535        #Sizer related to extrapolation at low q range
536        low_q_box = wx.StaticBox(self, -1, "Low Q")
537        self.low_extrapolation_sizer = wx.StaticBoxSizer(low_q_box, wx.VERTICAL)
538        self.low_q_sizer = wx.GridBagSizer(5,5)
539        #Sizer related to extrapolation at low q range
540        high_q_box = wx.StaticBox(self, -1, "High Q")
541        self.high_extrapolation_sizer = wx.StaticBoxSizer(high_q_box, wx.VERTICAL)
542        self.high_q_sizer = wx.GridBagSizer(5,5)
543        #sizer to define outputs
544        self.volume_surface_sizer = wx.GridBagSizer(5,5)
545        #Sizer related to invariant output
546        self.invariant_sizer = wx.GridBagSizer(5, 5)
547        #Sizer related to button
548        self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
549       
550    def _layout_data_name(self):
551        """
552            Draw widgets related to data's name
553        """
554        #Sizer hint
555        hint_msg = "First open data file from 'File' menu.  Then Highlight and right click on the data plot. \n"
556        hint_msg += "Finally, select 'Compute Invariant'. \n"
557        self.hint_msg_txt = wx.StaticText(self, -1, hint_msg) 
558        self.hint_msg_txt.SetForegroundColour("red")
559        msg = "Highlight = mouse the mouse's cursor on the data until the plot's color changes to yellow"
560        self.hint_msg_txt.SetToolTipString(msg)
561        self.hint_msg_sizer.Add(self.hint_msg_txt)
562        #Data name [string]
563        data_name_txt = wx.StaticText(self, -1, 'Data : ') 
564       
565        self.data_name_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH*5, 20), style=0) 
566        self.data_name_tcl.SetToolTipString("Data's name.")
567        self.data_name_sizer.AddMany([(data_name_txt, 0, wx.LEFT|wx.RIGHT, 10),
568                                       (self.data_name_tcl, 0, wx.EXPAND)])
569        #Data range [string]
570        data_range_txt = wx.StaticText(self, -1, 'Total Q Range (1/A): ') 
571        data_min_txt = wx.StaticText(self, -1, 'Min : ') 
572        self.data_min_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
573        self.data_min_tcl.SetToolTipString("The minimum value of q range.")
574        data_max_txt = wx.StaticText(self, -1, 'Max : ') 
575        self.data_max_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0) 
576        self.data_max_tcl.SetToolTipString("The maximum value of q range.")
577        self.data_range_sizer.AddMany([(data_range_txt, 0, wx.RIGHT, 10),
578                                       (data_min_txt, 0, wx.RIGHT, 10),
579                                       (self.data_min_tcl, 0, wx.RIGHT, 10),
580                                       (data_max_txt, 0, wx.RIGHT, 10),
581                                       (self.data_max_tcl, 0, wx.RIGHT, 10)])
582        self.data_name_boxsizer.AddMany([(self.hint_msg_sizer, 0 , wx.ALL, 10),
583                                         (self.data_name_sizer, 0 , wx.RIGHT, 10),
584                                         (self.data_range_sizer, 0 , wx.ALL, 10)])
585   
586    def _layout_bkg_scale(self):
587        """
588            Draw widgets related to background and scale
589        """
590        background_txt = wx.StaticText(self, -1, 'Background : ') 
591        self.background_tcl = InvTextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0) 
592        background_hint_txt = "background"
593        self.background_tcl.SetToolTipString(background_hint_txt)
594        background_unit_txt = wx.StaticText(self, -1, '[1/cm]') 
595        scale_txt = wx.StaticText(self, -1, 'Scale : ') 
596        self.scale_tcl = InvTextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
597        scale_hint_txt = "Scale"
598        self.scale_tcl.SetToolTipString(scale_hint_txt)
599        self.bkg_scale_sizer.AddMany([(background_txt, 0, wx.LEFT, 10),
600                                       (self.background_tcl, 0, wx.LEFT, 5),
601                                       (background_unit_txt, 0, wx.LEFT, 10),
602                                       (scale_txt, 0, wx.LEFT, 70),
603                                       (self.scale_tcl, 0, wx.LEFT, 40)])
604 
605    def _layout_contrast_porod(self):
606        """
607            Draw widgets related to porod constant and contrast
608        """
609        contrast_txt = wx.StaticText(self, -1, 'Contrast : ') 
610        self.contrast_tcl = InvTextCtrl(self, -1, size=(_BOX_WIDTH, 20), style=0)
611        contrast_hint_txt = "Contrast"
612        self.contrast_tcl.SetToolTipString(contrast_hint_txt)
613        contrast_unit_txt = wx.StaticText(self, -1, '[1/A^(2)]') 
614        porod_const_txt = wx.StaticText(self, -1, 'Porod Constant:') 
615        self.porod_constant_tcl = InvTextCtrl(self, -1, 
616                                              size=(_BOX_WIDTH, 20), style=0) 
617        porod_const_hint_txt = "Porod Constant"
618        self.porod_constant_tcl.SetToolTipString(porod_const_hint_txt)
619        optional_txt = wx.StaticText(self, -1, '(Optional)') 
620        self.contrast_porod_sizer.AddMany([(contrast_txt, 0, wx.LEFT, 10),
621                                           (self.contrast_tcl, 0, wx.LEFT, 20),
622                                           (contrast_unit_txt, 0, wx.LEFT, 10),
623                                           (porod_const_txt, 0, wx.LEFT, 50),
624                                       (self.porod_constant_tcl, 0, wx.LEFT, 0),
625                                       (optional_txt, 0, wx.LEFT, 10)])
626       
627    def _enable_fit_power_law_low(self, event=None):
628        """
629            Enable and disable the power value editing
630        """
631        if self.fix_enable_low.IsEnabled():
632            if self.fix_enable_low.GetValue():
633                self.power_low_tcl.Enable()
634            else:
635                self.power_low_tcl.Disable()
636           
637    def _enable_low_q_section(self, event=None):
638        """
639            Disable or enable some button if the user enable low q extrapolation
640        """
641        if self.enable_low_cbox.GetValue():
642            self.npts_low_tcl.Enable()
643            self.fix_enable_low.Enable()
644            self.fit_enable_low.Enable()
645            self.guinier.Enable()
646            self.power_law_low.Enable()
647
648        else:
649            self.npts_low_tcl.Disable()
650            self.fix_enable_low.Disable()
651            self.fit_enable_low.Disable()
652            self.guinier.Disable()
653            self.power_law_low.Disable()
654        self._enable_power_law_low()
655        self._enable_fit_power_law_low()
656        self.button_calculate.SetFocus()
657   
658    def _enable_power_law_low(self, event=None):
659        """
660            Enable editing power law section at low q range
661        """
662        if self.guinier.GetValue():
663            self.fix_enable_low.Disable()
664            self.fit_enable_low.Disable()
665            self.power_low_tcl.Disable()
666        else:
667            self.fix_enable_low.Enable()
668            self.fit_enable_low.Enable()
669            self.power_low_tcl.Enable()
670        self._enable_fit_power_law_low()
671           
672    def _layout_extrapolation_low(self):
673        """
674            Draw widgets related to extrapolation at low q range
675        """
676        self.enable_low_cbox = wx.CheckBox(self, -1, "Enable Extrapolate Low Q")
677        wx.EVT_CHECKBOX(self, self.enable_low_cbox.GetId(),
678                                         self._enable_low_q_section)
679        self.fix_enable_low = wx.RadioButton(self, -1, 'Fix',
680                                         (10, 10),style=wx.RB_GROUP)
681        self.fit_enable_low = wx.RadioButton(self, -1, 'Fit', (10, 10))
682        self.Bind(wx.EVT_RADIOBUTTON, self._enable_fit_power_law_low,
683                                     id=self.fix_enable_low.GetId())
684        self.Bind(wx.EVT_RADIOBUTTON, self._enable_fit_power_law_low, 
685                                        id=self.fit_enable_low.GetId())
686        self.guinier = wx.RadioButton(self, -1, 'Guinier',
687                                         (10, 10),style=wx.RB_GROUP)
688        self.power_law_low = wx.RadioButton(self, -1, 'Power Law', (10, 10))
689        self.Bind(wx.EVT_RADIOBUTTON, self._enable_power_law_low,
690                                     id=self.guinier.GetId())
691        self.Bind(wx.EVT_RADIOBUTTON, self._enable_power_law_low, 
692                                        id=self.power_law_low.GetId())
693       
694        npts_low_txt = wx.StaticText(self, -1, 'Npts')
695        self.npts_low_tcl = InvTextCtrl(self, -1, size=(_BOX_WIDTH*2/3, -1))
696        msg_hint = "Number of Q points to consider"
697        msg_hint +="while extrapolating the low-Q region"
698        self.npts_low_tcl.SetToolTipString(msg_hint)
699        power_txt = wx.StaticText(self, -1, 'Power')
700        self.power_low_tcl = InvTextCtrl(self, -1, size=(_BOX_WIDTH*2/3, -1))
701       
702        power_hint_txt = "Exponent to apply to the Power_law function."
703        self.power_low_tcl.SetToolTipString(power_hint_txt)
704        iy = 0
705        ix = 0
706        self.low_q_sizer.Add(self.enable_low_cbox,(iy, ix),(1,5),
707                            wx.TOP|wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
708        iy += 1
709        ix = 0
710        self.low_q_sizer.Add(npts_low_txt,(iy, ix),(1,1),
711                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
712        ix += 1
713        self.low_q_sizer.Add(self.npts_low_tcl, (iy, ix), (1,1),
714                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
715        iy += 1
716        ix = 0
717        self.low_q_sizer.Add(self.guinier,(iy, ix),(1,2),
718                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
719        iy += 1
720        ix = 0
721        self.low_q_sizer.Add(self.power_law_low,(iy, ix),(1,2),
722                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
723       
724        # Parameter controls for power law
725        ix = 1
726        iy += 1
727        self.low_q_sizer.Add(self.fix_enable_low,(iy, ix),(1,1),
728                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0)
729        ix += 1
730        self.low_q_sizer.Add(self.fit_enable_low,(iy, ix),(1,1),
731                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0)
732        ix = 1
733        iy += 1
734        self.low_q_sizer.Add(power_txt,(iy, ix),(1,1),
735                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0)
736        ix += 1
737        self.low_q_sizer.Add(self.power_low_tcl, (iy, ix), (1,1),
738                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
739        self.low_extrapolation_sizer.AddMany([(self.low_q_sizer, 0,
740                                                wx.BOTTOM|wx.RIGHT, 15)])
741       
742    def _enable_fit_power_law_high(self, event=None):
743        """
744            Enable and disable the power value editing
745        """
746        if self.fix_enable_high.IsEnabled():
747            if self.fix_enable_high.GetValue():
748                self.power_high_tcl.Enable()
749            else:
750                self.power_high_tcl.Disable()
751       
752    def _enable_high_q_section(self, event=None):
753        """
754            Disable or enable some button if the user enable high q extrapolation
755        """
756        if self.enable_high_cbox.GetValue():
757            self.npts_high_tcl.Enable()
758            self.power_law_high.Enable()
759            self.power_high_tcl.Enable()
760            self.fix_enable_high.Enable()
761            self.fit_enable_high.Enable()
762        else:
763            self.npts_high_tcl.Disable()
764            self.power_law_high.Disable()
765            self.power_high_tcl.Disable()
766            self.fix_enable_high.Disable()
767            self.fit_enable_high.Disable()
768        self._enable_fit_power_law_high()
769        self.button_calculate.SetFocus()
770 
771    def _layout_extrapolation_high(self):
772        """
773            Draw widgets related to extrapolation at high q range
774        """
775        self.enable_high_cbox = wx.CheckBox(self, -1, "Enable Extrapolate high-Q")
776        wx.EVT_CHECKBOX(self, self.enable_high_cbox.GetId(),
777                                         self._enable_high_q_section)
778     
779        self.fix_enable_high = wx.RadioButton(self, -1, 'Fix',
780                                         (10, 10),style=wx.RB_GROUP)
781        self.fit_enable_high = wx.RadioButton(self, -1, 'Fit', (10, 10))
782        self.Bind(wx.EVT_RADIOBUTTON, self._enable_fit_power_law_high,
783                                     id=self.fix_enable_high.GetId())
784        self.Bind(wx.EVT_RADIOBUTTON, self._enable_fit_power_law_high, 
785                                        id=self.fit_enable_high.GetId())
786       
787        self.power_law_high = wx.StaticText(self, -1, 'Power Law')
788        msg_hint ="Check to extrapolate data at high-Q"
789        self.power_law_high.SetToolTipString(msg_hint)
790        npts_high_txt = wx.StaticText(self, -1, 'Npts')
791        self.npts_high_tcl = InvTextCtrl(self, -1, size=(_BOX_WIDTH*2/3, -1))
792        msg_hint = "Number of Q points to consider"
793        msg_hint += "while extrapolating the high-Q region"
794        self.npts_high_tcl.SetToolTipString(msg_hint)
795        power_txt = wx.StaticText(self, -1, 'Power')
796        self.power_high_tcl = InvTextCtrl(self, -1, size=(_BOX_WIDTH*2/3, -1))
797        power_hint_txt = "Exponent to apply to the Power_law function."
798        self.power_high_tcl.SetToolTipString(power_hint_txt)
799        iy = 0
800        ix = 0
801        self.high_q_sizer.Add(self.enable_high_cbox,(iy, ix),(1,5),
802                            wx.TOP|wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
803        iy += 1
804        ix = 0
805        self.high_q_sizer.Add(npts_high_txt,(iy, ix),(1,1),
806                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
807        ix += 1
808        self.high_q_sizer.Add(self.npts_high_tcl, (iy, ix), (1,1),
809                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
810        iy += 2
811        ix = 0
812        self.high_q_sizer.Add(self.power_law_high,(iy, ix),(1,2),
813                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
814       
815        # Parameter controls for power law
816        ix = 1
817        iy += 1
818        self.high_q_sizer.Add(self.fix_enable_high,(iy, ix),(1,1),
819                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0)
820        ix += 1
821        self.high_q_sizer.Add(self.fit_enable_high,(iy, ix),(1,1),
822                           wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0)
823        ix = 1
824        iy += 1
825        self.high_q_sizer.Add(power_txt,(iy, ix),(1,1),
826                            wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
827        ix += 1
828        self.high_q_sizer.Add(self.power_high_tcl, (iy, ix), (1,1),
829                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
830        self.high_extrapolation_sizer.AddMany([(self.high_q_sizer, 0, 
831                                                wx.BOTTOM|wx.RIGHT, 10)])
832       
833    def _layout_extrapolation(self):
834        """
835            Draw widgets related to extrapolation
836        """
837        extra_hint = "Extrapolation Maximum Q Range [1/A]: "
838        extra_hint_txt = wx.StaticText(self, -1, extra_hint)
839        #Extrapolation range [string]
840        extrapolation_min_txt = wx.StaticText(self, -1, 'Min : ') 
841        self.extrapolation_min_tcl = OutputTextCtrl(self, -1, 
842                                                size=(_BOX_WIDTH, 20), style=0)
843        self.extrapolation_min_tcl.SetValue(str(Q_MINIMUM))
844        self.extrapolation_min_tcl.SetToolTipString("The minimum extrapolated q value.")
845        extrapolation_max_txt = wx.StaticText(self, -1, 'Max : ') 
846        self.extrapolation_max_tcl = OutputTextCtrl(self, -1,
847                                                  size=(_BOX_WIDTH, 20), style=0) 
848        self.extrapolation_max_tcl.SetValue(str(Q_MAXIMUM))
849        self.extrapolation_max_tcl.SetToolTipString("The maximum extrapolated q value.")
850        self.extrapolation_range_sizer.AddMany([(extra_hint_txt, 0, wx.LEFT, 10),
851                                                (extrapolation_min_txt, 0, wx.LEFT, 10),
852                                                (self.extrapolation_min_tcl,
853                                                            0, wx.LEFT, 10),
854                                                (extrapolation_max_txt, 0, wx.LEFT, 10),
855                                                (self.extrapolation_max_tcl,
856                                                            0, wx.LEFT, 10),
857                                                ])
858        self._layout_extrapolation_low()
859        self._layout_extrapolation_high()
860        self.extrapolation_low_high_sizer.AddMany([(self.low_extrapolation_sizer,
861                                                     0, wx.ALL, 10),
862                                                   (self.high_extrapolation_sizer,
863                                                    0, wx.ALL, 10)])
864        self.extrapolation_sizer.AddMany([(self.extrapolation_range_sizer, 0,
865                                            wx.RIGHT, 10),
866                                        (self.extrapolation_low_high_sizer, 0,
867                                           wx.ALL, 10)])
868       
869    def _layout_volume_surface_sizer(self):
870        """
871            Draw widgets related to volume and surface
872        """
873        unit_volume = ''
874        unit_surface = ''
875        uncertainty = "+/-" 
876        volume_txt = wx.StaticText(self, -1, 'Volume Fraction      ')
877        self.volume_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH,-1))
878        self.volume_tcl.SetToolTipString("Volume fraction.")
879        self.volume_err_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH,-1))
880        self.volume_err_tcl.SetToolTipString("Uncertainty on the volume fraction.")
881        volume_units_txt = wx.StaticText(self, -1, unit_volume)
882       
883        surface_txt = wx.StaticText(self, -1, 'Specific Surface')
884        self.surface_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH,-1))
885        self.surface_tcl.SetToolTipString("Specific surface value.")
886        self.surface_err_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH,-1))
887        self.surface_err_tcl.SetToolTipString("Uncertainty on the specific surface.")
888        surface_units_txt = wx.StaticText(self, -1, unit_surface)
889        iy = 0
890        ix = 0
891        self.volume_surface_sizer.Add(volume_txt, (iy, ix), (1,1),
892                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
893        ix += 1
894        self.volume_surface_sizer.Add(self.volume_tcl, (iy, ix), (1,1),
895                            wx.EXPAND|wx.ADJUST_MINSIZE, 10)
896        ix += 1
897        self.volume_surface_sizer.Add(wx.StaticText(self, -1, uncertainty),
898                         (iy, ix),(1,1),wx.EXPAND|wx.ADJUST_MINSIZE, 10) 
899        ix += 1
900        self.volume_surface_sizer.Add(self.volume_err_tcl, (iy, ix), (1,1),
901                            wx.EXPAND|wx.ADJUST_MINSIZE, 10) 
902        ix += 1
903        self.volume_surface_sizer.Add(volume_units_txt, (iy, ix), (1,1),
904                             wx.EXPAND|wx.ADJUST_MINSIZE, 10)
905        iy += 1
906        ix = 0
907        self.volume_surface_sizer.Add(surface_txt, (iy, ix), (1,1),
908                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
909        ix += 1
910        self.volume_surface_sizer.Add(self.surface_tcl, (iy, ix), (1,1),
911                            wx.EXPAND|wx.ADJUST_MINSIZE, 0)
912        ix += 1
913        self.volume_surface_sizer.Add(wx.StaticText(self, -1, uncertainty),
914                         (iy, ix),(1,1),wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
915        ix += 1
916        self.volume_surface_sizer.Add(self.surface_err_tcl, (iy, ix), (1,1),
917                            wx.EXPAND|wx.ADJUST_MINSIZE, 0) 
918        ix += 1
919        self.volume_surface_sizer.Add(surface_units_txt, (iy, ix), (1,1),
920                            wx.EXPAND|wx.ADJUST_MINSIZE, 10)
921       
922    def _layout_invariant_sizer(self):
923        """
924            Draw widgets related to invariant
925        """
926        uncertainty = "+/-" 
927        unit_invariant = '[1/(cm * A)]'
928        invariant_total_txt = wx.StaticText(self, -1, 'Invariant Total [Q*]')
929        self.invariant_total_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH,-1))
930        msg_hint = "Total invariant [Q*], including extrapolated regions."
931        self.invariant_total_tcl.SetToolTipString(msg_hint)
932        self.invariant_total_err_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH,-1))
933        self.invariant_total_err_tcl.SetToolTipString("Uncertainty on invariant.")
934        invariant_total_units_txt = wx.StaticText(self, -1, unit_invariant)
935   
936        #Invariant total
937        iy = 0
938        ix = 0
939        self.invariant_sizer.Add(invariant_total_txt, (iy, ix), (1,1),
940                             wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15)
941        ix += 1
942        self.invariant_sizer.Add(self.invariant_total_tcl, (iy, ix), (1,1),
943                          wx.EXPAND|wx.ADJUST_MINSIZE, 10)
944        ix += 1
945        self.invariant_sizer.Add( wx.StaticText(self, -1, uncertainty),
946                         (iy, ix),(1,1),wx.EXPAND|wx.ADJUST_MINSIZE, 10) 
947        ix += 1
948        self.invariant_sizer.Add(self.invariant_total_err_tcl, (iy, ix), (1,1),
949                             wx.EXPAND|wx.ADJUST_MINSIZE, 10)
950        ix += 1
951        self.invariant_sizer.Add(invariant_total_units_txt,(iy, ix), (1,1),
952                          wx.EXPAND|wx.ADJUST_MINSIZE, 10)
953 
954    def _layout_inputs_sizer(self):
955        """
956            Draw widgets related to inputs
957        """
958        self._layout_bkg_scale()
959        self._layout_contrast_porod()
960        self.inputs_sizer.AddMany([(self.bkg_scale_sizer, 0, wx.ALL, 5),
961                                    (self.contrast_porod_sizer, 0, wx.ALL, 5)])
962       
963    def _layout_outputs_sizer(self):
964        """
965            Draw widgets related to outputs
966        """
967        self._layout_volume_surface_sizer()
968        self._layout_invariant_sizer()
969        static_line = wx.StaticLine(self, -1)
970        self.outputs_sizer.AddMany([(self.volume_surface_sizer, 0, wx.ALL, 10),
971                                    (static_line, 0, wx.EXPAND, 0),
972                                    (self.invariant_sizer, 0, wx.ALL, 10)])
973    def _layout_button(self): 
974        """
975            Do the layout for the button widgets
976        """ 
977        #compute button
978        id = wx.NewId()
979        self.button_calculate = wx.Button(self, id, "Compute")
980        self.button_calculate.SetToolTipString("Compute invariant")
981        self.Bind(wx.EVT_BUTTON, self.compute_invariant, id=id)   
982        #detail button
983        id = wx.NewId()
984        self.button_details = wx.Button(self, id, "Details?")
985        self.button_details.SetToolTipString("Give Details on Computation")
986        self.Bind(wx.EVT_BUTTON, self.display_details, id=id)
987        details = "Details on Invariant Total Calculations"
988        details_txt = wx.StaticText(self, -1, details)
989        self.button_sizer.AddMany([((10,10), 0 , wx.LEFT,0),
990                                   (details_txt, 0 , 
991                                    wx.RIGHT|wx.BOTTOM|wx.TOP, 10),
992                                   (self.button_details, 0 , wx.ALL, 10),
993                        (self.button_calculate, 0 , wx.RIGHT|wx.TOP|wx.BOTTOM, 10)])
994       
995    def _do_layout(self):
996        """
997            Draw window content
998        """
999        self._define_structure()
1000        self._layout_data_name()
1001        self._layout_extrapolation()
1002        self._layout_inputs_sizer()
1003        self._layout_outputs_sizer()
1004        self._layout_button()
1005        self.main_sizer.AddMany([(self.data_name_boxsizer,0, wx.ALL, 10),
1006                                  (self.outputs_sizer, 0,
1007                                  wx.LEFT|wx.RIGHT|wx.BOTTOM, 10),
1008                                  (self.button_sizer,0,
1009                                  wx.LEFT|wx.RIGHT|wx.BOTTOM, 10),
1010                                 (self.inputs_sizer, 0,
1011                                  wx.LEFT|wx.RIGHT|wx.BOTTOM, 10),
1012                                  (self.extrapolation_sizer, 0,
1013                                  wx.LEFT|wx.RIGHT|wx.BOTTOM, 10)])
1014        self.SetSizer(self.main_sizer)
1015        self.SetAutoLayout(True)
1016   
1017class InvariantDialog(wx.Dialog):
1018    def __init__(self, parent=None, id=1,graph=None,
1019                 data=None, title="Invariant",base=None):
1020        wx.Dialog.__init__(self, parent, id, title, size=(PANEL_WIDTH,
1021                                                             PANEL_HEIGHT))
1022        self.panel = InvariantPanel(self)
1023        self.Centre()
1024        self.Show(True)
1025       
1026class InvariantWindow(wx.Frame):
1027    def __init__(self, parent=None, id=1,graph=None, 
1028                 data=None, title="Invariant",base=None):
1029       
1030        wx.Frame.__init__(self, parent, id, title, size=(PANEL_WIDTH +100,
1031                                                             PANEL_HEIGHT+100))
1032       
1033        self.panel = InvariantPanel(self)
1034        self.Centre()
1035        self.Show(True)
1036       
1037class MyApp(wx.App):
1038    def OnInit(self):
1039        wx.InitAllImageHandlers()
1040        frame = InvariantWindow()
1041        frame.Show(True)
1042        self.SetTopWindow(frame)
1043       
1044        return True
1045     
1046# end of class MyApp
1047
1048if __name__ == "__main__":
1049    app = MyApp(0)
1050    app.MainLoop()
Note: See TracBrowser for help on using the repository browser.