Ignore:
Timestamp:
Aug 16, 2012 11:46:12 AM (12 years ago)
Author:
Jae Cho <jhjcho@…>
Branches:
master, ESS_GUI, ESS_GUI_Docs, ESS_GUI_batch_fitting, ESS_GUI_bumps_abstraction, ESS_GUI_iss1116, ESS_GUI_iss879, ESS_GUI_iss959, ESS_GUI_opencl, ESS_GUI_ordering, ESS_GUI_sync_sascalc, costrafo411, magnetic_scatt, release-4.1.1, release-4.1.2, release-4.2.2, release_4.0.1, ticket-1009, ticket-1094-headless, ticket-1242-2d-resolution, ticket-1243, ticket-1249, ticket885, unittest-saveload
Children:
d8e3f7c
Parents:
9f51c2c
Message:

pylint cleanups

File:
1 edited

Legend:

Unmodified
Added
Removed

  • sansguiframe/src/sans/guiframe/local_perspectives/plotting/appearanceDialog.py

    r9f51c2c rf866fb5  
    11#!/usr/bin/python 
    2  
    32""" 
    4  
    53Dialog for appearance of plot symbols, color, size etc. 
    6  
    7  
    84/** 
    95    This software was developed by Institut Laue-Langevin as part of 
    … …  
    139 
    1410**/ 
    15  
    16  
    1711""" 
    18  
    1912import wx 
    2013import operator 
    2114 
    22  
    23 # main appearance dialog starts here: 
    24  
    25  
    2615class appearanceDialog(wx.Frame): 
    27  
    28     def __init__(self,parent,title): 
    29         super(appearanceDialog,self).__init__(parent, title=title,size=(570,450), style=wx.DEFAULT_FRAME_STYLE|wx.FRAME_FLOAT_ON_PARENT) 
     16    """ 
     17    Appearance dialog 
     18    """ 
     19    def __init__(self, parent, title): 
     20        """ 
     21        Initialization of the Panel 
     22        """ 
     23        super(appearanceDialog, self).__init__(parent, title=title,  
     24                        size=(570,450),  
     25                        style=wx.DEFAULT_FRAME_STYLE|wx.FRAME_FLOAT_ON_PARENT) 
    3026 
    3127        self.okay_clicked = False 
    3228        self.parent = parent 
    33         self.symbolLabels = self.parent.get_symbol_label() 
    34         self.colorLabels = self.parent.get_color_label() 
    35  
    36  
    37         self.InitUI() 
     29        self.symbo_labels = self.parent.get_symbol_label() 
     30        self.color_labels = self.parent.get_color_label() 
     31        self.init_ui() 
    3832        self.Centre() 
    3933        self.Show() 
    4034 
    41    
    42     def InitUI(self): 
    43  
    44         # create spacing needed 
     35    def init_ui(self): 
     36        """ 
     37        Create spacing needed 
     38        """ 
    4539        panel = wx.Panel(self) 
    4640 
    … …  
    5751        ihbox2 = wx.BoxSizer(wx.HORIZONTAL) 
    5852 
    59         symbolStaticBox = wx.StaticBox(panel, -1, 'Symbol') 
    60         symbolStaticBoxSizer = wx.StaticBoxSizer(symbolStaticBox, wx.VERTICAL) 
     53        symbolstaticbox = wx.StaticBox(panel, -1, 'Symbol') 
     54        symbolstaticboxsizer = wx.StaticBoxSizer(symbolstaticbox, wx.VERTICAL) 
    6155 
    6256        # add widgets - reverse order! 
    63  
    6457        # texts 
    65         symbolText = wx.StaticText(panel, label='Shape') 
    66         colorText = wx.StaticText(panel, label='Color') 
    67         sizeText = wx.StaticText(panel, label='Size ') 
    68         labelText = wx.StaticText(panel, label='Legend label') 
     58        symboltext = wx.StaticText(panel, label='Shape') 
     59        colortext = wx.StaticText(panel, label='Color') 
     60        sizetext = wx.StaticText(panel, label='Size ') 
     61        labeltext = wx.StaticText(panel, label='Legend Label') 
    6962 
    7063        # selection widgets 
    71         self.symbolListBox = wx.ListBox(panel,-1,size=(200,200)) 
    72         self.colorListBox = wx.ComboBox(panel,style=wx.CB_READONLY, size=(195,-1)) 
    73         self.sizeComboBox = wx.ComboBox(panel,style=wx.CB_READONLY, size=(90,-1)) 
    74         self.sizeComboBox.Bind(wx.EVT_COMBOBOX, self.combo_click) 
    75         self.sizeCustomButton = wx.Button(panel, label='Custom...') 
    76         self.sizeCustomButton.Bind(wx.EVT_BUTTON, self.customSize) 
    77         self.labelTextBox = wx.TextCtrl(panel,-1, "",size=(-1,-1)) 
     64        self.symbollistbox = wx.ListBox(panel, -1, size=(200, 200)) 
     65        self.colorlistbox = wx.ComboBox(panel, style=wx.CB_READONLY,  
     66                                        size=(185, -1)) 
     67        self.sizecombobox = wx.ComboBox(panel, style=wx.CB_READONLY,  
     68                                        size=(90, -1)) 
     69        self.sizecombobox.Bind(wx.EVT_COMBOBOX, self.combo_click) 
     70        self.sizecustombutton = wx.Button(panel, label='Custom...') 
     71        self.sizecustombutton.Bind(wx.EVT_BUTTON, self.custom_size) 
     72        self.labeltextbox = wx.TextCtrl(panel, -1, "",  size=(440, -1)) 
    7873 
    7974        # buttons 
    80         OkButton = wx.Button(panel, label='OK') 
    81         OkButton.Bind(wx.EVT_BUTTON,self.onOK) 
    82         cancelButton = wx.Button(panel, label='Cancel') 
    83         cancelButton.Bind(wx.EVT_BUTTON, self.CloseDlg) 
     75        okbutton = wx.Button(panel, label='OK') 
     76        okbutton.Bind(wx.EVT_BUTTON, self.on_ok) 
     77        cancelbutton = wx.Button(panel, label='Cancel') 
     78        cancelbutton.Bind(wx.EVT_BUTTON, self.close_dlg) 
    8479 
    8580        # now Add all the widgets to relevant spacer - tricky 
    86         ivbox1.Add(symbolText, flag =  wx.ALL  | wx.ALIGN_LEFT ,border=10) 
    87         ivbox1.Add(self.symbolListBox, flag = wx.ALL | wx.ALIGN_LEFT ,border=10) 
    88  
    89         ihbox1.Add(sizeText, flag = wx.ALL| wx.ALIGN_LEFT , border=10) 
    90         ihbox1.Add(self.sizeComboBox, flag =  wx.ALL|wx.RIGHT | wx.ALIGN_LEFT , border=10) 
    91         ihbox1.Add(self.sizeCustomButton, flag = wx.ALIGN_LEFT | wx.ALL, border=10) 
    92  
    93         ihbox2.Add(colorText,flag = wx.ALL | wx.ALIGN_LEFT, border=10) 
    94         ihbox2.Add(self.colorListBox, flag = wx.ALL  | wx.ALIGN_LEFT, border=10) 
    95  
    96  
    97  
    98         ivbox2.Add(ihbox1, flag =wx.ALIGN_LEFT,border=10) 
    99         ivbox2.Add(ihbox2, flag =wx.ALIGN_LEFT,border=10) 
    100  
    101          
    102         hbox1.Add(ivbox1,flag =wx.ALIGN_LEFT ,border=10) 
    103         hbox1.Add(ivbox2,flag =wx.ALIGN_LEFT ,border=10) 
    104  
    105     
    106         hbox2.Add(OkButton, flag = wx.ALL |  wx.ALIGN_RIGHT, border=10) 
    107         hbox2.Add(cancelButton, flag = wx.ALL | wx.ALIGN_RIGHT, border=10) 
    108  
    109         hbox3.Add(labelText, flag= wx.EXPAND | wx.RIGHT |  wx.ALIGN_LEFT, border=10) 
    110         hbox3.Add(self.labelTextBox, wx.EXPAND | wx.RIGHT |wx.ALIGN_LEFT , border=10) 
     81        ivbox1.Add(symboltext, flag=wx.ALL|wx.ALIGN_LEFT, border=10) 
     82        ivbox1.Add(self.symbollistbox, flag=wx.ALL|wx.ALIGN_LEFT, border=10) 
     83 
     84        ihbox1.Add(sizetext, flag=wx.ALL|wx.ALIGN_LEFT, border=10) 
     85        ihbox1.Add(self.sizecombobox,  
     86                   flag= wx.ALL|wx.RIGHT|wx.ALIGN_LEFT, border=10) 
     87        ihbox1.Add(self.sizecustombutton,  
     88                   flag=wx.ALIGN_LEFT|wx.ALL, border=10) 
     89 
     90        ihbox2.Add(colortext, flag=wx.ALL|wx.ALIGN_LEFT, border=10) 
     91        ihbox2.Add(self.colorlistbox, flag=wx.ALL|wx.ALIGN_LEFT, border=10) 
     92 
     93        ivbox2.Add(ihbox1, flag=wx.ALIGN_LEFT, border=10) 
     94        ivbox2.Add(ihbox2, flag=wx.ALIGN_LEFT, border=10) 
     95 
     96        hbox1.Add(ivbox1, flag=wx.ALIGN_LEFT, border=10) 
     97        hbox1.Add(ivbox2, flag=wx.ALIGN_LEFT, border=10) 
     98 
     99        hbox2.Add(okbutton, flag=wx.ALL| wx.ALIGN_RIGHT, border=10) 
     100        hbox2.Add(cancelbutton, flag=wx.ALL|wx.ALIGN_RIGHT, border=10) 
     101 
     102        hbox3.Add(labeltext, flag=wx.EXPAND|wx.ALL|wx.ALIGN_LEFT, border=10) 
     103        hbox3.Add(self.labeltextbox, flag=wx.EXPAND|wx.ALL|wx.ALIGN_LEFT, border=10) 
    111104   
    112         symbolStaticBoxSizer.Add(hbox1,flag = wx.ALL | wx.EXPAND,border=10) 
    113         vbox.Add(symbolStaticBoxSizer, flag = wx.ALL | wx.EXPAND,border=10) 
    114  
    115         vbox.Add(hbox3,flag = wx.ALL | wx.EXPAND | wx.ALIGN_RIGHT, border=10) 
    116  
    117  
    118         vbox.Add(hbox2,flag = wx.ALL  | wx.ALIGN_RIGHT, border=10) 
    119  
     105        symbolstaticboxsizer.Add(hbox1, flag=wx.ALL|wx.EXPAND, border=10) 
     106        vbox.Add(symbolstaticboxsizer, flag=wx.ALL|wx.EXPAND, border=10) 
     107        vbox.Add(hbox3, flag=wx.EXPAND|wx.RIGHT, border=10) 
     108        vbox.Add(wx.StaticLine(panel), 0, wx.ALL|wx.EXPAND, 5) 
     109        vbox.Add(hbox2, flag=wx.RIGHT|wx.ALIGN_RIGHT, border=10) 
    120110 
    121111        panel.SetSizer(vbox) 
    122112 
    123         self.populateSymbol() 
    124         self.populateColor() 
    125         self.populateSize() 
    126  
    127         self.SetDefaultItem(self.symbolListBox) 
    128  
    129     def customSize(self,e): 
    130         dlg = wx.TextEntryDialog(self, 
    131                                  'Enter custom size', 
    132                                  'Custom size', 
     113        self.populate_symbol() 
     114        self.populate_color() 
     115        self.populate_size() 
     116 
     117        self.SetDefaultItem(self.symbollistbox) 
     118 
     119    def custom_size(self, event): 
     120        """ 
     121        On custom size 
     122        """ 
     123        dlg = wx.TextEntryDialog(self,  
     124                                 'Enter custom size',  
     125                                 'Custom size',  
    133126                                 str(self.final_size)) 
    134127        if(dlg.ShowModal() == wx.ID_OK): 
    135128            if(float(dlg.GetValue()) < 0): 
    136                 dial = wx.MessageDialog(None,  
    137                                         'Unfortunately imaginary icons are not yet supported. Please enter a positive value', 
    138                                         'Error', 
    139                                         wx.OK | wx.ICON_ERROR) 
     129                msg = "Unfortunately imaginary icons are not yet supported." 
     130                msg += "Please enter a positive value" 
     131                dial = wx.MessageDialog(None, msg, 'Error', 
     132                                         wx.OK|wx.ICON_ERROR) 
    140133                dial.ShowModal() 
    141134                dlg.Destroy() 
    142                 self.customSize(e) 
     135                self.custom_size(event) 
    143136            else: 
    144137                self.final_size = dlg.GetValue() 
    … …  
    147140            dlg.Destroy() 
    148141 
    149     def setDefaults(self,size,color,symbol,label): 
     142    def set_defaults(self, size, color, symbol, label): 
     143        """ 
     144        Set Defaults 
     145        """ 
    150146        self.final_size = size 
    151147        # set up gui values 
    152         self.labelTextBox.SetValue(label) 
     148        self.labeltextbox.SetValue(label) 
    153149        if(size % 1 == 0 and size > 1 and size < 11): 
    154             self.sizeComboBox.SetSelection(int(size) - 1) 
     150            self.sizecombobox.SetSelection(int(size) - 1) 
    155151        else: 
    156             self.sizeComboBox.SetSelection(4) 
    157         self.symbolListBox.SetSelection(self.sorted_sym_dic[symbol]) 
    158         colorname = appearanceDialog.find_key(self.parent.get_color_label(),color) 
    159         self.colorListBox.SetStringSelection(colorname) 
    160  
    161     def populateSymbol(self): 
    162         self.sorted_symbolLabels = sorted(self.symbolLabels.iteritems(),key=operator.itemgetter(1)) 
     152            self.sizecombobox.SetSelection(4) 
     153        self.symbollistbox.SetSelection(self.sorted_sym_dic[symbol]) 
     154        colorname = appearanceDialog.find_key(self.parent.get_color_label(),  
     155                                              color) 
     156        self.colorlistbox.SetStringSelection(colorname) 
     157 
     158    def populate_symbol(self): 
     159        """ 
     160        Populate Symbols 
     161        """ 
     162        self.sorted_symbo_labels = sorted(self.symbo_labels.iteritems(),  
     163                                          key=operator.itemgetter(1)) 
    163164        self.sorted_sym_dic = {} 
    164165        i = 0 
    165         for label in self.sorted_symbolLabels: 
    166             self.symbolListBox.Append(str(label[0])) 
     166        for label in self.sorted_symbo_labels: 
     167            self.symbollistbox.Append(str(label[0])) 
    167168            self.sorted_sym_dic[str(label[0])] = i 
    168169            i += 1 
    169170 
    170     def populateColor(self): 
    171         sortedcolorLabels = sorted(self.colorLabels.iteritems(),key=operator.itemgetter(1)) 
    172          
    173         for color in sortedcolorLabels: 
    174              self.colorListBox.Append(str(color[0])) 
     171    def populate_color(self): 
     172        """ 
     173        Populate Colors 
     174        """ 
     175        sortedcolor_labels = sorted(self.color_labels.iteritems(),  
     176                                   key=operator.itemgetter(1)) 
     177        for color in sortedcolor_labels: 
     178            self.colorlistbox.Append(str(color[0])) 
    175179  
    176     def populateSize(self): 
    177  
    178         for i in range(1,11): 
    179             self.sizeComboBox.Append(str(i) + '.0') 
    180  
    181     def combo_click(self,e): 
    182         self.final_size = self.sizeComboBox.GetValue() 
    183  
    184     def CloseDlg(self,e): 
     180    def populate_size(self): 
     181        """ 
     182        Populate Size 
     183        """ 
     184        for i in range(1, 11): 
     185            self.sizecombobox.Append(str(i) + '.0') 
     186 
     187    def combo_click(self, event): 
     188        """ 
     189        Combox on click 
     190        """ 
     191        event.Skip() 
     192        self.final_size = self.sizecombobox.GetValue() 
     193 
     194    def close_dlg(self, event): 
     195        """ 
     196        On Close Dlg 
     197        """ 
     198        event.Skip() 
    185199        self.Destroy() 
    186200 
    187  
    188     def get_symbol_label(self): 
    189         """ 
    190         Associates label to symbol 
    191         """ 
    192         _labels = {} 
    193         i = 0 
    194         _labels['Circle'] = i 
    195         i += 1 
    196         _labels['Cross X '] = i 
    197         i += 1 
    198         _labels['Triangle Down'] = i 
    199         i += 1 
    200         _labels['Triangle Up'] = i 
    201         i += 1 
    202         _labels['Triangle Left'] = i 
    203         i += 1 
    204         _labels['Triangle Right'] = i 
    205         i += 1 
    206         _labels['Cross +'] = i 
    207         i += 1 
    208         _labels['Square'] = i 
    209         i += 1 
    210         _labels['Diamond'] = i 
    211         i += 1 
    212         _labels['Hexagon1'] = i 
    213         i += 1 
    214         _labels['Hexagon2'] = i 
    215         i += 1 
    216         _labels['Pentagon'] = i 
    217         i += 1 
    218         _labels['Line'] = i 
    219         i += 1 
    220         _labels['Dash'] = i 
    221         i += 1 
    222         _labels['Vline'] = i 
    223         i += 1 
    224         _labels['Step'] = i 
    225         return _labels 
    226      
    227     def get_color_label(self): 
    228         """ 
    229         Associates label to a specific color 
    230         """ 
    231         _labels = {} 
    232         i = 0 
    233         _labels['Blue'] = i 
    234         i += 1 
    235         _labels['Green'] = i 
    236         i += 1 
    237         _labels['Red'] = i 
    238         i += 1 
    239         _labels['Cyan'] = i 
    240         i += 1 
    241         _labels['Magenta'] = i 
    242         i += 1 
    243         _labels['Yellow'] = i 
    244         i += 1 
    245         _labels['Black'] = i 
    246         return _labels 
    247  
    248201    @staticmethod 
    249     def find_key(dic,val): 
     202    def find_key(dic, val): 
     203        """ 
     204        Find key 
     205        """ 
    250206        return [k for k, v in dic.iteritems() if v == val][0] 
    251207         
    252     def getCurrentValues(self): # returns (size,color,symbol,dataname) 
    253  
     208    def get_current_values(self):  
     209        """ 
     210        Get Current Values 
     211        :returns : (size, color, symbol, dataname) 
     212        """ 
    254213        size = float(self.final_size) 
    255         name = str(self.labelTextBox.GetValue()) 
    256         selTuple = self.symbolListBox.GetSelections() 
    257         symbol = appearanceDialog.find_key(self.sorted_sym_dic,int(selTuple[0])) 
    258         color = str(self.colorListBox.GetValue())  
    259   
    260         return(size,color,symbol,name) 
    261  
    262     def onOK(self,e): 
     214        name = str(self.labeltextbox.GetValue()) 
     215        seltuple = self.symbollistbox.GetSelections() 
     216        symbol = appearanceDialog.find_key(self.sorted_sym_dic,  
     217                                           int(seltuple[0])) 
     218        color = str(self.colorlistbox.GetValue())  
     219        return(size, color, symbol, name) 
     220 
     221    def on_ok(self, event): 
     222        """ 
     223        On OK button clicked 
     224        """ 
     225        event.Skip() 
    263226        self.okay_clicked = True 
    264  
    265227        self.Close() 
Note: See TracChangeset for help on using the changeset viewer.