Subversion Repositories Mobile Apps.GyroMouse

//

//  MouseHandler.swift

//  GyroServer

//

//  Created by Matteo Riva on 07/08/15.

//  Copyright © 2015 Matteo Riva. All rights reserved.

//

import CoreGraphics

import Cocoa

import Carbon

import CoreServices

class MouseHandler: NSObject, ServerHandlerDelegate {

    private let server = ServerHandler()

    private var activeScreen: Int {

        get {

            return UserDefaults.standard.integer(forKey: "activeScreen")

        }

    }

    private var mLoc: (x: CGFloat, y: CGFloat) = (x: 0, y: 0)

    private var isDragging = false

    override init() {

        super.init()

        resetPointerPosition(moveMouse: false)

        server.startBroadcast()

        NotificationCenter.default.addObserver(forName: ActiveScreenDidChangeNotification, object: nil, queue: OperationQueue.main) {[unowned self] (_) -> Void in

            self.resetPointerPosition(moveMouse: false)

        }

        NotificationCenter.default.addObserver(forName: ServerDidConnectNotification, object: nil, queue: OperationQueue.main) {[unowned self] (_) -> Void in

            self.resetPointerPosition(moveMouse: true)

            self.server.delegate = self

        }

        NotificationCenter.default.addObserver(forName: ServerDidDisconnectNotification, object: nil, queue: OperationQueue.main) {[unowned self] (_) -> Void in

            self.server.delegate = nil

            self.resetPointerPosition(moveMouse: false)

        }

    }

    deinit {

        NotificationCenter.default.removeObserver(self)

    }

    //MARK: - Mouse movements

    var averagedPoints: [CGPoint] = [ CGPoint(x: -1, y: -1), CGPoint(x: -1, y: -1), CGPoint(x: -1, y: -1), CGPoint(x: -1, y: -1), CGPoint(x: -1, y: -1), CGPoint(x: -1, y: -1) ]

    func computePointerMovement(grav: (X: Double, Y: Double, Z: Double), rotat: (X: Double, Z: Double), acc: (X: Double, Y: Double, Z: Double), moveVelocity: Double) {

        let frame = NSScreen.screens[activeScreen].frame

        let bounds = (origin: frame.origin, width: frame.maxX - 1, height: frame.maxY - 1)

        // compute new mouse location

        mLoc = (

            x: max(bounds.origin.x, min(mLoc.x + CGFloat((rotat.Z * -1) * (1 + abs(acc.Z)) * moveVelocity * abs(grav.Z)), bounds.width)),

            y: max(0,               min(mLoc.y + CGFloat((rotat.X * -1) * (1 + abs(acc.X)) * moveVelocity * abs(grav.Z)), bounds.height))

        )

        // initialize history array

        for i in 0...(averagedPoints.count - 1) {

            if ((averagedPoints[i].x == -1) && (averagedPoints[i].y == -1)) {

                averagedPoints[0].x = mLoc.x

                averagedPoints[0].y = mLoc.y

            }

        }

        // shift history array and insert new position at the end

        for i in 0...(averagedPoints.count - 2) {

            averagedPoints[i].x = averagedPoints[i + 1].x

            averagedPoints[i].y = averagedPoints[i + 1].y

        }

        averagedPoints[averagedPoints.count - 1].x = mLoc.x

        averagedPoints[averagedPoints.count - 1].y = mLoc.y

        // compute averaged coordinates

        var averagedPoint = CGPoint(x: 0, y: 0)

        for i in 0...(averagedPoints.count - 1) {

            averagedPoint.x += averagedPoints[i].x

            averagedPoint.y += averagedPoints[i].y

        }

        averagedPoint.x /= CGFloat(averagedPoints.count)

        averagedPoint.y /= CGFloat(averagedPoints.count)

        //print ("Updated mouse pos to \(averagedPoint.x),\(averagedPoint.y)")

        let mouseEvent = CGEvent(mouseEventSource: nil, mouseType: .mouseMoved, mouseCursorPosition: averagedPoint, mouseButton: .left)

        mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)

        if (isDragging) {

             let mouseEvent = CGEvent(mouseEventSource: nil, mouseType: .leftMouseDragged, mouseCursorPosition: averagedPoint, mouseButton: .left)

            mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)

        }

    }

    func resetPointerPosition(moveMouse: Bool) {

        let aIndex = UserDefaults.standard.integer(forKey: "activeScreen")

        let frame = NSScreen.screens[aIndex].frame

        let x = frame.midX

        let y = frame.midY - frame.origin.y

        NSLog("x: %f, y: %f, origin x: %f, y: %f, mid x: %f, y: %f", x,y,frame.origin.x, frame.origin.y, frame.midX, frame.midY)

        mLoc = (x: x, y: y)

        if moveMouse {

            let point = CGPoint(x: x, y: y)

            let mouseEvent = CGEvent(mouseEventSource: nil, mouseType: .mouseMoved, mouseCursorPosition: point, mouseButton: .left)

            mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)

        }

    }

    //MARK: - Mouse actions

    var rollGate = 0

    func scrollWithRoll(roll: Double, velocity: Double) {

        rollGate += 1

        if rollGate == 10 {

            let amount = (Int32) (roll * velocity * -50)

            let mouseEvent = CGEvent(scrollWheelEvent2Source: nil, units: .pixel, wheelCount: 1, wheel1: amount, wheel2: 0, wheel3: 0)

            mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)

            rollGate = 0

        }

    }

    func clickButton(isRightButton: Bool, isButtonDown: Bool) {

        let mousePos = CGPoint(x: mLoc.x, y: mLoc.y)

        let mouseEventType: CGEventType

        let mouseButton: CGMouseButton

        if (isRightButton) {

            mouseButton = .right

            mouseEventType = (isButtonDown ? .rightMouseDown : .rightMouseUp)

            isDragging = false

        } else {

            mouseButton = .left

            mouseEventType = (isButtonDown ? .leftMouseDown : .leftMouseUp)

            isDragging = isButtonDown

        }

        let mouseEvent = CGEvent(mouseEventSource: nil, mouseType: mouseEventType, mouseCursorPosition: mousePos, mouseButton: mouseButton)

        mouseEvent?.post(tap: CGEventTapLocation.cghidEventTap)

    }

    //MARK: - AppleScript launcher

    func runAppleScript(_ scriptText: String) {

        let script = NSAppleScript(source: scriptText)!

        var error: NSDictionary? = nil

        let resultMaybe = script.executeAndReturnError(&error) as NSAppleEventDescriptor?

        if (resultMaybe == nil)

        {

            let myStringDict = error as? [String:AnyObject]

            if (myStringDict?["NSAppleScriptErrorNumber"] as! Int16 == 1002) {

                let alert = NSAlert()

                alert.addButton(withTitle: "OK".localized)

                alert.messageText = "system_message".localized + "\n" + (myStringDict?["NSAppleScriptErrorBriefMessage"] as! String)

                alert.informativeText = "allow_perms".localized

                alert.alertStyle = .warning

                alert.runModal()

            }

            print (error as Any)

            return

        }

    }

}