source: issm/trunk/src/c/matlab/io/MatlabMatrixToDoubleMatrix.cpp@ 12330

/* \file MatlabMatrixToDoubleMatrix.cpp
 * \brief: convert a sparse or dense matlab matrix to a double* pointer
 */


#ifdef HAVE_CONFIG_H
        #include <config.h>
#else
#error "Cannot compile with HAVE_CONFIG_H symbol! run configure first!"
#endif


/*Matlab includes: */
#include "mex.h"

#include "../../shared/shared.h"

int MatlabMatrixToDoubleMatrix(double** pmatrix,int* pmatrix_rows,int* pmatrix_cols,const mxArray* mxmatrix){

        int     i,j,count,rows,cols;
        double *pmxdoublematrix = NULL;
        float  *pmxsinglematrix = NULL;

        /*output: */
        double* matrix=NULL;

        /*matlab indices: */
        mwIndex*    ir=NULL;
        mwIndex*    jc=NULL;

        /*Ok, first check if we are dealing with a sparse or full matrix: */
        if (mxIsSparse(mxmatrix)){

                /*Dealing with sparse matrix: recover size first: */
                pmxdoublematrix=(double*)mxGetPr(mxmatrix);
                rows=mxGetM(mxmatrix);
                cols=mxGetN(mxmatrix);
                
                if(rows*cols){
                        matrix=(double*)xcalloc(rows*cols,sizeof(double));

                        /*Now, get ir,jc and pr: */
                        ir=mxGetIr(mxmatrix);
                        jc=mxGetJc(mxmatrix);

                        /*Now, start inserting data into double* matrix: */
                        count=0;
                        for(i=0;i<cols;i++){
                                for(j=0;j<(jc[i+1]-jc[i]);j++){
                                        matrix[rows*ir[count]+i]=pmxdoublematrix[count];
                                        count++;
                                }
                        }
                }

        }
        else if(mxIsClass(mxmatrix,"double")){
                /*Dealing with dense matrix: recover pointer and size: */
                pmxdoublematrix=(double*)mxGetPr(mxmatrix);
                rows=mxGetM(mxmatrix);
                cols=mxGetN(mxmatrix);
                
                /*Create serial matrix: */
                if(rows*cols){
                        matrix=(double*)xcalloc(rows*cols,sizeof(double));

                        for(i=0;i<rows;i++){
                                for(j=0;j<cols;j++){
                                        matrix[cols*i+j]=(double)pmxdoublematrix[rows*j+i];
                                }
                        }
                }
        }
        else if(mxIsClass(mxmatrix,"single")){
                /*Dealing with dense matrix: recover pointer and size: */
                pmxsinglematrix=(float*)mxGetPr(mxmatrix);
                rows=mxGetM(mxmatrix);
                cols=mxGetN(mxmatrix);

                /*Create serial matrix: */
                if(rows*cols){
                        matrix=(double*)xcalloc(rows*cols,sizeof(double));

                        for(i=0;i<rows;i++){
                                for(j=0;j<cols;j++){
                                        matrix[cols*i+j]=(double)pmxsinglematrix[rows*j+i];
                                }
                        }
                }
        }
        else{
                _error_("Matlab matrix type Not implemented yet");
        }

        /*Assign output pointer: */
        *pmatrix=matrix;
        *pmatrix_rows=rows;
        *pmatrix_cols=cols;

        return 1;
}