Krolyxon
Commit 5105127

Branches

Refs →

nsxivimage.c

668 lines · 14.2 KB · C

Raw Blame History Download
1/* sxiv: image.c
2 * Copyright (c) 2011 Bert Muennich <muennich at informatik.hu-berlin.de>
3 *
4 * This program is free software; you can redistribute it and/or modify it
5 * under the terms of the GNU General Public License as published by the
6 * Free Software Foundation; either version 2 of the License, or (at your
7 * option) any later version.
8 *
9 * This program is distributed in the hope that it will be useful, but
10 * WITHOUT ANY WARRANTY; without even the implied warranty of
11 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
12 * General Public License for more details.
13 *
14 * You should have received a copy of the GNU General Public License along
15 * with this program; if not, write to the Free Software Foundation, Inc.,
16 * 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301, USA.
17 */
18
19#define _POSIX_C_SOURCE 200112L
20
21#include <string.h>
22#include <unistd.h>
23
24#ifdef EXIF_SUPPORT
25#include <libexif/exif-data.h>
26#endif
27
28#ifdef GIF_SUPPORT
29#include <stdlib.h>
30#include <sys/types.h>
31#include <gif_lib.h>
32#endif
33
34#include "image.h"
35#include "options.h"
36#include "util.h"
37
38#define _IMAGE_CONFIG
39#include "config.h"
40
41enum { MIN_GIF_DELAY = 50 };
42
43float zoom_min;
44float zoom_max;
45
46void img_init(img_t *img, win_t *win) {
47 zoom_min = zoom_levels[0] / 100.0;
48 zoom_max = zoom_levels[ARRLEN(zoom_levels) - 1] / 100.0;
49
50 if (img) {
51 img->im = NULL;
52 img->multi.cap = img->multi.cnt = 0;
53 img->multi.animate = 0;
54 img->zoom = options->zoom;
55 img->zoom = MAX(img->zoom, zoom_min);
56 img->zoom = MIN(img->zoom, zoom_max);
57 img->aa = options->aa;
58 img->alpha = 1;
59 img->slideshow = 0;
60 img->ss_delay = SLIDESHOW_DELAY * 1000;
61 }
62
63 if (win) {
64 imlib_context_set_display(win->env.dpy);
65 imlib_context_set_visual(win->env.vis);
66 imlib_context_set_colormap(win->env.cmap);
67 }
68}
69
70#ifdef EXIF_SUPPORT
71void exif_auto_orientate(const fileinfo_t *file) {
72 ExifData *ed;
73 ExifEntry *entry;
74 int byte_order, orientation;
75
76 if (!(ed = exif_data_new_from_file(file->path)))
77 return;
78 entry = exif_content_get_entry(ed->ifd[EXIF_IFD_0], EXIF_TAG_ORIENTATION);
79 if (entry) {
80 byte_order = exif_data_get_byte_order(ed);
81 orientation = exif_get_short(entry->data, byte_order);
82 }
83 exif_data_unref(ed);
84 if (!entry)
85 return;
86
87 switch (orientation) {
88 case 5:
89 imlib_image_orientate(1);
90 case 2:
91 imlib_image_flip_vertical();
92 break;
93
94 case 3:
95 imlib_image_orientate(2);
96 break;
97
98 case 7:
99 imlib_image_orientate(1);
100 case 4:
101 imlib_image_flip_horizontal();
102 break;
103
104 case 6:
105 imlib_image_orientate(1);
106 break;
107
108 case 8:
109 imlib_image_orientate(270);
110 break;
111 }
112}
113#endif /* EXIF_SUPPORT */
114
115#ifdef GIF_SUPPORT
116/* Originally based on, but in its current form merely inspired by Imlib2's
117 * src/modules/loaders/loader_gif.c:load(), written by Carsten Haitzler.
118 */
119int img_load_gif(img_t *img, const fileinfo_t *file) {
120 GifFileType *gif;
121 GifRowType *rows = NULL;
122 GifRecordType rec;
123 ColorMapObject *cmap;
124 DATA32 bgpixel, *data, *ptr;
125 DATA32 *prev_frame = NULL;
126 Imlib_Image *im;
127 int i, j, bg, r, g, b;
128 int x, y, w, h, sw, sh;
129 int intoffset[] = { 0, 4, 2, 1 };
130 int intjump[] = { 8, 8, 4, 2 };
131 int err = 0, transp = -1;
132 unsigned int delay = 0;
133
134 if (img->multi.cap == 0) {
135 img->multi.cap = 8;
136 img->multi.frames = (img_frame_t*)
137 s_malloc(sizeof(img_frame_t) * img->multi.cap);
138 }
139 img->multi.cnt = 0;
140 img->multi.sel = 0;
141
142 gif = DGifOpenFileName(file->path);
143 if (!gif) {
144 warn("could not open gif file: %s", file->name);
145 return 0;
146 }
147 bg = gif->SBackGroundColor;
148 sw = gif->SWidth;
149 sh = gif->SHeight;
150
151 do {
152 if (DGifGetRecordType(gif, &rec) == GIF_ERROR) {
153 err = 1;
154 break;
155 }
156 if (rec == EXTENSION_RECORD_TYPE) {
157 int ext_code;
158 GifByteType *ext = NULL;
159
160 DGifGetExtension(gif, &ext_code, &ext);
161 while (ext) {
162 if (ext_code == 0xf9) {
163 if (ext[1] & 1)
164 transp = (int) ext[4];
165 else
166 transp = -1;
167
168 delay = 10 * ((unsigned int) ext[3] << 8 | (unsigned int) ext[2]);
169 if (delay)
170 delay = MAX(delay, MIN_GIF_DELAY);
171 }
172 ext = NULL;
173 DGifGetExtensionNext(gif, &ext);
174 }
175 } else if (rec == IMAGE_DESC_RECORD_TYPE) {
176 if (DGifGetImageDesc(gif) == GIF_ERROR) {
177 err = 1;
178 break;
179 }
180 x = gif->Image.Left;
181 y = gif->Image.Top;
182 w = gif->Image.Width;
183 h = gif->Image.Height;
184
185 rows = (GifRowType*) s_malloc(h * sizeof(GifRowType));
186 for (i = 0; i < h; i++)
187 rows[i] = (GifRowType) s_malloc(w * sizeof(GifPixelType));
188 if (gif->Image.Interlace) {
189 for (i = 0; i < 4; i++) {
190 for (j = intoffset[i]; j < h; j += intjump[i])
191 DGifGetLine(gif, rows[j], w);
192 }
193 } else {
194 for (i = 0; i < h; i++)
195 DGifGetLine(gif, rows[i], w);
196 }
197
198 ptr = data = (DATA32*) s_malloc(sizeof(DATA32) * sw * sh);
199 cmap = gif->Image.ColorMap ? gif->Image.ColorMap : gif->SColorMap;
200 r = cmap->Colors[bg].Red;
201 g = cmap->Colors[bg].Green;
202 b = cmap->Colors[bg].Blue;
203 bgpixel = 0x00ffffff & (r << 16 | g << 8 | b);
204
205 for (i = 0; i < sh; i++) {
206 for (j = 0; j < sw; j++) {
207 if (i < y || i >= y + h || j < x || j >= x + w) {
208 if (transp >= 0 && prev_frame)
209 *ptr = prev_frame[i * sw + j];
210 else
211 *ptr = bgpixel;
212 } else if (rows[i-y][j-x] == transp) {
213 if (prev_frame)
214 *ptr = prev_frame[i * sw + j];
215 else
216 *ptr = bgpixel;
217 } else {
218 r = cmap->Colors[rows[i-y][j-x]].Red;
219 g = cmap->Colors[rows[i-y][j-x]].Green;
220 b = cmap->Colors[rows[i-y][j-x]].Blue;
221 *ptr = 0xff << 24 | r << 16 | g << 8 | b;
222 }
223 ptr++;
224 }
225 }
226
227 im = imlib_create_image_using_copied_data(sw, sh, data);
228
229 for (i = 0; i < h; i++)
230 free(rows[i]);
231 free(rows);
232 free(data);
233
234 if (!im) {
235 err = 1;
236 break;
237 }
238
239 imlib_context_set_image(im);
240 prev_frame = imlib_image_get_data_for_reading_only();
241
242 imlib_image_set_format("gif");
243 if (transp >= 0)
244 imlib_image_set_has_alpha(1);
245
246 if (img->multi.cnt == img->multi.cap) {
247 img->multi.cap *= 2;
248 img->multi.frames = (img_frame_t*)
249 s_realloc(img->multi.frames,
250 img->multi.cap * sizeof(img_frame_t));
251 }
252 img->multi.frames[img->multi.cnt].im = im;
253 img->multi.frames[img->multi.cnt].delay = delay ? delay : GIF_DELAY;
254 img->multi.cnt++;
255 }
256 } while (rec != TERMINATE_RECORD_TYPE);
257
258 DGifCloseFile(gif);
259
260 if (err && !file->loaded)
261 warn("corrupted gif file: %s", file->name);
262
263 if (img->multi.cnt > 1) {
264 imlib_context_set_image(img->im);
265 imlib_free_image();
266 img->im = img->multi.frames[0].im;
267 img->multi.animate = GIF_AUTOPLAY;
268 } else if (img->multi.cnt == 1) {
269 imlib_context_set_image(img->multi.frames[0].im);
270 imlib_free_image();
271 img->multi.cnt = 0;
272 img->multi.animate = 0;
273 }
274
275 imlib_context_set_image(img->im);
276
277 return !err;
278}
279#endif /* GIF_SUPPORT */
280
281int img_load(img_t *img, const fileinfo_t *file) {
282 const char *fmt;
283
284 if (!img || !file || !file->name || !file->path)
285 return 0;
286
287 if (access(file->path, R_OK) || !(img->im = imlib_load_image(file->path))) {
288 warn("could not open image: %s", file->name);
289 return 0;
290 }
291
292 imlib_context_set_image(img->im);
293 imlib_image_set_changes_on_disk();
294 imlib_context_set_anti_alias(img->aa);
295
296 fmt = imlib_image_format();
297 /* avoid unused-but-set-variable warning */
298 (void) fmt;
299
300#ifdef EXIF_SUPPORT
301 if (!strcmp(fmt, "jpeg"))
302 exif_auto_orientate(file);
303#endif
304#ifdef GIF_SUPPORT
305 if (!strcmp(fmt, "gif"))
306 img_load_gif(img, file);
307#endif
308
309 img->scalemode = options->scalemode;
310 img->re = 0;
311 img->checkpan = 0;
312
313 img->w = imlib_image_get_width();
314 img->h = imlib_image_get_height();
315
316 return 1;
317}
318
319void img_close(img_t *img, int decache) {
320 int i;
321
322 if (!img)
323 return;
324
325 if (img->multi.cnt) {
326 for (i = 0; i < img->multi.cnt; i++) {
327 imlib_context_set_image(img->multi.frames[i].im);
328 imlib_free_image();
329 }
330 img->multi.cnt = 0;
331 img->im = NULL;
332 } else if (img->im) {
333 imlib_context_set_image(img->im);
334 if (decache)
335 imlib_free_image_and_decache();
336 else
337 imlib_free_image();
338 img->im = NULL;
339 }
340}
341
342void img_check_pan(img_t *img, win_t *win) {
343 if (!img || !win)
344 return;
345
346 if (img->w * img->zoom > win->w) {
347 if (img->x > 0 && img->x + img->w * img->zoom > win->w)
348 img->x = 0;
349 if (img->x < 0 && img->x + img->w * img->zoom < win->w)
350 img->x = win->w - img->w * img->zoom;
351 } else {
352 img->x = (win->w - img->w * img->zoom) / 2;
353 }
354 if (img->h * img->zoom > win->h) {
355 if (img->y > 0 && img->y + img->h * img->zoom > win->h)
356 img->y = 0;
357 if (img->y < 0 && img->y + img->h * img->zoom < win->h)
358 img->y = win->h - img->h * img->zoom;
359 } else {
360 img->y = (win->h - img->h * img->zoom) / 2;
361 }
362}
363
364int img_fit(img_t *img, win_t *win) {
365 float oz, zw, zh;
366
367 if (!img || !win)
368 return 0;
369
370 oz = img->zoom;
371 zw = (float) win->w / (float) img->w;
372 zh = (float) win->h / (float) img->h;
373
374 img->zoom = MIN(zw, zh);
375 img->zoom = MAX(img->zoom, zoom_min);
376 img->zoom = MIN(img->zoom, zoom_max);
377
378 return oz != img->zoom;
379}
380
381void img_render(img_t *img, win_t *win) {
382 int sx, sy, sw, sh;
383 int dx, dy, dw, dh;
384
385 if (!img || !img->im || !win)
386 return;
387
388 if (img->scalemode != SCALE_ZOOM) {
389 img_fit(img, win);
390 if (img->scalemode == SCALE_DOWN && img->zoom > 1.0)
391 img->zoom = 1.0;
392 }
393
394 if (!img->re) {
395 /* rendered for the first time */
396 img->re = 1;
397 if (img->zoom * img->w <= win->w)
398 img->x = (win->w - img->w * img->zoom) / 2;
399 else
400 img->x = 0;
401 if (img->zoom * img->h <= win->h)
402 img->y = (win->h - img->h * img->zoom) / 2;
403 else
404 img->y = 0;
405 }
406
407 if (img->checkpan) {
408 img_check_pan(img, win);
409 img->checkpan = 0;
410 }
411
412 /* calculate source and destination offsets */
413 if (img->x < 0) {
414 sx = -img->x / img->zoom;
415 sw = win->w / img->zoom;
416 dx = 0;
417 dw = win->w;
418 } else {
419 sx = 0;
420 sw = img->w;
421 dx = img->x;
422 dw = img->w * img->zoom;
423 }
424 if (img->y < 0) {
425 sy = -img->y / img->zoom;
426 sh = win->h / img->zoom;
427 dy = 0;
428 dh = win->h;
429 } else {
430 sy = 0;
431 sh = img->h;
432 dy = img->y;
433 dh = img->h * img->zoom;
434 }
435
436 win_clear(win);
437
438 imlib_context_set_image(img->im);
439
440 if (imlib_image_has_alpha() && !img->alpha)
441 win_draw_rect(win, win->pm, dx, dy, dw, dh, True, 0, win->white);
442
443 imlib_context_set_drawable(win->pm);
444 imlib_render_image_part_on_drawable_at_size(sx, sy, sw, sh, dx, dy, dw, dh);
445
446 win_draw(win);
447}
448
449int img_fit_win(img_t *img, win_t *win) {
450 if (!img || !img->im || !win)
451 return 0;
452
453 img->scalemode = SCALE_FIT;
454 return img_fit(img, win);
455}
456
457int img_center(img_t *img, win_t *win) {
458 int ox, oy;
459
460 if (!img || !win)
461 return 0;
462
463 ox = img->x;
464 oy = img->y;
465
466 img->x = (win->w - img->w * img->zoom) / 2;
467 img->y = (win->h - img->h * img->zoom) / 2;
468
469 return ox != img->x || oy != img->y;
470}
471
472int img_zoom(img_t *img, win_t *win, float z) {
473 if (!img || !img->im || !win)
474 return 0;
475
476 z = MAX(z, zoom_min);
477 z = MIN(z, zoom_max);
478
479 img->scalemode = SCALE_ZOOM;
480
481 if (z != img->zoom) {
482 img->x = win->w / 2 - (win->w / 2 - img->x) * z / img->zoom;
483 img->y = win->h / 2 - (win->h / 2 - img->y) * z / img->zoom;
484 img->zoom = z;
485 img->checkpan = 1;
486 return 1;
487 } else {
488 return 0;
489 }
490}
491
492int img_zoom_in(img_t *img, win_t *win) {
493 int i;
494
495 if (!img || !img->im || !win)
496 return 0;
497
498 for (i = 1; i < ARRLEN(zoom_levels); i++) {
499 if (zoom_levels[i] > img->zoom * 100.0)
500 return img_zoom(img, win, zoom_levels[i] / 100.0);
501 }
502 return 0;
503}
504
505int img_zoom_out(img_t *img, win_t *win) {
506 int i;
507
508 if (!img || !img->im || !win)
509 return 0;
510
511 for (i = ARRLEN(zoom_levels) - 2; i >= 0; i--) {
512 if (zoom_levels[i] < img->zoom * 100.0)
513 return img_zoom(img, win, zoom_levels[i] / 100.0);
514 }
515 return 0;
516}
517
518int img_move(img_t *img, win_t *win, int dx, int dy) {
519 int ox, oy;
520
521 if (!img || !img->im || !win)
522 return 0;
523
524 ox = img->x;
525 oy = img->y;
526
527 img->x += dx;
528 img->y += dy;
529
530 img_check_pan(img, win);
531
532 return ox != img->x || oy != img->y;
533}
534
535int img_pan(img_t *img, win_t *win, direction_t dir, int screen) {
536 if (!img || !img->im || !win)
537 return 0;
538
539 switch (dir) {
540 case DIR_LEFT:
541 return img_move(img, win, win->w / (screen ? 1 : 5), 0);
542 case DIR_RIGHT:
543 return img_move(img, win, win->w / (screen ? 1 : 5) * -1, 0);
544 case DIR_UP:
545 return img_move(img, win, 0, win->h / (screen ? 1 : 5));
546 case DIR_DOWN:
547 return img_move(img, win, 0, win->h / (screen ? 1 : 5) * -1);
548 }
549
550 return 0;
551}
552
553int img_pan_edge(img_t *img, win_t *win, direction_t dir) {
554 int ox, oy;
555
556 if (!img || !img->im || !win)
557 return 0;
558
559 ox = img->x;
560 oy = img->y;
561
562 switch (dir) {
563 case DIR_LEFT:
564 img->x = 0;
565 break;
566 case DIR_RIGHT:
567 img->x = win->w - img->w * img->zoom;
568 break;
569 case DIR_UP:
570 img->y = 0;
571 break;
572 case DIR_DOWN:
573 img->y = win->h - img->h * img->zoom;
574 break;
575 }
576
577 img_check_pan(img, win);
578
579 return ox != img->x || oy != img->y;
580}
581
582void img_rotate(img_t *img, win_t *win, int d) {
583 int ox, oy, tmp;
584
585 if (!img || !img->im || !win)
586 return;
587
588 ox = d == 1 ? img->x : win->w - img->x - img->w * img->zoom;
589 oy = d == 3 ? img->y : win->h - img->y - img->h * img->zoom;
590
591 imlib_context_set_image(img->im);
592 imlib_image_orientate(d);
593
594 img->x = oy + (win->w - win->h) / 2;
595 img->y = ox + (win->h - win->w) / 2;
596
597 tmp = img->w;
598 img->w = img->h;
599 img->h = tmp;
600
601 img->checkpan = 1;
602}
603
604void img_rotate_left(img_t *img, win_t *win) {
605 img_rotate(img, win, 3);
606}
607
608void img_rotate_right(img_t *img, win_t *win) {
609 img_rotate(img, win, 1);
610}
611
612void img_toggle_antialias(img_t *img) {
613 if (img && img->im) {
614 img->aa ^= 1;
615 imlib_context_set_image(img->im);
616 imlib_context_set_anti_alias(img->aa);
617 }
618}
619
620int img_frame_goto(img_t *img, int n) {
621 if (!img || n < 0 || n >= img->multi.cnt)
622 return 0;
623
624 if (n == img->multi.sel)
625 return 0;
626
627 img->multi.sel = n;
628 img->im = img->multi.frames[n].im;
629
630 imlib_context_set_image(img->im);
631 img->w = imlib_image_get_width();
632 img->h = imlib_image_get_height();
633 img->checkpan = 1;
634
635 return 1;
636}
637
638int img_frame_navigate(img_t *img, int d) {
639 if (!img || !img->multi.cnt || !d)
640 return 0;
641
642 d += img->multi.sel;
643 if (d < 0)
644 d = 0;
645 else if (d >= img->multi.cnt)
646 d = img->multi.cnt - 1;
647
648 return img_frame_goto(img, d);
649}
650
651int img_frame_animate(img_t *img, int restart) {
652 if (!img || !img->multi.cnt)
653 return 0;
654
655 if (img->multi.sel + 1 >= img->multi.cnt) {
656 if (restart || (GIF_LOOP && !img->slideshow)) {
657 img_frame_goto(img, 0);
658 } else {
659 img->multi.animate = 0;
660 return 0;
661 }
662 } else if (!restart) {
663 img_frame_goto(img, img->multi.sel + 1);
664 }
665 img->multi.animate = 1;
666
667 return img->multi.frames[img->multi.sel].delay;
668}