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[]){

    //mexPrintf("nlhs: %d , nrhs: %d\n", nlhs, nrhs);

    // 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 camera 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, triggerModeSoftware));

        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/missing arguments ignored.");

        return;

    }

    // Get the class instance pointer from the second input

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

	// Call the various class methods

    // startCapture

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

        // Check parameters

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

            mexErrMsgTxt("startCapture: Unexpected/missing arguments.");

        // Call the method

        Camera_instance->startCapture();

        return;

    }

	// Call the various class methods

    // stopCapture

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

        // Check parameters

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

            mexErrMsgTxt("stopCapture: Unexpected/missing arguments.");

        // Call the method

        Camera_instance->stopCapture();

        return;

    }

	// trigger

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

        // Check parameters

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

            mexErrMsgTxt("trigger: Unexpected/missing arguments.");

        // Call the method

        Camera_instance->trigger();

        return;

    }

    // Call the various class methods

    // getFrame

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

        // Check parameters

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

            mexErrMsgTxt("getFrame: Unexpected/missing arguments.");

        // Call the method

        CameraFrame frame = Camera_instance->getFrame();

        // receive frame data in permuted dimensional order

        int dims[] = {frame.channels, frame.width, frame.height};

        plhs[0] = mxCreateNumericArray(3, dims, mxUINT8_CLASS, mxREAL);

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

        return;

    }

    // setCameraSettings

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

        // Check parameters

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

            mexErrMsgTxt("setCameraSettings: Unexpected/missing arguments.");

        // Call the method

		double *gain = (double*)mxGetData(prhs[2]);

		double *shutter = (double*)mxGetData(prhs[3]);

		CameraSettings settings;

        settings.gain = *gain;

        settings.shutter = *shutter;

        Camera_instance->setCameraSettings(settings);

        return;

    }

    // Got here, so command not recognized

    mexErrMsgTxt("Command not recognized.");

}