Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Allow setting color and background color, as well as resetting color, for standard error output. #56

Open
wants to merge 3 commits into
base: master
Choose a base branch
from
Open

Allow setting color and background color, as well as resetting color, for standard error output. #56

Changes from all commits
Commits
Show all changes
3 commits
Select commit Hold shift + click to select a range
1845597
fix UNICODE and switch warning
MangaD Aug 13, 2019
81d6259
allow setting color on standard error output (tested on linux only)
MangaD Nov 21, 2020
587bded
Merge branch 'master' into master
MangaD Nov 21, 2020
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
69 changes: 47 additions & 22 deletions rlutil.h
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -129,21 +129,27 @@ namespace rlutil {
/**
* Defs: Internal typedefs and macros
* RLUTIL_STRING_T - String type depending on which one of C or C++ is used
* RLUTIL_PRINT(str) - Printing macro independent of C/C++
* RLUTIL_PRINT(fp, str) - Printing macro independent of C/C++
*/

#ifdef __cplusplus
#ifndef RLUTIL_STRING_T
typedef std::string RLUTIL_STRING_T;
#endif // RLUTIL_STRING_T

#define RLUTIL_PRINT(st) do { std::cout << st; } while(false)
inline auto& RLUTIL_OUTPUT_STREAM = std::cout;
inline auto& RLUTIL_ERROR_STREAM = std::cerr;

#define RLUTIL_PRINT(fp, st) do { fp << st; } while(false)
#else // __cplusplus
#ifndef RLUTIL_STRING_T
typedef const char* RLUTIL_STRING_T;
#endif // RLUTIL_STRING_T

#define RLUTIL_PRINT(st) printf("%s", st)
typedef stdout RLUTIL_OUTPUT_STREAM;
typedef stderr RLUTIL_ERROR_STREAM;

#define RLUTIL_PRINT(fp, st) fprintf(fp, "%s", st)
#endif // __cplusplus

/**
Expand Down Expand Up @@ -392,6 +398,7 @@ RLUTIL_INLINE int getkey(void) {
case 'B': return KEY_DOWN;
case 'C': return KEY_RIGHT;
case 'D': return KEY_LEFT;
default: ;
}
} else return KEY_ESCAPE;
}
Expand Down Expand Up @@ -456,7 +463,13 @@ RLUTIL_INLINE RLUTIL_STRING_T getANSIBackgroundColor(const int c) {
/// Don't change the background color
///
/// See <Color Codes>
RLUTIL_INLINE void setColor(int c) {
RLUTIL_INLINE void setColor(int c,
#ifdef __cplusplus
std::ostream& fp = std::cout
#else
FILE* fp = stdout
#endif
) {
#if defined(_WIN32) && !defined(RLUTIL_USE_ANSI)
HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
CONSOLE_SCREEN_BUFFER_INFO csbi;
Expand All @@ -465,7 +478,7 @@ RLUTIL_INLINE void setColor(int c) {

SetConsoleTextAttribute(hConsole, (csbi.wAttributes & 0xFFF0) | (WORD)c); // Foreground colors take up the least significant byte
#else
RLUTIL_PRINT(getANSIColor(c));
RLUTIL_PRINT(fp, getANSIColor(c));
#endif
}

Expand All @@ -474,7 +487,13 @@ RLUTIL_INLINE void setColor(int c) {
/// Don't change the foreground color
///
/// See <Color Codes>
RLUTIL_INLINE void setBackgroundColor(int c) {
RLUTIL_INLINE void setBackgroundColor(int c,
#ifdef __cplusplus
std::ostream& fp = std::cout
#else
FILE* fp = stdout
#endif
) {
#if defined(_WIN32) && !defined(RLUTIL_USE_ANSI)
HANDLE hConsole = GetStdHandle(STD_OUTPUT_HANDLE);
CONSOLE_SCREEN_BUFFER_INFO csbi;
Expand All @@ -483,7 +502,7 @@ RLUTIL_INLINE void setBackgroundColor(int c) {

SetConsoleTextAttribute(hConsole, (csbi.wAttributes & 0xFF0F) | (((WORD)c) << 4)); // Background colors take up the second-least significant byte
#else
RLUTIL_PRINT(getANSIBackgroundColor(c));
RLUTIL_PRINT(fp, getANSIBackgroundColor(c));
#endif
}

Expand Down Expand Up @@ -517,11 +536,17 @@ RLUTIL_INLINE int saveDefaultColor(void) {
/// See <Color Codes>
/// See <setColor>
/// See <saveDefaultColor>
RLUTIL_INLINE void resetColor(void) {
RLUTIL_INLINE void resetColor(
#ifdef __cplusplus
std::ostream& fp = std::cout
#else
FILE* fp = stdout
#endif
) {
#if defined(_WIN32) && !defined(RLUTIL_USE_ANSI)
SetConsoleTextAttribute(GetStdHandle(STD_OUTPUT_HANDLE), (WORD)saveDefaultColor());
#else
RLUTIL_PRINT(ANSI_ATTRIBUTE_RESET);
RLUTIL_PRINT(fp, ANSI_ATTRIBUTE_RESET);
#endif
}

Expand All @@ -544,8 +569,8 @@ RLUTIL_INLINE void cls(void) {

SetConsoleCursorPosition(hConsole, coordScreen);
#else
RLUTIL_PRINT(ANSI_CLS);
RLUTIL_PRINT(ANSI_CURSOR_HOME);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, ANSI_CLS);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, ANSI_CURSOR_HOME);
#endif
}

Expand All @@ -560,11 +585,11 @@ RLUTIL_INLINE void locate(int x, int y) {
SetConsoleCursorPosition(GetStdHandle(STD_OUTPUT_HANDLE), coord);
#else // _WIN32 || USE_ANSI
#ifdef __cplusplus
RLUTIL_PRINT("\033[" << y << ";" << x << "H");
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, "\033[" << y << ";" << x << "H");
#else // __cplusplus
char buf[32];
sprintf(buf, "\033[%d;%df", y, x);
RLUTIL_PRINT(buf);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, buf);
#endif // __cplusplus
#endif // _WIN32 || USE_ANSI
}
Expand All @@ -587,13 +612,13 @@ RLUTIL_INLINE void setString(RLUTIL_STRING_T str) {
GetConsoleScreenBufferInfo(hConsoleOutput, &csbi);
WriteConsoleOutputCharacterA(hConsoleOutput, str, len, csbi.dwCursorPosition, &numberOfCharsWritten);
#else // _WIN32 || USE_ANSI
RLUTIL_PRINT(str);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, str);
#ifdef __cplusplus
RLUTIL_PRINT("\033[" << len << 'D');
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, "\033[" << len << 'D');
#else // __cplusplus
char buf[3 + 20 + 1]; // 20 = max length of 64-bit unsigned int when printed as dec
sprintf(buf, "\033[%uD", len);
RLUTIL_PRINT(buf);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, buf);
#endif // __cplusplus
#endif // _WIN32 || USE_ANSI
}
Expand All @@ -615,7 +640,7 @@ RLUTIL_INLINE void setCursorVisibility(char visible) {
structCursorInfo.bVisible = (visible ? TRUE : FALSE);
SetConsoleCursorInfo( hConsoleOutput, &structCursorInfo );
#else // _WIN32 || USE_ANSI
RLUTIL_PRINT((visible ? ANSI_CURSOR_SHOW : ANSI_CURSOR_HIDE));
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, (visible ? ANSI_CURSOR_SHOW : ANSI_CURSOR_HIDE));
#endif // _WIN32 || USE_ANSI
}

Expand Down Expand Up @@ -707,11 +732,11 @@ RLUTIL_INLINE void anykey() {
}

template <class T> void anykey(const T& msg) {
RLUTIL_PRINT(msg);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, msg);
#else
RLUTIL_INLINE void anykey(RLUTIL_STRING_T msg) {
if (msg)
RLUTIL_PRINT(msg);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, msg);
#endif // __cplusplus
getch();
}
Expand All @@ -726,9 +751,9 @@ RLUTIL_INLINE void setConsoleTitle(RLUTIL_STRING_T title) {
#if defined(_WIN32) && !defined(RLUTIL_USE_ANSI)
SetConsoleTitleA(true_title);
#else
RLUTIL_PRINT(ANSI_CONSOLE_TITLE_PRE);
RLUTIL_PRINT(true_title);
RLUTIL_PRINT(ANSI_CONSOLE_TITLE_POST);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, ANSI_CONSOLE_TITLE_PRE);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, true_title);
RLUTIL_PRINT(RLUTIL_OUTPUT_STREAM, ANSI_CONSOLE_TITLE_POST);
#endif // defined(_WIN32) && !defined(RLUTIL_USE_ANSI)
}

Expand Down