/* mosaic two images left/right or up/down
 *
 * 22/5/14
 * 	- from vips_mosaic()
 * 4/9/18
 * 	- add docs for transform output
 * 18/6/20 kleisauke
 * 	- convert to vips8
 */

/*

	This file is part of VIPS.

	VIPS is free software; you can redistribute it and/or modify
	it under the terms of the GNU Lesser 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 Lesser General Public License for more details.

	You should have received a copy of the GNU Lesser General Public License
	along with this program; if not, write to the Free Software
	Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA
	02110-1301  USA

 */

/*

	These files are distributed with VIPS - http://www.vips.ecs.soton.ac.uk

 */

#ifdef HAVE_CONFIG_H
#include <config.h>
#endif /*HAVE_CONFIG_H*/
#include <glib/gi18n-lib.h>

#include <stdio.h>

#include <vips/vips.h>

#include "pmosaicing.h"

typedef struct {
	VipsOperation parent_instance;

	VipsImage *ref;
	VipsImage *sec;
	VipsImage *out;
	VipsDirection direction;
	int xref;
	int yref;
	int xsec;
	int ysec;
	int mblend;
	int bandno;
	int hwindow;
	int harea;
	int dx0;
	int dy0;
	double scale1;
	double angle1;
	double dx1;
	double dy1;

} VipsMosaic;

typedef VipsOperationClass VipsMosaicClass;

G_DEFINE_TYPE(VipsMosaic, vips_mosaic, VIPS_TYPE_OPERATION);

static int
vips_mosaic_build(VipsObject *object)
{
	VipsMosaic *mosaic = (VipsMosaic *) object;

	VipsImage *x;
	int dx0;
	int dy0;
	double scale1;
	double angle1;
	double dx1;
	double dy1;

	g_object_set(mosaic, "out", vips_image_new(), NULL);

	if (VIPS_OBJECT_CLASS(vips_mosaic_parent_class)->build(object))
		return -1;

	/* A placeholder used to ensure that memory used by the analysis
	 * phase is freed as soon as possible.
	 */
	x = vips_image_new();

	switch (mosaic->direction) {
	case VIPS_DIRECTION_HORIZONTAL:
		if (vips__find_lroverlap(mosaic->ref, mosaic->sec, x,
				mosaic->bandno,
				mosaic->xref, mosaic->yref, mosaic->xsec, mosaic->ysec,
				mosaic->hwindow, mosaic->harea,
				&dx0, &dy0,
				&scale1, &angle1,
				&dx1, &dy1)) {
			g_object_unref(x);
			return -1;
		}
		g_object_unref(x);
		break;

	case VIPS_DIRECTION_VERTICAL:
		if (vips__find_tboverlap(mosaic->ref, mosaic->sec, x,
				mosaic->bandno,
				mosaic->xref, mosaic->yref, mosaic->xsec, mosaic->ysec,
				mosaic->hwindow, mosaic->harea,
				&dx0, &dy0,
				&scale1, &angle1,
				&dx1, &dy1)) {
			g_object_unref(x);
			return -1;
		}
		g_object_unref(x);
		break;

	default:
		g_assert_not_reached();

		/* Compiler warnings.
		 */
		dx0 = 0;
		dy0 = 0;
		scale1 = 1;
		angle1 = 1;
		dx1 = 0;
		dy1 = 0;
	}

	g_object_set(mosaic,
		"dx0", dx0,
		"dy0", dy0,
		"scale1", scale1,
		"angle1", angle1,
		"dx1", dx1,
		"dy1", dy1,
		NULL);

	if (vips_merge(mosaic->ref, mosaic->sec, &x,
			mosaic->direction, mosaic->dx0, mosaic->dy0,
			"mblend", mosaic->mblend,
			NULL))
		return -1;
	if (vips_image_write(x, mosaic->out)) {
		g_object_unref(x);
		return -1;
	}
	g_object_unref(x);

	return 0;
}

