port from perforce

ruins64k/tools/NvPerfUtility/include/NvPerfCounterData.h

@@ -0,0 +1,80 @@
+/*
+* 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 "nvperf_host.h"
+#include "nvperf_target.h"
+#include <string>
+#include <vector>
+
+namespace nv { namespace perf {
+    inline size_t CounterDataGetNumRanges(const uint8_t* pCounterDataImage)
+    {
+        NVPW_CounterData_GetNumRanges_Params getNumRangeParams = { NVPW_CounterData_GetRangeDescriptions_Params_STRUCT_SIZE };
+        getNumRangeParams.pCounterDataImage = pCounterDataImage;
+        NVPA_Status nvpaStatus = NVPW_CounterData_GetNumRanges(&getNumRangeParams);
+        if (nvpaStatus)
+        {
+            return 0;
+        }
+        return getNumRangeParams.numRanges;
+    }
+
+    // TODO: this function performs dynamic allocations; either need a non-malloc'ing variant, or move this to an appropriate place
+    inline std::string CounterDataGetRangeName(const uint8_t* pCounterDataImage, size_t rangeIndex, char delimiter, const char** ppLeafName = nullptr)
+    {
+        std::string rangeName;
+
+        NVPW_CounterData_GetRangeDescriptions_Params params = { NVPW_CounterData_GetRangeDescriptions_Params_STRUCT_SIZE };
+        params.pCounterDataImage = pCounterDataImage;
+        params.rangeIndex = rangeIndex;
+        NVPA_Status nvpaStatus = NVPW_CounterData_GetRangeDescriptions(&params);
+        if (nvpaStatus)
+        {
+            return "";
+        }
+
+        if (!params.numDescriptions)
+        {
+            return "";
+        }
+
+        std::vector<const char*> descriptions;
+        descriptions.resize(params.numDescriptions);
+        params.ppDescriptions = descriptions.data();
+        nvpaStatus = NVPW_CounterData_GetRangeDescriptions(&params);
+        if (nvpaStatus)
+        {
+            return "";
+        }
+
+        rangeName += descriptions[0];
+        for (size_t descriptionIdx = 1; descriptionIdx < params.numDescriptions; ++descriptionIdx)
+        {
+            const char* pDescription = params.ppDescriptions[descriptionIdx];
+            rangeName += delimiter;
+            rangeName += pDescription;
+        }
+
+        if (ppLeafName)
+        {
+            *ppLeafName = descriptions.back();
+        }
+
+        return rangeName;
+    }
+}}