Add UI buttons and modal dialogues.

Assets/Scripts/AppMouseKeyboard.cs

@@ -38,6 +38,9 @@ public class AppMouseKeyboard : MonoBehaviour
     InputTrackerMouse _inputTrackerMouse;
     InputTrackerKeyboard _inputTrackerKeyboard;
 
+    // Producer/Consumers
+    UIElementDrawer _uiElementDrawer;
+
     // Application state
     ConfigLoader _configLoader;
     GfxReplayPlayer _gfxReplayPlayer;
@@ -47,6 +50,9 @@ public class AppMouseKeyboard : MonoBehaviour
 
     protected void Awake()
     {
+        // Initialize Producer/Consumers.
+        _uiElementDrawer = new UIElementDrawer(_camera);
+
         // Initialize IKeyframeMessageConsumers.
         _serverKeyframeIdHandler = new ServerKeyframeIdHandler();
         _guiDrawer = new GuiDrawer(_appConfig, _camera);
@@ -63,6 +69,7 @@ protected void Awake()
             _cameraTransformHandler,
             _textRenderer,
             _loadingScreenOverlay,
+            _uiElementDrawer,
         };
 
         // Initialize IClientStateProducers.
@@ -72,6 +79,7 @@ protected void Awake()
         {
             _inputTrackerMouse,
             _inputTrackerKeyboard,
+            _uiElementDrawer,
         };
 
         // Initialize application state.

Assets/Scripts/ClientState.cs

@@ -11,6 +11,7 @@ public class ClientState
     public MouseInputData mouse;
     public Dictionary<string, string> connection_params_dict;
     public int? recentServerKeyframeId = null;
+    public UIElements ui;
 }
 
 [Serializable]
@@ -59,3 +60,10 @@ public class MouseInputData
     public float[] rayOrigin = new float[3];
     public float[] rayDirection = new float[3];
 }
+
+[Serializable]
+public class UIElements
+{
+    // Collection of UI buttons that were pressed since the last client message.
+    public List<string> buttonsPressed = new List<string>();
+}

Assets/Scripts/InputTrackerKeyboard.cs

@@ -120,6 +120,12 @@ public class InputTrackerKeyboard : IClientStateProducer
 
     public InputTrackerKeyboard()
     {
+        // Don't update if paused.
+        if (LoadProgressTracker.Instance.IsApplicationPaused)
+        {
+            return;
+        }
+
         // Bake the dict into an array for faster lookups.
         _keyMap = new int[KEY_COUNT];
         foreach (var kv in KEY_MAP)

Assets/Scripts/InputTrackerMouse.cs

@@ -31,6 +31,12 @@ public InputTrackerMouse(Camera camera)
 
     public void Update()
     {
+        // Don't update if paused.
+        if (LoadProgressTracker.Instance.IsApplicationPaused)
+        {
+            return;
+        }
+
         // Record all mouse actions that occurred OnEndFrame() call.
         // Note that multiple Unity frames may occur during that time.
         int mouseIndex = 0;

Assets/Scripts/Keyframe.cs

@@ -114,6 +114,7 @@ public class Message
     public List<TextMessage> texts;
     public AbsTransform camera;
     public int serverKeyframeId;
+    public Dialog dialog;
 }
 
 [Serializable]
@@ -141,3 +142,19 @@ public class TextMessage
     public string text;
     public List<float> position;
 }
+
+[Serializable]
+public class Dialog
+{
+    public string title;
+    public string text;
+    public Button[] buttons;
+}
+
+[Serializable]
+public class Button
+{
+    public string id;
+    public string text;
+    public bool enabled;
+}

Assets/Scripts/LoadProgressTracker.cs

@@ -15,6 +15,9 @@ public class LoadProgressTracker
     uint _successCount = 0;
     uint _failureCount = 0;
 
+    // TODO: Refactor.
+    public bool _modalDialogueShown = false;
+
     /// <summary>
     /// Get the LoadProgressTracker singleton.
     /// </summary>
@@ -62,6 +65,15 @@ public float EstimateProgress()
         return _progress;
     }
 
