TDebugConsole_Win.cpp

Go to the documentation of this file.

#include "ToshiPCH.h"
#include "TDebugConsole_Win.h"
#include "Toshi/TString8.h"
 
#include <conio.h>
#include <windows.h>
 
//-----------------------------------------------------------------------------
// Enables memory debugging.
// Note: Should be the last include!
//-----------------------------------------------------------------------------
#include "Core/TMemoryDebugOn.h"
 
static constexpr const WORD kTypeColours[]{
    FOREGROUND_GREEN,
    FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN,
    FOREGROUND_INTENSITY | FOREGROUND_RED,
    BACKGROUND_RED | FOREGROUND_INTENSITY | FOREGROUND_RED | FOREGROUND_GREEN
};
 
TSTATICASSERT( Toshi::TUtil::LogType_NUMOF == TARRAYSIZE( kTypeColours ) );
 
TOSHI_NAMESPACE_START
 
TDebugConsole::TDebugConsole()
{
    m_HWND    = FindConsole();
    m_Created = m_HWND != TNULL;
 
    m_LogListener.Connect(
        TUtil::GetLogEmitter(),
        this,
        TDebugConsole::OnLog,
        1
    );
}
 
TDebugConsole::~TDebugConsole()
{
    if ( IsCreated() )
    {
        if ( IsVisible() )
        {
            TWARN( "Press any key to exit...\n" );
            _getch();
        }
    }
}
 
TBOOL TDebugConsole::Show( TBOOL state )
{
    TBOOL oldState = m_Visible;
    m_Visible      = state;
 
    ShowWindow( m_HWND, state );
    UpdateWindow( m_HWND );
 
    return oldState;
}
 
TBOOL TDebugConsole::Toggle()
{
    Show( !m_Visible );
    return m_Visible;
}
 
HWND TDebugConsole::FindConsole() const
{
    TString8 str = TString8( TDebugConsoleUUID );
 
    // Generate title for the console so we can find it's HWND
    LARGE_INTEGER time;
    QueryPerformanceCounter( &time );
    TCHAR timeHex[ sizeof( time.QuadPart ) + 1 ] = { 0 };
    _ultoa_s( (unsigned long)time.QuadPart, timeHex, 0x10 );
    str += timeHex;
 
    // Save the current title
    TCHAR lpConsoleTitle[ 256 ];
    DWORD titleLength = GetConsoleTitleA( lpConsoleTitle, 256 );
    if ( titleLength == 0 ) { return TNULL; }
 
    // Set the genereated title and wait for 50ms so it's applied
    SetConsoleTitleA( str );
    ThreadSleep( 50 );
 
    // Find a window with the generated title and restore it
    HWND consoleHWND = FindWindowA( TNULL, str );
    SetConsoleTitleA( lpConsoleTitle );
 
    return consoleHWND;
}
 
static TBOOL s_bIsSimpleMode = TFALSE;
 
TBOOL __stdcall TDebugConsole::OnLog( TDebugConsole* a_pCaller, TUtil* a_pOwner, TUtil::LogEvent* pLogEvent )
{
    TASSERT( pLogEvent );
 
    HANDLE hStd;
    WORD   wOldColorAttrs;
 
    hStd = GetStdHandle( STD_OUTPUT_HANDLE );
 
    CONSOLE_SCREEN_BUFFER_INFO csbiInfo;
    GetConsoleScreenBufferInfo( hStd, &csbiInfo );
    wOldColorAttrs = csbiInfo.wAttributes;
 
    SetConsoleTextAttribute( hStd, kTypeColours[ pLogEvent->m_eType ] );
 
    if ( s_bIsSimpleMode )
    {
        TDebug_FinalPrintf(
            "[%s]: %s",
            TUtil::LogTypeToString( pLogEvent->m_eType ),
            pLogEvent->m_szString
        );
    }
    else
    {
        auto& appParams = TUtil::GetToshiParameters();
 
        TCHAR strTime[ 9 ];
        _strtime( strTime );
 
        TDebug_FinalPrintf(
            "[%s] [%s/%s] [%s]: %s",
            strTime,
            appParams.szLogAppName,
            appParams.szLogAppDirName != TNULL ? appParams.szLogAppDirName : "",
            TUtil::LogTypeToString( pLogEvent->m_eType ),
            pLogEvent->m_szString
        );
    }
 
    SetConsoleTextAttribute( hStd, wOldColorAttrs );
 
    return true;
}
 
TOSHI_NAMESPACE_END