Subversion Repositories seema-scanner

#include "mex.h"

#include "class_handle.hpp"

// Output printf statements in the Matlab workspace

#define printf mexPrintf

// The class that we are interfacing to

#include "Camera.h"

using namespace std;

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])

{	

    // Get the command string

    char cmd[64];

	if (nrhs < 1 || mxGetString(prhs[0], cmd, sizeof(cmd)))

		mexErrMsgTxt("First input should be a command string less than 64 characters long.");

    // Static methods

    // InterfaceCameraList

    if (!strcmp("GetInterfaceCameraList", cmd)) {

        // Call the method

        vector< vector<CameraInfo> > interfaceCameraList;

        interfaceCameraList = Camera::GetInterfaceCameraList();

        unsigned int nCameras = 0;

        for(unsigned int i=0; i<interfaceCameraList.size(); i++)

            nCameras += interfaceCameraList[i].size();

        const char* fieldNames[] = {"interfaceNum", "cameraNum", "vendor","model"};

        plhs[0] = mxCreateStructMatrix(1, nCameras, 4, fieldNames);

        unsigned int iCamera = 0;

        for (unsigned int i=0; i<interfaceCameraList.size(); i++) {

            for (unsigned int j=0; j<interfaceCameraList[i].size(); j++) {

                mxSetFieldByNumber(plhs[0], iCamera, 0, mxCreateDoubleScalar(i));

                mxSetFieldByNumber(plhs[0], iCamera, 1, mxCreateDoubleScalar(j));

                mxSetFieldByNumber(plhs[0], iCamera, 2, mxCreateString(interfaceCameraList[i][j].vendor.c_str()));

                mxSetFieldByNumber(plhs[0], iCamera, 3, mxCreateString(interfaceCameraList[i][j].model.c_str()));

                iCamera++;

            }

        }

        return;

    }

    // New Camera

    if (!strcmp("NewCamera", cmd)) {

        // Check parameters

        if (nlhs != 1)

            mexErrMsgTxt("newCamera: One output expected.");

        if (nrhs < 3)

            mexErrMsgTxt("newCamera: Expected interface and screen number argument.");

        unsigned int interfaceNum = (unsigned int)mxGetScalar(prhs[1]);

        unsigned int camNum = (unsigned int)mxGetScalar(prhs[2]);

        // Return a handle to a new C++ instance

        plhs[0] = convertPtr2Mat<Camera>(Camera::NewCamera(interfaceNum, camNum));

        return;

    }

    // Class methods

    // Check if there is a second input, which should be the class instance handle

    if (nrhs < 2)

		mexErrMsgTxt("Second input should be a class instance handle.");

    // Delete

    if (!strcmp("delete", cmd)) {

        // Destroy the C++ object

        destroyObject<Camera>(prhs[1]);

        // Warn if other commands were ignored

        if (nlhs != 0 || nrhs != 2)

            mexWarnMsgTxt("Delete: Unexpected arguments ignored.");

        return;

    }

    // Get the class instance pointer from the second input

    Camera *Camera_instance = convertMat2Ptr<Camera>(prhs[1]);

    // Call the various class methods

    // getSingleFrame

    if (!strcmp("getSingleFrame", cmd)) {

        // Check parameters

        if (nlhs < 1 || nrhs > 2)

            mexErrMsgTxt("getSingleFrame: Unexpected arguments.");

        // Call the method

        CameraFrame frame = Camera_instance->getSingleFrame();

        plhs[0] = mxCreateNumericMatrix(frame.width, frame.height, mxUINT8_CLASS, 0);

        memcpy(mxGetData(plhs[0]), frame.memory, frame.height*frame.width);

        //Camera_instance->unlockFrame();

        return;

    }

//     // SetGamma

//     if (!strcmp("SetGamma", cmd)) {

//         // Check parameters

//         if (nlhs < 0 || nrhs < 2)

//             mexErrMsgTxt("SetGamma: Unexpected arguments.");

//         // Call the method

//         float gamma = (float)mxGetScalar(prhs[2]);

//         nProjector_instance->SetGamma(gamma);

//         return;

//     }

//    // Test

//    if (!strcmp("test", cmd)) {

//        // Check parameters

//        if (nlhs < 0 || nrhs < 2)

//            mexErrMsgTxt("Test: Unexpected arguments.");

//        // Call the method

//        PatternProjector_instance->test();

//        return;

//    }

    // Got here, so command not recognized

    mexErrMsgTxt("Command not recognized.");

}