Test/Library/PackageCache/com.unity.inputsystem@2436e68c60c4/InputSystem/Editor/DeviceSimulator/InputSystemPlugin.cs

#if UNITY_EDITOR

using System;
using System.Collections.Generic;
using UnityEditor;
using UnityEditor.DeviceSimulation;
using UnityEditor.UIElements;
using UnityEngine.InputSystem.LowLevel;
using UnityEngine.UIElements;

namespace UnityEngine.InputSystem.Editor
{
    internal class InputSystemPlugin : DeviceSimulatorPlugin
    {
        internal Touchscreen SimulatorTouchscreen;

        // The Input Debugger reads this to show its touch simulation toggle as unavailable.
        internal static bool isSimulatorActive => EnhancedTouch.TouchSimulation.s_DeviceSimulatorCount > 0;

        private bool m_InputSystemEnabled;
        private bool m_Quitting;
        private bool m_ConflictingDevicesDisabled;
        private VisualElement m_RootElement;
        private EditorWindow m_LastFocusedWindow;
        private List<InputDevice> m_DisabledDevices;

        public override string title => "Input System";

        public override void OnCreate()
        {
            m_InputSystemEnabled = EditorPlayerSettingHelpers.newSystemBackendsEnabled;
            if (m_InputSystemEnabled)
            {
                // Monitor whether the editor is quitting to avoid risking unsafe EnableDevice while quitting
                UnityEditor.EditorApplication.quitting += OnQuitting;

                // Poll the active window so conflicting devices are only disabled while the simulator is focused.
                UnityEditor.EditorApplication.update += OnUpdate;

                m_DisabledDevices = new List<InputDevice>();

                // deviceSimulator is never null when the plugin is instantiated by a simulator window, but it can be null during unit tests
                if (deviceSimulator != null)
                    deviceSimulator.touchScreenInput += OnTouchEvent;
                InputSystem.onDeviceChange += OnDeviceChange;

                EnhancedTouch.TouchSimulation.IncreaseDeviceSimulatorCount();

                SimulatorTouchscreen = InputSystem.AddDevice<Touchscreen>("Device Simulator Touchscreen");
            }
        }

        internal void OnTouchEvent(TouchEvent touchEvent)
        {
            // Input System does not accept 0 as id
            var id = touchEvent.touchId + 1;

            InputSystem.QueueStateEvent(SimulatorTouchscreen,
                new TouchState
                {
                    touchId = id,
                    phase = ToInputSystem(touchEvent.phase),
                    position = touchEvent.position
                });
        }

        public override VisualElement OnCreateUI()
        {
            m_RootElement = new VisualElement();
            m_RootElement.Add(new HelpBox(
                L10n.Tr("Manages Input System devices while the Simulator is focused."),
                HelpBoxMessageType.Info));
            return m_RootElement;
        }

        private void OnUpdate()
        {
            if (!EditorApplication.isPlaying)
            {
                if (m_ConflictingDevicesDisabled)
                {
                    SetConflictingDevicesDisabled(false);
                    m_LastFocusedWindow = null;
                }
                return;
            }

            var focusedWindow = EditorWindow.focusedWindow;
            if (focusedWindow == m_LastFocusedWindow)
                return;
            m_LastFocusedWindow = focusedWindow;

            var simulatorFocused =
                m_RootElement != null
                && focusedWindow != null
                && focusedWindow.rootVisualElement.panel == m_RootElement.panel;

            SetConflictingDevicesDisabled(simulatorFocused);
        }

        // Exposed internally so tests can drive the focus transition without a live SimulatorWindow.
        // OnUpdate itself can't run in a unit test: it needs play mode and a real panel to compare against.
        internal void SetConflictingDevicesDisabled(bool disabled)
        {
            if (disabled == m_ConflictingDevicesDisabled)
                return;

            if (disabled)
            {
                // UGUI elements like a button don't get pressed when multiple pointers for example mouse and touchscreen are sending data at the same time
                foreach (var device in InputSystem.devices)
                    DisableConflictingDevice(device);
            }
            else
            {
                foreach (var device in m_DisabledDevices)
                {
                    // Note that m_Quitting is used here to mitigate the problem reported in issue tracker:
                    // https://issuetracker.unity3d.com/product/unity/issues/guid/UUM-10774.
                    // Enabling a device will call into IOCTL of backend which may be destroyed prior
                    // to this callback on Unity version. This is not a fix for the actual problem
                    // of shutdown order but a package fix to mitigate this problem.
                    // The core problem with the destruction order was still there in Unity 6.5.
                    if (device.added && !m_Quitting)
                        InputSystem.EnableDevice(device);
                }
                m_DisabledDevices.Clear();
            }

            m_ConflictingDevicesDisabled = disabled;
        }

        private void DisableConflictingDevice(InputDevice device)
        {
            if (device.native && (device is Mouse || device is Pen) && device.enabled)
            {
                InputSystem.DisableDevice(device);
                m_DisabledDevices.Add(device);
            }
        }

        private void OnDeviceChange(InputDevice device, InputDeviceChange change)
        {
            // Only disable newly added/reconnected devices while the simulator is the active window.
            if (!m_ConflictingDevicesDisabled)
                return;

            if (change == InputDeviceChange.Added || change == InputDeviceChange.Reconnected)
                DisableConflictingDevice(device);
        }

        private static UnityEngine.InputSystem.TouchPhase ToInputSystem(UnityEditor.DeviceSimulation.TouchPhase original)
        {
            switch (original)
            {
                case UnityEditor.DeviceSimulation.TouchPhase.Began:
                    return UnityEngine.InputSystem.TouchPhase.Began;
                case UnityEditor.DeviceSimulation.TouchPhase.Moved:
                    return UnityEngine.InputSystem.TouchPhase.Moved;
                case UnityEditor.DeviceSimulation.TouchPhase.Ended:
                    return UnityEngine.InputSystem.TouchPhase.Ended;
                case UnityEditor.DeviceSimulation.TouchPhase.Canceled:
                    return UnityEngine.InputSystem.TouchPhase.Canceled;
                case UnityEditor.DeviceSimulation.TouchPhase.Stationary:
                    return UnityEngine.InputSystem.TouchPhase.Stationary;
                default:
                    throw new ArgumentOutOfRangeException(nameof(original), original, "Unexpected value");
            }
        }

        public override void OnDestroy()
        {
            if (m_InputSystemEnabled)
            {
                // deviceSimulator is never null when the plugin is instantiated by a simulator window, but it can be null during unit tests
                if (deviceSimulator != null)
                    deviceSimulator.touchScreenInput -= OnTouchEvent;
                InputSystem.onDeviceChange -= OnDeviceChange;

                UnityEditor.EditorApplication.quitting -= OnQuitting;
                UnityEditor.EditorApplication.update -= OnUpdate;

                if (SimulatorTouchscreen != null)
                    InputSystem.RemoveDevice(SimulatorTouchscreen);

                SetConflictingDevicesDisabled(false);

                EnhancedTouch.TouchSimulation.DecreaseDeviceSimulatorCount();
                m_RootElement = null;
            }
        }

        private void OnQuitting()
        {
            m_Quitting = true;
        }
    }
}

#endif