PIO.h

Go to the documentation of this file.

#ifndef included_AMP_PIO
#define included_AMP_PIO
 
#include <cstdarg>
#include <cstdint>
#include <functional>
#include <iostream>
 
 
namespace AMP {
 
 
class ParallelStreamBuffer final : public std::streambuf
{
public:
    enum StreamOutputType : uint8_t {
        null    = 0,
        out     = 1,
        log     = 2,
        err     = 4,
        out_log = 3,
        err_log = 6
    };
 
    explicit ParallelStreamBuffer( StreamOutputType type );
 
    void setOutputType( StreamOutputType type );
 
    void setOutputStream( std::ofstream *stream );
 
    virtual ~ParallelStreamBuffer();
 
    int sync() override;
 
    std::streamsize xsputn( const char *text, std::streamsize n ) override;
 
    int overflow( int ch ) override;
 
    int underflow() override;
 
    void reset();
 
private:
    StreamOutputType d_type;
    size_t d_size;
    size_t d_buffer_size;
    char *d_buffer;
    std::ofstream *d_stream;
    inline void reserve( size_t size );
};
 
 
extern std::ostream pout;
 
extern std::ostream perr;
 
extern std::ostream plog;
 
inline int printp( const char *format, ... )
{
    va_list ap;
    va_start( ap, format );
    char tmp[4096];
    int n = vsnprintf( tmp, sizeof( tmp ), format, ap );
    va_end( ap );
    pout << tmp;
    pout.flush();
    return n;
}
 
 
void logOnlyNodeZero( const std::string &filename );
 
void logAllNodes( const std::string &filename, bool singleStream = false );
 
void stopLogging();
 
void overrideCout( std::function<void( const char * )> fun );
 
void overrideCerr( std::function<void( const char * )> fun );
 
 
} // namespace AMP
 
 
#endif