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
|
/* See LICENSE file for copyright and license details. */
#include "stream.h"
#include "util.h"
#include <sys/mman.h>
#include <fcntl.h>
#include <inttypes.h>
#include <string.h>
#include <unistd.h>
USAGE("[-o output-file] first-stream ... last-stream")
static void
concat_to_stdout(int argc, char *argv[])
{
struct stream *streams;
size_t frames = 0;
int i;
streams = emalloc((size_t)argc * sizeof(*streams));
for (i = 0; i < argc; i++) {
streams[i].file = argv[i];
streams[i].fd = eopen(streams[i].file, O_RDONLY);
einit_stream(streams + i);
if (i)
echeck_compat(streams + i, streams);
if (streams[i].frames > SIZE_MAX - frames)
eprintf("resulting video is too long\n");
frames += streams[i].frames;
}
streams->frames = frames;
fprint_stream_head(stdout, streams);
efflush(stdout, "<stdout>");
for (; argc--; streams++) {
for (; eread_stream(streams, SIZE_MAX); streams->ptr = 0)
ewriteall(STDOUT_FILENO, streams->buf, streams->ptr, "<stdout>");
close(streams->fd);
}
}
static void
concat_to_file(int argc, char *argv[], char *output_file)
{
struct stream stream, refstream;
int first = 0;
int fd = eopen(output_file, O_RDWR | O_CREAT | O_TRUNC, 0666);
char head[3 * sizeof(size_t) + 4 + sizeof(stream.pixfmt) + 6];
ssize_t headlen, size = 0;
char *data;
for (; argc--; argv++) {
stream.file = *argv;
stream.fd = eopen(stream.file, O_RDONLY);
einit_stream(&stream);
if (first) {
stream = refstream;
} else {
if (refstream.frames > SIZE_MAX - stream.frames)
eprintf("resulting video is too long\n");
refstream.frames += stream.frames;
echeck_compat(&stream, &refstream);
}
for (; eread_stream(&stream, SIZE_MAX); stream.ptr = 0) {
ewriteall(STDOUT_FILENO, stream.buf, stream.ptr, "<stdout>");
size += stream.ptr;
}
close(stream.fd);
}
sprintf(head, "%zu %zu %zu %s\n%cuivf%zn",
stream.frames, stream.width, stream.height, stream.pixfmt, 0, &headlen);
ewriteall(STDOUT_FILENO, head, (size_t)headlen, "<stdout>");
data = mmap(0, size + (size_t)headlen, PROT_READ | PROT_WRITE, MAP_PRIVATE, fd, 0);
memmove(data + headlen, data, size);
memcpy(data, head, (size_t)headlen);
munmap(data, size + (size_t)headlen);
close(fd);
}
int
main(int argc, char *argv[])
{
char *output_file = NULL;
ARGBEGIN {
case 'o':
output_file = EARG();
break;
default:
usage();
} ARGEND;
if (argc < 2)
usage();
if (output_file)
concat_to_file(argc, argv, output_file);
else
concat_to_stdout(argc, argv);
return 0;
}
|
