Commit 65bd8595ba8e39b32ee75771a9e5c182f30d0ae2
1 parent
338470d1
Exists in
master
and in
68 other branches
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 |