B_STS.h

Go to the documentation of this file.

 
#ifndef B_STS_H
#define B_STS_H
#include "../offload_acc_omp.h"
#include "../ascot5.h"
#include "../linint/linint.h"
#include "../spline/interp.h"
 
typedef struct {
    int psigrid_n_r;       
    int psigrid_n_z;       
    int psigrid_n_phi;     
    real psigrid_r_min;    
    real psigrid_r_max;    
    real psigrid_z_min;    
    real psigrid_z_max;    
    real psigrid_phi_min;  
    real psigrid_phi_max;  
    int Bgrid_n_r;       
    int Bgrid_n_z;       
    int Bgrid_n_phi;     
    real Bgrid_r_min;    
    real Bgrid_r_max;    
    real Bgrid_z_min;    
    real Bgrid_z_max;    
    real Bgrid_phi_min;  
    real Bgrid_phi_max;  
    real psi0;           
    real psi1;           
    int offload_array_length; 
    int n_axis;          
    real axis_min;       
    real axis_max;       
    real axis_grid;      
} B_STS_offload_data;
 
typedef struct {
    real psi0;           
    real psi1;           
    linint1D_data axis_r;
    linint1D_data axis_z;
    interp3D_data psi;   
    interp3D_data B_r;   
    interp3D_data B_phi ;
    interp3D_data B_z;   
} B_STS_data;
 
int B_STS_init_offload(B_STS_offload_data* offload_data, real** offload_array);
void B_STS_free_offload(B_STS_offload_data* offload_data, real** offload_array);
 
void B_STS_init(B_STS_data* Bdata, B_STS_offload_data* offload_data,
               real* offload_array);
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_STS_eval_psi(real* psi, real r, real phi, real z, B_STS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_STS_eval_psi_dpsi(real psi_dpsi[4], real r, real phi, real z,
                          B_STS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_STS_eval_rho_drho(real rho_drho[4], real r, real phi, real z,
                          B_STS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_STS_eval_B(real B[3], real r, real phi, real z, B_STS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_STS_eval_B_dB(real B_dB[12], real r, real phi, real z,
                      B_STS_data* Bdata);
DECLARE_TARGET_END
GPU_DECLARE_TARGET_SIMD_UNIFORM(Bdata)
a5err B_STS_get_axis_rz(real rz[2], B_STS_data* Bdata, real phi);
DECLARE_TARGET_END
#endif