dust3d/third_party/libigl/include/igl/matlab/MexStream.h

57 lines
1.5 KiB
C++

// This file is part of libigl, a simple c++ geometry processing library.
//
// Copyright (C) 2013 Alec Jacobson <alecjacobson@gmail.com>
//
// This Source Code Form is subject to the terms of the Mozilla Public License
// v. 2.0. If a copy of the MPL was not distributed with this file, You can
// obtain one at http://mozilla.org/MPL/2.0/.
#ifndef IGL_MATLAB_MEX_STREAM_H
#define IGL_MATLAB_MEX_STREAM_H
#include <iostream>
namespace igl
{
namespace matlab
{
// http://stackoverflow.com/a/249008/148668
// Class to implement "cout" for mex files to print to the matlab terminal
// window.
//
// Insert at the beginning of mexFunction():
// MexStream mout;
// std::streambuf *outbuf = std::cout.rdbuf(&mout);
// ...
// ALWAYS restore original buffer to avoid memory leak problems in matlab
// std::cout.rdbuf(outbuf);
//
class MexStream : public std::streambuf
{
public:
protected:
inline virtual std::streamsize xsputn(const char *s, std::streamsize n);
inline virtual int overflow(int c = EOF);
};
}
}
// Implementation
#include <mex.h>
inline std::streamsize igl::matlab::MexStream::xsputn(
const char *s,
std::streamsize n)
{
mexPrintf("%.*s",n,s);
mexEvalString("drawnow;"); // to dump string.
return n;
}
inline int igl::matlab::MexStream::overflow(int c)
{
if (c != EOF) {
mexPrintf("%.1s",&c);
mexEvalString("drawnow;"); // to dump string.
}
return 1;
}
#endif