sxiv/image.c

/* sxiv: image.c
 * Copyright (c) 2011 Bert Muennich <muennich at informatik.hu-berlin.de>
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License as published by
 * the Free Software Foundation; either version 2 of the License, or
 * (at your option) any later version.
 *  
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU General Public License for more details.
 *  
 * You should have received a copy of the GNU General Public License
 * along with this program; if not, write to the Free Software
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA.
 */

#include <string.h>
#include <unistd.h>

#include "config.h"
#include "icon.h"
#include "image.h"
#include "options.h"
#include "util.h"

int zl_cnt;
float zoom_min;
float zoom_max;

const short ori_left[8] = { 8, 7, 6, 5, 2, 1, 4, 3 };
const short ori_right[8] = { 6, 5, 8, 7, 4, 3, 2, 1 };

Imlib_Image *im_broken;

void img_init(img_t *img, win_t *win) {
	zl_cnt = sizeof(zoom_levels) / sizeof(zoom_levels[0]);
	zoom_min = zoom_levels[0] / 100.0;
	zoom_max = zoom_levels[zl_cnt - 1] / 100.0;

	im_broken = imlib_create_image_using_data(32, 32, icon_broken);

	if (img) {
		img->im = NULL;
		img->zoom = options->zoom;
		img->zoom = MAX(img->zoom, zoom_min);
		img->zoom = MIN(img->zoom, zoom_max);
		img->aa = options->aa;
	}

	if (win) {
		imlib_context_set_display(win->env.dpy);
		imlib_context_set_visual(win->env.vis);
		imlib_context_set_colormap(win->env.cmap);
	}
}

void img_free(img_t* img) {
	imlib_context_set_image(im_broken);
	imlib_free_image();
}

int img_check(const char *filename) {
	Imlib_Image *im;

	if (!filename)
		return 0;

	if (!access(filename, F_OK) && (im = imlib_load_image(filename))) {
		imlib_context_set_image(im);
		imlib_image_set_changes_on_disk();
		imlib_free_image();
		return 1;
	} else {
		warn("could not open file: %s", filename);
		return 0;
	}
}

int img_load(img_t *img, const char *filename) {
	if (!img || !filename)
		return 0;

	if (!access(filename, F_OK) && (img->im = imlib_load_image(filename))) {
		imlib_context_set_image(img->im);
		imlib_image_set_changes_on_disk();
		imlib_context_set_anti_alias(img->aa);
		img->scalemode = options->scalemode;
	} else {
		warn("could not open file: %s", filename);
		imlib_context_set_image(im_broken);
		imlib_context_set_anti_alias(0);
		img->scalemode = SCALE_DOWN;
	}

	img->ori = img->o_ori = 1;
	img->re = img->checkpan = 0;

	img->w = imlib_image_get_width();
	img->h = imlib_image_get_height();

	return 1;
}

int img_save(img_t *img) {
	const char *fmt;

	if (!img || !img->im)
		return 0;

	imlib_context_set_image(img->im);

	if (img->ori != img->o_ori) {
		fmt = imlib_image_format();
		if (strcmp(fmt, "png") == 0) {
			imlib_save_image(imlib_image_get_filename());
			return 1;
		}
	}

	return 0;
}

void img_close(img_t *img) {
	if (img && img->im) {
		imlib_context_set_image(img->im);
		imlib_free_image();
		img->im = NULL;
	}
}

void img_check_pan(img_t *img, win_t *win) {
	if (!img || !win)
		return;

	if (img->w * img->zoom > win->w) {
		if (img->x > 0 && img->x + img->w * img->zoom > win->w)
			img->x = 0;
		if (img->x < 0 && img->x + img->w * img->zoom < win->w)
			img->x = win->w - img->w * img->zoom;
	} else {
		img->x = (win->w - img->w * img->zoom) / 2;
	}
	if (img->h * img->zoom > win->h) {
		if (img->y > 0 && img->y + img->h * img->zoom > win->h)
			img->y = 0;
		if (img->y < 0 && img->y + img->h * img->zoom < win->h)
			img->y = win->h - img->h * img->zoom;
	} else {
		img->y = (win->h - img->h * img->zoom) / 2;
	}
}

int img_fit(img_t *img, win_t *win) {
	float oz, zw, zh;

	if (!img || !win)
		return 0;

	oz = img->zoom;
	zw = (float) win->w / (float) img->w;
	zh = (float) win->h / (float) img->h;

	img->zoom = MIN(zw, zh);
	img->zoom = MAX(img->zoom, zoom_min);
	img->zoom = MIN(img->zoom, zoom_max);

	return oz != img->zoom;
}

