Commit 65bd8595ba8e39b32ee75771a9e5c182f30d0ae2

Authored by Paulo Henrique Junqueira Amorim
1 parent 338470d1

ENH: wx.NewId to generate the id from menu objects

Showing 1 changed file with 6 additions and 8 deletions   Show diff stats
invesalius/gui/widgets/slice_menu.py
@@ -29,19 +29,19 @@ class SliceMenu(wx.Menu): @@ -29,19 +29,19 @@ class SliceMenu(wx.Menu):
29 def __init__(self): 29 def __init__(self):
30 wx.Menu.__init__(self) 30 wx.Menu.__init__(self)
31 self.ID_TO_TOOL_ITEM = {} 31 self.ID_TO_TOOL_ITEM = {}
32 - new_id = 0 32 +
33 33
34 submenu_wl = wx.Menu() 34 submenu_wl = wx.Menu()
35 for name in sorted(const.WINDOW_LEVEL): 35 for name in sorted(const.WINDOW_LEVEL):
  36 + new_id = wx.NewId()
36 wl_item = wx.MenuItem(submenu_wl, new_id,\ 37 wl_item = wx.MenuItem(submenu_wl, new_id,\
37 name, kind=wx.ITEM_RADIO) 38 name, kind=wx.ITEM_RADIO)
38 submenu_wl.AppendItem(wl_item) 39 submenu_wl.AppendItem(wl_item)
39 self.ID_TO_TOOL_ITEM[new_id] = name 40 self.ID_TO_TOOL_ITEM[new_id] = name
40 - new_id += 1  
41 41
42 submenu_pseudo_colors = wx.Menu() 42 submenu_pseudo_colors = wx.Menu()
43 for name in sorted(const.SLICE_COLOR_TABLE): 43 for name in sorted(const.SLICE_COLOR_TABLE):
44 - new_id += 1 44 + new_id = wx.NewId()
45 color_item = wx.MenuItem(submenu_wl, new_id,\ 45 color_item = wx.MenuItem(submenu_wl, new_id,\
46 name, kind=wx.ITEM_RADIO) 46 name, kind=wx.ITEM_RADIO)
47 submenu_pseudo_colors.AppendItem(color_item) 47 submenu_pseudo_colors.AppendItem(color_item)
@@ -62,18 +62,16 @@ class SliceMenu(wx.Menu): @@ -62,18 +62,16 @@ class SliceMenu(wx.Menu):
62 id = evt.GetId() 62 id = evt.GetId()
63 key = self.ID_TO_TOOL_ITEM[evt.GetId()] 63 key = self.ID_TO_TOOL_ITEM[evt.GetId()]
64 64
65 - if (id <= len(const.WINDOW_LEVEL)): 65 + if(key in const.WINDOW_LEVEL.keys()):
66 window, level = const.WINDOW_LEVEL[key] 66 window, level = const.WINDOW_LEVEL[key]
67 ps.Publisher().sendMessage('Bright and contrast adjustment image', 67 ps.Publisher().sendMessage('Bright and contrast adjustment image',
68 (window, level)) 68 (window, level))
69 ps.Publisher().sendMessage('Update slice viewer') 69 ps.Publisher().sendMessage('Update slice viewer')
70 70
71 - elif(id > len(const.WINDOW_LEVEL) and\  
72 - (id <= len(const.SLICE_COLOR_TABLE) + len(const.WINDOW_LEVEL))): 71 + elif(key in const.SLICE_COLOR_TABLE.keys()):
73 values = const.SLICE_COLOR_TABLE[key] 72 values = const.SLICE_COLOR_TABLE[key]
74 -  
75 ps.Publisher().sendMessage('Change color table from background image', values) 73 ps.Publisher().sendMessage('Change color table from background image', values)
76 ps.Publisher().sendMessage('Update slice viewer') 74 ps.Publisher().sendMessage('Update slice viewer')
77 - 75 +
78 evt.Skip() 76 evt.Skip()
79 77