#include "pps-pixbuf-cache.h"
#include "pps-colors.h"
#include "pps-job-scheduler.h"
#include "pps-view-private.h"
#include <config.h>

typedef enum {
	SCROLL_DIRECTION_DOWN,
	SCROLL_DIRECTION_UP
} ScrollDirection;

typedef struct _CacheJobInfo {
	PpsJob *job;
	gboolean page_ready;

	PpsJob *pending_job;
	PpsJobPriority pending_priority;

	/* Region of the page that needs to be drawn */
	cairo_region_t *region;
	GdkTexture *texture;

	/* Device scale factor of target widget */
	int device_scale;

	/* Selection data.
	 * Selection_points are the coordinates encapsulated in selection.
	 * target_points is the target selection size. */
	PpsRectangle target_points;
	PpsSelectionStyle selection_style;
	gboolean points_set;
	gboolean selection_stale;

	GdkTexture *selection_texture;
	cairo_surface_t *selection_surface;
	gdouble selection_scale;
	PpsRectangle selection_points;

	cairo_region_t *selection_region;
	gdouble selection_region_scale;
	PpsRectangle selection_region_points;
} CacheJobInfo;

struct _PpsPixbufCache {
	GObject parent;

	/* We keep a link to our containing view just for style information. */
	GtkWidget *view;
	PpsDocument *document;
	PpsDocumentModel *model;
	int start_page;
	int end_page;
	ScrollDirection scroll_direction;

	gsize max_size;

	/* preload_cache_size is the number of pages prior to the current
	 * visible area that we cache.  It's normally 1, but could be 2 in the
	 * case of twin pages.
	 */
	int preload_cache_size;
	guint job_list_len;

	CacheJobInfo *prev_job;
	CacheJobInfo *job_list;
	CacheJobInfo *next_job;
};

struct _PpsPixbufCacheClass {
	GObjectClass parent_class;

	void (*job_finished) (PpsPixbufCache *pixbuf_cache);
};

enum {
	JOB_FINISHED,
	N_SIGNALS,
};

static guint signals[N_SIGNALS] = {
	0,
};

static void pps_pixbuf_cache_init (PpsPixbufCache *pixbuf_cache);
static void pps_pixbuf_cache_class_init (PpsPixbufCacheClass *pixbuf_cache);
static void pps_pixbuf_cache_finalize (GObject *object);
static void pps_pixbuf_cache_dispose (GObject *object);
static void job_finished_cb (PpsJob *job,
                             PpsPixbufCache *pixbuf_cache);
static CacheJobInfo *find_job_cache (PpsPixbufCache *pixbuf_cache,
                                     int page);
static gboolean new_selection_surface_needed (PpsPixbufCache *pixbuf_cache,
                                              CacheJobInfo *job_info,
                                              gint page,
                                              gfloat scale);

/* These are used for iterating through the prev and next arrays */
#define FIRST_VISIBLE_PREV(pixbuf_cache) \
	(MAX (0, pixbuf_cache->preload_cache_size - pixbuf_cache->start_page))
#define VISIBLE_NEXT_LEN(pixbuf_cache) \
	(MIN (pixbuf_cache->preload_cache_size, pps_document_get_n_pages (pixbuf_cache->document) - (1 + pixbuf_cache->end_page)))
#define PAGE_CACHE_LEN(pixbuf_cache) \
	((pixbuf_cache->end_page - pixbuf_cache->start_page) + 1)

#define MAX_PRELOADED_PAGES 3

G_DEFINE_TYPE (PpsPixbufCache, pps_pixbuf_cache, G_TYPE_OBJECT)

static void
pps_pixbuf_cache_init (PpsPixbufCache *pixbuf_cache)
{
	pixbuf_cache->start_page = -1;
	pixbuf_cache->end_page = -1;
}

static void
pps_pixbuf_cache_class_init (PpsPixbufCacheClass *class)
{
	GObjectClass *object_class;

	object_class = G_OBJECT_CLASS (class);

	object_class->finalize = pps_pixbuf_cache_finalize;
	object_class->dispose = pps_pixbuf_cache_dispose;

	signals[JOB_FINISHED] =
	    g_signal_new ("job-finished",
	                  G_OBJECT_CLASS_TYPE (object_class),
	                  G_SIGNAL_RUN_LAST | G_SIGNAL_ACTION,
	                  G_STRUCT_OFFSET (PpsPixbufCacheClass, job_finished),
	                  NULL, NULL,
	                  g_cclosure_marshal_VOID__INT,
	                  G_TYPE_NONE, 1,
	                  G_TYPE_INT);
}

static void
pps_pixbuf_cache_finalize (GObject *object)
{
	PpsPixbufCache *pixbuf_cache;

	pixbuf_cache = PPS_PIXBUF_CACHE (object);

	if (pixbuf_cache->job_list) {
		g_slice_free1 (sizeof (CacheJobInfo) * pixbuf_cache->job_list_len,
		               pixbuf_cache->job_list);
		pixbuf_cache->job_list = NULL;
	}
	if (pixbuf_cache->prev_job) {
		g_slice_free1 (sizeof (CacheJobInfo) * pixbuf_cache->preload_cache_size,
		               pixbuf_cache->prev_job);
		pixbuf_cache->prev_job = NULL;
	}
	if (pixbuf_cache->next_job) {
		g_slice_free1 (sizeof (CacheJobInfo) * pixbuf_cache->preload_cache_size,
		               pixbuf_cache->next_job);
		pixbuf_cache->next_job = NULL;
	}

	g_clear_object (&pixbuf_cache->model);

	G_OBJECT_CLASS (pps_pixbuf_cache_parent_class)->finalize (object);
}

