TheRock/rocm-systems/projects/rocprofiler-sdk/source/lib/att-tool/filenames.hpp

// MIT License
//
// Copyright (c) 2024-2025 Advanced Micro Devices, Inc. All rights reserved.
//
// Permission is hereby granted, free of charge, to any person obtaining a copy
// of this software and associated documentation files (the "Software"), to deal
// in the Software without restriction, including without limitation the rights
// to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
// copies of the Software, and to permit persons to whom the Software is
// furnished to do so, subject to the following conditions:
//
// The above copyright notice and this permission notice shall be included in all
// copies or substantial portions of the Software.
//
// THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
// IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
// FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
// AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
// LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
// OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
// SOFTWARE.

#pragma once

#include "att_lib_wrapper.hpp"

#include <map>
#include <vector>
#include "util.hpp"

namespace rocprofiler
{
namespace att_wrapper
{
class FilenameMgr
{
public:
    struct Coord
    {
        int  se{0};
        int  sm{0};
        int  sl{0};
        int  id{0};
        bool operator==(const Coord& other) const
        {
            return se == other.se && sm == other.sm && sl == other.sl && id == other.id;
        }
        bool operator<(const Coord& other) const
        {
            if(se != other.se) return se < other.se;
            if(sm != other.sm) return sm < other.sm;
            if(sl != other.sl) return sl < other.sl;
            return id < other.id;
        }
    };
    struct TraceFile
    {
        std::string name{};
        size_t      begin{0};
        size_t      end{0};
    };

    using WaveName       = TraceFile;
    using ShaderDataName = TraceFile;
    using OtherSIMDName  = WaveName;

    FilenameMgr(const Fspath& _dir)
    : dir(_dir)
    , filename(_dir / "filenames.json")
    {}
    ~FilenameMgr();

    void addwave(const Fspath& file, Coord coord, size_t start, size_t end);

    void add_other_simd_data(
        int                                                                    se,
        const std::vector<rocprofiler_thread_trace_decoder_inst_other_simd_t>& records);

    /// Write shaderdata records for an SE and track the output file name.
    void add_shaderdata_data(int                                                  se,
                             const rocprofiler_thread_trace_decoder_shaderdata_t* records,
                             size_t                                               count);

    Fspath                                     dir{};
    Fspath                                     filename{};
    std::map<Coord, WaveName>                  streams{};
    std::vector<std::string>                   perfcounters{};
    int                                        gfxip = 9;
    std::map<int, std::vector<OtherSIMDName>>  other_simd_files{};
    std::map<int, std::vector<ShaderDataName>> shaderdata_files{};
};

}  // namespace att_wrapper
}  // namespace rocprofiler