void img_render(img_t *img, win_t *win) {
	int sx, sy, sw, sh;
	int dx, dy, dw, dh;

	if (!img || !win)
		return;

	if (img->scalemode != SCALE_ZOOM) {
		img_fit(img, win);
		if (img->scalemode == SCALE_DOWN && img->zoom > 1.0)
			img->zoom = 1.0;
	}

	if (!img->re) {
		/* rendered for the first time */
		img->re = 1;
		if (img->zoom * img->w <= win->w)
			img->x = (win->w - img->w * img->zoom) / 2;
		else
			img->x = 0;
		if (img->zoom * img->h <= win->h)
			img->y = (win->h - img->h * img->zoom) / 2;
		else
			img->y = 0;
	}
	
	if (img->checkpan) {
		img_check_pan(img, win);
		img->checkpan = 0;
	}

	/* calculate source and destination offsets */
	if (img->x < 0) {
		sx = -img->x / img->zoom;
		sw = win->w / img->zoom;
		dx = 0;
		dw = win->w;
	} else {
		sx = 0;
		sw = img->w;
		dx = img->x;
		dw = img->w * img->zoom;
	}
	if (img->y < 0) {
		sy = -img->y / img->zoom;
		sh = win->h / img->zoom;
		dy = 0;
		dh = win->h;
	} else {
		sy = 0;
		sh = img->h;
		dy = img->y;
		dh = img->h * img->zoom;
	}

	win_clear(win);

	if (img->im)
		imlib_context_set_image(img->im);
	else
		imlib_context_set_image(im_broken);
	
	imlib_context_set_drawable(win->pm);
	imlib_render_image_part_on_drawable_at_size(sx, sy, sw, sh, dx, dy, dw, dh);

	win_draw(win);
}

int img_fit_win(img_t *img, win_t *win) {
	if (!img || !img->im || !win)
		return 0;

	img->scalemode = SCALE_FIT;
	return img_fit(img, win);
}

int img_center(img_t *img, win_t *win) {
	int ox, oy;

	if (!img || !win)
		return 0;
	
	ox = img->x;
	oy = img->y;

	img->x = (win->w - img->w * img->zoom) / 2;
	img->y = (win->h - img->h * img->zoom) / 2;
	
	return ox != img->x || oy != img->y;
}

int img_zoom(img_t *img, float z) {
	if (!img || !img->im)
		return 0;

	z = MAX(z, zoom_min);
	z = MIN(z, zoom_max);

	img->scalemode = SCALE_ZOOM;

	if (z != img->zoom) {
		img->x -= (img->w * z - img->w * img->zoom) / 2;
		img->y -= (img->h * z - img->h * img->zoom) / 2;
		img->zoom = z;
		img->checkpan = 1;
		return 1;
	} else {
		return 0;
	}
}

int img_zoom_in(img_t *img) {
	int i;

	if (!img || !img->im)
		return 0;

	for (i = 1; i < zl_cnt; ++i) {
		if (zoom_levels[i] > img->zoom * 100.0)
			return img_zoom(img, zoom_levels[i] / 100.0);
	}
	return 0;
}

int img_zoom_out(img_t *img) {
	int i;

	if (!img || !img->im)
		return 0;

	for (i = zl_cnt - 2; i >= 0; --i) {
		if (zoom_levels[i] < img->zoom * 100.0)
			return img_zoom(img, zoom_levels[i] / 100.0);
	}
	return 0;
}

int img_move(img_t *img, win_t *win, int dx, int dy) {
	int ox, oy;

	if (!img || !img->im || !win)
		return 0;

	ox = img->x;
	oy = img->y;

	img->x += dx;
	img->y += dy;

	img_check_pan(img, win);

	return ox != img->x || oy != img->y;
}

int img_pan(img_t *img, win_t *win, pandir_t dir) {
	if (!img || !img->im || !win)
		return 0;

	switch (dir) {
		case PAN_LEFT:
			return img_move(img, win, win->w / 5, 0);
		case PAN_RIGHT:
			return img_move(img, win, win->w / 5 * -1, 0);
		case PAN_UP:
			return img_move(img, win, 0, win->h / 5);
		case PAN_DOWN:
			return img_move(img, win, 0, win->h / 5 * -1);
	}

	return 0;
}

void img_rotate(img_t *img, win_t *win, int d) {
	int ox, oy, tmp;

	if (!img || !img->im || !win)
		return;

	ox = d == 1 ? img->x : win->w - img->x - img->w * img->zoom;
	oy = d == 3 ? img->y : win->h - img->y - img->h * img->zoom;

	imlib_context_set_image(img->im);
	imlib_image_orientate(d);

	img->x = oy + (win->w - win->h) / 2;
	img->y = ox + (win->h - win->w) / 2;

	tmp = img->w;
	img->w = img->h;
	img->h = tmp;

	img->checkpan = 1;
}

void img_rotate_left(img_t *img, win_t *win) {
	if (img) {
		img_rotate(img, win, 3);
		img->ori = ori_left[img->ori];
	}
}

void img_rotate_right(img_t *img, win_t *win) {
	if (img) {
		img_rotate(img, win, 1);
		img->ori = ori_right[img->ori];
	}
}

void img_toggle_antialias(img_t *img) {
	if (img && img->im) {
		img->aa ^= 1;
		imlib_context_set_image(img->im);
		imlib_context_set_anti_alias(img->aa);
	}
}