static void
end_job (CacheJobInfo *job_info,
         gpointer data)
{
	g_signal_handlers_disconnect_by_func (job_info->job,
	                                      G_CALLBACK (job_finished_cb),
	                                      data);
	if (!pps_job_is_finished (job_info->job))
		pps_job_cancel (job_info->job);
	g_clear_object (&job_info->job);
}

static void
dispose_cache_job_info (CacheJobInfo *job_info,
                        gpointer data)
{
	if (job_info == NULL)
		return;

	if (job_info->job)
		end_job (job_info, data);

	g_clear_object (&job_info->texture);
	g_clear_object (&job_info->selection_texture);
	g_clear_pointer (&job_info->selection_surface, cairo_surface_destroy);
	g_clear_pointer (&job_info->region, cairo_region_destroy);
	g_clear_pointer (&job_info->selection_region, cairo_region_destroy);

	job_info->points_set = FALSE;
}

static void
pps_pixbuf_cache_dispose (GObject *object)
{
	PpsPixbufCache *pixbuf_cache;
	int i;

	pixbuf_cache = PPS_PIXBUF_CACHE (object);

	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		dispose_cache_job_info (pixbuf_cache->prev_job + i, pixbuf_cache);
		dispose_cache_job_info (pixbuf_cache->next_job + i, pixbuf_cache);
	}

	for (i = 0; i < PAGE_CACHE_LEN (pixbuf_cache); i++) {
		dispose_cache_job_info (pixbuf_cache->job_list + i, pixbuf_cache);
	}

	G_OBJECT_CLASS (pps_pixbuf_cache_parent_class)->dispose (object);
}

PpsPixbufCache *
pps_pixbuf_cache_new (GtkWidget *view,
                      PpsDocumentModel *model,
                      gsize max_size)
{
	PpsPixbufCache *pixbuf_cache;

	pixbuf_cache = (PpsPixbufCache *) g_object_new (PPS_TYPE_PIXBUF_CACHE, NULL);
	/* This is a backlink, so we don't ref this */
	pixbuf_cache->view = view;
	pixbuf_cache->model = g_object_ref (model);
	pixbuf_cache->document = pps_document_model_get_document (model);
	pixbuf_cache->max_size = max_size;

	return pixbuf_cache;
}

void
pps_pixbuf_cache_set_max_size (PpsPixbufCache *pixbuf_cache,
                               gsize max_size)
{
	if (pixbuf_cache->max_size == max_size)
		return;

	if (pixbuf_cache->max_size > max_size)
		pps_pixbuf_cache_clear (pixbuf_cache);
	pixbuf_cache->max_size = max_size;
}

static int
get_device_scale (PpsPixbufCache *pixbuf_cache)
{
	return gtk_widget_get_scale_factor (pixbuf_cache->view);
}

static void
copy_job_to_job_info (PpsJobRenderTexture *job_render,
                      CacheJobInfo *job_info,
                      PpsPixbufCache *pixbuf_cache)
{
	g_set_object (&job_info->texture, job_render->texture);

	job_info->points_set = FALSE;
	if (job_render->include_selection) {
		g_clear_pointer (&job_info->selection_region, cairo_region_destroy);

		job_info->selection_points = job_render->selection_points;
		job_info->selection_scale = job_render->scale * job_info->device_scale;
		g_assert (job_info->selection_points.x1 >= 0);

		job_info->selection_region_points = job_render->selection_points;
		job_info->selection_region = cairo_region_reference (job_render->selection_region);
		job_info->selection_region_scale = job_render->scale;

		g_set_object (&job_info->selection_texture, job_render->selection);
		job_info->points_set = TRUE;
	}
	g_clear_pointer (&job_info->selection_surface, cairo_surface_destroy);

	if (job_info->job)
		end_job (job_info, pixbuf_cache);

	job_info->page_ready = TRUE;
}

static void
job_finished_cb (PpsJob *job,
                 PpsPixbufCache *pixbuf_cache)
{
	CacheJobInfo *job_info;
	PpsJobRenderTexture *job_render = PPS_JOB_RENDER_TEXTURE (job);

	/* If the job is outside of our interest, we silently discard it */
	if ((job_render->page < (pixbuf_cache->start_page - pixbuf_cache->preload_cache_size)) ||
	    (job_render->page > (pixbuf_cache->end_page + pixbuf_cache->preload_cache_size))) {
		g_object_unref (job);
		return;
	}

	job_info = find_job_cache (pixbuf_cache, job_render->page);

	if (!pps_job_is_succeeded (job, NULL)) {
		g_clear_object (&job_info->job);
		return;
	}

	copy_job_to_job_info (job_render, job_info, pixbuf_cache);
	g_signal_emit (pixbuf_cache, signals[JOB_FINISHED], 0, job_render->page);

	if (job_info->pending_job) {
		job_info->job = job_info->pending_job;
		job_info->pending_job = NULL;
		pps_job_scheduler_push_job (job_info->job, job_info->pending_priority);
	}
}

/* This checks a job to see if the job would generate the right sized pixbuf
 * given a scale.  If it won't, it removes the job and clears it to NULL.
 */
static void
check_job_size_and_unref (PpsPixbufCache *pixbuf_cache,
                          CacheJobInfo *job_info,
                          gfloat scale)
{
	gint width, height;
	gint device_scale;

	g_assert (job_info);

	if (job_info->job == NULL)
		return;

	device_scale = get_device_scale (pixbuf_cache);
	if (job_info->device_scale == device_scale) {
		_get_page_size_for_scale_and_rotation (pps_job_get_document (job_info->job),
		                                       PPS_JOB_RENDER_TEXTURE (job_info->job)->page,
		                                       scale,
		                                       PPS_JOB_RENDER_TEXTURE (job_info->job)->rotation,
		                                       &width, &height);
		if (width * device_scale == PPS_JOB_RENDER_TEXTURE (job_info->job)->target_width &&
		    height * device_scale == PPS_JOB_RENDER_TEXTURE (job_info->job)->target_height)
			return;
	}

	end_job (job_info, pixbuf_cache);
}

