374 lines
9.3 KiB
C
374 lines
9.3 KiB
C
/* 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 );
|
|
}
|