Krolyxon
Commit 71e61b1

Branches

Refs →

nsxivimage.c

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