wxlogrecordhandler.h

Go to the documentation of this file.

#ifndef wxrecordhandler_h
#define wxrecordhandler_h
 
// WarpX includes
#include "wxobject.h"
#include "wxcreator.h"
#include "wxcryptset.h"
 
// std includes
#include <iostream>
#include <string>
#include <fstream>
#include <ios>
 
class WxLogRecordHandler : public WxObject
{
public:
    WxLogRecordHandler() : _outStream(0), _owns(false)
    {
        _outStream = &std::cout;
    }
 
    virtual ~WxLogRecordHandler()
    {
        // delete stream if owning it
        if (_owns)
            delete _outStream;
    }
 
    void setStream(std::ostream* stream)
    {
        _outStream = stream;
    }
 
    void write(const std::string& msg)
    {
        *_outStream << msg;
        *_outStream << std::flush;
    }
 
    void setOwns(bool handlerOwns)
    {
        _owns = handlerOwns;
    }
 
    bool owns() const
    {
        return _owns;
    }
 
private:
    std::ostream* _outStream;
    bool _owns;
};
 
class WxStreamHandler : public WxLogRecordHandler
{
public:
    WxStreamHandler() : WxLogRecordHandler()
    {
    }
 
    WxStreamHandler(std::ostream* stream)
    {
        setStream(stream);
    }
};
 
class WxFileHandler : public WxLogRecordHandler
{
public:
    WxFileHandler() : WxLogRecordHandler(), _file(0)
    {
    }
 
    WxFileHandler(const std::string& fname,
                  std::ios_base::openmode mode = std::ios_base::trunc)
    {
        // open a file with specified mode and set it as the output stream
        _file = new std::ofstream(fname.c_str(), mode);
        setStream(_file);
    }
 
    ~WxFileHandler()
    {
        delete _file;
    }
 
    void setup(const WxCryptSet& wxc)
    {
        // read file and open it for logging
        std::string fname = wxc.get<std::string>("file");
        if (_file)
            // need to close _file if it is already open
            delete _file;
        _file = new std::ofstream(fname.c_str());
        setStream(_file);
    }
 
private:
    std::ofstream* _file;
};
 
#endif // wxrecordhandler_h