From d706b0a490bb8228d74c97ba6211cce0d26e02b2 Mon Sep 17 00:00:00 2001
From: AndersMalmgren <anders.malmgren@agero.se>
Date: Mon, 17 Feb 2014 10:07:09 +0100
Subject: [PATCH] Read and write mouse wheel data

---
 FreePIE.Core.Plugins/MousePlugin.cs | 64 +++++++++++++++--------------
 1 file changed, 33 insertions(+), 31 deletions(-)

diff --git a/FreePIE.Core.Plugins/MousePlugin.cs b/FreePIE.Core.Plugins/MousePlugin.cs
index c9a8f0f6..f0308094 100644
--- a/FreePIE.Core.Plugins/MousePlugin.cs
+++ b/FreePIE.Core.Plugins/MousePlugin.cs
@@ -5,6 +5,7 @@
 using FreePIE.Core.Contracts;
 using FreePIE.Core.Plugins.Strategies;
 using SlimDX.DirectInput;
+using SlimDX.RawInput;
 
 namespace FreePIE.Core.Plugins
 {
@@ -15,6 +16,8 @@ public class MousePlugin : Plugin
         // Mouse position state variables
         private double deltaXOut;
         private double deltaYOut;
+        private int wheel;
+        public const int WheelMax = 120;
 
         private DirectInput directInputInstance = new DirectInput();
         private Mouse mouseDevice;
@@ -70,16 +73,6 @@ public override string FriendlyName
             get { return "Mouse"; }
         }
 
-        public override bool GetProperty(int index, IPluginProperty property)
-        {
-            return false;
-        }
-
-        public override bool SetProperties(Dictionary<string, object> properties)
-        {
-           return true;
-        }
-
         static private MouseKeyIO.MOUSEINPUT MouseInput(int x, int y, uint data, uint t, uint flag)
         {
            var mi = new MouseKeyIO.MOUSEINPUT {dx = x, dy = y, mouseData = data, time = t, dwFlags = flag};
@@ -89,12 +82,12 @@ static private MouseKeyIO.MOUSEINPUT MouseInput(int x, int y, uint data, uint t,
         public override void DoBeforeNextExecute()
         {
             // If a mouse command was given in the script, issue it all at once right here
-            if ((int)deltaXOut != 0 || (int)deltaYOut != 0)
+            if ((int)deltaXOut != 0 || (int)deltaYOut != 0 || wheel != 0)
             {
 
                 var input = new MouseKeyIO.INPUT[1];
                 input[0].type = MouseKeyIO.INPUT_MOUSE;
-                input[0].mi = MouseInput((int)deltaXOut, (int)deltaYOut, 0, 0, MouseKeyIO.MOUSEEVENTF_MOVE);
+                input[0].mi = MouseInput((int)deltaXOut, (int)deltaYOut, (uint)wheel, 0, MouseKeyIO.MOUSEEVENTF_MOVE | MouseKeyIO.MOUSEEVENTF_WHEEL);
 
                 MouseKeyIO.SendInput(1, input, Marshal.SizeOf(input[0].GetType()));
 
@@ -107,6 +100,8 @@ public override void DoBeforeNextExecute()
                 {
                     deltaYOut = deltaYOut - (int)deltaYOut;
                 }
+
+                wheel = 0;
             }
 
             currentMouseState = null;  // flush the mouse state
@@ -121,15 +116,7 @@ public double DeltaX
                 deltaXOut = deltaXOut + value;
             }
 
-            get
-            {
-                // Retrieve the mouse state only once per iteration to avoid getting
-                // zeros on subsequent calls
-                if (currentMouseState == null)
-                    currentMouseState = mouseDevice.GetCurrentState();
-
-                return currentMouseState.X;
-            }
+            get { return CurrentMouseState.X; }
         }
 
         public double DeltaY
@@ -139,26 +126,30 @@ public double DeltaY
                 deltaYOut = deltaYOut + value;
             }
 
+            get { return CurrentMouseState.Y; }
+        }
+
+        public int Wheel
+        {
+            get { return CurrentMouseState.Z; }
+            set { wheel = value; }
+            
+        }
+
+        private MouseState CurrentMouseState
+        {
             get
             {
-                // Retrieve the mouse state only once per iteration to avoid getting
-                // zeros on subsequent calls
                 if (currentMouseState == null)
                     currentMouseState = mouseDevice.GetCurrentState();
 
-                return currentMouseState.Y;
+                return currentMouseState;
             }
         }
 
         public bool IsButtonDown(int index)
         {
-
-            // Retrieve the mouse state only once per iteration to avoid getting
-            // zeros on subsequent calls
-            if (currentMouseState == null)
-                currentMouseState = mouseDevice.GetCurrentState();
-
-            return currentMouseState.IsPressed(index);
+            return CurrentMouseState.IsPressed(index);
         }
 
         public bool IsButtonPressed(int button)
@@ -242,6 +233,11 @@ public class MouseGlobal : UpdateblePluginGlobal<MousePlugin>
     {
         public MouseGlobal(MousePlugin plugin) : base(plugin) { }
 
+        public int wheelMax
+        {
+            get { return MousePlugin.WheelMax; }
+        }
+
         public double deltaX
         {
             get { return plugin.DeltaX; }
@@ -254,6 +250,12 @@ public double deltaY
             set { plugin.DeltaY = value; }
         }
 
+        public int wheel
+        {
+            get { return plugin.Wheel; }
+            set { plugin.Wheel = value; }
+        }
+
         public bool leftButton
         {
             get { return plugin.IsButtonDown(0); }