divide_by_table_ndxb.c

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

/* 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[]){
  double *bigT, *smallT, value;
  int    *ndx;
  int    i, j, N;  

  plhs[0] = mxDuplicateArray(prhs[0]);
  bigT = mxGetPr(plhs[0]);
  smallT = mxGetPr(prhs[1]); 
  ndx = mxGetData(prhs[2]);
  N = mxGetNumberOfElements(prhs[0]);
  j = mxGetNumberOfElements(prhs[2]);

  if(j == 1){
	value = smallT[*ndx];
	if(value == 0.0) value = 1.0;
	for (i = 0; i < N; i++) {
      *bigT++ /= value;
	}
  }
  else{
    for (i = 0; i < N; i++) {
      value = smallT[*ndx++];
	  if(value !=0) *bigT++ /= value;
	}
  }
}