git.sur5r.net Git - u-boot/blob - tools/easylogo/easylogo.c

   1 /*
   2 ** Easylogo TGA->header converter
   3 ** ==============================
   4 ** (C) 2000 by Paolo Scaffardi (arsenio@tin.it)
   5 ** AIRVENT SAM s.p.a - RIMINI(ITALY)
   6 ** (C) 2007-2008 Mike Frysinger <vapier@gentoo.org>
   7 **
   8 ** This is still under construction!
   9 */
  10
  11 #include <errno.h>
  12 #include <getopt.h>
  13 #include <stdbool.h>
  14 #include <stdio.h>
  15 #include <stdlib.h>
  16 #include <string.h>
  17 #include <unistd.h>
  18 #include <sys/stat.h>
  19
  20 #pragma pack(1)
  21
  22 /*#define ENABLE_ASCII_BANNERS */
  23
  24 typedef struct {
  25         unsigned char id;
  26         unsigned char ColorMapType;
  27         unsigned char ImageTypeCode;
  28         unsigned short ColorMapOrigin;
  29         unsigned short ColorMapLenght;
  30         unsigned char ColorMapEntrySize;
  31         unsigned short ImageXOrigin;
  32         unsigned short ImageYOrigin;
  33         unsigned short ImageWidth;
  34         unsigned short ImageHeight;
  35         unsigned char ImagePixelSize;
  36         unsigned char ImageDescriptorByte;
  37 } tga_header_t;
  38
  39 typedef struct {
  40         unsigned char r, g, b;
  41 } rgb_t;
  42
  43 typedef struct {
  44         unsigned char b, g, r;
  45 } bgr_t;
  46
  47 typedef struct {
  48         unsigned char Cb, y1, Cr, y2;
  49 } yuyv_t;
  50
  51 typedef struct {
  52         void *data, *palette;
  53         int width, height, pixels, bpp, pixel_size, size, palette_size, yuyv;
  54 } image_t;
  55
  56 void *xmalloc (size_t size)
  57 {
  58         void *ret = malloc (size);
  59         if (!ret) {
  60                 fprintf (stderr, "\nerror: malloc(%zu) failed: %s",
  61                         size, strerror(errno));
  62                 exit (1);
  63         }
  64         return ret;
  65 }
  66
  67 void StringUpperCase (char *str)
  68 {
  69         int count = strlen (str);
  70         char c;
  71
  72         while (count--) {
  73                 c = *str;
  74                 if ((c >= 'a') && (c <= 'z'))
  75                         *str = 'A' + (c - 'a');
  76                 str++;
  77         }
  78 }
  79
  80 void StringLowerCase (char *str)
  81 {
  82         int count = strlen (str);
  83         char c;
  84
  85         while (count--) {
  86                 c = *str;
  87                 if ((c >= 'A') && (c <= 'Z'))
  88                         *str = 'a' + (c - 'A');
  89                 str++;
  90         }
  91 }
  92 void pixel_rgb_to_yuyv (rgb_t * rgb_pixel, yuyv_t * yuyv_pixel)
  93 {
  94         unsigned int pR, pG, pB;
  95
  96         /* Transform (0-255) components to (0-100) */
  97         pR = rgb_pixel->r * 100 / 255;
  98         pG = rgb_pixel->g * 100 / 255;
  99         pB = rgb_pixel->b * 100 / 255;
 100
 101         /* Calculate YUV values (0-255) from RGB beetween 0-100 */
 102         yuyv_pixel->y1 = yuyv_pixel->y2 = 209 * (pR + pG + pB) / 300 + 16;
 103         yuyv_pixel->Cb = pB - (pR / 4) - (pG * 3 / 4) + 128;
 104         yuyv_pixel->Cr = pR - (pG * 3 / 4) - (pB / 4) + 128;
 105
 106         return;
 107 }
 108
 109 void printlogo_rgb (rgb_t * data, int w, int h)
 110 {
 111         int x, y;
 112
 113         for (y = 0; y < h; y++) {
 114                 for (x = 0; x < w; x++, data++)
 115                         if ((data->r <
 116                              30) /*&&(data->g == 0)&&(data->b == 0) */ )
 117                                 printf (" ");
 118                         else
 119                                 printf ("X");
 120                 printf ("\n");
 121         }
 122 }
 123
 124 void printlogo_yuyv (unsigned short *data, int w, int h)
 125 {
 126         int x, y;
 127
 128         for (y = 0; y < h; y++) {
 129                 for (x = 0; x < w; x++, data++)
 130                         if (*data == 0x1080)    /* Because of inverted on i386! */
 131                                 printf (" ");
 132                         else
 133                                 printf ("X");
 134                 printf ("\n");
 135         }
 136 }
 137
 138 static inline unsigned short le16_to_cpu (unsigned short val)
 139 {
 140         union {
 141                 unsigned char pval[2];
 142                 unsigned short val;
 143         } swapped;
 144
 145         swapped.val = val;
 146         return (swapped.pval[1] << 8) + swapped.pval[0];
 147 }
 148
 149 int image_load_tga (image_t * image, char *filename)
 150 {
 151         FILE *file;
 152         tga_header_t header;
 153         int i;
 154         unsigned char app;
 155         rgb_t *p;
 156
 157         if ((file = fopen (filename, "rb")) == NULL)
 158                 return -1;
 159
 160         fread (&header, sizeof (header), 1, file);
 161
 162         /* byte swap: tga is little endian, host is ??? */
 163         header.ColorMapOrigin = le16_to_cpu (header.ColorMapOrigin);
 164         header.ColorMapLenght = le16_to_cpu (header.ColorMapLenght);
 165         header.ImageXOrigin = le16_to_cpu (header.ImageXOrigin);
 166         header.ImageYOrigin = le16_to_cpu (header.ImageYOrigin);
 167         header.ImageWidth = le16_to_cpu (header.ImageWidth);
 168         header.ImageHeight = le16_to_cpu (header.ImageHeight);
 169
 170         image->width = header.ImageWidth;
 171         image->height = header.ImageHeight;
 172
 173         switch (header.ImageTypeCode) {
 174         case 2:         /* Uncompressed RGB */
 175                 image->yuyv = 0;
 176                 image->palette_size = 0;
 177                 image->palette = NULL;
 178                 break;
 179
 180         default:
 181                 printf ("Format not supported!\n");
 182                 return -1;
 183         }
 184
 185         image->bpp = header.ImagePixelSize;
 186         image->pixel_size = ((image->bpp - 1) / 8) + 1;
 187         image->pixels = image->width * image->height;
 188         image->size = image->pixels * image->pixel_size;
 189         image->data = xmalloc (image->size);
 190
 191         if (image->bpp != 24) {
 192                 printf ("Bpp not supported: %d!\n", image->bpp);
 193                 return -1;
 194         }
 195
 196         fread (image->data, image->size, 1, file);
 197
 198 /* Swapping R and B values */
 199
 200         p = image->data;
 201         for (i = 0; i < image->pixels; i++, p++) {
 202                 app = p->r;
 203                 p->r = p->b;
 204                 p->b = app;
 205         }
 206
 207 /* Swapping image */
 208
 209         if (!(header.ImageDescriptorByte & 0x20)) {
 210                 unsigned char *temp = xmalloc (image->size);
 211                 int linesize = image->pixel_size * image->width;
 212                 void *dest = image->data,
 213                         *source = temp + image->size - linesize;
 214
 215                 printf ("S");
 216                 if (temp == NULL) {
 217                         printf ("Cannot alloc temp buffer!\n");
 218                         return -1;
 219                 }
 220
 221                 memcpy (temp, image->data, image->size);
 222                 for (i = 0; i < image->height;
 223                      i++, dest += linesize, source -= linesize)
 224                         memcpy (dest, source, linesize);
 225
 226                 free (temp);
 227         }
 228 #ifdef ENABLE_ASCII_BANNERS
 229         printlogo_rgb (image->data, image->width, image->height);
 230 #endif
 231
 232         fclose (file);
 233         return 0;
 234 }
 235
 236 void image_free (image_t * image)
 237 {
 238         free (image->data);
 239         free (image->palette);
 240 }
 241
 242 int image_rgb_to_yuyv (image_t * rgb_image, image_t * yuyv_image)
 243 {
 244         rgb_t *rgb_ptr = (rgb_t *) rgb_image->data;
 245         yuyv_t yuyv;
 246         unsigned short *dest;
 247         int count = 0;
 248
 249         yuyv_image->pixel_size = 2;
 250         yuyv_image->bpp = 16;
 251         yuyv_image->yuyv = 1;
 252         yuyv_image->width = rgb_image->width;
 253         yuyv_image->height = rgb_image->height;
 254         yuyv_image->pixels = yuyv_image->width * yuyv_image->height;
 255         yuyv_image->size = yuyv_image->pixels * yuyv_image->pixel_size;
 256         dest = (unsigned short *) (yuyv_image->data =
 257                                    xmalloc (yuyv_image->size));
 258         yuyv_image->palette = 0;
 259         yuyv_image->palette_size = 0;
 260
 261         while ((count++) < rgb_image->pixels) {
 262                 pixel_rgb_to_yuyv (rgb_ptr++, &yuyv);
 263
 264                 if ((count & 1) == 0)   /* Was == 0 */
 265                         memcpy (dest, ((void *) &yuyv) + 2, sizeof (short));
 266                 else
 267                         memcpy (dest, (void *) &yuyv, sizeof (short));
 268
 269                 dest++;
 270         }
 271
 272 #ifdef ENABLE_ASCII_BANNERS
 273         printlogo_yuyv (yuyv_image->data, yuyv_image->width,
 274                         yuyv_image->height);
 275 #endif
 276         return 0;
 277 }
 278
 279 int image_rgb888_to_rgb565(image_t *rgb888_image, image_t *rgb565_image)
 280 {
 281         rgb_t *rgb_ptr = (rgb_t *) rgb888_image->data;
 282         unsigned short *dest;
 283         int count = 0;
 284
 285         rgb565_image->pixel_size = 2;
 286         rgb565_image->bpp = 16;
 287         rgb565_image->yuyv = 0;
 288         rgb565_image->width = rgb888_image->width;
 289         rgb565_image->height = rgb888_image->height;
 290         rgb565_image->pixels = rgb565_image->width * rgb565_image->height;
 291         rgb565_image->size = rgb565_image->pixels * rgb565_image->pixel_size;
 292         dest = (unsigned short *) (rgb565_image->data =
 293                                    xmalloc(rgb565_image->size));
 294         rgb565_image->palette = 0;
 295         rgb565_image->palette_size = 0;
 296
 297         while ((count++) < rgb888_image->pixels) {
 298
 299                 *dest++ = ((rgb_ptr->b & 0xF8) << 8) |
 300                         ((rgb_ptr->g & 0xFC) << 3) |
 301                         (rgb_ptr->r >> 3);
 302                 rgb_ptr++;
 303         }
 304
 305         return 0;
 306 }
 307
 308 enum comp_t {
 309         COMP_NONE,
 310         COMP_GZIP,
 311         COMP_LZMA,
 312 };
 313 static enum comp_t compression = COMP_NONE;
 314 static bool bss_storage = false;
 315
 316 int image_save_header (image_t * image, char *filename, char *varname)
 317 {
 318         FILE *file = fopen (filename, "w");
 319         char app[256], str[256] = "", def_name[64];
 320         int count = image->size, col = 0;
 321         unsigned char *dataptr = image->data;
 322
 323         if (file == NULL)
 324                 return -1;
 325
 326         /*  Author information */
 327         fprintf (file,
 328                  "/*\n * Generated by EasyLogo, (C) 2000 by Paolo Scaffardi\n *\n");
 329         fprintf (file,
 330                  " * To use this, include it and call: easylogo_plot(screen,&%s, width,x,y)\n *\n",
 331                  varname);
 332         fprintf (file,
 333                  " * Where:\t'screen'\tis the pointer to the frame buffer\n");
 334         fprintf (file, " *\t\t'width'\tis the screen width\n");
 335         fprintf (file, " *\t\t'x'\t\tis the horizontal position\n");
 336         fprintf (file, " *\t\t'y'\t\tis the vertical position\n */\n\n");
 337
 338         /* image compress */
 339         if (compression != COMP_NONE) {
 340                 const char *errstr = NULL;
 341                 unsigned char *compressed;
 342                 const char *comp_name;
 343                 struct stat st;
 344                 FILE *compfp;
 345                 size_t filename_len = strlen(filename);
 346                 char *compfilename = xmalloc(filename_len + 20);
 347                 char *compcmd = xmalloc(filename_len + 50);
 348
 349                 sprintf(compfilename, "%s.bin", filename);
 350                 switch (compression) {
 351                 case COMP_GZIP:
 352                         strcpy(compcmd, "gzip");
 353                         comp_name = "GZIP";
 354                         break;
 355                 case COMP_LZMA:
 356                         strcpy(compcmd, "lzma");
 357                         comp_name = "LZMA";
 358                         break;
 359                 default:
 360                         errstr = "\nerror: unknown compression method";
 361                         goto done;
 362                 }
 363                 strcat(compcmd, " > ");
 364                 strcat(compcmd, compfilename);
 365                 compfp = popen(compcmd, "w");
 366                 if (!compfp) {
 367                         errstr = "\nerror: popen() failed";
 368                         goto done;
 369                 }
 370                 if (fwrite(image->data, image->size, 1, compfp) != 1) {
 371                         errstr = "\nerror: writing data to gzip failed";
 372                         goto done;
 373                 }
 374                 if (pclose(compfp)) {
 375                         errstr = "\nerror: gzip process failed";
 376                         goto done;
 377                 }
 378
 379                 compfp = fopen(compfilename, "r");
 380                 if (!compfp) {
 381                         errstr = "\nerror: open() on gzip data failed";
 382                         goto done;
 383                 }
 384                 if (stat(compfilename, &st)) {
 385                         errstr = "\nerror: stat() on gzip file failed";
 386                         goto done;
 387                 }
 388                 compressed = xmalloc(st.st_size);
 389                 if (fread(compressed, st.st_size, 1, compfp) != 1) {
 390                         errstr = "\nerror: reading gzip data failed";
 391                         goto done;
 392                 }
 393                 fclose(compfp);
 394
 395                 unlink(compfilename);
 396
 397                 dataptr = compressed;
 398                 count = st.st_size;
 399                 fprintf(file, "#define EASYLOGO_ENABLE_%s %i\n\n", comp_name, count);
 400                 if (bss_storage)
 401                         fprintf (file, "static unsigned char EASYLOGO_DECOMP_BUFFER[%i];\n\n", image->size);
 402
 403  done:
 404                 free(compfilename);
 405                 free(compcmd);
 406
 407                 if (errstr) {
 408                         perror (errstr);
 409                         return -1;
 410                 }
 411         }
 412
 413         /*      Headers */
 414         fprintf (file, "#include <video_easylogo.h>\n\n");
 415         /*      Macros */
 416         strcpy (def_name, varname);
 417         StringUpperCase (def_name);
 418         fprintf (file, "#define DEF_%s_WIDTH\t\t%d\n", def_name,
 419                  image->width);
 420         fprintf (file, "#define DEF_%s_HEIGHT\t\t%d\n", def_name,
 421                  image->height);
 422         fprintf (file, "#define DEF_%s_PIXELS\t\t%d\n", def_name,
 423                  image->pixels);
 424         fprintf (file, "#define DEF_%s_BPP\t\t%d\n", def_name, image->bpp);
 425         fprintf (file, "#define DEF_%s_PIXEL_SIZE\t%d\n", def_name,
 426                  image->pixel_size);
 427         fprintf (file, "#define DEF_%s_SIZE\t\t%d\n\n", def_name,
 428                  image->size);
 429         /*  Declaration */
 430         fprintf (file, "unsigned char DEF_%s_DATA[] = {\n",
 431                  def_name);
 432
 433         /*      Data */
 434         while (count)
 435                 switch (col) {
 436                 case 0:
 437                         sprintf (str, " 0x%02x", *dataptr++);
 438                         col++;
 439                         count--;
 440                         break;
 441
 442                 case 16:
 443                         fprintf (file, "%s", str);
 444                         if (count > 0)
 445                                 fprintf (file, ",");
 446                         fprintf (file, "\n");
 447
 448                         col = 0;
 449                         break;
 450
 451                 default:
 452                         strcpy (app, str);
 453                         sprintf (str, "%s, 0x%02x", app, *dataptr++);
 454                         col++;
 455                         count--;
 456                         break;
 457                 }
 458
 459         if (col)
 460                 fprintf (file, "%s\n", str);
 461
 462         /*      End of declaration */
 463         fprintf (file, "};\n\n");
 464         /*      Variable */
 465         fprintf (file, "fastimage_t %s = {\n", varname);
 466         fprintf (file, "                DEF_%s_DATA,\n", def_name);
 467         fprintf (file, "                DEF_%s_WIDTH,\n", def_name);
 468         fprintf (file, "                DEF_%s_HEIGHT,\n", def_name);
 469         fprintf (file, "                DEF_%s_BPP,\n", def_name);
 470         fprintf (file, "                DEF_%s_PIXEL_SIZE,\n", def_name);
 471         fprintf (file, "                DEF_%s_SIZE\n};\n", def_name);
 472
 473         fclose (file);
 474
 475         return 0;
 476 }
 477
 478 #define DEF_FILELEN     256
 479
 480 static void usage (int exit_status)
 481 {
 482         puts (
 483                 "EasyLogo 1.0 (C) 2000 by Paolo Scaffardi\n"
 484                 "\n"
 485                 "Syntax:        easylogo [options] inputfile [outputvar [outputfile]]\n"
 486                 "\n"
 487                 "Options:\n"
 488                 "  -r     Output RGB888 instead of YUYV\n"
 489                 "  -s     Output RGB565 instead of YUYV\n"
 490                 "  -g     Compress with gzip\n"
 491                 "  -l     Compress with lzma\n"
 492                 "  -b     Preallocate space in bss for decompressing image\n"
 493                 "  -h     Help output\n"
 494                 "\n"
 495                 "Where: 'inputfile'   is the TGA image to load\n"
 496                 "       'outputvar'   is the variable name to create\n"
 497                 "       'outputfile'  is the output header file (default is 'inputfile.h')"
 498         );
 499         exit (exit_status);
 500 }
 501
 502 int main (int argc, char *argv[])
 503 {
 504         int c;
 505         bool use_rgb888 = false;
 506         bool use_rgb565 = false;
 507         char inputfile[DEF_FILELEN],
 508                 outputfile[DEF_FILELEN], varname[DEF_FILELEN];
 509
 510         image_t rgb888_logo, rgb565_logo, yuyv_logo;
 511
 512         while ((c = getopt(argc, argv, "hrsglb")) > 0) {
 513                 switch (c) {
 514                 case 'h':
 515                         usage (0);
 516                         break;
 517                 case 'r':
 518                         use_rgb888 = true;
 519                         puts("Using 24-bit RGB888 Output Fromat");
 520                         break;
 521                 case 's':
 522                         use_rgb565 = true;
 523                         puts("Using 16-bit RGB565 Output Fromat");
 524                         break;
 525                 case 'g':
 526                         compression = COMP_GZIP;
 527                         puts("Compressing with gzip");
 528                         break;
 529                 case 'l':
 530                         compression = COMP_LZMA;
 531                         puts("Compressing with lzma");
 532                         break;
 533                 case 'b':
 534                         bss_storage = true;
 535                         puts("Preallocating bss space for decompressing image");
 536                         break;
 537                 default:
 538                         usage (1);
 539                         break;
 540                 }
 541         }
 542
 543         c = argc - optind;
 544         if (c > 4 || c < 1)
 545                 usage (1);
 546
 547         strcpy (inputfile, argv[optind]);
 548
 549         if (c > 1)
 550                 strcpy (varname, argv[optind + 1]);
 551         else {
 552                 /* transform "input.tga" to just "input" */
 553                 char *dot;
 554                 strcpy (varname, inputfile);
 555                 dot = strchr (varname, '.');
 556                 if (dot)
 557                         *dot = '\0';
 558         }
 559
 560         if (c > 2)
 561                 strcpy (outputfile, argv[optind + 2]);
 562         else {
 563                 /* just append ".h" to input file name */
 564                 strcpy (outputfile, inputfile);
 565                 strcat (outputfile, ".h");
 566         }
 567
 568         /* Make sure the output is sent as soon as we printf() */
 569         setbuf(stdout, NULL);
 570
 571         printf ("Doing '%s' (%s) from '%s'...",
 572                 outputfile, varname, inputfile);
 573
 574         /* Import TGA logo */
 575
 576         printf ("L");
 577         if (image_load_tga(&rgb888_logo, inputfile) < 0) {
 578                 printf ("input file not found!\n");
 579                 exit (1);
 580         }
 581
 582         /* Convert, save, and free the image */
 583
 584         if (!use_rgb888 && !use_rgb565) {
 585                 printf ("C");
 586                 image_rgb_to_yuyv(&rgb888_logo, &yuyv_logo);
 587
 588                 printf("S");
 589                 image_save_header(&yuyv_logo, outputfile, varname);
 590                 image_free(&yuyv_logo);
 591         } else if (use_rgb565) {
 592                 printf("C");
 593                 image_rgb888_to_rgb565(&rgb888_logo, &rgb565_logo);
 594
 595                 printf("S");
 596                 image_save_header(&rgb565_logo, outputfile, varname);
 597                 image_free(&rgb565_logo);
 598         } else {
 599                 printf("S");
 600                 image_save_header(&rgb888_logo, outputfile, varname);
 601         }
 602
 603         /* Free original image and copy */
 604
 605         image_free(&rgb888_logo);
 606
 607         printf ("\n");
 608
 609         return 0;
 610 }