Krolyxon
Commit 8fc7cb7

Branches

Refs →

nsxivimage.c

675 lines · 14.5 KB · C

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