Krolyxon
Commit 6e575b0

Branches

Refs →

nsxivimage.c

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