Merge pull request #30 from Psychloor/LineTracer

[VR] Line tracer

ReModCE.Loader/ReMod.Loader.cs

@@ -16,7 +16,7 @@ namespace ReModCE.Loader
     public static class BuildInfo
     {
         public const string Name = "ReModCE";
-        public const string Author = "Requi, FenrixTheFox, Xaiver, Potato";
+        public const string Author = "Requi, FenrixTheFox, Xaiver, Potato, Psychloor";
         public const string Company = null;
         public const string Version = "1.0.0.3";
         public const string DownloadLink = "https://github.com/RequiDev/ReModCE/releases/latest/";

ReModCE/Components/LineTracerComponent.cs

@@ -0,0 +1,164 @@
+namespace ReModCE.Components
+{
+
+    using System.Collections.Generic;
+
+    using global::ReModCE.Managers;
+
+    using ReMod.Core;
+    using ReMod.Core.Managers;
+    using ReMod.Core.UI.QuickMenu;
+    using ReMod.Core.Unity;
+    using ReMod.Core.VRChat;
+
+    using UnityEngine;
+    using UnityEngine.XR;
+
+    using VRC;
+    using VRC.Core;
+
+    [ComponentPriority(8)]
+    public sealed class LineTracerComponent : ModComponent
+    {
+
+        private const string RightTrigger = "Oculus_CrossPlatform_SecondaryIndexTrigger";
+
+        private static readonly int Cull = Shader.PropertyToID("_Cull");
+
+        private static readonly int DstBlend = Shader.PropertyToID("_DstBlend");
+
+        private static readonly int SrcBlend = Shader.PropertyToID("_SrcBlend");
+
+        private static readonly int ZWrite = Shader.PropertyToID("_ZWrite");
+
+        // Requi told me no, do this
+        private ConfigValue<Color> FriendsColor;
+        private ConfigValue<Color> OthersColor;
+
+        private static Material lineMaterial;
+
+        private static RenderObjectListener renderObjectListener;
+
+        private readonly List<Player> cachedPlayers = new();
+
+        private ConfigValue<bool> lineTracerEnabled;
+
+        private bool materialSetup;
+
+        private Transform originTransform;
+
+        private bool riskyFunctionsAllowed;
+
+        public LineTracerComponent()
+        {
+            RenderObjectListener.RegisterSafe();
+
+            lineTracerEnabled = new ConfigValue<bool>(
+                nameof(lineTracerEnabled),
+                false,
+                "Enable Line Tracer (Right Trigger)");
+
+            FriendsColor = new ConfigValue<Color>(nameof(FriendsColor), Color.yellow);
+            OthersColor = new ConfigValue<Color>(nameof(OthersColor), Color.magenta);
+
+            RiskyFunctionsManager.Instance.OnRiskyFunctionsChanged += b => riskyFunctionsAllowed = b;
+        }
+
+        public override void OnEnterWorld(ApiWorld world, ApiWorldInstance instance)
+        {
+            cachedPlayers.Clear();
+        }
+
+        public override void OnPlayerJoined(Player player)
+        {
+            if (player.prop_APIUser_0.IsSelf) return;
+            cachedPlayers.Add(player);
+        }
+
+        public override void OnPlayerLeft(Player player)
+        {
+            cachedPlayers.Remove(player);
+        }
+
+        public override void OnUiManagerInit(UiManager uiManager)
+        {
+            ReMenuCategory espMenu = uiManager.MainMenu.GetCategoryPage("Visuals").GetCategory("ESP/Highlights");
+
+            espMenu.AddToggle(
+                "[VR] Line Tracer (Right Trigger)",
+                "Draw lines from your hand to each player in world",
+                lineTracerEnabled);
+
+            // Late enough that the camera is on now
+            renderObjectListener = VRCVrCamera.field_Private_Static_VRCVrCamera_0.field_Public_Camera_0.gameObject
+                                              .AddComponent<RenderObjectListener>();
+            renderObjectListener.hideFlags = HideFlags.HideAndDontSave;
+            renderObjectListener.RenderObject += OnRenderObject;
+        }
+
+        private static Transform GetOriginTransform()
+        {
+            VRCPlayer localPlayer = VRCPlayer.field_Internal_Static_VRCPlayer_0;
+            if (!localPlayer) return null;
+
+            Animator localAnimator = localPlayer.GetAvatarObject()?.GetComponent<Animator>();
+            if (localAnimator == null
+                || !localAnimator.isHuman) return null;
+
+            // try to grab from the tip of the finger all the way to the hand. otherwise fail
+            return localAnimator.GetBoneTransform(HumanBodyBones.RightIndexDistal)
+                   ?? localAnimator.GetBoneTransform(HumanBodyBones.RightIndexIntermediate)
+                   ?? localAnimator.GetBoneTransform(HumanBodyBones.RightIndexProximal)
+                   ?? localAnimator.GetBoneTransform(HumanBodyBones.RightHand);
+        }
+
+        private new void OnRenderObject()
+        {
+            // In World/Room
+            if (!RoomManager.field_Private_Static_Boolean_0) return;
+
+            if (!lineTracerEnabled
+                || !riskyFunctionsAllowed
+                || !XRDevice.isPresent) return;
+
+            if (Input.GetAxis(RightTrigger) < 0.4f) return;
+
+            if (!materialSetup) SetupMaterial();
+
+            // local player
+            if (!originTransform) originTransform = GetOriginTransform();
+            if (originTransform == null) return;
+
+            // Initialize GL
+            GL.Begin(1); // Lines
+            lineMaterial.SetPass(0);
+
+            // goes way faster to re-use the cached players
+            foreach (Player player in cachedPlayers)
+            {
+                if (!player) continue;
+                GL.Color(
+                    player.GetAPIUser().isFriend ? FriendsColor.Value : OthersColor.Value);
+                GL.Vertex(originTransform.position);
+                GL.Vertex(player.transform.position);
+            }
+
+            // End GL
+            GL.End();
+        }
+
+        private void SetupMaterial()
+        {
+            lineMaterial = Material.GetDefaultLineMaterial();
+            lineMaterial.SetInt(SrcBlend, 5);
+            lineMaterial.SetInt(DstBlend, 10);
+            lineMaterial.SetInt(Cull, 0);
+            lineMaterial.SetInt(ZWrite, 0);
+            lineMaterial.hideFlags = HideFlags.HideAndDontSave;
+
+            materialSetup = true;
+        }
+
+    }
+
+}