ibsimu.hpp

Go to the documentation of this file.

 
/* Copyright (c) 2010-2012 Taneli Kalvas. All rights reserved.
 *
 * You can redistribute this software and/or modify it under the terms
 * of the GNU General Public License as published by the Free Software
 * Foundation; either version 2 of the License, or (at your option)
 * any later version.
 * 
 * This library is distributed in the hope that it will be useful, but
 * WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
 * General Public License for more details.
 * 
 * You should have received a copy of the GNU General Public License
 * along with this library (file "COPYING" included in the package);
 * if not, write to the Free Software Foundation, Inc., 51 Franklin
 * Street, Fifth Floor, Boston, MA 02110-1301 USA
 * 
 * If you have questions about your rights to use or distribute this
 * software, please contact Berkeley Lab's Technology Transfer
 * Department at TTD@lbl.gov. Other questions, comments and bug
 * reports should be sent directly to the author via email at
 * taneli.kalvas@jyu.fi.
 * 
 * NOTICE. This software was developed under partial funding from the
 * U.S.  Department of Energy.  As such, the U.S. Government has been
 * granted for itself and others acting on its behalf a paid-up,
 * nonexclusive, irrevocable, worldwide license in the Software to
 * reproduce, prepare derivative works, and perform publicly and
 * display publicly.  Beginning five (5) years after the date
 * permission to assert copyright is obtained from the U.S. Department
 * of Energy, and subject to any subsequent five (5) year renewals,
 * the U.S. Government is granted for itself and others acting on its
 * behalf a paid-up, nonexclusive, irrevocable, worldwide license in
 * the Software to reproduce, prepare derivative works, distribute
 * copies to the public, perform publicly and display publicly, and to
 * permit others to do so.
 */
 
 
#ifndef IBSIMU_HPP
#define IBSIMU_HPP 1
 
 
#include <stdint.h>
#include <stdarg.h>
#include <string>
#include <iostream>
#include <fstream>
#include <sstream>
 
 
class Timer;
 
 
#define MSG_COUNT 5
 
enum message_type_e {
    MSG_VERBOSE = 0,
    MSG_WARNING,
    MSG_ERROR,
    MSG_DEBUG_GENERAL,
    MSG_DEBUG_DXF
};
 
 
enum rng_type_e {
    RNG_SOBOL = 0,
    RNG_MT
};
 
 
class IBSimu 
{
    Timer        *_t;
 
    bool          _hello;
    uint32_t      _threadcount;
    rng_type_e    _rng;
    
    struct nullstream: std::ostream {
        struct nullbuf: std::streambuf {
            int overflow( int c ) { return( traits_type::not_eof(c) ); }
        } m_sbuf;
        nullstream(): std::ios(&m_sbuf), std::ostream(&m_sbuf) {}
    };
 
    std::stringstream  _ss;
    nullstream         _ns;
    std::ofstream      _fo;
    std::ostream      *_os;             // Message output stream
 
    int32_t       _indent;
    int32_t       _message_threshold[MSG_COUNT];
 
    IBSimu( const IBSimu &ibs );
 
    const IBSimu &operator=( const IBSimu &ibs ) { return( *this ); }
 
    size_t convert_stringstream_to_lines( const std::stringstream &ss,
                                          size_t indentation,
                                          std::string &target,
                                          bool &lineunfinished );
 
public:
 
    IBSimu();
 
    ~IBSimu();
 
    std::ostream &set_message_output( std::ostream &os );
 
    std::ostream &set_message_output( const std::string &filename );
 
    std::ostream &message( message_type_e type, int32_t level );
 
    std::ostream &message( int32_t level );
 
    void flush( bool finishlines = true );
 
    void inc_indent( void );
 
    void dec_indent( void );
 
    bool output_is_cout();
 
    void set_message_threshold( message_type_e type, int32_t level );
 
    int32_t get_message_threshold( message_type_e type );
 
 
 
 
    void set_thread_count( uint32_t threadcount );
 
    uint32_t get_thread_count( void ) { return( _threadcount ); }
 
 
 
    void set_rng_type( rng_type_e type );
 
    rng_type_e get_rng_type( void ) { return( _rng ); }
    
 
 
    void halt( void );
};
 
 
extern IBSimu ibsimu;
 
 
#endif