Krolyxon
Commit 8dcf682

Branches

Refs →

nsxivimage.c

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