Console revamp - linux part.
Petr Mrázek
parent
b41ab40bb6
commit
f8cdd95b09
@ -0,0 +1,119 @@
|
||||
/*
|
||||
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.
|
||||
*/
|
||||
|
||||
#include "dfhack/Console.h"
|
||||
#include <cstdio>
|
||||
#include <cstdlib>
|
||||
#include <sstream>
|
||||
using namespace DFHack;
|
||||
|
||||
duthomhas::stdiostream dfout;
|
||||
FILE * dfout_C = 0;
|
||||
duthomhas::stdiobuf * stream_o = 0;
|
||||
|
||||
// FIXME: prime candidate for being a singleton...
|
||||
Console::Console()
|
||||
{
|
||||
// make our own weird streams so our IO isn't redirected
|
||||
dfout_C = fopen("/dev/tty", "w");
|
||||
stream_o = new duthomhas::stdiobuf(dfout_C);
|
||||
dfout.rdbuf(stream_o);
|
||||
std::cin.tie(&dfout);
|
||||
clear();
|
||||
// result is a terminal controlled by the parasitic code!
|
||||
}
|
||||
Console::~Console()
|
||||
{
|
||||
|
||||
}
|
||||
void Console::clear()
|
||||
{
|
||||
dfout << "\033c";
|
||||
dfout << "\033[3J\033[H";
|
||||
}
|
||||
void Console::gotoxy(int x, int y)
|
||||
{
|
||||
std::ostringstream oss;
|
||||
oss << "\033[" << y << ";" << x << "H";
|
||||
dfout << oss.str();
|
||||
}
|
||||
|
||||
const char * ANSI_CLS = "\033[2J";
|
||||
const char * ANSI_BLACK = "\033[22;30m";
|
||||
const char * ANSI_RED = "\033[22;31m";
|
||||
const char * ANSI_GREEN = "\033[22;32m";
|
||||
const char * ANSI_BROWN = "\033[22;33m";
|
||||
const char * ANSI_BLUE = "\033[22;34m";
|
||||
const char * ANSI_MAGENTA = "\033[22;35m";
|
||||
const char * ANSI_CYAN = "\033[22;36m";
|
||||
const char * ANSI_GREY = "\033[22;37m";
|
||||
const char * ANSI_DARKGREY = "\033[01;30m";
|
||||
const char * ANSI_LIGHTRED = "\033[01;31m";
|
||||
const char * ANSI_LIGHTGREEN = "\033[01;32m";
|
||||
const char * ANSI_YELLOW = "\033[01;33m";
|
||||
const char * ANSI_LIGHTBLUE = "\033[01;34m";
|
||||
const char * ANSI_LIGHTMAGENTA = "\033[01;35m";
|
||||
const char * ANSI_LIGHTCYAN = "\033[01;36m";
|
||||
const char * ANSI_WHITE = "\033[01;37m";
|
||||
|
||||
const char * getANSIColor(const int c)
|
||||
{
|
||||
switch (c)
|
||||
{
|
||||
case 0 : return ANSI_BLACK;
|
||||
case 1 : return ANSI_BLUE; // non-ANSI
|
||||
case 2 : return ANSI_GREEN;
|
||||
case 3 : return ANSI_CYAN; // non-ANSI
|
||||
case 4 : return ANSI_RED; // non-ANSI
|
||||
case 5 : return ANSI_MAGENTA;
|
||||
case 6 : return ANSI_BROWN;
|
||||
case 7 : return ANSI_GREY;
|
||||
case 8 : return ANSI_DARKGREY;
|
||||
case 9 : return ANSI_LIGHTBLUE; // non-ANSI
|
||||
case 10: return ANSI_LIGHTGREEN;
|
||||
case 11: return ANSI_LIGHTCYAN; // non-ANSI;
|
||||
case 12: return ANSI_LIGHTRED; // non-ANSI;
|
||||
case 13: return ANSI_LIGHTMAGENTA;
|
||||
case 14: return ANSI_YELLOW; // non-ANSI
|
||||
case 15: return ANSI_WHITE;
|
||||
default: return "";
|
||||
}
|
||||
}
|
||||
|
||||
void Console::color(int index)
|
||||
{
|
||||
dfout << getANSIColor(index);
|
||||
}
|
||||
|
||||
void Console::cursor(bool enable)
|
||||
{
|
||||
if(enable)
|
||||
{
|
||||
dfout <<"\033[?25h";
|
||||
}
|
||||
else
|
||||
{
|
||||
dfout <<"\033[?25l";
|
||||
}
|
||||
}
|
||||
@ -0,0 +1,46 @@
|
||||
/*
|
||||
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 "dfhack/Pragma.h"
|
||||
#include "dfhack/Export.h"
|
||||
#include "dfhack/Core.h"
|
||||
#include "dfhack/extra/stdiostream.h"
|
||||
|
||||
extern DFHACK_EXPORT duthomhas::stdiostream dfout;
|
||||
extern DFHACK_EXPORT FILE * dfout_C;
|
||||
|
||||
namespace DFHack
|
||||
{
|
||||
class DFHACK_EXPORT Console
|
||||
{
|
||||
public:
|
||||
Console();
|
||||
~Console();
|
||||
void clear();
|
||||
void gotoxy(int x, int y);
|
||||
void color(int index);
|
||||
void cursor(bool enable = true);
|
||||
};
|
||||
}
|
||||
@ -0,0 +1,302 @@
|
||||
// stdiostream.hpp
|
||||
//
|
||||
// Copyright (c) 2010 Michael Thomas Greer
|
||||
// Distributed under the Boost Software License, Version 1.0
|
||||
// (See http://www.boost.org/LICENSE_1_0.txt )
|
||||
//
|
||||
|
||||
#pragma once
|
||||
#ifndef DUTHOMHAS_STDIOSTREAM_HPP
|
||||
#define DUTHOMHAS_STDIOSTREAM_HPP
|
||||
|
||||
#include <cstdio>
|
||||
#include <iostream>
|
||||
#include <streambuf>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
||||
namespace duthomhas
|
||||
{
|
||||
|
||||
/* /////////////////////////////////////////////////////////////////////////
|
||||
basic_stdiobuf
|
||||
///////////////////////////////////////////////////////////////////////// */
|
||||
|
||||
template <
|
||||
typename CharType,
|
||||
typename CharTraits = std::char_traits <CharType>
|
||||
>
|
||||
class basic_stdiobuf: public std::basic_streambuf <CharType, CharTraits>
|
||||
{
|
||||
//------------------------------------------------------------------------
|
||||
public:
|
||||
//------------------------------------------------------------------------
|
||||
typedef CharType char_type;
|
||||
typedef CharTraits traits_type;
|
||||
typedef typename traits_type::int_type int_type;
|
||||
typedef typename traits_type::pos_type pos_type;
|
||||
typedef typename traits_type::off_type off_type;
|
||||
|
||||
typedef basic_stdiobuf <char_type, traits_type> this_type;
|
||||
|
||||
//......................................................................
|
||||
basic_stdiobuf( FILE* fp = NULL ):
|
||||
fp( fp )
|
||||
{ }
|
||||
|
||||
//......................................................................
|
||||
//BUG 1: Hey! I never get called! (How is that?)
|
||||
~basic_stdiobuf()
|
||||
{
|
||||
this->close();
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
bool is_open() const throw()
|
||||
{
|
||||
return fp != NULL;
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
this_type* open( const char* filename, std::ios_base::openmode mode )
|
||||
{
|
||||
if (is_open()) return NULL;
|
||||
|
||||
// Figure out the open mode flags . . . . . . . . . . . . . . . . . .
|
||||
std::string fmode;
|
||||
|
||||
bool is_ate = mode & std::ios_base::ate;
|
||||
bool is_bin = mode & std::ios_base::binary;
|
||||
mode &= ~(std::ios_base::ate | std::ios_base::binary);
|
||||
|
||||
#define _(flag) std::ios_base::flag
|
||||
if (mode == ( _(in) )) fmode = "r";
|
||||
else if (mode == ( _(out) & _(trunc))) fmode = "w";
|
||||
else if (mode == (_(app) & _(out) )) fmode = "a";
|
||||
else if (mode == ( _(in) & _(out) )) fmode = "r+";
|
||||
else if (mode == ( _(in) & _(out) & _(trunc))) fmode = "w+";
|
||||
else if (mode == (_(app) & _(in) & _(out) )) fmode = "a+";
|
||||
// I would prefer to throw an exception here,
|
||||
// but the standard only wants a NULL result.
|
||||
else return NULL;
|
||||
#undef _
|
||||
if (is_bin) fmode.insert( 1, 1, 'b' );
|
||||
|
||||
// Try opening the file . . . . . . . . . . . . . . . . . . . . . . .
|
||||
fp = std::fopen( filename, fmode.c_str() );
|
||||
if (!fp) return NULL;
|
||||
|
||||
// Reposition to EOF if wanted . . . . . . . . . . . . . . . . . . . .
|
||||
if (is_ate) std::fseek( fp, 0, SEEK_END );
|
||||
|
||||
return this;
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
this_type* close()
|
||||
{
|
||||
if (fp)
|
||||
{
|
||||
std::fclose( fp );
|
||||
fp = NULL;
|
||||
}
|
||||
pushbacks.clear();
|
||||
return this;
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
FILE* stdiofile() const
|
||||
{
|
||||
return fp;
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------
|
||||
protected:
|
||||
//------------------------------------------------------------------------
|
||||
|
||||
//......................................................................
|
||||
// Get the CURRENT character without advancing the file pointer
|
||||
virtual int_type underflow()
|
||||
{
|
||||
// Return anything previously pushed-back
|
||||
if (pushbacks.size())
|
||||
return pushbacks.back();
|
||||
|
||||
// Else do the right thing
|
||||
fpos_t pos;
|
||||
if (std::fgetpos( fp, &pos ) != 0)
|
||||
return traits_type::eof();
|
||||
|
||||
int c = std::fgetc( fp );
|
||||
std::fsetpos( fp, &pos );
|
||||
|
||||
return maybe_eof( c );
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
// Get the CURRENT character AND advance the file pointer
|
||||
virtual int_type uflow()
|
||||
{
|
||||
// Return anything previously pushed-back
|
||||
if (pushbacks.size())
|
||||
{
|
||||
int_type c = pushbacks.back();
|
||||
pushbacks.pop_back();
|
||||
return c;
|
||||
}
|
||||
|
||||
// Else do the right thing
|
||||
return maybe_eof( std::fgetc( fp ) );
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
virtual int_type pbackfail( int_type c = traits_type::eof() )
|
||||
{
|
||||
if (!is_open())
|
||||
return traits_type::eof();
|
||||
|
||||
// If the argument c is EOF and the file pointer is not at the
|
||||
// beginning of the character sequence, it is decremented by one.
|
||||
if (traits_type::eq_int_type( c, traits_type::eof() ))
|
||||
{
|
||||
pushbacks.clear();
|
||||
return std::fseek( fp, -1L, SEEK_CUR )
|
||||
? traits_type::eof()
|
||||
: 0;
|
||||
}
|
||||
|
||||
// Otherwise, make the argument the next value to be returned by
|
||||
// underflow() or uflow()
|
||||
pushbacks.push_back( c );
|
||||
return c;
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
virtual int_type overflow( int_type c = traits_type::eof() )
|
||||
{
|
||||
pushbacks.clear();
|
||||
|
||||
// Do nothing
|
||||
if (traits_type::eq_int_type( c, traits_type::eof() ))
|
||||
return 0;
|
||||
|
||||
// Else write a character
|
||||
return maybe_eof( std::fputc( c, fp ) );
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
virtual this_type* setbuf( char* s, std::streamsize n )
|
||||
{
|
||||
return std::setvbuf( fp, s, (s and n) ? _IOLBF : _IONBF, (size_t)n )
|
||||
? NULL
|
||||
: this;
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
virtual pos_type seekoff(
|
||||
off_type offset,
|
||||
std::ios_base::seekdir direction,
|
||||
std::ios_base::openmode which = std::ios_base::in | std::ios_base::out
|
||||
) {
|
||||
pushbacks.clear();
|
||||
return std::fseek( fp, offset,
|
||||
(direction == std::ios_base::beg) ? SEEK_SET :
|
||||
(direction == std::ios_base::cur) ? SEEK_CUR :
|
||||
SEEK_END
|
||||
) ? (-1) : std::ftell( fp );
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
virtual pos_type seekpos(
|
||||
pos_type position,
|
||||
std::ios_base::openmode which = std::ios_base::in | std::ios_base::out
|
||||
) {
|
||||
pushbacks.clear();
|
||||
return std::fseek( fp, position, SEEK_SET )
|
||||
? (-1)
|
||||
: std::ftell( fp );
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
virtual int sync()
|
||||
{
|
||||
pushbacks.clear();
|
||||
return std::fflush( fp )
|
||||
? traits_type::eof()
|
||||
: 0;
|
||||
}
|
||||
|
||||
//------------------------------------------------------------------------
|
||||
private:
|
||||
//------------------------------------------------------------------------
|
||||
FILE* fp;
|
||||
std::vector <int_type> pushbacks; // we'll treat this like a stack
|
||||
|
||||
//......................................................................
|
||||
// Utility function to make sure EOF gets translated to the proper value
|
||||
inline int_type maybe_eof( int value ) const
|
||||
{
|
||||
return (value == EOF) ? traits_type::eof() : value;
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
/* /////////////////////////////////////////////////////////////////////////
|
||||
basic_stdiostream
|
||||
///////////////////////////////////////////////////////////////////////// */
|
||||
|
||||
template <
|
||||
typename CharType,
|
||||
typename CharTraits = std::char_traits <CharType>
|
||||
>
|
||||
struct basic_stdiostream: public std::basic_iostream <CharType, CharTraits>
|
||||
{
|
||||
typedef CharType char_type;
|
||||
typedef CharTraits traits_type;
|
||||
|
||||
typedef basic_stdiobuf <char_type, traits_type> sbuf_type;
|
||||
typedef basic_stdiostream <char_type, traits_type> this_type;
|
||||
typedef std::basic_iostream <char_type, traits_type> base_type;
|
||||
|
||||
//......................................................................
|
||||
basic_stdiostream( FILE* fp = NULL ):
|
||||
base_type( new sbuf_type( fp ) )
|
||||
{ }
|
||||
|
||||
//......................................................................
|
||||
basic_stdiostream( const char* filename, std::ios_base::openmode mode ):
|
||||
//BUG 2: Oops! This is a potential memory leak!
|
||||
base_type( (new sbuf_type)->open( filename, mode ) )
|
||||
{ }
|
||||
|
||||
//......................................................................
|
||||
void open(
|
||||
const char* filename,
|
||||
std::ios_base::openmode mode = std::ios_base::in | std::ios_base::out
|
||||
) {
|
||||
sbuf_type* buf = static_cast <sbuf_type*> ( this->rdbuf() );
|
||||
if (!(buf->open( filename, mode )))
|
||||
this->setstate( std::ios_base::badbit );
|
||||
}
|
||||
|
||||
//......................................................................
|
||||
void close()
|
||||
{
|
||||
sbuf_type* buf = static_cast <sbuf_type*> ( this->rdbuf() );
|
||||
buf->close();
|
||||
}
|
||||
};
|
||||
|
||||
|
||||
/* /////////////////////////////////////////////////////////////////////////
|
||||
Useful typedefs
|
||||
///////////////////////////////////////////////////////////////////////// */
|
||||
|
||||
typedef basic_stdiobuf <char> stdiobuf;
|
||||
typedef basic_stdiostream <char> stdiostream;
|
||||
|
||||
} // namespace duthomhas
|
||||
|
||||
#endif
|
||||
|
||||
// end stdiostream.hpp
|
||||
Loading…
Reference in New Issue