logstream.h

来自「利用C」· C头文件 代码 · 共 53 行

// Copyright (C) 2003-2005 Anders Logg.
// Licensed under the GNU LGPL Version 2.1.
//
// Modified by Garth N. Wells 2005.
//
// First added:  2003-03-13
// Last changed: 2005-12-09

#ifndef __LOG_STREAM_H
#define __LOG_STREAM_H

#include <string>
#include <dolfin/common/types.h>

namespace dolfin
{

  class LogStream
  {
  public:
    
    enum Type {COUT, ENDL};
    
    LogStream(Type type);
    ~LogStream();
    
    LogStream& operator<<(const char* s);
    LogStream& operator<<(const std::string& s);
    LogStream& operator<<(int a);
    LogStream& operator<<(unsigned int a);
    LogStream& operator<<(real a);
    LogStream& operator<<(complex z);
    LogStream& operator<<(const LogStream& stream);

    void disp() const;
    
  private:
    
    void add(const char* msg);
    
    Type type;
    char* buffer;
    int current;
    
  };
  
  extern LogStream cout;
  extern LogStream endl;
  
}

#endif