2014-03-26 21:33:31 +01:00
|
|
|
/* hough transform for circles
|
|
|
|
*
|
|
|
|
* 7/3/14
|
|
|
|
* - from hough_line.c
|
|
|
|
*/
|
|
|
|
|
|
|
|
/*
|
|
|
|
|
|
|
|
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 <vips/intl.h>
|
|
|
|
|
|
|
|
#include <string.h>
|
|
|
|
|
|
|
|
#include <vips/vips.h>
|
2014-04-01 13:27:15 +02:00
|
|
|
#include <vips/internal.h>
|
2014-03-26 21:33:31 +01:00
|
|
|
|
|
|
|
#include "statistic.h"
|
|
|
|
#include "hough.h"
|
|
|
|
|
|
|
|
typedef struct _VipsHoughCircle {
|
|
|
|
VipsHough parent_instance;
|
|
|
|
|
2014-04-01 19:54:00 +02:00
|
|
|
int scale;
|
2014-03-26 21:33:31 +01:00
|
|
|
int min_radius;
|
|
|
|
int max_radius;
|
2014-04-01 19:54:00 +02:00
|
|
|
|
|
|
|
int width;
|
|
|
|
int height;
|
2014-03-26 21:33:31 +01:00
|
|
|
int bands;
|
|
|
|
|
|
|
|
} VipsHoughCircle;
|
|
|
|
|
|
|
|
typedef VipsHoughClass VipsHoughCircleClass;
|
|
|
|
|
|
|
|
G_DEFINE_TYPE( VipsHoughCircle, vips_hough_circle, VIPS_TYPE_HOUGH );
|
|
|
|
|
|
|
|
static int
|
|
|
|
vips_hough_circle_build( VipsObject *object )
|
|
|
|
{
|
|
|
|
VipsObjectClass *class = VIPS_OBJECT_GET_CLASS( object );
|
2014-04-01 19:54:00 +02:00
|
|
|
VipsStatistic *statistic = (VipsStatistic *) object;
|
2014-03-26 21:33:31 +01:00
|
|
|
VipsHoughCircle *hough_circle = (VipsHoughCircle *) object;
|
2014-04-01 19:54:00 +02:00
|
|
|
int range = hough_circle->max_radius - hough_circle->min_radius;
|
2014-03-26 21:33:31 +01:00
|
|
|
|
2014-04-01 19:54:00 +02:00
|
|
|
if( range <= 0 ) {
|
2014-03-26 21:33:31 +01:00
|
|
|
vips_error( class->nickname,
|
|
|
|
"%s", _( "parameters out of range" ) );
|
2014-04-01 13:27:15 +02:00
|
|
|
return( -1 );
|
2014-03-26 21:33:31 +01:00
|
|
|
}
|
|
|
|
|
2014-04-01 19:54:00 +02:00
|
|
|
hough_circle->width = statistic->in->Xsize / hough_circle->scale;
|
|
|
|
hough_circle->height = statistic->in->Ysize / hough_circle->scale;
|
|
|
|
hough_circle->bands = 1 + range / hough_circle->scale;
|
|
|
|
|
2014-03-26 21:33:31 +01:00
|
|
|
if( VIPS_OBJECT_CLASS( vips_hough_circle_parent_class )->
|
|
|
|
build( object ) )
|
|
|
|
return( -1 );
|
|
|
|
|
|
|
|
return( 0 );
|
|
|
|
}
|
|
|
|
|
|
|
|
static int
|
|
|
|
vips_hough_circle_init_accumulator( VipsHough *hough, VipsImage *accumulator )
|
|
|
|
{
|
|
|
|
VipsHoughCircle *hough_circle = (VipsHoughCircle *) hough;
|
|
|
|
|
|
|
|
vips_image_init_fields( accumulator,
|
2014-04-01 19:54:00 +02:00
|
|
|
hough_circle->width, hough_circle->height, hough_circle->bands,
|
2014-03-26 21:33:31 +01:00
|
|
|
VIPS_FORMAT_UINT, VIPS_CODING_NONE,
|
|
|
|
VIPS_INTERPRETATION_MATRIX,
|
|
|
|
1.0, 1.0 );
|
|
|
|
|
|
|
|
return( 0 );
|
|
|
|
}
|
|
|
|
|
|
|
|
static inline void
|
2014-04-01 13:27:15 +02:00
|
|
|
vips_hough_circle_vote_point( VipsImage *image, int x, int y, void *client )
|
2014-03-26 21:33:31 +01:00
|
|
|
{
|
2014-04-01 13:27:15 +02:00
|
|
|
guint *q = (guint *) VIPS_IMAGE_ADDR( image, x, y );
|
|
|
|
int r = *((int *) client);
|
|
|
|
|
|
|
|
g_assert( image->BandFmt == VIPS_FORMAT_UINT );
|
|
|
|
g_assert( x >= 0 );
|
|
|
|
g_assert( y >= 0 );
|
|
|
|
g_assert( x < image->Xsize );
|
2014-04-01 19:54:00 +02:00
|
|
|
g_assert( y < image->Ysize );
|
2014-04-01 13:27:15 +02:00
|
|
|
g_assert( r >= 0 );
|
|
|
|
g_assert( r < image->Bands );
|
|
|
|
|
|
|
|
q[r] += 1;
|
2014-03-26 21:33:31 +01:00
|
|
|
}
|
|
|
|
|
2014-04-01 13:27:15 +02:00
|
|
|
/* Vote endpoints, with clip.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
vips_hough_circle_vote_endpoints_clip( VipsImage *image,
|
|
|
|
int y, int x1, int x2, void *client )
|
2014-03-26 21:33:31 +01:00
|
|
|
{
|
2014-04-01 13:27:15 +02:00
|
|
|
if( y >= 0 &&
|
|
|
|
y < image->Ysize ) {
|
|
|
|
if( x1 >=0 &&
|
|
|
|
x1 < image->Xsize )
|
|
|
|
vips_hough_circle_vote_point( image, x1, y, client );
|
|
|
|
if( x2 >=0 &&
|
|
|
|
x2 < image->Xsize )
|
|
|
|
vips_hough_circle_vote_point( image, x2, y, client );
|
2014-03-26 21:33:31 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
2014-04-01 13:27:15 +02:00
|
|
|
/* Vote endpoints, no clip.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
vips_hough_circle_vote_endpoints_noclip( VipsImage *image,
|
|
|
|
int y, int x1, int x2, void *client )
|
2014-03-26 21:33:31 +01:00
|
|
|
{
|
2014-04-01 13:27:15 +02:00
|
|
|
vips_hough_circle_vote_point( image, x1, y, client );
|
|
|
|
vips_hough_circle_vote_point( image, x2, y, client );
|
2014-03-26 21:33:31 +01:00
|
|
|
}
|
|
|
|
|
|
|
|
/* Cast votes for all possible circles passing through x, y.
|
|
|
|
*/
|
|
|
|
static void
|
|
|
|
vips_hough_circle_vote( VipsHough *hough, VipsImage *accumulator, int x, int y )
|
|
|
|
{
|
|
|
|
VipsHoughCircle *hough_circle = (VipsHoughCircle *) hough;
|
2014-04-01 13:27:15 +02:00
|
|
|
int min_radius = hough_circle->min_radius;
|
|
|
|
int max_radius = hough_circle->max_radius;
|
|
|
|
int range = max_radius - min_radius;
|
2014-04-01 19:54:00 +02:00
|
|
|
int cx = x / hough_circle->scale;
|
|
|
|
int cy = y / hough_circle->scale;
|
2014-03-26 21:33:31 +01:00
|
|
|
|
2014-04-01 19:54:00 +02:00
|
|
|
int rb;
|
2014-03-26 21:33:31 +01:00
|
|
|
|
2014-04-01 13:27:15 +02:00
|
|
|
g_assert( range >= 0 );
|
|
|
|
|
2014-04-01 19:54:00 +02:00
|
|
|
for( rb = 0; rb < hough_circle->bands; rb++ ) {
|
|
|
|
/* r needs to be in scaled down image space.
|
|
|
|
*/
|
|
|
|
int r = rb + min_radius / hough_circle->scale;
|
2014-04-01 13:27:15 +02:00
|
|
|
|
|
|
|
VipsDrawScanline draw_scanline;
|
|
|
|
|
|
|
|
if( cx - r >= 0 &&
|
|
|
|
cx + r < accumulator->Xsize &&
|
|
|
|
cy - r >= 0 &&
|
|
|
|
cy + r < accumulator->Ysize )
|
|
|
|
draw_scanline = vips_hough_circle_vote_endpoints_noclip;
|
|
|
|
else
|
|
|
|
draw_scanline = vips_hough_circle_vote_endpoints_clip;
|
2014-03-26 21:33:31 +01:00
|
|
|
|
2014-04-01 13:27:15 +02:00
|
|
|
vips__draw_circle_direct( accumulator,
|
|
|
|
cx, cy, r, draw_scanline, &rb );
|
2014-03-26 21:33:31 +01:00
|
|
|
}
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
vips_hough_circle_class_init( VipsHoughClass *class )
|
|
|
|
{
|
|
|
|
GObjectClass *gobject_class = (GObjectClass *) class;
|
|
|
|
VipsObjectClass *object_class = (VipsObjectClass *) class;
|
|
|
|
VipsHoughClass *hclass = (VipsHoughClass *) class;
|
|
|
|
|
|
|
|
gobject_class->set_property = vips_object_set_property;
|
|
|
|
gobject_class->get_property = vips_object_get_property;
|
|
|
|
|
|
|
|
object_class->nickname = "hough_circle";
|
|
|
|
object_class->description = _( "find hough circle transform" );
|
|
|
|
object_class->build = vips_hough_circle_build;
|
|
|
|
|
|
|
|
hclass->init_accumulator = vips_hough_circle_init_accumulator;
|
|
|
|
hclass->vote = vips_hough_circle_vote;
|
|
|
|
|
2014-04-01 19:54:00 +02:00
|
|
|
VIPS_ARG_INT( class, "scale", 119,
|
|
|
|
_( "Scale" ),
|
|
|
|
_( "Scale down dimensions by this factor" ),
|
2014-03-26 21:33:31 +01:00
|
|
|
VIPS_ARGUMENT_OPTIONAL_INPUT,
|
2014-04-01 19:54:00 +02:00
|
|
|
G_STRUCT_OFFSET( VipsHoughCircle, scale ),
|
|
|
|
1, 100000, 3 );
|
2014-03-26 21:33:31 +01:00
|
|
|
|
|
|
|
VIPS_ARG_INT( class, "min_radius", 120,
|
|
|
|
_( "Min radius" ),
|
|
|
|
_( "Smallest radius to search for" ),
|
|
|
|
VIPS_ARGUMENT_OPTIONAL_INPUT,
|
|
|
|
G_STRUCT_OFFSET( VipsHoughCircle, min_radius ),
|
|
|
|
1, 100000, 10 );
|
|
|
|
|
2014-04-01 19:54:00 +02:00
|
|
|
VIPS_ARG_INT( class, "max_radius", 120,
|
2014-03-26 21:33:31 +01:00
|
|
|
_( "Max radius" ),
|
|
|
|
_( "Largest radius to search for" ),
|
|
|
|
VIPS_ARGUMENT_OPTIONAL_INPUT,
|
|
|
|
G_STRUCT_OFFSET( VipsHoughCircle, max_radius ),
|
|
|
|
1, 100000, 20 );
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
static void
|
|
|
|
vips_hough_circle_init( VipsHoughCircle *hough_circle )
|
|
|
|
{
|
2014-04-01 19:54:00 +02:00
|
|
|
hough_circle->scale = 3;
|
2014-03-26 21:33:31 +01:00
|
|
|
hough_circle->min_radius = 10;
|
|
|
|
hough_circle->max_radius = 20;
|
|
|
|
}
|
|
|
|
|
|
|
|
/**
|
|
|
|
* vips_hough_circle:
|
|
|
|
* @in: input image
|
|
|
|
* @out: output image
|
|
|
|
* @...: %NULL-terminated list of optional named arguments
|
|
|
|
*
|
|
|
|
* Optional arguments:
|
|
|
|
*
|
2014-04-01 19:54:00 +02:00
|
|
|
* @scale: scale down dimensions by this much
|
2014-03-26 21:33:31 +01:00
|
|
|
* @min_radius: smallest radius to search for
|
|
|
|
* @max_radius: largest radius to search for
|
|
|
|
*
|
|
|
|
* See also:
|
|
|
|
*
|
|
|
|
* Returns: 0 on success, -1 on error
|
|
|
|
*/
|
|
|
|
int
|
|
|
|
vips_hough_circle( VipsImage *in, VipsImage **out, ... )
|
|
|
|
{
|
|
|
|
va_list ap;
|
|
|
|
int result;
|
|
|
|
|
|
|
|
va_start( ap, out );
|
|
|
|
result = vips_call_split( "hough_circle", ap, in, out );
|
|
|
|
va_end( ap );
|
|
|
|
|
|
|
|
return( result );
|
|
|
|
}
|