 |
blind-apply-palette.c - blind - suckless command-line video editing utility |
 |
git clone git://git.suckless.org/blind |
 |
Log |
|
Files |
|
Refs |
|
README |
|
LICENSE |
|
--- |
|
blind-apply-palette.c (3026B) |
|
--- |
|
1 /* See LICENSE file for copyright and license details. */ |
|
2 #ifndef TYPE |
|
3 #include "common.h" |
|
4 |
|
5 USAGE("palette-stream") |
|
6 |
|
7 static double (*compare)(double x1, double y1, double z1, double a1, dou… |
|
8 /* TODO add more formulae: https://en.wikipedia.org/wiki/Color_differenc… |
|
9 |
|
10 static double |
|
11 distance_xyz(double x1, double y1, double z1, double a1, double x2, doub… |
|
12 { |
|
13 x2 -= x1, x2 *= x2; |
|
14 y2 -= y1, y2 *= y2; |
|
15 z2 -= z1, z2 *= z2; |
|
16 a2 -= a1, a2 *= a2; |
|
17 return sqrt(x2 + y2 + z2 + a2); |
|
18 } |
|
19 |
|
20 #define FILE "blind-apply-palette.c" |
|
21 #include "define-functions.h" |
|
22 |
|
23 int |
|
24 main(int argc, char *argv[]) |
|
25 { |
|
26 struct stream stream, palette; |
|
27 void (*process)(struct stream *stream, struct stream *palette, c… |
|
28 char *pal; |
|
29 |
|
30 compare = distance_xyz; |
|
31 |
|
32 UNOFLAGS(argc != 1); |
|
33 |
|
34 eopen_stream(&stream, NULL); |
|
35 eopen_stream(&palette, argv[0]); |
|
36 |
|
37 SELECT_PROCESS_FUNCTION(&stream); |
|
38 CHECK_N_CHAN(&stream, 4, 4); |
|
39 |
|
40 if (strcmp(stream.pixfmt, palette.pixfmt)) |
|
41 eprintf("videos use incompatible pixel formats\n"); |
|
42 |
|
43 echeck_dimensions(&stream, WIDTH | HEIGHT, NULL); |
|
44 echeck_dimensions(&palette, WIDTH | HEIGHT, NULL); |
|
45 pal = emalloc(palette.frame_size); |
|
46 fprint_stream_head(stdout, &stream); |
|
47 efflush(stdout, "<stdout>"); |
|
48 |
|
49 process(&stream, &palette, pal); |
|
50 |
|
51 free(pal); |
|
52 return 0; |
|
53 } |
|
54 |
|
55 #else |
|
56 |
|
57 static void |
|
58 PROCESS(struct stream *stream, struct stream *palette, char *pal) |
|
59 { |
|
60 size_t i, j, n, m; |
|
61 size_t palsiz = palette->width * palette->height; |
|
62 size_t best = 0; |
|
63 TYPE x, y, z, a, lx = 0, ly = 0, lz = 0, la = 0; |
|
64 TYPE cx, cy, cz, ca; |
|
65 double distance, best_distance = 0; |
|
66 while (eread_frame(palette, pal)) { |
|
67 m = stream->frame_size; |
|
68 do { |
|
69 n = MIN(stream->ptr, m) / stream->pixel_size; |
|
70 for (i = 0; i < n; i++) { |
|
71 x = ((TYPE *)(stream->buf + i * stream->… |
|
72 y = ((TYPE *)(stream->buf + i * stream->… |
|
73 z = ((TYPE *)(stream->buf + i * stream->… |
|
74 a = ((TYPE *)(stream->buf + i * stream->… |
|
75 if ((!i && m == stream->frame_size) || x… |
|
76 for (j = 0; j < palsiz; j++) { |
|
77 cx = ((TYPE *)(pal + j *… |
|
78 cy = ((TYPE *)(pal + j *… |
|
79 cz = ((TYPE *)(pal + j *… |
|
80 ca = ((TYPE *)(pal + j *… |
|
81 distance = compare((doub… |
|
82 (doub… |
|
83 if (!j || distance < bes… |
|
84 best_distance = … |
|
85 best = j; |
|
86 } |
|
87 } |
|
88 lx = x, ly = y, lz = z, la = a; |
|
89 } |
|
90 memcpy(stream->buf + i * stream->pixel_s… |
|
91 pal + best * stream->pixel_size, |
|
92 stream->pixel_size); |
|
93 } |
|
94 m -= n *= stream->pixel_size; |
|
95 ewriteall(STDOUT_FILENO, stream->buf, n, "<stdou… |
|
96 memmove(stream->buf, stream->buf + n, stream->pt… |
|
97 } while (m && eread_stream(stream, SIZE_MAX)); |
|
98 if (m) |
|
99 eprintf("%s: incomplete frame\n", stream->file); |
|
100 } |
|
101 } |
|
102 |
|
103 #endif |
