1 files changed, 19 insertions, 1 deletions
diff --git a/src/gallium/winsys/xlib/xm_winsys.c b/src/gallium/winsys/xlib/xm_winsys.c
index 9225ee510d..6071a5ad5e 100644
--- a/src/gallium/winsys/xlib/xm_winsys.c
+++ b/src/gallium/winsys/xlib/xm_winsys.c
@@ -54,6 +54,12 @@
 #define TILE_SIZE 32  /* avoid compilation errors */
 #endif
 
+#ifdef GALLIUM_TRACE
+#include "trace/tr_winsys.h"
+#include "trace/tr_screen.h"
+#include "trace/tr_context.h"
+#endif
+
 #include "xm_winsys_aub.h"
 
 
@@ -645,7 +651,11 @@ xmesa_get_pipe_winsys(struct xmesa_visual *xm_vis)
       ws->base.get_name = xm_get_name;
    }
 
-   return &ws->base;
+#ifdef GALLIUM_TRACE
+      return trace_winsys_create(&ws->base);
+#else
+      return &ws->base;
+#endif
 }
 
 
@@ -674,7 +684,15 @@ xmesa_create_pipe_context(XMesaContext xmesa, uint pixelformat)
    {
       struct pipe_screen *screen = softpipe_create_screen(pws);
 
+#ifdef GALLIUM_TRACE
+      screen = trace_screen_create(screen);
+#endif
+      
       pipe = softpipe_create(screen, pws, NULL);
+
+#ifdef GALLIUM_TRACE
+      pipe = trace_context_create(pipe);
+#endif
    }
 
    if (pipe)