Krolyxon
Commit 86dc686

Branches

Refs →

nsxivimage.c

772 lines · 16.3 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/types.h>
23#include <unistd.h>
24
25#include "image.h"
26#include "options.h"
27#include "util.h"
28
29#define _IMAGE_CONFIG
30#include "config.h"
31
32#if HAVE_LIBEXIF
33#include <libexif/exif-data.h>
34#endif
35
36#if HAVE_GIFLIB
37#include <gif_lib.h>
38enum { DEF_GIF_DELAY = 75 };
39#endif
40
41float zoom_min;
42float zoom_max;
43
44static int zoomdiff(float z1, float z2)
45{
46 return (int) (z1 * 1000.0 - z2 * 1000.0);
47}
48
49void img_init(img_t *img, win_t *win)
50{
51 zoom_min = zoom_levels[0] / 100.0;
52 zoom_max = zoom_levels[ARRLEN(zoom_levels) - 1] / 100.0;
53
54 imlib_context_set_display(win->env.dpy);
55 imlib_context_set_visual(win->env.vis);
56 imlib_context_set_colormap(win->env.cmap);
57
58 img->im = NULL;
59 img->win = win;
60 img->scalemode = options->scalemode;
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 = ANTI_ALIAS;
67 img->alpha = ALPHA_LAYER;
68 img->multi.cap = img->multi.cnt = 0;
69 img->multi.animate = options->animate;
70 img->multi.length = 0;
71
72 img->cmod = imlib_create_color_modifier();
73 imlib_context_set_color_modifier(img->cmod);
74 img->gamma = MIN(MAX(options->gamma, -GAMMA_RANGE), GAMMA_RANGE);
75
76 img->ss.on = options->slideshow > 0;
77 img->ss.delay = options->slideshow > 0 ? options->slideshow : SLIDESHOW_DELAY;
78}
79
80#if HAVE_LIBEXIF
81void exif_auto_orientate(const fileinfo_t *file)
82{
83 ExifData *ed;
84 ExifEntry *entry;
85 int byte_order, orientation = 0;
86
87 if ((ed = exif_data_new_from_file(file->path)) == NULL)
88 return;
89 byte_order = exif_data_get_byte_order(ed);
90 entry = exif_content_get_entry(ed->ifd[EXIF_IFD_0], EXIF_TAG_ORIENTATION);
91 if (entry != NULL)
92 orientation = exif_get_short(entry->data, byte_order);
93 exif_data_unref(ed);
94
95 switch (orientation) {
96 case 5:
97 imlib_image_orientate(1);
98 case 2:
99 imlib_image_flip_vertical();
100 break;
101 case 3:
102 imlib_image_orientate(2);
103 break;
104 case 7:
105 imlib_image_orientate(1);
106 case 4:
107 imlib_image_flip_horizontal();
108 break;
109 case 6:
110 imlib_image_orientate(1);
111 break;
112 case 8:
113 imlib_image_orientate(3);
114 break;
115 }
116}
117#endif
118
119#if HAVE_GIFLIB
120bool img_load_gif(img_t *img, const fileinfo_t *file)
121{
122 GifFileType *gif;
123 GifRowType *rows = NULL;
124 GifRecordType rec;
125 ColorMapObject *cmap;
126 DATA32 bgpixel, *data, *ptr;
127 DATA32 *prev_frame = NULL;
128 Imlib_Image im;
129 int i, j, bg, r, g, b;
130 int x, y, w, h, sw, sh;
131 int px, py, pw, ph;
132 int intoffset[] = { 0, 4, 2, 1 };
133 int intjump[] = { 8, 8, 4, 2 };
134 int transp = -1;
135 unsigned int disposal = 0, prev_disposal = 0;
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 emalloc(sizeof(img_frame_t) * img->multi.cap);
143 }
144 img->multi.cnt = img->multi.sel = 0;
145 img->multi.length = 0;
146
147#if defined(GIFLIB_MAJOR) && GIFLIB_MAJOR >= 5
148 gif = DGifOpenFileName(file->path, NULL);
149#else
150 gif = DGifOpenFileName(file->path);
151#endif
152 if (gif == NULL) {
153 error(0, 0, "%s: Error opening gif image", file->name);
154 return false;
155 }
156 bg = gif->SBackGroundColor;
157 sw = gif->SWidth;
158 sh = gif->SHeight;
159 px = py = pw = ph = 0;
160
161 do {
162 if (DGifGetRecordType(gif, &rec) == GIF_ERROR) {
163 err = true;
164 break;
165 }
166 if (rec == EXTENSION_RECORD_TYPE) {
167 int ext_code;
168 GifByteType *ext = NULL;
169
170 DGifGetExtension(gif, &ext_code, &ext);
171 while (ext) {
172 if (ext_code == GRAPHICS_EXT_FUNC_CODE) {
173 if (ext[1] & 1)
174 transp = (int) ext[4];
175 else
176 transp = -1;
177
178 delay = 10 * ((unsigned int) ext[3] << 8 | (unsigned int) ext[2]);
179 disposal = (unsigned int) ext[1] >> 2 & 0x7;
180 }
181 ext = NULL;
182 DGifGetExtensionNext(gif, &ext);
183 }
184 } else if (rec == IMAGE_DESC_RECORD_TYPE) {
185 if (DGifGetImageDesc(gif) == GIF_ERROR) {
186 err = true;
187 break;
188 }
189 x = gif->Image.Left;
190 y = gif->Image.Top;
191 w = gif->Image.Width;
192 h = gif->Image.Height;
193
194 rows = (GifRowType*) emalloc(h * sizeof(GifRowType));
195 for (i = 0; i < h; i++)
196 rows[i] = (GifRowType) emalloc(w * sizeof(GifPixelType));
197 if (gif->Image.Interlace) {
198 for (i = 0; i < 4; i++) {
199 for (j = intoffset[i]; j < h; j += intjump[i])
200 DGifGetLine(gif, rows[j], w);
201 }
202 } else {
203 for (i = 0; i < h; i++)
204 DGifGetLine(gif, rows[i], w);
205 }
206
207 ptr = data = (DATA32*) emalloc(sizeof(DATA32) * sw * sh);
208 cmap = gif->Image.ColorMap ? gif->Image.ColorMap : gif->SColorMap;
209 r = cmap->Colors[bg].Red;
210 g = cmap->Colors[bg].Green;
211 b = cmap->Colors[bg].Blue;
212 bgpixel = 0x00ffffff & (r << 16 | g << 8 | b);
213
214 for (i = 0; i < sh; i++) {
215 for (j = 0; j < sw; j++) {
216 if (i < y || i >= y + h || j < x || j >= x + w ||
217 rows[i-y][j-x] == transp)
218 {
219 if (prev_frame != NULL && (prev_disposal != 2 ||
220 i < py || i >= py + ph || j < px || j >= px + pw))
221 {
222 *ptr = prev_frame[i * sw + j];
223 } else {
224 *ptr = bgpixel;
225 }
226 } else {
227 r = cmap->Colors[rows[i-y][j-x]].Red;
228 g = cmap->Colors[rows[i-y][j-x]].Green;
229 b = cmap->Colors[rows[i-y][j-x]].Blue;
230 *ptr = 0xffu << 24 | r << 16 | g << 8 | b;
231 }
232 ptr++;
233 }
234 }
235
236 im = imlib_create_image_using_copied_data(sw, sh, data);
237
238 for (i = 0; i < h; i++)
239 free(rows[i]);
240 free(rows);
241 free(data);
242
243 if (im == NULL) {
244 err = true;
245 break;
246 }
247
248 imlib_context_set_image(im);
249 imlib_image_set_format("gif");
250 if (transp >= 0)
251 imlib_image_set_has_alpha(1);
252
253 if (disposal != 3)
254 prev_frame = imlib_image_get_data_for_reading_only();
255 prev_disposal = disposal;
256 px = x, py = y, pw = w, ph = h;
257
258 if (img->multi.cnt == img->multi.cap) {
259 img->multi.cap *= 2;
260 img->multi.frames = (img_frame_t*)
261 erealloc(img->multi.frames,
262 img->multi.cap * sizeof(img_frame_t));
263 }
264 img->multi.frames[img->multi.cnt].im = im;
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
300 if (access(file->path, R_OK) < 0 ||
301 (img->im = imlib_load_image(file->path)) == NULL)
302 {
303 if (file->flags & FF_WARN)
304 error(0, 0, "%s: Error opening image", file->name);
305 return false;
306 }
307
308 imlib_context_set_image(img->im);
309 imlib_image_set_changes_on_disk();
310
311#if HAVE_LIBEXIF
312 exif_auto_orientate(file);
313#endif
314
315 if ((fmt = imlib_image_format()) != NULL) {
316#if HAVE_GIFLIB
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->checkpan = true;
324 img->dirty = true;
325
326 return true;
327}
328
329CLEANUP void img_close(img_t *img, bool decache)
330{
331 int i;
332
333 if (img->multi.cnt > 0) {
334 for (i = 0; i < img->multi.cnt; i++) {
335 imlib_context_set_image(img->multi.frames[i].im);
336 imlib_free_image();
337 }
338 img->multi.cnt = 0;
339 img->im = NULL;
340 } else if (img->im != NULL) {
341 imlib_context_set_image(img->im);
342 if (decache)
343 imlib_free_image_and_decache();
344 else
345 imlib_free_image();
346 img->im = NULL;
347 }
348}
349
350void img_check_pan(img_t *img, bool moved)
351{
352 win_t *win;
353 int ox, oy;
354 float w, h;
355
356 win = img->win;
357 w = img->w * img->zoom;
358 h = img->h * img->zoom;
359 ox = img->x;
360 oy = img->y;
361
362 if (w < win->w)
363 img->x = (win->w - w) / 2;
364 else if (img->x > 0)
365 img->x = 0;
366 else if (img->x + w < win->w)
367 img->x = win->w - w;
368 if (h < win->h)
369 img->y = (win->h - h) / 2;
370 else if (img->y > 0)
371 img->y = 0;
372 else if (img->y + h < win->h)
373 img->y = win->h - h;
374
375 if (!moved && (ox != img->x || oy != img->y))
376 img->dirty = true;
377}
378
379bool img_fit(img_t *img)
380{
381 float z, zmax, zw, zh;
382
383 if (img->scalemode == SCALE_ZOOM)
384 return false;
385
386 zmax = img->scalemode == SCALE_DOWN ? 1.0 : zoom_max;
387 zw = (float) img->win->w / (float) img->w;
388 zh = (float) img->win->h / (float) img->h;
389
390 switch (img->scalemode) {
391 case SCALE_WIDTH:
392 z = zw;
393 break;
394 case SCALE_HEIGHT:
395 z = zh;
396 break;
397 default:
398 z = MIN(zw, zh);
399 break;
400 }
401
402 z = MAX(z, zoom_min);
403 z = MIN(z, zmax);
404
405 if (zoomdiff(z, img->zoom) != 0) {
406 img->zoom = z;
407 img->dirty = true;
408 return true;
409 } else {
410 return false;
411 }
412}
413
414void img_render(img_t *img)
415{
416 win_t *win;
417 int sx, sy, sw, sh;
418 int dx, dy, dw, dh;
419 Imlib_Image bg;
420 unsigned long c;
421
422 win = img->win;
423 img_fit(img);
424
425 if (img->checkpan) {
426 img_check_pan(img, false);
427 img->checkpan = false;
428 }
429
430 if (!img->dirty)
431 return;
432
433 /* calculate source and destination offsets:
434 * - part of image drawn on full window, or
435 * - full image drawn on part of window
436 */
437 if (img->x <= 0) {
438 sx = -img->x / img->zoom + 0.5;
439 sw = win->w / img->zoom;
440 dx = 0;
441 dw = win->w;
442 } else {
443 sx = 0;
444 sw = img->w;
445 dx = img->x;
446 dw = img->w * img->zoom;
447 }
448 if (img->y <= 0) {
449 sy = -img->y / img->zoom + 0.5;
450 sh = win->h / img->zoom;
451 dy = 0;
452 dh = win->h;
453 } else {
454 sy = 0;
455 sh = img->h;
456 dy = img->y;
457 dh = img->h * img->zoom;
458 }
459
460 win_clear(win);
461
462 imlib_context_set_image(img->im);
463 imlib_context_set_anti_alias(img->aa);
464 imlib_context_set_drawable(win->buf.pm);
465
466 if (imlib_image_has_alpha()) {
467 if ((bg = imlib_create_image(dw, dh)) == NULL)
468 error(EXIT_FAILURE, ENOMEM, NULL);
469 imlib_context_set_image(bg);
470 imlib_image_set_has_alpha(0);
471
472 if (img->alpha) {
473 int i, c, r;
474 DATA32 col[2] = { 0xFF666666, 0xFF999999 };
475 DATA32 * data = imlib_image_get_data();
476
477 for (r = 0; r < dh; r++) {
478 i = r * dw;
479 if (r == 0 || r == 8) {
480 for (c = 0; c < dw; c++)
481 data[i++] = col[!(c & 8) ^ !r];
482 } else {
483 memcpy(&data[i], &data[(r & 8) * dw], dw * sizeof(data[0]));
484 }
485 }
486 imlib_image_put_back_data(data);
487 } else {
488 c = win->fullscreen ? win->fscol.pixel : win->bgcol.pixel;
489 imlib_context_set_color(c >> 16 & 0xFF, c >> 8 & 0xFF, c & 0xFF, 0xFF);
490 imlib_image_fill_rectangle(0, 0, dw, dh);
491 }
492 imlib_blend_image_onto_image(img->im, 0, sx, sy, sw, sh, 0, 0, dw, dh);
493 imlib_context_set_color_modifier(NULL);
494 imlib_render_image_on_drawable(dx, dy);
495 imlib_free_image();
496 imlib_context_set_color_modifier(img->cmod);
497 } else {
498 imlib_render_image_part_on_drawable_at_size(sx, sy, sw, sh, dx, dy, dw, dh);
499 }
500 img->dirty = false;
501}
502
503bool img_fit_win(img_t *img, scalemode_t sm)
504{
505 float oz;
506
507 oz = img->zoom;
508 img->scalemode = sm;
509
510 if (img_fit(img)) {
511 img->x = img->win->w / 2 - (img->win->w / 2 - img->x) * img->zoom / oz;
512 img->y = img->win->h / 2 - (img->win->h / 2 - img->y) * img->zoom / oz;
513 img->checkpan = true;
514 return true;
515 } else {
516 return false;
517 }
518}
519
520bool img_zoom(img_t *img, float z)
521{
522 z = MAX(z, zoom_min);
523 z = MIN(z, zoom_max);
524
525 img->scalemode = SCALE_ZOOM;
526
527 if (zoomdiff(z, img->zoom) != 0) {
528 img->x = img->win->w / 2 - (img->win->w / 2 - img->x) * z / img->zoom;
529 img->y = img->win->h / 2 - (img->win->h / 2 - img->y) * z / img->zoom;
530 img->zoom = z;
531 img->checkpan = true;
532 img->dirty = true;
533 return true;
534 } else {
535 return false;
536 }
537}
538
539bool img_zoom_in(img_t *img)
540{
541 int i;
542 float z;
543
544 for (i = 1; i < ARRLEN(zoom_levels); i++) {
545 z = zoom_levels[i] / 100.0;
546 if (zoomdiff(z, img->zoom) > 0)
547 return img_zoom(img, z);
548 }
549 return false;
550}
551
552bool img_zoom_out(img_t *img)
553{
554 int i;
555 float z;
556
557 for (i = ARRLEN(zoom_levels) - 2; i >= 0; i--) {
558 z = zoom_levels[i] / 100.0;
559 if (zoomdiff(z, img->zoom) < 0)
560 return img_zoom(img, z);
561 }
562 return false;
563}
564
565bool img_move(img_t *img, float dx, float dy)
566{
567 float ox, oy;
568
569 ox = img->x;
570 oy = img->y;
571
572 img->x += dx;
573 img->y += dy;
574
575 img_check_pan(img, true);
576
577 if (ox != img->x || oy != img->y) {
578 img->dirty = true;
579 return true;
580 } else {
581 return false;
582 }
583}
584
585bool img_pan(img_t *img, direction_t dir, int d)
586{
587 /* d < 0: screen-wise
588 * d = 0: 1/5 of screen
589 * d > 0: num of pixels
590 */
591 float x, y;
592
593 if (d > 0) {
594 x = y = MAX(1, (float) d * img->zoom);
595 } else {
596 x = img->win->w / (d < 0 ? 1 : 5);
597 y = img->win->h / (d < 0 ? 1 : 5);
598 }
599
600 switch (dir) {
601 case DIR_LEFT:
602 return img_move(img, x, 0.0);
603 case DIR_RIGHT:
604 return img_move(img, -x, 0.0);
605 case DIR_UP:
606 return img_move(img, 0.0, y);
607 case DIR_DOWN:
608 return img_move(img, 0.0, -y);
609 }
610 return false;
611}
612
613bool img_pan_edge(img_t *img, direction_t dir)
614{
615 int ox, oy;
616
617 ox = img->x;
618 oy = img->y;
619
620 if (dir & DIR_LEFT)
621 img->x = 0;
622 if (dir & DIR_RIGHT)
623 img->x = img->win->w - img->w * img->zoom;
624 if (dir & DIR_UP)
625 img->y = 0;
626 if (dir & DIR_DOWN)
627 img->y = img->win->h - img->h * img->zoom;
628
629 img_check_pan(img, true);
630
631 if (ox != img->x || oy != img->y) {
632 img->dirty = true;
633 return true;
634 } else {
635 return false;
636 }
637}
638
639void img_rotate(img_t *img, degree_t d)
640{
641 int i, ox, oy, tmp;
642
643 imlib_context_set_image(img->im);
644 imlib_image_orientate(d);
645
646 for (i = 0; i < img->multi.cnt; i++) {
647 if (i != img->multi.sel) {
648 imlib_context_set_image(img->multi.frames[i].im);
649 imlib_image_orientate(d);
650 }
651 }
652 if (d == DEGREE_90 || d == DEGREE_270) {
653 ox = d == DEGREE_90 ? img->x : img->win->w - img->x - img->w * img->zoom;
654 oy = d == DEGREE_270 ? img->y : img->win->h - img->y - img->h * img->zoom;
655
656 img->x = oy + (img->win->w - img->win->h) / 2;
657 img->y = ox + (img->win->h - img->win->w) / 2;
658
659 tmp = img->w;
660 img->w = img->h;
661 img->h = tmp;
662 img->checkpan = true;
663 }
664 img->dirty = true;
665}
666
667void img_flip(img_t *img, flipdir_t d)
668{
669 int i;
670 void (*imlib_flip_op[3])(void) = {
671 imlib_image_flip_horizontal,
672 imlib_image_flip_vertical,
673 imlib_image_flip_diagonal
674 };
675
676 d = (d & (FLIP_HORIZONTAL | FLIP_VERTICAL)) - 1;
677
678 if (d < 0 || d >= ARRLEN(imlib_flip_op))
679 return;
680
681 imlib_context_set_image(img->im);
682 imlib_flip_op[d]();
683
684 for (i = 0; i < img->multi.cnt; i++) {
685 if (i != img->multi.sel) {
686 imlib_context_set_image(img->multi.frames[i].im);
687 imlib_flip_op[d]();
688 }
689 }
690 img->dirty = true;
691}
692
693void img_toggle_antialias(img_t *img)
694{
695 img->aa = !img->aa;
696 imlib_context_set_image(img->im);
697 imlib_context_set_anti_alias(img->aa);
698 img->dirty = true;
699}
700
701bool img_change_gamma(img_t *img, int d)
702{
703 /* d < 0: decrease gamma
704 * d = 0: reset gamma
705 * d > 0: increase gamma
706 */
707 int gamma;
708 double range;
709
710 if (d == 0)
711 gamma = 0;
712 else
713 gamma = MIN(MAX(img->gamma + d, -GAMMA_RANGE), GAMMA_RANGE);
714
715 if (img->gamma != gamma) {
716 imlib_reset_color_modifier();
717 if (gamma != 0) {
718 range = gamma <= 0 ? 1.0 : GAMMA_MAX - 1.0;
719 imlib_modify_color_modifier_gamma(1.0 + gamma * (range / GAMMA_RANGE));
720 }
721 img->gamma = gamma;
722 img->dirty = true;
723 return true;
724 } else {
725 return false;
726 }
727}
728
729bool img_frame_goto(img_t *img, int n)
730{
731 if (n < 0 || n >= img->multi.cnt || n == img->multi.sel)
732 return false;
733
734 img->multi.sel = n;
735 img->im = img->multi.frames[n].im;
736
737 imlib_context_set_image(img->im);
738 img->w = imlib_image_get_width();
739 img->h = imlib_image_get_height();
740 img->checkpan = true;
741 img->dirty = true;
742
743 return true;
744}
745
746bool img_frame_navigate(img_t *img, int d)
747{
748 if (img->multi.cnt == 0 || d == 0)
749 return false;
750
751 d += img->multi.sel;
752 if (d < 0)
753 d = 0;
754 else if (d >= img->multi.cnt)
755 d = img->multi.cnt - 1;
756
757 return img_frame_goto(img, d);
758}
759
760bool img_frame_animate(img_t *img)
761{
762 if (img->multi.cnt == 0)
763 return false;
764
765 if (img->multi.sel + 1 >= img->multi.cnt)
766 img_frame_goto(img, 0);
767 else
768 img_frame_goto(img, img->multi.sel + 1);
769 img->dirty = true;
770 return true;
771}
772