summaryrefslogtreecommitdiff
path: root/ff2ppm.c
blob: 7f6aa6124e4d7a63d84c306895789f1ec020770b (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
/* See LICENSE file for copyright and license details. */
#include <arpa/inet.h>

#include <errno.h>
#include <inttypes.h>
#include <stdint.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>

#include "arg.h"

char *argv0;

static void
usage(void)
{
	fprintf(stderr, "usage: %s [-b #rrggbb]\n", argv0);
	exit(1);
}

int
main(int argc, char *argv[])
{
	size_t rowlen;
	uint32_t hdr[4], width, height, i, j, k;
	uint16_t *row, mr = 0xffff, mg = 0xffff, mb = 0xffff;
	uint8_t *rowout;
	char *color;
	unsigned int r = 0xff, g = 0xff, b = 0xff;
	float a;

	argv0 = argv[0];
	ARGBEGIN {
	case 'b':
		for (color = EARGF(usage()); *color && *color == '#'; color++)
			;

		switch (strlen(color)) {
		case 3:
			if (sscanf(color, "%1x%1x%1x", &r, &g, &b) != 3)
				usage();
			mr = (r | r << 4) * 257;
			mg = (g | g << 4) * 257;
			mb = (b | b << 4) * 257;
			break;
		case 6:
			if (sscanf(color, "%2x%2x%2x", &r, &g, &b) != 3)
				usage();
			mr = r * 257;
			mg = g * 257;
			mb = b * 257;
			break;
		case 12:
			if (sscanf(color, "%4x%4x%4x", &r, &g, &b) != 3)
				usage();
			mr = r;
			mg = g;
			mb = b;
			break;
		default:
			usage();
		}
		break;
	default:
		usage();
	} ARGEND

	if (argc)
		usage();

	/* header */
	if (fread(hdr, sizeof(*hdr), 4, stdin) != 4) {
		fprintf(stderr, "%s: fread: %s\n", argv0, strerror(errno));
		return 1;
	}
	if (memcmp("farbfeld", hdr, sizeof("farbfeld") - 1)) {
		fprintf(stderr, "%s: invalid magic value\n", argv0);
		return 1;
	}
	if (!(width = ntohl(hdr[2]))) {
		fprintf(stderr, "%s: invalid width: zero\n", argv0);
		return 1;
	}
	if (!(height = ntohl(hdr[3]))) {
		fprintf(stderr, "%s: invalid height: zero\n", argv0);
		return 1;
	}
	if (width > SIZE_MAX / ((sizeof("RGBA") - 1) * sizeof(uint16_t))) {
		fprintf(stderr, "%s: row length integer overflow\n", argv0);
		return 1;
	}

	rowlen = width * (sizeof("RGBA") - 1);
	if (!(row = malloc(rowlen * sizeof(uint16_t)))) {
		fprintf(stderr, "%s: malloc: %s\n", argv0, strerror(errno));
		return 1;
	}
	if (!(rowout = malloc(width * sizeof("RGB") - 1))) {
		fprintf(stderr, "%s: malloc: %s\n", argv0, strerror(errno));
		return 1;
	}

	/* PPM binary */
	printf("P6\n%" PRIu32 " %" PRIu32 "\n255\n", width, height);

	/* write rows */
	for (i = 0; i < height; ++i) {
		if (fread(row, sizeof(uint16_t), rowlen, stdin) != rowlen) {
			fprintf(stderr, "%s: fread: %s\n", argv0, strerror(errno));
			return 1;
		}
		for (j = 0, k = 0; j < rowlen; j += 4, k += 3) {
			a = ntohs(row[j + 3]) / 65535.0f;
			rowout[k]     = ((ntohs(row[j]) * a)     + (mr * (1 - a))) / 257;
			rowout[k + 1] = ((ntohs(row[j + 1]) * a) + (mg * (1 - a))) / 257;
			rowout[k + 2] = ((ntohs(row[j + 2]) * a) + (mb * (1 - a))) / 257;
		}
		if (fwrite(rowout, 3, width, stdout) != width) {
			fprintf(stderr, "%s: fwrite: %s\n", argv0, strerror(errno));
			return 1;
		}
	}
	return 0;
}