ode.h

Go to the documentation of this file.

#ifndef G_U239L9IU4I9YYJGH4M9CD6ADMGT66
#define G_U239L9IU4I9YYJGH4M9CD6ADMGT66

#include <stdbool.h>
#include <math.h>
#include "vector.h"
#include "extern.h"
#include "restrict_begin.h"
#ifdef __cplusplus
extern "C" {
#endif

#define SG_ODE_FSTRICT 0x1

#define SG_ODE_FMAXNUM 0x2

#define SG_ODE_ETOLER 3

#define SG_ODE_ESTEPS 4

#define SG_ODE_ESTIFF 5

#define SG_ODE_EINVAL 6

enum {
    SG_ODE_TYPE_BOOL,
    SG_ODE_TYPE_DOUBLE,
    SG_ODE_TYPE_UNSIGNED,
    SG_ODE_TYPE_VECTOR,
};

typedef void SgDerivFn(void *f_ctx,
                       double t,
                       const SgVector *restrict y,
                       SgVector *restrict yp);

SG_EXTERN struct SgOde *sg_ode_try_new(struct SgVectorDriver drv);

SG_EXTERN void sg_ode_del(struct SgOde *self);

SG_EXTERN int sg_ode_traverse(struct SgOde *self,
                              int (*f)(void *ctx,
                                       void *data,
                                       int type,
                                       size_t len),
                              void *ctx);

SG_EXTERN int sg_ode_step(struct SgOde *self,
                          SgDerivFn *f,
                          void *restrict f_ctx,
                          double *restrict eps);

SG_EXTERN void sg_ode_interpolate(struct SgVectorDriver drv,
                                  double x,
                                  const SgVector *restrict y,
                                  double xout,
                                  SgVector *restrict yout,
                                  SgVector *restrict ypout,
                                  unsigned kold,
                                  SgVector *const restrict *phi,
                                  const double *restrict psi);

SG_EXTERN void sg_ode_de(struct SgOde *self,
                         SgDerivFn *f,
                         void *restrict f_ctx,
                         SgVector *restrict y,
                         double *restrict t,
                         double tout,
                         double *restrict relerr,
                         double *restrict abserr,
                         int *restrict iflag,
                         unsigned maxnum);

SG_EXTERN int sg_ode(void *f_ctx,
                     void (*f)(void *, double, const double *, double *),
                     size_t neqn,
                     double *restrict y,
                     double *restrict t,
                     double tout,
                     double relerr,
                     double abserr,
                     int flag,
                     double *restrict work,
                     int *restrict iwork);

#ifdef __cplusplus
}
#endif
#include "restrict_end.h"
#endif