/* Do all function that copies a job from an older cache to it's position in the
 * new cache.  It clears the old job if it doesn't have a place.
 */
static void
move_one_job (CacheJobInfo *job_info,
              PpsPixbufCache *pixbuf_cache,
              int page,
              CacheJobInfo *new_job_list,
              CacheJobInfo *new_prev_job,
              CacheJobInfo *new_next_job,
              int new_preload_cache_size,
              int start_page,
              int end_page,
              gint priority)
{
	CacheJobInfo *target_page = NULL;
	int page_offset;
	gint new_priority;

	if (page < (start_page - new_preload_cache_size) ||
	    page > (end_page + new_preload_cache_size)) {
		dispose_cache_job_info (job_info, pixbuf_cache);
		return;
	}

	/* find the target page to copy it over to. */
	if (page < start_page) {
		page_offset = (page - (start_page - new_preload_cache_size));

		g_assert (page_offset >= 0 &&
		          page_offset < new_preload_cache_size);
		target_page = new_prev_job + page_offset;
		new_priority = PPS_JOB_PRIORITY_LOW;
	} else if (page > end_page) {
		page_offset = (page - (end_page + 1));

		g_assert (page_offset >= 0 &&
		          page_offset < new_preload_cache_size);
		target_page = new_next_job + page_offset;
		new_priority = PPS_JOB_PRIORITY_LOW;
	} else {
		page_offset = page - start_page;
		g_assert (page_offset >= 0 &&
		          page_offset <= ((end_page - start_page) + 1));
		new_priority = PPS_JOB_PRIORITY_URGENT;
		target_page = new_job_list + page_offset;
	}

	*target_page = *job_info;
	job_info->job = NULL;
	job_info->region = NULL;
	job_info->texture = NULL;

	if (new_priority != priority && target_page->job) {
		pps_job_scheduler_update_job (target_page->job, new_priority);
	}
}

static gsize
pps_pixbuf_cache_get_page_size (PpsPixbufCache *pixbuf_cache,
                                gint page_index,
                                gdouble scale,
                                gint rotation)
{
	gint width, height;

	_get_page_size_for_scale_and_rotation (pixbuf_cache->document,
	                                       page_index, scale, rotation,
	                                       &width, &height);
	return height * cairo_format_stride_for_width (CAIRO_FORMAT_RGB24, width);
}

static gint
pps_pixbuf_cache_get_preload_size (PpsPixbufCache *pixbuf_cache,
                                   gint start_page,
                                   gint end_page,
                                   gdouble scale,
                                   gint rotation)
{
	gsize range_size = 0;
	gint new_preload_cache_size = 0;
	gint i;
	guint n_pages = pps_document_get_n_pages (pixbuf_cache->document);

	/* Get the size of the current range */
	for (i = start_page; i <= end_page; i++) {
		range_size += pps_pixbuf_cache_get_page_size (pixbuf_cache, i, scale, rotation);
	}

	if (range_size >= pixbuf_cache->max_size)
		return new_preload_cache_size;

	i = 1;
	while (((start_page - i > 0) || (end_page + i < n_pages)) &&
	       new_preload_cache_size < MAX_PRELOADED_PAGES) {
		gsize page_size;
		gboolean updated = FALSE;

		if (end_page + i < n_pages) {
			page_size = pps_pixbuf_cache_get_page_size (pixbuf_cache, end_page + i,
			                                            scale, rotation);
			if (page_size + range_size <= pixbuf_cache->max_size) {
				range_size += page_size;
				new_preload_cache_size++;
				updated = TRUE;
			} else {
				break;
			}
		}

		if (start_page - i > 0) {
			page_size = pps_pixbuf_cache_get_page_size (pixbuf_cache, start_page - i,
			                                            scale, rotation);
			if (page_size + range_size <= pixbuf_cache->max_size) {
				range_size += page_size;
				if (!updated)
					new_preload_cache_size++;
			} else {
				break;
			}
		}
		i++;
	}

	return new_preload_cache_size;
}

