##// END OF EJS Templates
Fix formatting.
Eli Rykoff -
Show More
@@ -1,143 +1,156
1 1 """Inputhook for OS X
2 2
3 3 Calls NSApp / CoreFoundation APIs via ctypes.
4 4 """
5 5
6 6 # obj-c boilerplate from appnope, used under BSD 2-clause
7 7
8 8 import ctypes
9 9 import ctypes.util
10 10 from threading import Event
11 11
12 12 objc = ctypes.cdll.LoadLibrary(ctypes.util.find_library('objc'))
13 13
14 14 void_p = ctypes.c_void_p
15 15
16 16 objc.objc_getClass.restype = void_p
17 17 objc.sel_registerName.restype = void_p
18 18 objc.objc_msgSend.restype = void_p
19 19 objc.objc_msgSend.argtypes = [void_p, void_p]
20 20
21 21 msg = objc.objc_msgSend
22 22
23 23 def _utf8(s):
24 24 """ensure utf8 bytes"""
25 25 if not isinstance(s, bytes):
26 26 s = s.encode('utf8')
27 27 return s
28 28
29 29 def n(name):
30 30 """create a selector name (for ObjC methods)"""
31 31 return objc.sel_registerName(_utf8(name))
32 32
33 33 def C(classname):
34 34 """get an ObjC Class by name"""
35 35 return objc.objc_getClass(_utf8(classname))
36 36
37 37 # end obj-c boilerplate from appnope
38 38
39 39 # CoreFoundation C-API calls we will use:
40 40 CoreFoundation = ctypes.cdll.LoadLibrary(ctypes.util.find_library('CoreFoundation'))
41 41
42 42 CFFileDescriptorCreate = CoreFoundation.CFFileDescriptorCreate
43 43 CFFileDescriptorCreate.restype = void_p
44 44 CFFileDescriptorCreate.argtypes = [void_p, ctypes.c_int, ctypes.c_bool, void_p]
45 45
46 46 CFFileDescriptorGetNativeDescriptor = CoreFoundation.CFFileDescriptorGetNativeDescriptor
47 47 CFFileDescriptorGetNativeDescriptor.restype = ctypes.c_int
48 48 CFFileDescriptorGetNativeDescriptor.argtypes = [void_p]
49 49
50 50 CFFileDescriptorEnableCallBacks = CoreFoundation.CFFileDescriptorEnableCallBacks
51 51 CFFileDescriptorEnableCallBacks.restype = None
52 52 CFFileDescriptorEnableCallBacks.argtypes = [void_p, ctypes.c_ulong]
53 53
54 54 CFFileDescriptorCreateRunLoopSource = CoreFoundation.CFFileDescriptorCreateRunLoopSource
55 55 CFFileDescriptorCreateRunLoopSource.restype = void_p
56 56 CFFileDescriptorCreateRunLoopSource.argtypes = [void_p, void_p, void_p]
57 57
58 58 CFRunLoopGetCurrent = CoreFoundation.CFRunLoopGetCurrent
59 59 CFRunLoopGetCurrent.restype = void_p
60 60
61 61 CFRunLoopAddSource = CoreFoundation.CFRunLoopAddSource
62 62 CFRunLoopAddSource.restype = None
63 63 CFRunLoopAddSource.argtypes = [void_p, void_p, void_p]
64 64
65 65 CFRelease = CoreFoundation.CFRelease
66 66 CFRelease.restype = None
67 67 CFRelease.argtypes = [void_p]
68 68
69 69 CFFileDescriptorInvalidate = CoreFoundation.CFFileDescriptorInvalidate
70 70 CFFileDescriptorInvalidate.restype = None
71 71 CFFileDescriptorInvalidate.argtypes = [void_p]
72 72
73 73 # From CFFileDescriptor.h
74 74 kCFFileDescriptorReadCallBack = 1
75 75 kCFRunLoopCommonModes = void_p.in_dll(CoreFoundation, 'kCFRunLoopCommonModes')
76 76
77 77
78 78 def _NSApp():
79 79 """Return the global NSApplication instance (NSApp)"""
80 80 objc.objc_msgSend.argtypes = [void_p, void_p]
81 81 return msg(C('NSApplication'), n('sharedApplication'))
82 82
83 83
84 84 def _wake(NSApp):
85 85 """Wake the Application"""
86 objc.objc_msgSend.argtypes = [void_p, void_p, void_p, void_p, void_p, void_p,
87 void_p, void_p, void_p, void_p, void_p]
88 event = msg(C('NSEvent'),
89 n('otherEventWithType:location:modifierFlags:'
90 'timestamp:windowNumber:context:subtype:data1:data2:'),
86 objc.objc_msgSend.argtypes = [
87 void_p,
88 void_p,
89 void_p,
90 void_p,
91 void_p,
92 void_p,
93 void_p,
94 void_p,
95 void_p,
96 void_p,
97 void_p]
98 event = msg(
99 C("NSEvent"),
100 n(
101 "otherEventWithType:location:modifierFlags:"
102 "timestamp:windowNumber:context:subtype:data1:data2:"
103 ),
91 104 15, # Type
92 105 0, # location
93 106 0, # flags
94 107 0, # timestamp
95 108 0, # window
96 109 None, # context
97 110 0, # subtype
98 111 0, # data1
99 112 0, # data2
100 113 )
101 114 objc.objc_msgSend.argtypes = [void_p, void_p, void_p, void_p]
102 115 msg(NSApp, n('postEvent:atStart:'), void_p(event), True)
103 116
104 117
105 118 _triggered = Event()
106 119
107 120 def _input_callback(fdref, flags, info):
108 121 """Callback to fire when there's input to be read"""
109 122 _triggered.set()
110 123 CFFileDescriptorInvalidate(fdref)
111 124 CFRelease(fdref)
112 125 NSApp = _NSApp()
113 126 objc.objc_msgSend.argtypes = [void_p, void_p, void_p]
114 127 msg(NSApp, n('stop:'), NSApp)
115 128 _wake(NSApp)
116 129
117 130 _c_callback_func_type = ctypes.CFUNCTYPE(None, void_p, void_p, void_p)
118 131 _c_input_callback = _c_callback_func_type(_input_callback)
119 132
120 133
121 134 def _stop_on_read(fd):
122 135 """Register callback to stop eventloop when there's data on fd"""
123 136 _triggered.clear()
124 137 fdref = CFFileDescriptorCreate(None, fd, False, _c_input_callback, None)
125 138 CFFileDescriptorEnableCallBacks(fdref, kCFFileDescriptorReadCallBack)
126 139 source = CFFileDescriptorCreateRunLoopSource(None, fdref, 0)
127 140 loop = CFRunLoopGetCurrent()
128 141 CFRunLoopAddSource(loop, source, kCFRunLoopCommonModes)
129 142 CFRelease(source)
130 143
131 144
132 145 def inputhook(context):
133 146 """Inputhook for Cocoa (NSApp)"""
134 147 NSApp = _NSApp()
135 148 _stop_on_read(context.fileno())
136 149 objc.objc_msgSend.argtypes = [void_p, void_p]
137 150 msg(NSApp, n('run'))
138 151 if not _triggered.is_set():
139 152 # app closed without firing callback,
140 153 # probably due to last window being closed.
141 154 # Run the loop manually in this case,
142 155 # since there may be events still to process (#9734)
143 156 CoreFoundation.CFRunLoopRun()
General Comments 0
You need to be logged in to leave comments. Login now