divide_by_table_ndxsd.c

来自「麻省理工学院的人工智能工具箱,很珍贵,希望对大家有用!」· C语言 代码 · 共 32 行

/* C mex version for divide_by_table_ndx.m in potential/Tables directory  */

#include "mex.h"

void mexFunction(int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[])
{
  mxArray *ptemp;
  double  *Tbig, *Tsmall, value;
  int     i, j, k, S, D;
  int     *small_ndx, *diff_ndx;

  plhs[0] = mxDuplicateArray(prhs[0]);
  Tbig = mxGetPr(plhs[0]); 
  Tsmall = mxGetPr(prhs[1]); 

  ptemp = mxGetField(prhs[2], 0, "small");
  small_ndx = mxGetData(ptemp);
  S = mxGetNumberOfElements(ptemp);

  ptemp = mxGetField(prhs[2], 0, "diff");
  diff_ndx = mxGetData(ptemp);
  D = mxGetNumberOfElements(ptemp);

  for (i = 0; i < S; i++) {
    for (j = 0; j < D; j++) {
      k = small_ndx[i] + diff_ndx[j];
      value = Tsmall[i];
      if (value != 0) Tbig[k] /= value;
    }
  }
}