+    /// <summary>
+    /// Whether the client is paused. True if loading or modal dialogue shown.
+    /// TODO: Either move out or rename this class.
+    /// </summary>
+    public bool IsApplicationPaused { get
+    {
+        return IsLoading || _modalDialogueShown;
+    }}
+
     /// <summary>
     /// Whether at least one asset is being loaded.
     /// </summary>

Assets/Scripts/TextRenderer.cs

@@ -14,7 +14,7 @@ public class TextRendererGUI : MonoBehaviour
 
         public void OnGUI()
         {
-            if (texts == null || LoadProgressTracker.Instance.IsLoading)
+            if (texts == null || LoadProgressTracker.Instance.IsApplicationPaused)
             {
                 return;
             }

Assets/Scripts/UIElementDrawer.cs

@@ -0,0 +1,116 @@
+using System.Collections;
+using System.Collections.Generic;
+using System.Linq;
+using UnityEngine;
+
+public class UIElementDrawer : IKeyframeMessageConsumer, IClientStateProducer
+{
+    /// <summary>
+    /// Modal dialogue with buttons.
+    /// </summary>
+    class ModalDialogue : MonoBehaviour
+    {
+        Camera _camera = null;
+        Dialog _dialog = null;
+        HashSet<string> _clickedButtons = new HashSet<string>();
+
+        public List<string> ClickedButtons { get { return _clickedButtons.ToList(); } }
+
+        public void Reset()
+        {
+            _clickedButtons.Clear();
+        }
+
+        public void Initialize(Camera camera)
+        {
+            _camera = camera;
+        }
+
+        public void UpdateState(Dialog dialog)
+        {
+            _dialog = dialog;
+
+            // TODO: Cleanup
+            LoadProgressTracker.Instance._modalDialogueShown = dialog != null;
+        }
+
+        void OnGUI()
+        {
+            if (_dialog == null || _camera == null || LoadProgressTracker.Instance.IsLoading)
+            {
+                return;
+            }
+
+            // TODO: Size
+            int width = _camera.pixelWidth;
+            int height = _camera.pixelHeight;
+            int windowWidth = width / 3;
+            int windowHeight = width / 5;
+            int windowXOffset = windowWidth - (windowWidth / 2);
+            int windowYOffset = windowHeight - (windowHeight / 2);
+
+            GUILayout.BeginVertical();
+            GUILayout.BeginHorizontal();
+            GUI.color = Color.white;
+            GUILayout.Window(
+                9, // TODO
+                new Rect(16, 16, 0, 0),
+                WindowUpdate,
+                _dialog.title,
+                GUILayout.Width(256)
+            );
+            GUILayout.EndHorizontal();
+            GUILayout.EndVertical();
+        }
+
+        void WindowUpdate(int windowId)
+        {
+            GUILayout.BeginVertical();
+            GUI.color = Color.white;
+            GUILayout.Label(_dialog.text);
+
+            GUILayout.BeginHorizontal();
+            foreach (var button in _dialog.buttons)
+            {
+                GUI.enabled = button.enabled;
+                bool clicked = GUILayout.Button(button.text);
+                if (clicked)
+                {
+                    _clickedButtons.Add(button.id);
+                }
+                GUI.enabled = true;
+            }
+            GUILayout.EndHorizontal();
+            GUILayout.EndVertical();
+        }
+    }
+
+    UIElements _uiElements = new UIElements();
+
+    ModalDialogue _modalDialogue;
+
+    public UIElementDrawer(Camera camera)
+    {
+        _modalDialogue = new GameObject("LoadingScreenOverlay").AddComponent<ModalDialogue>();
+        _modalDialogue.Initialize(camera);
+    }
+
+    public void ProcessMessage(Message message)
+    {
+        Dialog dialog = message.dialog;
+        _modalDialogue.UpdateState(dialog);
+    }
+
+    void IUpdatable.Update() {}
+
+    public void UpdateClientState(ref ClientState state)
+    {
+        _uiElements.buttonsPressed = _modalDialogue.ClickedButtons;
+        state.ui = _uiElements;
+    }
+
+    public void OnEndFrame()
+    {
+        _modalDialogue.Reset();
+    }
+}