dist_5D.h

Go to the documentation of this file.

 
#ifndef DIST_5D_H
#define DIST_5D_H
 
#include <stdlib.h>
#include "../ascot5.h"
#include "../particle.h"
 
typedef struct {
    int n_r;          
    real min_r;       
    real max_r;       
    int n_phi;        
    real min_phi;     
    real max_phi;     
    int n_z;          
    real min_z;       
    real max_z;       
    int n_ppara;      
    real min_ppara;   
    real max_ppara;   
    int n_pperp;      
    real min_pperp;   
    real max_pperp;   
    int n_time;       
    real min_time;    
    real max_time;    
    int n_q;          
    real min_q;       
    real max_q;       
} dist_5D_offload_data;
 
typedef struct {
    int n_r;          
    real min_r;       
    real max_r;       
    int n_phi;        
    real min_phi;     
    real max_phi;     
    int n_z;          
    real min_z;       
    real max_z;       
    int n_ppara;      
    real min_ppara;   
    real max_ppara;   
    int n_pperp;      
    real min_pperp;   
    real max_pperp;   
    int n_time;       
    real min_time;    
    real max_time;    
    int n_q;          
    real min_q;       
    real max_q;       
    size_t step_1;    
    size_t step_2;    
    size_t step_3;    
    size_t step_4;    
    size_t step_5;    
    size_t step_6;    
    real* histogram;  
} dist_5D_data;
 
size_t dist_5D_index(int i_r, int i_phi, int i_z, int i_ppara, int i_pperp,
                     int i_time, int i_q, size_t step_6, size_t step_5,
                     size_t step_4, size_t step_3, size_t step_2,
                     size_t step_1);
void dist_5D_init(dist_5D_data* dist_data,
                  dist_5D_offload_data* offload_data,
                  real* offload_array);
void dist_5D_update_fo(dist_5D_data* dist, particle_simd_fo* p_f,
                       particle_simd_fo* p_i);
void dist_5D_update_gc(dist_5D_data* dist, particle_simd_gc* p_f,
                       particle_simd_gc* p_i);
 
#endif