/*
 * Copyright (c) Contributors to the Open 3D Engine Project.
 * For complete copyright and license terms please see the LICENSE at the root of this distribution.
 *
 * SPDX-License-Identifier: Apache-2.0 OR MIT
 *
 */

#include "JobDiagnosticTracker.h"

namespace AssetProcessor
{
    bool JobDiagnosticInfo::operator==(const JobDiagnosticInfo& rhs) const
    {
        return m_errorCount == rhs.m_errorCount
            && m_warningCount == rhs.m_warningCount;
    }

    bool JobDiagnosticInfo::operator!=(const JobDiagnosticInfo& rhs) const
    {
        return !operator==(rhs);
    }

    //////////////////////////////////////////////////////////////////////////

    JobDiagnosticTracker::JobDiagnosticTracker()
    {
        BusConnect();
    }

    JobDiagnosticTracker::~JobDiagnosticTracker()
    {
        BusDisconnect();
    }

    JobDiagnosticInfo AssetProcessor::JobDiagnosticTracker::GetDiagnosticInfo(AZ::u64 jobRunKey) const
    {
        auto jobIter = m_jobInfo.find(jobRunKey);

        if(jobIter != m_jobInfo.end())
        {
            return jobIter->second;
        }

        return {};
    }

    void JobDiagnosticTracker::RecordDiagnosticInfo(AZ::u64 jobRunKey, JobDiagnosticInfo info)
    {
        if (info != JobDiagnosticInfo{})
        {
            // Only store non-empty entries
            m_jobInfo[jobRunKey] = info;
        }
    }

    WarningLevel JobDiagnosticTracker::GetWarningLevel() const
    {
        return m_warningLevel;
    }

    void JobDiagnosticTracker::SetWarningLevel(WarningLevel level)
    {
        m_warningLevel = level;
    }
}