-
-
Notifications
You must be signed in to change notification settings - Fork 57
/
catimg.c
253 lines (235 loc) · 9.24 KB
/
catimg.c
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
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "sh_image.h"
#include "sh_utils.h"
#include <unistd.h>
#include <signal.h>
#include <sys/ioctl.h>
#define USAGE "Usage: catimg [-hct] [-w width | -H height] [-l loops] [-r resolution] image-file\n\n" \
" -h: Displays this message\n" \
" -w: Terminal width/columns by default\n" \
" -H: Terminal height/row by default\n" \
" -l: Loops are only useful with GIF files. A value of 1 means that the GIF will " \
"be displayed twice because it loops once. A negative value means infinite " \
"looping\n" \
" -r: Resolution must be 1 or 2. By default catimg checks for unicode support to " \
"use higher resolution\n" \
" -c: Convert colors to a restricted palette\n" \
" -t: Disables true color (24-bit) support, falling back to 256 color\n"
#define ERR_WIDTH_OR_HEIGHT "[ERROR] '-w' and '-H' can't be used at the same time\n\n"
// Transparency threshold -- all pixels with alpha below 25%.
#define TRANSP_ALPHA 64
extern char *optarg;
extern int optind;
extern int optopt;
extern int opterr;
extern int optreset;
// For <C-C>
volatile int loops = -1, loop = -1;
volatile char stop = 0;
void intHandler() {
loops = loop;
stop = 1;
}
int getopt(int argc, char * const argv[], const char *optstring);
uint32_t pixelToInt(const color_t *pixel) {
if (pixel->a == 0)
return 0xffff;
else if (pixel->r == pixel->g && pixel->g == pixel->b)
return 232 + (pixel->r * 23)/255;
else
return (16 + ((pixel->r*5)/255)*36
+ ((pixel->g*5)/255)*6
+ (pixel->b*5)/255);
}
char supportsUTF8() {
const char* LC_ALL = getenv("LC_ALL");
const char* LANG = getenv("LANG");
const char* LC_CTYPE = getenv("LC_CTYPE");
const char* UTF = "UTF-8";
return (LC_ALL && strstr(LC_ALL, UTF))
|| (LANG && strstr(LANG, UTF))
|| (LC_CTYPE && strstr(LC_CTYPE, UTF));
}
int main(int argc, char *argv[])
{
init_hash_colors();
image_t img;
char *file;
char *num;
int c;
opterr = 0;
uint32_t cols = 0, rows = 0, precision = 0;
uint32_t max_cols = 0, max_rows = 0;
uint8_t convert = 0;
uint8_t true_color = 1;
uint8_t adjust_to_height = 0, adjust_to_width = 0;
float scale_cols = 0, scale_rows = 0;
while ((c = getopt (argc, argv, "H:w:l:r:hct")) != -1)
switch (c) {
case 'H':
rows = strtol(optarg, &num, 0);
if (adjust_to_width) {
// only either adjust to width or adjust to height is allowed, but not both.
printf(ERR_WIDTH_OR_HEIGHT);
printf(USAGE);
exit(1);
}
adjust_to_height = 1;
break;
case 'w':
cols = strtol(optarg, &num, 0) >> 1;
if (adjust_to_height) {
// only either adjust to width or adjust to height is allowed, but not both.
printf(ERR_WIDTH_OR_HEIGHT);
printf(USAGE);
exit(1);
}
adjust_to_width = 1;
break;
case 'l':
loops = strtol(optarg, &num, 0);
break;
case 'r':
precision = strtol(optarg, &num, 0);
break;
case 'h':
printf(USAGE);
exit(0);
break;
case 'c':
convert = 1;
break;
case 't':
true_color = 0;
break;
default:
printf(USAGE);
exit(1);
break;
}
if (argc > 1)
file = argv[argc-1];
else {
printf(USAGE);
exit(1);
}
if (precision == 0 || precision > 2) {
if (supportsUTF8())
precision = 2;
else
precision = 1;
}
// if precision is 2 we can use the terminal full width/height. Otherwise we can only use half
max_cols = terminal_columns() / (2 / precision);
max_rows = terminal_rows() * precision;
if (strcmp(file, "-") == 0) {
img_load_from_stdin(&img);
} else {
img_load_from_file(&img, file);
}
if (cols == 0 && rows == 0) {
scale_cols = max_cols / (float)img.width;
scale_rows = max_rows / (float)img.height;
if (adjust_to_height && scale_rows < scale_cols && max_rows < img.height)
// rows == 0 and adjust_to_height > adjust to height instead of width
img_resize(&img, scale_rows, scale_rows);
else if (max_cols < img.width)
img_resize(&img, scale_cols, scale_cols);
} else if (cols > 0 && cols < img.width) {
scale_cols = cols / (float)img.width;
img_resize(&img, scale_cols, scale_cols);
} else if (rows > 0 && rows < img.height) {
scale_rows = rows / (float)img.height;
img_resize(&img, scale_rows, scale_rows);
}
if (convert)
img_convert_colors(&img);
/*printf("Loaded %s: %ux%u. Console width: %u\n", file, img.width, img.height, cols);*/
// For GIF
if (img.frames > 1) {
system("clear");
signal(SIGINT, intHandler);
} else {
loops = 0;
}
// Save the cursor position and hide it
printf("\e[s\e[?25l");
while (loop++ < loops || loops < 0) {
uint32_t offset = 0;
for (uint32_t frame = 0; frame < img.frames; frame++) {
if (frame > 0 || loop > 0) {
if (frame > 0)
usleep(img.delays[frame - 1] * 10000);
else
usleep(img.delays[img.frames - 1] * 10000);
printf("\e[u");
}
uint32_t index, x, y;
for (y = 0; y < img.height; y += precision) {
for (x = 0; x < img.width; x++) {
index = y * img.width + x + offset;
const color_t* upperPixel = &img.pixels[index];
uint32_t fgCol = pixelToInt(upperPixel);
if (precision == 2) {
if (y < img.height - 1) {
const color_t* lowerPixel = &img.pixels[index + img.width];
uint32_t bgCol = pixelToInt(&img.pixels[index + img.width]);
if (upperPixel->a < TRANSP_ALPHA) { // first pixel is transparent
if (lowerPixel->a < TRANSP_ALPHA)
printf("\e[m ");
else
if (true_color)
printf("\x1b[0;38;2;%d;%d;%dm\u2584",
lowerPixel->r, lowerPixel->g, lowerPixel->b);
else
printf("\e[0;38;5;%um\u2584", bgCol);
} else {
if (lowerPixel->a < TRANSP_ALPHA)
if (true_color)
printf("\x1b[0;38;2;%d;%d;%dm\u2580",
upperPixel->r, upperPixel->g, upperPixel->b);
else
printf("\e[0;38;5;%um\u2580", fgCol);
else
if (true_color)
printf("\x1b[48;2;%d;%d;%dm\x1b[38;2;%d;%d;%dm\u2584",
upperPixel->r, upperPixel->g, upperPixel->b,
lowerPixel->r, lowerPixel->g, lowerPixel->b);
else
printf("\e[38;5;%u;48;5;%um\u2580", fgCol, bgCol);
}
} else { // this is the last line
if (upperPixel->a < TRANSP_ALPHA)
printf("\e[m ");
else
if (true_color)
printf("\x1b[0;38;2;%d;%d;%dm\u2580",
upperPixel->r, upperPixel->g, upperPixel->b);
else
printf("\e[38;5;%um\u2580", fgCol);
}
} else {
if (img.pixels[index].a < TRANSP_ALPHA)
printf("\e[m ");
else
if (true_color)
printf("\x1b[0;48;2;%d;%d;%dm ",
img.pixels[index].r, img.pixels[index].g, img.pixels[index].b);
else
printf("\e[48;5;%um ", fgCol);
}
}
printf("\e[m\n");
}
offset += img.width * img.height;
if (stop) frame = img.frames;
}
}
// Display the cursor again
printf("\e[?25h");
img_free(&img);
free_hash_colors();
return 0;
}