编辑:
在编辑答案之前,我使用了一个具有意外行为的未记录的函数,并且它的签名在不同版本的MATLAB之间不稳定所以在这里我提供了@CrisLuengo答案的扩展版本.
我们可以使用哈希映射在递归函数check_shared中存储数据元素的唯一地址及其关联的mxArrays,并获取数据大小.请注意,在这里我们可以检查单元格中的共享状态,并且我们无法检查单元格之外的元素以及与单元格元素具有相同的地址.*
#include "mex.h"
#include
typedef std::unordered_map TableType;
TableType check_shared(const mxArray* arr, TableType table = TableType())
{
switch (mxGetClassID(arr)) {
case mxCELL_CLASS:
for(int i = 0; i < mxGetNumberOfElements (arr); i++) {
table = check_shared(mxGetCell (arr,i), std::move(table));
}
break;
case mxSTRUCT_CLASS:
for (int i = 0; i < mxGetNumberOfFields (arr); i++) {
for (int j = 0; j < mxGetNumberOfElements (arr); j++) {
table = check_shared(mxGetFieldByNumber (arr, j, i), std::move(table));
}
}
break;
case mxVOID_CLASS:
case mxFUNCTION_CLASS:
case mxUNKNOWN_CLASS:
return table;
}
if (!mxIsEmpty (arr)) {
void* data = mxGetData(arr);
table[data] = arr;
}
return table;
}
uint64_t actual_size(const TableType& table)
{
uint64_t sz = 0;
for (const auto& entry : table) {
const mxArray * arr = entry.second;
sz += mxGetElementSize (arr) * mxGetNumberOfElements (arr);
}
return sz;
}
void mexFunction(int nlhs, mxArray *plhs[],
int nrhs, const mxArray *prhs[])
{
TableType table = check_shared(prhs[0]);
plhs[0] = mxCreateNumericMatrix(1,1, mxUINT64_CLASS, mxREAL );
uint64_t* result = static_cast(mxGetData (plhs[0]));
result[0] = actual_size(table);
}
(*)支持基本数据类型,如单元格,结构和数字数组.对于未知数据结构和classdef对象,该函数返回零.