Cog/Plugins/Cog/Source/CogEngine/Private/CogEngineWindow_OutputLog.cpp

#include "CogEngineWindow_OutputLog.h"

#include "CogDebugHelper.h"
#include "CogImguiHelper.h"
#include "CogWindowWidgets.h"
#include "Engine/Engine.h"
#include "HAL/PlatformApplicationMisc.h"
#include "Misc/StringBuilder.h"

//--------------------------------------------------------------------------------------------------------------------------
void FCogEngineWindow_OutputLog::Initialize()
{    
    Super::Initialize();

    bHasMenu = true;
    OutputDevice.OutputLog = this;

    Config = GetConfig<UCogEngineConfig_OutputLog>();
}

//--------------------------------------------------------------------------------------------------------------------------
void FCogEngineWindow_OutputLog::RenderHelp()
{
    ImGui::Text(
    "This window output the log based on each log categories verbosity. "
    "The verbosity of each log category can be configured in the 'Log Categories' window. "
    );
}

//--------------------------------------------------------------------------------------------------------------------------
void FCogEngineWindow_OutputLog::Clear()
{
    TextBuffer.clear();
    LineInfos.Empty();
}

//--------------------------------------------------------------------------------------------------------------------------
void FCogEngineWindow_OutputLog::AddLog(const TCHAR* Message, ELogVerbosity::Type Verbosity, const class FName& Category)
{
    static TAnsiStringBuilder<512> Format;

    Format.Reset();

    if (Message)
    {
        Format.Append(Message);
    }

    FLineInfo& LineInfo = LineInfos.AddDefaulted_GetRef();
    LineInfo.Frame = GFrameCounter % 1000;
    LineInfo.Verbosity = Verbosity;
    LineInfo.Category = Category;
    LineInfo.Start = TextBuffer.size();

    TextBuffer.append(Format.GetData(), Format.GetData() + Format.Len());

    LineInfo.End = TextBuffer.size();
}

//--------------------------------------------------------------------------------------------------------------------------
void FCogEngineWindow_OutputLog::DrawRow(const char* BufferStart, const FLineInfo& LineInfo, bool IsTableShown) const
{
    ImU32 Color;
    switch (LineInfo.Verbosity)
    {
        case ELogVerbosity::Error:      Color = FCogImguiHelper::ToImColor(Config->ErrorColor); break;
        case ELogVerbosity::Warning:    Color = FCogImguiHelper::ToImColor(Config->WarningColor); break;
        default:                        Color = FCogImguiHelper::ToImColor(Config->DefaultColor); break;
    }

    ImGui::PushStyleColor(ImGuiCol_Text, Color);

    if (IsTableShown)
    {
        ImGui::TableNextRow();

        if (Config->ShowFrame)
        {
            ImGui::TableNextColumn();
            ImGui::Text("%3d", LineInfo.Frame);
        }

        if (Config->ShowCategory)
        {
            ImGui::TableNextColumn();
            ImGui::Text("%s", TCHAR_TO_ANSI(*LineInfo.Category.ToString()));
        }

        if (Config->ShowVerbosity)
        {
            ImGui::TableNextColumn();
            ImGui::Text("%s", TCHAR_TO_ANSI(ToString(LineInfo.Verbosity)));
        }

        ImGui::TableNextColumn();
        const char* LineStart = BufferStart + LineInfo.Start;
        const char* LineEnd = BufferStart + LineInfo.End;
        ImGui::TextUnformatted(LineStart, LineEnd);
    }
    else
    {
        if (Config->ShowFrame)
        {
            ImGui::Text("[%3d] ", LineInfo.Frame);
            ImGui::SameLine();
        }

        if (Config->ShowCategory)
        {
            ImGui::Text("%s: ", TCHAR_TO_ANSI(*LineInfo.Category.ToString()));
            ImGui::SameLine();
        }

        if (Config->ShowVerbosity)
        {
            ImGui::Text("%s: ", TCHAR_TO_ANSI(ToString(LineInfo.Verbosity)));
            ImGui::SameLine();
        }

        const char* LineStart = BufferStart + LineInfo.Start;
        const char* LineEnd = BufferStart + LineInfo.End;
        ImGui::TextUnformatted(LineStart, LineEnd);
    }

    ImGui::PopStyleColor();
}

