JRedirectStream.hh

Go to the documentation of this file.

#ifndef __JLANG__JREDIRECTSTREAM__
#define __JLANG__JREDIRECTSTREAM__

#include <istream>
#include <ostream>

#include "JLang/JAbstractObjectStatus.hh"


/**
 * \author mdejong
 */

namespace JLANG {}
namespace JPP { using namespace JLANG; }

namespace JLANG {


  /**
   * This class can be used to temporarily redirect one output (input) stream to another output (input) stream.
   * The destructor restores the internal buffer of the first output (input) stream.
   */
  class JRedirectStream :
    public JAbstractObjectStatus
  {
  public:
    /**
     * Constructor.
     * The output stream from is redirected to output stream to.
     *
     * \param  from            output stream
     * \param  to              output stream
     */
    JRedirectStream(std::ostream& from,
                    std::ostream& to) :
      ios(from)
    {
      p = ios.rdbuf();

      ios.rdbuf(to.rdbuf());
    }


    /**
     * Constructor.
     * The input stream from is redirected to input stream to.
     *
     * \param  from            input stream
     * \param  to              input stream
     */
    JRedirectStream(std::istream& from,
                    std::istream& to) :
      ios(from)
    {
      p = ios.rdbuf();

      ios.rdbuf(to.rdbuf());
    }

    
    /**
     * Destructor.
     * Restore internal buffer.
     */
    ~JRedirectStream()
    {
      ios.rdbuf(p);
    }

    
    /**
     * Get status of object.
     *
     * \return                 true
     */
    virtual bool getStatus() const override 
    {
      return true;
    }

  private:
    std::ios&       ios;
    std::streambuf* p;
  };
}

#endif