static void
pps_pixbuf_cache_update_range (PpsPixbufCache *pixbuf_cache,
                               gint start_page,
                               gint end_page,
                               guint rotation,
                               gdouble scale)
{
	CacheJobInfo *new_job_list;
	CacheJobInfo *new_prev_job = NULL;
	CacheJobInfo *new_next_job = NULL;
	gint new_preload_cache_size;
	guint new_job_list_len;
	int i, page;

	new_preload_cache_size = pps_pixbuf_cache_get_preload_size (pixbuf_cache,
	                                                            start_page,
	                                                            end_page,
	                                                            scale,
	                                                            rotation);
	if (pixbuf_cache->start_page == start_page &&
	    pixbuf_cache->end_page == end_page &&
	    pixbuf_cache->preload_cache_size == new_preload_cache_size)
		return;

	new_job_list_len = (end_page - start_page) + 1;
	new_job_list = g_slice_alloc0 (sizeof (CacheJobInfo) * new_job_list_len);
	if (new_preload_cache_size > 0) {
		new_prev_job = g_slice_alloc0 (sizeof (CacheJobInfo) * new_preload_cache_size);
		new_next_job = g_slice_alloc0 (sizeof (CacheJobInfo) * new_preload_cache_size);
	}

	/* We go through each job in the old cache and either clear it or move
	 * it to a new location. */

	/* Start with the prev cache. */
	page = pixbuf_cache->start_page - pixbuf_cache->preload_cache_size;
	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		if (page < 0) {
			dispose_cache_job_info (pixbuf_cache->prev_job + i, pixbuf_cache);
		} else {
			move_one_job (pixbuf_cache->prev_job + i,
			              pixbuf_cache, page,
			              new_job_list, new_prev_job, new_next_job,
			              new_preload_cache_size,
			              start_page, end_page, PPS_JOB_PRIORITY_LOW);
		}
		page++;
	}

	page = pixbuf_cache->start_page;
	for (i = 0; i < PAGE_CACHE_LEN (pixbuf_cache) && page >= 0; i++) {
		move_one_job (pixbuf_cache->job_list + i,
		              pixbuf_cache, page,
		              new_job_list, new_prev_job, new_next_job,
		              new_preload_cache_size,
		              start_page, end_page, PPS_JOB_PRIORITY_URGENT);
		page++;
	}

	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		if (page >= pps_document_get_n_pages (pixbuf_cache->document)) {
			dispose_cache_job_info (pixbuf_cache->next_job + i, pixbuf_cache);
		} else {
			move_one_job (pixbuf_cache->next_job + i,
			              pixbuf_cache, page,
			              new_job_list, new_prev_job, new_next_job,
			              new_preload_cache_size,
			              start_page, end_page, PPS_JOB_PRIORITY_LOW);
		}
		page++;
	}

	if (pixbuf_cache->job_list) {
		g_slice_free1 (sizeof (CacheJobInfo) * pixbuf_cache->job_list_len,
		               pixbuf_cache->job_list);
	}
	if (pixbuf_cache->prev_job) {
		g_slice_free1 (sizeof (CacheJobInfo) * pixbuf_cache->preload_cache_size,
		               pixbuf_cache->prev_job);
	}
	if (pixbuf_cache->next_job) {
		g_slice_free1 (sizeof (CacheJobInfo) * pixbuf_cache->preload_cache_size,
		               pixbuf_cache->next_job);
	}

	pixbuf_cache->preload_cache_size = new_preload_cache_size;
	pixbuf_cache->job_list_len = new_job_list_len;

	pixbuf_cache->job_list = new_job_list;
	pixbuf_cache->prev_job = new_prev_job;
	pixbuf_cache->next_job = new_next_job;

	pixbuf_cache->start_page = start_page;
	pixbuf_cache->end_page = end_page;
}

static CacheJobInfo *
find_job_cache (PpsPixbufCache *pixbuf_cache,
                int page)
{
	int page_offset;

	if (page < (pixbuf_cache->start_page - pixbuf_cache->preload_cache_size) ||
	    page > (pixbuf_cache->end_page + pixbuf_cache->preload_cache_size))
		return NULL;

	if (page < pixbuf_cache->start_page) {
		page_offset = (page - (pixbuf_cache->start_page - pixbuf_cache->preload_cache_size));

		g_assert (page_offset >= 0 &&
		          page_offset < pixbuf_cache->preload_cache_size);
		return pixbuf_cache->prev_job + page_offset;
	}

	if (page > pixbuf_cache->end_page) {
		page_offset = (page - (pixbuf_cache->end_page + 1));

		g_assert (page_offset >= 0 &&
		          page_offset < pixbuf_cache->preload_cache_size);
		return pixbuf_cache->next_job + page_offset;
	}

	page_offset = page - pixbuf_cache->start_page;
	g_assert (page_offset >= 0 &&
	          page_offset <= PAGE_CACHE_LEN (pixbuf_cache));
	return pixbuf_cache->job_list + page_offset;
}

static void
pps_pixbuf_cache_clear_job_sizes (PpsPixbufCache *pixbuf_cache,
                                  gfloat scale)
{
	int i;

	for (i = 0; i < PAGE_CACHE_LEN (pixbuf_cache); i++) {
		check_job_size_and_unref (pixbuf_cache, pixbuf_cache->job_list + i, scale);
	}

	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		check_job_size_and_unref (pixbuf_cache, pixbuf_cache->prev_job + i, scale);
		check_job_size_and_unref (pixbuf_cache, pixbuf_cache->next_job + i, scale);
	}
}

static PpsRenderAnnotsFlags
pps_pixbuf_cache_get_annot_flags (PpsPixbufCache *pixbuf_cache)
{
	switch (pps_document_model_get_annotation_editing_state (pixbuf_cache->model)) {
	case PPS_ANNOTATION_EDITING_STATE_INK:
		return PPS_RENDER_ANNOTS_ALL & ~(PPS_RENDER_ANNOTS_INK);
	case PPS_ANNOTATION_EDITING_STATE_TEXT:
		return PPS_RENDER_ANNOTS_ALL & ~(PPS_RENDER_ANNOTS_FREETEXT);
	default:
		return PPS_RENDER_ANNOTS_ALL;
	}
}

static void
add_job (PpsPixbufCache *pixbuf_cache,
         CacheJobInfo *job_info,
         cairo_region_t *region,
         gint width,
         gint height,
         gint page,
         gint rotation,
         gfloat scale,
         PpsJobPriority priority,
         PpsRenderAnnotsFlags annot_flags)
{
	PpsJob *job;
	job_info->device_scale = get_device_scale (pixbuf_cache);
	job_info->page_ready = FALSE;

	if (job_info->region)
		cairo_region_destroy (job_info->region);
	job_info->region = region ? cairo_region_reference (region) : NULL;

	job = pps_job_render_texture_new (pixbuf_cache->document,
	                                  page,
	                                  rotation,
	                                  scale * job_info->device_scale,
	                                  width * job_info->device_scale,
	                                  height * job_info->device_scale,
	                                  annot_flags);

	if (new_selection_surface_needed (pixbuf_cache, job_info, page, scale)) {
		GdkRGBA text, base;

		get_accent_color (&base, &text);
		pps_job_render_texture_set_selection_info (PPS_JOB_RENDER_TEXTURE (job),
		                                           &(job_info->target_points),
		                                           job_info->selection_style,
		                                           &text, &base);
	}
	job_info->selection_stale = FALSE;

	g_signal_connect (job, "finished",
	                  G_CALLBACK (job_finished_cb),
	                  pixbuf_cache);
	if (!job_info->job) {
		job_info->job = job;
		pps_job_scheduler_push_job (job, priority);
		return;
	}

	if (job_info->pending_job) {
		g_signal_handlers_disconnect_by_func (
		    job_info->pending_job, G_CALLBACK (job_finished_cb), pixbuf_cache);
		if (!pps_job_is_finished (job_info->pending_job))
			pps_job_cancel (job_info->pending_job);
		g_clear_object (&job_info->pending_job);
	}
	job_info->pending_priority = priority;
	job_info->pending_job = job;
}

