python-whiteboard 1.0+git20140207-0.1 / usr / share / python-whiteboard / cursor.py

# -*- coding: utf-8 -*-

import Xlib.display
import Xlib.ext.xtest
import time

import PyQt4.Qt as qt

from configuration import Configuration


def clock():
	return int(time.time()*1000)


class Filter:
	def __init__(self):
		self.data = []
		conf = Configuration()
		self.limit = int(conf.getValueStr("smoothing"))
	
	def update(self,p):
		self.data.append(p)
		if len(self.data) > self.limit:
			self.data.pop(0)
			
		n = len(self.data)
		res = [0,0]
		for p in self.data:
			res[0] += p[0]
			res[1] += p[1]
		res[0] /= n
		res[1] /= n
		return res

class Click:
	UP_TIMEOUT = 100
	
	def __init__(self,cursor):
		self.initialTime = clock()
		self.cursor = cursor
		self.cursor.mouse_up()
		self.cursor.mouse_down()
	
	# Called when needing to update without IR Data
	def updateWithoutData(self):
		if (clock() - self.initialTime) > Click.UP_TIMEOUT:
			self.cursor.mouse_up()
			return False
		
		return True
	
	# Update event with IR Data
	def updateWithData(self):
		self.initialTime = clock()


class FakeCursor:
	LEFT_BUTTON = 1
	MIDDLE_BUTTON = 2
	RIGHT_BUTTON = 3
	ONLY_MOVE = 4
	RIGHT_CLICK_TIMEOUT = 750
	CLICK_TIMEOUT = 60
	ZONE1, ZONE2, ZONE3, ZONE4 = range(4)
	
	def __init__(self,wii):
		self.display = Xlib.display.Display()
		self.screen = self.display.screen()
		self.wii = wii
		self.click = None
		self.filt = None
		self.clickType = FakeCursor.LEFT_BUTTON
		self.zones = {}
		self.mutex = qt.QMutex()
		self.mustFinish = False
		self.thread = None
		self.noClicks = False
		# Start of click - measure timeout for right click
		self.begin = None
		# Keepalive of IR signal
		self.lastdata = 0
		# Starts click (left or right) on timeout or move
		self.run = False
		# Delay to only move, before click
		self.toclick = False
	
	
	def setZone(self,zone,clickType):
		self.zones[zone] = clickType
	
	def setZones(self,actions):
		for z,a in zip((FakeCursor.ZONE1,FakeCursor.ZONE2,FakeCursor.ZONE3,FakeCursor.ZONE4),actions):
			if a == '2':
				self.setZone(z,FakeCursor.RIGHT_BUTTON)
			elif a == '0':
				self.setZone(z,FakeCursor.LEFT_BUTTON)
			elif a == '3':
				self.setZone(z,FakeCursor.MIDDLE_BUTTON)
			elif a == '1':
				self.setZone(z,FakeCursor.ONLY_MOVE)
			else:
				self.setZone(z,FakeCursor.LEFT_BUTTON)
	
	
	def move(self,pos):
		Xlib.ext.xtest.fake_input(self.display, Xlib.X.MotionNotify, x=pos[0], y=pos[1])
		self.display.sync()
	
	
	def mouse_down(self):
		if self.noClicks or self.clickType == FakeCursor.ONLY_MOVE: 
			self.clickType = None
			return
		button = self.clickType
		Xlib.ext.xtest.fake_input(self.display, Xlib.X.ButtonPress, button)
		self.display.sync()
	
	
	def mouse_up(self):
		if self.clickType is None:
			return
		button = self.clickType
		Xlib.ext.xtest.fake_input(self.display, Xlib.X.ButtonRelease, button)
		self.display.sync()
	
	
	# Returns True if point is within screen limits
	def checkLimits(self,pos):
		if pos[1] < 0:
			# Zone 1
			self.clickType = self.zones[FakeCursor.ZONE1]
			return False
		if pos[0] > self.screen['width_in_pixels']:
			# Zone 2
			self.clickType = self.zones[FakeCursor.ZONE2]
			return False
		if pos[1] > self.screen['height_in_pixels']:
			# Zone 3
			self.clickType = self.zones[FakeCursor.ZONE3]
			return False
		if pos[0] < 0:
			# Zone 4
			self.clickType = self.zones[FakeCursor.ZONE4]
			return False
		
		return True
	
	
	# This function fabricates the callback function that is to be
	# passed to the wiimote object. It's called every time that IR data
	# is available. It is necessary to do it this way because the callback
	# has to be aware of the cursor object.
	def makeCallback(self):
		def callback(q):
			self.mutex.lock()
			if self.checkLimits(q):
				if not self.filt:
					self.filt = Filter()
				p = self.filt.update(q)
				
				self.move(p)
				
				if self.run:
					if not self.click:
						self.click = Click(self)
					else:
						self.click.updateWithData()
				else:
					if not self.begin:
						self.begin = clock()
						self.clkP = p
					else:
						self.lastdata = clock()
						# timeout for "only move" flash
						if clock() - self.begin > FakeCursor.CLICK_TIMEOUT:
							self.toclick = True
						
						# timeout for "right click"
						if clock() - self.begin > FakeCursor.RIGHT_CLICK_TIMEOUT:
							self.begin = None
							self.run = True
							self.clickType = FakeCursor.RIGHT_BUTTON
						else:
							dX = p[0]-self.clkP[0]
							dY = p[1]-self.clkP[1]
							# if the cursor moves, start click (drag) immediatly
							if (dX*dX)+(dY*dY) > 16:
								self.begin = None
								self.run = True
								# Returns to the originlal point and clicks there
								self.move(self.clkP)
								self.click = Click(self)

			self.mutex.unlock()
		
		return callback
	
	
	# This function creates the cursor thread. It makes the callback to the
	# thread and enables the wiimote device
	def runThread(self):
		def runFunc():
			while 1:
				qt.QThread.usleep(50)
				self.mutex.lock()
				# A click had occourred: a short burst (noot an impulse) with no movement
				if self.toclick and not self.run and clock() - self.lastdata > Click.UP_TIMEOUT:
					self.begin = None
					self.lastdata = None
					self.toclick = False
					self.run = True
					self.click = Click(self)
					
				if self.click and self.click.updateWithoutData() == False:
					self.click = None
					self.filt = None
					self.begin = None
					self.lastdata = None
					self.run = False
					self.toclick = False
					self.clickType = FakeCursor.LEFT_BUTTON
				if self.mustFinish:
					self.mutex.unlock()
					break
				self.mutex.unlock()
		
		from threads import CreateThreadClass
		
		self.mustFinish = False
		self.wii.disable()
		self.wii.putCallbackIR(self.makeCallback())
		self.wii.enable()
		thread = CreateThreadClass(runFunc)
		self.thread = thread()
		self.thread.start()
	
	
	# Destroys the cursor thread and disables the wiimote device
	def finish(self):
		self.mutex.lock()
		self.mustFinish = True
		self.mutex.unlock()
		self.thread.wait()
		self.wii.disable()
		self.wii.putCallbackIR(None)
		self.thread = None