static void
vips_mosaic_class_init(VipsMosaicClass *class)
{
	GObjectClass *gobject_class = G_OBJECT_CLASS(class);
	VipsObjectClass *object_class = (VipsObjectClass *) class;

	gobject_class->set_property = vips_object_set_property;
	gobject_class->get_property = vips_object_get_property;

	object_class->nickname = "mosaic";
	object_class->description = _("mosaic two images");
	object_class->build = vips_mosaic_build;

	VIPS_ARG_IMAGE(class, "ref", 1,
		_("Reference"),
		_("Reference image"),
		VIPS_ARGUMENT_REQUIRED_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, ref));

	VIPS_ARG_IMAGE(class, "sec", 2,
		_("Secondary"),
		_("Secondary image"),
		VIPS_ARGUMENT_REQUIRED_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, sec));

	VIPS_ARG_IMAGE(class, "out", 3,
		_("Output"),
		_("Output image"),
		VIPS_ARGUMENT_REQUIRED_OUTPUT,
		G_STRUCT_OFFSET(VipsMosaic, out));

	VIPS_ARG_ENUM(class, "direction", 4,
		_("Direction"),
		_("Horizontal or vertical mosaic"),
		VIPS_ARGUMENT_REQUIRED_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, direction),
		VIPS_TYPE_DIRECTION, VIPS_DIRECTION_HORIZONTAL);

	VIPS_ARG_INT(class, "xref", 5,
		_("xref"),
		_("Position of reference tie-point"),
		VIPS_ARGUMENT_REQUIRED_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, xref),
		0, 1000000000, 1);

	VIPS_ARG_INT(class, "yref", 6,
		_("yref"),
		_("Position of reference tie-point"),
		VIPS_ARGUMENT_REQUIRED_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, yref),
		0, 1000000000, 1);

	VIPS_ARG_INT(class, "xsec", 7,
		_("xsec"),
		_("Position of secondary tie-point"),
		VIPS_ARGUMENT_REQUIRED_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, xsec),
		0, 1000000000, 1);

	VIPS_ARG_INT(class, "ysec", 8,
		_("ysec"),
		_("Position of secondary tie-point"),
		VIPS_ARGUMENT_REQUIRED_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, ysec),
		0, 1000000000, 1);

	VIPS_ARG_INT(class, "hwindow", 9,
		_("hwindow"),
		_("Half window size"),
		VIPS_ARGUMENT_OPTIONAL_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, hwindow),
		0, 1000000000, 5);

	VIPS_ARG_INT(class, "harea", 10,
		_("harea"),
		_("Half area size"),
		VIPS_ARGUMENT_OPTIONAL_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, harea),
		0, 1000000000, 15);

	VIPS_ARG_INT(class, "mblend", 11,
		_("Max blend"),
		_("Maximum blend size"),
		VIPS_ARGUMENT_OPTIONAL_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, mblend),
		0, 10000, 10);

	VIPS_ARG_INT(class, "bandno", 12,
		_("Search band"),
		_("Band to search for features on"),
		VIPS_ARGUMENT_OPTIONAL_INPUT,
		G_STRUCT_OFFSET(VipsMosaic, bandno),
		0, 10000, 0);

	VIPS_ARG_INT(class, "dx0", 13,
		_("Integer offset"),
		_("Detected integer offset"),
		VIPS_ARGUMENT_OPTIONAL_OUTPUT,
		G_STRUCT_OFFSET(VipsMosaic, dx0),
		-10000000, 10000000, 0);

	VIPS_ARG_INT(class, "dy0", 14,
		_("Integer offset"),
		_("Detected integer offset"),
		VIPS_ARGUMENT_OPTIONAL_OUTPUT,
		G_STRUCT_OFFSET(VipsMosaic, dy0),
		-10000000, 10000000, 0);

	VIPS_ARG_DOUBLE(class, "scale1", 15,
		_("Scale"),
		_("Detected scale"),
		VIPS_ARGUMENT_OPTIONAL_OUTPUT,
		G_STRUCT_OFFSET(VipsMosaic, scale1),
		-10000000.0, 10000000.0, 1.0);

	VIPS_ARG_DOUBLE(class, "angle1", 16,
		_("Angle"),
		_("Detected rotation"),
		VIPS_ARGUMENT_OPTIONAL_OUTPUT,
		G_STRUCT_OFFSET(VipsMosaic, angle1),
		-10000000.0, 10000000.0, 0.0);

	VIPS_ARG_DOUBLE(class, "dx1", 17,
		_("First-order displacement"),
		_("Detected first-order displacement"),
		VIPS_ARGUMENT_OPTIONAL_OUTPUT,
		G_STRUCT_OFFSET(VipsMosaic, dx1),
		-10000000.0, 10000000.0, 0.0);

	VIPS_ARG_DOUBLE(class, "dy1", 17,
		_("First-order displacement"),
		_("Detected first-order displacement"),
		VIPS_ARGUMENT_OPTIONAL_OUTPUT,
		G_STRUCT_OFFSET(VipsMosaic, dy1),
		-10000000.0, 10000000.0, 0.0);
}

static void
vips_mosaic_init(VipsMosaic *mosaic)
{
	mosaic->mblend = 10;
	mosaic->hwindow = 5;
	mosaic->harea = 15;
	mosaic->scale1 = 1.0;
}

/**
 * vips_mosaic:
 * @ref: reference image
 * @sec: secondary image
 * @out: (out): output image
 * @direction: horizontal or vertical join
 * @xref: position in reference image
 * @yref: position in reference image
 * @xsec: position in secondary image
 * @ysec: position in secondary image
 * @...: %NULL-terminated list of optional named arguments
 *
 * Optional arguments:
 *
 * * @bandno: %gint, band to search for features
 * * @hwindow: %gint, half window size
 * * @harea: %gint, half search size
 * * @mblend: %gint, maximum blend size
 * * @dx0: %gint, output, detected displacement
 * * @dy0: %gint, output, detected displacement
 * * @scale1: %gdouble, output, detected first order scale
 * * @angle1: %gdouble, output, detected first order rotation
 * * @dx1: %gdouble, output, detected first order displacement
 * * @dy1: %gdouble, output, detected first order displacement
 *
 * This operation joins two images left-right (with @ref on the left) or
 * top-bottom (with @ref above) given an approximate overlap.
 *
 * @sec is positioned so that the pixel (@xsec, @ysec) in @sec lies on top of
 * the pixel (@xref, @yref) in @ref. The overlap area is divided into three
 * sections, 20 high-contrast points in band @bandno of image @ref are found
 * in each, and a window of pixels of size @hwindow around each high-contrast
 * point is searched for in @sec over an area of @harea.
 *
 * A linear model is fitted to the 60 tie-points, points a long way from the
 * fit are discarded, and the model refitted until either too few points
 * remain or the model reaches good agreement.
 *
 * The detected displacement is used with vips_merge() to join the two images
 * together.
 *
 * You can read out the detected transform with @dx0, @dy0, @scale1, @angle1,
 * @dx1, @dy1.
 *
 * See also: vips_merge(), vips_insert().
 *
 * Returns: 0 on success, -1 on error
 */
int
vips_mosaic(VipsImage *ref, VipsImage *sec, VipsImage **out,
	VipsDirection direction, int xref, int yref, int xsec, int ysec, ...)
{
	va_list ap;
	int result;

	va_start(ap, ysec);
	result = vips_call_split("mosaic", ap, ref, sec, out,
		direction, xref, yref, xsec, ysec);
	va_end(ap);

	return result;
}
