Krolyxon
Commit bfab4dc

Branches

Refs →

nsxivimage.c

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