Files
fn_registry/cpp/functions/datascience/samples_to_grid_2d.h
T
egutierrez d76c831247 feat(cpp/datascience): CPU stats + MCMC primitives
Nuevo dominio cpp/functions/datascience con primitivas puras CPU para post-
proceso de samples Monte Carlo y diagnostico de cadenas MCMC. Diseñadas como
gemelas CPU de los kernels GPU (rng pareja con gpu_rng_glsl, MH 1D/ND con
mc_metropolis_hastings_gpu) para validar numericamente y para datasets
pequeños donde el dispatch GPU no compensa.

- rng: xoshiro256++ con uniform / normal (Box-Muller) / below (Lemire) /
  categorical. Determinista bit-exacto dado seed.
- stats_summary: sum (Kahan), mean, var/std (Welford one-pass), min, max,
  quantile / percentile (R type-7).
- autocorr: r(k), ACF, tau_int (Sokal) — diagnostico ACF y ESS.
- rhat_ess: Gelman-Rubin clasico y split + ESS basico (multi-chain).
- beta_dist: lgamma (Lanczos), beta_pdf, beta_cdf (continued fraction),
  beta_quantile, mean/var/std — para inferencia Beta-Binomial.
- drawdown: max_dd absoluto/pct + underwater series para sesiones
  simuladas y backtests.
- samples_to_grid_2d: binning 2D CPU para alimentar heatmap_cpp_viz /
  contour_cpp_viz desde samples (x[], y[]).
- metropolis_hastings: MH 1D y ND con target log-pdf como std::function
  (no normalizada).

Co-Authored-By: Claude Opus 4.7 (1M context) <noreply@anthropic.com>
2026-05-04 11:52:26 +02:00

38 lines
1.6 KiB
C++

#pragma once
#include <cstddef>
namespace fn::ds {
// Binning 2D puro CPU. samples_x[i], samples_y[i] son los pares (x, y) a
// binnear. nx*ny celdas en row-major (idx = row * nx + col), donde row
// crece con y. Samples fuera del rango se descartan.
//
// out_counts[nx*ny] se llena con conteos uint32. (No se hace clear interno —
// el caller decide si acumular sobre llamadas previas o no.)
void samples_to_grid_2d_counts(const double* samples_x,
const double* samples_y,
std::size_t n,
double xmin, double xmax,
double ymin, double ymax,
int nx, int ny,
unsigned int* out_counts);
// Variante que normaliza los counts a float[0,1] sobre el max (densidad
// relativa). Adecuado para alimentar heatmap_cpp_viz / contour_cpp_viz / surface_plot_3d.
// Si todos los counts son 0 se llena out_density con 0.
void samples_to_grid_2d_density(const double* samples_x,
const double* samples_y,
std::size_t n,
double xmin, double xmax,
double ymin, double ymax,
int nx, int ny,
float* out_density);
// Helper: convierte counts uint -> density float in-place sobre out.
// max_count se calcula y se normaliza out[i] = counts[i]/max_count.
void counts_to_density(const unsigned int* counts, int nx, int ny,
float* out_density);
} // namespace fn::ds