source: sasview/src/sas/perspectives/invariant/invariant_details.py @ 78f75d02

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 78f75d02 was 78a205a, checked in by Mathieu Doucet <doucetm@…>, 10 years ago

pylint fixes

  • Property mode set to 100644
File size: 22.3 KB
RevLine 
[a3efdeb]1"""
2    Invariant panel
3"""
[9ce7641c]4import wx
5import sys
[f6518f8]6
[79492222]7from sas.guiframe.utils import format_number
[518d35d]8from invariant_widgets import OutputTextCtrl
[9ce7641c]9# Dimensions related to chart
[78a205a]10RECTANGLE_WIDTH = 400.0
[9ce7641c]11RECTANGLE_HEIGHT = 20
[78a205a]12# Invariant panel size
[9ce7641c]13_BOX_WIDTH = 76
[d0cc0bbc]14
[78a205a]15# scale to use for a bar of value zero
16RECTANGLE_SCALE = 0.0001
17DEFAULT_QSTAR = 1.0
18
[4a2b054]19if sys.platform.count("win32") > 0:
[9ce7641c]20    _STATICBOX_WIDTH = 450
21    PANEL_WIDTH = 500
[d0cc0bbc]22    PANEL_HEIGHT = 430
[9ce7641c]23    FONT_VARIANT = 0
24else:
25    _STATICBOX_WIDTH = 480
26    PANEL_WIDTH = 530
[d0cc0bbc]27    PANEL_HEIGHT = 430
[9ce7641c]28    FONT_VARIANT = 1
[78a205a]29
30ERROR_COLOR = wx.Colour(255, 0, 0, 128)
31EXTRAPOLATION_COLOR = wx.Colour(169, 169, 168, 128)
32INVARIANT_COLOR = wx.Colour(67, 208, 128, 128)
33
34
[d0cc0bbc]35class InvariantContainer(wx.Object):
[45802d4]36    """
[d7a39e5]37    This class stores some values resulting resulting from invariant
[78a205a]38    calculations. Given the value of total invariant, this class can also
[d7a39e5]39    determine the percentage of invariants resulting from extrapolation.
[45802d4]40    """
[d0cc0bbc]41    def __init__(self):
[78a205a]42        # invariant at low range
[eed601e]43        self.qstar_low = None
[78a205a]44        # invariant at low range error
[eed601e]45        self.qstar_low_err = None
[78a205a]46        # invariant
[eed601e]47        self.qstar = None
[78a205a]48        # invariant error
[eed601e]49        self.qstar_err = None
[78a205a]50        # invariant at high range
[eed601e]51        self.qstar_high = None
[78a205a]52        # invariant at high range error
[eed601e]53        self.qstar_high_err = None
[78a205a]54        # invariant total
[d0cc0bbc]55        self.qstar_total = None
[78a205a]56        # invariant error
[d0cc0bbc]57        self.qstar_total_err = None
[78a205a]58        # scale
[eed601e]59        self.qstar_low_percent = None
60        self.qstar_high_percent = None
61        self.qstar_percent = None
[d0cc0bbc]62        # warning message
63        self.existing_warning = False
64        self.warning_msg = "No Details on calculations available...\n"
[78a205a]65
[d0cc0bbc]66    def compute_percentage(self):
67        """
[d7a39e5]68        Compute percentage of each invariant
[d0cc0bbc]69        """
[eed601e]70        if self.qstar_total is None:
71            self.qstar_percent = None
72            self.qstar_low = None
73            self.qstar_high = None
74            self.check_values()
[78a205a]75            return
76
77        # compute invariant percentage
[eed601e]78        if self.qstar is None:
79            self.qstar_percent = None
80        else:
81            try:
[78a205a]82                self.qstar_percent = float(self.qstar) / float(self.qstar_total)
[eed601e]83            except:
84                self.qstar_percent = 'error'
[78a205a]85        # compute low q invariant percentage
[eed601e]86        if self.qstar_low is None:
87            self.qstar_low_percent = None
88        else:
89            try:
[4a2b054]90                self.qstar_low_percent = float(self.qstar_low)\
[78a205a]91                                            / float(self.qstar_total)
[eed601e]92            except:
93                self.qstar_low_percent = 'error'
[78a205a]94        # compute high q invariant percentage
[eed601e]95        if self.qstar_high is None:
96            self.qstar_high_percent = None
97        else:
98            try:
[4a2b054]99                self.qstar_high_percent = float(self.qstar_high)\
[78a205a]100                                                / float(self.qstar_total)
[eed601e]101            except:
102                self.qstar_high_percent = 'error'
[ee54fcf0]103        wx.CallAfter(self.check_values)
[78a205a]104
[d0cc0bbc]105    def check_values(self):
106        """
[d7a39e5]107        check the validity if invariant
[d0cc0bbc]108        """
[eed601e]109        if self.qstar_total is None and self.qstar is None:
110            self.warning_msg = "Invariant not calculated.\n"
[78a205a]111            return
[eed601e]112        if self.qstar_total == 0:
[4a2b054]113            self.existing_warning = True
114            self.warning_msg = "Invariant is zero. \n"
115            self.warning_msg += "The calculations are likely "
116            self.warning_msg += "to be unreliable!\n"
[78a205a]117            return
118        # warning to the user when the extrapolated invariant is greater than %5
[3fab6ef]119        msg = ''
[eed601e]120        if self.qstar_percent == 'error':
[ee54fcf0]121            try:
122                float(self.qstar)
123            except:
124                self.existing_warning = True
125                msg += 'Error occurred when computing invariant from data.\n '
[f6518f8]126        if self.qstar_percent > 1:
127            self.existing_warning = True
128            msg += "Invariant Q  contribution is greater "
129            msg += "than 100% .\n"
[eed601e]130        if self.qstar_low_percent == 'error':
[ee54fcf0]131            try:
132                float(self.qstar_low)
133            except:
134                self.existing_warning = True
135                msg += "Error occurred when computing extrapolated invariant"
136                msg += " at low-Q region.\n"
[78a205a]137        elif self.qstar_low_percent is not None:
[f6518f8]138            if self.qstar_low_percent >= 0.05:
139                self.existing_warning = True
140                msg += "Extrapolated contribution at Low Q is higher "
141                msg += "than 5% of the invariant.\n"
142            elif self.qstar_low_percent < 0:
143                self.existing_warning = True
144                msg += "Extrapolated contribution at Low Q < 0.\n"
145            elif self.qstar_low_percent > 1:
146                self.existing_warning = True
147                msg += "Extrapolated contribution at Low Q is greater "
148                msg += "than 100% .\n"
[eed601e]149        if self.qstar_high_percent == 'error':
[ee54fcf0]150            try:
151                float(self.qstar_high)
152            except:
153                self.existing_warning = True
154                msg += 'Error occurred when computing extrapolated'
155                msg += ' invariant at high-Q region.\n'
[f6518f8]156        elif self.qstar_high_percent is not None:
157            if self.qstar_high_percent >= 0.05:
158                self.existing_warning = True
159                msg += "Extrapolated contribution at High Q is higher "
160                msg += "than 5% of the invariant.\n"
161            elif self.qstar_high_percent < 0:
162                self.existing_warning = True
163                msg += "Extrapolated contribution at High Q < 0.\n"
164            elif self.qstar_high_percent > 1:
165                self.existing_warning = True
166                msg += "Extrapolated contribution at High Q is greater "
167                msg += "than 100% .\n"
[eed601e]168        if (self.qstar_low_percent not in [None, "error"]) and \
169            (self.qstar_high_percent not in [None, "error"])\
170            and self.qstar_low_percent + self.qstar_high_percent >= 0.05:
[d0cc0bbc]171            self.existing_warning = True
[3fab6ef]172            msg += "The sum of all extrapolated contributions is higher "
173            msg += "than 5% of the invariant.\n"
[78a205a]174
[d0cc0bbc]175        if self.existing_warning:
[3fab6ef]176            self.warning_msg = ''
[78a205a]177            self.warning_msg += msg
[4a2b054]178            self.warning_msg += "The calculations are likely to be"
179            self.warning_msg += " unreliable!\n"
[d0cc0bbc]180        else:
181            self.warning_msg = "No Details on calculations available...\n"
[78a205a]182
[d3fac18]183class InvariantDetailsPanel(wx.Dialog):
[9ce7641c]184    """
[78a205a]185    This panel describes proportion of invariants
[9ce7641c]186    """
[78a205a]187    def __init__(self, parent=None, id=-1, qstar_container=None,
188                 title="Invariant Details",
189                 size=(PANEL_WIDTH, PANEL_HEIGHT)):
[4a2b054]190        wx.Dialog.__init__(self, parent=parent, id=id, title=title, size=size)
[78a205a]191
192        # Font size
[f75ea4a]193        self.SetWindowVariant(variant=FONT_VARIANT)
194        self.parent = parent
[78a205a]195        # self.qstar_container
[f75ea4a]196        self.qstar_container = qstar_container
[78a205a]197        # warning message
[f75ea4a]198        self.warning_msg = self.qstar_container.warning_msg
[78a205a]199
200        # Define scale of each bar
[f75ea4a]201        self.low_inv_percent = self.qstar_container.qstar_low_percent
202        self.low_scale = self.get_scale(percentage=self.low_inv_percent,
[78a205a]203                                        scale_name="Extrapolated at Low Q")
[f75ea4a]204        self.inv_percent = self.qstar_container.qstar_percent
[78a205a]205        self.inv_scale = self.get_scale(percentage=self.inv_percent,
206                                        scale_name="Inv in Q range")
[f75ea4a]207        self.high_inv_percent = self.qstar_container.qstar_high_percent
208        self.high_scale = self.get_scale(percentage=self.high_inv_percent,
209                                         scale_name="Extrapolated at High Q")
[78a205a]210
211        # Default color the extrapolation bar is grey
[9cec2dd]212        self.extrapolation_color_low = EXTRAPOLATION_COLOR
213        self.extrapolation_color_high = EXTRAPOLATION_COLOR
214        self.invariant_color = INVARIANT_COLOR
[78a205a]215        # change color of high and low bar when necessary
[f75ea4a]216        self.set_color_bar()
[78a205a]217        # draw the panel itself
[f75ea4a]218        self._do_layout()
219        self.set_values()
[78a205a]220
[9ce7641c]221    def _define_structure(self):
222        """
[d7a39e5]223        Define main sizers needed for this panel
[9ce7641c]224        """
[78a205a]225        # Box sizers must be defined first before defining buttons/textctrls
[4a2b054]226        # (MAC).
[9ce7641c]227        self.main_sizer = wx.BoxSizer(wx.VERTICAL)
[78a205a]228        # Sizer related to chart
[9ce7641c]229        chart_box = wx.StaticBox(self, -1, "Invariant Chart")
230        self.chart_sizer = wx.StaticBoxSizer(chart_box, wx.VERTICAL)
[4a2b054]231        self.chart_sizer.SetMinSize((PANEL_WIDTH - 50, 110))
[78a205a]232        # Sizer related to invariant values
233        self.invariant_sizer = wx.GridBagSizer(4, 4)
[d0cc0bbc]234        invariant_box = wx.StaticBox(self, -1, "Numerical Values")
[78a205a]235        self.invariant_box_sizer = wx.StaticBoxSizer(invariant_box, wx.HORIZONTAL)
[d0cc0bbc]236
[4a2b054]237        self.invariant_box_sizer.SetMinSize((PANEL_WIDTH - 50, -1))
[78a205a]238        # Sizer related to warning message
[9ce7641c]239        warning_box = wx.StaticBox(self, -1, "Warning")
240        self.warning_sizer = wx.StaticBoxSizer(warning_box, wx.VERTICAL)
[78a205a]241        self.warning_sizer.SetMinSize((PANEL_WIDTH - 50, -1))
242        # Sizer related to button
[9ce7641c]243        self.button_sizer = wx.BoxSizer(wx.HORIZONTAL)
[78a205a]244
[9ce7641c]245    def _layout_shart(self):
246        """
[d7a39e5]247        Draw widgets related to chart
[9ce7641c]248        """
249        self.panel_chart = wx.Panel(self)
250        self.panel_chart.Bind(wx.EVT_PAINT, self.on_paint)
[78a205a]251        self.chart_sizer.Add(self.panel_chart, 1, wx.EXPAND | wx.ALL, 0)
252
[9ce7641c]253    def _layout_invariant(self):
254        """
[d7a39e5]255        Draw widgets related to invariant
[9ce7641c]256        """
[78a205a]257        uncertainty = "+/-"
[3df5ecf]258        unit_invariant = '[1/(cm * A^3)]'
[78a205a]259
[eed601e]260        invariant_txt = wx.StaticText(self, -1, 'Q* from Data ')
261        invariant_txt.SetToolTipString("Invariant in the data set's Q range.")
[4a2b054]262        self.invariant_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1))
263        hint_msg = "Invariant in the data set's Q range."
264        self.invariant_tcl.SetToolTipString(hint_msg)
265        self.invariant_err_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1))
266        hint_msg = "Uncertainty on the invariant from data's range."
267        self.invariant_err_tcl.SetToolTipString(hint_msg)
[9ce7641c]268        invariant_units_txt = wx.StaticText(self, -1, unit_invariant)
[4a2b054]269        hint_msg = "Unit of the invariant from data's Q range"
270        invariant_units_txt.SetToolTipString(hint_msg)
[78a205a]271
[eed601e]272        invariant_low_txt = wx.StaticText(self, -1, 'Q* from Low-Q')
[4a2b054]273        hint_msg = "Extrapolated invariant from low-Q range."
274        invariant_low_txt.SetToolTipString(hint_msg)
275        self.invariant_low_tcl = OutputTextCtrl(self, -1, size=(_BOX_WIDTH, -1))
276        hint_msg = "Extrapolated invariant from low-Q range."
277        self.invariant_low_tcl.SetToolTipString(hint_msg)
278        self.invariant_low_err_tcl = OutputTextCtrl(self, -1,
279                                                    size=(_BOX_WIDTH, -1))
280        hint_msg = "Uncertainty on the invariant from low-Q range."
281        self.invariant_low_err_tcl.SetToolTipString(hint_msg)
282        invariant_low_units_txt = wx.StaticText(self, -1, unit_invariant)
283        hint_msg = "Unit of the extrapolated invariant from  low-Q range"
284        invariant_low_units_txt.SetToolTipString(hint_msg)
[78a205a]285
[eed601e]286        invariant_high_txt = wx.StaticText(self, -1, 'Q* from High-Q')
[4a2b054]287        hint_msg = "Extrapolated invariant from  high-Q range"
288        invariant_high_txt.SetToolTipString(hint_msg)
289        self.invariant_high_tcl = OutputTextCtrl(self, -1,
290                                                 size=(_BOX_WIDTH, -1))
291        hint_msg = "Extrapolated invariant from  high-Q range"
292        self.invariant_high_tcl.SetToolTipString(hint_msg)
293        self.invariant_high_err_tcl = OutputTextCtrl(self, -1,
294                                                     size=(_BOX_WIDTH, -1))
295        hint_msg = "Uncertainty on the invariant from high-Q range."
[98eefdb]296        self.invariant_high_err_tcl.SetToolTipString(hint_msg)
[4a2b054]297        invariant_high_units_txt = wx.StaticText(self, -1, unit_invariant)
298        hint_msg = "Unit of the extrapolated invariant from  high-Q range"
299        invariant_high_units_txt.SetToolTipString(hint_msg)
[78a205a]300
301        # Invariant low
[d0cc0bbc]302        iy = 0
[78a205a]303        ix = 0
[4a2b054]304        self.invariant_sizer.Add(invariant_low_txt, (iy, ix), (1, 1),
[78a205a]305                                 wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
[9ce7641c]306        ix += 1
[4a2b054]307        self.invariant_sizer.Add(self.invariant_low_tcl, (iy, ix), (1, 1),
[78a205a]308                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)
[9ce7641c]309        ix += 1
[78a205a]310        self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty),
311                                 (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0)
[9ce7641c]312        ix += 1
[4a2b054]313        self.invariant_sizer.Add(self.invariant_low_err_tcl, (iy, ix), (1, 1),
[78a205a]314                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)
[9ce7641c]315        ix += 1
[78a205a]316        self.invariant_sizer.Add(invariant_low_units_txt,
317                                 (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0)
318        # Invariant
[9ce7641c]319        iy += 1
[78a205a]320        ix = 0
[4a2b054]321        self.invariant_sizer.Add(invariant_txt, (iy, ix), (1, 1),
[78a205a]322                                 wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
[9ce7641c]323        ix += 1
[4a2b054]324        self.invariant_sizer.Add(self.invariant_tcl, (iy, ix), (1, 1),
[78a205a]325                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)
[9ce7641c]326        ix += 1
327        self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty),
[78a205a]328                                 (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0)
[9ce7641c]329        ix += 1
[4a2b054]330        self.invariant_sizer.Add(self.invariant_err_tcl, (iy, ix), (1, 1),
[78a205a]331                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)
332        ix += 1
333        self.invariant_sizer.Add(invariant_units_txt,
334                                 (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0)
335        # Invariant high
[9ce7641c]336        iy += 1
[78a205a]337        ix = 0
[4a2b054]338        self.invariant_sizer.Add(invariant_high_txt, (iy, ix), (1, 1),
[78a205a]339                                 wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 15)
[9ce7641c]340        ix += 1
[4a2b054]341        self.invariant_sizer.Add(self.invariant_high_tcl, (iy, ix), (1, 1),
[78a205a]342                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)
[9ce7641c]343        ix += 1
344        self.invariant_sizer.Add(wx.StaticText(self, -1, uncertainty),
[78a205a]345                                 (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0)
[9ce7641c]346        ix += 1
[4a2b054]347        self.invariant_sizer.Add(self.invariant_high_err_tcl, (iy, ix), (1, 1),
[78a205a]348                                 wx.EXPAND | wx.ADJUST_MINSIZE, 0)
[9ce7641c]349        ix += 1
[78a205a]350        self.invariant_sizer.Add(invariant_high_units_txt,
351                                 (iy, ix), (1, 1), wx.EXPAND | wx.ADJUST_MINSIZE, 0)
352        self.invariant_box_sizer.Add(self.invariant_sizer, 0, wx.TOP | wx.BOTTOM, 10)
353
[9ce7641c]354    def _layout_warning(self):
355        """
[d7a39e5]356        Draw widgets related to warning
[9ce7641c]357        """
[78a205a]358        # Warning [string]
359        self.warning_msg_txt = wx.StaticText(self, -1, self.warning_msg)
[d0cc0bbc]360        if self.qstar_container.existing_warning:
[78a205a]361            self.warning_msg_txt.SetForegroundColour('red')
[277fad8]362        self.warning_sizer.AddMany([(self.warning_msg_txt, 0,
[78a205a]363                                     wx.LEFT | wx.EXPAND | wx.ADJUST_MINSIZE, 10)])
364
[9ce7641c]365    def _layout_button(self):
366        """
[d7a39e5]367        Draw widgets related to button
[9ce7641c]368        """
[78a205a]369        # Close button
[9ce7641c]370        id = wx.NewId()
371        button_ok = wx.Button(self, id, "Ok")
372        button_ok.SetToolTipString("Give Details on Computation")
373        self.Bind(wx.EVT_BUTTON, self.on_close, id=id)
[78a205a]374        self.button_sizer.AddMany([((20, 20), 0, wx.LEFT, 350),
375                                   (button_ok, 0, wx.RIGHT, 10)])
[9ce7641c]376    def _do_layout(self):
377        """
[d7a39e5]378        Draw window content
[9ce7641c]379        """
380        self._define_structure()
381        self._layout_shart()
382        self._layout_invariant()
383        self._layout_warning()
384        self._layout_button()
[d0cc0bbc]385        self.main_sizer.AddMany([(self.chart_sizer, 0, wx.ALL, 10),
[78a205a]386                                 (self.invariant_box_sizer, 0, wx.ALL, 10),
387                                 (self.warning_sizer, 0, wx.ALL, 10),
388                                 (self.button_sizer, 0, wx.ALL, 10)])
[9ce7641c]389        self.SetSizer(self.main_sizer)
[78a205a]390
391
[9ce7641c]392    def set_values(self):
393        """
[d7a39e5]394        Set value of txtcrtl
[9ce7641c]395        """
[4a2b054]396        value = format_number(self.qstar_container.qstar)
397        self.invariant_tcl.SetValue(value)
398        value = format_number(self.qstar_container.qstar_err)
[78a205a]399        self.invariant_err_tcl.SetValue(value)
[4a2b054]400        value = format_number(self.qstar_container.qstar_low)
401        self.invariant_low_tcl.SetValue(value)
402        value = format_number(self.qstar_container.qstar_low_err)
[78a205a]403        self.invariant_low_err_tcl.SetValue(value)
[4a2b054]404        value = format_number(self.qstar_container.qstar_high)
405        self.invariant_high_tcl.SetValue(value)
406        value = format_number(self.qstar_container.qstar_high_err)
[78a205a]407        self.invariant_high_err_tcl.SetValue(value)
[d0cc0bbc]408
409    def get_scale(self, percentage, scale_name='scale'):
[9ce7641c]410        """
[78a205a]411        Check scale receive in this panel.
[9ce7641c]412        """
[eed601e]413        scale = RECTANGLE_SCALE
[78a205a]414        try:
[eed601e]415            if percentage in [None, 0.0, "error"]:
[78a205a]416                scale = RECTANGLE_SCALE
417                return scale
[f6518f8]418            elif percentage < 0:
419                scale = RECTANGLE_SCALE
420                return scale
421            scale = float(percentage)
[9ce7641c]422        except:
[eed601e]423            scale = RECTANGLE_SCALE
[a3efdeb]424            self.warning_msg += "Recieve an invalid scale for %s\n"
425            self.warning_msg += "check this value : %s\n" % str(percentage)
[eed601e]426        return  scale
[78a205a]427
[9ce7641c]428    def set_color_bar(self):
429        """
[d7a39e5]430        Change the color for low and high bar when necessary
[9ce7641c]431        """
[9cec2dd]432        self.extrapolation_color_low = EXTRAPOLATION_COLOR
433        self.extrapolation_color_high = EXTRAPOLATION_COLOR
434        self.invariant_color = INVARIANT_COLOR
[78a205a]435        # warning to the user when the extrapolated invariant is greater than %5
[f6518f8]436        if self.low_scale >= 0.05 or self.low_scale > 1 or self.low_scale < 0:
[78a205a]437            self.extrapolation_color_low = ERROR_COLOR
[f6518f8]438        if self.high_scale >= 0.05 or self.high_scale > 1 or self.high_scale < 0:
[78a205a]439            self.extrapolation_color_high = ERROR_COLOR
[9cec2dd]440        if self.inv_scale > 1 or self.inv_scale < 0:
[78a205a]441            self.invariant_color = ERROR_COLOR
442
[9ce7641c]443    def on_close(self, event):
444        """
[d7a39e5]445        Close the current window
[9ce7641c]446        """
447        self.Close()
[78a205a]448
[9ce7641c]449    def on_paint(self, event):
450        """
[d7a39e5]451        Draw the chart
[9ce7641c]452        """
453        dc = wx.PaintDC(self.panel_chart)
454        try:
455            gc = wx.GraphicsContext.Create(dc)
456        except NotImplementedError:
[4a2b054]457            msg = "This build of wxPython does not support "
458            msg += "the wx.GraphicsContext family of classes."
459            dc.DrawText(msg, 25, 25)
[9ce7641c]460            return
[78a205a]461        # Start the drawing
[9ce7641c]462        font = wx.SystemSettings.GetFont(wx.SYS_DEFAULT_GUI_FONT)
463        font.SetWeight(wx.BOLD)
464        gc.SetFont(font)
465        # Draw a rectangle
466        path = gc.CreatePath()
[78a205a]467        path.AddRectangle(-RECTANGLE_WIDTH / 2, -RECTANGLE_HEIGHT / 2,
468                          RECTANGLE_WIDTH / 2, RECTANGLE_HEIGHT / 2)
[d0cc0bbc]469        x_origine = 20
[9ce7641c]470        y_origine = 15
[78a205a]471        # Draw low rectangle
472        gc.PushState()
[f6518f8]473        label = "Q* from Low-Q "
[9ce7641c]474        PathFunc = gc.DrawPath
475        w, h = gc.GetTextExtent(label)
476        gc.DrawText(label, x_origine, y_origine)
[78a205a]477        # Translate the rectangle
478        x_center = x_origine + RECTANGLE_WIDTH * self.low_scale / 2 + w + 10
[9ce7641c]479        y_center = y_origine + h
[78a205a]480        gc.Translate(x_center, y_center)
[9ce7641c]481        gc.SetPen(wx.Pen("black", 1))
482        gc.SetBrush(wx.Brush(self.extrapolation_color_low))
[eed601e]483        if self.low_inv_percent is None:
484            low_percent = 'Not Computed'
485        elif self.low_inv_percent == 'error':
486            low_percent = 'Error'
487        else:
[78a205a]488            low_percent = format_number(self.low_inv_percent * 100) + '%'
[d0cc0bbc]489        x_center = 20
490        y_center = -h
491        gc.DrawText(low_percent, x_center, y_center)
[9ce7641c]492        # Increase width by self.low_scale
[78a205a]493        gc.Scale(self.low_scale, 1.0)
[9ce7641c]494        PathFunc(path)
[78a205a]495        gc.PopState()
496        # Draw rectangle for invariant
497        gc.PushState()  # save it again
498        y_origine += 20
[f6518f8]499        gc.DrawText("Q* from Data ", x_origine, y_origine)
[9ce7641c]500        # offset to the lower part of the window
[78a205a]501        x_center = x_origine + RECTANGLE_WIDTH * self.inv_scale / 2 + w + 10
[9ce7641c]502        y_center = y_origine + h
503        gc.Translate(x_center, y_center)
504        # 128 == half transparent
[78a205a]505        gc.SetBrush(wx.Brush(self.invariant_color))
[9ce7641c]506        # Increase width by self.inv_scale
[eed601e]507        if self.inv_percent is None:
508            inv_percent = 'Not Computed'
509        elif self.inv_percent == 'error':
510            inv_percent = 'Error'
511        else:
[78a205a]512            inv_percent = format_number(self.inv_percent * 100) + '%'
513        x_center = 20
[d0cc0bbc]514        y_center = -h
515        gc.DrawText(inv_percent, x_center, y_center)
[78a205a]516        gc.Scale(self.inv_scale, 1.0)
[9ce7641c]517        gc.DrawPath(path)
518        gc.PopState()
519        # restore saved state
[78a205a]520        # Draw rectangle for high invariant
521        gc.PushState()
522        y_origine += 20
523        gc.DrawText("Q* from High-Q ", x_origine, y_origine)
524        # define the position of the new rectangle
525        x_center = x_origine + RECTANGLE_WIDTH * self.high_scale / 2 + w + 10
[9ce7641c]526        y_center = y_origine + h
527        gc.Translate(x_center, y_center)
[78a205a]528        gc.SetBrush(wx.Brush(self.extrapolation_color_high))
[9ce7641c]529        # increase scale by self.high_scale
[eed601e]530        if self.high_inv_percent is None:
531            high_percent = 'Not Computed'
532        elif self.high_inv_percent == 'error':
533            high_percent = 'Error'
534        else:
[78a205a]535            high_percent = format_number(self.high_inv_percent * 100) + '%'
[d0cc0bbc]536        x_center = 20
537        y_center = -h
538        gc.DrawText(high_percent, x_center, y_center)
[78a205a]539
540        gc.Scale(self.high_scale, 1.0)
[9ce7641c]541        gc.DrawPath(path)
542        gc.PopState()
Note: See TracBrowser for help on using the repository browser.