console_helper.h

Go to the documentation of this file.

#ifndef OOKII_CONSOLE_HELPER_H_
#define OOKII_CONSOLE_HELPER_H_
 
#pragma once
 
#if !defined(OOKII_NO_PLATFORM_HEADERS) && (!defined(OOKII_CONSOLE_NOT_INLINE) || defined(OOKII_CONSOLE_DEFINITION))
 
#if _WIN32
#define WIN32_LEAN_AND_MEAN
#define NOMINMAX
#define NOSERVICE
#define NOMCX
#define NOIME
#define NOSOUND
#define NOCOMM
#define NOKANJI
#define NORPC
#define NOPROXYSTUB
#define NOIMAGE
#define NOTAPE
#include <Windows.h>
#include <io.h>
#else
#include <unistd.h>
#include <sys/ioctl.h>
#endif
 
#include <iostream>
 
#endif
 
namespace ookii
{
#ifndef OOKII_CONSOLE_NOT_INLINE
    inline
#endif
        short get_console_width(short default_width = 80) noexcept
#if defined(OOKII_CONSOLE_NOT_INLINE) && !defined(OOKII_CONSOLE_DEFINITION)
        ;
#else
    {
#if _WIN32
 
        HANDLE handle = GetStdHandle(STD_OUTPUT_HANDLE);
        CONSOLE_SCREEN_BUFFER_INFO info;
        if (GetConsoleScreenBufferInfo(handle, &info))
            return info.srWindow.Right - info.srWindow.Left + 1;
 
#elif defined(TIOCGWINSZ)
 
        struct winsize ws;
 
        if (ioctl(STDOUT_FILENO, TIOCGWINSZ, &ws) == 0)
            return static_cast<short>(ws.ws_col);
 
#endif
 
        return default_width;
    }
#endif
 
    template<typename CharType>
    struct console_stream
    {
    };
 
    template<>
    struct console_stream<char>
    {
        static inline std::ostream &cout()
        {
            return std::cout;
        };
 
        static inline std::ostream &cerr()
        {
            return std::cerr;
        }
 
        static inline std::istream &cin()
        {
            return std::cin;
        }
    };
 
    template<>
    struct console_stream<wchar_t>
    {
        static inline std::wostream &cout()
        {
            return std::wcout;
        };
 
        static inline std::wostream &cerr()
        {
            return std::wcerr;
        }
 
        static inline std::wistream &cin()
        {
            return std::wcin;
        }
    };
 
}
 
#endif