Rewrite tools/pokemon_animation.c

This commit is contained in:
Rangi 2021-09-02 18:37:36 -04:00
parent a18f83b911
commit 021702628c
4 changed files with 142 additions and 214 deletions

View File

@ -25,6 +25,15 @@ void *malloc_verbose(size_t size) {
return m; return m;
} }
void *calloc_verbose(size_t size) {
errno = 0;
void *m = calloc(size, 1);
if (!m) {
error_exit("Could not allocate %zu bytes: %s\n", size, strerror(errno));
}
return m;
}
FILE *fopen_verbose(const char *filename, char rw) { FILE *fopen_verbose(const char *filename, char rw) {
char mode[3] = {rw, 'b', '\0'}; char mode[3] = {rw, 'b', '\0'};
errno = 0; errno = 0;
@ -100,4 +109,19 @@ uint32_t read_png_width_verbose(const char *filename) {
return (bytes[0] << 24) | (bytes[1] << 16) | (bytes[2] << 8) | bytes[3]; return (bytes[0] << 24) | (bytes[1] << 16) | (bytes[2] << 8) | bytes[3];
} }
void read_dimensions(const char *filename, int *width) {
long filesize;
uint8_t *bytes = read_u8(filename, &filesize);
if (filesize != 1) {
error_exit("%s: invalid dimensions file\n", filename);
}
uint8_t dimensions = bytes[0];
free(bytes);
*width = dimensions & 0xF;
int height = dimensions >> 4;
if (*width != height || (*width != 5 && *width != 6 && *width != 7)) {
error_exit("%s: invalid dimensions: %dx%d tiles\n", filename, *width, height);
}
}
#endif // GUARD_COMMON_H #endif // GUARD_COMMON_H

View File

@ -1,10 +1,10 @@
#include "common.h" #include "common.h"
void usage() { void usage() {
fputs("Usage: png_dimensions in.png out.dimensions\n", stderr); fputs("Usage: png_dimensions front.png front.dimensions\n", stderr);
} }
uint8_t read_dimensions(const char *filename) { uint8_t read_png_dimensions(const char *filename) {
uint32_t width_px = read_png_width_verbose(filename); uint32_t width_px = read_png_width_verbose(filename);
if (width_px != 40 && width_px != 48 && width_px != 56) { if (width_px != 40 && width_px != 48 && width_px != 56) {
error_exit("Not a valid width for \"%s\": %" PRIu32 " px\n", filename, width_px); error_exit("Not a valid width for \"%s\": %" PRIu32 " px\n", filename, width_px);
@ -19,7 +19,7 @@ int main(int argc, char *argv[]) {
exit(1); exit(1);
} }
uint8_t output_byte = read_dimensions(argv[1]); uint8_t output_byte = read_png_dimensions(argv[1]);
write_u8(argv[2], &output_byte, 1); write_u8(argv[2], &output_byte, 1);
return 0; return 0;
} }

View File

@ -1,112 +1,104 @@
#include <stdio.h> #include "common.h"
#include <unistd.h>
#include <stdlib.h> struct Options {
#include <string.h> bool use_bitmasks;
#include <stdint.h> bool use_frames;
#include <stdbool.h> };
#include <getopt.h>
void usage() {
fputs("Usage: pokemon_animation [-b|--bitmasks] [-f|--frames] front.animated.tilemap front.dimensions\n", stderr);
}
void parse_args(int argc, char *argv[], struct Options *options) {
struct option long_options[] = {
{"bitmasks", no_argument, 0, 'b'},
{"frames", no_argument, 0, 'f'},
{"help", no_argument, 0, 'h'},
{0}
};
for (int opt; (opt = getopt_long(argc, argv, "bfh", long_options)) != -1;) {
switch (opt) {
case 'b':
options->use_bitmasks = true;
break;
case 'f':
options->use_frames = true;
break;
case 'h':
usage();
exit(0);
break;
default:
usage();
exit(1);
}
}
}
struct Frame { struct Frame {
uint8_t* data; uint8_t *data;
int size; int size;
int bitmask; int bitmask;
}; };
struct Frames { struct Frames {
struct Frame* frames; struct Frame *frames;
int num_frames; int num_frames;
int frame_size;
}; };
struct Bitmask { struct Bitmask {
uint8_t* data; uint8_t *data;
int bitlength; int bitlength;
}; };
struct Bitmasks { struct Bitmasks {
struct Bitmask* bitmasks; struct Bitmask *bitmasks;
int num_bitmasks; int num_bitmasks;
}; };
int bitmask_exists(const struct Bitmask *bitmask, const struct Bitmasks *bitmasks) {
void make_frames(struct Frames* frames, struct Bitmasks* bitmasks, char* tilemap_filename, char* dimensions_filename); for (int i = 0; i < bitmasks->num_bitmasks; i++) {
int bitmask_exists(struct Bitmask *bitmask, struct Bitmasks *bitmasks); struct Bitmask existing = bitmasks->bitmasks[i];
void print_frames(struct Frames* frames); if (bitmask->bitlength != existing.bitlength) {
continue;
}
void make_frames(struct Frames* frames, struct Bitmasks* bitmasks, char* tilemap_filename, char* dimensions_filename) { bool match = true;
uint8_t* tilemap; int length = (bitmask->bitlength + 7) / 8;
uint8_t* this_frame; for (int j = 0; j < length; j++) {
FILE* f; if (bitmask->data[j] != existing.data[j]) {
size_t size; match = false;
int width; break;
int height; }
uint8_t byte; }
int frame_size; if (match) {
int num_frames; return i;
int i, j; }
f = fopen(tilemap_filename, "rb");
if (f == NULL) {
fprintf(stderr, "could not open file %s\n", tilemap_filename);
exit(1);
} }
return -1;
}
fseek(f, 0, SEEK_END); void make_frames(const uint8_t *tilemap, long tilemap_size, int width, struct Frames *frames, struct Bitmasks *bitmasks) {
size = ftell(f); int num_tiles_per_frame = width * width;
if (!size) { int num_frames = tilemap_size / num_tiles_per_frame - 1;
fprintf(stderr, "empty file %s\n", tilemap_filename);
exit(1);
}
rewind(f);
tilemap = malloc(size); frames->frames = malloc_verbose((sizeof *frames->frames) * num_frames);
if (!tilemap) { frames->num_frames = num_frames;
fprintf(stderr, "malloc failure\n");
exit(1);
}
if (size != fread(tilemap, 1, size, f)) {
fprintf(stderr, "failed to read file %s\n", tilemap_filename);
exit(1);
}
fclose(f);
f = fopen(dimensions_filename, "rb"); bitmasks->bitmasks = malloc_verbose((sizeof *bitmasks->bitmasks) * num_frames);
if (f == NULL) {
fprintf(stderr, "could not open file %s\n", dimensions_filename);
exit(1);
}
if (1 != fread(&byte, 1, 1, f)) {
fprintf(stderr, "failed to read file %s\n", dimensions_filename);
exit(1);
}
fclose(f);
width = byte & 0xf;
height = byte >> 4;
frame_size = width * height;
num_frames = size / frame_size - 1;
//fprintf(stderr, "num_frames: %d\n", num_frames);
bitmasks->bitmasks = malloc((sizeof (struct Bitmask)) * num_frames);
bitmasks->num_bitmasks = 0; bitmasks->num_bitmasks = 0;
frames->frames = malloc((sizeof (struct Frame)) * num_frames); const uint8_t *first_frame = &tilemap[0];
frames->frame_size = frame_size; const uint8_t *this_frame = &tilemap[num_tiles_per_frame];
frames->num_frames = 0; for (int i = 0; i < num_frames; i++) {
struct Frame *frame = malloc_verbose(sizeof *frame);
uint8_t *first_frame = tilemap; frame->data = malloc_verbose(num_tiles_per_frame);
this_frame = tilemap + frame_size;
for (i = 0; i < num_frames; i++) {
struct Frame *frame = (struct Frame*)malloc(sizeof(struct Frame));
frame->data = malloc(frame_size);
frame->size = 0; frame->size = 0;
struct Bitmask *bitmask = (struct Bitmask*)malloc(sizeof(struct Bitmask));
bitmask->data = calloc((frame_size + 7) / 8, 1); struct Bitmask *bitmask = malloc_verbose(sizeof *bitmask);
bitmask->data = calloc_verbose((num_tiles_per_frame + 7) / 8);
bitmask->bitlength = 0; bitmask->bitlength = 0;
for (j = 0; j < frame_size; j++) {
for (int j = 0; j < num_tiles_per_frame; j++) {
if (bitmask->bitlength % 8 == 0) { if (bitmask->bitlength % 8 == 0) {
bitmask->data[bitmask->bitlength / 8] = 0; bitmask->data[bitmask->bitlength / 8] = 0;
} }
@ -133,151 +125,76 @@ void make_frames(struct Frames* frames, struct Bitmasks* bitmasks, char* tilemap
free(bitmask); free(bitmask);
} }
frames->frames[i] = *frame; frames->frames[i] = *frame;
frames->num_frames++; this_frame += num_tiles_per_frame;
this_frame += frame_size;
} }
//for (i = 0; i < frames->num_frames; i++) {
//free(frames->frames[i].data);
//free(frames->frames[i]);
//}
//free(frames->frames);
//fprintf(stderr, "num bitmasks: %d\n", bitmasks->num_bitmasks);
//for (i = 0; i < bitmasks->num_bitmasks; i++) {
// free(bitmasks->bitmasks[i].data);
// fprintf(stderr, "freed bitmask %d\n", i);
//free(bitmasks->bitmasks[i]);
//}
//free(bitmasks->bitmasks);
//fprintf(stderr, "freed bitmasks\n");
free(tilemap);
} }
int bitmask_exists(struct Bitmask *bitmask, struct Bitmasks *bitmasks) { void print_frames(struct Frames *frames) {
int i, j; for (int i = 0; i < frames->num_frames; i++) {
struct Bitmask existing;
for (i = 0; i < bitmasks->num_bitmasks; i++) {
existing = bitmasks->bitmasks[i];
if (bitmask->bitlength != existing.bitlength) {
continue;
}
bool match = true;
for (j = 0; j < (bitmask->bitlength + 7) / 8; j++) {
if (bitmask->data[j] != existing.data[j]) {
match = false;
break;
}
}
if (match) {
return i;
}
}
return -1;
}
void print_frames(struct Frames* frames) {
int i;
int j;
for (i = 0; i < frames->num_frames; i++) {
printf("\tdw .frame%d\n", i + 1); printf("\tdw .frame%d\n", i + 1);
} }
for (i = 0; i < frames->num_frames; i++) { for (int i = 0; i < frames->num_frames; i++) {
struct Frame *frame = &frames->frames[i]; const struct Frame *frame = &frames->frames[i];
printf(".frame%d\n", i + 1); printf(".frame%d\n", i + 1);
printf("\tdb $%02x ; bitmask\n", frame->bitmask); printf("\tdb $%02x ; bitmask\n", frame->bitmask);
if (frame->size > 0) { if (frame->size > 0) {
for (j = 0; j < frame->size; j++) { for (int j = 0; j < frame->size; j++) {
if (j % 12 == 0) { if (j % 12 == 0) {
if (j) { if (j) {
printf("\n"); putchar('\n');
} }
printf("\tdb $%02x", frame->data[j]); printf("\tdb $%02x", frame->data[j]);
} else { } else {
printf(", $%02x", frame->data[j]); printf(", $%02x", frame->data[j]);
} }
} }
printf("\n"); putchar('\n');
} }
} }
} }
void print_bitmasks(struct Bitmasks* bitmasks) { void print_bitmasks(const struct Bitmasks *bitmasks) {
int i, j, k; for (int i = 0; i < bitmasks->num_bitmasks; i++) {
int length; struct Bitmask bitmask = bitmasks->bitmasks[i];
struct Bitmask bitmask;
for (i = 0; i < bitmasks->num_bitmasks; i++) {
printf("; %d\n", i); printf("; %d\n", i);
bitmask = bitmasks->bitmasks[i]; int length = (bitmask.bitlength + 7) / 8;
length = (bitmask.bitlength + 7) / 8; for (int j = 0; j < length; j++) {
for (j = 0; j < length; j++) {
printf("\tdb %%"); printf("\tdb %%");
for (k = 0; k < 8; k++) { for (int k = 0; k < 8; k++) {
if ((bitmask.data[j] >> (7 - k)) & 1) { putchar(((bitmask.data[j] >> (7 - k)) & 1) ? '1' : '0');
printf("1");
} else {
printf("0");
};
} }
printf("\n"); putchar('\n');
} }
} }
} }
// HOW ARE YOU GENTLEMEN. int main(int argc, char *argv[]) {
char* cats (char* head, char* tail) { struct Options options = {0};
char* string; parse_args(argc, argv, &options);
string = malloc(strlen(head) + strlen(tail) + 1);
strcpy(string, head);
strcat(string, tail);
return string;
}
static void usage(void) {
printf("Usage: pokemon_animation [-b] [-f] tilemap_file dimensions_file\n");
exit(1);
}
int main(int argc, char* argv[]) {
struct Frames frames = {0};
struct Bitmasks bitmasks = {0};
int ch;
bool use_bitmasks = false, use_frames = false;
char* tilemap_filename;
char* dimensions_filename;
while ((ch = getopt(argc, argv, "bf")) != -1) {
switch (ch) {
case 'b':
use_bitmasks = true;
break;
case 'f':
use_frames = true;
break;
default:
usage();
}
}
argc -= optind; argc -= optind;
argv += optind; argv += optind;
if (argc < 2) { if (argc < 2) {
usage(); usage();
exit(1);
} }
tilemap_filename = argv[0];
dimensions_filename = argv[1];
//ext = strrchr(argv[3], '.'); int width;
//if (!ext || ext == argv[3]) { read_dimensions(argv[1], &width);
// fprintf(stderr, "need a file extension to determine what to write to %s\n", argv[3]); long tilemap_size;
//} uint8_t *tilemap = read_u8(argv[0], &tilemap_size);
make_frames(&frames, &bitmasks, tilemap_filename, dimensions_filename); struct Frames frames = {0};
if (use_frames) { struct Bitmasks bitmasks = {0};
make_frames(tilemap, tilemap_size, width, &frames, &bitmasks);
if (options.use_frames) {
print_frames(&frames); print_frames(&frames);
} }
if (use_bitmasks) { if (options.use_bitmasks) {
print_bitmasks(&bitmasks); print_bitmasks(&bitmasks);
} }
free(tilemap);
return 0; return 0;
} }

View File

@ -67,19 +67,8 @@ int get_tile_index(const uint8_t *tile, const uint8_t *tiles, int num_tiles, int
return -1; return -1;
} }
uint8_t read_dimensions(const char *filename) { uint8_t *read_tiles(const char *filename, int width, long *tiles_size) {
long filesize; int frame_size = width * width * TILE_SIZE;
uint8_t *bytes = read_u8(filename, &filesize);
if (filesize != 1) {
error_exit("%s: invalid dimensions file\n", filename);
}
uint8_t dimensions = bytes[0];
free(bytes);
return dimensions;
}
uint8_t *read_tiles(const char *filename, int width, int height, long *tiles_size) {
int frame_size = width * height * TILE_SIZE;
uint8_t *tiles = read_u8(filename, tiles_size); uint8_t *tiles = read_u8(filename, tiles_size);
if (!*tiles_size) { if (!*tiles_size) {
@ -87,7 +76,7 @@ uint8_t *read_tiles(const char *filename, int width, int height, long *tiles_siz
} else if (*tiles_size % TILE_SIZE) { } else if (*tiles_size % TILE_SIZE) {
error_exit("%s: not divisible into 8x8-px 2bpp tiles\n", filename); error_exit("%s: not divisible into 8x8-px 2bpp tiles\n", filename);
} else if (*tiles_size % frame_size) { } else if (*tiles_size % frame_size) {
error_exit("%s: not divisible into %dx%d-tile frames\n", filename, width, height); error_exit("%s: not divisible into %dx%d-tile frames\n", filename, width, width);
} }
int num_frames = *tiles_size / frame_size; int num_frames = *tiles_size / frame_size;
@ -169,18 +158,16 @@ int main(int argc, char *argv[]) {
exit(1); exit(1);
} }
uint8_t dimensions = read_dimensions(argv[1]); int width;
int width = dimensions & 0xF; read_dimensions(argv[1], &width);
int height = dimensions >> 4;
long tiles_size; long tiles_size;
uint8_t *tiles = read_tiles(argv[0], width, height, &tiles_size); uint8_t *tiles = read_tiles(argv[0], width, &tiles_size);
if (options.out_filename) { if (options.out_filename) {
write_graphics(options.out_filename, tiles, tiles_size, width * height, options.girafarig); write_graphics(options.out_filename, tiles, tiles_size, width * width, options.girafarig);
} }
if (options.map_filename) { if (options.map_filename) {
write_tilemap(options.map_filename, tiles, tiles_size, width * height, options.girafarig); write_tilemap(options.map_filename, tiles, tiles_size, width * width, options.girafarig);
} }
free(tiles); free(tiles);