matlab的mexFunction中使用std::cout

编程技术  /  houtizong 发布于 3年前   103

在matlab、C++混合编程的时候，可能会用到之前的代码。代码中会有许多的输出（使用std::cout）;
如果将这些std::cout重新用mexPrintf()写一遍，不仅费时费力，还造成原来的程序在别的地方无法编译的问题。

那么，有没有办法将std::cout重定向到mexPrintf呢？

原理没深究，直接转载代码：http://stackoverflow.com/questions/243696/correctly-over-loading-a-stringbuf-to-replace-cout-in-a-matlab-mex-file
先定义一个自己的streambuf类：

class mstream : public std::streambuf {public:protected:    virtual std::streamsize xsputn (const char *s, std::streamsize n) {        mexPrintf ("%.*s", n, s);        return n;    }    virtual int overflow (int c = EOF) {        if (c != EOF) {            mexPrintf ("%.1s", &c);        }        return 1;    }};

mexFunction中这么用就可以了：

void mexFunction (int nlhs, mxArray *plhs[], int nrhs, const mxArray *prhs[]) {    mstream mout;    std::streambuf *outbuf = std::cout.rdbuf (&mout);//重定向cout        std::cout << "haha" << std::endl;    std::cout.rdbuf (outbuf);//结束重定向}