| /* |
| * Copyright (C) 2019 The Android Open Source Project |
| * |
| * 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. |
| */ |
| |
| #ifndef SRC_TRACE_PROCESSOR_IMPORTERS_PROTO_GPU_EVENT_PARSER_H_ |
| #define SRC_TRACE_PROCESSOR_IMPORTERS_PROTO_GPU_EVENT_PARSER_H_ |
| |
| #include <optional> |
| #include <vector> |
| |
| #include "perfetto/ext/base/string_writer.h" |
| #include "perfetto/protozero/field.h" |
| #include "protos/perfetto/trace/android/gpu_mem_event.pbzero.h" |
| #include "protos/perfetto/trace/gpu/gpu_render_stage_event.pbzero.h" |
| #include "src/trace_processor/importers/common/args_tracker.h" |
| #include "src/trace_processor/importers/proto/proto_incremental_state.h" |
| #include "src/trace_processor/importers/proto/vulkan_memory_tracker.h" |
| #include "src/trace_processor/storage/trace_storage.h" |
| |
| #include "protos/perfetto/trace/gpu/vulkan_memory_event.pbzero.h" |
| |
| namespace perfetto { |
| |
| namespace protos { |
| namespace pbzero { |
| |
| class GpuRenderStageEvent_Decoder; |
| |
| } // namespace pbzero |
| } // namespace protos |
| |
| namespace trace_processor { |
| |
| class TraceProcessorContext; |
| |
| struct ProtoEnumHasher { |
| template <typename T> |
| std::size_t operator()(T t) const { |
| return static_cast<std::size_t>(t); |
| } |
| }; |
| |
| // Class for parsing graphics related events. |
| class GpuEventParser { |
| public: |
| using ConstBytes = protozero::ConstBytes; |
| using VulkanMemoryEventSource = VulkanMemoryEvent::Source; |
| using VulkanMemoryEventOperation = VulkanMemoryEvent::Operation; |
| explicit GpuEventParser(TraceProcessorContext*); |
| |
| void ParseGpuCounterEvent(int64_t ts, ConstBytes); |
| void ParseGpuRenderStageEvent(int64_t ts, |
| PacketSequenceStateGeneration*, |
| ConstBytes); |
| void ParseGraphicsFrameEvent(int64_t timestamp, ConstBytes); |
| void ParseGpuLog(int64_t ts, ConstBytes); |
| |
| void ParseVulkanMemoryEvent(PacketSequenceStateGeneration*, ConstBytes); |
| void UpdateVulkanMemoryAllocationCounters(UniquePid, |
| const VulkanMemoryEvent::Decoder&); |
| |
| void ParseVulkanApiEvent(int64_t, ConstBytes); |
| |
| void ParseGpuMemTotalEvent(int64_t, ConstBytes); |
| |
| private: |
| const StringId GetFullStageName( |
| PacketSequenceStateGeneration* sequence_state, |
| const protos::pbzero::GpuRenderStageEvent_Decoder& event) const; |
| void InsertGpuTrack( |
| const protos::pbzero:: |
| GpuRenderStageEvent_Specifications_Description_Decoder& hw_queue); |
| std::optional<std::string> FindDebugName(int32_t vk_object_type, |
| uint64_t vk_handle) const; |
| const StringId ParseRenderSubpasses( |
| const protos::pbzero::GpuRenderStageEvent_Decoder& event) const; |
| |
| TraceProcessorContext* const context_; |
| VulkanMemoryTracker vulkan_memory_tracker_; |
| // For GpuCounterEvent |
| std::unordered_map<uint32_t, TrackId> gpu_counter_track_ids_; |
| // For GpuRenderStageEvent |
| const StringId description_id_; |
| const StringId gpu_render_stage_scope_id_; |
| std::vector<std::optional<TrackId>> gpu_hw_queue_ids_; |
| size_t gpu_hw_queue_counter_ = 0; |
| // Map of stage ID -> pair(stage name, stage description) |
| std::vector<std::pair<StringId, StringId>> gpu_render_stage_ids_; |
| // For VulkanMemoryEvent |
| std::unordered_map<VulkanMemoryEvent::AllocationScope, |
| int64_t /*counter_value*/, |
| ProtoEnumHasher> |
| vulkan_driver_memory_counters_; |
| std::unordered_map<uint32_t /*memory_type*/, int64_t /*counter_value*/> |
| vulkan_device_memory_counters_allocate_; |
| std::unordered_map<uint32_t /*memory_type*/, int64_t /*counter_value*/> |
| vulkan_device_memory_counters_bind_; |
| // For GpuLog |
| const StringId gpu_log_track_name_id_; |
| const StringId gpu_log_scope_id_; |
| const StringId tag_id_; |
| const StringId log_message_id_; |
| std::array<StringId, 7> log_severity_ids_; |
| // For Vulkan events. |
| // For VulkanApiEvent.VkDebugUtilsObjectName. |
| // Map of vk handle -> vk object name. |
| using DebugMarkerMap = std::unordered_map<uint64_t, std::string>; |
| // Map of VkObjectType -> DebugMarkerMap. |
| std::unordered_map<int32_t, DebugMarkerMap> debug_marker_names_; |
| // For VulkanApiEvent.VkQueueSubmit. |
| StringId vk_event_track_id_; |
| StringId vk_event_scope_id_; |
| StringId vk_queue_submit_id_; |
| // For GpuMemTotalEvent |
| const StringId gpu_mem_total_name_id_; |
| const StringId gpu_mem_total_unit_id_; |
| const StringId gpu_mem_total_global_desc_id_; |
| const StringId gpu_mem_total_proc_desc_id_; |
| }; |
| } // namespace trace_processor |
| } // namespace perfetto |
| |
| #endif // SRC_TRACE_PROCESSOR_IMPORTERS_PROTO_GPU_EVENT_PARSER_H_ |