Krolyxon
Commit c4007be

Branches

Refs →

nsxivimage.c

739 lines · 15.7 KB · C

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