flex_ostreambuf.h

// Copyright 2009-2018 Toby Schneider (http://gobysoft.org/index.wt/people/toby)
//                     GobySoft, LLC (2013-)
//                     Massachusetts Institute of Technology (2007-2014)
//                     Community contributors (see AUTHORS file)
//
//
// This file is part of the Goby Underwater Autonomy Project Libraries
// ("The Goby Libraries").
//
// The Goby Libraries are free software: you can redistribute them and/or modify
// them under the terms of the GNU Lesser General Public License as published by
// the Free Software Foundation, either version 2.1 of the License, or
// (at your option) any later version.
//
// The Goby Libraries are distributed in the hope that they will be useful,
// but WITHOUT ANY WARRANTY; without even the implied warranty of
// MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
// GNU Lesser General Public License for more details.
//
// You should have received a copy of the GNU Lesser General Public License
// along with Goby.  If not, see <http://www.gnu.org/licenses/>.

#ifndef FlexOStreamBuf20091110H
#define FlexOStreamBuf20091110H

#include <cstdio>
#include <deque>
#include <iostream>
#include <sstream>
#include <vector>

#include <boost/thread.hpp>

#include <boost/date_time.hpp>
#include <boost/shared_ptr.hpp>

#include "goby/common/protobuf/logger.pb.h"

#include "term_color.h"

class Group;

namespace goby
{
namespace common
{
class FlexNCurses;

namespace logger_lock
{
enum LockAction
{
    none,
    lock
};
} // namespace logger_lock

namespace logger
{
extern boost::recursive_mutex mutex;

enum Verbosity
{
    UNKNOWN = 4,
    QUIET = protobuf::GLogConfig::QUIET,
    WARN = protobuf::GLogConfig::WARN,
    VERBOSE = protobuf::GLogConfig::VERBOSE,
    //GUI = protobuf::GLogConfig::GUI,
    DEBUG1 = protobuf::GLogConfig::DEBUG1,
    DEBUG2 = protobuf::GLogConfig::DEBUG2,
    DEBUG3 = protobuf::GLogConfig::DEBUG3,
    DIE = -1
};
}; // namespace logger

class FlexOstream;

class FlexOStreamBuf : public std::streambuf
{
  public:
    FlexOStreamBuf(FlexOstream* parent);
    ~FlexOStreamBuf();

    int sync();

    int overflow(int c = EOF);

    void name(const std::string& s) { name_ = s; }

    void add_stream(logger::Verbosity verbosity, std::ostream* os);

    bool is_quiet() const { return highest_verbosity_ == logger::QUIET; }

    bool is_gui() const { return is_gui_; }

    void enable_gui();

    logger::Verbosity highest_verbosity() const { return highest_verbosity_; }

    void group_name(const std::string& s) { group_name_ = s; }

    void set_die_flag(bool b) { die_flag_ = b; }

    void set_verbosity_depth(logger::Verbosity depth) { current_verbosity_ = depth; }

    logger::Verbosity verbosity_depth() { return current_verbosity_; }

    void add_group(const std::string& name, Group g);

    void refresh();

    void set_lock_action(logger_lock::LockAction lock_action) { lock_action_ = lock_action; }

    logger_lock::LockAction lock_action() { return lock_action_; }

  private:
    void display(std::string& s);
    void strip_escapes(std::string& s);

  private:
    std::deque<std::string> buffer_;

    class StreamConfig
    {
      public:
        StreamConfig(std::ostream* os, logger::Verbosity verbosity) : os_(os), verbosity_(verbosity)
        {
        }

        void set_verbosity(logger::Verbosity verbosity) { verbosity_ = verbosity; }

        std::ostream* os() const { return os_; }
        logger::Verbosity verbosity() const { return verbosity_; }

      private:
        std::ostream* os_;
        logger::Verbosity verbosity_;
    };

    std::string name_;
    std::string group_name_;

    std::map<std::string, Group> groups_;

    bool die_flag_;
    logger::Verbosity current_verbosity_;

    FlexNCurses* curses_;
    boost::shared_ptr<boost::thread> input_thread_;

    boost::posix_time::ptime start_time_;

    std::vector<StreamConfig> streams_;

    bool is_gui_;

    logger::Verbosity highest_verbosity_;

    logger_lock::LockAction lock_action_;
    FlexOstream* parent_;
};
} // namespace common
} // namespace goby
#endif