regress_simple.h

Go to the documentation of this file.

 
#ifndef LIBA_REGRESS_SIMPLE_H
#define LIBA_REGRESS_SIMPLE_H
 
#include "a.h"

 
/* clang-format off */
#define A_REGRESS_SIMPLE_2(coef, bias) {a_real_c(coef), a_real_c(bias)}
/* clang-format on */
#define A_REGRESS_SIMPLE_1(coef) A_REGRESS_SIMPLE_2(coef, 0)
#define A_REGRESS_SIMPLE_0() A_REGRESS_SIMPLE_1(1)
 
typedef struct a_regress_simple a_regress_simple;
 
#if defined(__cplusplus)
extern "C" {
#endif /* __cplusplus */

A_EXTERN void a_regress_simple_init(a_regress_simple *ctx, a_real coef, a_real bias);

A_EXTERN a_real a_regress_simple_eval(a_regress_simple const *ctx, a_real val);

A_EXTERN a_real a_regress_simple_evar(a_regress_simple const *ctx, a_real val);

A_EXTERN void a_regress_simple_ols_(a_regress_simple *ctx, a_size n, a_real const *x, a_real const *y, a_real x_mean, a_real y_mean);

A_EXTERN void a_regress_simple_olsx(a_regress_simple *ctx, a_size n, a_real const *x, a_real const *y, a_real x_mean);

A_EXTERN void a_regress_simple_olsy(a_regress_simple *ctx, a_size n, a_real const *x, a_real const *y, a_real y_mean);

A_EXTERN void a_regress_simple_ols(a_regress_simple *ctx, a_size n, a_real const *x, a_real const *y);

A_EXTERN void a_regress_simple_zero(a_regress_simple *ctx);
 
#if defined(__cplusplus)
} /* extern "C" */
namespace a
{
typedef struct a_regress_simple regress_simple;
} /* namespace a */
#endif /* __cplusplus */

struct a_regress_simple
{
    a_real coef; 
    a_real bias; 
#if defined(__cplusplus)
    A_INLINE void init(a_real a = 1, a_real b = 0)
    {
        a_regress_simple_init(this, a, b);
    }
    A_INLINE a_real eval(a_real val) const
    {
        return a_regress_simple_eval(this, val);
    }
    A_INLINE a_real evar(a_real val) const
    {
        return a_regress_simple_evar(this, val);
    }
    A_INLINE void ols(a_size n, a_real const *x, a_real const *y, a_real x_mean, a_real y_mean)
    {
        a_regress_simple_ols_(this, n, x, y, x_mean, y_mean);
    }
    A_INLINE void olsx(a_size n, a_real const *x, a_real const *y, a_real x_mean)
    {
        a_regress_simple_olsx(this, n, x, y, x_mean);
    }
    A_INLINE void olsy(a_size n, a_real const *x, a_real const *y, a_real y_mean)
    {
        a_regress_simple_olsy(this, n, x, y, y_mean);
    }
    A_INLINE void ols(a_size n, a_real const *x, a_real const *y)
    {
        a_regress_simple_ols(this, n, x, y);
    }
    A_INLINE void zero() { a_regress_simple_zero(this); }
#endif /* __cplusplus */
};

 
#endif /* a/regress_simple.h */