From 2cffce805aef4f2430328b8c39f6262c94cfe362 Mon Sep 17 00:00:00 2001
From: jorolf <jorolf@gmx.de>
Date: Wed, 7 Nov 2018 00:48:54 +0100
Subject: [PATCH] add flashlight mod

---
 .../Mods/OsuModFlashlight.cs                  | 84 ++++++++++++++++++-
 1 file changed, 83 insertions(+), 1 deletion(-)

diff --git a/osu.Game.Rulesets.Osu/Mods/OsuModFlashlight.cs b/osu.Game.Rulesets.Osu/Mods/OsuModFlashlight.cs
index a337439593..b6e16ff537 100644
--- a/osu.Game.Rulesets.Osu/Mods/OsuModFlashlight.cs
+++ b/osu.Game.Rulesets.Osu/Mods/OsuModFlashlight.cs
@@ -1,12 +1,94 @@
 // Copyright (c) 2007-2018 ppy Pty Ltd <contact@ppy.sh>.
 // Licensed under the MIT Licence - https://raw.githubusercontent.com/ppy/osu/master/LICENCE
 
+using System;
+using osu.Framework.Allocation;
+using osu.Framework.Graphics;
+using osu.Framework.Graphics.OpenGL.Vertices;
+using osu.Framework.Graphics.Primitives;
+using osu.Framework.Graphics.Shaders;
+using osu.Framework.Graphics.Textures;
+using osu.Framework.Input;
+using osu.Framework.Input.Events;
 using osu.Game.Rulesets.Mods;
+using osu.Game.Rulesets.Osu.Objects;
+using osu.Game.Rulesets.UI;
+using OpenTK;
 
 namespace osu.Game.Rulesets.Osu.Mods
 {
-    public class OsuModFlashlight : ModFlashlight
+    public class OsuModFlashlight : ModFlashlight, IApplicableToRulesetContainer<OsuHitObject>
     {
         public override double ScoreMultiplier => 1.12;
+
+        public void ApplyToRulesetContainer(RulesetContainer<OsuHitObject> rulesetContainer)
+        {
+            rulesetContainer.KeyBindingInputManager.Add(new Flashlight
+            {
+                RelativeSizeAxes = Axes.Both,
+            });
+        }
+
+        private class Flashlight : Drawable, IRequireHighFrequencyMousePosition
+        {
+            private Shader shader;
+            private readonly MousePositionWrapper mousePosWrapper = new MousePositionWrapper();
+
+            protected override DrawNode CreateDrawNode() => new FlashlightDrawNode();
+
+            protected override void ApplyDrawNode(DrawNode node)
+            {
+                base.ApplyDrawNode(node);
+
+                var flashNode = (FlashlightDrawNode)node;
+
+                flashNode.Shader = shader;
+                flashNode.ScreenSpaceDrawQuad = ScreenSpaceDrawQuad;
+                flashNode.MousePosWrapper = mousePosWrapper;
+                flashNode.FlashlightSize = 100f;
+            }
+
+            [BackgroundDependencyLoader]
+            private void load(ShaderManager shaderManager)
+            {
+                shader = shaderManager.Load(VertexShaderDescriptor.POSITION, "Flashlight");
+            }
+
+            protected override bool OnMouseMove(MouseMoveEvent e)
+            {
+                mousePosWrapper.MousePosition = e.ScreenSpaceMousePosition;
+                return base.OnMouseMove(e);
+            }
+        }
+
+        private class MousePositionWrapper
+        {
+            public Vector2 MousePosition;
+        }
+
+        private class FlashlightDrawNode : DrawNode
+        {
+            public Shader Shader;
+            public Quad ScreenSpaceDrawQuad;
+            public MousePositionWrapper MousePosWrapper;
+            public float FlashlightSize;
+            private bool sizeSet;
+
+            public override void Draw(Action<TexturedVertex2D> vertexAction)
+            {
+                base.Draw(vertexAction);
+
+                Shader.Bind();
+                // ReSharper disable once AssignmentInConditionalExpression
+                if(sizeSet = !sizeSet)
+                    Shader.GetUniform<float>("flashlightSize").UpdateValue(ref FlashlightSize);
+
+                Shader.GetUniform<Vector2>("mousePos").UpdateValue(ref MousePosWrapper.MousePosition);
+
+                Texture.WhitePixel.DrawQuad(ScreenSpaceDrawQuad, DrawColourInfo.Colour, vertexAction: vertexAction);
+
+                Shader.Unbind();
+            }
+        }
     }
 }