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
|
/* See LICENSE file for copyright and license details. */
#include "common.h"
USAGE("[-e] [theta0-stream]")
static int equal = 0;
#define PROCESS(TYPE, SUFFIX)\
static void\
process_##SUFFIX(struct stream *grad, struct stream *theta0)\
{\
size_t i, n, m = 0;\
TYPE *theta0xyza;\
TYPE x, theta0x = 0;\
TYPE y, theta0y = 0;\
TYPE z, theta0z = 0;\
TYPE a, theta0a = 0;\
echeck_dimensions(grad, WIDTH | HEIGHT, NULL);\
do {\
if (!m) {\
m = grad->frame_size;\
if (theta0) {\
while (theta0->ptr < theta0->frame_size)\
if (!eread_stream(theta0, theta0->frame_size - theta0->ptr))\
return;\
theta0xyza = (TYPE *)theta0->buf;\
theta0x = (theta0xyza)[0];\
theta0y = (theta0xyza)[1];\
theta0z = (theta0xyza)[2];\
theta0a = (theta0xyza)[3];\
memmove(theta0->buf, theta0->buf + theta0->frame_size,\
theta0->ptr -= theta0->frame_size);\
}\
}\
n = MIN(grad->ptr, m) / grad->pixel_size;\
if (equal) {\
for (i = 0; i < n; i++) {\
a = ((TYPE *)(grad->buf))[4 * i + 3];\
a = (a ? sin(a + theta0y) / a : sin(a + theta0y)) / 2 + 0.5;\
((TYPE *)(grad->buf))[4 * i + 0] = a;\
((TYPE *)(grad->buf))[4 * i + 1] = a;\
((TYPE *)(grad->buf))[4 * i + 2] = a;\
((TYPE *)(grad->buf))[4 * i + 3] = a;\
}\
} else {\
for (i = 0; i < n; i++) {\
x = ((TYPE *)(grad->buf))[4 * i + 0];\
y = ((TYPE *)(grad->buf))[4 * i + 1];\
z = ((TYPE *)(grad->buf))[4 * i + 2];\
a = ((TYPE *)(grad->buf))[4 * i + 3];\
x = (x ? sin(x + theta0x) / x : sin(x + theta0x)) / 2 + 0.5;\
y = (y ? sin(y + theta0y) / y : sin(y + theta0y)) / 2 + 0.5;\
z = (z ? sin(z + theta0z) / z : sin(z + theta0z)) / 2 + 0.5;\
a = (a ? sin(a + theta0a) / a : sin(a + theta0a)) / 2 + 0.5;\
((TYPE *)(grad->buf))[4 * i + 0] = x;\
((TYPE *)(grad->buf))[4 * i + 1] = y;\
((TYPE *)(grad->buf))[4 * i + 2] = z;\
((TYPE *)(grad->buf))[4 * i + 3] = a;\
}\
}\
n *= grad->pixel_size;\
m -= n;\
ewriteall(STDOUT_FILENO, grad->buf, n, "<stdout>");\
memmove(grad->buf, grad->buf + n, grad->ptr -= n);\
} while (eread_stream(grad, SIZE_MAX));\
if (grad->ptr)\
eprintf("%s: incomplete frame\n", grad->file);\
}
PROCESS(double, lf)
PROCESS(float, f)
int
main(int argc, char *argv[])
{
struct stream stream, theta0;
int have_theta0;
void (*process)(struct stream *grad, struct stream *theta0);
ARGBEGIN {
case 'e':
equal = 1;
break;
default:
usage();
} ARGEND;
if (argc > 1)
usage();
eopen_stream(&stream, NULL);
if ((have_theta0 = argc == 1)) {
eopen_stream(&theta0, argv[0]);
if (theta0.width != 1 || theta0.height != 1)
eprintf("theta0-stream must be of dimension 1x1\n");
}
if (!strcmp(stream.pixfmt, "xyza"))
process = process_lf;
else if (!strcmp(stream.pixfmt, "xyza f"))
process = process_f;
else
eprintf("pixel format %s is not supported, try xyza\n", stream.pixfmt);
if (have_theta0 && strcmp(stream.pixfmt, theta0.pixfmt))
eprintf("videos use incompatible pixel formats\n");
fprint_stream_head(stdout, &stream);
efflush(stdout, "<stdout>");
process(&stream, have_theta0 ? &theta0 : NULL);
return 0;
}
|
