interp.h

Go to the documentation of this file.

 
#ifndef INTERP_H
#define INTERP_H
#include "../offload_acc_omp.h"
#include "../ascot5.h"
#include "../error.h"
 
enum boundaryCondition {
    NATURALBC  = 0, 
    PERIODICBC = 1  
};
 
enum splinesize {
    NSIZE_COMP1D =  2,
    NSIZE_COMP2D =  4,
    NSIZE_COMP3D =  8,
    NSIZE_EXPL1D =  4,
    NSIZE_EXPL2D = 16,
    NSIZE_EXPL3D = 64
};
 
typedef struct {
    int n_x;     
    int bc_x;    
    real x_min;  
    real x_max;  
    real x_grid; 
    real* c;     
} interp1D_data;
 
typedef struct {
    int n_x;     
    int n_y;     
    int bc_x;    
    int bc_y;    
    real x_min;  
    real x_max;  
    real x_grid; 
    real y_min;  
    real y_max;  
    real y_grid; 
    real* c;     
} interp2D_data;
 
typedef struct {
    int n_x;     
    int n_y;     
    int n_z;     
    int bc_x;    
    int bc_y;    
    int bc_z;    
    real x_min;  
    real x_max;  
    real x_grid; 
    real y_min;  
    real y_max;  
    real y_grid; 
    real z_min;  
    real z_max;  
    real z_grid; 
    real* c;     
} interp3D_data;
 
int interp1Dcomp_init_coeff(real* c, real* f,
                            int n_x, int bc_x,
                            real x_min, real x_max);
 
int interp2Dcomp_init_coeff(real* c, real* f,
                            int n_x, int n_y, int bc_x, int bc_y,
                            real x_min, real x_max,
                            real y_min, real y_max);
 
int interp3Dcomp_init_coeff(real* c, real* f,
                            int n_x, int n_y, int n_z,
                            int bc_x, int bc_y, int bc_z,
                            real x_min, real x_max,
                            real y_min, real y_max,
                            real z_min, real z_max);
 
int interp1Dexpl_init_coeff(real* c, real* f,
                            int n_x, int bc_x,
                            real x_min, real x_max);
 
int interp2Dexpl_init_coeff(real* c, real* f,
                            int n_x, int n_y, int bc_x, int bc_y,
                            real x_min, real x_max,
                            real y_min, real y_max);
 
int interp3Dexpl_init_coeff(real* c, real* f,
                            int n_x, int n_y, int n_z,
                            int bc_x, int bc_y, int bc_z,
                            real x_min, real x_max,
                            real y_min, real y_max,
                            real z_min, real z_max);
 
void interp1Dcomp_init_spline(interp1D_data* str, real* c,
                              int n_x, int bc_x,
                              real x_min, real x_max);
 
void interp2Dcomp_init_spline(interp2D_data* str, real* c,
                              int n_x, int n_y, int bc_x, int bc_y,
                              real x_min, real x_max,
                              real y_min, real y_max);
 
void interp3Dcomp_init_spline(interp3D_data* str, real* c,
                              int n_x, int n_y, int n_z,
                              int bc_x, int bc_y, int bc_z,
                              real x_min, real x_max,
                              real y_min, real y_max,
                              real z_min, real z_max);
 
void interp1Dexpl_init_spline(interp1D_data* str, real* c,
                              int n_x, int bc_x,
                              real x_min, real x_max);
 
void interp2Dexpl_init_spline(interp2D_data* str, real* c,
                              int n_x, int n_y, int bc_x, int bc_y,
                              real x_min, real x_max,
                              real y_min, real y_max);
 
void interp3Dexpl_init_spline(interp3D_data* str, real* c,
                              int n_x, int n_y, int n_z,
                              int bc_x, int bc_y, int bc_z,
                              real x_min, real x_max,
                              real y_min, real y_max,
                              real z_min, real z_max);
 
GPU_DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp1Dcomp_eval_f(real* f, interp1D_data* str, real x);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp2Dcomp_eval_f(real* f, interp2D_data* str, real x, real y);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp3Dcomp_eval_f(real* f, interp3D_data* str,
                         real x, real y, real z);
DECLARE_TARGET_END
 
DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp1Dexpl_eval_f(real* f, interp1D_data* str, real x);
DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp2Dexpl_eval_f(real* f, interp2D_data* str, real x, real y);
DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp3Dexpl_eval_f(real* f, interp3D_data* str,
                          real x, real y, real z);
 
GPU_DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp1Dcomp_eval_df(real* f_df, interp1D_data* str, real x);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp2Dcomp_eval_df(real* f_df, interp2D_data* str, real x, real y);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp3Dcomp_eval_df(real* f_df, interp3D_data* str,
                           real x, real y, real z);
DECLARE_TARGET_END
 
DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp1Dexpl_eval_df(real* f_df, interp1D_data* str, real x);
DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp2Dexpl_eval_df(real* f_df, interp2D_data* str, real x, real y);
DECLARE_TARGET_SIMD_UNIFORM(str)
a5err interp3Dexpl_eval_df(real* f_df, interp3D_data* str,
                           real x, real y, real z);
#endif