Доступ к разреженной матрице, созданной в MEX из MATLAB - PullRequest
1 голос
/ 27 мая 2011

Я создал разреженную матрицу в MEX-файле, следуя примеру, показанному в здесь . Теперь, как мне получить доступ к этой матрице из MATLAB в целом.

#define NZMAX 4
#define ROWS  4
#define COLS  2

int            rows=ROWS, cols=COLS;
mxArray       *ptr_array; /* Pointer to created sparse array. */
static double  static_pr_data[NZMAX] = {5.8, 6.2, 5.9, 6.1};
static int     static_ir_data[NZMAX] = {0, 2, 1, 3};
static int     static_jc_data[COLS+1] = {0, 2, 4};
double        *start_of_pr;
int           *start_of_ir, *start_of_jc;
mxArray       *array_ptr;

/* Create a sparse array. */    
array_ptr = mxCreateSparse(rows, cols, NZMAX, mxREAL); 

/* Place pr data into the newly created sparse array. */ 
start_of_pr = (double *)mxGetPr(array_ptr); 
memcpy(start_of_pr, static_pr_data, NZMAX*sizeof(double));

/* Place ir data into the newly created sparse array. */ 
start_of_ir = (int *)mxGetIr(array_ptr); 
memcpy(start_of_ir, static_ir_data, NZMAX*sizeof(int));

/* Place jc data into the newly created sparse array. */ 
start_of_jc = (int *)mxGetJc(array_ptr); 
memcpy(start_of_jc, static_jc_data, NZMAX*sizeof(int));

/* ... Use the sparse array in some fashion. */
/* When finished with the mxArray, deallocate it. */
mxDestroyArray(array_ptr);

Также при сохранении значений в static_pr_data, ic_data и jc_data необходимо ли хранить значения в основном формате столбца? Можно ли сохранить основной формат строки (поскольку это ускорит мои вычисления)?

Ответы [ 2 ]

1 голос
/ 29 мая 2011

В примере, на который вы ссылаетесь, последний оператор -

mxDestroyArray(array_ptr);

Вместо того, чтобы уничтожать массив, вам нужно вернуть его в качестве вывода вашей MEX-функции.Ваш исходный код MEX-функции C / C ++ должен иметь функцию с именем mexFunction (точка входа для MEX-функции), которая выглядит следующим образом:

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

Для возврата вывода из MEX-функцииназначьте его в массив plhs, например:

plhs[0] = array_ptr; // instead of mxDestroyArray(array_ptr);

Скомпилируйте код в MEX-функцию (назовем его sparsetest).Вызовите его из MATLAB следующим образом:

>> output = sparsetest;

Теперь output - это переменная MATLAB, содержащая разреженную матрицу, созданную вами в вашей MEX-функции.

Что касается хранения данных в строке -основной формат, это невозможно.MATLAB обрабатывает только разреженные матрицы по столбцам.

0 голосов
/ 13 ноября 2014

Подсказка для получения матричного поля из структуры matlab с использованием engGetVariable()

У меня есть структура с полем, которое является матрицей.В C ++ соответствующая структура является двойной **, например.Попытка получить доступ к полю с помощью engGetVariable (engine, MyStruct.theField) не удалась.Я использую переменную temp для хранения MyStruct.theField, а затем использую engGetVariable (engine, tempVar), и код для получения поля матрицы из структуры выглядит так:

// Fetch struct field using a temp variable
std::string tempName = std::string(field_name) + "_temp";
std::string fetchField = tempName + " = " + std::string(struct_name) 
        + "." + std::string(field_name) + "; ";
matlabExecute(ep, fetchField);
mxArray *matlabArray = engGetVariable(ep, tempName.c_str());

// Get variable elements
const int count = mxGetNumberOfElements(matlabArray);
T *data = (T*) mxGetData(matlabArray);
for (int i = 0; i < count; i++)
    vector[i] = _isnan(data[i]) ? (T) (int) -9999 : (T) data[i];

// Clear temp variable
std::string clearTempVar = "clear " + tempName + "; ";
matlabExecute(ep, clearTempVar);

// Destroy mx object
mxDestroyArray(matlabArray);
...