[6cfe703] | 1 | #!/usr/bin/python |
---|
| 2 | |
---|
| 3 | # fitDialog.py |
---|
| 4 | |
---|
| 5 | import wx |
---|
[e40b651] | 6 | from PlotPanel import PlotPanel |
---|
| 7 | from plottables import Theory1D |
---|
| 8 | import math,pylab,fittings |
---|
[831149e] | 9 | import transform |
---|
| 10 | |
---|
[05da1f89] | 11 | def format_number(value, high=False): |
---|
| 12 | """ |
---|
| 13 | Return a float in a standardized, human-readable formatted string |
---|
| 14 | """ |
---|
[052a66bc] | 15 | try: |
---|
| 16 | value = float(value) |
---|
| 17 | except: |
---|
| 18 | print "returning 0" |
---|
| 19 | return "0" |
---|
| 20 | |
---|
[05da1f89] | 21 | if high: |
---|
| 22 | return "%-6.4g" % value |
---|
| 23 | else: |
---|
| 24 | return "%-5.3g" % value |
---|
| 25 | |
---|
[831149e] | 26 | |
---|
[6cfe703] | 27 | class LinearFit(wx.Dialog): |
---|
[f52bea1] | 28 | def __init__(self, parent, plottable, push_data,transform, id, title): |
---|
[1c94a9f1] | 29 | wx.Dialog.__init__(self, parent, id, title, size=(400, 350)) |
---|
[05da1f89] | 30 | |
---|
[6cfe703] | 31 | """ |
---|
[b43a009] | 32 | Dialog window pops- up when select Linear fit on Context menu |
---|
| 33 | Displays fitting parameters |
---|
[6cfe703] | 34 | """ |
---|
| 35 | self.parent = parent |
---|
[f52bea1] | 36 | self.transform = transform |
---|
[05da1f89] | 37 | |
---|
[1c94a9f1] | 38 | # Registered owner for close event |
---|
| 39 | self._registered_close = None |
---|
| 40 | |
---|
[e40b651] | 41 | #dialog panel self call function to plot the fitting function |
---|
| 42 | self.push_data = push_data |
---|
| 43 | #dialog self plottable |
---|
| 44 | self.plottable = plottable |
---|
[05da1f89] | 45 | |
---|
[ddff053] | 46 | # Receive transformations of x and y |
---|
[b43a009] | 47 | self.xLabel,self.yLabel,self.Avalue,self.Bvalue,\ |
---|
| 48 | self.ErrAvalue,self.ErrBvalue,self.Chivalue= self.transform() |
---|
[05da1f89] | 49 | |
---|
[e40b651] | 50 | #Dialog interface |
---|
[6cfe703] | 51 | vbox = wx.BoxSizer(wx.VERTICAL) |
---|
[106ef4d] | 52 | sizer = wx.GridBagSizer(5,5) |
---|
| 53 | |
---|
[05da1f89] | 54 | _BOX_WIDTH = 100 |
---|
[6cfe703] | 55 | |
---|
[05da1f89] | 56 | self.tcA = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20)) |
---|
| 57 | self.tcA.SetToolTipString("Fit value for the slope parameter.") |
---|
| 58 | self.tcErrA = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20)) |
---|
| 59 | self.tcErrA.SetToolTipString("Error on the slope parameter.") |
---|
| 60 | self.tcB = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20)) |
---|
| 61 | self.tcA.SetToolTipString("Fit value for the constant parameter.") |
---|
| 62 | self.tcErrB = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20)) |
---|
| 63 | self.tcErrB.SetToolTipString("Error on the constant parameter.") |
---|
| 64 | self.tcChi = wx.TextCtrl(self, -1,size=(_BOX_WIDTH,20)) |
---|
| 65 | self.tcChi.SetToolTipString("Chi^2 over degrees of freedom.") |
---|
| 66 | self.xminFit = wx.TextCtrl(self,-1,size=(_BOX_WIDTH,20)) |
---|
[6a8adb0] | 67 | self.xminFit.SetToolTipString("Enter the minimum value on the x-axis to be included in the fit.") |
---|
[05da1f89] | 68 | self.xmaxFit = wx.TextCtrl(self,-1,size=(_BOX_WIDTH,20)) |
---|
[6a8adb0] | 69 | self.xmaxFit.SetToolTipString("Enter the maximum value on the x-axis to be included in the fit.") |
---|
[05da1f89] | 70 | self.initXmin = wx.TextCtrl(self,-1,size=(_BOX_WIDTH,20)) |
---|
[1c94a9f1] | 71 | self.initXmin.SetToolTipString("Minimum value on the x-axis for the plotted data.") |
---|
[05da1f89] | 72 | self.initXmax = wx.TextCtrl(self,-1,size=(_BOX_WIDTH,20)) |
---|
[1c94a9f1] | 73 | self.initXmax.SetToolTipString("Maximum value on the x-axis for the plotted data.") |
---|
[05da1f89] | 74 | |
---|
| 75 | # Make the info box not editable |
---|
| 76 | #_BACKGROUND_COLOR = '#ffdf85' |
---|
| 77 | _BACKGROUND_COLOR = self.GetBackgroundColour() |
---|
| 78 | self.initXmin.SetEditable(False) |
---|
| 79 | self.initXmin.SetBackgroundColour(_BACKGROUND_COLOR) |
---|
| 80 | self.initXmax.SetEditable(False) |
---|
| 81 | self.initXmax.SetBackgroundColour(_BACKGROUND_COLOR) |
---|
| 82 | |
---|
| 83 | |
---|
| 84 | # Buttons on the bottom |
---|
| 85 | self.static_line_1 = wx.StaticLine(self, -1) |
---|
| 86 | self.btFit =wx.Button(self,-1,'Fit') |
---|
| 87 | self.btFit.Bind(wx.EVT_BUTTON, self._onFit) |
---|
| 88 | self.btFit.SetToolTipString("Perform fit.") |
---|
| 89 | self.btClose =wx.Button(self, wx.ID_CANCEL,'Close') |
---|
[1c94a9f1] | 90 | self.btClose.Bind(wx.EVT_BUTTON, self._on_close) |
---|
[05da1f89] | 91 | |
---|
| 92 | # Intro |
---|
| 93 | explanation = "Perform fit for y(x) = Ax + B" |
---|
| 94 | |
---|
| 95 | vbox.Add(sizer) |
---|
[8e44d51] | 96 | |
---|
[106ef4d] | 97 | ix = 0 |
---|
[6cfe703] | 98 | iy = 1 |
---|
[05da1f89] | 99 | sizer.Add(wx.StaticText(self, -1, explanation),(iy, ix),\ |
---|
| 100 | (1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) |
---|
| 101 | iy += 2 |
---|
| 102 | sizer.Add(wx.StaticText(self, -1, 'Parameter A'),(iy, ix),\ |
---|
[106ef4d] | 103 | (1,1), wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) |
---|
[6cfe703] | 104 | ix += 1 |
---|
[106ef4d] | 105 | sizer.Add(self.tcA,(iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 106 | ix += 1 |
---|
[05da1f89] | 107 | sizer.Add(wx.StaticText(self, -1, '+/-'),(iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 108 | ix += 1 |
---|
[106ef4d] | 109 | sizer.Add(self.tcErrA, (iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 110 | iy += 1 |
---|
[106ef4d] | 111 | ix = 0 |
---|
[05da1f89] | 112 | sizer.Add(wx.StaticText(self, -1, 'Parameter B'),(iy, ix),(1,1),\ |
---|
[106ef4d] | 113 | wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) |
---|
[6cfe703] | 114 | ix += 1 |
---|
[106ef4d] | 115 | sizer.Add(self.tcB, (iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 116 | ix += 1 |
---|
[05da1f89] | 117 | sizer.Add(wx.StaticText(self, -1, '+/-'),(iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 118 | ix += 1 |
---|
[106ef4d] | 119 | sizer.Add(self.tcErrB, (iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 120 | iy += 1 |
---|
[106ef4d] | 121 | ix = 0 |
---|
[05da1f89] | 122 | sizer.Add(wx.StaticText(self, -1, 'Chi2/dof'),(iy, ix),(1,1),\ |
---|
[106ef4d] | 123 | wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) |
---|
[6cfe703] | 124 | ix += 1 |
---|
[106ef4d] | 125 | sizer.Add(self.tcChi, (iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[05da1f89] | 126 | iy += 2 |
---|
[6cfe703] | 127 | ix = 1 |
---|
[05da1f89] | 128 | sizer.Add(wx.StaticText(self, -1, 'Min'),(iy, ix),(1,1),\ |
---|
[106ef4d] | 129 | wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 130 | ix += 2 |
---|
[05da1f89] | 131 | sizer.Add(wx.StaticText(self, -1, 'Max'),(iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 132 | iy += 1 |
---|
[831149e] | 133 | ix = 0 |
---|
[1c94a9f1] | 134 | sizer.Add(wx.StaticText(self, -1, 'Maximum range (linear scale)'),(iy, ix),(1,1),\ |
---|
[831149e] | 135 | wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) |
---|
| 136 | ix +=1 |
---|
[83fe2cc] | 137 | sizer.Add(self.initXmin, (iy, ix),(1,1),\ |
---|
[831149e] | 138 | wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
| 139 | ix += 2 |
---|
[83fe2cc] | 140 | sizer.Add(self.initXmax, (iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[b43a009] | 141 | |
---|
[ddff053] | 142 | iy += 1 |
---|
[831149e] | 143 | ix = 0 |
---|
[6a8adb0] | 144 | sizer.Add(wx.StaticText(self, -1, 'Fit range of '+self.xLabel),(iy, ix),(1,1),\ |
---|
[831149e] | 145 | wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 15) |
---|
| 146 | ix += 1 |
---|
[83fe2cc] | 147 | sizer.Add(self.xminFit, (iy, ix),(1,1),\ |
---|
[ddff053] | 148 | wx.LEFT|wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
| 149 | ix += 2 |
---|
[83fe2cc] | 150 | sizer.Add(self.xmaxFit, (iy, ix),(1,1), wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
[6cfe703] | 151 | iy += 1 |
---|
[106ef4d] | 152 | ix = 1 |
---|
| 153 | |
---|
[05da1f89] | 154 | vbox.Add(self.static_line_1, 0, wx.EXPAND, 0) |
---|
| 155 | |
---|
| 156 | sizer_button = wx.BoxSizer(wx.HORIZONTAL) |
---|
| 157 | sizer_button.Add((20, 20), 1, wx.EXPAND|wx.ADJUST_MINSIZE, 0) |
---|
| 158 | sizer_button.Add(self.btFit, 0, wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10) |
---|
| 159 | sizer_button.Add(self.btClose, 0, wx.LEFT|wx.RIGHT|wx.ADJUST_MINSIZE, 10) |
---|
| 160 | vbox.Add(sizer_button, 0, wx.EXPAND|wx.BOTTOM|wx.TOP, 10) |
---|
| 161 | |
---|
| 162 | |
---|
| 163 | |
---|
[106ef4d] | 164 | sizer.Add(self.btFit, (iy, ix),(1,1), wx.LEFT|wx.ADJUST_MINSIZE, 0) |
---|
| 165 | |
---|
[05da1f89] | 166 | |
---|
| 167 | #panel.SetSizer(sizer) |
---|
[6cfe703] | 168 | self.SetSizer(vbox) |
---|
| 169 | self.Centre() |
---|
[8e44d51] | 170 | |
---|
[e40b651] | 171 | # Receives the type of model for the fitting |
---|
| 172 | from LineModel import LineModel |
---|
| 173 | self.model = LineModel() |
---|
[1fdb81d] | 174 | |
---|
| 175 | #Display the fittings values |
---|
| 176 | self.default_A = self.model.getParam('A') |
---|
| 177 | self.default_B = self.model.getParam('B') |
---|
| 178 | self.cstA = fittings.Parameter(self.model, 'A', self.default_A) |
---|
| 179 | self.cstB = fittings.Parameter(self.model, 'B', self.default_B) |
---|
| 180 | |
---|
| 181 | # Set default value of parameter in fit dialog |
---|
[90640f2] | 182 | |
---|
[b43a009] | 183 | if self.Avalue==None: |
---|
[052a66bc] | 184 | self.tcA.SetLabel(format_number(self.default_A)) |
---|
[b43a009] | 185 | else : |
---|
[052a66bc] | 186 | self.tcA.SetLabel(format_number(self.Avalue)) |
---|
[b43a009] | 187 | if self.Bvalue==None: |
---|
[052a66bc] | 188 | self.tcB.SetLabel(format_number(self.default_B)) |
---|
[b43a009] | 189 | else: |
---|
[052a66bc] | 190 | self.tcB.SetLabel(format_number(self.Bvalue)) |
---|
[b43a009] | 191 | if self.ErrAvalue==None: |
---|
[052a66bc] | 192 | self.tcErrA.SetLabel(format_number(0.0)) |
---|
[b43a009] | 193 | else: |
---|
[052a66bc] | 194 | self.tcErrA.SetLabel(format_number(self.ErrAvalue)) |
---|
[b43a009] | 195 | if self.ErrBvalue==None: |
---|
[052a66bc] | 196 | self.tcErrB.SetLabel(format_number(0.0)) |
---|
[b43a009] | 197 | else: |
---|
[052a66bc] | 198 | self.tcErrB.SetLabel(format_number(self.ErrBvalue)) |
---|
[b43a009] | 199 | if self.Chivalue==None: |
---|
[052a66bc] | 200 | self.tcChi.SetLabel(format_number(0.0)) |
---|
[b43a009] | 201 | else: |
---|
[052a66bc] | 202 | self.tcChi.SetLabel(format_number(self.Chivalue)) |
---|
[831149e] | 203 | if self.plottable.x !=[]: |
---|
[83fe2cc] | 204 | #store the values of View in self.x,self.y,self.dx,self.dy |
---|
| 205 | self.x,self.y,self.dx,self.dy= self.plottable.returnValuesOfView() |
---|
[90640f2] | 206 | |
---|
[1c94a9f1] | 207 | self.mini =min(self.x) |
---|
| 208 | self.maxi =max(self.x) |
---|
[90640f2] | 209 | |
---|
[05da1f89] | 210 | |
---|
[1c94a9f1] | 211 | self.initXmin.SetValue(format_number(min(self.plottable.x))) |
---|
| 212 | self.initXmax.SetValue(format_number(max(self.plottable.x))) |
---|
[90640f2] | 213 | |
---|
[05da1f89] | 214 | self.xminFit.SetLabel(format_number(self.mini)) |
---|
| 215 | self.xmaxFit.SetLabel(format_number(self.maxi)) |
---|
[5e14aee] | 216 | |
---|
[90640f2] | 217 | |
---|
[1c94a9f1] | 218 | def register_close(self, owner): |
---|
| 219 | """ |
---|
| 220 | Method to register the close event to a parent |
---|
| 221 | window that needs notification when the dialog |
---|
| 222 | is closed |
---|
| 223 | @param owner: parent window |
---|
| 224 | """ |
---|
| 225 | self._registered_close = owner |
---|
| 226 | |
---|
| 227 | def _on_close(self, event): |
---|
| 228 | """ |
---|
| 229 | Close event. |
---|
| 230 | Notify registered owner if available. |
---|
| 231 | """ |
---|
| 232 | event.Skip() |
---|
| 233 | if self._registered_close is not None: |
---|
| 234 | self._registered_close() |
---|
| 235 | |
---|
[6cfe703] | 236 | def _onFit(self ,event): |
---|
[f52bea1] | 237 | """ |
---|
| 238 | Performs the fit. Receive an event when clicking on the button Fit.Computes chisqr , |
---|
| 239 | A and B parameters of the best linear fit y=Ax +B |
---|
| 240 | Push a plottable to |
---|
| 241 | """ |
---|
| 242 | tempx=[] |
---|
| 243 | tempy=[] |
---|
[7a03e65] | 244 | tempdy = [] |
---|
[ddff053] | 245 | |
---|
[83fe2cc] | 246 | |
---|
| 247 | |
---|
[1fdb81d] | 248 | |
---|
[106ef4d] | 249 | # Check if View contains a x array .we online fit when x exits |
---|
| 250 | # makes transformation for y as a line to fit |
---|
[83fe2cc] | 251 | if self.x != []: |
---|
[ddff053] | 252 | |
---|
| 253 | |
---|
[83fe2cc] | 254 | if(self.checkFitValues(self.xminFit) == True): |
---|
[f193585] | 255 | #Check if the field of Fit Dialog contain values and use the x max and min of the user |
---|
[1c94a9f1] | 256 | xminView,xmaxView = self._checkVal(self.xminFit.GetValue(),self.xmaxFit.GetValue()) |
---|
| 257 | xmin = self.floatInvTransform(xminView) |
---|
| 258 | xmax = self.floatInvTransform(xmaxView) |
---|
| 259 | |
---|
[ddff053] | 260 | |
---|
[f193585] | 261 | # Store the transformed values of view x, y,dy in variables before the fit |
---|
[b43a009] | 262 | if self.yLabel.lower() == "log10(y)": |
---|
| 263 | if (self.xLabel.lower() == "log10(x)"): |
---|
[83fe2cc] | 264 | for i in range(len(self.x)): |
---|
| 265 | if self.x[i]>= math.log10(xmin): |
---|
| 266 | tempy.append(math.log10(self.y[i])) |
---|
| 267 | tempdy.append(transform.errToLogX(self.y[i],0,self.dy[i],0)) |
---|
[f193585] | 268 | else: |
---|
[83fe2cc] | 269 | for i in range(len(self.y)): |
---|
| 270 | tempy.append(math.log10(self.y[i])) |
---|
| 271 | tempdy.append(transform.errToLogX(self.y[i],0,self.dy[i],0)) |
---|
[f193585] | 272 | else: |
---|
[83fe2cc] | 273 | tempy = self.y |
---|
| 274 | tempdy = self.dy |
---|
[f193585] | 275 | |
---|
[b43a009] | 276 | if (self.xLabel.lower() == "log10(x)"): |
---|
[83fe2cc] | 277 | for x_i in self.x: |
---|
[34ae302] | 278 | if x_i >= math.log10(xmin): |
---|
[f193585] | 279 | tempx.append(math.log10(x_i)) |
---|
| 280 | else: |
---|
[83fe2cc] | 281 | tempx = self.x |
---|
[b43a009] | 282 | |
---|
[f193585] | 283 | #Find the fitting parameters |
---|
[ad8bcd6] | 284 | # Always use the same defaults, so that fit history doesn't play a role! |
---|
| 285 | self.cstA = fittings.Parameter(self.model, 'A', self.default_A) |
---|
| 286 | self.cstB = fittings.Parameter(self.model, 'B', self.default_B) |
---|
[ddff053] | 287 | |
---|
[b43a009] | 288 | if (self.xLabel.lower() == "log10(x)"): |
---|
[34ae302] | 289 | chisqr, out, cov = fittings.sansfit(self.model, [self.cstA, self.cstB], |
---|
| 290 | tempx, tempy,tempdy,math.log10(xmin),math.log10(xmax)) |
---|
| 291 | else: |
---|
| 292 | chisqr, out, cov = fittings.sansfit(self.model, |
---|
[f193585] | 293 | [self.cstA, self.cstB],tempx, tempy,tempdy,xminView,xmaxView) |
---|
[05da1f89] | 294 | |
---|
| 295 | # Use chi2/dof |
---|
| 296 | if len(tempx)>0: |
---|
| 297 | chisqr = chisqr/len(tempx) |
---|
| 298 | |
---|
[f193585] | 299 | #Check that cov and out are iterable before displaying them |
---|
| 300 | if cov ==None: |
---|
| 301 | errA =0.0 |
---|
| 302 | errB =0.0 |
---|
| 303 | else: |
---|
| 304 | errA= math.sqrt(cov[0][0]) |
---|
| 305 | errB= math.sqrt(cov[1][1]) |
---|
| 306 | if out==None: |
---|
| 307 | cstA=0.0 |
---|
| 308 | cstB=0.0 |
---|
| 309 | else: |
---|
| 310 | cstA=out[0] |
---|
| 311 | cstB=out[1] |
---|
| 312 | # Reset model with the right values of A and B |
---|
| 313 | self.model.setParam('A', float(cstA)) |
---|
| 314 | self.model.setParam('B', float(cstB)) |
---|
[3aa7074] | 315 | |
---|
[f193585] | 316 | tempx = [] |
---|
| 317 | tempy = [] |
---|
| 318 | y_model = 0.0 |
---|
| 319 | # load tempy with the minimum transformation |
---|
| 320 | |
---|
[b43a009] | 321 | if self.xLabel == "log10(x)": |
---|
[f193585] | 322 | y_model = self.model.run(math.log10(xmin)) |
---|
| 323 | tempx.append(xmin) |
---|
| 324 | else: |
---|
| 325 | y_model = self.model.run(xminView) |
---|
| 326 | tempx.append(xminView) |
---|
| 327 | |
---|
[b43a009] | 328 | if self.yLabel == "log10(y)": |
---|
[f193585] | 329 | tempy.append(math.pow(10,y_model)) |
---|
| 330 | print "tempy",tempy |
---|
| 331 | else: |
---|
| 332 | tempy.append(y_model) |
---|
| 333 | |
---|
| 334 | # load tempy with the maximum transformation |
---|
[b43a009] | 335 | if self.xLabel == "log10(x)": |
---|
[f193585] | 336 | y_model = self.model.run(math.log10(xmax)) |
---|
| 337 | tempx.append(xmax) |
---|
| 338 | else: |
---|
| 339 | y_model = self.model.run(xmaxView) |
---|
| 340 | tempx.append(xmaxView) |
---|
| 341 | |
---|
[b43a009] | 342 | if self.yLabel == "log10(y)": |
---|
[f193585] | 343 | tempy.append(math.pow(10,y_model)) |
---|
| 344 | else: |
---|
| 345 | tempy.append(y_model) |
---|
[83fe2cc] | 346 | #Set the fit parameter display when FitDialog is opened again |
---|
[2e07e8f] | 347 | self.Avalue=cstB |
---|
| 348 | self.Bvalue=cstA |
---|
[b43a009] | 349 | self.ErrAvalue=errA |
---|
| 350 | self.ErrBvalue=errB |
---|
| 351 | self.Chivalue=chisqr |
---|
| 352 | self.push_data(tempx,tempy,xminView,xmaxView,xmin,xmax,self._ongetValues()) |
---|
[f193585] | 353 | |
---|
| 354 | # Display the fitting value on the Fit Dialog |
---|
[2e07e8f] | 355 | self._onsetValues(cstB, cstA, errA,errB,chisqr) |
---|
[b43a009] | 356 | |
---|
| 357 | |
---|
[7a03e65] | 358 | |
---|
[6cfe703] | 359 | def _onsetValues(self,cstA,cstB,errA,errB,Chi): |
---|
[f52bea1] | 360 | """ |
---|
| 361 | Display the value on fit Dialog |
---|
| 362 | """ |
---|
[05da1f89] | 363 | self.tcA.SetValue(format_number(cstA)) |
---|
| 364 | self.tcB.SetValue(format_number(cstB)) |
---|
| 365 | self.tcErrA.SetValue(format_number(errA)) |
---|
| 366 | self.tcErrB.SetValue(format_number(errB)) |
---|
| 367 | self.tcChi.SetValue(format_number(Chi)) |
---|
[b43a009] | 368 | |
---|
| 369 | def _ongetValues(self): |
---|
| 370 | """ |
---|
| 371 | Display the value on fit Dialog |
---|
| 372 | """ |
---|
| 373 | return self.Avalue, self.Bvalue,self.ErrAvalue,self.ErrBvalue,self.Chivalue |
---|
[e40b651] | 374 | |
---|
| 375 | |
---|
[831149e] | 376 | def _checkVal(self,usermin, usermax): |
---|
[6cfe703] | 377 | """ |
---|
[831149e] | 378 | Ensure that fields parameter contains a min and a max value |
---|
| 379 | within x min and x max range |
---|
[6cfe703] | 380 | """ |
---|
[831149e] | 381 | if float(usermin) < float(usermax): |
---|
| 382 | if float(usermin) >= float(self.mini) and float(usermin) < float(self.maxi): |
---|
[05da1f89] | 383 | self.xminFit.SetValue(format_number(float(usermin))) |
---|
[831149e] | 384 | else: |
---|
[05da1f89] | 385 | self.xminFit.SetValue(format_number(float(self.mini))) |
---|
[831149e] | 386 | |
---|
| 387 | if float(usermax) > float(self.mini) and float(usermax) <= float(self.maxi): |
---|
[05da1f89] | 388 | self.xmaxFit.SetLabel(format_number(float(usermax))) |
---|
[831149e] | 389 | else: |
---|
[05da1f89] | 390 | self.xmaxFit.SetLabel(format_number(float(self.maxi))) |
---|
[831149e] | 391 | |
---|
[83fe2cc] | 392 | mini =float(self.xminFit.GetValue()) |
---|
| 393 | maxi =float(self.xmaxFit.GetValue()) |
---|
[831149e] | 394 | |
---|
| 395 | return mini, maxi |
---|
| 396 | def floatTransform(self,x): |
---|
| 397 | """ |
---|
| 398 | transform a float.It is use to determine the x.View min and x.View max for values |
---|
| 399 | not in x |
---|
| 400 | """ |
---|
[1c94a9f1] | 401 | #TODO: refactor this with proper object-oriented design |
---|
| 402 | # This code stinks. |
---|
| 403 | |
---|
[b43a009] | 404 | if ( self.xLabel=="x" ): |
---|
[831149e] | 405 | return transform.toX(x) |
---|
| 406 | |
---|
[b43a009] | 407 | if ( self.xLabel=="x^(2)" ): |
---|
[831149e] | 408 | return transform.toX2(x) |
---|
| 409 | |
---|
[b43a009] | 410 | if (self.xLabel=="log10(x)" ): |
---|
[831149e] | 411 | if x >0: |
---|
[34ae302] | 412 | return x |
---|
[831149e] | 413 | else: |
---|
| 414 | raise ValueError,"cannot compute log of a negative number" |
---|
[f193585] | 415 | |
---|
[1c94a9f1] | 416 | def floatInvTransform(self,x): |
---|
| 417 | """ |
---|
| 418 | transform a float.It is use to determine the x.View min and x.View max for values |
---|
| 419 | not in x |
---|
| 420 | """ |
---|
| 421 | #TODO: refactor this. This is just a hack to make the |
---|
| 422 | # functionality work without rewritting the whole code |
---|
| 423 | # with good design (which really should be done...). |
---|
| 424 | |
---|
| 425 | if ( self.xLabel=="x^(2)" ): |
---|
| 426 | return math.sqrt(x) |
---|
| 427 | |
---|
| 428 | elif (self.xLabel=="log10(x)" ): |
---|
| 429 | return math.pow(10, x) |
---|
| 430 | |
---|
| 431 | return x |
---|
| 432 | |
---|
[f193585] | 433 | def checkFitValues(self,item): |
---|
| 434 | """ |
---|
| 435 | Check the validity of input values |
---|
| 436 | """ |
---|
| 437 | flag = True |
---|
| 438 | value = item.GetValue() |
---|
| 439 | # Check for possible values entered |
---|
[b43a009] | 440 | if (self.xLabel=="log10(x)"): |
---|
[f193585] | 441 | if (float(value) > 0): |
---|
| 442 | item.SetBackgroundColour(wx.WHITE) |
---|
| 443 | item.Refresh() |
---|
| 444 | else: |
---|
| 445 | flag = False |
---|
| 446 | item.SetBackgroundColour("pink") |
---|
| 447 | item.Refresh() |
---|
| 448 | |
---|
| 449 | return flag |
---|
[831149e] | 450 | |
---|
[83fe2cc] | 451 | def setFitRange(self,xmin,xmax,xminTrans,xmaxTrans): |
---|
| 452 | """ |
---|
| 453 | Set fit parameters |
---|
| 454 | """ |
---|
[05da1f89] | 455 | self.xminFit.SetValue(format_number(xmin)) |
---|
| 456 | self.xmaxFit.SetValue(format_number(xmax)) |
---|
[ddff053] | 457 | |
---|
[1c94a9f1] | 458 | def set_fit_region(self, xmin, xmax): |
---|
| 459 | """ |
---|
| 460 | Set the fit region |
---|
| 461 | @param xmin: minimum x-value to be included in fit |
---|
| 462 | @param xmax: maximum x-value to be included in fit |
---|
| 463 | """ |
---|
| 464 | # Check values |
---|
| 465 | try: |
---|
| 466 | float(xmin) |
---|
| 467 | float(xmax) |
---|
| 468 | except: |
---|
| 469 | raise ValueError, "LinearFit.set_fit_region: fit range must be floats" |
---|
| 470 | self.xminFit.SetValue(format_number(xmin)) |
---|
| 471 | self.xmaxFit.SetValue(format_number(xmax)) |
---|
[05da1f89] | 472 | |
---|
| 473 | class MyApp(wx.App): |
---|
| 474 | def OnInit(self): |
---|
| 475 | wx.InitAllImageHandlers() |
---|
| 476 | plot = Theory1D([],[]) |
---|
| 477 | dialog = LinearFit(None, plot, self.onFitDisplay,self.returnTrans, -1, 'Linear Fit') |
---|
| 478 | if dialog.ShowModal() == wx.ID_OK: |
---|
| 479 | pass |
---|
| 480 | dialog.Destroy() |
---|
| 481 | |
---|
| 482 | return 1 |
---|
| 483 | |
---|
| 484 | def onFitDisplay(self, tempx,tempy,xminView,xmaxView,xmin,xmax,func): |
---|
| 485 | pass |
---|
| 486 | |
---|
| 487 | def returnTrans(self): |
---|
| 488 | return '','',0,0,0,0,0 |
---|
[6cfe703] | 489 | |
---|
[05da1f89] | 490 | # end of class MyApp |
---|
[6cfe703] | 491 | |
---|
[05da1f89] | 492 | if __name__ == "__main__": |
---|
| 493 | app = MyApp(0) |
---|
| 494 | app.MainLoop() |
---|