Krolyxon
Commit 6695cd4

Branches

Refs →

nsxivimage.c

777 lines · 16.6 KB · C

Raw Blame History Download
1/* Copyright 2011, 2012 Bert Muennich
2 *
3 * This file is part of sxiv.
4 *
5 * sxiv is free software; you can redistribute it and/or modify
6 * it under the terms of the GNU General Public License as published
7 * by the Free Software Foundation; either version 2 of the License,
8 * or (at your option) any later version.
9 *
10 * sxiv is distributed in the hope that it will be useful,
11 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
13 * GNU General Public License for more details.
14 *
15 * You should have received a copy of the GNU General Public License
16 * along with sxiv. If not, see <http://www.gnu.org/licenses/>.
17 */
18
19#include <errno.h>
20#include <stdlib.h>
21#include <string.h>
22#include <sys/stat.h>
23#include <sys/types.h>
24#include <unistd.h>
25
26#include "image.h"
27#include "options.h"
28#include "util.h"
29
30#define _IMAGE_CONFIG
31#include "config.h"
32
33#if HAVE_LIBEXIF
34#include <libexif/exif-data.h>
35#endif
36
37#if HAVE_GIFLIB
38#include <gif_lib.h>
39enum { DEF_GIF_DELAY = 75 };
40#endif
41
42float zoom_min;
43float zoom_max;
44
45static int zoomdiff(float z1, float z2)
46{
47 return (int) (z1 * 1000.0 - z2 * 1000.0);
48}
49
50void img_init(img_t *img, win_t *win)
51{
52 zoom_min = zoom_levels[0] / 100.0;
53 zoom_max = zoom_levels[ARRLEN(zoom_levels) - 1] / 100.0;
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->scalemode = options->scalemode;
62 img->zoom = options->zoom;
63 img->zoom = MAX(img->zoom, zoom_min);
64 img->zoom = MIN(img->zoom, zoom_max);
65 img->checkpan = false;
66 img->dirty = false;
67 img->aa = ANTI_ALIAS;
68 img->alpha = ALPHA_LAYER;
69 img->multi.cap = img->multi.cnt = 0;
70 img->multi.animate = options->animate;
71 img->multi.framedelay = options->framerate > 0 ? 1000 / options->framerate : 0;
72 img->multi.length = 0;
73
74 img->cmod = imlib_create_color_modifier();
75 imlib_context_set_color_modifier(img->cmod);
76 img->gamma = MIN(MAX(options->gamma, -GAMMA_RANGE), GAMMA_RANGE);
77
78 img->ss.on = options->slideshow > 0;
79 img->ss.delay = options->slideshow > 0 ? options->slideshow : SLIDESHOW_DELAY * 10;
80}
81
82#if HAVE_LIBEXIF
83void exif_auto_orientate(const fileinfo_t *file)
84{
85 ExifData *ed;
86 ExifEntry *entry;
87 int byte_order, orientation = 0;
88
89 if ((ed = exif_data_new_from_file(file->path)) == NULL)
90 return;
91 byte_order = exif_data_get_byte_order(ed);
92 entry = exif_content_get_entry(ed->ifd[EXIF_IFD_0], EXIF_TAG_ORIENTATION);
93 if (entry != NULL)
94 orientation = exif_get_short(entry->data, byte_order);
95 exif_data_unref(ed);
96
97 switch (orientation) {
98 case 5:
99 imlib_image_orientate(1);
100 case 2:
101 imlib_image_flip_vertical();
102 break;
103 case 3:
104 imlib_image_orientate(2);
105 break;
106 case 7:
107 imlib_image_orientate(1);
108 case 4:
109 imlib_image_flip_horizontal();
110 break;
111 case 6:
112 imlib_image_orientate(1);
113 break;
114 case 8:
115 imlib_image_orientate(3);
116 break;
117 }
118}
119#endif
120
121#if HAVE_GIFLIB
122bool img_load_gif(img_t *img, const fileinfo_t *file)
123{
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 px, py, pw, ph;
134 int intoffset[] = { 0, 4, 2, 1 };
135 int intjump[] = { 8, 8, 4, 2 };
136 int transp = -1;
137 unsigned int disposal = 0, prev_disposal = 0;
138 unsigned int delay = 0;
139 bool err = false;
140
141 if (img->multi.cap == 0) {
142 img->multi.cap = 8;
143 img->multi.frames = (img_frame_t*)
144 emalloc(sizeof(img_frame_t) * img->multi.cap);
145 }
146 img->multi.cnt = img->multi.sel = 0;
147 img->multi.length = 0;
148
149#if defined(GIFLIB_MAJOR) && GIFLIB_MAJOR >= 5
150 gif = DGifOpenFileName(file->path, NULL);
151#else
152 gif = DGifOpenFileName(file->path);
153#endif
154 if (gif == NULL) {
155 error(0, 0, "%s: Error opening gif image", file->name);
156 return false;
157 }
158 bg = gif->SBackGroundColor;
159 sw = gif->SWidth;
160 sh = gif->SHeight;
161 px = py = pw = ph = 0;
162
163 do {
164 if (DGifGetRecordType(gif, &rec) == GIF_ERROR) {
165 err = true;
166 break;
167 }
168 if (rec == EXTENSION_RECORD_TYPE) {
169 int ext_code;
170 GifByteType *ext = NULL;
171
172 DGifGetExtension(gif, &ext_code, &ext);
173 while (ext) {
174 if (ext_code == GRAPHICS_EXT_FUNC_CODE) {
175 if (ext[1] & 1)
176 transp = (int) ext[4];
177 else
178 transp = -1;
179
180 delay = 10 * ((unsigned int) ext[3] << 8 | (unsigned int) ext[2]);
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*) emalloc(h * sizeof(GifRowType));
197 for (i = 0; i < h; i++)
198 rows[i] = (GifRowType) emalloc(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*) emalloc(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_frame != NULL && (prev_disposal != 2 ||
222 i < py || i >= py + ph || j < px || j >= px + pw))
223 {
224 *ptr = prev_frame[i * sw + j];
225 } else {
226 *ptr = bgpixel;
227 }
228 } else {
229 r = cmap->Colors[rows[i-y][j-x]].Red;
230 g = cmap->Colors[rows[i-y][j-x]].Green;
231 b = cmap->Colors[rows[i-y][j-x]].Blue;
232 *ptr = 0xffu << 24 | r << 16 | g << 8 | b;
233 }
234 ptr++;
235 }
236 }
237
238 im = imlib_create_image_using_copied_data(sw, sh, data);
239
240 for (i = 0; i < h; i++)
241 free(rows[i]);
242 free(rows);
243 free(data);
244
245 if (im == NULL) {
246 err = true;
247 break;
248 }
249
250 imlib_context_set_image(im);
251 imlib_image_set_format("gif");
252 if (transp >= 0)
253 imlib_image_set_has_alpha(1);
254
255 if (disposal != 3)
256 prev_frame = imlib_image_get_data_for_reading_only();
257 prev_disposal = disposal;
258 px = x, py = y, pw = w, ph = h;
259
260 if (img->multi.cnt == img->multi.cap) {
261 img->multi.cap *= 2;
262 img->multi.frames = (img_frame_t*)
263 erealloc(img->multi.frames,
264 img->multi.cap * sizeof(img_frame_t));
265 }
266 img->multi.frames[img->multi.cnt].im = im;
267 delay = img->multi.framedelay > 0 ? img->multi.framedelay : delay;
268 img->multi.frames[img->multi.cnt].delay = delay > 0 ? delay : DEF_GIF_DELAY;
269 img->multi.length += img->multi.frames[img->multi.cnt].delay;
270 img->multi.cnt++;
271 }
272 } while (rec != TERMINATE_RECORD_TYPE);
273
274#if defined(GIFLIB_MAJOR) && GIFLIB_MAJOR >= 5 && GIFLIB_MINOR >= 1
275 DGifCloseFile(gif, NULL);
276#else
277 DGifCloseFile(gif);
278#endif
279
280 if (err && (file->flags & FF_WARN))
281 error(0, 0, "%s: Corrupted gif file", file->name);
282
283 if (img->multi.cnt > 1) {
284 imlib_context_set_image(img->im);
285 imlib_free_image();
286 img->im = img->multi.frames[0].im;
287 } else if (img->multi.cnt == 1) {
288 imlib_context_set_image(img->multi.frames[0].im);
289 imlib_free_image();
290 img->multi.cnt = 0;
291 }
292
293 imlib_context_set_image(img->im);
294
295 return !err;
296}
297#endif /* HAVE_GIFLIB */
298
299bool img_load(img_t *img, const fileinfo_t *file)
300{
301 const char *fmt;
302 struct stat st;
303
304 if (access(file->path, R_OK) == -1 ||
305 stat(file->path, &st) == -1 || !S_ISREG(st.st_mode) ||
306 (img->im = imlib_load_image(file->path)) == NULL)
307 {
308 if (file->flags & FF_WARN)
309 error(0, 0, "%s: Error opening image", file->name);
310 return false;
311 }
312
313 imlib_context_set_image(img->im);
314 imlib_image_set_changes_on_disk();
315
316#if HAVE_LIBEXIF
317 exif_auto_orientate(file);
318#endif
319
320 if ((fmt = imlib_image_format()) != NULL) {
321#if HAVE_GIFLIB
322 if (STREQ(fmt, "gif"))
323 img_load_gif(img, file);
324#endif
325 }
326 img->w = imlib_image_get_width();
327 img->h = imlib_image_get_height();
328 img->checkpan = true;
329 img->dirty = true;
330
331 return true;
332}
333
334CLEANUP void img_close(img_t *img, bool decache)
335{
336 int i;
337
338 if (img->multi.cnt > 0) {
339 for (i = 0; i < img->multi.cnt; i++) {
340 imlib_context_set_image(img->multi.frames[i].im);
341 imlib_free_image();
342 }
343 img->multi.cnt = 0;
344 img->im = NULL;
345 } else if (img->im != NULL) {
346 imlib_context_set_image(img->im);
347 if (decache)
348 imlib_free_image_and_decache();
349 else
350 imlib_free_image();
351 img->im = NULL;
352 }
353}
354
355void img_check_pan(img_t *img, bool moved)
356{
357 win_t *win;
358 int ox, oy;
359 float w, h;
360
361 win = img->win;
362 w = img->w * img->zoom;
363 h = img->h * img->zoom;
364 ox = img->x;
365 oy = img->y;
366
367 if (w < win->w)
368 img->x = (win->w - w) / 2;
369 else if (img->x > 0)
370 img->x = 0;
371 else if (img->x + w < win->w)
372 img->x = win->w - w;
373 if (h < win->h)
374 img->y = (win->h - h) / 2;
375 else if (img->y > 0)
376 img->y = 0;
377 else if (img->y + h < win->h)
378 img->y = win->h - h;
379
380 if (!moved && (ox != img->x || oy != img->y))
381 img->dirty = true;
382}
383
384bool img_fit(img_t *img)
385{
386 float z, zmax, zw, zh;
387
388 if (img->scalemode == SCALE_ZOOM)
389 return false;
390
391 zmax = img->scalemode == SCALE_DOWN ? 1.0 : zoom_max;
392 zw = (float) img->win->w / (float) img->w;
393 zh = (float) img->win->h / (float) img->h;
394
395 switch (img->scalemode) {
396 case SCALE_WIDTH:
397 z = zw;
398 break;
399 case SCALE_HEIGHT:
400 z = zh;
401 break;
402 default:
403 z = MIN(zw, zh);
404 break;
405 }
406
407 z = MAX(z, zoom_min);
408 z = MIN(z, zmax);
409
410 if (zoomdiff(z, img->zoom) != 0) {
411 img->zoom = z;
412 img->dirty = true;
413 return true;
414 } else {
415 return false;
416 }
417}
418
419void img_render(img_t *img)
420{
421 win_t *win;
422 int sx, sy, sw, sh;
423 int dx, dy, dw, dh;
424 Imlib_Image bg;
425 unsigned long c;
426
427 win = img->win;
428 img_fit(img);
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 * - part of image drawn on full window, or
440 * - full image drawn on part of window
441 */
442 if (img->x <= 0) {
443 sx = -img->x / img->zoom + 0.5;
444 sw = win->w / img->zoom;
445 dx = 0;
446 dw = win->w;
447 } else {
448 sx = 0;
449 sw = img->w;
450 dx = img->x;
451 dw = img->w * img->zoom;
452 }
453 if (img->y <= 0) {
454 sy = -img->y / img->zoom + 0.5;
455 sh = win->h / img->zoom;
456 dy = 0;
457 dh = win->h;
458 } else {
459 sy = 0;
460 sh = img->h;
461 dy = img->y;
462 dh = img->h * img->zoom;
463 }
464
465 win_clear(win);
466
467 imlib_context_set_image(img->im);
468 imlib_context_set_anti_alias(img->aa);
469 imlib_context_set_drawable(win->buf.pm);
470
471 if (imlib_image_has_alpha()) {
472 if ((bg = imlib_create_image(dw, dh)) == NULL)
473 error(EXIT_FAILURE, ENOMEM, NULL);
474 imlib_context_set_image(bg);
475 imlib_image_set_has_alpha(0);
476
477 if (img->alpha) {
478 int i, c, r;
479 DATA32 col[2] = { 0xFF666666, 0xFF999999 };
480 DATA32 * data = imlib_image_get_data();
481
482 for (r = 0; r < dh; r++) {
483 i = r * dw;
484 if (r == 0 || r == 8) {
485 for (c = 0; c < dw; c++)
486 data[i++] = col[!(c & 8) ^ !r];
487 } else {
488 memcpy(&data[i], &data[(r & 8) * dw], dw * sizeof(data[0]));
489 }
490 }
491 imlib_image_put_back_data(data);
492 } else {
493 c = win->fullscreen ? win->fscol.pixel : win->bgcol.pixel;
494 imlib_context_set_color(c >> 16 & 0xFF, c >> 8 & 0xFF, c & 0xFF, 0xFF);
495 imlib_image_fill_rectangle(0, 0, dw, dh);
496 }
497 imlib_blend_image_onto_image(img->im, 0, sx, sy, sw, sh, 0, 0, dw, dh);
498 imlib_context_set_color_modifier(NULL);
499 imlib_render_image_on_drawable(dx, dy);
500 imlib_free_image();
501 imlib_context_set_color_modifier(img->cmod);
502 } else {
503 imlib_render_image_part_on_drawable_at_size(sx, sy, sw, sh, dx, dy, dw, dh);
504 }
505 img->dirty = false;
506}
507
508bool img_fit_win(img_t *img, scalemode_t sm)
509{
510 float oz;
511
512 oz = img->zoom;
513 img->scalemode = sm;
514
515 if (img_fit(img)) {
516 img->x = img->win->w / 2 - (img->win->w / 2 - img->x) * img->zoom / oz;
517 img->y = img->win->h / 2 - (img->win->h / 2 - img->y) * img->zoom / oz;
518 img->checkpan = true;
519 return true;
520 } else {
521 return false;
522 }
523}
524
525bool img_zoom(img_t *img, float z)
526{
527 z = MAX(z, zoom_min);
528 z = MIN(z, zoom_max);
529
530 img->scalemode = SCALE_ZOOM;
531
532 if (zoomdiff(z, img->zoom) != 0) {
533 img->x = img->win->w / 2 - (img->win->w / 2 - img->x) * z / img->zoom;
534 img->y = img->win->h / 2 - (img->win->h / 2 - img->y) * z / img->zoom;
535 img->zoom = z;
536 img->checkpan = true;
537 img->dirty = true;
538 return true;
539 } else {
540 return false;
541 }
542}
543
544bool img_zoom_in(img_t *img)
545{
546 int i;
547 float z;
548
549 for (i = 1; i < ARRLEN(zoom_levels); i++) {
550 z = zoom_levels[i] / 100.0;
551 if (zoomdiff(z, img->zoom) > 0)
552 return img_zoom(img, z);
553 }
554 return false;
555}
556
557bool img_zoom_out(img_t *img)
558{
559 int i;
560 float z;
561
562 for (i = ARRLEN(zoom_levels) - 2; i >= 0; i--) {
563 z = zoom_levels[i] / 100.0;
564 if (zoomdiff(z, img->zoom) < 0)
565 return img_zoom(img, z);
566 }
567 return false;
568}
569
570bool img_move(img_t *img, float dx, float dy)
571{
572 float ox, oy;
573
574 ox = img->x;
575 oy = img->y;
576
577 img->x += dx;
578 img->y += dy;
579
580 img_check_pan(img, true);
581
582 if (ox != img->x || oy != img->y) {
583 img->dirty = true;
584 return true;
585 } else {
586 return false;
587 }
588}
589
590bool img_pan(img_t *img, direction_t dir, int d)
591{
592 /* d < 0: screen-wise
593 * d = 0: 1/5 of screen
594 * d > 0: num of pixels
595 */
596 float x, y;
597
598 if (d > 0) {
599 x = y = MAX(1, (float) d * img->zoom);
600 } else {
601 x = img->win->w / (d < 0 ? 1 : 5);
602 y = img->win->h / (d < 0 ? 1 : 5);
603 }
604
605 switch (dir) {
606 case DIR_LEFT:
607 return img_move(img, x, 0.0);
608 case DIR_RIGHT:
609 return img_move(img, -x, 0.0);
610 case DIR_UP:
611 return img_move(img, 0.0, y);
612 case DIR_DOWN:
613 return img_move(img, 0.0, -y);
614 }
615 return false;
616}
617
618bool img_pan_edge(img_t *img, direction_t dir)
619{
620 int ox, oy;
621
622 ox = img->x;
623 oy = img->y;
624
625 if (dir & DIR_LEFT)
626 img->x = 0;
627 if (dir & DIR_RIGHT)
628 img->x = img->win->w - img->w * img->zoom;
629 if (dir & DIR_UP)
630 img->y = 0;
631 if (dir & DIR_DOWN)
632 img->y = img->win->h - img->h * img->zoom;
633
634 img_check_pan(img, true);
635
636 if (ox != img->x || oy != img->y) {
637 img->dirty = true;
638 return true;
639 } else {
640 return false;
641 }
642}
643
644void img_rotate(img_t *img, degree_t d)
645{
646 int i, ox, oy, tmp;
647
648 imlib_context_set_image(img->im);
649 imlib_image_orientate(d);
650
651 for (i = 0; i < img->multi.cnt; i++) {
652 if (i != img->multi.sel) {
653 imlib_context_set_image(img->multi.frames[i].im);
654 imlib_image_orientate(d);
655 }
656 }
657 if (d == DEGREE_90 || d == DEGREE_270) {
658 ox = d == DEGREE_90 ? img->x : img->win->w - img->x - img->w * img->zoom;
659 oy = d == DEGREE_270 ? img->y : img->win->h - img->y - img->h * img->zoom;
660
661 img->x = oy + (img->win->w - img->win->h) / 2;
662 img->y = ox + (img->win->h - img->win->w) / 2;
663
664 tmp = img->w;
665 img->w = img->h;
666 img->h = tmp;
667 img->checkpan = true;
668 }
669 img->dirty = true;
670}
671
672void img_flip(img_t *img, flipdir_t d)
673{
674 int i;
675 void (*imlib_flip_op[3])(void) = {
676 imlib_image_flip_horizontal,
677 imlib_image_flip_vertical,
678 imlib_image_flip_diagonal
679 };
680
681 d = (d & (FLIP_HORIZONTAL | FLIP_VERTICAL)) - 1;
682
683 if (d < 0 || d >= ARRLEN(imlib_flip_op))
684 return;
685
686 imlib_context_set_image(img->im);
687 imlib_flip_op[d]();
688
689 for (i = 0; i < img->multi.cnt; i++) {
690 if (i != img->multi.sel) {
691 imlib_context_set_image(img->multi.frames[i].im);
692 imlib_flip_op[d]();
693 }
694 }
695 img->dirty = true;
696}
697
698void img_toggle_antialias(img_t *img)
699{
700 img->aa = !img->aa;
701 imlib_context_set_image(img->im);
702 imlib_context_set_anti_alias(img->aa);
703 img->dirty = true;
704}
705
706bool img_change_gamma(img_t *img, int d)
707{
708 /* d < 0: decrease gamma
709 * d = 0: reset gamma
710 * d > 0: increase gamma
711 */
712 int gamma;
713 double range;
714
715 if (d == 0)
716 gamma = 0;
717 else
718 gamma = MIN(MAX(img->gamma + d, -GAMMA_RANGE), GAMMA_RANGE);
719
720 if (img->gamma != gamma) {
721 imlib_reset_color_modifier();
722 if (gamma != 0) {
723 range = gamma <= 0 ? 1.0 : GAMMA_MAX - 1.0;
724 imlib_modify_color_modifier_gamma(1.0 + gamma * (range / GAMMA_RANGE));
725 }
726 img->gamma = gamma;
727 img->dirty = true;
728 return true;
729 } else {
730 return false;
731 }
732}
733
734bool img_frame_goto(img_t *img, int n)
735{
736 if (n < 0 || n >= img->multi.cnt || n == img->multi.sel)
737 return false;
738
739 img->multi.sel = n;
740 img->im = img->multi.frames[n].im;
741
742 imlib_context_set_image(img->im);
743 img->w = imlib_image_get_width();
744 img->h = imlib_image_get_height();
745 img->checkpan = true;
746 img->dirty = true;
747
748 return true;
749}
750
751bool img_frame_navigate(img_t *img, int d)
752{
753 if (img->multi.cnt == 0 || d == 0)
754 return false;
755
756 d += img->multi.sel;
757 if (d < 0)
758 d = 0;
759 else if (d >= img->multi.cnt)
760 d = img->multi.cnt - 1;
761
762 return img_frame_goto(img, d);
763}
764
765bool img_frame_animate(img_t *img)
766{
767 if (img->multi.cnt == 0)
768 return false;
769
770 if (img->multi.sel + 1 >= img->multi.cnt)
771 img_frame_goto(img, 0);
772 else
773 img_frame_goto(img, img->multi.sel + 1);
774 img->dirty = true;
775 return true;
776}
777