raysense 0.6.2

/*
 *   Copyright (c) 2025-2026 Anton Kundenko <singaraiona@gmail.com>
 *   All rights reserved.

 *   Permission is hereby granted, free of charge, to any person obtaining a copy
 *   of this software and associated documentation files (the "Software"), to deal
 *   in the Software without restriction, including without limitation the rights
 *   to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 *   copies of the Software, and to permit persons to whom the Software is
 *   furnished to do so, subject to the following conditions:

 *   The above copyright notice and this permission notice shall be included in all
 *   copies or substantial portions of the Software.

 *   THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 *   IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 *   FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 *   AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 *   LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 *   OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 *   SOFTWARE.
 */

#ifndef RAY_PROFILE_H
#define RAY_PROFILE_H

#include <stdint.h>
#include <stdbool.h>

#if defined(RAY_OS_WINDOWS)
#include <windows.h>
#else
#include <time.h>
/* clock_gettime / CLOCK_MONOTONIC may be hidden under strict -std=c17
 * without _POSIX_C_SOURCE.  Provide fallback declarations. */
#ifndef CLOCK_MONOTONIC
#define CLOCK_MONOTONIC 1
int clock_gettime(int clk_id, struct timespec *tp);
#endif
#endif

/* ===== Span-based execution profiler =====
 *
 * Zero overhead when inactive — every call guards on g_ray_profile.active.
 * Activated by REPL :t command; lives entirely outside hot morsel loops.
 */

#define RAY_PROFILE_SPANS_MAX 2048

typedef enum {
    RAY_PROF_SPAN_START,
    RAY_PROF_SPAN_END,
    RAY_PROF_SPAN_TICK
} ray_prof_span_type_t;

typedef struct {
    ray_prof_span_type_t type;
    const char*          msg;
    int64_t              ts;   /* nanoseconds (monotonic) */
} ray_prof_span_t;

/* Progress callback — set by REPL to render progress bar.
 * Called at morsel boundaries; receives done/total/label. */
typedef void (*ray_progress_fn)(int64_t done, int64_t total, const char* label);

typedef struct {
    bool              active;
    int32_t           n;
    /* Progress tracking */
    int64_t           progress_total;
    int64_t           progress_done;
    const char*       progress_label;
    int64_t           progress_last_render; /* ns timestamp of last render */
    ray_progress_fn   progress_cb;          /* set by REPL; NULL = no-op  */
    ray_prof_span_t   spans[RAY_PROFILE_SPANS_MAX];
} ray_profile_t;

/* Single global instance */
extern ray_profile_t g_ray_profile;

static inline int64_t ray_profile_now_ns(void) {
#if defined(RAY_OS_WINDOWS)
    LARGE_INTEGER freq, cnt;
    QueryPerformanceFrequency(&freq);
    QueryPerformanceCounter(&cnt);
    return (int64_t)((double)cnt.QuadPart / (double)freq.QuadPart * 1e9);
#else
    struct timespec ts;
    clock_gettime(CLOCK_MONOTONIC, &ts);
    return (int64_t)ts.tv_sec * 1000000000LL + (int64_t)ts.tv_nsec;
#endif
}

static inline void ray_profile_reset(void) {
    g_ray_profile.n = 0;
    g_ray_profile.progress_total = 0;
    g_ray_profile.progress_done = 0;
    g_ray_profile.progress_label = NULL;
    g_ray_profile.progress_last_render = 0;
}

static inline void ray_profile_span_start(const char* name) {
    if (!g_ray_profile.active) return;
    if (g_ray_profile.n >= RAY_PROFILE_SPANS_MAX) return;
    ray_prof_span_t* s = &g_ray_profile.spans[g_ray_profile.n++];
    s->type = RAY_PROF_SPAN_START;
    s->msg  = name;
    s->ts   = ray_profile_now_ns();
}

static inline void ray_profile_span_end(const char* name) {
    if (!g_ray_profile.active) return;
    if (g_ray_profile.n >= RAY_PROFILE_SPANS_MAX) return;
    ray_prof_span_t* s = &g_ray_profile.spans[g_ray_profile.n++];
    s->type = RAY_PROF_SPAN_END;
    s->msg  = name;
    s->ts   = ray_profile_now_ns();
}

static inline void ray_profile_tick(const char* msg) {
    if (!g_ray_profile.active) return;
    if (g_ray_profile.n >= RAY_PROFILE_SPANS_MAX) return;
    ray_prof_span_t* s = &g_ray_profile.spans[g_ray_profile.n++];
    s->type = RAY_PROF_SPAN_TICK;
    s->msg  = msg;
    s->ts   = ray_profile_now_ns();
}

/* Progress bar — called between morsels / pipeline stages */
static inline void ray_profile_progress_begin(const char* label, int64_t total) {
    if (!g_ray_profile.active) return;
    g_ray_profile.progress_label = label;
    g_ray_profile.progress_total = total;
    g_ray_profile.progress_done  = 0;
}

/* Throttled: renders at most every 100ms to avoid terminal spam */
#define RAY_PROGRESS_RENDER_INTERVAL_NS (100 * 1000000LL)

static inline void ray_profile_progress_advance(int64_t delta) {
    if (!g_ray_profile.active) return;
    g_ray_profile.progress_done += delta;
    if (g_ray_profile.progress_cb && g_ray_profile.progress_total > 0) {
        int64_t now = ray_profile_now_ns();
        if (now - g_ray_profile.progress_last_render > RAY_PROGRESS_RENDER_INTERVAL_NS) {
            g_ray_profile.progress_last_render = now;
            g_ray_profile.progress_cb(g_ray_profile.progress_done,
                                      g_ray_profile.progress_total,
                                      g_ray_profile.progress_label);
        }
    }
}

static inline void ray_profile_progress_end(void) {
    if (!g_ray_profile.active) return;
    g_ray_profile.progress_label = NULL;
    g_ray_profile.progress_total = 0;
    g_ray_profile.progress_done  = 0;
}

#endif /* RAY_PROFILE_H */