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

Last change on this file since 876 was 876, checked in by Sam Hocevar, 15 years ago
  • Small optimisation in import_text().
  • Property svn:keywords set to Id
File size: 13.0 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 876 2006-04-25 15:04:19Z sam $
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        /* if 4 first letters are CACA */
70        if(size >= 4 &&
71            buf[0] == 'C' && buf[1] == 'A' && buf[2] == 'C' && buf[3] != 'A')
72            return import_caca(data, size);
73
74        /* If 2 first characters are ESC[ (not reliable at all) */
75        if(size >= 2 && buf[0] == 0x1b && buf[1] == '[')
76            return import_ansi(data, size);
77
78        /* Otherwise, import it as text */
79        return import_text(data, size);
80    }
81    return NULL;
82}
83
84/** \brief Get available import formats
85 *
86 *  Return a list of available import formats. The list is a NULL-terminated
87 *  array of strings, interleaving a string containing the internal value for
88 *  the import format, to be used with cucul_import_canvas(), and a string
89 *  containing the natural language description for that import format.
90 *
91 *  \return An array of strings.
92 */
93char const * const * cucul_get_import_list(void)
94{
95    static char const * const list[] =
96    {
97        "", "autodetect",
98        "text", "plain text",
99        "caca", "native libcaca format",
100        "ansi", "ANSI coloured text",
101        NULL, NULL
102    };
103
104    return list;
105}
106
107/*
108 * XXX: the following functions are local.
109 */
110
111static cucul_canvas_t *import_caca(void const *data, unsigned int size)
112{
113    cucul_canvas_t *cv;
114    uint8_t const *buf = (uint8_t const *)data;
115    unsigned int width, height, n;
116
117    if(size < 16)
118        return NULL;
119
120    if(buf[0] != 'C' || buf[1] != 'A' || buf[2] != 'C' || buf[3] != 'A')
121        return NULL;
122
123    if(buf[4] != 'C' || buf[5] != 'A' || buf[6] != 'N' || buf[7] != 'V')
124        return NULL;
125
126    width = ((uint32_t)buf[8] << 24) | ((uint32_t)buf[9] << 16)
127           | ((uint32_t)buf[10] << 8) | (uint32_t)buf[11];
128    height = ((uint32_t)buf[12] << 24) | ((uint32_t)buf[13] << 16)
129            | ((uint32_t)buf[14] << 8) | (uint32_t)buf[15];
130
131    if(!width || !height)
132        return NULL;
133
134    if(size != 16 + width * height * 8)
135        return NULL;
136
137    cv = cucul_create_canvas(width, height);
138
139    if(!cv)
140        return NULL;
141
142    for(n = height * width; n--; )
143    {
144        cv->chars[n] = ((uint32_t)buf[16 + 0 + 8 * n] << 24)
145                     | ((uint32_t)buf[16 + 1 + 8 * n] << 16)
146                     | ((uint32_t)buf[16 + 2 + 8 * n] << 8)
147                     | (uint32_t)buf[16 + 3 + 8 * n];
148        cv->attr[n] = ((uint32_t)buf[16 + 4 + 8 * n] << 24)
149                    | ((uint32_t)buf[16 + 5 + 8 * n] << 16)
150                    | ((uint32_t)buf[16 + 6 + 8 * n] << 8)
151                    | (uint32_t)buf[16 + 7 + 8 * n];
152    }
153
154    return cv;
155}
156
157static cucul_canvas_t *import_text(void const *data, unsigned int size)
158{
159    cucul_canvas_t *cv;
160    char const *text = (char const *)data;
161    unsigned int width = 1, height = 1, x = 0, y = 0, i;
162
163    cv = cucul_create_canvas(width, height);
164    cucul_set_color(cv, CUCUL_COLOR_DEFAULT, CUCUL_COLOR_TRANSPARENT);
165
166    for(i = 0; i < size; i++)
167    {
168        unsigned char ch = *text++;
169
170        if(ch == '\r')
171            continue;
172
173        if(ch == '\n')
174        {
175            x = 0;
176            y++;
177            continue;
178        }
179
180        if(x >= width || y >= height)
181        {
182            width = x + 1;
183            height = y + 1;
184            cucul_set_canvas_size(cv, width, height);
185        }
186
187        cucul_putchar(cv, x, y, ch);
188        x++;
189    }
190
191    return cv;
192}
193
194#define IS_ALPHA(x) (x>='A' && x<='z')
195void updateCanvasSize(cucul_canvas_t *cv, int x, int y, int *max_x, int *max_y);
196unsigned char get_ansi_command(unsigned char const *buffer, int size);
197int parse_tuple(unsigned int *ret, unsigned char const *buffer, int size);
198void manage_modifiers(char c, int *fg, int *bg, int *old_fg, int *old_bg);
199
200static cucul_canvas_t *import_ansi(void const *data, unsigned int size)
201{
202    cucul_canvas_t *cv;
203    unsigned char const *buffer = (unsigned char const*)data;
204    unsigned int i, sent_size = 0;
205    unsigned char c;
206    unsigned int count = 0;
207    unsigned int tuple[1024]; /* Should be enough. Will it be ? */
208    int x = 0, y = 0, max_x = 80, max_y = 25;
209    int save_x = 0, save_y = 0;
210    unsigned int j, add = 0;
211    int fg, bg, old_fg, old_bg;
212
213    fg = old_fg = CUCUL_COLOR_LIGHTGRAY;
214    bg = old_bg = CUCUL_COLOR_BLACK;
215
216    cv = cucul_create_canvas(max_x, max_y);
217
218    for(i = 0; i < size; i++)
219    {
220        if((buffer[i] == 0x1b) && (buffer[i + 1] == '['))  /* ESC code */
221        {
222            i++; // ESC
223            i++; // [
224            sent_size = size - i;
225            c = get_ansi_command(&buffer[i], sent_size);
226            add = parse_tuple(tuple, &buffer[i], sent_size);
227            count = 0;
228
229            while(tuple[count] != 0x1337)
230                count++;  /* Gruik */
231
232            switch(c)
233            {
234            case 'f':
235            case 'H':
236                if(tuple[0] != 0x1337)
237                {
238                    x = tuple[0];
239                    if(tuple[1] != 0x1337)
240                        y = tuple[1];
241                }
242                else
243                {
244                    x = 0;
245                    y = 0;
246                }
247                updateCanvasSize(cv, x, y, &max_x, &max_y);
248                break;
249            case 'A':
250                if(tuple[0] == 0x1337)
251                    y -= 1;
252                else
253                    y -= tuple[0];
254                if(y < 0) y = 0;
255                updateCanvasSize(cv, x, y, &max_x, &max_y);
256                break;
257            case 'B':
258                if(tuple[0] == 0x1337)
259                    y++;
260                else
261                    y += tuple[0];
262                updateCanvasSize(cv, x, y, &max_x, &max_y);
263                break;
264            case 'C':
265                if(tuple[0] == 0x1337)
266                    x++;
267                else
268                    x += tuple[0];
269                updateCanvasSize(cv, x, y, &max_x, &max_y);
270                break;
271            case 'D':
272                if(tuple[0] == 0x1337)
273                    x--;
274                else
275                    x -= tuple[0];
276                if(x < 0) x = 0;
277                updateCanvasSize(cv, x, y, &max_x, &max_y);
278                break;
279            case 's':
280                save_x = x;
281                save_y = y;
282                break;
283            case 'u':
284                x = save_x;
285                y = save_y;
286                updateCanvasSize(cv, x, y, &max_x, &max_y);
287                break;
288            case 'J':
289                if(tuple[0] == 2)
290                {
291                    x = 0;
292                    y = 0;
293                    updateCanvasSize(cv, x, y, &max_x, &max_y);
294                }
295                break;
296            case 'K':
297                // CLEAR END OF LINE
298                break;
299            case 'm':
300                for(j = 0; j < count; j++)
301                    manage_modifiers(tuple[j], &fg, &bg, &old_fg, &old_bg);
302                cucul_set_color(cv, fg, bg);
303                break;
304            default:
305                /*printf("Unknow command %c (%c)\n", c, IS_ALPHA(c)?c:'.');*/
306                break;
307            }
308        }
309        else
310        {
311            if(buffer[i] == '\n')
312            {
313                x = 0;
314                y++;
315                updateCanvasSize(cv, x, y, &max_x, &max_y);
316            }
317            else if(buffer[i] == '\r')
318            {
319                    // DOS sucks.
320            }
321            else
322            {
323                if((buffer[i] >= 0x20) || (buffer[i] <= 0x7E))
324                {
325                    _cucul_putchar32(cv, x, y,_cucul_cp437_to_utf32(buffer[i]));
326
327                    //  cucul_putchar(cv, x, y, buffer[i]);
328                }
329                else
330                    cucul_putchar(cv, x, y, '?');
331                x++;
332                updateCanvasSize(cv, x, y, &max_x, &max_y);
333            }
334        }
335
336        i += add; // add is tuple char count, then +[ +command
337        add = 0;
338    }
339
340    return cv;
341}
342
343/* XXX : ANSI loader helpers */
344
345unsigned char get_ansi_command(unsigned char const *buffer, int size)
346{
347    int i;
348
349    for(i = 0; i < size; i++)
350        if(IS_ALPHA(buffer[i]))
351            return buffer[i];
352
353    return 0;
354}
355
356int parse_tuple(unsigned int *ret, unsigned char const *buffer, int size)
357{
358    int i = 0;
359    int j = 0;
360    int t = 0;
361    unsigned char nbr[1024];
362
363    ret[0] = 0x1337;
364
365    for(i = 0; i < size; i++)
366    {
367        if(IS_ALPHA(buffer[i]))
368        {
369            if(j != 0)
370            {
371                ret[t] = atoi((char*)nbr);
372                t++;
373            }
374            ret[t] = 0x1337;
375            j = 0;
376            return i;
377        }
378
379        if(buffer[i] != ';')
380        {
381            nbr[j] = buffer[i];
382            nbr[j + 1] = 0;
383            j++;
384        }
385        else
386        {
387            ret[t] = atoi((char*)nbr);
388            t++;
389            ret[t] = 0x1337;
390            j = 0;
391        }
392    }
393    return size;
394}
395
396
397
398void manage_modifiers(char c, int *fg, int *bg, int *old_fg, int *old_bg)
399{
400    switch(c)
401    {
402    case 0:
403        *fg = CUCUL_COLOR_DEFAULT;
404        *bg = CUCUL_COLOR_DEFAULT;
405        break;
406    case 1: // BOLD
407        break;
408    case 4: // Underline
409        break;
410    case 5: // blink
411        break;
412    case 7: // reverse
413        *fg = 15 - *fg;
414        *bg = 15 - *bg;
415        break;
416    case 8: // invisible
417        *old_fg = *fg;
418        *old_bg = *bg;
419        *fg = CUCUL_COLOR_TRANSPARENT;
420        *bg = CUCUL_COLOR_TRANSPARENT;
421        break;
422    case 28: // not invisible
423        *fg = *old_fg;
424        *bg = *old_bg;
425        break;
426    case 30: *fg = CUCUL_COLOR_BLACK; break;
427    case 31: *fg = CUCUL_COLOR_RED; break;
428    case 32: *fg = CUCUL_COLOR_GREEN; break;
429    case 33: *fg = CUCUL_COLOR_BROWN; break;
430    case 34: *fg = CUCUL_COLOR_BLUE; break;
431    case 35: *fg = CUCUL_COLOR_MAGENTA; break;
432    case 36: *fg = CUCUL_COLOR_CYAN; break;
433    case 37: *fg = CUCUL_COLOR_WHITE; break;
434    case 39: *fg = CUCUL_COLOR_LIGHTGRAY; break;
435    case 40: *bg = CUCUL_COLOR_BLACK; break;
436    case 41: *bg = CUCUL_COLOR_RED; break;
437    case 42: *bg = CUCUL_COLOR_GREEN; break;
438    case 43: *bg = CUCUL_COLOR_BROWN; break;
439    case 44: *bg = CUCUL_COLOR_BLUE; break;
440    case 45: *bg = CUCUL_COLOR_MAGENTA; break;
441    case 46: *bg = CUCUL_COLOR_CYAN; break;
442    case 47: *bg = CUCUL_COLOR_WHITE; break;
443    case 49: *bg = CUCUL_COLOR_BLACK; break;
444
445    case 90: *fg = CUCUL_COLOR_DARKGRAY; break;
446    case 91: *fg = CUCUL_COLOR_LIGHTRED; break;
447    case 92: *fg = CUCUL_COLOR_LIGHTGREEN; break;
448    case 93: *fg = CUCUL_COLOR_YELLOW; break;
449    case 94: *fg = CUCUL_COLOR_LIGHTBLUE; break;
450    case 95: *fg = CUCUL_COLOR_LIGHTMAGENTA; break;
451    case 96: *fg = CUCUL_COLOR_LIGHTCYAN; break;
452    case 97: *fg = CUCUL_COLOR_WHITE; break;
453    case 100: *bg = CUCUL_COLOR_DARKGRAY; break;
454    case 101: *bg = CUCUL_COLOR_LIGHTRED; break;
455    case 102: *bg = CUCUL_COLOR_LIGHTGREEN; break;
456    case 103: *bg = CUCUL_COLOR_YELLOW; break;
457    case 104: *bg = CUCUL_COLOR_LIGHTBLUE; break;
458    case 105: *bg = CUCUL_COLOR_LIGHTMAGENTA; break;
459    case 106: *bg = CUCUL_COLOR_LIGHTCYAN; break;
460    case 107: *bg = CUCUL_COLOR_WHITE; break;
461
462    default:
463        /*   printf("Unknow option to 'm' %d (%c)\n", c, IS_ALPHA(c)?c:'.'); */
464        break;
465    }
466}
467
468void updateCanvasSize(cucul_canvas_t *cv, int x, int y, int *max_x, int *max_y)
469{
470    if(x > *max_x)
471        *max_x = x;
472
473    if(y > *max_y)
474        *max_y = y;
475
476    cucul_set_canvas_size(cv, *max_x, *max_y);
477}
478
Note: See TracBrowser for help on using the repository browser.