Krolyxon
Commit b56989f

Branches

Refs →

nsxivimage.c

732 lines · 15.4 KB · C

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