#include "lsl_common.h"
#include <string.h>

/* function [SampleData,Timestamp] = lsl_pull_sample_d(LibHandle,Inlet,NumChannels,Timeout) */

void mexFunction( int nlhs, mxArray *plhs[], 
		  int nrhs, const mxArray*prhs[] ) 
{
    /* handle of the desired field */
    mxArray *field;
    /* temp pointer */
    uintptr_t *pTmp;
    /* function handle */
    lsl_pull_sample_d_t func;
    
    /* input/output variables */
    uintptr_t in;
    int numchannels;
    double timeout;
    int errcode;
    double buffer[8192];
    double timestamp;
    
    if (nrhs != 4)
        mexErrMsgTxt("4 input argument(s) required."); 
    if (nlhs < 1)
        mexErrMsgTxt("At least 1 output argument(s) required."); 
    if (nlhs > 2)
        mexErrMsgTxt("At most 2 output argument(s) required."); 
    
    /* get function handle */
    field = mxGetField(prhs[0],0,"lsl_pull_sample_d");
    if (!field)
        mexErrMsgTxt("The field does not seem to exist.");
    pTmp = (uintptr_t*)mxGetData(field);
    if (!pTmp)
        mexErrMsgTxt("The field seems to be empty.");
    func = (lsl_pull_sample_d_t*)*pTmp;
    
    /* get additional inputs */
    in = *(uintptr_t*)mxGetData(prhs[1]);
    if (mxGetClassID(prhs[2]) != mxDOUBLE_CLASS)
        mexErrMsgTxt("The numchannels argument must be passed as a double.");
    numchannels = (int)*(double*)mxGetData(prhs[2]);
    if (mxGetClassID(prhs[3]) != mxDOUBLE_CLASS)
        mexErrMsgTxt("The timeout argument must be passed as a double.");
    timeout = *(double*)mxGetData(prhs[3]);    
    
    /* invoke & return */
    timestamp = func(in,buffer,numchannels,timeout,&errcode);
    if (errcode) {
        if (errcode == lsl_timeout_error)
            mexErrMsgIdAndTxt("lsl:timeout_error","The operation timed out.");
        if (errcode == lsl_lost_error)
            mexErrMsgIdAndTxt("lsl:lost_error","The stream has been lost.");
        if (errcode == lsl_internal_error)
            mexErrMsgIdAndTxt("lsl:internal_error","An internal error occurred.");
        mexErrMsgIdAndTxt("lsl:unknown_error","An unknown error occurred.");
    }
    if (timestamp) {
        /* got a sample */
        plhs[0] = mxCreateNumericMatrix(1,numchannels,mxDOUBLE_CLASS,mxREAL);
        /* copy over into the result vector */
        memcpy(mxGetData(plhs[0]),buffer,sizeof(buffer[0])*numchannels);        
        if (nlhs == 2) {
            /* also assign the time stamp */
            plhs[1] = mxCreateNumericMatrix(1,1,mxDOUBLE_CLASS,mxREAL);
            *(double*)mxGetData(plhs[1]) = timestamp;
        }
    } else {
        /* got no sample: return empty arrays */
        plhs[0] = mxCreateNumericMatrix(0,0,mxDOUBLE_CLASS,mxREAL);
        if (nlhs == 2)
            plhs[1] = mxCreateNumericMatrix(0,0,mxDOUBLE_CLASS,mxREAL);
    }
}