Krolyxon
Commit 9dabc5f

Branches

Refs →

nsxivimage.c

786 lines · 16.7 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 "sxiv.h"
20#define _IMAGE_CONFIG
21#include "config.h"
22
23#include <errno.h>
24#include <stdlib.h>
25#include <string.h>
26#include <sys/stat.h>
27#include <sys/types.h>
28#include <unistd.h>
29
30#if HAVE_LIBEXIF
31#include <libexif/exif-data.h>
32#endif
33
34#if HAVE_GIFLIB
35#include <gif_lib.h>
36enum { DEF_GIF_DELAY = 75 };
37#endif
38
39float zoom_min;
40float zoom_max;
41
42static int zoomdiff(float z1, float z2)
43{
44 return (int) (z1 * 1000.0 - z2 * 1000.0);
45}
46
47void img_init(img_t *img, win_t *win)
48{
49 zoom_min = zoom_levels[0] / 100.0;
50 zoom_max = zoom_levels[ARRLEN(zoom_levels) - 1] / 100.0;
51
52 imlib_context_set_display(win->env.dpy);
53 imlib_context_set_visual(win->env.vis);
54 imlib_context_set_colormap(win->env.cmap);
55
56 img->im = NULL;
57 img->win = win;
58 img->scalemode = options->scalemode;
59 img->zoom = options->zoom;
60 img->zoom = MAX(img->zoom, zoom_min);
61 img->zoom = MIN(img->zoom, zoom_max);
62 img->checkpan = false;
63 img->dirty = false;
64 img->aa = ANTI_ALIAS;
65 img->alpha = ALPHA_LAYER;
66 img->multi.cap = img->multi.cnt = 0;
67 img->multi.animate = options->animate;
68 img->multi.framedelay = options->framerate > 0 ? 1000 / options->framerate : 0;
69 img->multi.length = 0;
70
71 img->cmod = imlib_create_color_modifier();
72 imlib_context_set_color_modifier(img->cmod);
73 img->gamma = MIN(MAX(options->gamma, -GAMMA_RANGE), GAMMA_RANGE);
74
75 img->ss.on = options->slideshow > 0;
76 img->ss.delay = options->slideshow > 0 ? options->slideshow : SLIDESHOW_DELAY * 10;
77}
78
79#if HAVE_LIBEXIF
80void exif_auto_orientate(const fileinfo_t *file)
81{
82 ExifData *ed;
83 ExifEntry *entry;
84 int byte_order, orientation = 0;
85
86 if ((ed = exif_data_new_from_file(file->path)) == NULL)
87 return;
88 byte_order = exif_data_get_byte_order(ed);
89 entry = exif_content_get_entry(ed->ifd[EXIF_IFD_0], EXIF_TAG_ORIENTATION);
90 if (entry != NULL)
91 orientation = exif_get_short(entry->data, byte_order);
92 exif_data_unref(ed);
93
94 switch (orientation) {
95 case 5:
96 imlib_image_orientate(1);
97 case 2:
98 imlib_image_flip_vertical();
99 break;
100 case 3:
101 imlib_image_orientate(2);
102 break;
103 case 7:
104 imlib_image_orientate(1);
105 case 4:
106 imlib_image_flip_horizontal();
107 break;
108 case 6:
109 imlib_image_orientate(1);
110 break;
111 case 8:
112 imlib_image_orientate(3);
113 break;
114 }
115}
116#endif
117
118#if HAVE_GIFLIB
119bool img_load_gif(img_t *img, const fileinfo_t *file)
120{
121 GifFileType *gif;
122 GifRowType *rows = NULL;
123 GifRecordType rec;
124 ColorMapObject *cmap;
125 DATA32 bgpixel, *data, *ptr;
126 DATA32 *prev_frame = NULL;
127 Imlib_Image im;
128 int i, j, bg, r, g, b;
129 int x, y, w, h, sw, sh;
130 int px, py, pw, ph;
131 int intoffset[] = { 0, 4, 2, 1 };
132 int intjump[] = { 8, 8, 4, 2 };
133 int transp = -1;
134 unsigned int disposal = 0, prev_disposal = 0;
135 unsigned int delay = 0;
136 bool err = false;
137
138 if (img->multi.cap == 0) {
139 img->multi.cap = 8;
140 img->multi.frames = (img_frame_t*)
141 emalloc(sizeof(img_frame_t) * img->multi.cap);
142 }
143 img->multi.cnt = img->multi.sel = 0;
144 img->multi.length = 0;
145
146#if defined(GIFLIB_MAJOR) && GIFLIB_MAJOR >= 5
147 gif = DGifOpenFileName(file->path, NULL);
148#else
149 gif = DGifOpenFileName(file->path);
150#endif
151 if (gif == NULL) {
152 error(0, 0, "%s: Error opening gif image", file->name);
153 return false;
154 }
155 bg = gif->SBackGroundColor;
156 sw = gif->SWidth;
157 sh = gif->SHeight;
158 px = py = pw = ph = 0;
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 == GRAPHICS_EXT_FUNC_CODE) {
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 disposal = (unsigned int) ext[1] >> 2 & 0x7;
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*) emalloc(h * sizeof(GifRowType));
194 for (i = 0; i < h; i++)
195 rows[i] = (GifRowType) emalloc(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*) emalloc(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 && (prev_disposal != 2 ||
219 i < py || i >= py + ph || j < px || j >= px + pw))
220 {
221 *ptr = prev_frame[i * sw + j];
222 } else {
223 *ptr = bgpixel;
224 }
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 = 0xffu << 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 px = x, py = y, pw = w, ph = h;
256
257 if (img->multi.cnt == img->multi.cap) {
258 img->multi.cap *= 2;
259 img->multi.frames = (img_frame_t*)
260 erealloc(img->multi.frames,
261 img->multi.cap * sizeof(img_frame_t));
262 }
263 img->multi.frames[img->multi.cnt].im = im;
264 delay = img->multi.framedelay > 0 ? img->multi.framedelay : delay;
265 img->multi.frames[img->multi.cnt].delay = delay > 0 ? delay : DEF_GIF_DELAY;
266 img->multi.length += img->multi.frames[img->multi.cnt].delay;
267 img->multi.cnt++;
268 }
269 } while (rec != TERMINATE_RECORD_TYPE);
270
271#if defined(GIFLIB_MAJOR) && GIFLIB_MAJOR >= 5 && GIFLIB_MINOR >= 1
272 DGifCloseFile(gif, NULL);
273#else
274 DGifCloseFile(gif);
275#endif
276
277 if (err && (file->flags & FF_WARN))
278 error(0, 0, "%s: Corrupted gif file", file->name);
279
280 if (img->multi.cnt > 1) {
281 imlib_context_set_image(img->im);
282 imlib_free_image();
283 img->im = img->multi.frames[0].im;
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 }
289
290 imlib_context_set_image(img->im);
291
292 return !err;
293}
294#endif /* HAVE_GIFLIB */
295
296bool img_load(img_t *img, const fileinfo_t *file)
297{
298 const char *fmt;
299 struct stat st;
300
301 if (access(file->path, R_OK) == -1 ||
302 stat(file->path, &st) == -1 || !S_ISREG(st.st_mode) ||
303 (img->im = imlib_load_image(file->path)) == NULL)
304 {
305 if (file->flags & FF_WARN)
306 error(0, 0, "%s: Error opening image", file->name);
307 return false;
308 }
309
310 imlib_context_set_image(img->im);
311 imlib_image_set_changes_on_disk();
312
313#if HAVE_LIBEXIF
314 exif_auto_orientate(file);
315#endif
316
317 if ((fmt = imlib_image_format()) != NULL) {
318#if HAVE_GIFLIB
319 if (STREQ(fmt, "gif"))
320 img_load_gif(img, file);
321#endif
322 }
323 img->w = imlib_image_get_width();
324 img->h = imlib_image_get_height();
325 img->checkpan = true;
326 img->dirty = true;
327
328 return true;
329}
330
331CLEANUP void img_close(img_t *img, bool decache)
332{
333 int i;
334
335 if (img->multi.cnt > 0) {
336 for (i = 0; i < img->multi.cnt; i++) {
337 imlib_context_set_image(img->multi.frames[i].im);
338 imlib_free_image();
339 }
340 img->multi.cnt = 0;
341 img->im = NULL;
342 } else if (img->im != NULL) {
343 imlib_context_set_image(img->im);
344 if (decache)
345 imlib_free_image_and_decache();
346 else
347 imlib_free_image();
348 img->im = NULL;
349 }
350}
351
352void img_check_pan(img_t *img, bool moved)
353{
354 win_t *win;
355 int ox, oy;
356 float w, h;
357
358 win = img->win;
359 w = img->w * img->zoom;
360 h = img->h * img->zoom;
361 ox = img->x;
362 oy = img->y;
363
364 if (w < win->w)
365 img->x = (win->w - w) / 2;
366 else if (img->x > 0)
367 img->x = 0;
368 else if (img->x + w < win->w)
369 img->x = win->w - w;
370 if (h < win->h)
371 img->y = (win->h - h) / 2;
372 else if (img->y > 0)
373 img->y = 0;
374 else if (img->y + h < win->h)
375 img->y = win->h - h;
376
377 if (!moved && (ox != img->x || oy != img->y))
378 img->dirty = true;
379}
380
381bool img_fit(img_t *img)
382{
383 float z, zmax, zw, zh;
384
385 if (img->scalemode == SCALE_ZOOM)
386 return false;
387
388 zmax = img->scalemode == SCALE_DOWN ? 1.0 : zoom_max;
389 zw = (float) img->win->w / (float) img->w;
390 zh = (float) img->win->h / (float) img->h;
391
392 switch (img->scalemode) {
393 case SCALE_WIDTH:
394 z = zw;
395 break;
396 case SCALE_HEIGHT:
397 z = zh;
398 break;
399 default:
400 z = MIN(zw, zh);
401 break;
402 }
403
404 z = MAX(z, zoom_min);
405 z = MIN(z, zmax);
406
407 if (zoomdiff(z, img->zoom) != 0) {
408 img->zoom = z;
409 img->dirty = true;
410 return true;
411 } else {
412 return false;
413 }
414}
415
416void img_render(img_t *img)
417{
418 win_t *win;
419 int sx, sy, sw, sh;
420 int dx, dy, dw, dh;
421 Imlib_Image bg;
422 unsigned long c;
423
424 win = img->win;
425 img_fit(img);
426
427 if (img->checkpan) {
428 img_check_pan(img, false);
429 img->checkpan = false;
430 }
431
432 if (!img->dirty)
433 return;
434
435 /* calculate source and destination offsets:
436 * - part of image drawn on full window, or
437 * - full image drawn on part of window
438 */
439 if (img->x <= 0) {
440 sx = -img->x / img->zoom + 0.5;
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 + 0.5;
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 imlib_context_set_anti_alias(img->aa);
466 imlib_context_set_drawable(win->buf.pm);
467
468 if (imlib_image_has_alpha()) {
469 if ((bg = imlib_create_image(dw, dh)) == NULL)
470 error(EXIT_FAILURE, ENOMEM, NULL);
471 imlib_context_set_image(bg);
472 imlib_image_set_has_alpha(0);
473
474 if (img->alpha) {
475 int i, c, r;
476 DATA32 col[2] = { 0xFF666666, 0xFF999999 };
477 DATA32 * data = imlib_image_get_data();
478
479 for (r = 0; r < dh; r++) {
480 i = r * dw;
481 if (r == 0 || r == 8) {
482 for (c = 0; c < dw; c++)
483 data[i++] = col[!(c & 8) ^ !r];
484 } else {
485 memcpy(&data[i], &data[(r & 8) * dw], dw * sizeof(data[0]));
486 }
487 }
488 imlib_image_put_back_data(data);
489 } else {
490 c = win->fullscreen ? win->fscol.pixel : win->bgcol.pixel;
491 imlib_context_set_color(c >> 16 & 0xFF, c >> 8 & 0xFF, c & 0xFF, 0xFF);
492 imlib_image_fill_rectangle(0, 0, dw, dh);
493 }
494 imlib_blend_image_onto_image(img->im, 0, sx, sy, sw, sh, 0, 0, dw, dh);
495 imlib_context_set_color_modifier(NULL);
496 imlib_render_image_on_drawable(dx, dy);
497 imlib_free_image();
498 imlib_context_set_color_modifier(img->cmod);
499 } else {
500 imlib_render_image_part_on_drawable_at_size(sx, sy, sw, sh, dx, dy, dw, dh);
501 }
502 img->dirty = false;
503}
504
505bool img_fit_win(img_t *img, scalemode_t sm)
506{
507 float oz;
508
509 oz = img->zoom;
510 img->scalemode = sm;
511
512 if (img_fit(img)) {
513 img->x = img->win->w / 2 - (img->win->w / 2 - img->x) * img->zoom / oz;
514 img->y = img->win->h / 2 - (img->win->h / 2 - img->y) * img->zoom / oz;
515 img->checkpan = true;
516 return true;
517 } else {
518 return false;
519 }
520}
521
522bool img_zoom(img_t *img, float z)
523{
524 z = MAX(z, zoom_min);
525 z = MIN(z, zoom_max);
526
527 img->scalemode = SCALE_ZOOM;
528
529 if (zoomdiff(z, img->zoom) != 0) {
530 int x, y;
531
532 win_cursor_pos(img->win, &x, &y);
533 if (x < 0 || x >= img->win->w || y < 0 || y >= img->win->h) {
534 x = img->win->w / 2;
535 y = img->win->h / 2;
536 }
537 img->x = x - (x - img->x) * z / img->zoom;
538 img->y = y - (y - img->y) * z / img->zoom;
539 img->zoom = z;
540 img->checkpan = true;
541 img->dirty = true;
542 return true;
543 } else {
544 return false;
545 }
546}
547
548bool img_zoom_in(img_t *img)
549{
550 int i;
551 float z;
552
553 for (i = 1; i < ARRLEN(zoom_levels); i++) {
554 z = zoom_levels[i] / 100.0;
555 if (zoomdiff(z, img->zoom) > 0)
556 return img_zoom(img, z);
557 }
558 return false;
559}
560
561bool img_zoom_out(img_t *img)
562{
563 int i;
564 float z;
565
566 for (i = ARRLEN(zoom_levels) - 2; i >= 0; i--) {
567 z = zoom_levels[i] / 100.0;
568 if (zoomdiff(z, img->zoom) < 0)
569 return img_zoom(img, z);
570 }
571 return false;
572}
573
574bool img_pos(img_t *img, float x, float y)
575{
576 float ox, oy;
577
578 ox = img->x;
579 oy = img->y;
580
581 img->x = x;
582 img->y = y;
583
584 img_check_pan(img, true);
585
586 if (ox != img->x || oy != img->y) {
587 img->dirty = true;
588 return true;
589 } else {
590 return false;
591 }
592}
593
594bool img_move(img_t *img, float dx, float dy)
595{
596 return img_pos(img, img->x + dx, img->y + dy);
597}
598
599bool img_pan(img_t *img, direction_t dir, int d)
600{
601 /* d < 0: screen-wise
602 * d = 0: 1/PAN_FRACTION of screen
603 * d > 0: num of pixels
604 */
605 float x, y;
606
607 if (d > 0) {
608 x = y = MAX(1, (float) d * img->zoom);
609 } else {
610 x = img->win->w / (d < 0 ? 1 : PAN_FRACTION);
611 y = img->win->h / (d < 0 ? 1 : PAN_FRACTION);
612 }
613
614 switch (dir) {
615 case DIR_LEFT:
616 return img_move(img, x, 0.0);
617 case DIR_RIGHT:
618 return img_move(img, -x, 0.0);
619 case DIR_UP:
620 return img_move(img, 0.0, y);
621 case DIR_DOWN:
622 return img_move(img, 0.0, -y);
623 }
624 return false;
625}
626
627bool img_pan_edge(img_t *img, direction_t dir)
628{
629 int ox, oy;
630
631 ox = img->x;
632 oy = img->y;
633
634 if (dir & DIR_LEFT)
635 img->x = 0;
636 if (dir & DIR_RIGHT)
637 img->x = img->win->w - img->w * img->zoom;
638 if (dir & DIR_UP)
639 img->y = 0;
640 if (dir & DIR_DOWN)
641 img->y = img->win->h - img->h * img->zoom;
642
643 img_check_pan(img, true);
644
645 if (ox != img->x || oy != img->y) {
646 img->dirty = true;
647 return true;
648 } else {
649 return false;
650 }
651}
652
653void img_rotate(img_t *img, degree_t d)
654{
655 int i, ox, oy, tmp;
656
657 imlib_context_set_image(img->im);
658 imlib_image_orientate(d);
659
660 for (i = 0; i < img->multi.cnt; i++) {
661 if (i != img->multi.sel) {
662 imlib_context_set_image(img->multi.frames[i].im);
663 imlib_image_orientate(d);
664 }
665 }
666 if (d == DEGREE_90 || d == DEGREE_270) {
667 ox = d == DEGREE_90 ? img->x : img->win->w - img->x - img->w * img->zoom;
668 oy = d == DEGREE_270 ? img->y : img->win->h - img->y - img->h * img->zoom;
669
670 img->x = oy + (img->win->w - img->win->h) / 2;
671 img->y = ox + (img->win->h - img->win->w) / 2;
672
673 tmp = img->w;
674 img->w = img->h;
675 img->h = tmp;
676 img->checkpan = true;
677 }
678 img->dirty = true;
679}
680
681void img_flip(img_t *img, flipdir_t d)
682{
683 int i;
684 void (*imlib_flip_op[3])(void) = {
685 imlib_image_flip_horizontal,
686 imlib_image_flip_vertical,
687 imlib_image_flip_diagonal
688 };
689
690 d = (d & (FLIP_HORIZONTAL | FLIP_VERTICAL)) - 1;
691
692 if (d < 0 || d >= ARRLEN(imlib_flip_op))
693 return;
694
695 imlib_context_set_image(img->im);
696 imlib_flip_op[d]();
697
698 for (i = 0; i < img->multi.cnt; i++) {
699 if (i != img->multi.sel) {
700 imlib_context_set_image(img->multi.frames[i].im);
701 imlib_flip_op[d]();
702 }
703 }
704 img->dirty = true;
705}
706
707void img_toggle_antialias(img_t *img)
708{
709 img->aa = !img->aa;
710 imlib_context_set_image(img->im);
711 imlib_context_set_anti_alias(img->aa);
712 img->dirty = true;
713}
714
715bool img_change_gamma(img_t *img, int d)
716{
717 /* d < 0: decrease gamma
718 * d = 0: reset gamma
719 * d > 0: increase gamma
720 */
721 int gamma;
722 double range;
723
724 if (d == 0)
725 gamma = 0;
726 else
727 gamma = MIN(MAX(img->gamma + d, -GAMMA_RANGE), GAMMA_RANGE);
728
729 if (img->gamma != gamma) {
730 imlib_reset_color_modifier();
731 if (gamma != 0) {
732 range = gamma <= 0 ? 1.0 : GAMMA_MAX - 1.0;
733 imlib_modify_color_modifier_gamma(1.0 + gamma * (range / GAMMA_RANGE));
734 }
735 img->gamma = gamma;
736 img->dirty = true;
737 return true;
738 } else {
739 return false;
740 }
741}
742
743bool img_frame_goto(img_t *img, int n)
744{
745 if (n < 0 || n >= img->multi.cnt || n == img->multi.sel)
746 return false;
747
748 img->multi.sel = n;
749 img->im = img->multi.frames[n].im;
750
751 imlib_context_set_image(img->im);
752 img->w = imlib_image_get_width();
753 img->h = imlib_image_get_height();
754 img->checkpan = true;
755 img->dirty = true;
756
757 return true;
758}
759
760bool img_frame_navigate(img_t *img, int d)
761{
762 if (img->multi.cnt == 0 || d == 0)
763 return false;
764
765 d += img->multi.sel;
766 if (d < 0)
767 d = 0;
768 else if (d >= img->multi.cnt)
769 d = img->multi.cnt - 1;
770
771 return img_frame_goto(img, d);
772}
773
774bool img_frame_animate(img_t *img)
775{
776 if (img->multi.cnt == 0)
777 return false;
778
779 if (img->multi.sel + 1 >= img->multi.cnt)
780 img_frame_goto(img, 0);
781 else
782 img_frame_goto(img, img->multi.sel + 1);
783 img->dirty = true;
784 return true;
785}
786