/** @brief MEX entry point */
void
mexFunction(int nout, mxArray *out[],
int nin, const mxArray *in[])
{
enum {IN_I = 0,
IN_END } ;
enum {OUT_SEEDS = 0,
OUT_FRAMES } ;
int verbose = 0 ;
int opt ;
int next = IN_END ;
mxArray const *optarg ;
/* algorithm parameters */
double delta = -1 ;
double max_area = -1 ;
double min_area = -1 ;
double max_variation = -1 ;
double min_diversity = -1 ;
int nel ;
int ndims ;
mwSize const* dims ;
vl_mser_pix const *data ;
VL_USE_MATLAB_ENV ;
/** -----------------------------------------------------------------
** Check the arguments
** -------------------------------------------------------------- */
if (nin < 1) {
mexErrMsgTxt("At least one input argument is required.") ;
}
if (nout > 2) {
mexErrMsgTxt("Too many output arguments.");
}
if(mxGetClassID(in[IN_I]) != mxUINT8_CLASS) {
mexErrMsgTxt("I must be of class UINT8") ;
}
/* get dimensions */
nel = mxGetNumberOfElements(in[IN_I]) ;
ndims = mxGetNumberOfDimensions(in[IN_I]) ;
dims = mxGetDimensions(in[IN_I]) ;
data = mxGetData(in[IN_I]) ;
while ((opt = uNextOption(in, nin, options, &next, &optarg)) >= 0) {
switch (opt) {
case opt_verbose :
++ verbose ;
break ;
case opt_delta :
if (!uIsRealScalar(optarg) || (delta = *mxGetPr(optarg)) < 0) {
mexErrMsgTxt("'Delta' must be non-negative.") ;
}
break ;
case opt_max_area :
if (!uIsRealScalar(optarg) ||
(max_area = *mxGetPr(optarg)) < 0 ||
max_area > 1) {
mexErrMsgTxt("'MaxArea' must be in the range [0,1].") ;
}
break ;
case opt_min_area :
if (!uIsRealScalar(optarg) ||
(min_area = *mxGetPr(optarg)) < 0 ||
min_area > 1) {
mexErrMsgTxt("'MinArea' must be in the range [0,1].") ;
}
break ;
case opt_max_variation :
if (!uIsRealScalar(optarg) ||
(max_variation = *mxGetPr(optarg)) < 0) {
mexErrMsgTxt("'MaxVariation' must be non negative.") ;
}
break ;
case opt_min_diversity :
if (!uIsRealScalar(optarg) ||
(min_diversity = *mxGetPr(optarg)) < 0 ||
min_diversity > 1.0) {
mexErrMsgTxt("'MinDiversity' must be in the [0,1] range.") ;
}
break ;
default :
assert(0) ;
break ;
}
}
/* -----------------------------------------------------------------
* Run algorithm
* -------------------------------------------------------------- */
{
VlMserFilt *filt ;
vl_uint const *regions ;
float const *frames ;
int i, j, nregions, nframes, dof ;
mwSize odims [2] ;
double *pt ;
/* new filter */
{
int * vlDims = mxMalloc(sizeof(int) * ndims) ;
for (i = 0 ; i < ndims ; ++i) vlDims [i] = dims [i] ;
filt = vl_mser_new (ndims, vlDims) ;
mxFree(vlDims) ;
}
if (!filt) {
mexErrMsgTxt("Could not create an MSER filter.") ;
}
if (delta >= 0) vl_mser_set_delta (filt, (vl_mser_pix) delta) ;
if (max_area >= 0) vl_mser_set_max_area (filt, max_area) ;
if (min_area >= 0) vl_mser_set_min_area (filt, min_area) ;
if (max_variation >= 0) vl_mser_set_max_variation (filt, max_variation) ;
if (min_diversity >= 0) vl_mser_set_min_diversity (filt, min_diversity) ;
if (verbose) {
mexPrintf("mser: parameters:\n") ;
mexPrintf("mser: delta = %d\n", vl_mser_get_delta (filt)) ;
mexPrintf("mser: max_area = %g\n", vl_mser_get_max_area (filt)) ;
mexPrintf("mser: min_area = %g\n", vl_mser_get_min_area (filt)) ;
mexPrintf("mser: max_variation = %g\n", vl_mser_get_max_variation (filt)) ;
mexPrintf("mser: min_diversity = %g\n", vl_mser_get_min_diversity (filt)) ;
}
/* process the image */
vl_mser_process (filt, data) ;
/* save regions back to array */
nregions = vl_mser_get_regions_num (filt) ;
regions = vl_mser_get_regions (filt) ;
odims [0] = nregions ;
out [OUT_SEEDS] = mxCreateNumericArray (1, odims, mxDOUBLE_CLASS,mxREAL) ;
pt = mxGetPr (out [OUT_SEEDS]) ;
for (i = 0 ; i < nregions ; ++i)
pt [i] = regions [i] + 1 ;
/* optionally compute and save ellipsoids */
if (nout > 1) {
vl_mser_ell_fit (filt) ;
nframes = vl_mser_get_ell_num (filt) ;
dof = vl_mser_get_ell_dof (filt) ;
frames = vl_mser_get_ell (filt) ;
odims [0] = dof ;
odims [1] = nframes ;
out [OUT_FRAMES] = mxCreateNumericArray (2, odims, mxDOUBLE_CLASS, mxREAL) ;
pt = mxGetPr (out [OUT_FRAMES]) ;
for (i = 0 ; i < nframes ; ++i) {
for (j = 0 ; j < dof ; ++j) {
pt [i * dof + j] = frames [i * dof + j] + ((j < ndims)?1.0:0.0) ;
}
}
}
if (verbose) {
VlMserStats const* s = vl_mser_get_stats (filt) ;
int tot = s-> num_extremal ;
mexPrintf("mser: statistics:\n") ;
mexPrintf("mser: %d extremal regions of which\n", tot) ;
#define REMAIN(test,num) \
mexPrintf("mser: %5d (%7.3g %% of previous) " test "\n", \
tot-(num),100.0*(double)(tot-(num))/(tot+VL_EPSILON_D)) ; \
tot -= (num) ;
REMAIN("maximally stable,", s-> num_unstable ) ;
REMAIN("stable enough,", s-> num_abs_unstable) ;
REMAIN("small enough,", s-> num_too_big ) ;
REMAIN("big enough,", s-> num_too_small ) ;
REMAIN("diverse enough.", s-> num_duplicates ) ;
}
/* cleanup */
vl_mser_delete (filt) ;
}
}
/** @brief MSER driver entry point
**/
int
main(int argc, char **argv)
{
/* algorithm parameters */
double delta = -1 ;
double max_area = -1 ;
double min_area = -1 ;
double max_variation = -1 ;
double min_diversity = -1 ;
int bright_on_dark = 1 ;
int dark_on_bright = 1 ;
vl_bool err = VL_ERR_OK ;
char err_msg [1024] ;
int n ;
int exit_code = 0 ;
int verbose = 0 ;
VlFileMeta frm = {0, "%.frame", VL_PROT_ASCII, "", 0} ;
VlFileMeta piv = {0, "%.mser", VL_PROT_ASCII, "", 0} ;
VlFileMeta met = {0, "%.meta", VL_PROT_ASCII, "", 0} ;
#define ERRF(msg, arg) { \
err = VL_ERR_BAD_ARG ; \
snprintf(err_msg, sizeof(err_msg), msg, arg) ; \
break ; \
}
#define ERR(msg) { \
err = VL_ERR_BAD_ARG ; \
snprintf(err_msg, sizeof(err_msg), msg) ; \
break ; \
}
/* ------------------------------------------------------------------
* Parse options
* --------------------------------------------------------------- */
while (!err) {
int ch = getopt_long(argc, argv, opts, longopts, 0) ;
/* If there are no files passed as input, print the help and settings */
if (ch == -1 && argc - optind == 0)
ch = 'h';
/* end of option list? */
if (ch == -1) break;
/* process options */
switch (ch) {
/* .......................................................... */
case '?' :
ERRF("Invalid option '%s'.", argv [optind - 1]) ;
break ;
case ':' :
ERRF("Missing mandatory argument for option '%s'.",
argv [optind - 1]) ;
break ;
case 'h' :
printf (help_message, argv [0]) ;
printf ("MSERs filespec: `%s'\n", piv.pattern) ;
printf ("Frames filespec: `%s'\n", frm.pattern) ;
printf ("Meta filespec: `%s'\n", met.pattern) ;
printf ("Version: driver %s; libvl %s\n",
VL_XSTRINGIFY(VL_MSER_DRIVER_VERSION),
vl_get_version_string()) ;
exit (0) ;
break ;
case 'v' :
++ verbose ;
break ;
/* .......................................................... */
case 'd' :
n = sscanf (optarg, "%lf", &delta) ;
if (n == 0 || delta < 0)
ERRF("The argument of '%s' must be a non-negative number.",
argv [optind - 1]) ;
break ;
/* ........................................................... */
case opt_max_area :
n = sscanf (optarg, "%lf", &max_area) ;
if (n == 0 || max_area < 0 || max_area > 1)
ERR("max-area argument must be in the [0,1] range.") ;
break ;
case opt_min_area :
n = sscanf (optarg, "%lf", &min_area) ;
if (n == 0 || min_area < 0 || min_area > 1)
ERR("min-area argument must be in the [0,1] range.") ;
break ;
case opt_max_variation :
n = sscanf (optarg, "%lf", &max_variation) ;
if (n == 0 || max_variation < 0)
ERR("max-variation argument must be non-negative.") ;
break ;
case opt_min_diversity :
n = sscanf (optarg, "%lf", &min_diversity) ;
if (n == 0 || min_diversity < 0 || min_diversity > 1)
ERR("min-diversity argument must be in the [0,1] range.") ;
break ;
/* ........................................................... */
case opt_frame :
err = vl_file_meta_parse (&frm, optarg) ;
if (err)
ERRF("The arguments of '%s' is invalid.", argv [optind - 1]) ;
break ;
case opt_seed :
err = vl_file_meta_parse (&piv, optarg) ;
if (err)
ERRF("The arguments of '%s' is invalid.", argv [optind - 1]) ;
break ;
case opt_meta :
err = vl_file_meta_parse (&met, optarg) ;
if (err)
ERRF("The arguments of '%s' is invalid.", argv [optind - 1]) ;
if (met.protocol != VL_PROT_ASCII)
ERR("meta file supports only ASCII protocol") ;
break ;
case opt_bright :
n = sscanf (optarg, "%d", &bright_on_dark) ;
if (n == 0 || (bright_on_dark != 0 && bright_on_dark != 1))
ERR("bright_on_dark must be 0 or 1.") ;
break ;
case opt_dark :
n = sscanf (optarg, "%d", &dark_on_bright) ;
if (n == 0 || (dark_on_bright != 0 && dark_on_bright != 1))
ERR("dark_on_bright must be 0 or 1.") ;
break ;
/* .......................................................... */
case 0 :
default :
abort() ;
}
}
/* check for parsing errors */
if (err) {
fprintf(stderr, "%s: error: %s (%d)\n",
argv [0],
err_msg, err) ;
exit (1) ;
}
/* parse other arguments (filenames) */
argc -= optind ;
argv += optind ;
/* make sure at least one file */
if (piv.active == 0 && frm.active == 0) {
frm.active = 1 ;
}
if (verbose > 1) {
printf("mser: frames output\n") ;
printf("mser: active %d\n", frm.active ) ;
printf("mser: pattern %s\n", frm.pattern) ;
printf("mser: protocol %s\n", vl_string_protocol_name (frm.protocol)) ;
printf("mser: seeds output\n") ;
printf("mser: active %d\n", piv.active ) ;
printf("mser: pattern %s\n", piv.pattern) ;
printf("mser: protocol %s\n", vl_string_protocol_name (piv.protocol)) ;
printf("mser: meta output\n") ;
printf("mser: active %d\n", met.active ) ;
printf("mser: pattern %s\n", met.pattern) ;
printf("mser: protocol %s\n", vl_string_protocol_name (met.protocol)) ;
}
/* ------------------------------------------------------------------
* Process one image per time
* --------------------------------------------------------------- */
while (argc--) {
char basename [1024] ;
char const *name = *argv++ ;
VlMserFilt *filt = 0 ;
VlMserFilt *filtinv = 0 ;
vl_uint8 *data = 0 ;
vl_uint8 *datainv = 0 ;
VlPgmImage pim ;
vl_uint const *regions ;
vl_uint const *regionsinv ;
float const *frames ;
float const *framesinv ;
enum {ndims = 2} ;
int dims [ndims] ;
int nregions = 0, nregionsinv = 0, nframes = 0, nframesinv =0;
int i, j, dof ;
vl_size q ;
FILE *in = 0 ;
/* Open files ------------------------------------------------ */
/* get basenmae from filename */
q = vl_string_basename (basename, sizeof(basename), name, 1) ;
err = (q >= sizeof(basename)) ;
if (err) {
snprintf(err_msg, sizeof(err_msg),
"Basename of '%s' is too long", name);
err = VL_ERR_OVERFLOW ;
goto done ;
}
if (verbose) {
printf("mser: processing '%s'\n", name) ;
}
if (verbose > 1) {
printf("mser: basename is '%s'\n", basename) ;
}
#define WERR(name) \
if (err == VL_ERR_OVERFLOW) { \
snprintf(err_msg, sizeof(err_msg), \
"Output file name too long.") ; \
goto done ; \
} else if (err) { \
snprintf(err_msg, sizeof(err_msg), \
"Could not open '%s' for writing.", name) ; \
goto done ; \
}
/* open input file */
in = fopen (name, "rb") ;
if (!in) {
err = VL_ERR_IO ;
snprintf(err_msg, sizeof(err_msg),
"Could not open '%s' for reading.", name) ;
goto done ;
}
/* open output files */
err = vl_file_meta_open (&piv, basename, "w") ; WERR(piv.name) ;
err = vl_file_meta_open (&frm, basename, "w") ; WERR(frm.name) ;
err = vl_file_meta_open (&met, basename, "w") ; WERR(met.name) ;
if (verbose > 1) {
if (piv.active) printf("mser: writing seeds to '%s'\n", piv.name);
if (frm.active) printf("mser: writing frames to '%s'\n", frm.name);
if (met.active) printf("mser: writing meta to '%s'\n", met.name);
}
/* Read image data -------------------------------------------- */
/* read source image header */
err = vl_pgm_extract_head (in, &pim) ;
if (err) {
err = VL_ERR_IO ;
snprintf(err_msg, sizeof(err_msg),
"PGM header corrputed.") ;
goto done ;
}
if (verbose) {
printf("mser: image is %d by %d pixels\n",
pim. width,
pim. height) ;
}
/* allocate buffer */
data = malloc(vl_pgm_get_npixels (&pim) *
vl_pgm_get_bpp (&pim)) ;
if (!data) {
err = VL_ERR_ALLOC ;
snprintf(err_msg, sizeof(err_msg),
"Could not allocate enough memory.") ;
goto done ;
}
/* read PGM */
err = vl_pgm_extract_data (in, &pim, data) ;
if (err) {
snprintf(err_msg, sizeof(err_msg),
"PGM body corrputed.") ;
goto done ;
}
/* Process data ---------------------------------------------- */
dims[0] = pim.width ;
dims[1] = pim.height ;
filt = vl_mser_new (ndims, dims) ;
filtinv = vl_mser_new (ndims, dims) ;
if (!filt || !filtinv) {
snprintf(err_msg, sizeof(err_msg),
"Could not create an MSER filter.") ;
goto done ;
}
if (delta >= 0) vl_mser_set_delta (filt, (vl_mser_pix) delta) ;
if (max_area >= 0) vl_mser_set_max_area (filt, max_area) ;
if (min_area >= 0) vl_mser_set_min_area (filt, min_area) ;
if (max_variation >= 0) vl_mser_set_max_variation (filt, max_variation) ;
if (min_diversity >= 0) vl_mser_set_min_diversity (filt, min_diversity) ;
if (delta >= 0) vl_mser_set_delta (filtinv, (vl_mser_pix) delta) ;
if (max_area >= 0) vl_mser_set_max_area (filtinv, max_area) ;
if (min_area >= 0) vl_mser_set_min_area (filtinv, min_area) ;
if (max_variation >= 0) vl_mser_set_max_variation (filtinv, max_variation) ;
if (min_diversity >= 0) vl_mser_set_min_diversity (filtinv, min_diversity) ;
if (verbose) {
printf("mser: parameters:\n") ;
printf("mser: delta = %d\n", vl_mser_get_delta (filt)) ;
printf("mser: max_area = %g\n", vl_mser_get_max_area (filt)) ;
printf("mser: min_area = %g\n", vl_mser_get_min_area (filt)) ;
printf("mser: max_variation = %g\n", vl_mser_get_max_variation (filt)) ;
printf("mser: min_diversity = %g\n", vl_mser_get_min_diversity (filt)) ;
}
if (dark_on_bright)
{
vl_mser_process (filt, (vl_mser_pix*) data) ;
/* Save result ----------------------------------------------- */
nregions = vl_mser_get_regions_num (filt) ;
regions = vl_mser_get_regions (filt) ;
if (piv.active) {
for (i = 0 ; i < nregions ; ++i) {
fprintf(piv.file, "%d ", regions [i]) ;
}
}
if (frm.active) {
vl_mser_ell_fit (filt) ;
nframes = vl_mser_get_ell_num (filt) ;
dof = vl_mser_get_ell_dof (filt) ;
frames = vl_mser_get_ell (filt) ;
for (i = 0 ; i < nframes ; ++i) {
for (j = 0 ; j < dof ; ++j) {
fprintf(frm.file, "%f ", *frames++) ;
}
fprintf(frm.file, "\n") ;
}
}
}
if (bright_on_dark)
{
/* allocate buffer */
datainv = malloc(vl_pgm_get_npixels (&pim) *
vl_pgm_get_bpp (&pim)) ;
for (i = 0; i < vl_pgm_get_npixels (&pim); i++) {
datainv[i] = ~data[i]; /* 255 - data[i] */
}
if (!datainv) {
err = VL_ERR_ALLOC ;
snprintf(err_msg, sizeof(err_msg),
"Could not allocate enough memory.") ;
goto done ;
}
vl_mser_process (filtinv, (vl_mser_pix*) datainv) ;
/* Save result ----------------------------------------------- */
nregionsinv = vl_mser_get_regions_num (filtinv) ;
regionsinv = vl_mser_get_regions (filtinv) ;
if (piv.active) {
for (i = 0 ; i < nregionsinv ; ++i) {
fprintf(piv.file, "%d ", -regionsinv [i]) ;
}
}
if (frm.active) {
vl_mser_ell_fit (filtinv) ;
nframesinv = vl_mser_get_ell_num (filtinv) ;
dof = vl_mser_get_ell_dof (filtinv) ;
framesinv = vl_mser_get_ell (filtinv) ;
for (i = 0 ; i < nframesinv ; ++i) {
for (j = 0 ; j < dof ; ++j) {
fprintf(frm.file, "%f ", *framesinv++) ;
}
fprintf(frm.file, "\n") ;
}
}
}
if (met.active) {
fprintf(met.file, "<mser\n") ;
fprintf(met.file, " input = '%s'\n", name) ;
if (piv.active) {
fprintf(met.file, " seeds = '%s'\n", piv.name) ;
}
if (frm.active) {
fprintf(met.file," frames = '%s'\n", frm.name) ;
}
fprintf(met.file, ">\n") ;
}
/* Next guy ----------------------------------------------- */
done :
/* release filter */
if (filt) {
vl_mser_delete (filt) ;
filt = 0 ;
}
if (filtinv) {
vl_mser_delete (filtinv) ;
filtinv = 0 ;
}
/* release image data */
if (data) {
free (data) ;
data = 0 ;
}
if (datainv) {
free (datainv) ;
datainv = 0 ;
}
/* close files */
if (in) {
fclose (in) ;
in = 0 ;
}
vl_file_meta_close (&frm) ;
vl_file_meta_close (&piv) ;
vl_file_meta_close (&met) ;
/* if bad print error message */
if (err) {
fprintf
(stderr,
"mser: err: %s (%d)\n",
err_msg,
err) ;
exit_code = 1 ;
}
}
/* quit */
return exit_code ;
}
/**
* @brief Python entry point for MSER
*
* @param pyArray
* @param delta
* @param max_area
* @param min_area
* @param max_variation
* @param min_diversity
* @return
*/
PyObject * vl_mser_python(
PyArrayObject & pyArray,
double delta,
double max_area,
double min_area,
double max_variation,
double min_diversity)
{
// check data type
assert(pyArray.descr->type_num == PyArray_UBYTE);
assert(pyArray.flags & NPY_FORTRAN);
unsigned char * data = (unsigned char *) pyArray.data;
unsigned int width = pyArray.dimensions[0];
unsigned int height = pyArray.dimensions[1];
VlMserFilt *filt = 0;
int i, j, dof, nframes, q;
vl_uint const *regions;
float const *frames;
npy_intp nregions;
// image dims
const int ndims = 2;
int dims[ndims];
dims[0] = width;
dims[1] = height;
// mser filter
filt = vl_mser_new(ndims, dims);
// set parameters
if (delta >= 0)
vl_mser_set_delta(filt, (vl_mser_pix) delta);
if (max_area >= 0)
vl_mser_set_max_area(filt, max_area);
if (min_area >= 0)
vl_mser_set_min_area(filt, min_area);
if (max_variation >= 0)
vl_mser_set_max_variation(filt, max_variation);
if (min_diversity >= 0)
vl_mser_set_min_diversity(filt, min_diversity);
// do mser computation
vl_mser_process(filt, (vl_mser_pix*) data);
// fit ellipses
vl_mser_ell_fit(filt);
// get results
nregions = (npy_intp) vl_mser_get_regions_num(filt);
regions = vl_mser_get_regions(filt);
nframes = vl_mser_get_ell_num(filt);
dof = vl_mser_get_ell_dof(filt);
frames = vl_mser_get_ell(filt);
// convert results to PyArrayObjects
npy_intp odims[2];
odims[0] = dof;
odims[1] = nframes;
// allocate pyarray objects
PyArrayObject * _regions = (PyArrayObject*) PyArray_SimpleNew(
1, (npy_intp*) &nregions, PyArray_DOUBLE);
PyArrayObject * _frames = (PyArrayObject*) PyArray_NewFromDescr(
&PyArray_Type, PyArray_DescrFromType(PyArray_DOUBLE),
2, odims, NULL, NULL, NPY_F_CONTIGUOUS, NULL);
// check if valid pointers
assert(_regions);
assert(_frames);
// fill pyarray objects
double * _regions_buf = (double *) _regions->data;
for (i = 0; i < nregions; ++i) {
_regions_buf[i] = regions[i];
}
double * _frames_buf = (double *) _frames->data;
for (j = 0; j < dof; ++j) {
for (i = 0; i < nframes; ++i) {
_frames_buf[i * dof + j] = frames[i * dof + j]; //+ ((j < ndims) ? 1.0 : 0.0);
}
}
// cleanup
vl_mser_delete (filt) ;
// construct tuple to return both results: (regions, frames)
PyObject * tuple = PyTuple_New(2);
PyTuple_SetItem(tuple, 0, PyArray_Return(_regions));
PyTuple_SetItem(tuple, 1, PyArray_Return(_frames));
return tuple;
}
