From 2ded7736166ffb25373eec83f288d20ac6f51462 Mon Sep 17 00:00:00 2001
From: Konstantin Seurer <konstantin.seurer@gmail.com>
Date: Mon, 23 Feb 2026 14:28:58 +0100
Subject: [PATCH] util/u_trace: Add an option to emit additional code

radv will generate additional wrapper functions for the tracepoints.

Reviewed-by: Danylo Piliaiev <dpiliaiev@igalia.com>
Part-of: <https://gitlab.freedesktop.org/mesa/mesa/-/merge_requests/41271>
---
 src/util/perf/u_trace.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/src/util/perf/u_trace.py b/src/util/perf/u_trace.py
index 4fb8d3cb43e..82d93f8b02b 100644
--- a/src/util/perf/u_trace.py
+++ b/src/util/perf/u_trace.py
@@ -373,6 +373,8 @@ static ALWAYS_INLINE void trace_${trace_name}(
 }
 #endif
 
+${additional_code}
+
 #endif /* ${guard_name} */
 """
 
@@ -608,7 +610,7 @@ void __trace_${trace_name}(
 """
 
 def utrace_generate(cpath, hpath, ctx_param, trace_toggle_name=None,
-                    trace_toggle_defaults=[]):
+                    trace_toggle_defaults=[], additional_code=""):
     """Parameters:
 
     - cpath: c file to generate.
@@ -644,7 +646,8 @@ def utrace_generate(cpath, hpath, ctx_param, trace_toggle_name=None,
                     HEADERS=[h for h in HEADERS if h.scope & HeaderScope.HEADER],
                     FORWARD_DECLS=FORWARD_DECLS,
                     TRACEPOINTS=TRACEPOINTS,
-                    TRACEPOINTS_TOGGLES=TRACEPOINTS_TOGGLES))
+                    TRACEPOINTS_TOGGLES=TRACEPOINTS_TOGGLES,
+                    additional_code=additional_code))
             except:
                 print(exceptions.text_error_template().render())