gpu_profiler.h

Go to the documentation of this file.

#ifndef GPU_PROFILER_H
#define GPU_PROFILER_H
 
#include "adaptive_sampler.h"
#include "metric_stack.h"
#include "perf_timer.h"
#include <stdint.h>
 
#ifdef TRACY_ENABLE
#include "../deps/tracy/public/tracy/TracyC.h"
#include "tracy_gpu.h"
#endif
 
enum {
    MAX_GPU_STAGES = 32,
    MAX_GPU_STAGE_NAME = 32,
    GPU_QUERY_BUFFER_COUNT = 2
};
 
/* --- Profiler Colors (Nord Theme) --- */
static const int GPU_PROFILER_TOTAL_FRAME_COLOR = 0xECEFF4;
static const int GPU_PROFILER_ENV_COLOR = 0x88C0D0;
static const int GPU_PROFILER_SCENE_COLOR = 0xD08770;
static const int GPU_PROFILER_AUTO_EXPOSURE_COLOR = 0xEBCB8B;
static const int GPU_PROFILER_BLOOM_COLOR = 0x5E81AC;
static const int GPU_PROFILER_DOF_COLOR = 0xA3BE8C;
static const int GPU_PROFILER_MOTION_BLUR_COLOR = 0xBF616A;
static const int GPU_PROFILER_COMPOSITE_COLOR = 0x81A1C1;
static const int GPU_PROFILER_POSTPROCESS_COLOR = 0xB48EAD;
static const int GPU_PROFILER_UI_COLOR = 0x4C566A;
static const int GPU_PROFILER_GI_SYNC_COLOR = 0x8FBCBB;
static const int GPU_PROFILER_GI_DEBUG_COLOR = 0xB48EAD;
static const int GPU_PROFILER_NBODY_COLOR = 0xD8DEE9;
static const int GPU_PROFILER_SHOCKWAVE_COLOR = 0xE5A3C9;
 
/* --- Capture Settings --- */
static const float GPU_PROFILER_WINDOW_DURATION_S = 0.5F;
static const float GPU_PROFILER_WINDOW_TRANSITION_S = 0.2F;
static const float GPU_PROFILER_ROW_HEIGHT = 28.0F;
 
typedef struct {
    char name[MAX_GPU_STAGE_NAME];
    uint32_t color;
    AdaptiveSampler duration_sampler; 
    AdaptiveSampler offset_sampler;   
    float start_offset_ms;            
    float duration_ms;                
    float prev_start_offset_ms; 
    float prev_duration_ms;     
    float alpha;      
    float prev_alpha; 
    int depth;        
    int parent_index; 
} GPUStage;
 
typedef struct {
    char name[MAX_GPU_STAGE_NAME];
    uint32_t color;
    int depth;
    int parent_index;
} GPUStageInfo;
 
typedef struct {
    GPUTimer queries[MAX_GPU_STAGES];
    GPUStageInfo stage_info[MAX_GPU_STAGES];
    int stage_count;
    uint64_t frame_index;
} GPUQueryBuffer;
 
typedef struct GPUProfiler {
    GPUStage stages[MAX_GPU_STAGES];
    int stage_count; 
    int recording_count; 
    // Double buffering for queries
    GPUQueryBuffer buffers[GPU_QUERY_BUFFER_COUNT];
    int write_index; 
    int read_index; 
    // State tracking for the current frame
    int current_stage_index;
    MetricStack hierarchy_stack; 
    float transition_progress; 
    bool enabled; 
} GPUProfiler;
 
void gpu_profiler_init(GPUProfiler* profiler);
 
void gpu_profiler_cleanup(GPUProfiler* profiler);
 
void gpu_profiler_begin_frame(GPUProfiler* profiler, uint64_t frame_index);
 
void gpu_profiler_set_enabled(GPUProfiler* profiler, bool enabled);
 
void gpu_profiler_start_stage(GPUProfiler* profiler, const char* name,
                              uint32_t color);
 
void gpu_profiler_end_stage(GPUProfiler* profiler);
 
void gpu_profiler_reset_samplers(GPUProfiler* profiler, double current_time);
 
typedef struct {
    GPUProfiler* profiler;
#ifdef TRACY_ENABLE
    TracyCZoneCtx tracy_ctx;
    void* tracy_gpu_ctx;
#endif
} GPUStageRAII;
 
static inline void gpu_stage_cleanup_raii(GPUStageRAII* stage_raii)
{
    gpu_profiler_end_stage(stage_raii->profiler);
#ifdef TRACY_ENABLE
    TracyCZoneEnd(stage_raii->tracy_ctx);
    tracy_gpu_zone_end(stage_raii->tracy_gpu_ctx);
#endif
}
 
#ifdef TRACY_ENABLE
#define GPU_STAGE_PROFILER(profiler_ptr, name, color)                        \
    TracyCZoneN(_tracy_ctx##__LINE__, name, 1);                          \
    void* _tracy_gpu_ctx##__LINE__ =                                     \
        tracy_gpu_zone_begin(name, __func__, __FILE__, __LINE__, color); \
    GPUStageRAII _stage_raii##__LINE__                                   \
        __attribute__((cleanup(gpu_stage_cleanup_raii)))                 \
        __attribute__((unused)) = {profiler_ptr, _tracy_ctx##__LINE__,   \
                                   _tracy_gpu_ctx##__LINE__};            \
    gpu_profiler_start_stage(profiler_ptr, name, color)
#else
#define GPU_STAGE_PROFILER(profiler_ptr, name, color)        \
    GPUStageRAII _stage_raii##__LINE__                   \
        __attribute__((cleanup(gpu_stage_cleanup_raii))) \
        __attribute__((unused)) = {profiler_ptr};        \
    gpu_profiler_start_stage(profiler_ptr, name, color)
#endif
 
#endif  // GPU_PROFILER_H