/*
* Copyright 2014-2021 NVIDIA Corporation.  All rights reserved.
*
* Licensed under the Apache License, Version 2.0 (the "License");
* you may not use this file except in compliance with the License.
* You may obtain a copy of the License at
*
*     http://www.apache.org/licenses/LICENSE-2.0
*
* Unless required by applicable law or agreed to in writing, software
* distributed under the License is distributed on an "AS IS" BASIS,
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
* See the License for the specific language governing permissions and
* limitations under the License.
*/
#pragma once

#include <stdint.h>
#include <vector>
#include "NvPerfInit.h"
#include "NvPerfMetricsConfigBuilder.h"

namespace nv { namespace perf {

    struct CounterConfiguration
    {
        std::vector<uint8_t> configImage;
        std::vector<uint8_t> counterDataPrefix;
        size_t numPipelinedPasses;
        size_t numIsolatedPasses;
    };

    /// Transforms configBuilder into configuration.
    inline bool CreateConfiguration(
        MetricsConfigBuilder& configBuilder,
        CounterConfiguration& configuration)
    {
        bool res = false;
        res = configBuilder.PrepareConfigImage();
        if (!res)
        {
            //std::cerr << "FAILED: D3D12CreateConfiguration - failed PrepareConfigImage\n";
            return false;
        }

        const size_t configImageSize = configBuilder.GetConfigImageSize();
        if (!configImageSize)
        {
            // std::cerr << "FAILED: GetConfigImageSize - failed PrepareConfigImage\n";
            return false;
        }
        configuration.configImage.resize(configImageSize);
        if (!configBuilder.GetConfigImage(configuration.configImage.size(), &configuration.configImage[0]))
        {
            //std::cerr << "FAILED: GetConfigImage - failed PrepareConfigImage\n";
            return false;
        }

        const size_t counterDataPrefixSize = configBuilder.GetCounterDataPrefixSize();
        if (!counterDataPrefixSize)
        {
            //std::cerr << "FAILED: GetCounterDataPrefixSize - failed PrepareConfigImage\n";
            return false;
        }
        configuration.counterDataPrefix.resize(counterDataPrefixSize);
        if (!configBuilder.GetCounterDataPrefix(configuration.counterDataPrefix.size(), &configuration.counterDataPrefix[0]))
        {
            //std::cerr << "FAILED: GetCounterDataPrefix - failed PrepareConfigImage\n";
            return false;
        }

        NVPW_Config_GetNumPasses_Params getNumPassesParams = { NVPW_Config_GetNumPasses_Params_STRUCT_SIZE };
        getNumPassesParams.pConfig = &configuration.configImage[0];
        NVPA_Status nvpaStatus = NVPW_Config_GetNumPasses(&getNumPassesParams);
        if (nvpaStatus)
        {
            return false;
        }
        configuration.numPipelinedPasses = getNumPassesParams.numPipelinedPasses;
        configuration.numIsolatedPasses = getNumPassesParams.numIsolatedPasses;

        return true;
    }


    /// Adds pMetricNames[0..numMetrics-1] into configBuilder, then transforms configBuilder into configuration.
    inline bool CreateConfiguration(
        MetricsConfigBuilder& configBuilder,
        size_t numMetrics,
        const char* const pMetricNames[],
        CounterConfiguration& configuration)
    {
        bool succeeded = configBuilder.AddMetrics(pMetricNames, numMetrics);
        if (!succeeded)
        {
            return false;
        }

        succeeded = CreateConfiguration(configBuilder, configuration);
        if (!succeeded)
        {
            return false;
        }
        return true;
    }

}}