174 lines
4.5 KiB
C++
174 lines
4.5 KiB
C++
/*
|
|
https://github.com/peterix/dfhack
|
|
Copyright (c) 2009-2011 Petr Mrázek (peterix@gmail.com)
|
|
|
|
This software is provided 'as-is', without any express or implied
|
|
warranty. In no event will the authors be held liable for any
|
|
damages arising from the use of this software.
|
|
|
|
Permission is granted to anyone to use this software for any
|
|
purpose, including commercial applications, and to alter it and
|
|
redistribute it freely, subject to the following restrictions:
|
|
|
|
1. The origin of this software must not be misrepresented; you must
|
|
not claim that you wrote the original software. If you use this
|
|
software in a product, an acknowledgment in the product documentation
|
|
would be appreciated but is not required.
|
|
|
|
2. Altered source versions must be plainly marked as such, and
|
|
must not be misrepresented as being the original software.
|
|
|
|
3. This notice may not be removed or altered from any source
|
|
distribution.
|
|
*/
|
|
|
|
#pragma once
|
|
#include "Pragma.h"
|
|
#include "Export.h"
|
|
|
|
#include <list>
|
|
#include <fstream>
|
|
#include <assert.h>
|
|
#include <iostream>
|
|
#include <string>
|
|
#include <stdarg.h>
|
|
#include <sstream>
|
|
|
|
namespace dfproto
|
|
{
|
|
class CoreTextNotification;
|
|
}
|
|
|
|
namespace DFHack
|
|
{
|
|
class DFHACK_EXPORT color_ostream : public std::ostream
|
|
{
|
|
public:
|
|
enum color_value
|
|
{
|
|
COLOR_RESET = -1,
|
|
COLOR_BLACK = 0,
|
|
COLOR_BLUE,
|
|
COLOR_GREEN,
|
|
COLOR_CYAN,
|
|
COLOR_RED,
|
|
COLOR_MAGENTA,
|
|
COLOR_BROWN,
|
|
COLOR_GREY,
|
|
COLOR_DARKGREY,
|
|
COLOR_LIGHTBLUE,
|
|
COLOR_LIGHTGREEN,
|
|
COLOR_LIGHTCYAN,
|
|
COLOR_LIGHTRED,
|
|
COLOR_LIGHTMAGENTA,
|
|
COLOR_YELLOW,
|
|
COLOR_WHITE,
|
|
COLOR_MAX = COLOR_WHITE
|
|
};
|
|
|
|
private:
|
|
color_value cur_color;
|
|
|
|
class buffer : public std::stringbuf
|
|
{
|
|
public:
|
|
color_ostream *owner;
|
|
|
|
buffer(color_ostream *owner) : owner(owner) {}
|
|
virtual ~buffer() { }
|
|
|
|
protected:
|
|
virtual int sync() {
|
|
owner->flush_buffer(true);
|
|
return 0;
|
|
}
|
|
};
|
|
|
|
buffer *buf() { return (buffer*)rdbuf(); }
|
|
|
|
void flush_buffer(bool flush);
|
|
|
|
protected:
|
|
// These must be strictly balanced, because
|
|
// they might grab and hold mutexes.
|
|
virtual void begin_batch();
|
|
virtual void end_batch();
|
|
|
|
virtual void add_text(color_value color, const std::string &text) = 0;
|
|
|
|
virtual void flush_proxy() {};
|
|
|
|
friend class color_ostream_proxy;
|
|
public:
|
|
color_ostream();
|
|
virtual ~color_ostream();
|
|
|
|
/// Print a formatted string, like printf
|
|
void print(const char *format, ...);
|
|
void vprint(const char *format, va_list args);
|
|
|
|
/// Print a formatted string, like printf, in red
|
|
void printerr(const char *format, ...);
|
|
void vprinterr(const char *format, va_list args);
|
|
|
|
/// Set color (ANSI color number)
|
|
void color(color_value c);
|
|
/// Reset color to default
|
|
void reset_color(void);
|
|
|
|
virtual bool is_console() { return false; }
|
|
virtual color_ostream *proxy_target() { return NULL; }
|
|
};
|
|
|
|
inline color_ostream &operator << (color_ostream &out, color_ostream::color_value clr)
|
|
{
|
|
out.color(clr);
|
|
return out;
|
|
}
|
|
|
|
class DFHACK_EXPORT color_ostream_wrapper : public color_ostream
|
|
{
|
|
std::ostream &out;
|
|
|
|
protected:
|
|
virtual void add_text(color_value color, const std::string &text);
|
|
virtual void flush_proxy();
|
|
|
|
public:
|
|
color_ostream_wrapper(std::ostream &os) : out(os) {}
|
|
};
|
|
|
|
class DFHACK_EXPORT buffered_color_ostream : public color_ostream
|
|
{
|
|
protected:
|
|
virtual void add_text(color_value color, const std::string &text);
|
|
|
|
public:
|
|
typedef std::pair<color_value,std::string> fragment_type;
|
|
|
|
buffered_color_ostream() {}
|
|
~buffered_color_ostream() {}
|
|
|
|
const std::list<fragment_type> &fragments() { return buffer; }
|
|
|
|
protected:
|
|
std::list<fragment_type> buffer;
|
|
};
|
|
|
|
class DFHACK_EXPORT color_ostream_proxy : public buffered_color_ostream
|
|
{
|
|
protected:
|
|
color_ostream *target;
|
|
|
|
virtual void flush_proxy();
|
|
|
|
public:
|
|
color_ostream_proxy(color_ostream &target) : target(&target) {}
|
|
~color_ostream_proxy();
|
|
|
|
virtual color_ostream *proxy_target() { return target; }
|
|
|
|
void decode(dfproto::CoreTextNotification *data);
|
|
};
|
|
}
|