1.8.2

* Added support for games which use the new InputSystem module and have disabled LegacyInputModule

src/Helpers/InputHelper.cs

@@ -5,18 +5,41 @@ using UnityEngine;
 namespace Explorer
 {
     /// <summary>
-    /// Version-agnostic UnityEngine Input module using Reflection.
+    /// Version-agnostic Input module using Reflection.
     /// </summary>
     public static class InputHelper
     {
         // If Input module failed to load at all
         public static bool NO_INPUT;
 
-        // Base UnityEngine.Input class
-        private static Type Input => _input ?? (_input = ReflectionHelpers.GetTypeByName("UnityEngine.Input"));
+        // If using new InputSystem module
+        public static bool USING_NEW_INPUT;
+
+        // Cached Types
+        private static Type TInput => _input ?? (_input = ReflectionHelpers.GetTypeByName("UnityEngine.Input"));
         private static Type _input;
 
-        // Cached member infos
+        private static Type TKeyboard => _keyboardSys ?? (_keyboardSys = ReflectionHelpers.GetTypeByName("UnityEngine.InputSystem.Keyboard"));
+        private static Type _keyboardSys;
+
+        private static Type TMouse => _mouseSys ?? (_mouseSys = ReflectionHelpers.GetTypeByName("UnityEngine.InputSystem.Mouse"));
+        private static Type _mouseSys;
+
+        private static Type TKey => _key ?? (_key = ReflectionHelpers.GetTypeByName("UnityEngine.InputSystem.Key"));
+        private static Type _key;
+
+        // Cached member infos (new system)
+        private static PropertyInfo _keyboardCurrent;
+        private static PropertyInfo _kbItemProp;
+        private static PropertyInfo _isPressed;
+        private static PropertyInfo _wasPressedThisFrame;
+        private static PropertyInfo _mouseCurrent;
+        private static PropertyInfo _leftButton;
+        private static PropertyInfo _rightButton;
+        private static PropertyInfo _position;
+        private static MethodInfo _readValueMethod;
+
+        // Cached member infos (legacy)
         private static PropertyInfo _mousePosition;
         private static MethodInfo _getKey;
         private static MethodInfo _getKeyDown;
@@ -25,49 +48,153 @@ namespace Explorer
 
         public static void Init()
         {
-            if (Input == null && !TryManuallyLoadInput())
+            if (TKeyboard != null || TryManuallyLoadNewInput())
             {
-                NO_INPUT = true;
+                InitNewInput();
                 return;
             }
 
-            // Cache reflection now that we know Input is loaded
+            if (TInput != null || TryManuallyLoadLegacyInput())
+            {
+                InitLegacyInput();
+                return;
+            }
 
-            _mousePosition = Input.GetProperty("mousePosition");
-
-            _getKey = Input.GetMethod("GetKey", new Type[] { typeof(KeyCode) });
-            _getKeyDown = Input.GetMethod("GetKeyDown", new Type[] { typeof(KeyCode) });
-            _getMouseButton = Input.GetMethod("GetMouseButton", new Type[] { typeof(int) });
-            _getMouseButtonDown = Input.GetMethod("GetMouseButtonDown", new Type[] { typeof(int) });
+            ExplorerCore.LogWarning("Could not find any Input module!");
+            NO_INPUT = true;
         }
 
-#pragma warning disable IDE1006 // Camel-case property (Unity style)
-        public static Vector3 mousePosition
+        private static void InitNewInput()
+        {
+            ExplorerCore.Log("Initializing new InputSystem support...");
+
+            USING_NEW_INPUT = true;
+
+            _keyboardCurrent = TKeyboard.GetProperty("current");
+            _kbItemProp = TKeyboard.GetProperty("Item", new Type[] { TKey });
+
+            var btnControl = ReflectionHelpers.GetTypeByName("UnityEngine.InputSystem.Controls.ButtonControl");
+            _isPressed = btnControl.GetProperty("isPressed");
+            _wasPressedThisFrame = btnControl.GetProperty("wasPressedThisFrame");
+
+            _mouseCurrent = TMouse.GetProperty("current");
+            _leftButton = TMouse.GetProperty("leftButton");
+            _rightButton = TMouse.GetProperty("rightButton");
+
+            _position = ReflectionHelpers.GetTypeByName("UnityEngine.InputSystem.Pointer")
+                                         .GetProperty("position");
+
+            _readValueMethod = ReflectionHelpers.GetTypeByName("UnityEngine.InputSystem.InputControl`1")
+                               .MakeGenericType(typeof(Vector2))
+                               .GetMethod("ReadValue");
+        }
+
+        private static void InitLegacyInput()
+        {
+            ExplorerCore.Log("Initializing Legacy Input support...");
+
+            _mousePosition = TInput.GetProperty("mousePosition");
+            _getKey = TInput.GetMethod("GetKey", new Type[] { typeof(KeyCode) });
+            _getKeyDown = TInput.GetMethod("GetKeyDown", new Type[] { typeof(KeyCode) });
+            _getMouseButton = TInput.GetMethod("GetMouseButton", new Type[] { typeof(int) });
+            _getMouseButtonDown = TInput.GetMethod("GetMouseButtonDown", new Type[] { typeof(int) });
+        }
+
+        private static bool TryManuallyLoadNewInput()
+        {
+            if (ReflectionHelpers.LoadModule("Unity.InputSystem") && TKeyboard != null)
+            {
+                ExplorerCore.Log("Loaded new InputSystem module!");
+                return true;
+            }
+            else
+            {
+                return false;
+            }
+        }
+
+        private static bool TryManuallyLoadLegacyInput()
+        {
+            if ((ReflectionHelpers.LoadModule("UnityEngine.InputLegacyModule") || ReflectionHelpers.LoadModule("UnityEngine.CoreModule"))
+                && TInput != null)
+            {
+                ExplorerCore.Log("Loaded legacy InputModule!");
+                return true;
+            }
+            else
+            {
+                return false;
+            }
+        }
+
+        public static Vector3 MousePosition
         {
             get
             {
                 if (NO_INPUT) return Vector3.zero;
+
+                if (USING_NEW_INPUT)
+                {
+                    var mouse = _mouseCurrent.GetValue(null, null);
+                    var pos = _position.GetValue(mouse, null);
+
+                    return (Vector2)_readValueMethod.Invoke(pos, new object[0]);
+                }
+
                 return (Vector3)_mousePosition.GetValue(null, null);
             }
         }
-#pragma warning restore IDE1006
 
         public static bool GetKeyDown(KeyCode key)
         {
             if (NO_INPUT) return false;
+
+            if (USING_NEW_INPUT)
+            {
+                var parsed = Enum.Parse(TKey, key.ToString());
+                var currentKB = _keyboardCurrent.GetValue(null, null);
+                var actualKey = _kbItemProp.GetValue(currentKB, new object[] { parsed });
+
+                return (bool)_wasPressedThisFrame.GetValue(actualKey, null);
+            }
+
             return (bool)_getKeyDown.Invoke(null, new object[] { key });
         }
 
         public static bool GetKey(KeyCode key)
         {
             if (NO_INPUT) return false;
+
+            if (USING_NEW_INPUT)
+            {
+                var parsed = Enum.Parse(TKey, key.ToString());
+                var currentKB = _keyboardCurrent.GetValue(null, null);
+                var actualKey = _kbItemProp.GetValue(currentKB, new object[] { parsed });
+
+                return (bool)_isPressed.GetValue(actualKey, null);
+            }
+
             return (bool)_getKey.Invoke(null, new object[] { key });
         }
 
-        /// <param name="btn">1 = left, 2 = middle, 3 = right, etc</param>
+        /// <param name="btn">0/1 = left, 2 = middle, 3 = right, etc</param>
         public static bool GetMouseButtonDown(int btn)
         {
             if (NO_INPUT) return false;
+
+            if (USING_NEW_INPUT)
+            {
+                var mouse = _mouseCurrent.GetValue(null, null);
+
+                PropertyInfo btnProp;
+                if (btn < 2) btnProp = _leftButton;
+                else         btnProp = _rightButton;
+
+                var actualBtn = btnProp.GetValue(mouse, null);
+
+                return (bool)_wasPressedThisFrame.GetValue(actualBtn, null);
+            }
+
             return (bool)_getMouseButtonDown.Invoke(null, new object[] { btn });
         }
 
@@ -75,24 +202,21 @@ namespace Explorer
         public static bool GetMouseButton(int btn)
         {
             if (NO_INPUT) return false;
+
+            if (USING_NEW_INPUT)
+            {
+                var mouse = _mouseCurrent.GetValue(null, null);
+
+                PropertyInfo btnProp;
+                if (btn < 2) btnProp = _leftButton;
+                else btnProp = _rightButton;
+
+                var actualBtn = btnProp.GetValue(mouse, null);
+
+                return (bool)_isPressed.GetValue(actualBtn, null);
+            }
+
             return (bool)_getMouseButton.Invoke(null, new object[] { btn });
         }
-
-        private static bool TryManuallyLoadInput()
-        {
-            ExplorerCore.Log("UnityEngine.Input is null, trying to load manually....");
-
-            if ((ReflectionHelpers.LoadModule("UnityEngine.InputLegacyModule") || ReflectionHelpers.LoadModule("UnityEngine.CoreModule")) 
-                && Input != null)
-            {
-                ExplorerCore.Log("Ok!");
-                return true;
-            }
-            else
-            {
-                ExplorerCore.Log("Could not load Input module!");
-                return false;
-            }
-        }
     }
 }