Krolyxon
Commit fdbc5cf

Branches

Refs →

nsxivimage.c

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