Controller.m

//
//  Controller.m
//  MiddleClick
//
//  Created by Alex Galonsky on 11/9/09.
//  Copyright 2009 __MyCompanyName__. All rights reserved.
//

#import "Controller.h"
#import <Cocoa/Cocoa.h>
#import "TrayMenu.h"
#include <math.h>
#include <unistd.h>
#include <CoreFoundation/CoreFoundation.h>
#import <Foundation/Foundation.h> 
#import "WakeObserver.h"

/***************************************************************************
 *
 * Multitouch API
 *
 ***************************************************************************/

typedef struct { float x,y; } mtPoint;
typedef struct { mtPoint pos,vel; } mtReadout;

typedef struct {
    int frame;
    double timestamp;
    int identifier, state, foo3, foo4;
    mtReadout normalized;
    float size;
    int zero1;
    float angle, majorAxis, minorAxis; // ellipsoid
    mtReadout mm;
    int zero2[2];
    float unk2;
} Finger;

typedef void *MTDeviceRef;
typedef int (*MTContactCallbackFunction)(int,Finger*,int,double,int);

MTDeviceRef MTDeviceCreateDefault();
CFMutableArrayRef MTDeviceCreateList(void); 
void MTRegisterContactFrameCallback(MTDeviceRef, MTContactCallbackFunction);
void MTDeviceStart(MTDeviceRef, int); // thanks comex
void MTDeviceStop(MTDeviceRef);


NSDate *touchStartTime;
float middleclickX, middleclickY;
float middleclickX2, middleclickY2;
MTDeviceRef dev;

BOOL needToClick;
BOOL maybeMiddleClick;
BOOL pressed;

@implementation Controller

- (void) start
{
	pressed = NO;
	needToClick = NO;
	NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];	
    [NSApplication sharedApplication];
	
	
	//Get list of all multi touch devices
	NSMutableArray* deviceList = (NSMutableArray*)MTDeviceCreateList(); //grab our device list
	
	
	//Iterate and register callbacks for multitouch devices.
	for(int i = 0; i<[deviceList count]; i++) //iterate available devices
	{
        MTRegisterContactFrameCallback((MTDeviceRef)[deviceList objectAtIndex:i], callback); //assign callback for device
        MTDeviceStart((MTDeviceRef)[deviceList objectAtIndex:i],0); //start sending events
	}
	
	//register a callback to know when osx come back from sleep
	WakeObserver *wo = [[WakeObserver alloc] init];
	[[[NSWorkspace sharedWorkspace] notificationCenter] addObserver: wo selector: @selector(receiveWakeNote:) name: NSWorkspaceDidWakeNotification object: NULL];
	
	
	//add traymenu
    TrayMenu *menu = [[TrayMenu alloc] initWithController:self];
    [NSApp setDelegate:menu];
    [NSApp run];
	
	[pool release];
}

- (BOOL)getClickMode
{
	return needToClick;
}

- (void)setMode:(BOOL)click
{
	needToClick = click;
}

int callback(int device, Finger *data, int nFingers, double timestamp, int frame) {
	NSAutoreleasePool *pool = [[NSAutoreleasePool alloc] init];	
	
	if(needToClick)
	{
		
		if(nFingers == 3)
		{
			if(!pressed)
			{
				NSLog(@"Pressed");
				#if __MAC_OS_X_VERSION_MIN_REQUIRED >= 1060
				  CGEventCreateKeyboardEvent(NULL, (CGKeyCode)55, true);
				#else
				  CGPostKeyboardEvent( (CGCharCode)0, (CGKeyCode)55, true );
				#endif
				pressed = YES;
			}
			
		}
		
		if(nFingers == 0) {
			if(pressed)
			{
				NSLog(@"Released");
				#if __MAC_OS_X_VERSION_MIN_REQUIRED >= 1060
					CGEventCreateKeyboardEvent(NULL, (CGKeyCode)55, false);
				#else
					CGPostKeyboardEvent( (CGCharCode)0, (CGKeyCode)55, false );
				#endif					
				
				pressed = NO;
			}
		}
	}
	else 
	{
		if (nFingers==0){
			touchStartTime = NULL;
			if(middleclickX+middleclickY) {
				float delta = ABS(middleclickX-middleclickX2)+ABS(middleclickY-middleclickY2); 
				if (delta < 0.4f) {
					// Emulate a middle click
					
					// get the current pointer location
					CGEventRef ourEvent = CGEventCreate(NULL);
					CGPoint ourLoc = CGEventGetLocation(ourEvent);
					
					/*
					 // CMD+Click code
					 CGPostKeyboardEvent( (CGCharCode)0, (CGKeyCode)55, true );
					 CGPostMouseEvent( ourLoc, 1, 1, 1);
					 CGPostMouseEvent( ourLoc, 1, 1, 0);
					 CGPostKeyboardEvent( (CGCharCode)0, (CGKeyCode)55, false );
					 */
					
					// Real middle click
					#if __MAC_OS_X_VERSION_MIN_REQUIRED >= 1060
						CGEventPost (kCGHIDEventTap, CGEventCreateMouseEvent (NULL,kCGEventOtherMouseDown,ourLoc,kCGMouseButtonCenter));
						CGEventPost (kCGHIDEventTap, CGEventCreateMouseEvent (NULL,kCGEventOtherMouseUp,ourLoc,kCGMouseButtonCenter));
					#else
						CGPostMouseEvent( ourLoc, 1, 3, 0, 0, 1);
						CGPostMouseEvent( ourLoc, 1, 3, 0, 0, 0);
					#endif
					
				}
			}
			
		} else if (nFingers>0 && touchStartTime == NULL){		
			NSDate *now = [[NSDate alloc] init];
			touchStartTime = [now retain];
			[now release];
			
			maybeMiddleClick = YES;
			middleclickX = 0.0f;
			middleclickY = 0.0f;
		} else {
			if (maybeMiddleClick==YES){
				NSTimeInterval elapsedTime = -[touchStartTime timeIntervalSinceNow];  
				if (elapsedTime > 0.5f)
					maybeMiddleClick = NO;
			}
		}
		
		if (nFingers>3) {
			maybeMiddleClick = NO;
			middleclickX = 0.0f;
			middleclickY = 0.0f;
		}
		
		if (nFingers==3) {
			Finger *f1 = &data[0];
			Finger *f2 = &data[1];
			Finger *f3 = &data[2];
			
			if (maybeMiddleClick==YES) {
				middleclickX = (f1->normalized.pos.x+f2->normalized.pos.x+f3->normalized.pos.x);
				middleclickY = (f1->normalized.pos.y+f2->normalized.pos.y+f3->normalized.pos.y);
				middleclickX2 = middleclickX;
				middleclickY2 = middleclickY;
				maybeMiddleClick=NO;
			} else {
				middleclickX2 = (f1->normalized.pos.x+f2->normalized.pos.x+f3->normalized.pos.x);
				middleclickY2 = (f1->normalized.pos.y+f2->normalized.pos.y+f3->normalized.pos.y);
			}
		}
	}

	
	[pool release];
	return 0;
}

@end