diag.h

Go to the documentation of this file.

 
#ifndef DIAG_H
#define DIAG_H
#include "ascot5.h"
#include "particle.h"
#include "B_field.h"
#include "diag/dist_5D.h"
#include "diag/dist_6D.h"
#include "diag/dist_rho5D.h"
#include "diag/dist_rho6D.h"
#include "diag/dist_com.h"
#include "diag/diag_orb.h"
#include "diag/diag_transcoef.h"
 
typedef struct {
    int diagorb_collect;   
    int dist5D_collect;    
    int dist6D_collect;    
    int distrho5D_collect; 
    int distrho6D_collect; 
    int distCOM_collect;    
    int diagtrcof_collect; 
    diag_orb_offload_data diagorb;     
    dist_5D_offload_data dist5D;       
    dist_6D_offload_data dist6D;       
    dist_rho5D_offload_data distrho5D; 
    dist_rho6D_offload_data distrho6D; 
    dist_COM_offload_data distCOM;     
    diag_transcoef_offload_data diagtrcof; 
    size_t offload_dist5D_index;   
    size_t offload_dist6D_index;   
    size_t offload_distrho5D_index;
    size_t offload_distrho6D_index;
    size_t offload_distCOM_index;  
    size_t offload_diagorb_index;  
    size_t offload_diagtrcof_index;
    size_t offload_dist_length;    
    size_t offload_array_length;   
} diag_offload_data;
 
typedef struct {
    int diagorb_collect;   
    int dist5D_collect;    
    int dist6D_collect;    
    int distrho5D_collect; 
    int distrho6D_collect; 
    int distCOM_collect;   
    int diagtrcof_collect; 
    diag_orb_data diagorb;     
    dist_5D_data dist5D;       
    dist_6D_data dist6D;       
    dist_rho5D_data distrho5D; 
    dist_rho6D_data distrho6D; 
    dist_COM_data distCOM;     
    diag_transcoef_data diagtrcof; 
} diag_data;
 
int diag_init_offload(diag_offload_data* data, real** offload_array, int Nmrk);
 
void diag_free_offload(diag_offload_data* data, real** offload_array);
 
void diag_sum(diag_offload_data* data, real* array1, real* array2);
 
void diag_init(diag_data* data, diag_offload_data* offload_data,
               real* offload_array);
 
void diag_free(diag_data* data);
 
void diag_update_fo(diag_data* data, B_field_data* Bdata, particle_simd_fo* p_f,
                    particle_simd_fo* p_i);
 
void diag_update_gc(diag_data* data, B_field_data* Bdata, particle_simd_gc* p_f,
                    particle_simd_gc* p_i);
 
void diag_update_ml(diag_data* data, particle_simd_ml* p_f,
                    particle_simd_ml* p_i);
 
 
#endif