rate_equations.h

Go to the documentation of this file.

 
#pragma once
 
#include "parameters.h"
 
#ifdef __cplusplus
extern "C" {
#endif
 
/* SUNDIALS N_Vector */
#include <sundials/sundials_nvector.h>
#include <sundials/sundials_types.h>
 
#ifdef __cplusplus
}
#endif
 
// ── SUNDIALS RHS callback type ────────────────────────────────────────────────
using CVRhsFn_t = int(*)(sunrealtype, N_Vector, N_Vector, void*);
 
// ── User data struct passed to SUNDIALS RHS ───────────────────────────────────
 
struct UserData {
    Parameters* P;            // non-const: preconditioner writes storage
    CVRhsFn_t   rhs_fn;       // RHS function pointer (for FD probing in preconditioner)
    // Dynamic window state (used by active_window)
    int x_lo_i;    // lower active SIA index (0-based)
    int x_hi_i;    // upper active SIA index (inclusive, 0-based)
    int x_lo_v;    // lower active VAC index (0-based)
    int x_hi_v;    // upper active VAC index (inclusive, 0-based)
    bool window_active;
};
 
// ── Startup: size-only kernel tables ─────────────────────────────────────────
 
void build_kernel_tables(Parameters& P);
 
// ── RHS callbacks ─────────────────────────────────────────────────────────────
 
int rhs_full_CD(sunrealtype t, N_Vector y, N_Vector ydot, void* user_data);
 
int rhs_bin_moment(sunrealtype t, N_Vector y, N_Vector ydot, void* user_data);
 
int prec_setup(sunrealtype t, N_Vector y, N_Vector fy,
               sunbooleantype jok, sunbooleantype* jcurPtr,
               sunrealtype gamma, void* user_data);
 
int prec_solve(sunrealtype t, N_Vector y, N_Vector fy,
               N_Vector r, N_Vector z,
               sunrealtype gamma, sunrealtype delta,
               int lr, void* user_data);