vrd_sse/VRD.cpp at master · sagargp/vrd_sse · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/*=================================================================
 * VRD.cpp - Compute Variance Ridge Detector on LAB images
 *
 * Input:   LAB-uint8 image, radius of edge detection
 * Output:  2D array of edge magnitudes
 *
 * Copyright 2012 Randolph Voorhies
 *
 *=================================================================*/
#include "mex.h"
#include "vrd_sse.h"
#include <stdint.h>

void mexFunction( int nlhs, mxArray *plhs[],
                  int nrhs, const mxArray *prhs[])
{
  if(nrhs != 2)
    mexErrMsgTxt("Inputs must be [imagearray, radius]");
  if(! (nlhs == 1 || nlhs == 3) )
    mexErrMsgTxt("VRD provides either one output ([magnitude], or three outputs [magnitude, horizontal, vertical]");

  if(!(mxIsClass(prhs[0],"uint8") && mxGetNumberOfDimensions(prhs[0]) == 3))
  {
    char buffer[512];
    sprintf(buffer, "First argument must be a LAB-uint8 image. Type is [%s], nDims are [%d]",
        mxGetClassName(prhs[0]), mxGetNumberOfDimensions(prhs[0]));
    mexErrMsgTxt(buffer);
  }

  // Get the input dimensions
  mwSize const * dims = mxGetDimensions(prhs[0]);

  // Get the desired radius
  int radius  = mxGetScalar(prhs[1]);

  // Get the input array
  uint8_t * mxInput = static_cast<uint8_t*>(mxGetData(prhs[0]));

  // Pack the input pixels into a LABx configuration
  float * const img = static_cast<float*>(calloc(dims[0]*dims[1]*4, sizeof(float)));
  int const npixels = dims[0] * dims[1];
  for(int i=0; i<3; ++i)
  {
    float *imgit = img + i;
    for(int j=0; j<npixels; ++j)
    {
      *imgit = *mxInput;
      mxInput += 1;
      imgit   += 4;
    }
  }

  // Create the output array
  if(nlhs == 1)
  {
    mxArray *mxOutput = mxCreateNumericArray(2, &dims[0], mxSINGLE_CLASS, mxComplexity(0));
    vrd_sse(img, dims[0], dims[1], radius, static_cast<float*>(mxGetData(mxOutput)));
    plhs[0] = mxOutput;
  }
  else
  {
    mxArray *mxOutput = mxCreateNumericArray(2, &dims[0], mxSINGLE_CLASS, mxComplexity(0));
    mxArray *mxGradV  = mxCreateNumericArray(2, &dims[0], mxSINGLE_CLASS, mxComplexity(0));
    mxArray *mxGradH  = mxCreateNumericArray(2, &dims[0], mxSINGLE_CLASS, mxComplexity(0));

    // Do the VRDing
    vrd_sse(img, dims[0], dims[1], radius, static_cast<float*>(mxGetData(mxOutput)), static_cast<float*>(mxGetData(mxGradV)), static_cast<float*>(mxGetData(mxGradH)));
    plhs[0] = mxOutput;
    plhs[1] = mxGradV;
    plhs[2] = mxGradH;
  }


  free(img);
}