X-Git-Url: http://git.tdb.fi/?p=libs%2Fcore.git;a=blobdiff_plain;f=source%2Fio%2Fconsole.cpp;h=a03ad20747bf251c40c57f682e027f760756cb7e;hp=4d686db1aafb5ec33dc74ad2cd212a5a84d0c853;hb=3f7aa81d9212811b323d89949adcccda212cbed3;hpb=d16185720fa344263367dbd50c61bfc8183d99a4

diff --git a/source/io/console.cpp b/source/io/console.cpp
index 4d686db..a03ad20 100644
--- a/source/io/console.cpp
+++ b/source/io/console.cpp
@@ -1,5 +1,6 @@
 #ifndef WIN32
 #include <errno.h>
+#include <unistd.h>
 #include <fcntl.h>
 #include <termios.h>
 #include <sys/ioctl.h>
@@ -12,7 +13,19 @@ using namespace std;
 
 namespace {
 
-#ifndef WIN32
+#ifdef WIN32
+DWORD stream_to_sys(Msp::IO::Console::Stream stream)
+{
+	switch(stream)
+	{
+	case Msp::IO::Console::CIN: return STD_INPUT_HANDLE;
+	case Msp::IO::Console::COUT: return STD_OUTPUT_HANDLE;
+	case Msp::IO::Console::CERR: return STD_ERROR_HANDLE;
+	default: throw invalid_argument("stream_to_sys");
+	}
+}
+
+#else
 termios orig_attr;
 #endif
 
@@ -21,35 +34,28 @@ termios orig_attr;
 namespace Msp {
 namespace IO {
 
-Console::Console(unsigned n)
+Console::Console(Stream s):
+	stream(s)
 {
-	if(n>2)
-		throw invalid_argument("Console::Console");
-
-	mode = (n==0 ? M_READ : M_WRITE);
+	mode = (stream==CIN ? M_READ : M_WRITE);
 
 #ifdef WIN32
-	switch(n)
-	{
-	case 0: *handle = GetStdHandle(STD_INPUT_HANDLE); break;
-	case 1: *handle = GetStdHandle(STD_OUTPUT_HANDLE); break;
-	case 2: *handle = GetStdHandle(STD_ERROR_HANDLE); break;
-	}
+	*handle = GetStdHandle(stream_to_sys(stream));
 #else
-	*handle = n;
+	*handle = stream;
 
-	if(n==0)
+	if(stream==CIN)
 		tcgetattr(*handle, &orig_attr);
 #endif
 
-	if(n==0)
+	if(stream==CIN)
 		set_events(P_INPUT);
 }
 
 Console::~Console()
 {
 #ifndef WIN32
-	if(handle==0)
+	if(stream==CIN)
 		tcsetattr(*handle, TCSADRAIN, &orig_attr);
 #endif
 }
@@ -88,14 +94,28 @@ void Console::set_line_buffer(bool l)
 
 #ifdef WIN32
 	DWORD m;
-	GetConsoleMode(*handle, &m);
-	SetConsoleMode(*handle, (m&~ENABLE_LINE_INPUT) | (l?ENABLE_LINE_INPUT:0));
+	if(!GetConsoleMode(*handle, &m))
+		throw system_error("GetConsoleMode");
+	if(!SetConsoleMode(*handle, (m&~ENABLE_LINE_INPUT) | (l?ENABLE_LINE_INPUT:0)))
+		throw system_error("SetConsoleMode");
 #else
-	// XXX ICANON does more than just set line buffering, may need a bit more thought
 	termios t;
-	tcgetattr(*handle, &t);
+	if(tcgetattr(*handle, &t)==-1)
+		throw system_error("tcgetattr");
 	t.c_lflag = (t.c_lflag&~ICANON) | (l?ICANON:0);
-	tcsetattr(*handle, TCSADRAIN, &t);
+	// man termios warns that VMIN and VTIME may share indices with VEOF and VEOL
+	if(l)
+	{
+		t.c_cc[VEOF] = orig_attr.c_cc[VEOF];
+		t.c_cc[VEOL] = orig_attr.c_cc[VEOL];
+	}
+	else
+	{
+		t.c_cc[VMIN] = 1;
+		t.c_cc[VTIME] = 0;
+	}
+	if(tcsetattr(*handle, TCSADRAIN, &t)==-1)
+		throw system_error("tcsetattr");
 #endif
 }
 
@@ -104,17 +124,31 @@ void Console::get_size(unsigned &rows, unsigned &cols)
 	check_access(M_WRITE);
 
 #ifdef WIN32
-	// XXX Figure out how to do this
-	rows = 24;
-	cols = 80;
+	CONSOLE_SCREEN_BUFFER_INFO sbi;
+	if(!GetConsoleScreenBufferInfo(*handle, &sbi))
+		throw system_error("GetConsoleScreenBufferInfo");
+	// Right/bottom coords are inclusive
+	rows = sbi.srWindow.Bottom+1-sbi.srWindow.Top;
+	cols = sbi.srWindow.Right+1-sbi.srWindow.Left;
 #else
 	struct winsize wsz;
-	ioctl(*handle, TIOCGWINSZ, &wsz);
+	if(ioctl(*handle, TIOCGWINSZ, &wsz)==-1)
+		throw system_error("ioctl TIOCGWINSZ");
 	rows = wsz.ws_row;
 	cols = wsz.ws_col;
 #endif
 }
 
+void Console::redirect(Base &other)
+{
+	Handle other_handle = other.get_handle(mode&M_RDWR);
+#ifdef WIN32
+	SetStdHandle(stream_to_sys(stream), *other_handle);
+#else
+	dup2(*other_handle, *handle);
+#endif
+}
+
 unsigned Console::do_write(const char *buf, unsigned len)
 {
 	check_access(M_WRITE);
@@ -133,25 +167,25 @@ unsigned Console::do_read(char *buf, unsigned len)
 	return ret;
 }
 
-Console &Console::instance(unsigned n)
+Console &Console::instance(Stream s)
 {
-	static Console in(0);
-	static Console out(1);
-	static Console err(2);
+	static Console in(CIN);
+	static Console out(COUT);
+	static Console err(CERR);
 
-	switch(n)
+	switch(s)
 	{
-	case 0: return in;
-	case 1: return out;
-	case 2: return err;
+	case CIN: return in;
+	case COUT: return out;
+	case CERR: return err;
 	}
 
 	throw invalid_argument("Console::instance");
 }
 
-Console &cin = Console::instance(0);
-Console &cout = Console::instance(1);
-Console &cerr = Console::instance(2);
+Console &cin = Console::instance(Console::CIN);
+Console &cout = Console::instance(Console::COUT);
+Console &cerr = Console::instance(Console::CERR);
 
 } // namespace IO
 } // namespace Msp