// Copyright (c) 2008-2023 the Urho3D project
// License: MIT

#include "../Precompiled.h"

#include "../Core/Profiler.h"

#include <cstdio>

#include "../DebugNew.h"

namespace Urho3D
{

Profiler::Profiler(Context* context) :
    Object(context),
    current_(nullptr),
    root_(nullptr),
    intervalFrames_(0)
{
    current_ = root_ = new ProfilerBlock(nullptr, "RunFrame");
}

Profiler::~Profiler()
{
    delete root_;
    root_ = nullptr;
}

void Profiler::BeginFrame()
{
    // End the previous frame if any
    if (root_->count_)
        EndFrame();

    root_->Begin();
}

void Profiler::EndFrame()
{
    EndBlock();
    ++intervalFrames_;
    root_->EndFrame();
    current_ = root_;
}

void Profiler::BeginInterval()
{
    root_->BeginInterval();
    intervalFrames_ = 0;
}

const String& Profiler::PrintData(bool showUnused, bool showTotal, unsigned maxDepth) const
{
    static String output;

    if (!showTotal)
        output  = "Block                            Cnt     Avg      Max     Frame     Total\n\n";
    else
    {
        output  = "Block                                       Last frame                       Whole execution time\n\n";
        output += "                                 Cnt     Avg      Max      Total      Cnt      Avg       Max        Total\n\n";
    }

    if (!maxDepth)
        maxDepth = 1;

    PrintData(root_, output, 0, maxDepth, showUnused, showTotal);

    return output;
}

void Profiler::PrintData(ProfilerBlock* block, String& output, unsigned depth, unsigned maxDepth, bool showUnused,
    bool showTotal) const
{
    static const int LINE_MAX_LENGTH = 256;
    static const int NAME_MAX_LENGTH = 30;

    char line[LINE_MAX_LENGTH];
    char indentedName[LINE_MAX_LENGTH];

    if (depth >= maxDepth)
        return;

    // Do not print any block that does not collect critical data
    if (showUnused || block->intervalCount_ || (showTotal && block->totalCount_))
    {
        memset(indentedName, ' ', NAME_MAX_LENGTH);
        indentedName[depth++] = 0;
        strncat(indentedName, block->name_, NAME_MAX_LENGTH - depth);
        indentedName[strlen(indentedName)] = ' ';
        indentedName[NAME_MAX_LENGTH] = 0;

        if (!showTotal)
        {
            float avg = (float)block->intervalTime_ / block->intervalCount_ / 1000.0f;
            float max = block->intervalMaxTime_ / 1000.0f;
            float frame = (float)block->intervalTime_ / (intervalFrames_ ? intervalFrames_ : 1) / 1000.0f;
            float all = block->intervalTime_ / 1000.0f;

            sprintf(line, "%s %5u %8.3f %8.3f %8.3f %9.3f\n", indentedName, Min(block->intervalCount_, 99999U),
                avg, max, frame, all);
        }
        else
        {
            float avg = (block->frameCount_ ? (float)block->frameTime_ / block->frameCount_ : 0.0f) / 1000.0f;
            float max = block->frameMaxTime_ / 1000.0f;
            float all = block->frameTime_ / 1000.0f;

            float totalAvg = (float)block->totalTime_ / block->totalCount_ / 1000.0f;
            float totalMax = block->totalMaxTime_ / 1000.0f;
            float totalAll = block->totalTime_ / 1000.0f;

            sprintf(line, "%s %5u %8.3f %8.3f %9.3f  %7u %9.3f %9.3f %11.3f\n", indentedName, Min(block->frameCount_, 99999U),
                avg, max, all, Min(block->totalCount_, 99999U), totalAvg, totalMax, totalAll);
        }

        output += String(line);
    }

    for (Vector<ProfilerBlock*>::ConstIterator i = block->children_.Begin(); i != block->children_.End(); ++i)
        PrintData(*i, output, depth, maxDepth, showUnused, showTotal);
}

}