开通博客赚积分
发布资源赚积分
分类

源码开发语言/平台
当前位置：首页 > 源码/资料 > 图形图像 > 图形图像处理 > 查看源码
pdfbgen_tdata.cc
资源名称：contourletHMT.rar　[点击查看]
上传用户：l56789
上传日期：2022-02-25
资源大小：2422k
文件大小：2k
源码类别：

图形图像处理

开发平台：

Matlab

pdfbgen_tdata.cc：源码内容
  1. //-----------------------------------------------------------------------------
  2. // pdfbgen_tdata.cc  - Generate data from a Tying HMT Model for PDFB
  3. //
  4. // input arguments: 
  5. // nObs - number of observations (trees) to be generated randomly (nRoots)
  6. // modelfile - input file containing HMT parameters used in random generation
  7. //             (for modelfile format, see thmt.hh)
  8. // datafile - output file containing generated data in tree structures
  9. //            (for datafile format, see tree.hh)
  10. //-----------------------------------------------------------------------------
  12. #include <stdlib.h>
  13. #include <iostream>
  14. #include "pdfbthmt.hh"
  17. tree<double>* pdfbgen_tdata(const mxArray* model, int ntrees, int* levndir, 
  18.     int nl)
  19. {
  20.    tree<double>* data = new tree<double>(ntrees, 4, nl, 0);
  22.    THMT thmt(model, levndir);
  24.    // Generate data
  25.    thmt.generate_data(*data, ntrees);
  27.    return data;
  28. }
  30. //---------------------------------------------------------------------------  
  31. void mexFunction( int nlhs, mxArray *plhs[], 
  32.   int nrhs, const mxArray *prhs[] )     
  33. {
  34.     double *data, *templevndir; 
  35.     int nObs, nl, i, n, *levndir, *intptr, temp;
  36.     tree<double>* dataTree;
  37.     const mxArray *constdatacell;
  38.     mxArray *initcell, *datacell;
  39.     
  40.     /* Check for proper number of arguments */
  41.     
  42.     if (nrhs != 4){ 
  43. mexErrMsgTxt("Wrong number of arguments."); 
  44.     } else if (nlhs > 1) {
  45. mexErrMsgTxt("Too many output arguments."); 
  46.     } 
  48.     nl = (int)mxGetScalar(prhs[1]);
  49.     nObs = (int)mxGetScalar(prhs[2]);
  51.     levndir = new int[nl];
  52.     templevndir = mxGetPr(prhs[3]);
  53.     for(i = 0; i<nl; i++)
  54.       levndir[i] = (int)(templevndir[i]);
  56.     intptr = new int(nl);
  57.     plhs[0] = mxCreateCellArray(1, intptr);
  58.     temp = nObs;
  60.     for (i = 0; i < nl; i++)
  61.     {
  62.       initcell = mxCreateDoubleMatrix(temp, 1, mxREAL);
  63.       mxSetCell(plhs[0], i, initcell);
  64.       temp = temp*4;
  65.     }
  67.     dataTree = pdfbgen_tdata(prhs[0], nObs, levndir, nl);
  69.     for (i = 0; i < nl; i++)
  70.     {
  71.       constdatacell = plhs[0];
  72.       datacell = mxGetCell(constdatacell, i);
  73.       constdatacell = datacell;
  74.       data = mxGetPr(constdatacell);
  76.       for (n = 0; n < (*dataTree)[i].size(); n++)
  77. *data++ = (*dataTree)[i][n];
  78.     }
  80.     return;    
  81. }