grab the keyboard during move/resize to make sure the popup doesnt get left onscreen...
[dana/openbox.git] / scripts / focus.py
1 ###########################################################################
2 ###          Functions for helping out with your window focus.          ###
3 ###########################################################################
4
5 ###########################################################################
6 ###         Options that affect the behavior of the focus module.       ###
7 ###                                                                     ###
8 # raise the window also when it is focused                              ###
9 cycle_raise = 1                                                         ###
10 # raise as you cycle in stacked mode                                    ###
11 stacked_cycle_raise = 0                                                 ###
12 # show a pop-up list of windows while cycling                           ###
13 stacked_cycle_popup_list = 1                                            ###
14 # send focus somewhere when nothing is left with the focus, if possible ###
15 fallback = 0                                                            ###
16 ###                                                                     ###
17 ###                                                                     ###
18 # Provides:                                                             ###
19 # def focus_next_stacked(data, forward=1):                              ###
20 #   """Focus the next (or previous, with forward=0) window in a stacked ###
21 #      order."""                                                        ###
22 # def focus_prev_stacked(data):                                         ###
23 #   """Focus the previous window in a stacked order."""                 ###
24 # def focus_next(data, num=1, forward=1):                               ###
25 #   """Focus the next (or previous, with forward=0) window in a linear  ###
26 #      order."""                                                        ###
27 # def focus_prev(data, num=1):                                          ###
28 #   """Focus the previous window in a linear order."""                  ###
29 ###                                                                     ###
30 # All of these functions call be used as callbacks for bindings         ###
31 # directly.                                                             ###
32 ###                                                                     ###
33 ###########################################################################
34
35 import otk
36 import ob
37
38 # maintain a list of clients, stacked in focus order
39 _clients = []
40 # maintaint he current focused window
41 _doing_stacked = 0
42
43 def _new_win(data):
44     global _clients
45     global _doing_stacked
46     global _cyc_w;
47
48     if _doing_stacked:
49         _clients.insert(_clients.index(_cyc_w), data.client.window())
50         _create_popup_list(data)
51         _hilite_popup_list(data)
52     else:
53         if not len(_clients):
54             _clients.append(data.client.window())
55         else:
56             _clients.insert(1, data.client.window()) # insert in 2nd slot
57
58 def _close_win(data):
59     global _clients
60     global _cyc_w;
61     global _doing_stacked
62
63     if not _doing_stacked:
64         # not in the middle of stacked cycling, so who cares
65         _clients.remove(data.client.window())
66     else:
67         # have to fix the cycling if we remove anything
68         win = data.client.window()
69         if _cyc_w == win:
70             _do_stacked_cycle(data, 1) # cycle off the window first, forward
71         _clients.remove(win)
72         _create_popup_list(data)
73
74 def _focused(data):
75     global _clients
76     global _doing_stacked
77     global _cyc_w
78     
79     if data.client:
80         if not _doing_stacked: # only move the window when we're not cycling
81             win = data.client.window()
82             # move it to the top
83             _clients.remove(win)
84             _clients.insert(0, win)
85         else: # if we are cycling, then update our pointer
86             _cyc_w = data.client.window()
87             _hilite_popup_list(data)
88     elif fallback: 
89         # pass around focus
90         desktop = ob.openbox.screen(_cyc_screen).desktop()
91         for w in _clients:
92             client = ob.openbox.findClient(w)
93             if client and (client.desktop() == desktop and \
94                            client.normal() and client.focus()):
95                 break
96         if _doing_stacked:
97             _cyc_w = 0
98             _hilite_popup_list(data)
99
100 _cyc_mask = 0
101 _cyc_key = 0
102 _cyc_w = 0 # last window cycled to
103 _cyc_screen = 0
104
105 def _do_stacked_cycle(data, forward):
106     global _cyc_w
107     global stacked_cycle_raise
108     global _clients
109
110     clients = _clients[:] # make a copy
111
112     if not forward:
113         clients.reverse()
114
115     try:
116         i = clients.index(_cyc_w) + 1
117     except ValueError:
118         i = 1
119     clients = clients[i:] + clients[:i]
120         
121     desktop = ob.openbox.screen(data.screen).desktop()
122     for w in clients:
123         client = ob.openbox.findClient(w)
124         if client and (client.desktop() == desktop and \
125                        client.normal() and client.focus()):
126             if stacked_cycle_raise:
127                 ob.openbox.screen(data.screen).raiseWindow(client)
128             return
129
130 def _focus_stacked_ungrab(data):
131     global _cyc_mask;
132     global _cyc_key;
133     global _doing_stacked;
134
135     if data.action == ob.KeyAction.Release:
136         # have all the modifiers this started with been released?
137         if not _cyc_mask & data.state:
138             _destroy_popup_list()
139             ob.kungrab()
140             ob.mungrab()
141             _doing_stacked = 0;
142             if cycle_raise:
143                 client = ob.openbox.findClient(_cyc_w)
144                 if client:
145                     ob.openbox.screen(data.screen).raiseWindow(client)
146
147 _list_widget = 0
148 _list_labels = []
149 _list_windows = []
150
151 def _hilite_popup_list(data):
152     global _cyc_w, _doing_stacked
153     global _list_widget, _list_labels, _list_windows
154     found = 0
155
156     if not _list_widget and _doing_stacked:
157         _create_popup_list(data)
158     
159     if _list_widget:
160         i = 0
161         for w in _list_windows:
162             if w == _cyc_w:
163                 _list_labels[i].focus()
164                 found = 1
165             else:
166                 _list_labels[i].unfocus()
167             i += 1
168     if not found:
169         _create_popup_list(data)
170
171 def _destroy_popup_list():
172     global _list_widget, _list_labels, _list_windows
173     if _list_widget:
174         _list_windows = []
175         _list_labels = []
176         _list_widget = 0
177     
178 def _create_popup_list(data):
179     global _list_widget, _list_labels, _list_windows, _clients
180
181     if _list_widget:
182         _destroy_popup_list()
183     
184     style = ob.openbox.screen(data.screen).style()
185     _list_widget = otk.Widget(ob.openbox, style,
186                               otk.Widget.Vertical, 0,
187                               style.bevelWidth(), 1)
188     t = style.titlebarFocusBackground()
189     _list_widget.setTexture(t)
190
191     titles = []
192     font = style.labelFont()
193     height = font.height()
194     longest = 0
195     for c in _clients:
196         client = ob.openbox.findClient(c)
197         desktop = ob.openbox.screen(data.screen).desktop()
198         if client and ((client.desktop() == desktop or
199                         client.desktop() == 0xffffffff) and \
200                        client.normal() and (client.canFocus() or
201                                             client.focusNotify())):
202             t = client.title()
203             if len(t) > 50: # limit the length of titles
204                 t = t[:24] + "..." + t[-24:]
205             titles.append(t)
206             _list_windows.append(c)
207             l = font.measureString(t)
208             if l > longest: longest = l
209     if len(titles) > 1:
210         for t in titles:
211             w = otk.FocusLabel(_list_widget)
212             w.fitSize(longest, height)
213             w.setText(t)
214             w.unfocus()
215             _list_labels.append(w)
216         _list_widget.update()
217         area = otk.display.screenInfo(data.screen).rect()
218         _list_widget.move(area.x() + (area.width() -
219                                       _list_widget.width()) / 2,
220                           area.y() + (area.height() -
221                                       _list_widget.height()) / 2)
222         _list_widget.show(1)
223     else:
224         _destroy_popup_list() # nothing (or only 1) to list
225
226 def focus_next_stacked(data, forward=1):
227     """Focus the next (or previous, with forward=0) window in a stacked
228        order."""
229     global _cyc_mask
230     global _cyc_key
231     global _cyc_w
232     global _cyc_screen
233     global _doing_stacked
234
235     if _doing_stacked:
236         if _cyc_key == data.key:
237             _do_stacked_cycle(data,forward)
238     else:
239         _cyc_mask = data.state
240         _cyc_key = data.key
241         _cyc_w = 0
242         _cyc_screen = data.screen
243         _doing_stacked = 1
244
245         global stacked_cycle_popup_list
246         if stacked_cycle_popup_list:
247             _create_popup_list(data)
248
249         ob.kgrab(data.screen, _focus_stacked_ungrab)
250         # the pointer grab causes pointer events during the keyboard grab to
251         # go away, which means we don't get enter notifies when the popup
252         # disappears, screwing up the focus
253         ob.mgrab(data.screen)
254         focus_next_stacked(data, forward) # start with the first press
255
256 def focus_prev_stacked(data):
257     """Focus the previous window in a stacked order."""
258     focus_next_stacked(data, forward=0)
259
260 def focus_next(data, num=1, forward=1):
261     """Focus the next (or previous, with forward=0) window in a linear
262        order."""
263     screen = ob.openbox.screen(data.screen)
264     count = screen.clientCount()
265
266     if not count: return # no clients
267     
268     target = 0
269     if data.client:
270         client_win = data.client.window()
271         found = 0
272         r = range(count)
273         if not forward:
274             r.reverse()
275         for i in r:
276             if found:
277                 target = i
278                 found = 2
279                 break
280             elif screen.client(i).window() == client_win:
281                 found = 1
282         if found == 1: # wraparound
283             if forward: target = 0
284             else: target = count - 1
285
286     t = target
287     curdesk = screen.desktop()
288     while 1:
289         client = screen.client(t)
290         if client.normal() and \
291                (client.desktop() == curdesk or client.desktop() == 0xffffffff)\
292                and client.focus():
293             if cycle_raise:
294                 screen.raiseWindow(client)
295             return
296         if forward:
297             t += num
298             if t >= count: t -= count
299         else:
300             t -= num
301             if t < 0: t += count
302         if t == target: return # nothing to focus
303
304 def focus_prev(data, num=1):
305     """Focus the previous window in a linear order."""
306     focus_next(data, num, forward=0)
307
308
309 ob.ebind(ob.EventAction.NewWindow, _new_win)
310 ob.ebind(ob.EventAction.CloseWindow, _close_win)
311 ob.ebind(ob.EventAction.Focus, _focused)
312
313 print "Loaded focus.py"