static void
add_job_if_needed (PpsPixbufCache *pixbuf_cache,
                   CacheJobInfo *job_info,
                   gint page,
                   gint rotation,
                   gfloat scale,
                   PpsJobPriority priority,
                   PpsRenderAnnotsFlags annot_flags)
{
	gint device_scale = get_device_scale (pixbuf_cache);
	gint width, height;

	if (job_info->job)
		return;

	_get_page_size_for_scale_and_rotation (pixbuf_cache->document,
	                                       page, scale, rotation,
	                                       &width, &height);

	if (job_info->texture &&
	    job_info->device_scale == device_scale &&
	    gdk_texture_get_width (job_info->texture) == width * device_scale &&
	    gdk_texture_get_height (job_info->texture) == height * device_scale &&
	    (!job_info->points_set || !job_info->selection_stale))
		return;

	/* Free old surfaces for non visible pages */
	if (priority == PPS_JOB_PRIORITY_LOW) {
		g_clear_object (&job_info->texture);
		g_clear_object (&job_info->selection_texture);
	}

	add_job (pixbuf_cache, job_info, NULL,
	         width, height, page, rotation, scale,
	         priority, annot_flags);
}

static void
add_prev_jobs_if_needed (PpsPixbufCache *pixbuf_cache,
                         gint rotation,
                         gfloat scale,
                         PpsRenderAnnotsFlags annot_flags)
{
	CacheJobInfo *job_info;
	int page;
	int i;

	for (i = pixbuf_cache->preload_cache_size - 1; i >= FIRST_VISIBLE_PREV (pixbuf_cache); i--) {
		job_info = (pixbuf_cache->prev_job + i);
		page = pixbuf_cache->start_page - pixbuf_cache->preload_cache_size + i;

		add_job_if_needed (pixbuf_cache, job_info,
		                   page, rotation, scale,
		                   PPS_JOB_PRIORITY_LOW, annot_flags);
	}
}

static void
add_next_jobs_if_needed (PpsPixbufCache *pixbuf_cache,
                         gint rotation,
                         gfloat scale,
                         PpsRenderAnnotsFlags annot_flags)
{
	CacheJobInfo *job_info;
	int page;
	int i;

	for (i = 0; i < VISIBLE_NEXT_LEN (pixbuf_cache); i++) {
		job_info = (pixbuf_cache->next_job + i);
		page = pixbuf_cache->end_page + 1 + i;

		add_job_if_needed (pixbuf_cache, job_info,
		                   page, rotation, scale,
		                   PPS_JOB_PRIORITY_LOW, annot_flags);
	}
}

static void
pps_pixbuf_cache_add_jobs_if_needed (PpsPixbufCache *pixbuf_cache,
                                     gint rotation,
                                     gfloat scale,
                                     PpsRenderAnnotsFlags annot_flags)
{
	CacheJobInfo *job_info;
	int page;
	int i;

	for (i = 0; i < PAGE_CACHE_LEN (pixbuf_cache); i++) {
		job_info = (pixbuf_cache->job_list + i);
		page = pixbuf_cache->start_page + i;

		add_job_if_needed (pixbuf_cache, job_info,
		                   page, rotation, scale,
		                   PPS_JOB_PRIORITY_URGENT, annot_flags);
	}

	if (pixbuf_cache->scroll_direction == SCROLL_DIRECTION_UP) {
		add_prev_jobs_if_needed (pixbuf_cache, rotation, scale, annot_flags);
		add_next_jobs_if_needed (pixbuf_cache, rotation, scale, annot_flags);
	} else {
		add_next_jobs_if_needed (pixbuf_cache, rotation, scale, annot_flags);
		add_prev_jobs_if_needed (pixbuf_cache, rotation, scale, annot_flags);
	}
}

static ScrollDirection
pps_pixbuf_cache_get_scroll_direction (PpsPixbufCache *pixbuf_cache,
                                       gint start_page,
                                       gint end_page)
{
	if (start_page < pixbuf_cache->start_page)
		return SCROLL_DIRECTION_UP;

	if (end_page > pixbuf_cache->end_page)
		return SCROLL_DIRECTION_DOWN;

	if (start_page > pixbuf_cache->start_page)
		return SCROLL_DIRECTION_DOWN;

	if (end_page < pixbuf_cache->end_page)
		return SCROLL_DIRECTION_UP;

	return pixbuf_cache->scroll_direction;
}

