shithub: rgbds

ref: c02b04f7d239b14e94811f8124c26ed5b99cc98f
dir: /src/gfx/gb.c/

/*
 * Copyright © 2013 stag019 <stag019@gmail.com>
 *
 * Permission to use, copy, modify, and distribute this software for any
 * purpose with or without fee is hereby granted, provided that the above
 * copyright notice and this permission notice appear in all copies.
 *
 * THE SOFTWARE IS PROVIDED "AS IS" AND THE AUTHOR DISCLAIMS ALL WARRANTIES
 * WITH REGARD TO THIS SOFTWARE INCLUDING ALL IMPLIED WARRANTIES OF
 * MERCHANTABILITY AND FITNESS. IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR
 * ANY SPECIAL, DIRECT, INDIRECT, OR CONSEQUENTIAL DAMAGES OR ANY DAMAGES
 * WHATSOEVER RESULTING FROM LOSS OF USE, DATA OR PROFITS, WHETHER IN AN
 * ACTION OF CONTRACT, NEGLIGENCE OR OTHER TORTIOUS ACTION, ARISING OUT OF
 * OR IN CONNECTION WITH THE USE OR PERFORMANCE OF THIS SOFTWARE.
 */

#include <stdio.h>
#include <stdlib.h>

#include "gfx/main.h"

void
transpose_tiles(struct GBImage *gb, int width)
{
	uint8_t *newdata;
	int i;
	int newbyte;

	newdata = calloc(gb->size, 1);
	for (i = 0; i < gb->size; i++) {
		newbyte = i / (8 * depth) * width * 8 * depth;
		newbyte = newbyte % gb->size + 8 * depth * (newbyte / gb->size) + i % (8 * depth);
		newdata[newbyte] = gb->data[i];
	}

	free(gb->data);

	gb->data = newdata;
}

void
png_to_gb(struct PNGImage png, struct GBImage *gb)
{
	int x, y, byte;
	png_byte index;

	for (y = 0; y < png.height; y++) {
		for (x = 0; x < png.width; x++) {
			index = png.data[y][x];
			index &= (1 << depth) - 1;

			if (!gb->horizontal) {
				byte = y * depth + x / 8 * png.height / 8 * 8 * depth;
			} else {
				byte = y * depth + x / 8 * png.height / 8 * 8 * depth;
			}
			gb->data[byte] |= (index & 1) << (7 - x % 8);
			if (depth == 2) {
				gb->data[byte + 1] |= (index >> 1) << (7 - x % 8);
			}
		}
	}

	if (!gb->horizontal) {
		transpose_tiles(gb, png.width / 8);
	}
}

void
output_file(struct Options opts, struct GBImage gb)
{
	FILE *f;

	f = fopen(opts.outfile, "wb");
	if (!f) {
		err(1, "Opening output file '%s' failed", opts.outfile);
	}
	fwrite(gb.data, 1, gb.size - gb.trim * 8 * depth, f);

	fclose(f);
}

int
get_tile_index(uint8_t *tile, uint8_t **tiles, int num_tiles, int tile_size)
{
	int i, j;
	for (i = 0; i < num_tiles; i++) {
		for (j = 0; j < tile_size; j++) {
			if (tile[j] != tiles[i][j]) {
				break;
			}
		}
		if (j >= tile_size) {
			return i;
		}
	}
	return -1;
}

void
create_tilemap(struct Options opts, struct GBImage *gb, struct Tilemap *tilemap)
{
	int i, j;
	int gb_i;
	int tile_size;
	int max_tiles;
	int num_tiles;
	int index;
	int gb_size;
	uint8_t *tile;
	uint8_t **tiles;

	tile_size = sizeof(uint8_t) * depth * 8;
	gb_size = gb->size - (gb->trim * tile_size);
	max_tiles = gb_size / tile_size;
	tiles = malloc(sizeof(uint8_t*) * max_tiles);
	num_tiles = 0;

	tilemap->data = malloc(sizeof(uint8_t) * max_tiles);
	tilemap->size = 0;

	gb_i = 0;
	while (gb_i < gb_size) {
		tile = malloc(tile_size);
		for (i = 0; i < tile_size; i++) {
			tile[i] = gb->data[gb_i];
			gb_i++;
		}
		if (opts.unique) {
			index = get_tile_index(tile, tiles, num_tiles, tile_size);
			if (index < 0) {
				index = num_tiles;
				tiles[num_tiles] = tile;
				num_tiles++;
			}
		} else {
			index = num_tiles;
			tiles[num_tiles] = tile;
			num_tiles++;
		}
		tilemap->data[tilemap->size] = index;
		tilemap->size++;
	}

	if (opts.unique) {
		free(gb->data);
		gb->data = malloc(tile_size * num_tiles);
		for (i = 0; i < num_tiles; i++) {
			tile = tiles[i];
			for (j = 0; j < tile_size; j++) {
				gb->data[i * tile_size + j] = tile[j];
			}
		}
		gb->size = i * tile_size;
	}

	for (i = 0; i < num_tiles; i++) {
		free(tiles[i]);
	}
	free(tiles);
}

void
output_tilemap_file(struct Options opts, struct Tilemap tilemap)
{
	FILE *f;

	f = fopen(opts.mapfile, "wb");
	if (!f) {
		err(1, "Opening tilemap file '%s' failed", opts.mapfile);
	}

	fwrite(tilemap.data, 1, tilemap.size, f);
	fclose(f);

	if (opts.mapout) {
		free(opts.mapfile);
	}
}

void
output_palette_file(struct Options opts, struct PNGImage png)
{
	FILE *f;
	int i, colors, color;
	png_color *palette;

	if (png_get_PLTE(png.png, png.info, &palette, &colors)) {
		f = fopen(opts.palfile, "wb");
		if (!f) {
			err(1, "Opening palette file '%s' failed", opts.palfile);
		}
		for (i = 0; i < colors; i++) {
			color = palette[i].blue >> 3 << 10 | palette[i].green >> 3 << 5 | palette[i].red >> 3;
			fwrite(&color, 2, 1, f);
		}
		fclose(f);
	}

	if (opts.palout) {
		free(opts.palfile);
	}
}