#include "compiler.h"

  enum {
  	MODE_GEN_INFO,

  	MODE_GEN_DATA,
  	MODE_GEN_BMP

  };

  typedef struct bitmap_s {		/* bitmap description */
  	uint16_t width;
  	uint16_t height;
  	uint8_t	palette[256*3];
  	uint8_t	*data;
  } bitmap_t;
  
  #define DEFAULT_CMAP_SIZE	16	/* size of default color map	*/

  void usage(const char *prog)
  {

  	fprintf(stderr, "Usage: %s [--gen-info|--gen-data|--gen-bmp] file
  ",
  		prog);

  }

  /*
   * Neutralize little endians.
   */
  uint16_t le_short(uint16_t x)
  {
      uint16_t val;
      uint8_t *p = (uint8_t *)(&x);
  
      val =  (*p++ & 0xff) << 0;
      val |= (*p & 0xff) << 8;
  
      return val;
  }
  
  void skip_bytes (FILE *fp, int n)
  {
  	while (n-- > 0)
  		fgetc (fp);
  }

  __attribute__ ((__noreturn__))
  int error (char * msg, FILE *fp)
  {
  	fprintf (stderr, "ERROR: %s
  ", msg);
  
  	fclose (fp);
  
  	exit (EXIT_FAILURE);
  }

  void gen_info(bitmap_t *b, uint16_t n_colors)
  {
  	printf("/*
  "
  		" * Automatically generated by \"tools/bmp_logo\"
  "
  		" *
  "
  		" * DO NOT EDIT
  "
  		" *
  "
  		" */
  
  
  "
  		"#ifndef __BMP_LOGO_H__
  "
  		"#define __BMP_LOGO_H__
  
  "
  		"#define BMP_LOGO_WIDTH\t\t%d
  "
  		"#define BMP_LOGO_HEIGHT\t\t%d
  "
  		"#define BMP_LOGO_COLORS\t\t%d
  "
  		"#define BMP_LOGO_OFFSET\t\t%d
  
  "
  		"extern unsigned short bmp_logo_palette[];
  "
  		"extern unsigned char bmp_logo_bitmap[];
  
  "
  		"#endif /* __BMP_LOGO_H__ */
  ",
  		b->width, b->height, n_colors,
  		DEFAULT_CMAP_SIZE);
  }

  int main (int argc, char *argv[])
  {

  	int	mode, i, x;

  	int	size;

  	FILE	*fp;
  	bitmap_t bmp;
  	bitmap_t *b = &bmp;

  	uint16_t data_offset, n_colors, hdr_size;

  	if (argc < 3) {
  		usage(argv[0]);

  		exit (EXIT_FAILURE);
  	}

  	if (!strcmp(argv[1], "--gen-info"))
  		mode = MODE_GEN_INFO;
  	else if (!strcmp(argv[1], "--gen-data"))
  		mode = MODE_GEN_DATA;

  	else if (!strcmp(argv[1], "--gen-bmp"))
  		mode = MODE_GEN_BMP;

  	else {
  		usage(argv[0]);
  		exit(EXIT_FAILURE);
  	}
  
  	fp = fopen(argv[2], "rb");
  	if (!fp) {
  		perror(argv[2]);

  		exit (EXIT_FAILURE);
  	}

  	if (fgetc (fp) != 'B' || fgetc (fp) != 'M')
  		error ("Input file is not a bitmap", fp);

  	/*
  	 * read width and height of the image, and the number of colors used;

  	 * ignore the rest
  	 */

  	skip_bytes (fp, 8);

  	if (fread (&data_offset, sizeof (uint16_t), 1, fp) != 1)
  		error ("Couldn't read bitmap data offset", fp);

  	skip_bytes(fp, 2);
  	if (fread(&hdr_size,   sizeof(uint16_t), 1, fp) != 1)
  		error("Couldn't read bitmap header size", fp);
  	if (hdr_size < 40)
  		error("Invalid bitmap header", fp);
  	skip_bytes(fp, 2);

  	if (fread (&b->width,   sizeof (uint16_t), 1, fp) != 1)
  		error ("Couldn't read bitmap width", fp);

  	skip_bytes (fp, 2);

  	if (fread (&b->height,  sizeof (uint16_t), 1, fp) != 1)
  		error ("Couldn't read bitmap height", fp);

  	skip_bytes (fp, 22);

  	if (fread (&n_colors, sizeof (uint16_t), 1, fp) != 1)
  		error ("Couldn't read bitmap colors", fp);

  	skip_bytes(fp, hdr_size - 34);

  
  	/*
  	 * Repair endianess.
  	 */

  	data_offset = le_short(data_offset);

  	b->width = le_short(b->width);
  	b->height = le_short(b->height);
  	n_colors = le_short(n_colors);

  	size = b->width * b->height;

  
  	/* assume we are working with an 8-bit file */
  	if ((n_colors == 0) || (n_colors > 256 - DEFAULT_CMAP_SIZE)) {
  		/* reserve DEFAULT_CMAP_SIZE color map entries for default map */
  		n_colors = 256 - DEFAULT_CMAP_SIZE;
  	}

  	if (mode == MODE_GEN_INFO) {
  		gen_info(b, n_colors);
  		goto out;
  	}
  
  	printf("/*
  "

  		" * Automatically generated by \"tools/bmp_logo\"
  "
  		" *
  "
  		" * DO NOT EDIT
  "
  		" *
  "
  		" */
  
  
  "

  		"#ifndef __BMP_LOGO_DATA_H__
  "
  		"#define __BMP_LOGO_DATA_H__
  
  ");

  	/* read and print the palette information */

  	printf("unsigned short bmp_logo_palette[] = {
  ");

  
  	for (i=0; i<n_colors; ++i) {
  		b->palette[(int)(i*3+2)] = fgetc(fp);
  		b->palette[(int)(i*3+1)] = fgetc(fp);
  		b->palette[(int)(i*3+0)] = fgetc(fp);
  		x=fgetc(fp);

  		printf ("%s0x0%X%X%X,%s",
  			((i%8) == 0) ? "\t" : "  ",

  			(b->palette[(int)(i*3+0)] >> 4) & 0x0F,
  			(b->palette[(int)(i*3+1)] >> 4) & 0x0F,
  			(b->palette[(int)(i*3+2)] >> 4) & 0x0F,

  			((i%8) == 7) ? "
  " : ""

  		);

  	}

  	/* seek to offset indicated by file header */

  	if (mode == MODE_GEN_BMP) {
  		/* copy full bmp file */
  		fseek(fp, 0L, SEEK_END);
  		size = ftell(fp);
  		fseek(fp, 0L, SEEK_SET);
  	} else {
  		fseek(fp, (long)data_offset, SEEK_SET);
  	}
  
  	/* allocate memory */
  	b->data = (uint8_t *)malloc(size);
  	if (!b->data)
  		error("Error allocating memory for file", fp);

  	/* read the bitmap; leave room for default color map */
  	printf ("
  ");
  	printf ("};
  ");
  	printf ("
  ");

  	printf("unsigned char bmp_logo_bitmap[] = {
  ");

  	if (mode == MODE_GEN_BMP) {
  		/* write full bmp */
  		for (i = 0; i < size; i++)
  			b->data[i] = (uint8_t)fgetc(fp);
  	} else {
  		for (i = (b->height - 1) * b->width; i >= 0; i -= b->width) {
  			for (x = 0; x < b->width; x++) {
  				b->data[i + x] = (uint8_t)fgetc(fp)

  						+ DEFAULT_CMAP_SIZE;

  			}

  		}
  	}

  	for (i = 0; i < size; ++i) {

  		if ((i%8) == 0)
  			putchar ('\t');
  		printf ("0x%02X,%c",
  			b->data[i],
  			((i%8) == 7) ? '
  ' : ' '
  		);
  	}
  	printf ("
  "
  		"};
  
  "

  		"#endif /* __BMP_LOGO_DATA_H__ */
  "

  	);

  out:
  	fclose(fp);
  	return 0;

  }