""" Invariant panel """ import wx import sys from sas.guiframe.utils import format_number from invariant_widgets import OutputTextCtrl # Dimensions related to chart RECTANGLE_WIDTH = 400.0 RECTANGLE_HEIGHT = 20 # Invariant panel size _BOX_WIDTH = 76 # scale to use for a bar of value zero RECTANGLE_SCALE = 0.0001 DEFAULT_QSTAR = 1.0 if sys.platform.count("win32") > 0: _STATICBOX_WIDTH = 450 PANEL_WIDTH = 500 PANEL_HEIGHT = 430 FONT_VARIANT = 0 else: _STATICBOX_WIDTH = 480 PANEL_WIDTH = 530 PANEL_HEIGHT = 430 FONT_VARIANT = 1 ERROR_COLOR = wx.Colour(255, 0, 0, 128) EXTRAPOLATION_COLOR = wx.Colour(169, 169, 168, 128) INVARIANT_COLOR = wx.Colour(67, 208, 128, 128) class InvariantContainer(wx.Object): """ This class stores some values resulting resulting from invariant calculations. Given the value of total invariant, this class can also determine the percentage of invariants resulting from extrapolation. """ def __init__(self): # invariant at low range self.qstar_low = None # invariant at low range error self.qstar_low_err = None # invariant self.qstar = None # invariant error self.qstar_err = None # invariant at high range self.qstar_high = None # invariant at high range error self.qstar_high_err = None # invariant total self.qstar_total = None # invariant error self.qstar_total_err = None # scale self.qstar_low_percent = None self.qstar_high_percent = None self.qstar_percent = None # warning message self.existing_warning = False self.warning_msg = "No Details on calculations available...\n" def compute_percentage(self): """ Compute percentage of each invariant """ if self.qstar_total is None: self.qstar_percent = None self.qstar_low = None self.qstar_high = None self.check_values() return # compute invariant percentage if self.qstar is None: self.qstar_percent = None else: try: self.qstar_percent = float(self.qstar) / float(self.qstar_total) except: self.qstar_percent = 'error' # compute low q invariant percentage if self.qstar_low is None: self.qstar_low_percent = None else: try: self.qstar_low_percent = float(self.qstar_low)\ / float(self.qstar_total) except: self.qstar_low_percent = 'error' # compute high q invariant percentage if self.qstar_high is None: self.qstar_high_percent = None else: try: self.qstar_high_percent = float(self.qstar_high)\ / float(self.qstar_total) except: self.qstar_high_percent = 'error' wx.CallAfter(self.check_values) def check_values(self): """ check the validity if invariant """ if self.qstar_total is None and self.qstar is None: self.warning_msg = "Invariant not calculated.\n" return if self.qstar_total == 0: self.existing_warning = True self.warning_msg = "Invariant is zero. \n" self.warning_msg += "The calculations are likely " self.warning_msg += "to be unreliable!\n" return # warning to the user when the extrapolated invariant is greater than %5 msg = '' if self.qstar_percent == 'error': try: float(self.qstar) except: self.existing_warning = True msg += 'Error occurred when computing invariant from data.\n ' if self.qstar_percent > 1: self.existing_warning = True msg += "Invariant Q contribution is greater " msg += "than 100% .\n" if self.qstar_low_percent == 'error': try: float(self.qstar_low) except: self.existing_warning = True msg += "Error occurred when computing extrapolated invariant" msg += " at low-Q region.\n" elif self.qstar_low_percent is not None: if self.qstar_low_percent >= 0.05: self.existing_warning = True msg += "Extrapolated contribution at Low Q is higher " msg += "than 5% of the invariant.\n" elif self.qstar_low_percent < 0: self.existing_warning = True msg += "Extrapolated contribution at Low Q < 0.\n" elif self.qstar_low_percent > 1: self.existing_warning = True msg += "Extrapolated contribution at Low Q is greater " msg += "than 100% .\n" if self.qstar_high_percent == 'error': try: float(self.qstar_high) except: self.existing_warning = True msg += 'Error occurred when computing extrapolated' msg += ' invariant at high-Q region.\n' elif self.qstar_high_percent is not None: if self.qstar_high_percent >= 0.05: self.existing_warning = True msg += "Extrapolated contribution at High Q is higher " msg += "than 5% of the invariant.\n" elif self.qstar_high_percent < 0: self.existing_warning = True msg += "Extrapolated contribution at High Q < 0.\n" elif self.qstar_high_percent > 1: self.existing_warning = True msg += "Extrapolated contribution at High Q is greater " msg += "than 100% .\n" if (self.qstar_low_percent not in [None, "error"]) and \ (self.qstar_high_percent not in [None, "error"])\ and self.qstar_low_percent + self.qstar_high_percent >= 0.05: self.existing_warning = True msg += "The sum of all extrapolated contributions is higher " msg += "than 5% of the invariant.\n" if self.existing_warning: self.warning_msg = '' self.warning_msg += msg self.warning_msg += "The calculations are likely to be" self.warning_msg += " unreliable!\n" else: self.warning_msg = "No Details on calculations available...\n" class InvariantDetailsPanel(wx.Dialog): """ This panel describes proportion of invariants """ def __init__(self, parent=None, id=-1, qstar_container=None, title="Invariant Details", size=(PANEL_WIDTH, PANEL_HEIGHT)): wx.Dialog.__init__(self, parent=parent, id=id, title=title, size=size) # Font size self.SetWindowVariant(variant=FONT_VARIANT) self.parent = parent # self.qstar_container self.qstar_container = qstar_container # warning message self.warning_msg = self.qstar_container.warning_msg # Define scale of each bar self.low_inv_percent = self.qstar_container.qstar_low_percent self.low_scale = self.get_scale(percentage=self.low_inv_percent, scale_name="Extrapolated at Low Q") self.inv_percent = self.qstar_container.qstar_percent self.inv_scale = self.get_scale(percentage=self.inv_percent, scale_name="Inv in Q range") self.high_inv_percent = self.qstar_container.qstar_high_percent self.high_scale = self.get_scale(percentage=self.high_inv_percent, scale_name="Extrapolated at High Q") # Default color the extrapolation bar is grey self.extrapolation_color_low = EXTRAPOLATION_COLOR self.extrapolation_color_high = EXTRAPOLATION_COLOR self.invariant_color = INVARIANT_COLOR # change color of high and low bar when necessary self.set_color_bar() # draw the panel itself self._do_layout() self.set_values() def _define_structure(self): """ Define main sizers needed for this panel """ # Box sizers must be defined first before defining buttons/textctrls # (MAC). self.main_sizer = wx.BoxSizer(wx.VERTICAL) # Sizer related to chart chart_box = wx.StaticBox(self, -1, "Invariant Chart") self.chart_sizer = wx.StaticBoxSizer(chart_box, wx.VERTICAL) self.chart_sizer.SetMinSize((PANEL_WIDTH - 50, 110)) # Sizer related to invariant values self.invariant_sizer = wx.GridBagSizer(4, 4) invariant_box = wx.StaticBox(self, -1, "Numerical Values") self.invariant_box_sizer = wx.StaticBoxSizer(invariant_box, wx.HORIZONTAL) self.invariant_box_sizer.SetMinSize((PANEL_WIDTH - 50, -1)) # Sizer related to warning message warning_box = wx.StaticBox(self, -1, "Warning") self.warning_sizer = wx.StaticBoxSizer(warning_box, wx.VERTICAL) self.warning_sizer.SetMinSize((PANEL_WIDTH - 50, -1)) # Sizer related to button self.button_sizer = wx.BoxSizer(wx.HORIZONTAL) def _layout_shart(self): """ Draw widgets related to chart """ self.panel_chart = wx.Panel(self) self.panel_chart.Bind(wx.EVT_PAINT, self.on_paint) self.chart_sizer.Add(self.panel_chart, 1, wx.EXPAND | wx.ALL, 0) def _layout_invariant(self): """ Draw widgets related to invariant """ uncertainty = "+/-" unit_invariant = '[1/(cm * A^3)]' invariant_txt = wx.StaticText(self, -1, 'Q* from Data ') invariant_txt.SetToolTipString("Invariant in the data set's Q range.") self.invariant_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1)) hint_msg = "Invariant in the data set's Q range." self.invariant_tcl.SetToolTipString(hint_msg) self.invariant_err_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1)) hint_msg = "Uncertainty on the invariant from data's range." self.invariant_err_tcl.SetToolTipString(hint_msg) invariant_units_txt = wx.StaticText(self, -1, unit_invariant) hint_msg = "Unit of the invariant from data's Q range" invariant_units_txt.SetToolTipString(hint_msg) invariant_low_txt = wx.StaticText(self, -1, 'Q* from Low-Q') hint_msg = "Extrapolated invariant from low-Q range." invariant_low_txt.SetToolTipString(hint_msg) self.invariant_low_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1)) hint_msg = "Extrapolated invariant from low-Q range." self.invariant_low_tcl.SetToolTipString(hint_msg) self.invariant_low_err_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1)) hint_msg = "Uncertainty on the invariant from low-Q range." self.invariant_low_err_tcl.SetToolTipString(hint_msg) invariant_low_units_txt = wx.StaticText(self, -1, unit_invariant) hint_msg = "Unit of the extrapolated invariant from low-Q range" invariant_low_units_txt.SetToolTipString(hint_msg) invariant_high_txt = wx.StaticText(self, -1, 'Q* from High-Q') hint_msg = "Extrapolated invariant from high-Q range" invariant_high_txt.SetToolTipString(hint_msg) self.invariant_high_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1)) hint_msg = "Extrapolated invariant from high-Q range" self.invariant_high_tcl.SetToolTipString(hint_msg) self.invariant_high_err_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1)) hint_msg = "Uncertainty on the invariant from high-Q range." self.invariant_high_err_tcl.SetToolTipString(hint_msg) invariant_high_units_txt = wx.StaticText(self, -1, unit_invariant) hint_msg = "Unit of the extrapolated invariant from high-Q range" invariant_high_units_txt.SetToolTipString(hint_msg) # Invariant low iy = 0 ix = 0 self.invariant_sizer.Add(invariant_low_txt, (iy, ix), (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15) ix += 1 self.invariant_sizer.Add(self.invariant_low_tcl, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty), (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(self.invariant_low_err_tcl, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(invariant_low_units_txt, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) # Invariant iy += 1 ix = 0 self.invariant_sizer.Add(invariant_txt, (iy, ix), (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15) ix += 1 self.invariant_sizer.Add(self.invariant_tcl, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty), (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(self.invariant_err_tcl, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(invariant_units_txt, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) # Invariant high iy += 1 ix = 0 self.invariant_sizer.Add(invariant_high_txt, (iy, ix), (1, 1), wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15) ix += 1 self.invariant_sizer.Add(self.invariant_high_tcl, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty), (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(self.invariant_high_err_tcl, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) ix += 1 self.invariant_sizer.Add(invariant_high_units_txt, (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0) self.invariant_box_sizer.Add(self.invariant_sizer, 0, wx.TOP | wx.BOTTOM, 10) def _layout_warning(self): """ Draw widgets related to warning """ # Warning [string] self.warning_msg_txt = wx.StaticText(self, -1, self.warning_msg) if self.qstar_container.existing_warning: self.warning_msg_txt.SetForegroundColour('red') self.warning_sizer.AddMany([(self.warning_msg_txt, 0, wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 10)]) def _layout_button(self): """ Draw widgets related to button """ # Close button id = wx.NewId() button_ok = wx.Button(self, id, "Ok") button_ok.SetToolTipString("Give Details on Computation") self.Bind(wx.EVT_BUTTON, self.on_close, id=id) self.button_sizer.AddMany([((20, 20), 0, wx.LEFT, 350), (button_ok, 0, wx.RIGHT, 10)]) def _do_layout(self): """ Draw window content """ self._define_structure() self._layout_shart() self._layout_invariant() self._layout_warning() self._layout_button() self.main_sizer.AddMany([(self.chart_sizer, 0, wx.ALL, 10), (self.invariant_box_sizer, 0, wx.ALL, 10), (self.warning_sizer, 0, wx.ALL, 10), (self.button_sizer, 0, wx.ALL, 10)]) self.SetSizer(self.main_sizer) def set_values(self): """ Set value of txtcrtl """ value = format_number(self.qstar_container.qstar) self.invariant_tcl.SetValue(value) value = format_number(self.qstar_container.qstar_err) self.invariant_err_tcl.SetValue(value) value = format_number(self.qstar_container.qstar_low) self.invariant_low_tcl.SetValue(value) value = format_number(self.qstar_container.qstar_low_err) self.invariant_low_err_tcl.SetValue(value) value = format_number(self.qstar_container.qstar_high) self.invariant_high_tcl.SetValue(value) value = format_number(self.qstar_container.qstar_high_err) self.invariant_high_err_tcl.SetValue(value) def get_scale(self, percentage, scale_name='scale'): """ Check scale receive in this panel. """ scale = RECTANGLE_SCALE try: if percentage in [None, 0.0, "error"]: scale = RECTANGLE_SCALE return scale elif percentage < 0: scale = RECTANGLE_SCALE return scale scale = float(percentage) except: scale = RECTANGLE_SCALE self.warning_msg += "Recieve an invalid scale for %s\n" self.warning_msg += "check this value : %s\n" % str(percentage) return scale def set_color_bar(self): """ Change the color for low and high bar when necessary """ self.extrapolation_color_low = EXTRAPOLATION_COLOR self.extrapolation_color_high = EXTRAPOLATION_COLOR self.invariant_color = INVARIANT_COLOR # warning to the user when the extrapolated invariant is greater than %5 if self.low_scale >= 0.05 or self.low_scale > 1 or self.low_scale < 0: self.extrapolation_color_low = ERROR_COLOR if self.high_scale >= 0.05 or self.high_scale > 1 or self.high_scale < 0: self.extrapolation_color_high = ERROR_COLOR if self.inv_scale > 1 or self.inv_scale < 0: self.invariant_color = ERROR_COLOR def on_close(self, event): """ Close the current window """ self.Close() def on_paint(self, event): """ Draw the chart """ dc = wx.PaintDC(self.panel_chart) try: gc = wx.GraphicsContext.Create(dc) except NotImplementedError: msg = "This build of wxPython does not support " msg += "the wx.GraphicsContext family of classes." dc.DrawText(msg, 25, 25) return # Start the drawing font = wx.SystemSettings.GetFont(wx.SYS_DEFAULT_GUI_FONT) font.SetWeight(wx.BOLD) gc.SetFont(font) # Draw a rectangle path = gc.CreatePath() path.AddRectangle(-RECTANGLE_WIDTH / 2, -RECTANGLE_HEIGHT / 2, RECTANGLE_WIDTH / 2, RECTANGLE_HEIGHT / 2) x_origine = 20 y_origine = 15 # Draw low rectangle gc.PushState() label = "Q* from Low-Q " PathFunc = gc.DrawPath w, h = gc.GetTextExtent(label) gc.DrawText(label, x_origine, y_origine) # Translate the rectangle x_center = x_origine + RECTANGLE_WIDTH * self.low_scale / 2 + w + 10 y_center = y_origine + h gc.Translate(x_center, y_center) gc.SetPen(wx.Pen("black", 1)) gc.SetBrush(wx.Brush(self.extrapolation_color_low)) if self.low_inv_percent is None: low_percent = 'Not Computed' elif self.low_inv_percent == 'error': low_percent = 'Error' else: low_percent = format_number(self.low_inv_percent * 100) + '%' x_center = 20 y_center = -h gc.DrawText(low_percent, x_center, y_center) # Increase width by self.low_scale gc.Scale(self.low_scale, 1.0) PathFunc(path) gc.PopState() # Draw rectangle for invariant gc.PushState() # save it again y_origine += 20 gc.DrawText("Q* from Data ", x_origine, y_origine) # offset to the lower part of the window x_center = x_origine + RECTANGLE_WIDTH * self.inv_scale / 2 + w + 10 y_center = y_origine + h gc.Translate(x_center, y_center) # 128 == half transparent gc.SetBrush(wx.Brush(self.invariant_color)) # Increase width by self.inv_scale if self.inv_percent is None: inv_percent = 'Not Computed' elif self.inv_percent == 'error': inv_percent = 'Error' else: inv_percent = format_number(self.inv_percent * 100) + '%' x_center = 20 y_center = -h gc.DrawText(inv_percent, x_center, y_center) gc.Scale(self.inv_scale, 1.0) gc.DrawPath(path) gc.PopState() # restore saved state # Draw rectangle for high invariant gc.PushState() y_origine += 20 gc.DrawText("Q* from High-Q ", x_origine, y_origine) # define the position of the new rectangle x_center = x_origine + RECTANGLE_WIDTH * self.high_scale / 2 + w + 10 y_center = y_origine + h gc.Translate(x_center, y_center) gc.SetBrush(wx.Brush(self.extrapolation_color_high)) # increase scale by self.high_scale if self.high_inv_percent is None: high_percent = 'Not Computed' elif self.high_inv_percent == 'error': high_percent = 'Error' else: high_percent = format_number(self.high_inv_percent * 100) + '%' x_center = 20 y_center = -h gc.DrawText(high_percent, x_center, y_center) gc.Scale(self.high_scale, 1.0) gc.DrawPath(path) gc.PopState()