7aa112b9518fc253d3ee5a8faad6ccbe841e329d
[winnie] / src / wm.cc
1 #include <algorithm>
2 #include <stdexcept>
3 #include <stdio.h>      // TODO
4
5 #include "gfx.h"
6 #include "wm.h"
7 #include "window.h"
8 #include "mouse.h"
9 #include "mouse_cursor.h"
10
11 WindowManager *wm;
12 static WindowManager wminst;
13
14 static void display(Window *win);
15 static void mouse(Window *win, int bn, bool pressed);
16 static void motion(Window *win, int x, int y);
17
18 void WindowManager::create_frame(Window *win)
19 {
20         Window *frame = new Window;
21         Window *parent = win->get_parent();
22
23         frame->set_display_callback(display);
24         frame->set_mouse_button_callback(mouse);
25         frame->set_mouse_motion_callback(motion);
26
27         frame->add_child(win);
28         frame->set_focusable(false);
29
30         windows.push_back(frame);
31
32         Rect win_rect = win->get_rect();
33         frame->move(win_rect.x - frame_thickness, 
34                         win_rect.y - frame_thickness - titlebar_thickness);
35         frame->resize(win_rect.width + frame_thickness * 2, 
36                         win_rect.height + frame_thickness * 2 + titlebar_thickness);
37
38         parent->add_child(frame);
39 }
40
41 void WindowManager::destroy_frame(Window *win)
42 {
43         Window *frame = win->parent;
44         if(!frame) {
45                 return;
46         }
47
48         std::list<Window*>::iterator it;
49         it = std::find(windows.begin(), windows.end(), frame);
50         if(it != windows.end()) {
51                 root_win->add_child(win);
52                 windows.erase(it);
53                 delete frame;
54         }
55 }
56
57 WindowManager::WindowManager()
58 {
59         if(!wm) {
60                 wm = this;
61         } else {
62                 throw std::runtime_error("Trying to create a second instance of WindowManager!\n");
63         }
64
65         root_win = new Window;
66         root_win->resize(get_screen_size().width, get_screen_size().height);
67         root_win->move(0, 0);
68         root_win->set_managed(false);
69
70         focused_win = 0;
71
72         bg_color[0] = 210;
73         bg_color[1] = 106;
74         bg_color[2] = 106;
75
76         frame_thickness = 2;
77         titlebar_thickness = 4;
78
79         frame_fcolor[0] = frame_fcolor[1] = frame_fcolor[2] = 142;
80         frame_ucolor[0] = frame_ucolor[1] = frame_ucolor[2] = 210;
81
82         mouse_cursor.set_image(mouse_cursor_width, mouse_cursor_height);
83         unsigned char *pixels = mouse_cursor.get_image();
84
85         for(int i=0; i<mouse_cursor_height; i++) {
86                 for(int j=0; j<mouse_cursor_width; j++) {
87                         int val = mouse_cursor_bw[i * mouse_cursor_width + j];
88                         *pixels++ = val;
89                         *pixels++ = val;
90                         *pixels++ = val;
91                         *pixels++ = 255;
92                 }
93         }
94 }
95
96 WindowManager::~WindowManager()
97 {
98         delete root_win;
99 }
100
101 void WindowManager::invalidate_region(const Rect &rect)
102 {
103         dirty_rects.push_back(rect);
104 }
105
106 void WindowManager::process_windows()
107 {
108         if(dirty_rects.empty()) {
109                 return;
110         }
111
112         std::list<Rect>::iterator drit = dirty_rects.begin();
113         Rect uni = *drit++;
114         while(drit != dirty_rects.end()) {
115                 uni = rect_union(uni, *drit++);
116         }
117         dirty_rects.clear();
118
119         fill_rect(uni, bg_color[0], bg_color[1], bg_color[2]);
120
121         root_win->draw_children(uni);
122
123         // draw mouse cursor
124         int mouse_x, mouse_y;
125         get_pointer_pos(&mouse_x, &mouse_y);
126
127         blit_key(mouse_cursor.get_image(), mouse_cursor.get_rect(),
128                         get_framebuffer(), get_screen_size(), mouse_x, mouse_y,
129                         0, 0, 0);
130
131         Rect mouse_rect = {mouse_x, mouse_y, mouse_cursor.get_width(), mouse_cursor.get_height()};
132         invalidate_region(mouse_rect);
133
134         gfx_update();
135 }
136
137 void WindowManager::add_window(Window *win)
138 {
139         if(!win || win == root_win) {
140                 return;
141         }
142
143         root_win->add_child(win);
144
145         if(windows.empty()) {
146                 focused_win = win;
147         }
148
149         if(win->get_managed()) {
150                 create_frame(win);
151         }
152
153         windows.push_back(win);
154 }
155
156 void WindowManager::remove_window(Window *win)
157 {
158         std::list<Window*>::iterator it;
159         it = std::find(windows.begin(), windows.end(), win);
160
161         if(it != windows.end()) {
162                 windows.erase(it);
163         }
164 }
165
166 void WindowManager::set_focused_window(Window *win)
167 {
168         if(win == focused_win) {
169                 return;
170         }
171
172         if(focused_win) {
173                 // invalidate the frame (if any)
174                 Window *parent = focused_win->get_parent();
175                 if(parent && parent != root_win) {
176                         parent->invalidate();
177                 }
178         }
179
180         if(!win) {
181                 focused_win = 0;
182                 return;
183         }
184
185         if(win->get_focusable()) {
186                 focused_win = win;
187                 return;
188         }
189
190         Window **children = win->get_children();
191         for(int i=0; i<win->get_children_count(); i++) {
192                 if(children[0]->get_focusable()) {
193                         set_focused_window(children[0]);
194                         return;
195                 }
196         }
197
198         focused_win = 0;
199 }
200
201 const Window *WindowManager::get_focused_window() const
202 {
203         return focused_win;
204 }
205
206 Window *WindowManager::get_focused_window()
207 {
208         return focused_win;
209 }
210
211 Window *WindowManager::get_window_at_pos(int pointer_x, int pointer_y)
212 {
213         std::list<Window*>::reverse_iterator rit = windows.rbegin();
214         while(rit != windows.rend()) {
215                 Window *w = *rit++;
216                 Window *parent = w->get_parent();
217
218                 if(parent == root_win && w->contains_point(pointer_x, pointer_y)) {
219                         return w;
220                 }
221         }
222
223         return 0;
224 }
225
226 static void display(Window *win)
227 {
228         fill_rect(win->get_rect(), 255, 211, 5);
229 }
230
231 static int prev_x = -1, prev_y;
232
233 static void mouse(Window *win, int bn, bool pressed)
234 {
235         if(bn == 0) {
236                 if(pressed) {
237                         get_pointer_pos(&prev_x, &prev_y);
238                 } else {
239                         prev_x = -1;
240                 }
241         }
242 }
243
244 static void motion(Window *win, int x, int y)
245 {
246         int left_bn = get_button(0);
247         if(left_bn && prev_x != -1) {
248                 int dx = x - prev_x;
249                 int dy = y - prev_y;
250                 prev_x = x - dx;
251                 prev_y = y - dy;
252
253                 Rect rect = win->get_rect();
254                 win->move(rect.x + dx, rect.y + dy);
255         }
256 }