hist.h

Go to the documentation of this file.

#ifndef HIST_H
#define HIST_H
 
#include <stdlib.h>
#include "../ascot5.h"
#include "../particle.h"
 
#define HIST_ALLDIM 16

typedef enum {
    R,      
    PHI,    
    Z,      
    RHO,    
    THETA,  
    PPAR,   
    PPERP,  
    PR,     
    PPHI,   
    PZ,     
    EKIN,   
    XI,     
    MU,     
    PTOR,   
    TIME,   
    CHARGE, 
} hist_coordinate;

typedef struct {
    hist_coordinate name; 
    real min;             
    real max;             
    size_t n;             
} hist_axis;

typedef struct {
    hist_axis axes[HIST_ALLDIM];   
    size_t strides[HIST_ALLDIM-1]; 
    size_t nbin;                   
    real* bins;                    
} histogram;
 
int hist_init(histogram* data, int dimensions, hist_coordinate* coordinates,
              real* binmin, real* binmax, size_t* nbin);
void hist_free(histogram* data);
void hist_offload(histogram* data);
void hist_update_fo(histogram* hist, particle_simd_fo* p_f,
                    particle_simd_fo* p_i);
void hist_update_gc(histogram* hist, particle_simd_gc* p_f,
                    particle_simd_gc* p_i);
#endif