dist_com.h

Go to the documentation of this file.

 
#ifndef DIST_COM_H
#define DIST_COM_H
 
#include <stdlib.h>
#include "../ascot5.h"
#include "../particle.h"
#include "../B_field.h"
 
typedef struct {
    int n_mu;            
    real min_mu;         
    real max_mu;         
    int n_Ekin;          
    real min_Ekin;       
    real max_Ekin;       
    int n_Ptor;          
    real min_Ptor;       
    real max_Ptor;       
} dist_COM_offload_data;
 
typedef struct {
    int n_mu;            
    real min_mu;         
    real max_mu;         
    int n_Ekin;          
    real min_Ekin;       
    real max_Ekin;       
    int n_Ptor;          
    real min_Ptor;       
    real max_Ptor;       
    size_t step_1;       
    size_t step_2;       
    real* histogram;  
} dist_COM_data;
 
void dist_COM_init(dist_COM_data* dist_data,
                   dist_COM_offload_data* offload_data,
                   real* offload_array);
void dist_COM_update_fo(dist_COM_data* dist, B_field_data*Bdata,
                        particle_simd_fo* p_f, particle_simd_fo* p_i);
void dist_COM_update_gc(dist_COM_data* dist, B_field_data* Bdata,
                        particle_simd_gc* p_f, particle_simd_gc* p_i);
 
#endif