Krolyxon
Commit 5621975

Branches

Refs →

nsxivimage.c

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