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
126
127
128
129
|
/* See LICENSE file for copyright and license details. */
#include "stream.h"
#include "util.h"
#include <signal.h>
#include <stdio.h>
#include <stdint.h>
#include <string.h>
USAGE("[-d] frame-rate ffmpeg-arguments ...")
static int draft = 0;
static int fd;
#define PROCESS(TYPE, SUFFIX)\
do {\
char *buf = stream->buf;\
TYPE *pixel, r, g, b;\
uint16_t *pixels, *end;\
uint16_t pixbuf[BUFSIZ / sizeof(uint16_t)];\
long int a, y, u, v;\
size_t ptr;\
pixels = pixbuf;\
end = pixbuf + ELEMENTSOF(pixbuf);\
if (draft) {\
for (ptr = 0; ptr < n; ptr += 4 * sizeof(TYPE)) {\
pixel = (TYPE *)(buf + ptr);\
ciexyz_to_scaled_yuv##SUFFIX(pixel[0], pixel[1], pixel[2], &r, &g, &b);\
y = (long int)r + 16L * 256L;\
u = (long int)g + 128L * 256L;\
v = (long int)b + 128L * 256L;\
*pixels++ = 0xFFFFU;\
*pixels++ = htole16((uint16_t)CLIP(0, y, 0xFFFFL));\
*pixels++ = htole16((uint16_t)CLIP(0, u, 0xFFFFL));\
*pixels++ = htole16((uint16_t)CLIP(0, v, 0xFFFFL));\
if (pixels == end)\
ewriteall(fd, pixels = pixbuf, sizeof(pixbuf), "<subprocess>");\
}\
} else {\
for (ptr = 0; ptr < n; ptr += 4 * sizeof(TYPE)) {\
pixel = (TYPE *)(buf + ptr);\
a = (long int)(pixel[3] * 0xFFFFL);\
ciexyz_to_srgb##SUFFIX(pixel[0], pixel[1], pixel[2], &r, &g, &b);\
r = srgb_encode##SUFFIX(r);\
g = srgb_encode##SUFFIX(g);\
b = srgb_encode##SUFFIX(b);\
srgb_to_yuv##SUFFIX(r, g, b, pixel + 0, pixel + 1, pixel + 2);\
y = (long int)(pixel[0] * 0xFFFFL) + 16L * 256L;\
u = (long int)(pixel[1] * 0xFFFFL) + 128L * 256L;\
v = (long int)(pixel[2] * 0xFFFFL) + 128L * 256L;\
*pixels++ = htole16((uint16_t)CLIP(0, a, 0xFFFFL));\
*pixels++ = htole16((uint16_t)CLIP(0, y, 0xFFFFL));\
*pixels++ = htole16((uint16_t)CLIP(0, u, 0xFFFFL));\
*pixels++ = htole16((uint16_t)CLIP(0, v, 0xFFFFL));\
if (pixels == end)\
ewriteall(fd, pixels = pixbuf, sizeof(pixbuf), "<subprocess>");\
}\
}\
ewriteall(fd, pixbuf, (size_t)(pixels - pixbuf) * sizeof(*pixels), "<subprocess>");\
} while (0)
static void process_xyza (struct stream *stream, size_t n) {PROCESS(double,);}
static void process_xyzaf(struct stream *stream, size_t n) {PROCESS(float, _f);}
int
main(int argc, char *argv[])
{
struct stream stream;
char geometry[2 * 3 * sizeof(size_t) + 2];
char *frame_rate;
const char **cmd;
size_t n = 0;
int status, pipe_rw[2];
pid_t pid;
void (*process)(struct stream *stream, size_t n) = NULL;
ARGBEGIN {
case 'd':
draft = 1;
break;
default:
usage();
} ARGEND;
if (argc < 2)
usage();
frame_rate = *argv++, argc--;
cmd = ecalloc((size_t)argc + 12, sizeof(*cmd));
cmd[n++] = "ffmpeg";
cmd[n++] = "-f", cmd[n++] = "rawvideo";
cmd[n++] = "-pix_fmt", cmd[n++] = "ayuv64le";
cmd[n++] = "-r", cmd[n++] = frame_rate;
cmd[n++] = "-s:v", cmd[n++] = geometry;
cmd[n++] = "-i", cmd[n++] = "-";
memcpy(cmd + n, argv, (size_t)argc * sizeof(*cmd));
eopen_stream(&stream, NULL);
sprintf(geometry, "%zux%zu", stream.width, stream.height);
if (!strcmp(stream.pixfmt, "xyza"))
process = process_xyza;
else if (!strcmp(stream.pixfmt, "xyza f"))
process = process_xyzaf;
else
eprintf("pixel format %s is not supported, try xyza\n", stream.pixfmt);
epipe(pipe_rw);
pid = efork();
if (!pid) {
pdeath(SIGKILL);
close(pipe_rw[1]);
edup2(pipe_rw[0], STDIN_FILENO);
close(pipe_rw[0]);
eexecvp("ffmpeg", (char **)(void *)cmd);
}
free(cmd);
close(pipe_rw[0]);
fd = pipe_rw[1];
process_stream(&stream, process);
close(fd);
ewaitpid(pid, &status, 0);
return !!status;
}
|
