source: libcaca/trunk/cucul/import.c @ 884

Last change on this file since 884 was 884, checked in by Jean-Yves Lamoureux, 15 years ago
  • Improved ANSI format detection
  • Property svn:keywords set to Id
File size: 11.3 KB
Line 
1/*
2 *  libcucul      Canvas for ultrafast compositing of Unicode letters
3 *  Copyright (c) 2002-2006 Sam Hocevar <sam@zoy.org>
4 *                All Rights Reserved
5 *
6 *  $Id: import.c 884 2006-04-25 18:23:33Z jylam $
7 *
8 *  This library is free software; you can redistribute it and/or
9 *  modify it under the terms of the Do What The Fuck You Want To
10 *  Public License, Version 2, as published by Sam Hocevar. See
11 *  http://sam.zoy.org/wtfpl/COPYING for more details.
12 */
13
14/*
15 *  This file contains various import functions.
16 */
17
18#include "config.h"
19#include "common.h"
20
21#if !defined(__KERNEL__)
22#   include <stdio.h>
23#   include <stdlib.h>
24#   include <string.h>
25#endif
26
27#include "cucul.h"
28#include "cucul_internals.h"
29
30static cucul_canvas_t *import_caca(void const *, unsigned int);
31static cucul_canvas_t *import_text(void const *, unsigned int);
32static cucul_canvas_t *import_ansi(void const *, unsigned int);
33
34/** \brief Import a buffer into a canvas
35 *
36 *  This function imports a memory area into an internal libcucul canvas.
37 *
38 *  Valid values for \c format are:
39 *
40 *  \li \c "": attempt to autodetect the file format.
41 *
42 *  \li \c "ansi": import ANSI files.
43 *
44 *  \li \c "caca": import native libcaca files.
45 *
46 *  \param data The memory area to be loaded into a canvas.
47 *  \param size The length of the memory area.
48 *  \param format A string describing the input format.
49 *  \return A libcucul canvas, or NULL in case of error.
50 */
51cucul_canvas_t * cucul_import_canvas(void const *data, unsigned int size,
52                                     char const *format)
53{
54    char const *buf = (char const*) data;
55
56    if(size==0 || data==NULL)
57        return NULL;
58
59    if(!strcasecmp("caca", format))
60        return import_caca(data, size);
61    if(!strcasecmp("text", format))
62        return import_text(data, size);
63    if(!strcasecmp("ansi", format))
64        return import_ansi(data, size);
65
66    /* Autodetection */
67    if(!strcasecmp("", format))
68    {
69        unsigned int i=0;
70        /* if 4 first letters are CACA */
71        if(size >= 4 &&
72            buf[0] == 'C' && buf[1] == 'A' && buf[2] == 'C' && buf[3] != 'A')
73            return import_caca(data, size);
74
75        /* If we find ESC[ tuple, we guess it's an ANSI file */
76        while(i<size-1)
77        {
78            if((buf[i] == 0x1b) && (buf[i+1] == '['))
79                return import_ansi(data, size);
80            i++;
81        }
82       
83        /* Otherwise, import it as text */
84        return import_text(data, size);
85    }
86    return NULL;
87}
88
89/** \brief Get available import formats
90 *
91 *  Return a list of available import formats. The list is a NULL-terminated
92 *  array of strings, interleaving a string containing the internal value for
93 *  the import format, to be used with cucul_import_canvas(), and a string
94 *  containing the natural language description for that import format.
95 *
96 *  \return An array of strings.
97 */
98char const * const * cucul_get_import_list(void)
99{
100    static char const * const list[] =
101    {
102        "", "autodetect",
103        "text", "plain text",
104        "caca", "native libcaca format",
105        "ansi", "ANSI coloured text",
106        NULL, NULL
107    };
108
109    return list;
110}
111
112/*
113 * XXX: the following functions are local.
114 */
115
116static cucul_canvas_t *import_caca(void const *data, unsigned int size)
117{
118    cucul_canvas_t *cv;
119    uint8_t const *buf = (uint8_t const *)data;
120    unsigned int width, height, n;
121
122    if(size < 16)
123        return NULL;
124
125    if(buf[0] != 'C' || buf[1] != 'A' || buf[2] != 'C' || buf[3] != 'A')
126        return NULL;
127
128    if(buf[4] != 'C' || buf[5] != 'A' || buf[6] != 'N' || buf[7] != 'V')
129        return NULL;
130
131    width = ((uint32_t)buf[8] << 24) | ((uint32_t)buf[9] << 16)
132           | ((uint32_t)buf[10] << 8) | (uint32_t)buf[11];
133    height = ((uint32_t)buf[12] << 24) | ((uint32_t)buf[13] << 16)
134            | ((uint32_t)buf[14] << 8) | (uint32_t)buf[15];
135
136    if(!width || !height)
137        return NULL;
138
139    if(size != 16 + width * height * 8)
140        return NULL;
141
142    cv = cucul_create_canvas(width, height);
143
144    if(!cv)
145        return NULL;
146
147    for(n = height * width; n--; )
148    {
149        cv->chars[n] = ((uint32_t)buf[16 + 0 + 8 * n] << 24)
150                     | ((uint32_t)buf[16 + 1 + 8 * n] << 16)
151                     | ((uint32_t)buf[16 + 2 + 8 * n] << 8)
152                     | (uint32_t)buf[16 + 3 + 8 * n];
153        cv->attr[n] = ((uint32_t)buf[16 + 4 + 8 * n] << 24)
154                    | ((uint32_t)buf[16 + 5 + 8 * n] << 16)
155                    | ((uint32_t)buf[16 + 6 + 8 * n] << 8)
156                    | (uint32_t)buf[16 + 7 + 8 * n];
157    }
158
159    return cv;
160}
161
162static cucul_canvas_t *import_text(void const *data, unsigned int size)
163{
164    cucul_canvas_t *cv;
165    char const *text = (char const *)data;
166    unsigned int width = 1, height = 1, x = 0, y = 0, i;
167
168    cv = cucul_create_canvas(width, height);
169    cucul_set_color(cv, CUCUL_COLOR_DEFAULT, CUCUL_COLOR_TRANSPARENT);
170
171    for(i = 0; i < size; i++)
172    {
173        unsigned char ch = *text++;
174
175        if(ch == '\r')
176            continue;
177
178        if(ch == '\n')
179        {
180            x = 0;
181            y++;
182            continue;
183        }
184
185        if(x >= width || y >= height)
186        {
187            if(x >= width)
188                width = x + 1;
189
190            if(y >= height)
191                height = y + 1;
192
193            cucul_set_canvas_size(cv, width, height);
194        }
195
196        cucul_putchar(cv, x, y, ch);
197        x++;
198    }
199
200    return cv;
201}
202
203#define IS_ALPHA(x) (x>='A' && x<='z')
204#define END_TUP 0x1337
205unsigned char _get_ansi_command(unsigned char const *buffer, int size);
206int _parse_tuple(unsigned int *ret, unsigned char const *buffer, int size);
207void _manage_modifiers(int i, int *fg, int *bg, int *save_fg, int *save_bg, int *bold);
208
209static cucul_canvas_t *import_ansi(void const *data, unsigned int size)
210{
211    cucul_canvas_t *cv;
212    unsigned char const *buffer = (unsigned char const*)data;
213    unsigned int i, sent_size = 0;
214    unsigned char c;
215    unsigned int count = 0;
216    unsigned int tuple[1024]; /* Should be enough. Will it be ? */
217    int x = 0, y = 0, width = 80, height = 25;
218    int save_x = 0, save_y = 0;
219    unsigned int j, add = 0;
220    int fg, bg, save_fg, save_bg, bold;
221
222    fg = save_fg = CUCUL_COLOR_LIGHTGRAY;
223    bg = save_bg = CUCUL_COLOR_BLACK;
224    bold = 0;
225
226    cv = cucul_create_canvas(width, height);
227
228    for(i = 0; i < size; i++)
229    {
230        if((buffer[i] == 0x1b) && (buffer[i + 1] == '['))  /* ESC code */
231        {
232            i++; // ESC
233            i++; // [
234            sent_size = size - i;
235            c = _get_ansi_command(&buffer[i], sent_size);
236            add = _parse_tuple(tuple, &buffer[i], sent_size);
237            count = 0;
238
239            while(tuple[count] != END_TUP)
240                count++;  /* Gruik */
241
242            switch(c)
243            {
244            case 'f':
245            case 'H':
246                if(tuple[0] != END_TUP)
247                {
248                    y = tuple[0] - 1;
249                    x = tuple[1] == END_TUP ? 0 : tuple[1] - 1;
250                }
251                else
252                {
253                    y = 0;
254                    x = 0;
255                }
256                break;
257            case 'A':
258                y -= tuple[0] == END_TUP ? 1 : tuple[0];
259                if(y < 0)
260                    y = 0;
261                break;
262            case 'B':
263                y += tuple[0] == END_TUP ? 1 : tuple[0];
264                break;
265            case 'C':
266                x += tuple[0] == END_TUP ? 1 : tuple[0];
267                break;
268            case 'D':
269                x -= tuple[0] == END_TUP ? 1 : tuple[0];
270                if(x < 0)
271                    x = 0;
272                break;
273            case 's':
274                save_x = x;
275                save_y = y;
276                break;
277            case 'u':
278                x = save_x;
279                y = save_y;
280                break;
281            case 'J':
282                if(tuple[0] == 2)
283                {
284                    x = 0;
285                    y = 0;
286                }
287                break;
288            case 'K':
289                // CLEAR END OF LINE
290                break;
291            case 'm':
292                for(j = 0; j < count; j++)
293                    _manage_modifiers(tuple[j], &fg, &bg, &save_fg, &save_bg, &bold);
294                if(bold && fg < 8)
295                    fg += 8;
296                if(bold && bg < 8)
297                    bg += 8;
298                cucul_set_color(cv, fg, bg);
299                break;
300            default:
301                break;
302            }
303        }
304        else
305        {
306            if(buffer[i] == '\n')
307            {
308                x = 0;
309                y++;
310            }
311            else if(buffer[i] == '\r')
312            {
313                    // DOS sucks.
314            }
315            else
316            {
317
318                _cucul_putchar32(cv, x, y,_cucul_cp437_to_utf32(buffer[i]));
319
320                x++;
321            }
322        }
323
324        if(x >= width || y >= height)
325        {
326            if(x >= width)
327                width = x + 1;
328
329            if(y >= height)
330                height = y + 1;
331
332            cucul_set_canvas_size(cv, width, height);
333        }
334
335        i += add; // add is tuple char count
336        add = 0;
337    }
338
339    return cv;
340}
341
342/* XXX : ANSI loader helpers */
343
344unsigned char _get_ansi_command(unsigned char const *buffer, int size)
345{
346    int i;
347
348    for(i = 0; i < size; i++)
349        if(IS_ALPHA(buffer[i]))
350            return buffer[i];
351
352    return 0;
353}
354
355int _parse_tuple(unsigned int *ret, unsigned char const *buffer, int size)
356{
357    int i = 0;
358    int j = 0;
359    int t = 0;
360    unsigned char nbr[1024];
361
362    ret[0] = END_TUP;
363
364    for(i = 0; i < size; i++)
365    {
366        if(IS_ALPHA(buffer[i]))
367        {
368            if(j != 0)
369            {
370                ret[t] = atoi((char*)nbr);
371                t++;
372            }
373            ret[t] = END_TUP;
374            j = 0;
375            return i;
376        }
377
378        if(buffer[i] != ';')
379        {
380            nbr[j] = buffer[i];
381            nbr[j + 1] = 0;
382            j++;
383        }
384        else
385        {
386            ret[t] = atoi((char*)nbr);
387            t++;
388            ret[t] = END_TUP;
389            j = 0;
390        }
391    }
392    return size;
393}
394
395
396
397void _manage_modifiers(int i, int *fg, int *bg, int *save_fg, int *save_bg, int *bold)
398{
399    static uint8_t const ansi2cucul[] =
400    {
401        CUCUL_COLOR_BLACK,
402        CUCUL_COLOR_RED,
403        CUCUL_COLOR_GREEN,
404        CUCUL_COLOR_BROWN,
405        CUCUL_COLOR_BLUE,
406        CUCUL_COLOR_MAGENTA,
407        CUCUL_COLOR_CYAN,
408        CUCUL_COLOR_LIGHTGRAY
409    };
410
411    if(i >= 30 && i <= 37)
412        *fg = ansi2cucul[i - 30];
413    else if(i == 39)
414        *fg = CUCUL_COLOR_DEFAULT;
415    else if(i >= 40 && i <= 47)
416        *bg = ansi2cucul[i - 40];
417    else if(i == 49)
418        *bg = CUCUL_COLOR_DEFAULT;
419    else if(i >= 90 && i <= 97)
420        *fg = ansi2cucul[i - 90] + 8;
421    else if(i >= 100 && i <= 107)
422        *bg = ansi2cucul[i - 100] + 8;
423    else switch(i)
424    {
425    case 0:
426        *fg = CUCUL_COLOR_DEFAULT;
427        *bg = CUCUL_COLOR_DEFAULT;
428        *bold = 0;
429        break;
430    case 1: /* BOLD */
431        *bold = 1;
432        break;
433    case 4: // Underline
434        break;
435    case 5: // blink
436        break;
437    case 7: // reverse
438        *fg = 15 - *fg;
439        *bg = 15 - *bg;
440        break;
441    case 8: // invisible
442        *save_fg = *fg;
443        *save_bg = *bg;
444        *fg = CUCUL_COLOR_TRANSPARENT;
445        *bg = CUCUL_COLOR_TRANSPARENT;
446        break;
447    case 28: // not invisible
448        *fg = *save_fg;
449        *bg = *save_bg;
450        break;
451    default:
452        break;
453    }
454}
455
Note: See TracBrowser for help on using the repository browser.