void
pps_pixbuf_cache_set_page_range (PpsPixbufCache *pixbuf_cache,
                                 gint start_page,
                                 gint end_page,
                                 GList *selection_list)
{
	gdouble scale = pps_document_model_get_scale (pixbuf_cache->model);
	gint rotation = pps_document_model_get_rotation (pixbuf_cache->model);
	PpsRenderAnnotsFlags annot_flags = pps_pixbuf_cache_get_annot_flags (pixbuf_cache);

	g_return_if_fail (PPS_IS_PIXBUF_CACHE (pixbuf_cache));

	g_return_if_fail (start_page >= 0 && start_page < pps_document_get_n_pages (pixbuf_cache->document));
	g_return_if_fail (end_page >= 0 && end_page < pps_document_get_n_pages (pixbuf_cache->document));
	g_return_if_fail (end_page >= start_page);

	pixbuf_cache->scroll_direction = pps_pixbuf_cache_get_scroll_direction (pixbuf_cache, start_page, end_page);

	/* First, resize the page_range as needed.  We cull old pages
	 * mercilessly. */
	pps_pixbuf_cache_update_range (pixbuf_cache, start_page, end_page, rotation, scale);

	/* Then, we update the current jobs to see if any of them are the wrong
	 * size, we remove them if we need to. */
	pps_pixbuf_cache_clear_job_sizes (pixbuf_cache, scale);

	/* Next, we update the target selection for our pages */
	pps_pixbuf_cache_set_selection_list (pixbuf_cache, selection_list);

	/* Finally, we add the new jobs for all the sizes that don't have a
	 * pixbuf */
	pps_pixbuf_cache_add_jobs_if_needed (pixbuf_cache, rotation, scale, annot_flags);
}

GdkTexture *
pps_pixbuf_cache_get_texture (PpsPixbufCache *pixbuf_cache,
                              gint page)
{
	CacheJobInfo *job_info;

	job_info = find_job_cache (pixbuf_cache, page);
	if (job_info == NULL)
		return NULL;

	if (job_info->page_ready)
		return job_info->texture;

	/* We don't need to wait for the idle to handle the callback */
	if (job_info->job &&
	    PPS_JOB_RENDER_TEXTURE (job_info->job)->page_ready) {
		copy_job_to_job_info (PPS_JOB_RENDER_TEXTURE (job_info->job), job_info, pixbuf_cache);
		g_signal_emit (pixbuf_cache, signals[JOB_FINISHED], 0, job_info->region);
	}

	return job_info->texture;
}

static gboolean
new_selection_surface_needed (PpsPixbufCache *pixbuf_cache,
                              CacheJobInfo *job_info,
                              gint page,
                              gfloat scale)
{
	if (job_info->selection_stale)
		return TRUE;
	if (job_info->selection_texture)
		return job_info->selection_scale != scale;
	return job_info->points_set;
}

static gboolean
new_selection_region_needed (PpsPixbufCache *pixbuf_cache,
                             CacheJobInfo *job_info,
                             gint page,
                             gfloat scale)
{
	if (job_info->selection_stale)
		return TRUE;
	if (job_info->selection_region)
		return job_info->selection_region_scale != scale;
	return job_info->points_set;
}

static void
clear_selection_surface_if_needed (PpsPixbufCache *pixbuf_cache,
                                   CacheJobInfo *job_info,
                                   gint page,
                                   gfloat scale)
{
	if (new_selection_surface_needed (pixbuf_cache, job_info, page, scale)) {
		g_clear_object (&job_info->selection_texture);
		job_info->selection_points.x1 = -1;
	}
}

static void
clear_selection_region_if_needed (PpsPixbufCache *pixbuf_cache,
                                  CacheJobInfo *job_info,
                                  gint page,
                                  gfloat scale)
{
	if (new_selection_region_needed (pixbuf_cache, job_info, page, scale)) {
		g_clear_pointer (&job_info->selection_region, cairo_region_destroy);
		job_info->selection_region_points.x1 = -1;
	}
}

/* Clears the cache of jobs and pixbufs.
 */
void
pps_pixbuf_cache_clear (PpsPixbufCache *pixbuf_cache)
{
	int i;

	if (!pixbuf_cache->job_list)
		return;

	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		dispose_cache_job_info (pixbuf_cache->prev_job + i, pixbuf_cache);
		dispose_cache_job_info (pixbuf_cache->next_job + i, pixbuf_cache);
	}

	for (i = 0; i < PAGE_CACHE_LEN (pixbuf_cache); i++) {
		dispose_cache_job_info (pixbuf_cache->job_list + i, pixbuf_cache);
	}
}

void
pps_pixbuf_cache_style_changed (PpsPixbufCache *pixbuf_cache)
{
	gint i;

	if (!pixbuf_cache->job_list)
		return;

	/* FIXME: doesn't update running jobs. */
	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		CacheJobInfo *job_info;

		job_info = pixbuf_cache->prev_job + i;
		if (job_info->selection_texture) {
			g_clear_object (&job_info->selection_texture);
			g_clear_pointer (&job_info->selection_surface, cairo_surface_destroy);
			job_info->selection_points.x1 = -1;
		}

		job_info = pixbuf_cache->next_job + i;

		if (job_info->selection_texture) {
			g_clear_object (&job_info->selection_texture);
			g_clear_pointer (&job_info->selection_surface, cairo_surface_destroy);
			job_info->selection_points.x1 = -1;
		}
	}

	for (i = 0; i < PAGE_CACHE_LEN (pixbuf_cache); i++) {
		CacheJobInfo *job_info;

		job_info = pixbuf_cache->job_list + i;

		if (job_info->selection_texture) {
			g_clear_object (&job_info->selection_texture);
			g_clear_pointer (&job_info->selection_surface, cairo_surface_destroy);
			job_info->selection_points.x1 = -1;
		}
	}
}

