/*=========================================================================

  Program:   Image Guided Surgery Software Toolkit
  Module:    $RCSfile: igstkMultipleOutput.h,v $
  Language:  C++
  Date:      $Date: 2006/04/03 18:46:59 $
  Version:   $Revision: 1.6 $

  Copyright (c) ISIS Georgetown University. All rights reserved.
  See IGSTKCopyright.txt or http://www.igstk.org/HTML/Copyright.htm for details.

     This software is distributed WITHOUT ANY WARRANTY; without even
     the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR
     PURPOSE.  See the above copyright notices for more information.

=========================================================================*/

#ifndef __igstkMultipleOutput_h
#define __igstkMultipleOutput_h


#include <iostream>
#include <fstream>
#include <set>

namespace igstk
{

/** \class MultipleOutput
    \brief Class MultipleOutput allows writing simultaneously to multiple
    streams.  Note that the class derives from std::streambuf and contains a
    std::set<> of std::ostream.
*/

class MultipleOutput : virtual public std::streambuf
{

public:

  typedef std::ostream               StreamType;

  
public:

  /** Constructor */
  MultipleOutput();

  /** Destructor */
  ~MultipleOutput();


  /** Register a additional output stream into the list of ostreams to write
   * to. The messages will be sent to the streams in the same order that the
   * streams have been added here.  */
  void AddOutputStream( StreamType & output );


  /** Broadcast a flush operation to all the output streams */
  void Flush();

  /** Operator that will receive different input types and will forward  them
   * to the multiple outputs. */
  template <class T>
  MultipleOutput& operator << ( T tt )
  {
    ContainerType::iterator itr = m_Output.begin();
    ContainerType::iterator end = m_Output.end();
    while( itr != end )
      {
      *(*itr) << tt;
      ++itr;
      }
    return *this;
  }


private:
  
  typedef std::set< StreamType * >   ContainerType;

  ContainerType        m_Output;

};

}

#endif //__igstk_MultipleOutput_h_