itkThreadLogger.h

Go to the documentation of this file.

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

  Program:   Insight Segmentation & Registration Toolkit
  Module:    $RCSfile: itkThreadLogger.h,v $
  Language:  C++
  Date:      $Date: 2009-03-03 15:09:52 $
  Version:   $Revision: 1.7 $

  Copyright (c) Insight Software Consortium. All rights reserved.
  See ITKCopyright.txt or http://www.itk.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 __itkThreadLogger_h
#define __itkThreadLogger_h

#include "itkMacro.h"
#include "itkMultiThreader.h"
#include "itkLogger.h"
#include "itkSimpleFastMutexLock.h"

#include <string>
#include <queue>

namespace itk
{
class ITKCommon_EXPORT ThreadLogger : public Logger
{

public:
  
  typedef ThreadLogger              Self;
  typedef Logger                    Superclass;
  typedef SmartPointer<Self>        Pointer;
  typedef SmartPointer<const Self>  ConstPointer;

  itkTypeMacro( ThreadLogger, Logger );

  itkNewMacro( Self );

  typedef  Logger::OutputType   OutputType;

  typedef  Logger::PriorityLevelType  PriorityLevelType;

  typedef enum
    {
    SET_PRIORITY_LEVEL,
    SET_LEVEL_FOR_FLUSHING,
    ADD_LOG_OUTPUT,
    WRITE,
    FLUSH
    } OperationType;

  virtual void SetPriorityLevel( PriorityLevelType level );

  virtual PriorityLevelType GetPriorityLevel() const;

  virtual void SetLevelForFlushing( PriorityLevelType level );

  virtual PriorityLevelType GetLevelForFlushing() const;

  virtual void AddLogOutput( OutputType* output );

  virtual void Write(PriorityLevelType level, std::string const & content);

  virtual void Flush();

protected:

  ThreadLogger();

  virtual ~ThreadLogger();

  virtual void PrintSelf(std::ostream &os, Indent indent) const;

  static ITK_THREAD_RETURN_TYPE ThreadFunction(void*);

private:

  typedef std::queue<OperationType> OperationContainerType;

  typedef std::queue<std::string>  MessageContainerType;

  typedef std::queue<PriorityLevelType>  LevelContainerType;

  typedef std::queue<OutputType::Pointer>  OutputContainerType;

  MultiThreader::Pointer  m_Threader;

  int m_ThreadID;

  OperationContainerType  m_OperationQ;

  MessageContainerType  m_MessageQ;

  LevelContainerType  m_LevelQ;

  OutputContainerType m_OutputQ;

  SimpleFastMutexLock m_Mutex;

  SimpleFastMutexLock m_WaitMutex;

};  // class ThreadLogger


} // namespace itk


#endif  // __itkThreadLogger_h