GdkTexture *
pps_pixbuf_cache_get_selection_texture (PpsPixbufCache *pixbuf_cache,
                                        gint page,
                                        gfloat scale)
{
	CacheJobInfo *job_info;

	/* the document does not implement the selection interface */
	if (!PPS_IS_SELECTION (pixbuf_cache->document))
		return NULL;

	job_info = find_job_cache (pixbuf_cache, page);
	if (job_info == NULL)
		return NULL;

	/* No selection on this page */
	if (!job_info->points_set)
		return NULL;

	/* If we have a running job, we just return what we have under the
	 * assumption that it'll be updated later and we can scale it as need
	 * be */
	if (job_info->job && PPS_JOB_RENDER_TEXTURE (job_info->job)->include_selection)
		return job_info->selection_texture;

	/* Now, lets see if we need to resize the image.  If we do, we clear the
	 * old one. */
	clear_selection_surface_if_needed (pixbuf_cache, job_info, page, scale);

	/* Finally, we see if the two scales are the same, and get a new pixbuf
	 * if needed.  We do this synchronously for now.  At some point, we
	 * _should_ be able to get rid of the doc_mutex, so the synchronicity
	 * doesn't kill us.  Rendering a few glyphs should really be fast.
	 */
	if (pps_rect_cmp (&(job_info->target_points), &(job_info->selection_points))) {
		PpsRectangle *old_points;
		GdkRGBA text, base;
		PpsRenderContext *rc;
		PpsPage *pps_page;
		gint width, height;

		/* we need to get a new selection pixbuf */
		pps_document_doc_mutex_lock (pixbuf_cache->document);
		if (job_info->selection_points.x1 < 0) {
			g_assert (job_info->selection_texture == NULL);
			old_points = NULL;
		} else {
			old_points = &(job_info->selection_points);
		}

		pps_page = pps_document_get_page (pixbuf_cache->document, page);
		_get_page_size_for_scale_and_rotation (pixbuf_cache->document,
		                                       page,
		                                       scale * job_info->device_scale,
		                                       0, &width, &height);

		rc = pps_render_context_new (pps_page, 0, scale * job_info->device_scale, pps_pixbuf_cache_get_annot_flags (pixbuf_cache));
		pps_render_context_set_target_size (rc, width, height);
		g_object_unref (pps_page);

		get_accent_color (&base, &text);
		pps_selection_render_selection (PPS_SELECTION (pixbuf_cache->document),
		                                rc, &job_info->selection_surface,
		                                &(job_info->target_points),
		                                old_points,
		                                job_info->selection_style,
		                                &text, &base);

		job_info->selection_points = job_info->target_points;
		job_info->selection_scale = scale * job_info->device_scale;

		g_clear_object (&job_info->selection_texture);

		job_info->selection_texture = pps_document_misc_texture_from_surface (job_info->selection_surface);
		g_object_unref (rc);
		pps_document_doc_mutex_unlock (pixbuf_cache->document);
	}
	return job_info->selection_texture;
}

cairo_region_t *
pps_pixbuf_cache_get_selection_region (PpsPixbufCache *pixbuf_cache,
                                       gint page,
                                       gfloat scale)
{
	CacheJobInfo *job_info;

	/* the document does not implement the selection interface */
	if (!PPS_IS_SELECTION (pixbuf_cache->document))
		return NULL;

	job_info = find_job_cache (pixbuf_cache, page);
	if (job_info == NULL)
		return NULL;

	/* No selection on this page */
	if (!job_info->points_set)
		return NULL;

	/* If we have a running job, we just return what we have under the
	 * assumption that it'll be updated later and we can scale it as need
	 * be */
	if (job_info->job && PPS_JOB_RENDER_TEXTURE (job_info->job)->include_selection)
		return job_info->selection_region && !cairo_region_is_empty (job_info->selection_region) ? job_info->selection_region : NULL;

	/* Now, lets see if we need to resize the region.  If we do, we clear the
	 * old one. */
	clear_selection_region_if_needed (pixbuf_cache, job_info, page, scale);

	/* Finally, we see if the two scales are the same, and get a new region
	 * if needed.
	 */
	if (pps_rect_cmp (&(job_info->target_points), &(job_info->selection_region_points))) {
		PpsRenderContext *rc;
		PpsPage *pps_page;
		gint width, height;

		pps_document_doc_mutex_lock (pixbuf_cache->document);
		pps_page = pps_document_get_page (pixbuf_cache->document, page);

		_get_page_size_for_scale_and_rotation (pixbuf_cache->document,
		                                       page, scale, 0,
		                                       &width, &height);

		rc = pps_render_context_new (pps_page, 0, 0., PPS_RENDER_ANNOTS_ALL);
		pps_render_context_set_target_size (rc, width, height);
		g_object_unref (pps_page);

		if (job_info->selection_region)
			cairo_region_destroy (job_info->selection_region);
		job_info->selection_region =
		    pps_selection_get_selection_region (PPS_SELECTION (pixbuf_cache->document),
		                                        rc, job_info->selection_style,
		                                        &(job_info->target_points));
		job_info->selection_region_points = job_info->target_points;
		job_info->selection_region_scale = scale;
		g_object_unref (rc);
		pps_document_doc_mutex_unlock (pixbuf_cache->document);
	}
	return job_info->selection_region && !cairo_region_is_empty (job_info->selection_region) ? job_info->selection_region : NULL;
}

static void
clear_job_selection (CacheJobInfo *job_info)
{
	job_info->selection_stale = FALSE;
	job_info->points_set = FALSE;
	job_info->selection_points.x1 = -1;

	g_clear_object (&job_info->selection_texture);
	g_clear_pointer (&job_info->selection_region, cairo_region_destroy);
}

static void
update_job_selection (CacheJobInfo *job_info,
                      PpsViewSelection *selection)
{
	if (pps_rect_cmp (&job_info->target_points, &selection->rect) != 0 || !job_info->points_set || job_info->selection_style != selection->style) {
		job_info->selection_stale = TRUE;
		job_info->points_set = TRUE;
		job_info->target_points = selection->rect;
		job_info->selection_style = selection->style;
	}
}

