gistfile1.txt

'''
WM_PEN: Support of WM_PEN message (Window platform)
'''

__all__ = ['WM_PenProvider']

import os
from ctypes import *
from ..provider import TouchProvider
from ..factory import TouchFactory
from ..touch import Touch
from ...utils import curry

if 'PYMT_DOC' not in os.environ:
    from ...base import getWindow


WNDPROC = WINFUNCTYPE(c_long, c_int, c_int, c_int, c_int)
GWL_WNDPROC            = -4
PEN_OR_TOUCH_SIGNATURE = 0xFF515700
PEN_OR_TOUCH_MASK      = 0xFFFFFF00
PEN_EVENT_TOUCH_MASK   = 0x80
WM_MOUSEMOVE           = 512
WM_LBUTTONDOWN         = 513
WM_LBUTTONUP           = 514



class WM_PenProvider(TouchProvider):

    def is_pen_message(self, msg):
        info = windll.user32.GetMessageExtraInfo()
        if (info & PEN_OR_TOUCH_MASK) == PEN_OR_TOUCH_SIGNATURE: #its a touch or a pen
            if not info & PEN_EVENT_TOUCH_MASK:
                return True


    def pen_handler(self, msg, wParam, lParam):
        x = c_int16(lParam & 0xffff).value
        y = c_int16(lParam >> 16).value
        
        if msg == WM_LBUTTONDOWN:
            self.pen_events.append(('down', x, y))
            self.pen_status = True
            
        if msg == WM_MOUSEMOVE and self.pen_status:
            self.pen_events.append(('move', x, y))
            
        if msg == WM_LBUTTONUP:
            self.pen_events.append(('up', x, y))
            self.pen_status = False
            
        return 1


    def pen_wndProc( self, hwnd, msg, wParam, lParam ):
        if self.is_pen_message(msg):
            return self.pen_handler(msg, wParam, lParam)
        
        return windll.user32.CallWindowProcW( self.old_windProc, hwnd, msg, wParam, lParam)



    def start(self):
        self.uid = 0
        self.pen = None
        self.pen_status = None
        self.pen_events = []
        
        self.hwnd = windll.user32.GetActiveWindow()

        #inject our own wndProc to handle messages before window manager does
        self.new_windProc = WNDPROC(self.pen_wndProc)
        self.old_windProc = windll.user32.SetWindowLongW(
            self.hwnd,
            GWL_WNDPROC,
            self.new_windProc
        )
        

    def update(self, dispatch_fn):
       
        #dispatch pen events
        while len(self.pen_events):
            type,x,y = self.pen_events.pop(0)
            
            print type,x,y
            if  type == 'down':
                self.uid += 1
                self.pen = WM_Pen(self.device,self.uid, [x,y])
                dispatch_fn(type, self.pen )
                
            if  type == 'move':
                self.pen.move([x,y])
                dispatch_fn(type, self.pen )
                
            if type == 'up':
                dispatch_fn(type, self.pen)
                self.pen = None
                

    def stop(self):
        self.new_windProc = windll.user32.SetWindowLongW(
            self.hwnd,
            GWL_WNDPROC,
            self.old_windProc
        )




class WM_Pen(Touch):

    def depack(self, args):
        self.sx, self.sy = args[0], args[1]
        super(WM_Pen, self).depack(args)

    def __str__(self):
        return "Pen, id:%d, pos:(%f,%f, device:%s )" % (self.id, self.sx, self.sy, self.device)

TouchFactory.register('wm_pen', WM_PenProvider)