2017-06-18 04:09:25 +00:00
|
|
|
#ifndef GUARD_COMMON_H
|
|
|
|
#define GUARD_COMMON_H
|
|
|
|
|
2021-09-02 04:21:10 +00:00
|
|
|
#include <stdio.h>
|
|
|
|
#include <stdlib.h>
|
|
|
|
#include <stdint.h>
|
|
|
|
#include <stdbool.h>
|
|
|
|
#include <inttypes.h>
|
|
|
|
#include <string.h>
|
|
|
|
#include <errno.h>
|
|
|
|
#include <unistd.h>
|
|
|
|
#include <getopt.h>
|
2017-06-18 04:09:25 +00:00
|
|
|
|
2021-09-21 21:37:43 +00:00
|
|
|
#ifndef PROGRAM_NAME
|
|
|
|
#error Define PROGRAM_NAME before including common.h!
|
|
|
|
#endif
|
|
|
|
#ifndef USAGE_OPTS
|
|
|
|
#error Define USAGE_OPTS before including common.h!
|
|
|
|
#endif
|
|
|
|
|
|
|
|
#define error_exit(...) exit((fprintf(stderr, PROGRAM_NAME ": " __VA_ARGS__), 1))
|
|
|
|
|
|
|
|
void usage_exit(int status) {
|
|
|
|
fprintf(stderr, "Usage: " PROGRAM_NAME " " USAGE_OPTS "\n");
|
|
|
|
exit(status);
|
|
|
|
}
|
2021-09-02 07:04:40 +00:00
|
|
|
|
2021-09-02 04:21:10 +00:00
|
|
|
int getopt_long_index;
|
2021-09-02 05:21:15 +00:00
|
|
|
#define getopt_long(argc, argv, optstring, longopts) getopt_long(argc, argv, optstring, longopts, &getopt_long_index)
|
2021-09-02 04:21:10 +00:00
|
|
|
|
|
|
|
void *malloc_verbose(size_t size) {
|
2021-09-02 05:21:15 +00:00
|
|
|
errno = 0;
|
2021-09-02 04:21:10 +00:00
|
|
|
void *m = malloc(size);
|
|
|
|
if (!m) {
|
2021-09-02 07:04:40 +00:00
|
|
|
error_exit("Could not allocate %zu bytes: %s\n", size, strerror(errno));
|
2021-09-02 04:21:10 +00:00
|
|
|
}
|
|
|
|
return m;
|
|
|
|
}
|
|
|
|
|
2021-09-02 22:37:36 +00:00
|
|
|
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;
|
|
|
|
}
|
|
|
|
|
2021-09-02 04:21:10 +00:00
|
|
|
FILE *fopen_verbose(const char *filename, char rw) {
|
|
|
|
char mode[3] = {rw, 'b', '\0'};
|
2021-09-02 05:21:15 +00:00
|
|
|
errno = 0;
|
2017-06-18 04:09:25 +00:00
|
|
|
FILE *f = fopen(filename, mode);
|
|
|
|
if (!f) {
|
2021-09-02 07:04:40 +00:00
|
|
|
error_exit("Could not open file \"%s\": %s\n", filename, strerror(errno));
|
2017-06-18 04:09:25 +00:00
|
|
|
}
|
|
|
|
return f;
|
|
|
|
}
|
|
|
|
|
2021-09-02 04:21:10 +00:00
|
|
|
void fread_verbose(uint8_t *data, size_t size, const char *filename, FILE *f) {
|
2021-09-02 05:21:15 +00:00
|
|
|
errno = 0;
|
2021-09-02 04:21:10 +00:00
|
|
|
if (fread(data, 1, size, f) != size) {
|
|
|
|
fclose(f);
|
2021-09-02 07:04:40 +00:00
|
|
|
error_exit("Could not read from file \"%s\": %s\n", filename, strerror(errno));
|
2017-06-18 04:09:25 +00:00
|
|
|
}
|
2021-09-02 04:21:10 +00:00
|
|
|
}
|
|
|
|
|
|
|
|
void fwrite_verbose(const uint8_t *data, size_t size, const char *filename, FILE *f) {
|
2021-09-02 05:21:15 +00:00
|
|
|
errno = 0;
|
2021-09-02 04:21:10 +00:00
|
|
|
if (fwrite(data, 1, size, f) != size) {
|
|
|
|
fclose(f);
|
2021-09-02 07:04:40 +00:00
|
|
|
error_exit("Could not write to file \"%s\": %s\n", filename, strerror(errno));
|
2017-12-28 06:25:25 +00:00
|
|
|
}
|
2021-09-02 04:21:10 +00:00
|
|
|
}
|
|
|
|
|
2021-09-02 05:21:15 +00:00
|
|
|
long file_size_verbose(const char *filename, FILE *f) {
|
|
|
|
long size = -1;
|
|
|
|
errno = 0;
|
2021-09-02 04:21:10 +00:00
|
|
|
if (!fseek(f, 0, SEEK_END)) {
|
|
|
|
size = ftell(f);
|
|
|
|
if (size != -1) {
|
|
|
|
rewind(f);
|
|
|
|
}
|
|
|
|
}
|
2021-09-02 05:21:15 +00:00
|
|
|
if (size == -1) {
|
2021-09-02 07:04:40 +00:00
|
|
|
error_exit("Could not measure file \"%s\": %s\n", filename, strerror(errno));
|
2021-09-02 04:21:10 +00:00
|
|
|
}
|
|
|
|
return size;
|
|
|
|
}
|
|
|
|
|
|
|
|
uint8_t *read_u8(const char *filename, long *size) {
|
|
|
|
FILE *f = fopen_verbose(filename, 'r');
|
2021-09-02 05:21:15 +00:00
|
|
|
*size = file_size_verbose(filename, f);
|
2021-09-02 04:21:10 +00:00
|
|
|
uint8_t *data = malloc_verbose(*size);
|
|
|
|
fread_verbose(data, *size, filename, f);
|
2017-06-18 04:09:25 +00:00
|
|
|
fclose(f);
|
|
|
|
return data;
|
|
|
|
}
|
|
|
|
|
2021-09-02 04:21:10 +00:00
|
|
|
void write_u8(const char *filename, uint8_t *data, size_t size) {
|
|
|
|
FILE *f = fopen_verbose(filename, 'w');
|
|
|
|
fwrite_verbose(data, size, filename, f);
|
|
|
|
fclose(f);
|
|
|
|
}
|
|
|
|
|
|
|
|
uint32_t read_png_width_verbose(const char *filename) {
|
|
|
|
FILE *f = fopen_verbose(filename, 'r');
|
|
|
|
uint8_t header[16] = {0};
|
|
|
|
fread_verbose(header, sizeof(header), filename, f);
|
|
|
|
static uint8_t expected_header[16] = {
|
|
|
|
0x89, 'P', 'N', 'G', '\r', '\n', 0x1A, '\n', // signature
|
|
|
|
0, 0, 0, 13, // IHDR chunk length
|
|
|
|
'I', 'H', 'D', 'R', // IHDR chunk type
|
|
|
|
};
|
|
|
|
if (memcmp(header, expected_header, sizeof(header))) {
|
2017-06-18 04:09:25 +00:00
|
|
|
fclose(f);
|
2021-09-02 07:04:40 +00:00
|
|
|
error_exit("Not a valid PNG file: \"%s\"\n", filename);
|
2017-06-18 04:09:25 +00:00
|
|
|
}
|
2021-09-02 04:21:10 +00:00
|
|
|
uint8_t bytes[4] = {0};
|
|
|
|
fread_verbose(bytes, sizeof(bytes), filename, f);
|
|
|
|
fclose(f);
|
|
|
|
return (bytes[0] << 24) | (bytes[1] << 16) | (bytes[2] << 8) | bytes[3];
|
2017-06-18 04:09:25 +00:00
|
|
|
}
|
|
|
|
|
2021-09-02 22:37:36 +00:00
|
|
|
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);
|
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2017-06-18 04:09:25 +00:00
|
|
|
#endif // GUARD_COMMON_H
|