Krolyxon
Commit 4383a65

Branches

Refs →

nsxivimage.c

722 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
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 intoffset[] = { 0, 4, 2, 1 };
137 int intjump[] = { 8, 8, 4, 2 };
138 int transp = -1;
139 unsigned int disposal = 0, prev_disposal = 0;
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 disposal = (unsigned int) ext[1] >> 2 & 0x7;
182 }
183 ext = NULL;
184 DGifGetExtensionNext(gif, &ext);
185 }
186 } else if (rec == IMAGE_DESC_RECORD_TYPE) {
187 if (DGifGetImageDesc(gif) == GIF_ERROR) {
188 err = true;
189 break;
190 }
191 x = gif->Image.Left;
192 y = gif->Image.Top;
193 w = gif->Image.Width;
194 h = gif->Image.Height;
195
196 rows = (GifRowType*) s_malloc(h * sizeof(GifRowType));
197 for (i = 0; i < h; i++)
198 rows[i] = (GifRowType) s_malloc(w * sizeof(GifPixelType));
199 if (gif->Image.Interlace) {
200 for (i = 0; i < 4; i++) {
201 for (j = intoffset[i]; j < h; j += intjump[i])
202 DGifGetLine(gif, rows[j], w);
203 }
204 } else {
205 for (i = 0; i < h; i++)
206 DGifGetLine(gif, rows[i], w);
207 }
208
209 ptr = data = (DATA32*) s_malloc(sizeof(DATA32) * sw * sh);
210 cmap = gif->Image.ColorMap ? gif->Image.ColorMap : gif->SColorMap;
211 r = cmap->Colors[bg].Red;
212 g = cmap->Colors[bg].Green;
213 b = cmap->Colors[bg].Blue;
214 bgpixel = 0x00ffffff & (r << 16 | g << 8 | b);
215
216 for (i = 0; i < sh; i++) {
217 for (j = 0; j < sw; j++) {
218 if (i < y || i >= y + h || j < x || j >= x + w ||
219 rows[i-y][j-x] == transp)
220 {
221 if (prev_disposal != 2 && prev_frame != NULL)
222 *ptr = prev_frame[i * sw + j];
223 else
224 *ptr = bgpixel;
225 } else {
226 r = cmap->Colors[rows[i-y][j-x]].Red;
227 g = cmap->Colors[rows[i-y][j-x]].Green;
228 b = cmap->Colors[rows[i-y][j-x]].Blue;
229 *ptr = 0xff << 24 | r << 16 | g << 8 | b;
230 }
231 ptr++;
232 }
233 }
234
235 im = imlib_create_image_using_copied_data(sw, sh, data);
236
237 for (i = 0; i < h; i++)
238 free(rows[i]);
239 free(rows);
240 free(data);
241
242 if (im == NULL) {
243 err = true;
244 break;
245 }
246
247 imlib_context_set_image(im);
248 imlib_image_set_format("gif");
249 if (transp >= 0)
250 imlib_image_set_has_alpha(1);
251
252 if (disposal != 3)
253 prev_frame = imlib_image_get_data_for_reading_only();
254 prev_disposal = disposal;
255
256 if (img->multi.cnt == img->multi.cap) {
257 img->multi.cap *= 2;
258 img->multi.frames = (img_frame_t*)
259 s_realloc(img->multi.frames,
260 img->multi.cap * sizeof(img_frame_t));
261 }
262 img->multi.frames[img->multi.cnt].im = im;
263 img->multi.frames[img->multi.cnt].delay = delay ? delay : GIF_DELAY;
264 img->multi.cnt++;
265 }
266 } while (rec != TERMINATE_RECORD_TYPE);
267
268 DGifCloseFile(gif);
269
270 if (err && !file->loaded)
271 warn("corrupted gif file: %s", file->name);
272
273 if (img->multi.cnt > 1) {
274 imlib_context_set_image(img->im);
275 imlib_free_image();
276 img->im = img->multi.frames[0].im;
277 img->multi.animate = GIF_AUTOPLAY;
278 } else if (img->multi.cnt == 1) {
279 imlib_context_set_image(img->multi.frames[0].im);
280 imlib_free_image();
281 img->multi.cnt = 0;
282 img->multi.animate = false;
283 }
284
285 imlib_context_set_image(img->im);
286
287 return !err;
288}
289#endif /* GIF_SUPPORT */
290
291bool img_load(img_t *img, const fileinfo_t *file) {
292 const char *fmt;
293
294 if (img == NULL || file == NULL || file->name == NULL || file->path == NULL)
295 return false;
296
297 if (access(file->path, R_OK) < 0 ||
298 (img->im = imlib_load_image(file->path)) == NULL)
299 {
300 warn("could not open image: %s", file->name);
301 return false;
302 }
303
304 imlib_context_set_image(img->im);
305 imlib_image_set_changes_on_disk();
306 imlib_context_set_anti_alias(img->aa);
307
308 fmt = imlib_image_format();
309 /* avoid unused-but-set-variable warning */
310 (void) fmt;
311
312#if EXIF_SUPPORT
313 if (STREQ(fmt, "jpeg"))
314 exif_auto_orientate(file);
315#endif
316#if GIF_SUPPORT
317 if (STREQ(fmt, "gif"))
318 img_load_gif(img, file);
319#endif
320
321 img->w = imlib_image_get_width();
322 img->h = imlib_image_get_height();
323 img->scalemode = options->scalemode;
324 img->re = false;
325 img->checkpan = false;
326 img->dirty = true;
327
328 return true;
329}
330
331void img_close(img_t *img, bool decache) {
332 int i;
333
334 if (img == NULL)
335 return;
336
337 if (img->multi.cnt > 0) {
338 for (i = 0; i < img->multi.cnt; i++) {
339 imlib_context_set_image(img->multi.frames[i].im);
340 imlib_free_image();
341 }
342 img->multi.cnt = 0;
343 img->im = NULL;
344 } else if (img->im != NULL) {
345 imlib_context_set_image(img->im);
346 if (decache)
347 imlib_free_image_and_decache();
348 else
349 imlib_free_image();
350 img->im = NULL;
351 }
352}
353
354void img_check_pan(img_t *img, bool moved) {
355 win_t *win;
356 int ox, oy;
357
358 if (img == NULL || img->im == NULL || img->win == NULL)
359 return;
360
361 win = img->win;
362 ox = img->x;
363 oy = img->y;
364
365 if (img->w * img->zoom > win->w) {
366 if (img->x > 0 && img->x + img->w * img->zoom > win->w)
367 img->x = 0;
368 if (img->x < 0 && img->x + img->w * img->zoom < win->w)
369 img->x = win->w - img->w * img->zoom;
370 } else {
371 img->x = (win->w - img->w * img->zoom) / 2;
372 }
373 if (img->h * img->zoom > win->h) {
374 if (img->y > 0 && img->y + img->h * img->zoom > win->h)
375 img->y = 0;
376 if (img->y < 0 && img->y + img->h * img->zoom < win->h)
377 img->y = win->h - img->h * img->zoom;
378 } else {
379 img->y = (win->h - img->h * img->zoom) / 2;
380 }
381
382 if (!moved && (ox != img->x || oy != img->y))
383 img->dirty = true;
384}
385
386bool img_fit(img_t *img) {
387 float z, zmax, zw, zh;
388
389 if (img == NULL || img->im == NULL || img->win == NULL)
390 return false;
391 if (img->scalemode == SCALE_ZOOM)
392 return false;
393
394 zmax = img->scalemode == SCALE_DOWN ? 1.0 : zoom_max;
395 zw = (float) img->win->w / (float) img->w;
396 zh = (float) img->win->h / (float) img->h;
397
398 z = MIN(zw, zh);
399 z = MAX(z, zoom_min);
400 z = MIN(z, zmax);
401
402 if (zoomdiff(z, img->zoom)) {
403 img->zoom = z;
404 img->dirty = true;
405 return true;
406 } else {
407 return false;
408 }
409}
410
411void img_render(img_t *img) {
412 win_t *win;
413 int sx, sy, sw, sh;
414 int dx, dy, dw, dh;
415
416 if (img == NULL || img->im == NULL || img->win == NULL)
417 return;
418
419 win = img->win;
420 img_fit(img);
421
422 if (!img->re) {
423 /* rendered for the first time */
424 img->re = true;
425 if (img->zoom * img->w <= win->w)
426 img->x = (win->w - img->w * img->zoom) / 2;
427 else
428 img->x = 0;
429 if (img->zoom * img->h <= win->h)
430 img->y = (win->h - img->h * img->zoom) / 2;
431 else
432 img->y = 0;
433 }
434
435 if (img->checkpan) {
436 img_check_pan(img, false);
437 img->checkpan = false;
438 }
439
440 if (!img->dirty)
441 return;
442
443 /* calculate source and destination offsets */
444 if (img->x < 0) {
445 sx = -img->x / img->zoom;
446 sw = win->w / img->zoom;
447 dx = 0;
448 dw = win->w;
449 } else {
450 sx = 0;
451 sw = img->w;
452 dx = img->x;
453 dw = img->w * img->zoom;
454 }
455 if (img->y < 0) {
456 sy = -img->y / img->zoom;
457 sh = win->h / img->zoom;
458 dy = 0;
459 dh = win->h;
460 } else {
461 sy = 0;
462 sh = img->h;
463 dy = img->y;
464 dh = img->h * img->zoom;
465 }
466
467 win_clear(win);
468
469 imlib_context_set_image(img->im);
470
471 if (!img->alpha && imlib_image_has_alpha())
472 win_draw_rect(win, win->pm, dx, dy, dw, dh, True, 0, win->white);
473
474 imlib_context_set_drawable(win->pm);
475 imlib_render_image_part_on_drawable_at_size(sx, sy, sw, sh, dx, dy, dw, dh);
476
477 win_draw(win);
478
479 img->dirty = false;
480}
481
482bool img_fit_win(img_t *img) {
483 if (img == NULL || img->im == NULL)
484 return false;
485
486 img->scalemode = SCALE_FIT;
487 return img_fit(img);
488}
489
490bool img_center(img_t *img) {
491 int ox, oy;
492
493 if (img == NULL || img->im == NULL || img->win == NULL)
494 return false;
495
496 ox = img->x;
497 oy = img->y;
498
499 img->x = (img->win->w - img->w * img->zoom) / 2;
500 img->y = (img->win->h - img->h * img->zoom) / 2;
501
502 if (ox != img->x || oy != img->y) {
503 img->dirty = true;
504 return true;
505 } else {
506 return false;
507 }
508}
509
510bool img_zoom(img_t *img, float z) {
511 if (img == NULL || img->im == NULL || img->win == NULL)
512 return false;
513
514 z = MAX(z, zoom_min);
515 z = MIN(z, zoom_max);
516
517 img->scalemode = SCALE_ZOOM;
518
519 if (zoomdiff(z, img->zoom)) {
520 img->x = img->win->w / 2 - (img->win->w / 2 - img->x) * z / img->zoom;
521 img->y = img->win->h / 2 - (img->win->h / 2 - img->y) * z / img->zoom;
522 img->zoom = z;
523 img->checkpan = true;
524 img->dirty = true;
525 return true;
526 } else {
527 return false;
528 }
529}
530
531bool img_zoom_in(img_t *img) {
532 int i;
533
534 if (img == NULL || img->im == NULL)
535 return false;
536
537 for (i = 1; i < ARRLEN(zoom_levels); i++) {
538 if (zoom_levels[i] > img->zoom * 100.0)
539 return img_zoom(img, zoom_levels[i] / 100.0);
540 }
541 return false;
542}
543
544bool img_zoom_out(img_t *img) {
545 int i;
546
547 if (img == NULL || img->im == NULL)
548 return false;
549
550 for (i = ARRLEN(zoom_levels) - 2; i >= 0; i--) {
551 if (zoom_levels[i] < img->zoom * 100.0)
552 return img_zoom(img, zoom_levels[i] / 100.0);
553 }
554 return false;
555}
556
557bool img_move(img_t *img, int dx, int dy) {
558 int ox, oy;
559
560 if (img == NULL || img->im == NULL)
561 return false;
562
563 ox = img->x;
564 oy = img->y;
565
566 img->x += dx;
567 img->y += dy;
568
569 img_check_pan(img, true);
570
571 if (ox != img->x || oy != img->y) {
572 img->dirty = true;
573 return true;
574 } else {
575 return false;
576 }
577}
578
579bool img_pan(img_t *img, direction_t dir, bool screen) {
580 if (img == NULL || img->im == NULL || img->win == NULL)
581 return false;
582
583 switch (dir) {
584 case DIR_LEFT:
585 return img_move(img, img->win->w / (screen ? 1 : 5), 0);
586 case DIR_RIGHT:
587 return img_move(img, img->win->w / (screen ? 1 : 5) * -1, 0);
588 case DIR_UP:
589 return img_move(img, 0, img->win->h / (screen ? 1 : 5));
590 case DIR_DOWN:
591 return img_move(img, 0, img->win->h / (screen ? 1 : 5) * -1);
592 }
593 return false;
594}
595
596bool img_pan_edge(img_t *img, direction_t dir) {
597 int ox, oy;
598
599 if (img == NULL || img->im == NULL || img->win == NULL)
600 return false;
601
602 ox = img->x;
603 oy = img->y;
604
605 switch (dir) {
606 case DIR_LEFT:
607 img->x = 0;
608 break;
609 case DIR_RIGHT:
610 img->x = img->win->w - img->w * img->zoom;
611 break;
612 case DIR_UP:
613 img->y = 0;
614 break;
615 case DIR_DOWN:
616 img->y = img->win->h - img->h * img->zoom;
617 break;
618 }
619
620 img_check_pan(img, true);
621
622 if (ox != img->x || oy != img->y) {
623 img->dirty = true;
624 return true;
625 } else {
626 return false;
627 }
628}
629
630void img_rotate(img_t *img, int d) {
631 win_t *win;
632 int ox, oy, tmp;
633
634 if (img == NULL || img->im == NULL || img->win == NULL)
635 return;
636
637 win = img->win;
638 ox = d == 1 ? img->x : win->w - img->x - img->w * img->zoom;
639 oy = d == 3 ? img->y : win->h - img->y - img->h * img->zoom;
640
641 imlib_context_set_image(img->im);
642 imlib_image_orientate(d);
643
644 img->x = oy + (win->w - win->h) / 2;
645 img->y = ox + (win->h - win->w) / 2;
646
647 tmp = img->w;
648 img->w = img->h;
649 img->h = tmp;
650
651 img->checkpan = true;
652 img->dirty = true;
653}
654
655void img_rotate_left(img_t *img) {
656 img_rotate(img, 3);
657}
658
659void img_rotate_right(img_t *img) {
660 img_rotate(img, 1);
661}
662
663void img_toggle_antialias(img_t *img) {
664 if (img == NULL || img->im == NULL)
665 return;
666
667 img->aa = !img->aa;
668 imlib_context_set_image(img->im);
669 imlib_context_set_anti_alias(img->aa);
670 img->dirty = true;
671}
672
673bool img_frame_goto(img_t *img, int n) {
674 if (img == NULL || img->im == NULL)
675 return false;
676 if (n < 0 || n >= img->multi.cnt || n == img->multi.sel)
677 return false;
678
679 img->multi.sel = n;
680 img->im = img->multi.frames[n].im;
681
682 imlib_context_set_image(img->im);
683 img->w = imlib_image_get_width();
684 img->h = imlib_image_get_height();
685 img->checkpan = true;
686 img->dirty = true;
687
688 return true;
689}
690
691bool img_frame_navigate(img_t *img, int d) {
692 if (img == NULL|| img->im == NULL || img->multi.cnt == 0 || d == 0)
693 return false;
694
695 d += img->multi.sel;
696 if (d < 0)
697 d = 0;
698 else if (d >= img->multi.cnt)
699 d = img->multi.cnt - 1;
700
701 return img_frame_goto(img, d);
702}
703
704bool img_frame_animate(img_t *img, bool restart) {
705 if (img == NULL || img->im == NULL || img->multi.cnt == 0)
706 return false;
707
708 if (img->multi.sel + 1 >= img->multi.cnt) {
709 if (restart || (GIF_LOOP && !img->slideshow)) {
710 img_frame_goto(img, 0);
711 } else {
712 img->multi.animate = false;
713 return false;
714 }
715 } else if (!restart) {
716 img_frame_goto(img, img->multi.sel + 1);
717 }
718 img->multi.animate = true;
719 img->dirty = true;
720
721 return true;
722}