Krolyxon
Commit 78f48fe

Branches

Refs →

nsxivimage.c

709 lines · 15.0 KB · C

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