Krolyxon
Commit f2e0c49

Branches

Refs →

nsxivimage.c

825 lines · 17.0 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#define _POSIX_C_SOURCE 200112L
20#define _IMAGE_CONFIG
21#define _RENDER_CONFIG
22
23#include <stdlib.h>
24#include <string.h>
25#include <sys/types.h>
26#include <unistd.h>
27
28#if HAVE_GIFLIB
29#include <gif_lib.h>
30enum { MIN_GIF_DELAY = 25 };
31#endif
32
33#include "exif.h"
34#include "image.h"
35#include "options.h"
36#include "util.h"
37#include "config.h"
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
47static void img_apply_gamma(img_t *img)
48{
49 if (img == NULL || img->im == NULL || img->cmod == NULL)
50 return;
51
52 if (img->gamma == 0) {
53 imlib_context_set_color_modifier(NULL);
54 } else {
55 double range = img->gamma <= 0 ? 1.0 : GAMMA_MAX - 1.0;
56
57 imlib_context_set_color_modifier(img->cmod);
58 imlib_reset_color_modifier();
59 imlib_modify_color_modifier_gamma(1.0 + img->gamma * (range / GAMMA_RANGE));
60 }
61}
62
63void img_init(img_t *img, win_t *win)
64{
65 zoom_min = zoom_levels[0] / 100.0;
66 zoom_max = zoom_levels[ARRLEN(zoom_levels) - 1] / 100.0;
67
68 if (img == NULL || win == NULL)
69 return;
70
71 imlib_context_set_display(win->env.dpy);
72 imlib_context_set_visual(win->env.vis);
73 imlib_context_set_colormap(win->env.cmap);
74
75 img->im = NULL;
76 img->win = win;
77 img->zoom = options->zoom;
78 img->zoom = MAX(img->zoom, zoom_min);
79 img->zoom = MIN(img->zoom, zoom_max);
80 img->checkpan = false;
81 img->dirty = false;
82 img->aa = RENDER_ANTI_ALIAS;
83 img->alpha = !RENDER_WHITE_ALPHA;
84 img->multi.cap = img->multi.cnt = 0;
85 img->multi.animate = false;
86 img->multi.repeat = 0;
87
88 img->cmod = imlib_create_color_modifier();
89 img->gamma = MIN(MAX(options->gamma, -GAMMA_RANGE), GAMMA_RANGE);
90}
91
92void exif_auto_orientate(const fileinfo_t *file)
93{
94 switch (exif_orientation(file)) {
95 case 5:
96 imlib_image_orientate(1);
97 case 2:
98 imlib_image_flip_vertical();
99 break;
100
101 case 3:
102 imlib_image_orientate(2);
103 break;
104
105 case 7:
106 imlib_image_orientate(1);
107 case 4:
108 imlib_image_flip_horizontal();
109 break;
110
111 case 6:
112 imlib_image_orientate(1);
113 break;
114
115 case 8:
116 imlib_image_orientate(3);
117 break;
118 }
119}
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 s_malloc(sizeof(img_frame_t) * img->multi.cap);
145 }
146 img->multi.cnt = 0;
147 img->multi.sel = 0;
148 img->multi.repeat = 0;
149
150#if defined(GIFLIB_MAJOR) && GIFLIB_MAJOR >= 5
151 gif = DGifOpenFileName(file->path, NULL);
152#else
153 gif = DGifOpenFileName(file->path);
154#endif
155 if (gif == NULL) {
156 warn("could not open gif file: %s", file->name);
157 return false;
158 }
159 bg = gif->SBackGroundColor;
160 sw = gif->SWidth;
161 sh = gif->SHeight;
162 px = py = pw = ph = 0;
163
164 do {
165 if (DGifGetRecordType(gif, &rec) == GIF_ERROR) {
166 err = true;
167 break;
168 }
169 if (rec == EXTENSION_RECORD_TYPE) {
170 int ext_code;
171 GifByteType *ext = NULL;
172
173 DGifGetExtension(gif, &ext_code, &ext);
174 while (ext) {
175 if (ext_code == GRAPHICS_EXT_FUNC_CODE) {
176 if (ext[1] & 1)
177 transp = (int) ext[4];
178 else
179 transp = -1;
180
181 delay = 10 * ((unsigned int) ext[3] << 8 | (unsigned int) ext[2]);
182 if (delay)
183 delay = MAX(delay, MIN_GIF_DELAY);
184
185 disposal = (unsigned int) ext[1] >> 2 & 0x7;
186 } else if (ext_code == APPLICATION_EXT_FUNC_CODE) {
187 if (ext[0] == 11 && memcmp(ext+1, "NETSCAPE2.0", 11) == 0) {
188 DGifGetExtensionNext(gif, &ext);
189 if (ext && ext[0] == 3 && ext[1] == 1)
190 img->multi.repeat = ((int) ext[3] << 8 | (int) ext[2]) - 1;
191 }
192 }
193 ext = NULL;
194 DGifGetExtensionNext(gif, &ext);
195 }
196 } else if (rec == IMAGE_DESC_RECORD_TYPE) {
197 if (DGifGetImageDesc(gif) == GIF_ERROR) {
198 err = true;
199 break;
200 }
201 x = gif->Image.Left;
202 y = gif->Image.Top;
203 w = gif->Image.Width;
204 h = gif->Image.Height;
205
206 rows = (GifRowType*) s_malloc(h * sizeof(GifRowType));
207 for (i = 0; i < h; i++)
208 rows[i] = (GifRowType) s_malloc(w * sizeof(GifPixelType));
209 if (gif->Image.Interlace) {
210 for (i = 0; i < 4; i++) {
211 for (j = intoffset[i]; j < h; j += intjump[i])
212 DGifGetLine(gif, rows[j], w);
213 }
214 } else {
215 for (i = 0; i < h; i++)
216 DGifGetLine(gif, rows[i], w);
217 }
218
219 ptr = data = (DATA32*) s_malloc(sizeof(DATA32) * sw * sh);
220 cmap = gif->Image.ColorMap ? gif->Image.ColorMap : gif->SColorMap;
221 r = cmap->Colors[bg].Red;
222 g = cmap->Colors[bg].Green;
223 b = cmap->Colors[bg].Blue;
224 bgpixel = 0x00ffffff & (r << 16 | g << 8 | b);
225
226 for (i = 0; i < sh; i++) {
227 for (j = 0; j < sw; j++) {
228 if (i < y || i >= y + h || j < x || j >= x + w ||
229 rows[i-y][j-x] == transp)
230 {
231 if (prev_frame != NULL && (prev_disposal != 2 ||
232 i < py || i >= py + ph || j < px || j >= px + pw))
233 {
234 *ptr = prev_frame[i * sw + j];
235 } else {
236 *ptr = bgpixel;
237 }
238 } else {
239 r = cmap->Colors[rows[i-y][j-x]].Red;
240 g = cmap->Colors[rows[i-y][j-x]].Green;
241 b = cmap->Colors[rows[i-y][j-x]].Blue;
242 *ptr = 0xff << 24 | r << 16 | g << 8 | b;
243 }
244 ptr++;
245 }
246 }
247
248 im = imlib_create_image_using_copied_data(sw, sh, data);
249
250 for (i = 0; i < h; i++)
251 free(rows[i]);
252 free(rows);
253 free(data);
254
255 if (im == NULL) {
256 err = true;
257 break;
258 }
259
260 imlib_context_set_image(im);
261 imlib_image_set_format("gif");
262 if (transp >= 0)
263 imlib_image_set_has_alpha(1);
264
265 if (disposal != 3)
266 prev_frame = imlib_image_get_data_for_reading_only();
267 prev_disposal = disposal;
268 px = x, py = y, pw = w, ph = h;
269
270 if (img->multi.cnt == img->multi.cap) {
271 img->multi.cap *= 2;
272 img->multi.frames = (img_frame_t*)
273 s_realloc(img->multi.frames,
274 img->multi.cap * sizeof(img_frame_t));
275 }
276 img->multi.frames[img->multi.cnt].im = im;
277 img->multi.frames[img->multi.cnt].delay = delay ? delay : GIF_DELAY;
278 img->multi.cnt++;
279 }
280 } while (rec != TERMINATE_RECORD_TYPE);
281
282 DGifCloseFile(gif);
283
284 if (err && !file->loaded)
285 warn("corrupted gif file: %s", file->name);
286
287 if (img->multi.cnt > 1) {
288 imlib_context_set_image(img->im);
289 imlib_free_image();
290 img->im = img->multi.frames[0].im;
291 img->multi.animate = GIF_AUTOPLAY;
292 } else if (img->multi.cnt == 1) {
293 imlib_context_set_image(img->multi.frames[0].im);
294 imlib_free_image();
295 img->multi.cnt = 0;
296 img->multi.animate = false;
297 }
298
299 imlib_context_set_image(img->im);
300
301 return !err;
302}
303#endif /* HAVE_GIFLIB */
304
305bool img_load(img_t *img, const fileinfo_t *file)
306{
307 const char *fmt;
308
309 if (img == NULL || file == NULL || file->name == NULL || file->path == NULL)
310 return false;
311
312 if (access(file->path, R_OK) < 0 ||
313 (img->im = imlib_load_image(file->path)) == NULL)
314 {
315 warn("could not open image: %s", file->name);
316 return false;
317 }
318
319 imlib_context_set_image(img->im);
320 imlib_image_set_changes_on_disk();
321
322 if ((fmt = imlib_image_format()) == NULL) {
323 warn("could not open image: %s", file->name);
324 return false;
325 }
326 if (STREQ(fmt, "jpeg"))
327 exif_auto_orientate(file);
328#if HAVE_GIFLIB
329 if (STREQ(fmt, "gif"))
330 img_load_gif(img, file);
331#endif
332
333 img_apply_gamma(img);
334
335 img->w = imlib_image_get_width();
336 img->h = imlib_image_get_height();
337 img->scalemode = options->scalemode;
338 img->re = false;
339 img->checkpan = false;
340 img->dirty = true;
341
342 return true;
343}
344
345void img_close(img_t *img, bool decache)
346{
347 int i;
348
349 if (img == NULL)
350 return;
351
352 if (img->multi.cnt > 0) {
353 for (i = 0; i < img->multi.cnt; i++) {
354 imlib_context_set_image(img->multi.frames[i].im);
355 imlib_free_image();
356 }
357 img->multi.cnt = 0;
358 img->im = NULL;
359 } else if (img->im != NULL) {
360 imlib_context_set_image(img->im);
361 if (decache)
362 imlib_free_image_and_decache();
363 else
364 imlib_free_image();
365 img->im = NULL;
366 }
367
368 if (img->cmod)
369 imlib_context_set_color_modifier(NULL);
370}
371
372void img_check_pan(img_t *img, bool moved)
373{
374 win_t *win;
375 int ox, oy;
376
377 if (img == NULL || img->im == NULL || img->win == NULL)
378 return;
379
380 win = img->win;
381 ox = img->x;
382 oy = img->y;
383
384 if (img->w * img->zoom > win->w) {
385 if (img->x > 0 && img->x + img->w * img->zoom > win->w)
386 img->x = 0;
387 if (img->x < 0 && img->x + img->w * img->zoom < win->w)
388 img->x = win->w - img->w * img->zoom;
389 } else {
390 img->x = (win->w - img->w * img->zoom) / 2;
391 }
392 if (img->h * img->zoom > win->h) {
393 if (img->y > 0 && img->y + img->h * img->zoom > win->h)
394 img->y = 0;
395 if (img->y < 0 && img->y + img->h * img->zoom < win->h)
396 img->y = win->h - img->h * img->zoom;
397 } else {
398 img->y = (win->h - img->h * img->zoom) / 2;
399 }
400
401 if (!moved && (ox != img->x || oy != img->y))
402 img->dirty = true;
403}
404
405bool img_fit(img_t *img)
406{
407 float z, zmax, zw, zh;
408
409 if (img == NULL || img->im == NULL || img->win == NULL)
410 return false;
411 if (img->scalemode == SCALE_ZOOM)
412 return false;
413
414 zmax = img->scalemode == SCALE_DOWN ? 1.0 : zoom_max;
415 zw = (float) img->win->w / (float) img->w;
416 zh = (float) img->win->h / (float) img->h;
417
418 switch (img->scalemode) {
419 case SCALE_WIDTH:
420 z = zw;
421 break;
422 case SCALE_HEIGHT:
423 z = zh;
424 break;
425 default:
426 z = MIN(zw, zh);
427 break;
428 }
429
430 z = MAX(z, zoom_min);
431 z = MIN(z, zmax);
432
433 if (zoomdiff(z, img->zoom) != 0) {
434 img->zoom = z;
435 img->dirty = true;
436 return true;
437 } else {
438 return false;
439 }
440}
441
442void img_render(img_t *img)
443{
444 win_t *win;
445 int sx, sy, sw, sh;
446 int dx, dy, dw, dh;
447
448 if (img == NULL || img->im == NULL || img->win == NULL)
449 return;
450
451 win = img->win;
452 img_fit(img);
453
454 if (!img->re) {
455 /* rendered for the first time */
456 img->re = true;
457 if (img->zoom * img->w <= win->w)
458 img->x = (win->w - img->w * img->zoom) / 2;
459 else
460 img->x = 0;
461 if (img->zoom * img->h <= win->h)
462 img->y = (win->h - img->h * img->zoom) / 2;
463 else
464 img->y = 0;
465 }
466
467 if (img->checkpan) {
468 img_check_pan(img, false);
469 img->checkpan = false;
470 }
471
472 if (!img->dirty)
473 return;
474
475 /* calculate source and destination offsets */
476 if (img->x < 0) {
477 sx = -img->x / img->zoom;
478 sw = win->w / img->zoom;
479 dx = 0;
480 dw = win->w;
481 } else {
482 sx = 0;
483 sw = img->w;
484 dx = img->x;
485 dw = img->w * img->zoom;
486 }
487 if (img->y < 0) {
488 sy = -img->y / img->zoom;
489 sh = win->h / img->zoom;
490 dy = 0;
491 dh = win->h;
492 } else {
493 sy = 0;
494 sh = img->h;
495 dy = img->y;
496 dh = img->h * img->zoom;
497 }
498
499 win_clear(win);
500
501 imlib_context_set_image(img->im);
502 imlib_context_set_anti_alias(img->aa);
503
504 if (!img->alpha && imlib_image_has_alpha())
505 win_draw_rect(win, win->pm, dx, dy, dw, dh, True, 0, win->white);
506
507 imlib_context_set_drawable(win->pm);
508 imlib_render_image_part_on_drawable_at_size(sx, sy, sw, sh, dx, dy, dw, dh);
509
510 img->dirty = false;
511}
512
513bool img_fit_win(img_t *img, scalemode_t sm)
514{
515 if (img == NULL || img->im == NULL)
516 return false;
517
518 img->scalemode = sm;
519 return img_fit(img);
520}
521
522bool img_center(img_t *img)
523{
524 int ox, oy;
525
526 if (img == NULL || img->im == NULL || img->win == NULL)
527 return false;
528
529 ox = img->x;
530 oy = img->y;
531
532 img->x = (img->win->w - img->w * img->zoom) / 2;
533 img->y = (img->win->h - img->h * img->zoom) / 2;
534
535 if (ox != img->x || oy != img->y) {
536 img->dirty = true;
537 return true;
538 } else {
539 return false;
540 }
541}
542
543bool img_zoom(img_t *img, float z)
544{
545 if (img == NULL || img->im == NULL || img->win == NULL)
546 return false;
547
548 z = MAX(z, zoom_min);
549 z = MIN(z, zoom_max);
550
551 img->scalemode = SCALE_ZOOM;
552
553 if (zoomdiff(z, img->zoom) != 0) {
554 img->x = img->win->w / 2 - (img->win->w / 2 - img->x) * z / img->zoom;
555 img->y = img->win->h / 2 - (img->win->h / 2 - img->y) * z / img->zoom;
556 img->zoom = z;
557 img->checkpan = true;
558 img->dirty = true;
559 return true;
560 } else {
561 return false;
562 }
563}
564
565bool img_zoom_in(img_t *img)
566{
567 int i;
568 float z;
569
570 if (img == NULL || img->im == NULL)
571 return false;
572
573 for (i = 1; i < ARRLEN(zoom_levels); i++) {
574 z = zoom_levels[i] / 100.0;
575 if (zoomdiff(z, img->zoom) > 0)
576 return img_zoom(img, z);
577 }
578 return false;
579}
580
581bool img_zoom_out(img_t *img)
582{
583 int i;
584 float z;
585
586 if (img == NULL || img->im == NULL)
587 return false;
588
589 for (i = ARRLEN(zoom_levels) - 2; i >= 0; i--) {
590 z = zoom_levels[i] / 100.0;
591 if (zoomdiff(z, img->zoom) < 0)
592 return img_zoom(img, z);
593 }
594 return false;
595}
596
597bool img_move(img_t *img, float dx, float dy)
598{
599 float ox, oy;
600
601 if (img == NULL || img->im == NULL)
602 return false;
603
604 ox = img->x;
605 oy = img->y;
606
607 img->x += dx;
608 img->y += dy;
609
610 img_check_pan(img, true);
611
612 if (ox != img->x || oy != img->y) {
613 img->dirty = true;
614 return true;
615 } else {
616 return false;
617 }
618}
619
620bool img_pan(img_t *img, direction_t dir, int d)
621{
622 /* d < 0: screen-wise
623 * d = 0: 1/5 of screen
624 * d > 0: num of pixels
625 */
626 float x, y;
627
628 if (img == NULL || img->im == NULL || img->win == NULL)
629 return false;
630
631 if (d > 0) {
632 x = y = MAX(1, (float) d * img->zoom);
633 } else {
634 x = img->win->w / (d < 0 ? 1 : 5);
635 y = img->win->h / (d < 0 ? 1 : 5);
636 }
637
638 switch (dir) {
639 case DIR_LEFT:
640 return img_move(img, x, 0.0);
641 case DIR_RIGHT:
642 return img_move(img, -x, 0.0);
643 case DIR_UP:
644 return img_move(img, 0.0, y);
645 case DIR_DOWN:
646 return img_move(img, 0.0, -y);
647 }
648 return false;
649}
650
651bool img_pan_edge(img_t *img, direction_t dir)
652{
653 int ox, oy;
654
655 if (img == NULL || img->im == NULL || img->win == NULL)
656 return false;
657
658 ox = img->x;
659 oy = img->y;
660
661 switch (dir) {
662 case DIR_LEFT:
663 img->x = 0;
664 break;
665 case DIR_RIGHT:
666 img->x = img->win->w - img->w * img->zoom;
667 break;
668 case DIR_UP:
669 img->y = 0;
670 break;
671 case DIR_DOWN:
672 img->y = img->win->h - img->h * img->zoom;
673 break;
674 }
675
676 img_check_pan(img, true);
677
678 if (ox != img->x || oy != img->y) {
679 img->dirty = true;
680 return true;
681 } else {
682 return false;
683 }
684}
685
686void img_rotate(img_t *img, degree_t d)
687{
688 int ox, oy, tmp;
689
690 if (img == NULL || img->im == NULL || img->win == NULL)
691 return;
692
693 imlib_context_set_image(img->im);
694 imlib_image_orientate(d);
695
696 if (d == DEGREE_90 || d == DEGREE_270) {
697 ox = d == DEGREE_90 ? img->x : img->win->w - img->x - img->w * img->zoom;
698 oy = d == DEGREE_270 ? img->y : img->win->h - img->y - img->h * img->zoom;
699
700 img->x = oy + (img->win->w - img->win->h) / 2;
701 img->y = ox + (img->win->h - img->win->w) / 2;
702
703 tmp = img->w;
704 img->w = img->h;
705 img->h = tmp;
706 img->checkpan = true;
707 }
708
709 img->dirty = true;
710}
711
712void img_flip(img_t *img, flipdir_t d)
713{
714 if (img == NULL || img->im == NULL)
715 return;
716
717 imlib_context_set_image(img->im);
718
719 switch (d) {
720 case FLIP_HORIZONTAL:
721 imlib_image_flip_horizontal();
722 break;
723 case FLIP_VERTICAL:
724 imlib_image_flip_vertical();
725 break;
726 }
727 img->dirty = true;
728}
729
730void img_toggle_antialias(img_t *img)
731{
732 if (img == NULL || img->im == NULL)
733 return;
734
735 img->aa = !img->aa;
736 imlib_context_set_image(img->im);
737 imlib_context_set_anti_alias(img->aa);
738 img->dirty = true;
739}
740
741bool img_change_gamma(img_t *img, int d)
742{
743 /* d < 0: decrease gamma
744 * d = 0: reset gamma
745 * d > 0: increase gamma
746 */
747 int gamma;
748
749 if (img == NULL || img->im == NULL)
750 return false;
751
752 if (d == 0)
753 gamma = 0;
754 else if (d < 0)
755 gamma = MAX(-GAMMA_RANGE, img->gamma - 1);
756 else
757 gamma = MIN(+GAMMA_RANGE, img->gamma + 1);
758
759 if (img->gamma != gamma) {
760 img->gamma = gamma;
761 img_apply_gamma(img);
762 img->dirty = true;
763 return true;
764 } else {
765 return false;
766 }
767}
768
769bool img_frame_goto(img_t *img, int n)
770{
771 if (img == NULL || img->im == NULL)
772 return false;
773 if (n < 0 || n >= img->multi.cnt || n == img->multi.sel)
774 return false;
775
776 img->multi.sel = n;
777 img->im = img->multi.frames[n].im;
778
779 imlib_context_set_image(img->im);
780 img->w = imlib_image_get_width();
781 img->h = imlib_image_get_height();
782 img->checkpan = true;
783 img->dirty = true;
784
785 return true;
786}
787
788bool img_frame_navigate(img_t *img, int d)
789{
790 if (img == NULL|| img->im == NULL || img->multi.cnt == 0 || d == 0)
791 return false;
792
793 d += img->multi.sel;
794 if (d < 0)
795 d = 0;
796 else if (d >= img->multi.cnt)
797 d = img->multi.cnt - 1;
798
799 return img_frame_goto(img, d);
800}
801
802bool img_frame_animate(img_t *img, bool restart)
803{
804 if (img == NULL || img->im == NULL || img->multi.cnt == 0)
805 return false;
806
807 if (img->multi.sel + 1 >= img->multi.cnt) {
808 if (restart || GIF_LOOP == 1) {
809 img_frame_goto(img, 0);
810 } else if (GIF_LOOP == -1 && img->multi.repeat != 0) {
811 if (img->multi.repeat > 0)
812 img->multi.repeat--;
813 img_frame_goto(img, 0);
814 } else {
815 img->multi.animate = false;
816 return false;
817 }
818 } else if (!restart) {
819 img_frame_goto(img, img->multi.sel + 1);
820 }
821 img->multi.animate = true;
822 img->dirty = true;
823
824 return true;
825}