编辑:

在编辑答案之前,我使用了一个具有意外行为的未记录的函数,并且它的签名在不同版本的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对象,该函数返回零.

Logo

魔乐社区(Modelers.cn) 是一个中立、公益的人工智能社区,提供人工智能工具、模型、数据的托管、展示与应用协同服务,为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作,由全产业链共同建设、共同运营、共同享有,推动国产AI生态繁荣发展。

更多推荐