diag.c

Go to the documentation of this file.

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include "ascot5.h"
#include "simulate.h"
#include "B_field.h"
#include "diag.h"
#include "diag/diag_orb.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_transcoef.h"
#include "particle.h"
 
void diag_arraysum(size_t start, size_t stop, real* array1, real* array2);

int diag_init(diag_data* data, int Nmrk) {
 
    if(data->diagorb_collect) {
        data->diagorb.Nmrk = Nmrk;
 
        switch(data->diagorb.record_mode) {
 
            case simulate_mode_fo:
                data->diagorb.Nfld = DIAG_ORB_FOFIELDS;
                break;
 
            case simulate_mode_gc:
                data->diagorb.Nfld = DIAG_ORB_GCFIELDS;
                break;
 
            case simulate_mode_ml:
                data->diagorb.Nfld = DIAG_ORB_MLFIELDS;
                break;
 
            case simulate_mode_hybrid:
                data->diagorb.Nfld = DIAG_ORB_HYBRIDFIELDS;
                break;
 
        }
    }
    if(data->diagtrcof_collect) {
        data->diagtrcof.Nmrk = Nmrk;
    }
 
    if(data->dist5D_collect) {
        dist_5D_init(&data->dist5D);
    }
 
    if(data->dist6D_collect) {
        dist_6D_init(&data->dist6D);
    }
 
    if(data->distrho5D_collect) {
        dist_rho5D_init(&data->distrho5D);
    }
 
    if(data->distrho6D_collect) {
        dist_rho6D_init(&data->distrho6D);
    }
 
    if(data->distCOM_collect) {
        dist_COM_init(&data->distCOM);
    }
 
    if(data->diagorb_collect) {
        diag_orb_init(&data->diagorb);
    }
 
    if(data->diagtrcof_collect) {
        diag_transcoef_init(&data->diagtrcof);
    }
 
    return 0;
}

void diag_free(diag_data* data) {
    if(data->dist5D_collect) {
        dist_5D_free(&data->dist5D);
    }
    if(data->dist6D_collect) {
        dist_6D_free(&data->dist6D);
    }
    if(data->distrho5D_collect) {
        dist_rho5D_free(&data->distrho5D);
    }
    if(data->distrho6D_collect) {
        dist_rho6D_free(&data->distrho6D);
    }
    if(data->distCOM_collect) {
        dist_COM_free(&data->distCOM);
    }
    if(data->diagorb_collect) {
        diag_orb_free(&data->diagorb);
    }
    if(data->diagtrcof_collect) {
        diag_transcoef_free(&data->diagtrcof);
    }
}

void diag_offload(diag_data* data) {
    if(data->dist5D_collect) {
        dist_5D_offload(&data->dist5D);
    }
    if(data->dist6D_collect) {
        dist_6D_offload(&data->dist6D);
    }
    if(data->distrho5D_collect) {
        dist_rho5D_offload(&data->distrho5D);
    }
    if(data->distrho6D_collect) {
        dist_rho6D_offload(&data->distrho6D);
    }
    if(data->distCOM_collect) {
        dist_COM_offload(&data->distCOM);
    }
}

void diag_update_fo(diag_data* data, B_field_data* Bdata, particle_simd_fo* p_f,
                    particle_simd_fo* p_i) {
    if(data->diagorb_collect) {
        diag_orb_update_fo(&data->diagorb, p_f, p_i);
    }
    if(data->dist5D_collect) {
        dist_5D_update_fo(&data->dist5D, p_f, p_i);
    }
    if(data->dist6D_collect) {
        dist_6D_update_fo(&data->dist6D, p_f, p_i);
    }
    if(data->distrho5D_collect) {
        dist_rho5D_update_fo(&data->distrho5D, p_f, p_i);
    }
    if(data->distrho6D_collect) {
        dist_rho6D_update_fo(&data->distrho6D, p_f, p_i);
    }
    if(data->distCOM_collect){
        dist_COM_update_fo(&data->distCOM, Bdata, p_f, p_i);
    }
    if(data->diagtrcof_collect){
        diag_transcoef_update_fo(&data->diagtrcof, p_f, p_i);
    }
}

