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_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(diag_data* data, int Nmrk);
void diag_free(diag_data* data);
void diag_offload(diag_data* data);
 
void diag_sum(diag_data* data1, diag_data* data2);
 
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