Krolyxon
Commit e2d4b9c

Branches

Refs →

nsxivimage.c

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