void diag_update_gc(diag_data* data, B_field_data* Bdata, particle_simd_gc* p_f,
                    particle_simd_gc* p_i) {
    if(data->diagorb_collect) {
        diag_orb_update_gc(&data->diagorb, p_f, p_i);
    }
 
    if(data->dist5D_collect){
        dist_5D_update_gc(&data->dist5D, p_f, p_i);
    }
 
    if(data->dist6D_collect){
        dist_6D_update_gc(&data->dist6D, p_f, p_i);
    }
 
    if(data->distrho5D_collect){
        dist_rho5D_update_gc(&data->distrho5D, p_f, p_i);
    }
 
    if(data->distrho6D_collect){
        dist_rho6D_update_gc(&data->distrho6D, p_f, p_i);
    }
 
    if(data->distCOM_collect){
        dist_COM_update_gc(&data->distCOM, Bdata, p_f, p_i);
    }
 
    if(data->diagtrcof_collect){
        diag_transcoef_update_gc(&data->diagtrcof, p_f, p_i);
    }
}

void diag_update_ml(diag_data* data, particle_simd_ml* p_f,
                    particle_simd_ml* p_i) {
    if(data->diagorb_collect) {
        diag_orb_update_ml(&data->diagorb, p_f, p_i);
    }
 
    if(data->diagtrcof_collect){
        diag_transcoef_update_ml(&data->diagtrcof, p_f, p_i);
    }
}

void diag_sum(diag_data* data1, diag_data* data2) {
    if(data1->diagorb_collect) {
    }
 
    if(data1->diagtrcof_collect) {
    }
 
    if(data1->dist5D_collect){
        size_t stop = (size_t)(data1->dist5D.n_r) * (size_t)(data1->dist5D.n_z)
            * (size_t)(data1->dist5D.n_ppara) * (size_t)(data1->dist5D.n_pperp)
            * (size_t)(data1->dist5D.n_time) * (size_t)(data1->dist5D.n_q);
        diag_arraysum(0, stop, data1->dist5D.histogram,
                      data2->dist5D.histogram);
    }
 
    if(data1->dist6D_collect){
        size_t stop = (size_t)(data1->dist6D.n_r)
            * (size_t)(data1->dist6D.n_phi)
            * (size_t)(data1->dist6D.n_z) * (size_t)(data1->dist6D.n_pr)
            * (size_t)(data1->dist6D.n_pphi) * (size_t)(data1->dist6D.n_pz)
            * (size_t)(data1->dist6D.n_time) * (size_t)(data1->dist6D.n_q);
        diag_arraysum(0, stop, data1->dist6D.histogram,
                      data2->dist6D.histogram);
    }
 
    if(data1->distrho5D_collect){
        size_t stop = (size_t)(data1->distrho5D.n_rho)
            * (size_t)(data1->distrho5D.n_theta)
            * (size_t)(data1->distrho5D.n_phi)
            * (size_t)(data1->distrho5D.n_ppara)
            * (size_t)(data1->distrho5D.n_pperp)
            * (size_t)(data1->distrho5D.n_time)
            * (size_t)(data1->distrho5D.n_q);
        diag_arraysum(0, stop, data1->distrho5D.histogram,
                      data2->distrho5D.histogram);
    }
 
    if(data1->distrho6D_collect){
        size_t stop = (size_t)(data1->distrho6D.n_rho)
            * (size_t)(data1->distrho6D.n_theta)
            * (size_t)(data1->distrho6D.n_phi)
            * (size_t)(data1->distrho6D.n_pr)
            * (size_t)(data1->distrho6D.n_pphi)
            * (size_t)(data1->distrho6D.n_pz)
            * (size_t)(data1->distrho6D.n_time)
            * (size_t)(data1->distrho6D.n_q);
        diag_arraysum(0, stop, data1->distrho6D.histogram,
                      data2->distrho6D.histogram);
    }
 
    if(data1->distCOM_collect){
        size_t stop = (size_t)(data1->distCOM.n_mu)
            * (size_t)(data1->distCOM.n_Ekin) * (size_t)(data1->distCOM.n_Ptor);
        diag_arraysum(0, stop, data1->dist6D.histogram,
                      data2->dist6D.histogram);
    }
}

void diag_arraysum(size_t start, size_t stop, real* array1, real* array2) {
    for(size_t i = start; i < stop; i++) {
        array1[i] += array2[i];
    }
}