1# Copyright © 2021 Igalia S.L. 2# SPDX-License-Identifier: MIT 3 4import argparse 5import sys 6 7# 8# TODO can we do this with less boilerplate? 9# 10parser = argparse.ArgumentParser() 11parser.add_argument('-p', '--import-path', required=True) 12parser.add_argument('--utrace-src', required=True) 13parser.add_argument('--utrace-hdr', required=True) 14parser.add_argument('--perfetto-hdr', required=True) 15args = parser.parse_args() 16sys.path.insert(0, args.import_path) 17 18 19from u_trace import Header, HeaderScope 20from u_trace import ForwardDecl 21from u_trace import Tracepoint 22from u_trace import TracepointArg as Arg 23from u_trace import TracepointArgStruct as ArgStruct 24from u_trace import utrace_generate 25from u_trace import utrace_generate_perfetto_utils 26 27Header('vk_enum_to_str.h', scope=HeaderScope.SOURCE|HeaderScope.PERFETTO) 28Header('vk_format.h') 29Header('tu_cmd_buffer.h', scope=HeaderScope.SOURCE) 30Header('tu_device.h', scope=HeaderScope.SOURCE) 31 32# we can't use tu_common.h because it includes ir3 headers which are not 33# compatible with C++ 34ForwardDecl('struct tu_cmd_buffer') 35ForwardDecl('struct tu_device') 36ForwardDecl('struct tu_framebuffer') 37ForwardDecl('struct tu_tiling_config') 38 39# List of the default tracepoints enabled. By default tracepoints are enabled, 40# set tp_default_enabled=False to disable them by default. 41tu_default_tps = [] 42 43# 44# Tracepoint definitions: 45# 46 47def begin_end_tp(name, args=[], tp_struct=None, tp_print=None, 48 tp_default_enabled=True, marker_tp=True, 49 queue_tp=True): 50 global tu_default_tps 51 if tp_default_enabled: 52 tu_default_tps.append(name) 53 Tracepoint('start_{0}'.format(name), 54 toggle_name=name, 55 args=args, 56 tp_struct=tp_struct, 57 tp_perfetto='tu_perfetto_start_{0}'.format(name) if queue_tp else None, 58 tp_print=tp_print if queue_tp else None, 59 tp_markers='tu_cs_trace_start' if marker_tp else None) 60 Tracepoint('end_{0}'.format(name), 61 toggle_name=name, 62 tp_perfetto='tu_perfetto_end_{0}'.format(name), 63 tp_markers='tu_cs_trace_end' if marker_tp else None) 64 65begin_end_tp('cmd_buffer', 66 args=[ArgStruct(type='const struct tu_cmd_buffer *', var='cmd')], 67 tp_struct=[Arg(type='VkCommandBufferLevel', name='level', var='cmd->vk.level', c_format='%s', to_prim_type='vk_CommandBufferLevel_to_str({})'), 68 Arg(type='uint8_t', name='render_pass_continue', var='!!(cmd->usage_flags & VK_COMMAND_BUFFER_USAGE_RENDER_PASS_CONTINUE_BIT)', c_format='%u')]) 69 70begin_end_tp('render_pass', 71 args=[ArgStruct(type='const struct tu_framebuffer *', var='fb'), 72 ArgStruct(type='const struct tu_tiling_config *', var='tiling')], 73 tp_struct=[Arg(type='uint16_t', name='width', var='fb->width', c_format='%u'), 74 Arg(type='uint16_t', name='height', var='fb->height', c_format='%u'), 75 Arg(type='uint8_t', name='attachment_count', var='fb->attachment_count', c_format='%u'), 76 # Arg(type='uint8_t', name='samples', var='fb->samples', c_format='%u'), 77 Arg(type='uint16_t', name='numberOfBins', var='tiling->tile_count.width * tiling->tile_count.height', c_format='%u'), 78 Arg(type='uint16_t', name='binWidth', var='tiling->tile0.width', c_format='%u'), 79 Arg(type='uint16_t', name='binHeight', var='tiling->tile0.height', c_format='%u')]) 80 81begin_end_tp('binning_ib') 82begin_end_tp('draw_ib_sysmem') 83begin_end_tp('draw_ib_gmem') 84 85begin_end_tp('gmem_clear', 86 args=[Arg(type='enum VkFormat', var='format', c_format='%s', to_prim_type='vk_format_description({})->short_name'), 87 Arg(type='uint8_t', var='samples', c_format='%u')]) 88 89begin_end_tp('sysmem_clear', 90 args=[Arg(type='enum VkFormat', var='format', c_format='%s', to_prim_type='vk_format_description({})->short_name'), 91 Arg(type='uint8_t', var='uses_3d_ops', c_format='%u'), 92 Arg(type='uint8_t', var='samples', c_format='%u')]) 93 94begin_end_tp('sysmem_clear_all', 95 args=[Arg(type='uint8_t', var='mrt_count', c_format='%u'), 96 Arg(type='uint8_t', var='rect_count', c_format='%u')]) 97 98begin_end_tp('gmem_load', 99 args=[Arg(type='enum VkFormat', var='format', c_format='%s', to_prim_type='vk_format_description({})->short_name'), 100 Arg(type='uint8_t', var='force_load', c_format='%u')]) 101 102begin_end_tp('gmem_store', 103 args=[Arg(type='enum VkFormat', var='format', c_format='%s', to_prim_type='vk_format_description({})->short_name'), 104 Arg(type='uint8_t', var='fast_path', c_format='%u'), 105 Arg(type='uint8_t', var='unaligned', c_format='%u')]) 106 107begin_end_tp('sysmem_resolve', 108 args=[Arg(type='enum VkFormat', var='format', c_format='%s', to_prim_type='vk_format_description({})->short_name')]) 109 110begin_end_tp('blit', 111 # TODO: add source megapixels count and target megapixels count arguments 112 args=[Arg(type='uint8_t', var='uses_3d_blit', c_format='%u'), 113 Arg(type='enum VkFormat', var='src_format', c_format='%s', to_prim_type='vk_format_description({})->short_name'), 114 Arg(type='enum VkFormat', var='dst_format', c_format='%s', to_prim_type='vk_format_description({})->short_name'), 115 Arg(type='uint8_t', var='layers', c_format='%u')]) 116 117begin_end_tp('compute', 118 args=[Arg(type='uint8_t', var='indirect', c_format='%u'), 119 Arg(type='uint16_t', var='local_size_x', c_format='%u'), 120 Arg(type='uint16_t', var='local_size_y', c_format='%u'), 121 Arg(type='uint16_t', var='local_size_z', c_format='%u'), 122 Arg(type='uint16_t', var='num_groups_x', c_format='%u'), 123 Arg(type='uint16_t', var='num_groups_y', c_format='%u'), 124 Arg(type='uint16_t', var='num_groups_z', c_format='%u')]) 125 126 127# Annotations for Cmd(Begin|End)DebugUtilsLabelEXT 128for suffix in ["", "_rp"]: 129 begin_end_tp('cmd_buffer_annotation' + suffix, 130 args=[ArgStruct(type='unsigned', var='len'), 131 ArgStruct(type='const char *', var='str'),], 132 tp_struct=[Arg(type='uint8_t', name='dummy', var='0', c_format='%hhu'), 133 Arg(type='char', name='str', var='str', c_format='%s', length_arg='len + 1', copy_func='strncpy'),]) 134 135utrace_generate(cpath=args.utrace_src, 136 hpath=args.utrace_hdr, 137 ctx_param='struct tu_device *dev', 138 trace_toggle_name='tu_gpu_tracepoint', 139 trace_toggle_defaults=tu_default_tps) 140utrace_generate_perfetto_utils(hpath=args.perfetto_hdr) 141