Krolyxon
Commit 0155223

Branches

Refs →

nsxivimage.c

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