tuiototouch.py

#! /usr/bin/python
#
#	tuiototouch, a bridge between TUIO and Linux input for multitouch
#
#	The copyright owners for the contents of this file are:
#		Ecole Nationale de l'Aviation Civile, France (2010-2011)
#
#	main and only file
#
#	Contributors:
#		Simon Charvet <charvet@lii-enac.fr>
#
#
#	This program is provided to you as free software;
#	you can redistribute it	and/or modify it under the terms of the
#	GNU General Public License as published by the Free Software
#	Foundation; either version 2 of the License, or (at your option)
#	any later version.

import getopt
import tuio
import uinput
import time
xRes = 1000
xOffset = 0
xCombined = 1000

yRes = 1000
yOffset = 0
yCombined = 1000

class Device(object):

    def __init__(self, capabilities=()):
	self.objects=dict()
        self.empty=1
        capabilities += (uinput.BTN_TOUCH,
                         uinput.ABS_MT_POSITION_X + (0, xCombined, 0, 0),
                         uinput.ABS_MT_POSITION_Y + (0, yCombined, 0, 0),
                         uinput.ABS_MT_TRACKING_ID + (0, 10, 0, 0))
        self.device = uinput.Device(capabilities, name="TUIO-multitouch")

    def update(self,list_obj,list_cursors):
        list=set(list_obj).union(set(list_cursors))
        self.objects.clear()
        for obj in list:
            self.objects[obj.sessionid]=obj

    def display(self):
        if (len(self.objects)==0) & self.empty:
            self.emit(uinput.BTN_TOUCH, 0)
            self.empty=0
        i=0
        for key in self.objects:
            if not self.empty:
                self.emit(uinput.BTN_TOUCH, 1)
            self.empty=1
            self.treatment(self.objects[key])
            i+=1
            if i==len(self.objects):
                self.emit((0, 0), 0, syn=False)
                self.empty=1

    def treatment(self, obj):
        self.emit(uinput.ABS_MT_TRACKING_ID, obj.sessionid, syn=False)
        self.emit(uinput.ABS_MT_POSITION_X, xOffset + (obj.xpos*xRes), syn=False)
        self.emit(uinput.ABS_MT_POSITION_Y, yOffset + (obj.ypos*yRes), syn=False)
        self.emit((0, 2), 0, syn=False)

    def emit(self, event, value, syn=True):
        evtype, evcode = event
        print "type :%d code : %d value : %d" % (evtype, evcode, value)
        self.device.emit(event, int(value), syn)


class DeviceWME(Device):

    def __init__(self):
        capabilities = (uinput.ABS_X + (0, xCombined, 0, 0),
                        uinput.ABS_Y + (0, yCombined, 0, 0))
        Device.__init__(self, capabilities)
        self.x_mouse=0
        self.y_mouse=0

    def display(self):
        if (len(self.objects)==0) & self.empty:
            self.emit(uinput.BTN_TOUCH, 0)
            self.empty=0
        i=0
        for key in self.objects:
            if not self.empty:
                self.emit(uinput.BTN_TOUCH, 1)
            self.empty=1
            obj = self.objects[key]
            if i==0:
                modified=0
                if self.x_mouse!=xOffset + (obj.xpos*xRes):
                    modified=1
                    self.x_mouse= xOffset + (obj.xpos*xRes)
                    self.emit(uinput.ABS_X, self.x_mouse, syn=False)
                if self.y_mouse!=yOffset + (obj.ypos*yRes):
                    modified=1
                    self.y_mouse= yOffset + (obj.ypos*yRes)
                    self.emit(uinput.ABS_Y, self.y_mouse, syn=False)
                if modified:
                    self.emit((0, 0), 0, syn=False)
            self.treatment(obj)
            i+=1
            if i==len(self.objects):
                self.emit((0, 0), 0, syn=False)
                self.empty=1


if __name__ == "__main__":
    import sys
    host = "127.0.0.1"
    port = 3333
    noWME = False

    options, remainder = getopt.getopt(sys.argv[1:], "", ['no-mouse-emu', 'host=', 'port=', 'xoffset=', 'yoffset=', 'xsize=', 'ysize=', 'xcombined=', 'ycombined='])

    for opt, arg in options:   
	if opt in ('--no-mouse-emu'):
		noWME = True
       	if opt in ('--host'):
		host = arg
	if opt in ('--port'):
		port = int(arg)
	if opt in ('--xoffset'):
		xOffset = int(arg)
	if opt in ('--yoffset'):
		yOffset = int(arg)
	if opt in ('--xsize'):
		xRes = int(arg)
	if opt in ('--ysize'):
		yRes = int(arg)
	if opt in ('--xcombined'):
		xCombined = int(arg)
	if opt in ('--ycombined'):
		yCombined = int(arg)

    device = DeviceWME()

    if noWME:
	device = Device()

    tracking=tuio.Tracking(host, port)    
    try:
        while 1:
            tracking.update()
            objects=tracking.objects()
            cursors=tracking.cursors()
            device.update(objects,cursors)
            device.display()
            time.sleep(0.01)
    except KeyboardInterrupt:
        tracking.stop()