ThreadStream.h

//-*-C++-*-
/***************************************************************************
 *
 *   Copyright (C) 2007 by Willem van Straten
 *   Licensed under the Academic Free License version 2.1
 *
 ***************************************************************************/
 
// psrchive/Util/genutil/ThreadStream.h
 
#ifndef __ThreadStream_h
#define __ThreadStream_h
 
#ifdef HAVE_CONFIG_H
#include <config.h>
#endif
 
#ifdef HAVE_PTHREAD
#include <pthread.h>
#endif
 
#include "Warning.h"
 
class ThreadStream
{
public:
 
  ThreadStream ();
 
  ~ThreadStream ();
 
  std::ostream& get ();
 
  void * get_raw ();
 
  void set (std::ostream*);
 
  void set (void *);
 
private:
 
#ifdef HAVE_PTHREAD
  pthread_key_t key;
#else
  std::ostream* stream;
#endif
 
};
 
template<class T>
ThreadStream& operator<< (ThreadStream& ts, const T& t)
{
  ts.get() << t;
  return ts;
}
 
inline ThreadStream& operator<< (ThreadStream& ts, manipulator m)
{
  ts.get() << m;
  return ts;
}
 
#endif