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