aboutsummaryrefslogtreecommitdiffstats
path: root/src/vu-crop.c
blob: 2fcd6f7b3bbfd6ec8d724a32d216012cf84db37b (plain) (blame)
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
/* See LICENSE file for copyright and license details. */
#include "arg.h"
#include "stream.h"
#include "util.h"

#include <fcntl.h>
#include <inttypes.h>
#include <string.h>
#include <unistd.h>

static void
usage(void)
{
	eprintf("usage: [-t] width height left top %s\n", argv0);
}

int
main(int argc, char *argv[])
{
	struct stream stream;
	char *buf, *image, *p;
	size_t width = 0, height = 0, left = 0, top = 0;
	size_t off, yoff, x, y, irown, orown, ptr, n, m;
	ssize_t r;
	int tile = 0;

	ARGBEGIN {
	case 't':
		tile = 1;
		break;
	default:
		usage();
	} ARGEND;

	if (argc != 4)
		usage();

	if (tozu(argv[0], 1, SIZE_MAX, &width))
		eprintf("width be an integer in [1, %zu]\n", SIZE_MAX);
	if (tozu(argv[1], 1, SIZE_MAX, &height))
		eprintf("height be an integer in [1, %zu]\n", SIZE_MAX);
	if (tozu(argv[2], 0, SIZE_MAX, &left))
		eprintf("left position be an integer in [0, %zu]\n", SIZE_MAX);
	if (tozu(argv[3], 0, SIZE_MAX, &top))
		eprintf("top position be an integer in [0, %zu]\n", SIZE_MAX);

	stream.file = "<stdin>";
	stream.fd = STDIN_FILENO;
	einit_stream(&stream);
	if (left + width > stream.width || top + height > stream.height)
		eprintf("crop area extends beyond original image\n");
	if (tile) {
		fprint_stream_head(stdout, &stream);
	} else {
		x = stream.width,  stream.width  = width;
		y = stream.height, stream.height = height;
		fprint_stream_head(stdout, &stream);
		stream.width  = x;
		stream.height = y;
	}
	fflush(stdout);
	if (ferror(stdout))
		eprintf("<stdout>:");

	if (stream.width > SIZE_MAX / stream.pixel_size)
		eprintf("<stdin>: video frame is too large\n");
	n = irown = stream.width * stream.pixel_size;
	if (n > SIZE_MAX / stream.height)
		eprintf("<stdin>: video frame is too large\n");
	n *= stream.height;
	if (!(buf = malloc(n)))
		eprintf("malloc:");
	orown = width * stream.pixel_size;
	m = tile ? n : height * orown;
	if (!(image = malloc(m)))
		eprintf("malloc:");

	left *= stream.pixel_size;
	if (tile) {
		off  = (orown  - (left % orown))  % orown;
		yoff = (height - (top  % height)) % height;
	} else {
		off = top * irown;
	}
	memcpy(buf, stream.buf, ptr = stream.ptr);
	for (;;) {
		for (; ptr < n; ptr += (size_t)r) {
			r = read(stream.fd, buf + ptr, n - ptr);
			if (r < 0) {
				eprintf("read %s:", stream.file);
			} else if (r == 0) {
				if (!ptr)
					break;
				eprintf("%s: incomplete frame", stream.file);
			}
		}
		if (!ptr)
			break;

		if (!tile) {
			for (y = 0; y < height; y++)
				memcpy(image + y * orown, buf + y * irown + off, orown);
		} else {
			for (ptr = y = 0; y < stream.height; y++) {
				p = buf + ((y + yoff) % height) * irown + left;
				for (x = 0; x < irown; x++, ptr++)
					image[ptr++] = p[(x + off) % orown];
			}
			
		}

		for (ptr = 0; ptr < m; ptr += (size_t)r) {
			r = write(STDOUT_FILENO, image + ptr, m - ptr);
			if (r < 0)
				eprintf("write <stdout>");
		}
	}

	return 0;
}