neutral.c

Go to the documentation of this file.

 
#include <stdio.h>
#include "ascot5.h"
#include "error.h"
#include "print.h"
#include "neutral.h"
#include "neutral/N0_1D.h"
#include "neutral/N0_3D.h"
 
int neutral_init_offload(neutral_offload_data* offload_data,
                         real** offload_array) {
    int err = 0;
 
    switch(offload_data->type) {
        case neutral_type_1D:
            err = N0_1D_init_offload(&(offload_data->N01D), offload_array);
            offload_data->offload_array_length =
                offload_data->N01D.offload_array_length;
            break;
        case neutral_type_3D:
            err = N0_3D_init_offload(&(offload_data->N03D), offload_array);
            offload_data->offload_array_length =
                offload_data->N03D.offload_array_length;
            break;
        default:
            /* Unregonized input. Produce error. */
            print_err("Error: Unregonized neutral data type.");
            err = 1;
            break;
    }
    if(!err) {
        print_out(VERBOSE_IO, "Estimated memory usage %.1f MB\n",
                  offload_data->offload_array_length
                  * sizeof(real) / (1024.0*1024.0) );
    }
 
    return err;
}
 
void neutral_free_offload(neutral_offload_data* offload_data,
                          real** offload_array) {
    switch(offload_data->type) {
        case neutral_type_1D:
            N0_1D_free_offload(&(offload_data->N01D), offload_array);
            break;
        case neutral_type_3D:
            N0_3D_free_offload(&(offload_data->N03D), offload_array);
            break;
    }
}
 
int neutral_init(neutral_data* ndata, neutral_offload_data* offload_data,
                  real* offload_array) {
    int err = 0;
 
    switch(offload_data->type) {
        case neutral_type_1D:
            N0_1D_init(&(ndata->N01D),
                       &(offload_data->N01D), offload_array);
            break;
        case neutral_type_3D:
            N0_3D_init(&(ndata->N03D),
                       &(offload_data->N03D), offload_array);
            break;
        default:
            /* Unregonized input. Produce error. */
            print_err("Error: Unregonized neutral data type.\n");
            err = 1;
            break;
    }
    ndata->type = offload_data->type;
 
    return err;
}
 
a5err neutral_eval_n0(real* n0, real rho, real r, real phi, real z, real t,
                      neutral_data* ndata) {
    a5err err = 0;
 
    switch(ndata->type) {
        case neutral_type_1D:
            err = N0_1D_eval_n0(n0, rho, &(ndata->N01D));
            break;
        case neutral_type_3D:
            err = N0_3D_eval_n0(n0, r, phi, z, &(ndata->N03D));
            break;
        default:
            /* Unregonized input. Produce error. */
            err = error_raise( ERR_UNKNOWN_INPUT, __LINE__, EF_NEUTRAL);
            break;
    }
 
    if(err) {
        /* Return some reasonable values to avoid further errors */
        n0[0] = 0;
    }
 
    return err;
}
 
a5err neutral_eval_t0(real* t0, real rho, real r, real phi, real z, real t,
                      neutral_data* ndata) {
    a5err err = 0;
 
    switch(ndata->type) {
        case neutral_type_1D:
            err = N0_1D_eval_t0(t0, rho, &(ndata->N01D));
            break;
        case neutral_type_3D:
            err = N0_3D_eval_t0(t0, r, phi, z, &(ndata->N03D));
            break;
        default:
            /* Unregonized input. Produce error. */
            err = error_raise( ERR_UNKNOWN_INPUT, __LINE__, EF_NEUTRAL);
            break;
    }
 
    if(err) {
        /* Return some reasonable values to avoid further errors */
        t0[0] = 1;
    }
 
    return err;
}
 
int neutral_get_n_species(neutral_data* ndata) {
    int n = 0;
    switch(ndata->type) {
        case neutral_type_1D:
            n = N0_1D_get_n_species(&(ndata->N01D));
            break;
        case neutral_type_3D:
            n = N0_3D_get_n_species(&(ndata->N03D));
            break;
    }
 
    return n;
}