@@ -43,6 +43,7 @@ public UIBase() {
43
43
44
44
// add the menu
45
45
MainMenu = ( MainMenuPanel ) uiView . AddUIComponent ( typeof ( MainMenuPanel ) ) ;
46
+ MainMenu . gameObject . AddComponent < CustomKeyHandler > ( ) ;
46
47
#if DEBUG
47
48
DebugMenu = ( DebugMenuPanel ) uiView . AddUIComponent ( typeof ( DebugMenuPanel ) ) ;
48
49
#endif
@@ -52,6 +53,9 @@ public UIBase() {
52
53
53
54
~ UIBase ( ) {
54
55
UnityEngine . Object . Destroy ( MainMenuButton ) ;
56
+ CustomKeyHandler keyHandler = MainMenu . GetComponent < CustomKeyHandler > ( ) ;
57
+ if ( keyHandler != null )
58
+ UnityEngine . Object . Destroy ( keyHandler ) ;
55
59
}
56
60
57
61
public bool IsVisible ( ) {
@@ -68,13 +72,18 @@ public void ToggleMainMenu() {
68
72
internal void RebuildMenu ( ) {
69
73
Close ( ) ;
70
74
if ( MainMenu != null ) {
75
+ CustomKeyHandler keyHandler = MainMenu . GetComponent < CustomKeyHandler > ( ) ;
76
+ if ( keyHandler != null )
77
+ UnityEngine . Object . Destroy ( keyHandler ) ;
78
+
71
79
UnityEngine . Object . Destroy ( MainMenu ) ;
72
80
#if DEBUG
73
81
UnityEngine . Object . Destroy ( DebugMenu ) ;
74
82
#endif
75
83
}
76
84
var uiView = UIView . GetAView ( ) ;
77
85
MainMenu = ( MainMenuPanel ) uiView . AddUIComponent ( typeof ( MainMenuPanel ) ) ;
86
+ MainMenu . gameObject . AddComponent < CustomKeyHandler > ( ) ;
78
87
#if DEBUG
79
88
DebugMenu = ( DebugMenuPanel ) uiView . AddUIComponent ( typeof ( DebugMenuPanel ) ) ;
80
89
#endif
@@ -99,6 +108,7 @@ public void Show() {
99
108
SetToolMode ( TrafficManagerMode . Activated ) ;
100
109
_uiShown = true ;
101
110
MainMenuButton . UpdateSprites ( ) ;
111
+ UIView . SetFocus ( MainMenu ) ;
102
112
}
103
113
104
114
public void Close ( ) {
0 commit comments