source: neercs/trunk/src/screens.c @ 2467

Last change on this file since 2467 was 2467, checked in by jylam, 6 years ago
  • Re-indented properly (not touched to grab.c, although it'd be quite fun)
File size: 10.0 KB
Line 
1/*
2 *  neercs        console-based window manager
3 *  Copyright (c) 2006 Sam Hocevar <sam@zoy.org>
4 *                2008 Jean-Yves Lamoureux <jylam@lnxscene.org>
5 *                All Rights Reserved
6 *
7 *  $Id: main.c 2401 2008-06-15 12:50:12Z jylam $
8 *
9 *  This program is free software. It comes without any warranty, to
10 *  the extent permitted by applicable law. You can redistribute it
11 *  and/or modify it under the terms of the Do What The Fuck You Want
12 *  To Public License, Version 2, as published by Sam Hocevar. See
13 *  http://sam.zoy.org/wtfpl/COPYING for more details.
14 */
15
16#include "config.h"
17
18#include <stdio.h>
19#include <string.h>
20#include <stdlib.h>
21#include <sys/types.h>
22#include <signal.h>
23#include <sys/wait.h>
24#include <errno.h>
25#include <unistd.h>
26
27#include <cucul.h>
28#include <caca.h>
29
30#include "neercs.h"
31
32struct screen* create_screen_grab(int w, int h, int pid)
33{
34    struct screen *s = (struct screen*) malloc(sizeof(struct screen));
35
36    s->cv = cucul_create_canvas(w, h);
37    cucul_set_color_ansi(s->cv, CUCUL_BLACK, CUCUL_BLACK);
38    cucul_clear_canvas(s->cv);
39    s->init = 0;
40
41    s->buf = NULL;
42    s->title = NULL;
43    s->total = 0;
44    s->w = w+1;
45    s->h = h+1;
46    s->bell = 0;
47    s->pid = pid;
48
49    s->fd = create_pty_grab(pid, w, h);
50
51    if(s->fd < 0)
52    {
53        cucul_free_canvas(s->cv);
54        free(s);
55        return NULL;
56    }
57    return s;
58}
59
60struct screen* create_screen(int w, int h, char *command)
61{
62    struct screen *s = (struct screen*) malloc(sizeof(struct screen));
63
64    s->cv = cucul_create_canvas(w, h);
65    cucul_set_color_ansi(s->cv, CUCUL_BLACK, CUCUL_BLACK);
66    cucul_clear_canvas(s->cv);
67    s->init = 0;
68
69    s->buf = NULL;
70    s->title = NULL;
71    s->total = 0;
72    s->w = w+1;
73    s->h = h+1;
74    s->bell = 0;
75    s->visible = 1;
76    s->scroll = 0;
77    s->fd = create_pty(command, w, h, &s->pid);
78
79    s->orig_w = s->w;
80    s->orig_h = s->h;
81    s->orig_x = s->x;
82    s->orig_y = s->y;
83
84
85    if(s->fd < 0)
86    {
87        cucul_free_canvas(s->cv);
88        free(s);
89        return NULL;
90    }
91    return s;
92}
93
94int destroy_screen(struct screen *s)
95{
96    if(s->fd>=0)
97        close(s->fd);
98    if(s->buf)
99        free(s->buf);
100    if(s->title)
101        free(s->title);
102    s->buf = NULL;
103    if(s->cv)
104        cucul_free_canvas(s->cv);
105    s->cv = NULL;
106    free(s);
107    return 1;
108}
109
110int add_screen(struct screen_list *list, struct screen *s)
111{
112    if(list == NULL || s == NULL) return -1;
113
114    else
115    {
116        list->screen = (struct screen**) realloc(list->screen,
117                                                 sizeof(sizeof(struct screen*))
118                                                 * (list->count+1));
119        if(!list->screen)
120            fprintf(stderr, "Can't allocate memory at %s:%d\n", __FUNCTION__, __LINE__);
121        list->screen[list->count] = s;
122        list->count++;
123    }
124
125    return list->count-1;
126}
127
128int remove_screen(struct screen_list *list, int n, int please_kill)
129{
130
131    if(n>=list->count) return -1;
132
133    if(please_kill)
134    {
135        int status = 0;
136        int ret = 0;
137        /* FIXME */
138        close(list->screen[n]->fd);
139        list->screen[n]->fd = -1;
140        kill(list->screen[n]->pid, SIGINT);
141        ret = waitpid(list->screen[n]->pid, &status,
142                      WNOHANG|WUNTRACED|WCONTINUED);
143        if(!ret)
144            kill(list->screen[n]->pid, SIGQUIT);
145        ret = waitpid(list->screen[n]->pid, &status,
146                      WNOHANG|WUNTRACED|WCONTINUED);
147        if(!ret)
148            kill(list->screen[n]->pid, SIGABRT);
149        ret = waitpid(list->screen[n]->pid, &status,
150                      WNOHANG|WUNTRACED|WCONTINUED);
151        if(!ret)
152            kill(list->screen[n]->pid, SIGKILL);
153
154    }
155    destroy_screen(list->screen[n]);
156
157    memmove(&list->screen[n],
158            &list->screen[n+1],
159            sizeof(struct screen*)*(list->count-(n+1)));
160
161    list->screen = (struct screen**) realloc(list->screen,
162                                             sizeof(sizeof(struct screen*))
163                                             * (list->count));
164    if(!list->screen)
165        fprintf(stderr, "Can't allocate memory at %s:%d\n", __FUNCTION__, __LINE__);
166
167
168
169    list->count--;
170    return 1;
171}
172
173
174
175void refresh_screens(cucul_canvas_t *cv,
176                     caca_display_t *dp,
177                     struct screen_list *screen_list)
178{
179    int i;
180
181    if(!screen_list->count) return;
182
183    screen_list->width  = cucul_get_canvas_width(cv);
184    screen_list->height = cucul_get_canvas_height(cv) - (screen_list->mini*6);
185
186    if(!screen_list->dont_update_coords)
187        update_windows_props(cv, screen_list);
188
189    if(screen_list->screen[screen_list->pty]->title)
190        caca_set_display_title(dp, screen_list->screen[screen_list->pty]->title);
191    else
192        caca_set_display_title(dp, PACKAGE_STRING);
193
194    cucul_set_color_ansi(cv, CUCUL_DEFAULT, CUCUL_DEFAULT);
195    cucul_clear_canvas(cv);
196    cucul_set_color_ansi(cv, CUCUL_LIGHTRED, CUCUL_BLACK);
197
198    for(i = screen_list->count - 1; i >=0; i--)
199    {
200        if(i!=screen_list->pty && screen_list->screen[i]->visible)
201        {
202            cucul_blit(cv,
203                       screen_list->screen[i]->x,
204                       screen_list->screen[i]->y,
205                       screen_list->screen[i]->cv, NULL);
206
207            cucul_draw_cp437_box(cv,
208                                 screen_list->screen[i]->x - 1,
209                                 screen_list->screen[i]->y - 1,
210                                 screen_list->screen[i]->w + 2,
211                                 screen_list->screen[i]->h + 2);
212            if(screen_list->screen[i]->title)
213                cucul_printf(cv,
214                             screen_list->screen[i]->x,
215                             screen_list->screen[i]->y - 1,
216                             " %.*s ",
217                             screen_list->screen[i]->w - 3,
218                             screen_list->screen[i]->title);
219        }
220    }
221
222    cucul_blit(cv,
223               screen_list->screen[screen_list->pty]->x,
224               screen_list->screen[screen_list->pty]->y,
225               screen_list->screen[screen_list->pty]->cv, NULL);
226
227    if(screen_list->screen[screen_list->pty]->bell)
228    {
229        cucul_set_color_ansi(cv, CUCUL_RED, CUCUL_BLACK);
230        screen_list->screen[screen_list->pty]->bell = 0;
231        screen_list->in_bell--;
232    }
233    else
234    {
235        cucul_set_color_ansi(cv, CUCUL_LIGHTGREEN, CUCUL_BLACK);
236    }
237
238    cucul_draw_cp437_box(cv,
239                         screen_list->screen[screen_list->pty]->x - 1,
240                         screen_list->screen[screen_list->pty]->y - 1,
241                         screen_list->screen[screen_list->pty]->w + 2,
242                         screen_list->screen[screen_list->pty]->h + 2);
243
244    if(screen_list->screen[screen_list->pty]->title)
245    {
246        cucul_printf(cv,
247                     screen_list->screen[screen_list->pty]->x,
248                     screen_list->screen[screen_list->pty]->y - 1,
249                     " %.*s ",
250                     screen_list->screen[screen_list->pty]->w - 3,
251                     screen_list->screen[screen_list->pty]->title);
252    }
253
254    cucul_gotoxy(cv,
255                 screen_list->screen[screen_list->pty]->x +
256                 cucul_get_cursor_x(screen_list->screen[screen_list->pty]->cv),
257                 screen_list->screen[screen_list->pty]->y +
258                 cucul_get_cursor_y(screen_list->screen[screen_list->pty]->cv));
259
260
261    if(screen_list->mini)
262    {
263        draw_thumbnails(cv, screen_list);
264    }
265    if(screen_list->status)
266    {
267        draw_status(cv, screen_list);
268    }
269    if(screen_list->help)
270    {
271        draw_help(cv, screen_list);
272    }
273
274    caca_refresh_display(dp);
275}
276
277
278int update_screens_contents(struct screen_list* screen_list)
279{
280    int i, refresh = 0;
281    int maxfd = 0;
282    struct timeval tv;
283    fd_set fdset;
284    int ret;
285
286    /* Read data, if any */
287    FD_ZERO(&fdset);
288    for(i = 0; i < screen_list->count; i++)
289    {
290        if(screen_list->screen[i]->fd >= 0)
291            FD_SET(screen_list->screen[i]->fd, &fdset);
292        if(screen_list->screen[i]->fd > maxfd)
293            maxfd = screen_list->screen[i]->fd;
294    }
295    tv.tv_sec = 0;
296    tv.tv_usec = 50000;
297    ret = select(maxfd + 1, &fdset, NULL, NULL, &tv);
298
299    if(ret < 0)
300    {
301        if(errno == EINTR)
302            ; /* We probably got a SIGWINCH, ignore it */
303        else
304        {
305            for(i = 0; i < screen_list->count; i++)
306                if(screen_list->screen[i]->total)
307                    break;
308            if(i == screen_list->count)
309                return 0;
310        }
311    }
312    else if(ret)
313    {
314
315    for(i = 0; i < screen_list->count; i++)
316    {
317        /* FIXME: try a new strategy: read all filedescriptors until
318         * each of them starved at least once. */
319
320        if(screen_list->screen[i]->fd < 0 ||
321           !FD_ISSET(screen_list->screen[i]->fd, &fdset))
322            continue;
323
324        for(;;)
325        {
326            ssize_t nr;
327
328            screen_list->screen[i]->buf =
329                realloc(screen_list->screen[i]->buf,
330                        screen_list->screen[i]->total + 1024);
331            if(!screen_list->screen[i]->buf)
332                fprintf(stderr, "Can't allocate memory at %s:%d\n", __FUNCTION__, __LINE__);
333
334            nr = read(screen_list->screen[i]->fd,
335                      screen_list->screen[i]->buf +
336                      screen_list->screen[i]->total, 1024);
337
338            if(nr > 0)
339            {
340                screen_list->screen[i]->total += nr;
341                continue;
342            }
343
344            if(nr == 0 || errno != EWOULDBLOCK)
345            {
346                remove_screen(screen_list, i, 0);
347                if(i < screen_list->prevpty) screen_list->prevpty--;
348                if(i == screen_list->pty)
349                {
350                    screen_list->pty = screen_list->prevpty;
351                    screen_list->prevpty = 0;
352                }
353                if(i < (screen_list->pty)) (screen_list->pty)--;
354                /* Don't skip the element which is now at position i */
355                i--;
356                refresh = 1;
357            }
358
359            break;
360        }
361    }
362    }
363    return refresh;
364}
365
366
Note: See TracBrowser for help on using the repository browser.