/* This function will reset the selection on pages that no longer have them, and
 * will update the target_selection on those that need it.  It will _not_ free
 * the previous selection_list -- that's up to caller to do.
 */
void
pps_pixbuf_cache_set_selection_list (PpsPixbufCache *pixbuf_cache,
                                     GList *selection_list)
{
	PpsViewSelection *selection;
	GList *list = selection_list;
	int page;
	int i;

	g_return_if_fail (PPS_IS_PIXBUF_CACHE (pixbuf_cache));

	if (!PPS_IS_SELECTION (pixbuf_cache->document))
		return;

	if (pixbuf_cache->start_page == -1 || pixbuf_cache->end_page == -1)
		return;

	/* We check each area to see what needs updating, and what needs freeing; */
	page = pixbuf_cache->start_page - pixbuf_cache->preload_cache_size;
	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		if (page < 0) {
			page++;
			continue;
		}

		selection = NULL;
		while (list) {
			if (((PpsViewSelection *) list->data)->page == page) {
				selection = list->data;
				break;
			} else if (((PpsViewSelection *) list->data)->page > page)
				break;
			list = list->next;
		}

		if (selection)
			update_job_selection (pixbuf_cache->prev_job + i, selection);
		else
			clear_job_selection (pixbuf_cache->prev_job + i);
		page++;
	}

	page = pixbuf_cache->start_page;
	for (i = 0; i < PAGE_CACHE_LEN (pixbuf_cache); i++) {
		selection = NULL;
		while (list) {
			if (((PpsViewSelection *) list->data)->page == page) {
				selection = list->data;
				break;
			} else if (((PpsViewSelection *) list->data)->page > page)
				break;
			list = list->next;
		}

		if (selection)
			update_job_selection (pixbuf_cache->job_list + i, selection);
		else
			clear_job_selection (pixbuf_cache->job_list + i);
		page++;
	}

	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		if (page >= pps_document_get_n_pages (pixbuf_cache->document))
			break;

		selection = NULL;
		while (list) {
			if (((PpsViewSelection *) list->data)->page == page) {
				selection = list->data;
				break;
			} else if (((PpsViewSelection *) list->data)->page > page)
				break;
			list = list->next;
		}

		if (selection)
			update_job_selection (pixbuf_cache->next_job + i, selection);
		else
			clear_job_selection (pixbuf_cache->next_job + i);
		page++;
	}
}

/* Returns what the pixbuf cache thinks is */

GList *
pps_pixbuf_cache_get_selection_list (PpsPixbufCache *pixbuf_cache)
{
	PpsViewSelection *selection;
	GList *retval = NULL;
	int page;
	int i;

	g_return_val_if_fail (PPS_IS_PIXBUF_CACHE (pixbuf_cache), NULL);

	if (pixbuf_cache->start_page == -1 || pixbuf_cache->end_page == -1)
		return NULL;

	/* We check each area to see what needs updating, and what needs freeing; */
	page = pixbuf_cache->start_page - pixbuf_cache->preload_cache_size;
	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		if (page < 0) {
			page++;
			continue;
		}

		if (pixbuf_cache->prev_job[i].selection_points.x1 != -1) {
			selection = g_new0 (PpsViewSelection, 1);
			selection->page = page;
			selection->rect = pixbuf_cache->prev_job[i].selection_points;
			if (pixbuf_cache->prev_job[i].selection_region)
				selection->covered_region = cairo_region_reference (pixbuf_cache->prev_job[i].selection_region);
			retval = g_list_prepend (retval, selection);
		}

		page++;
	}

	page = pixbuf_cache->start_page;
	for (i = 0; i < PAGE_CACHE_LEN (pixbuf_cache); i++) {
		if (pixbuf_cache->job_list[i].selection_points.x1 != -1) {
			selection = g_new0 (PpsViewSelection, 1);
			selection->page = page;
			selection->rect = pixbuf_cache->job_list[i].selection_points;
			if (pixbuf_cache->job_list[i].selection_region)
				selection->covered_region = cairo_region_reference (pixbuf_cache->job_list[i].selection_region);
			retval = g_list_prepend (retval, selection);
		}

		page++;
	}

	for (i = 0; i < pixbuf_cache->preload_cache_size; i++) {
		if (page >= pps_document_get_n_pages (pixbuf_cache->document))
			break;

		if (pixbuf_cache->next_job[i].selection_points.x1 != -1) {
			selection = g_new0 (PpsViewSelection, 1);
			selection->page = page;
			selection->rect = pixbuf_cache->next_job[i].selection_points;
			if (pixbuf_cache->next_job[i].selection_region)
				selection->covered_region = cairo_region_reference (pixbuf_cache->next_job[i].selection_region);
			retval = g_list_prepend (retval, selection);
		}

		page++;
	}

	return g_list_reverse (retval);
}

void
pps_pixbuf_cache_reload_page (PpsPixbufCache *pixbuf_cache,
                              cairo_region_t *region,
                              gint page,
                              gint rotation,
                              gdouble scale)
{
	CacheJobInfo *job_info;
	gint width, height;
	PpsRenderAnnotsFlags annot_flags = pps_pixbuf_cache_get_annot_flags (pixbuf_cache);

	job_info = find_job_cache (pixbuf_cache, page);
	if (job_info == NULL)
		return;

	_get_page_size_for_scale_and_rotation (pixbuf_cache->document,
	                                       page, scale, rotation,
	                                       &width, &height);
	add_job (pixbuf_cache, job_info, region,
	         width, height, page, rotation, scale,
	         PPS_JOB_PRIORITY_URGENT, annot_flags);
}
