| 1 | |
|---|
| 2 | ################################################################################ |
|---|
| 3 | #This software was developed by the University of Tennessee as part of the |
|---|
| 4 | #Distributed Data Analysis of Neutron Scattering Experiments (DANSE) |
|---|
| 5 | #project funded by the US National Science Foundation. |
|---|
| 6 | # |
|---|
| 7 | #See the license text in license.txt |
|---|
| 8 | # |
|---|
| 9 | #copyright 2009, University of Tennessee |
|---|
| 10 | ################################################################################ |
|---|
| 11 | import wx |
|---|
| 12 | from wx.lib.scrolledpanel import ScrolledPanel |
|---|
| 13 | |
|---|
| 14 | MAX_NBR_DATA = 4 |
|---|
| 15 | WIDTH = 430 |
|---|
| 16 | HEIGHT = 350 |
|---|
| 17 | |
|---|
| 18 | |
|---|
| 19 | class DialogPanel(ScrolledPanel): |
|---|
| 20 | def __init__(self, *args, **kwds): |
|---|
| 21 | ScrolledPanel.__init__(self, *args, **kwds) |
|---|
| 22 | self.SetupScrolling() |
|---|
| 23 | |
|---|
| 24 | |
|---|
| 25 | class BatchDataDialog(wx.Dialog): |
|---|
| 26 | """ |
|---|
| 27 | The current design of Batch fit allows only of type of data in the data |
|---|
| 28 | set. This allows the user to make a quick selection of the type of data |
|---|
| 29 | to use in fit tab. |
|---|
| 30 | """ |
|---|
| 31 | def __init__(self, parent=None, *args, **kwds): |
|---|
| 32 | wx.Dialog.__init__(self, parent, *args, **kwds) |
|---|
| 33 | self.SetSize((WIDTH, HEIGHT)) |
|---|
| 34 | self.data_1d_selected = None |
|---|
| 35 | self.data_2d_selected = None |
|---|
| 36 | self._do_layout() |
|---|
| 37 | |
|---|
| 38 | def _do_layout(self): |
|---|
| 39 | """ |
|---|
| 40 | Draw the content of the current dialog window |
|---|
| 41 | """ |
|---|
| 42 | vbox = wx.BoxSizer(wx.VERTICAL) |
|---|
| 43 | box_description = wx.StaticBox(self, wx.ID_ANY, str("Hint")) |
|---|
| 44 | hint_sizer = wx.StaticBoxSizer(box_description, wx.VERTICAL) |
|---|
| 45 | selection_sizer = wx.GridBagSizer(5, 5) |
|---|
| 46 | button_sizer = wx.BoxSizer(wx.HORIZONTAL) |
|---|
| 47 | self.data_1d_selected = wx.RadioButton(self, wx.ID_ANY, 'Data1D', |
|---|
| 48 | style=wx.RB_GROUP) |
|---|
| 49 | self.data_2d_selected = wx.RadioButton(self, wx.ID_ANY, 'Data2D') |
|---|
| 50 | self.data_1d_selected.SetValue(True) |
|---|
| 51 | self.data_2d_selected.SetValue(False) |
|---|
| 52 | button_cancel = wx.Button(self, wx.ID_CANCEL, "Cancel") |
|---|
| 53 | button_OK = wx.Button(self, wx.ID_OK, "Ok") |
|---|
| 54 | button_OK.SetFocus() |
|---|
| 55 | hint = "Selected Data set contains both 1D and 2D Data.\n" |
|---|
| 56 | hint += "Please select on type of analysis before proceeding.\n" |
|---|
| 57 | hint_sizer.Add(wx.StaticText(self, wx.ID_ANY, hint)) |
|---|
| 58 | #draw area containing radio buttons |
|---|
| 59 | ix = 0 |
|---|
| 60 | iy = 0 |
|---|
| 61 | selection_sizer.Add(self.data_1d_selected, (iy, ix), |
|---|
| 62 | (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15) |
|---|
| 63 | iy += 1 |
|---|
| 64 | selection_sizer.Add(self.data_2d_selected, (iy, ix), |
|---|
| 65 | (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15) |
|---|
| 66 | #contruction the sizer contaning button |
|---|
| 67 | button_sizer.Add((20, 20), 1, wx.EXPAND | wx.ADJUST_MINSIZE, 0) |
|---|
| 68 | button_sizer.Add(button_cancel, 0, |
|---|
| 69 | wx.LEFT | wx.RIGHT | wx.ADJUST_MINSIZE, 10) |
|---|
| 70 | button_sizer.Add(button_OK, 0, |
|---|
| 71 | wx.LEFT | wx.RIGHT | wx.ADJUST_MINSIZE, 10) |
|---|
| 72 | vbox.Add(hint_sizer, 0, wx.EXPAND | wx.ALL, 10) |
|---|
| 73 | vbox.Add(selection_sizer, 0, wx.TOP | wx.BOTTOM, 10) |
|---|
| 74 | vbox.Add(wx.StaticLine(self, wx.ID_ANY), 0, wx.EXPAND, 0) |
|---|
| 75 | vbox.Add(button_sizer, 0, wx.TOP | wx.BOTTOM, 10) |
|---|
| 76 | self.SetSizer(vbox) |
|---|
| 77 | self.Layout() |
|---|
| 78 | |
|---|
| 79 | def get_data(self): |
|---|
| 80 | """ |
|---|
| 81 | return 1 if user requested Data1D , 2 if user requested Data2D |
|---|
| 82 | """ |
|---|
| 83 | if self.data_1d_selected.GetValue(): |
|---|
| 84 | return 1 |
|---|
| 85 | else: |
|---|
| 86 | return 2 |
|---|
| 87 | |
|---|
| 88 | |
|---|
| 89 | class DataDialog(wx.Dialog): |
|---|
| 90 | """ |
|---|
| 91 | Allow file selection at loading time |
|---|
| 92 | """ |
|---|
| 93 | def __init__(self, data_list, parent=None, text='', |
|---|
| 94 | nb_data=MAX_NBR_DATA, *args, **kwds): |
|---|
| 95 | wx.Dialog.__init__(self, parent, *args, **kwds) |
|---|
| 96 | self.SetTitle("Data Selection") |
|---|
| 97 | self._max_data = nb_data |
|---|
| 98 | self._nb_selected_data = nb_data |
|---|
| 99 | |
|---|
| 100 | self.SetSize((WIDTH, HEIGHT)) |
|---|
| 101 | self.list_of_ctrl = [] |
|---|
| 102 | if not data_list: |
|---|
| 103 | return |
|---|
| 104 | select_data_text = " %s Data selected.\n" % str(self._nb_selected_data) |
|---|
| 105 | self._data_text_ctrl = wx.StaticText(self, wx.ID_ANY, str(select_data_text)) |
|---|
| 106 | |
|---|
| 107 | self._data_text_ctrl.SetForegroundColour('blue') |
|---|
| 108 | self._sizer_main = wx.BoxSizer(wx.VERTICAL) |
|---|
| 109 | self._sizer_txt = wx.BoxSizer(wx.VERTICAL) |
|---|
| 110 | self._sizer_button = wx.BoxSizer(wx.HORIZONTAL) |
|---|
| 111 | self._choice_sizer = wx.GridBagSizer(5, 5) |
|---|
| 112 | self._panel = DialogPanel(self, style=wx.RAISED_BORDER, |
|---|
| 113 | size=(WIDTH - 20, HEIGHT / 3)) |
|---|
| 114 | self.__do_layout(data_list, text=text) |
|---|
| 115 | |
|---|
| 116 | def __do_layout(self, data_list, text=''): |
|---|
| 117 | """ |
|---|
| 118 | layout the dialog |
|---|
| 119 | """ |
|---|
| 120 | if not data_list or len(data_list) <= 1: |
|---|
| 121 | return |
|---|
| 122 | #add text |
|---|
| 123 | if text.strip() == "": |
|---|
| 124 | text = "Fitting: We recommend that you selected" |
|---|
| 125 | text += " no more than '%s' data\n" % str(self._max_data) |
|---|
| 126 | text += "for adequate plot display size. \n" |
|---|
| 127 | text += "unchecked data won't be send to fitting . \n" |
|---|
| 128 | text_ctrl = wx.StaticText(self, wx.ID_ANY, str(text)) |
|---|
| 129 | self._sizer_txt.Add(text_ctrl) |
|---|
| 130 | iy = 0 |
|---|
| 131 | ix = 0 |
|---|
| 132 | data_count = 0 |
|---|
| 133 | for i in range(len(data_list)): |
|---|
| 134 | data_count += 1 |
|---|
| 135 | cb = wx.CheckBox(self._panel, wx.ID_ANY, str(data_list[i].name), (10, 10)) |
|---|
| 136 | wx.EVT_CHECKBOX(self, cb.GetId(), self._count_selected_data) |
|---|
| 137 | if data_count <= MAX_NBR_DATA: |
|---|
| 138 | cb.SetValue(True) |
|---|
| 139 | else: |
|---|
| 140 | cb.SetValue(False) |
|---|
| 141 | self.list_of_ctrl.append((cb, data_list[i])) |
|---|
| 142 | self._choice_sizer.Add(cb, (iy, ix), |
|---|
| 143 | (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15) |
|---|
| 144 | iy += 1 |
|---|
| 145 | self._panel.SetSizer(self._choice_sizer) |
|---|
| 146 | #add sizer |
|---|
| 147 | self._sizer_button.Add((20, 20), 1, wx.EXPAND | wx.ADJUST_MINSIZE, 0) |
|---|
| 148 | button_cancel = wx.Button(self, wx.ID_CANCEL, "Cancel") |
|---|
| 149 | self._sizer_button.Add(button_cancel, 0, |
|---|
| 150 | wx.LEFT | wx.RIGHT | wx.ADJUST_MINSIZE, 10) |
|---|
| 151 | button_OK = wx.Button(self, wx.ID_OK, "Ok") |
|---|
| 152 | button_OK.SetFocus() |
|---|
| 153 | self._sizer_button.Add(button_OK, 0, |
|---|
| 154 | wx.LEFT | wx.RIGHT | wx.ADJUST_MINSIZE, 10) |
|---|
| 155 | static_line = wx.StaticLine(self, wx.ID_ANY) |
|---|
| 156 | self._sizer_txt.Add(self._panel, 0, wx.EXPAND | wx.ALL, 10) |
|---|
| 157 | self._sizer_main.Add(self._sizer_txt, 0, wx.EXPAND | wx.ALL, 10) |
|---|
| 158 | self._sizer_main.Add(self._data_text_ctrl, 0, wx.EXPAND | wx.ALL, 10) |
|---|
| 159 | self._sizer_main.Add(static_line, 0, wx.EXPAND, 0) |
|---|
| 160 | self._sizer_main.Add(self._sizer_button, 0, wx.EXPAND | wx.ALL, 10) |
|---|
| 161 | self.SetSizer(self._sizer_main) |
|---|
| 162 | self.Layout() |
|---|
| 163 | |
|---|
| 164 | def get_data(self): |
|---|
| 165 | """ |
|---|
| 166 | return the selected data |
|---|
| 167 | """ |
|---|
| 168 | temp = [] |
|---|
| 169 | for item in self.list_of_ctrl: |
|---|
| 170 | cb, data = item |
|---|
| 171 | if cb.GetValue(): |
|---|
| 172 | temp.append(data) |
|---|
| 173 | return temp |
|---|
| 174 | |
|---|
| 175 | def _count_selected_data(self, event): |
|---|
| 176 | """ |
|---|
| 177 | count selected data |
|---|
| 178 | """ |
|---|
| 179 | if event.GetEventObject().GetValue(): |
|---|
| 180 | self._nb_selected_data += 1 |
|---|
| 181 | else: |
|---|
| 182 | self._nb_selected_data -= 1 |
|---|
| 183 | select_data_text = " %s Data selected.\n" % str(self._nb_selected_data) |
|---|
| 184 | self._data_text_ctrl.SetLabel(select_data_text) |
|---|
| 185 | if self._nb_selected_data <= self._max_data: |
|---|
| 186 | self._data_text_ctrl.SetForegroundColour('blue') |
|---|
| 187 | else: |
|---|
| 188 | self._data_text_ctrl.SetForegroundColour('red') |
|---|
