Exemple #1
0
/* The computational routine */
void knnfield(const mwSize *dimsA, const mwSize *dimsB, nnf_data_t* videoA, nnf_data_t* videoB, NNFieldParams params, int32_t *outMatrix, nnf_data_t * distMatrix)
{
    Video<nnf_data_t> A;
    int dA[4];
    for (int i = 0; i < 4; ++i) {
        dA[i] = dimsA[i];
    }
    A.initFromMxArray(4, dA, videoA);

    Video<nnf_data_t> B;
    int dB[4];
    for (int i = 0; i < 4; ++i) {
        dB[i] = dimsB[i];
    }
    B.initFromMxArray(4, dB, videoB);

    NNField field(&A,&B, params);
    NNFieldOutput output = field.compute();
    Video<int32_t> &nnf  = output.nnf;
    Video<nnf_data_t> &cost  = output.error;

    const int32_t* pData = nnf.dataReader();
    const nnf_data_t* pCost = cost.dataReader();

    int channel_stride = nnf.voxelCount();
    int in_nn_stride   = 3*channel_stride;
    int out_nn_stride  = 3*channel_stride;


    for (unsigned int idx = 0; idx < dimsA[0]*dimsA[1]*dimsA[2]; ++idx) // each voxel
    {
        for(int k = 0; k < params.knn; ++ k) { // each NN
            for (unsigned int c = 0; c < 3; ++c) { // each warp channel
                outMatrix[idx + c*channel_stride + k*out_nn_stride] = pData[idx + c*channel_stride + k*in_nn_stride];
            }
            assert(outMatrix[idx + 0*channel_stride + k*out_nn_stride] < B.getWidth()-params.patch_size_space+1);
            assert(outMatrix[idx + 1*channel_stride + k*out_nn_stride] < B.getHeight()-params.patch_size_space+1);
            assert(outMatrix[idx + 2*channel_stride + k*out_nn_stride] < B.frameCount()-params.patch_size_time+1);
        }
    }

    if(distMatrix != nullptr) {
        for (unsigned int idx = 0; idx < dimsA[0]*dimsA[1]*dimsA[2]; ++idx) // each voxel
        {
            for(int k = 0; k < params.knn; ++ k) { // each NN
                distMatrix[idx + k*channel_stride] = pCost[idx + k*channel_stride];
            }
        }
    }
}
Exemple #2
0
void mexFunction(
    int          nlhs,
    mxArray*      plhs[],
    int          nrhs,
    const mxArray* prhs[]
) {

    char* filename;
    int first_frame, last_frame;
    bool color;
    parse_arguments(nlhs, prhs, nrhs, &filename, &first_frame, &last_frame, &color);

    Video* video = NULL;

    if (first_frame < 0) {
        std::cout << "reading all video!" << std::endl;
        video = new Video(filename);
        first_frame = 1;
        last_frame = video->frames;//frames counted by grabbing all frames in video
    } else {
        std::cout << "reading frames: " << first_frame << "-" << last_frame << std::endl;
        video = new Video(filename, last_frame - first_frame);
    }



    int n_of_dims, channels;
    mwSize* dims;
    if (!color) {
        n_of_dims = 3;
        channels = 1;
        dims = new mwSize[3];
        dims[1] = video->getWidth();
        dims[0] = video->getHeight();
        dims[2] = last_frame - first_frame + 1 ;
    } else {
        n_of_dims = 4;
        channels = 3;
        dims = new mwSize[4];
        dims[1] = video->getWidth();
        dims[0] = video->getHeight();
        dims[3] = last_frame - first_frame + 1 ;
        dims[2] = 3;
    }
    IplImage* img = 0;

    //create 3 dimensional storage and assign it to output arguments
    plhs[0] = mxCreateNumericArray(n_of_dims, dims , mxUINT8_CLASS, mxREAL);
    //get pointer to data (cast necessary since data is not double, mxGetData returns a void*)
    unsigned char* video_ptr = (unsigned char*)mxGetData(plhs[0]);


    //frame counter
    int f = 0;
    //we skip frames until f == first_frame (we want to grab from here)
    //and we stop grabbing after f == last_frame
    while ((img = video->getNextFrame()) && f < last_frame) {
        ++f;
        //check if we are in the correct range (following annotation)
        if (f >= first_frame) { // && f <= last_frame ... no need to check this after last_frame we grab the last one
            video->convertFrame(video_ptr, img, color);
            //move pointer ahead one frame
            video_ptr += video->getHeight() * video->getWidth() * channels;
        }
    }
    mxFree(filename);
    return;
}