//--------------------------------------------------------------------------------------------------------------------------
void FCogEngineWindow_OutputLog::Copy() const
{
    const auto Buffer = StringCast<TCHAR>(TextBuffer.c_str());
    const wchar_t* BufferData = Buffer.Get();
    if (BufferData == nullptr)
    { return; }

    FStringBuilderBase StringBuilder;
    for (const FLineInfo& LineInfo : LineInfos)
    {
        StringBuilder.Append(FString::Printf(TEXT("[%3d] [%s] [%s] "), LineInfo.Frame, *LineInfo.Category.ToString(), ToString(LineInfo.Verbosity)));
        StringBuilder.Append(BufferData + LineInfo.Start, LineInfo.End - LineInfo.Start);
        StringBuilder.Append("\n");
    };
    
    FPlatformApplicationMisc::ClipboardCopy(StringBuilder.ToString());
}

//--------------------------------------------------------------------------------------------------------------------------
void FCogEngineWindow_OutputLog::RenderContent()
{
    Super::RenderContent();

    if (ImGui::BeginMenuBar())
    {
        if (ImGui::BeginMenu("Options"))
        {
            if (ImGui::MenuItem("Copy"))
            {
                Copy();
            }
            ImGui::Separator();

            ImGui::Checkbox("Auto Scroll", &Config->AutoScroll);
            ImGui::Checkbox("Show Frame", &Config->ShowFrame);
            ImGui::Checkbox("Show Category", &Config->ShowCategory);
            ImGui::Checkbox("Show Verbosity", &Config->ShowVerbosity);
            ImGui::Checkbox("Show As Table", &Config->ShowAsTable);

            ImGui::Separator();
            
            constexpr ImGuiColorEditFlags ColorEditFlags = ImGuiColorEditFlags_NoInputs | ImGuiColorEditFlags_AlphaPreviewHalf;
            FCogImguiHelper::ColorEdit4("Default Color", Config->DefaultColor, ColorEditFlags);
            FCogImguiHelper::ColorEdit4("Warning Color", Config->WarningColor, ColorEditFlags);
            FCogImguiHelper::ColorEdit4("Error Color", Config->ErrorColor, ColorEditFlags);

            ImGui::Separator();

            if (ImGui::Button("Reset Settings", ImVec2(-1, 0)))
            {
                ResetConfig();
            }
            
            ImGui::EndMenu();
        }

        ImGui::SameLine();

        if (ImGui::MenuItem("Clear"))
        {
            Clear();
        }

        ImGui::SameLine();

        ImGui::SetNextItemWidth(ImGui::GetFontSize() * 9);
        if (ImGui::BeginCombo("##Verbosity", FCogDebugHelper::VerbosityToString((ELogVerbosity::Type)Config->VerbosityFilter)))
        {
            for (int32 i = ELogVerbosity::Error; i <= (int32)ELogVerbosity::VeryVerbose; ++i)
            {
	            const bool IsSelected = i == Config->VerbosityFilter;
	            const ELogVerbosity::Type Verbosity = (ELogVerbosity::Type)i;

                if (ImGui::Selectable(FCogDebugHelper::VerbosityToString(Verbosity), IsSelected))
                {
                    Config->VerbosityFilter = i;
                }
            }
            ImGui::EndCombo();
        }

        FCogWindowWidgets::SearchBar("##Filter", Filter);

        ImGui::EndMenuBar();
    }

    int32 ColumnCount = 1;
    ColumnCount += (int32)Config->ShowFrame;
    ColumnCount += (int32)Config->ShowCategory;
    ColumnCount += (int32)Config->ShowVerbosity;

    bool IsTableShown = false;
    if (Config->ShowAsTable)
    {
        if (ImGui::BeginTable("LogTable", ColumnCount, ImGuiTableFlags_SizingFixedFit | ImGuiTableFlags_Resizable | ImGuiTableFlags_BordersOuter | ImGuiTableFlags_BordersV | ImGuiTableFlags_ScrollX))
        {
            IsTableShown = true;
            if (Config->ShowFrame)
            {
                ImGui::TableSetupColumn("Frame", ImGuiTableColumnFlags_WidthFixed, FCogWindowWidgets::GetFontWidth() * 4);
            }

            if (Config->ShowCategory)
            {
                ImGui::TableSetupColumn("Category", ImGuiTableColumnFlags_WidthFixed, FCogWindowWidgets::GetFontWidth() * 10);
            }

            if (Config->ShowVerbosity)
            {
                ImGui::TableSetupColumn("Verbosity", ImGuiTableColumnFlags_WidthFixed, FCogWindowWidgets::GetFontWidth() * 10);
            }

            ImGui::TableSetupColumn("Message", ImGuiTableColumnFlags_WidthStretch);
        }
    }

    if (IsTableShown == false)
    {
        ImGui::BeginChild("Scrolling", ImVec2(0, 0), false, ImGuiWindowFlags_AlwaysVerticalScrollbar | ImGuiWindowFlags_AlwaysHorizontalScrollbar);
    }

    const char* BufferStart = TextBuffer.begin();

    if (Filter.IsActive())
    {
        for (int32 LineIndex = 0; LineIndex < LineInfos.Num(); LineIndex++)
        {
            const FLineInfo& LineInfo = LineInfos[LineIndex];
            const char* LineStart = BufferStart + LineInfo.Start;
            const char* LineEnd = BufferStart + LineInfo.End;
            if (Filter.PassFilter(LineStart, LineEnd))
            {
                DrawRow(BufferStart, LineInfo, IsTableShown);
            }
        }
    }
    else if (Config->VerbosityFilter != ELogVerbosity::VeryVerbose)
    {
        for (int32 LineIndex = 0; LineIndex < LineInfos.Num(); LineIndex++)
        {
            const FLineInfo& LineInfo = LineInfos[LineIndex];

            if (LineInfo.Verbosity <= (ELogVerbosity::Type)Config->VerbosityFilter)
            {
                DrawRow(BufferStart, LineInfo, IsTableShown);
            }
        }
    }
    else
    {
        ImGuiListClipper Clipper;
        Clipper.Begin(LineInfos.Num());
        while (Clipper.Step())
        {
            for (int32 LineIndex = Clipper.DisplayStart; LineIndex < Clipper.DisplayEnd; LineIndex++)
            {
                if (LineInfos.IsValidIndex(LineIndex))
                {
                    const FLineInfo& LineInfo = LineInfos[LineIndex];
                    DrawRow(BufferStart, LineInfo, IsTableShown);
                }
            }
        }
        Clipper.End();
    }

    if (Config->AutoScroll && ImGui::GetScrollY() >= ImGui::GetScrollMaxY())
    {
        ImGui::SetScrollHereY(1.0f);
    }

    if (IsTableShown)
    {
        ImGui::EndTable();
    }
    else
    {
        ImGui::EndChild();
    }

    if (ImGui::BeginPopupContextItem("OutputLog"))
    {
        if (ImGui::MenuItem("Clear"))
        {
            Clear();
        }

        if (ImGui::MenuItem("Copy"))
        {
            Copy();
        }

        ImGui::EndPopup();
    }
}

//--------------------------------------------------------------------------------------------------------------------------
// FCogLogOutputDevice
//--------------------------------------------------------------------------------------------------------------------------
FCogLogOutputDevice::FCogLogOutputDevice()
{
    GLog->AddOutputDevice(this);
}

//--------------------------------------------------------------------------------------------------------------------------
FCogLogOutputDevice::~FCogLogOutputDevice()
{
    if (GLog != nullptr)
    {
        GLog->RemoveOutputDevice(this);
    }
}

//--------------------------------------------------------------------------------------------------------------------------
void FCogLogOutputDevice::Serialize(const TCHAR* Message, const ELogVerbosity::Type Verbosity, const FName& Category)
{
    if (OutputLog != nullptr)
    {
        OutputLog->AddLog(Message, Verbosity, Category);
    }
}